diff --git a/skills/kb-deep-analyzer/deep_analyzer.py b/skills/kb-deep-analyzer/deep_analyzer.py index 8697eb6..7f51062 100644 --- a/skills/kb-deep-analyzer/deep_analyzer.py +++ b/skills/kb-deep-analyzer/deep_analyzer.py @@ -53,11 +53,11 @@ class SimpleAIClient: SHORT_VIDEO_THRESHOLD = 600 # 10分钟 LONG_VIDEO_THRESHOLD = 1800 # 30分钟 - def __init__(self, api_key: str = None, provider: str = "mock", model: str = "gpt-4"): + def __init__(self, api_key: str = None, provider: str = "mock", model: str = "gpt-4", base_url: str = ""): self.api_key = api_key or os.environ.get("AI_API_KEY", "") self.provider = provider self.model = model - self.base_url = os.environ.get("AI_BASE_URL", "") + self.base_url = base_url or os.environ.get("AI_BASE_URL", "") def generate(self, prompt: str, max_tokens: int = 1000) -> str: """生成文本""" @@ -252,8 +252,8 @@ class DeepAnalyzer: """有字幕的长视频详细分析""" result = AnalysisResult() - if duration >= self.LONG_VIDEO_THRESHOLD and content_type == "video": - # 使用分段分析 + if duration >= self.SHORT_VIDEO_THRESHOLD and content_type == "video": + # 10分钟以上使用分段分析 result = self._analyze_in_chunks(title, subtitle, duration) else: # 直接分析