f3daa5cf97
Fixes #128 Two bugs: 1. doctor warn message said 'agent-reach configure groq-api-key' but the actual CLI arg is 'groq-key' (see cli.py:75 choices list and :993 handler). Same typo existed in cli.py _install_xiaoyuzhou_deps() output. 2. check() had a logic flaw: the inner 'if not has_key' was nested inside 'if not os.environ.get(...)', so when GROQ_API_KEY env var was absent but config.json read succeeded, the outer condition never triggered the warning. Refactored to a flat has_key variable — cleaner and correct. Verified: 36/36 tests pass, manual unit tests confirm warn uses 'groq-key' and ok is returned when groq_api_key is present in config.json. Co-authored-by: 小白(Agent) <neo@agent-reach.dev>
58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""Xiaoyuzhou Podcast (小宇宙播客) — transcribe podcasts via Groq Whisper API."""
|
|
|
|
import os
|
|
import shutil
|
|
from .base import Channel
|
|
|
|
|
|
class XiaoyuzhouChannel(Channel):
|
|
name = "xiaoyuzhou"
|
|
description = "小宇宙播客转文字"
|
|
backends = ["groq-whisper", "ffmpeg"]
|
|
tier = 1
|
|
|
|
def can_handle(self, url: str) -> bool:
|
|
from urllib.parse import urlparse
|
|
d = urlparse(url).netloc.lower()
|
|
return "xiaoyuzhoufm.com" in d
|
|
|
|
def check(self, config=None):
|
|
# Check ffmpeg
|
|
if not shutil.which("ffmpeg"):
|
|
return "off", (
|
|
"需要 ffmpeg(音频转码和切片)。安装:\n"
|
|
" Ubuntu/Debian: apt install -y ffmpeg\n"
|
|
" macOS: brew install ffmpeg"
|
|
)
|
|
|
|
# Check script exists
|
|
script = os.path.expanduser("~/.agent-reach/tools/xiaoyuzhou/transcribe.sh")
|
|
if not os.path.isfile(script):
|
|
return "off", (
|
|
"转录脚本未安装。运行:\n"
|
|
" agent-reach install --env=auto\n"
|
|
" 或手动复制 transcribe.sh 到 ~/.agent-reach/tools/xiaoyuzhou/"
|
|
)
|
|
|
|
# Check GROQ_API_KEY — prefer env var, fall back to config file
|
|
has_key = bool(os.environ.get("GROQ_API_KEY"))
|
|
if not has_key:
|
|
config_path = os.path.expanduser("~/.agent-reach/config.json")
|
|
if os.path.isfile(config_path):
|
|
try:
|
|
import json
|
|
with open(config_path) as f:
|
|
cfg = json.load(f)
|
|
has_key = bool(cfg.get("groq_api_key"))
|
|
except Exception:
|
|
pass
|
|
if not has_key:
|
|
return "warn", (
|
|
"需要配置 Groq API Key(免费)。步骤:\n"
|
|
" 1. 注册 https://console.groq.com\n"
|
|
" 2. 运行: agent-reach configure groq-key gsk_xxxxx"
|
|
)
|
|
|
|
return "ok", "完整可用(播客下载 + Whisper 转录)"
|