diff --git a/skills/kb-deep-analyzer/deep_analyzer.py b/skills/kb-deep-analyzer/deep_analyzer.py index a3b4339..926b54a 100644 --- a/skills/kb-deep-analyzer/deep_analyzer.py +++ b/skills/kb-deep-analyzer/deep_analyzer.py @@ -59,24 +59,32 @@ class SimpleAIClient: self.provider = provider or os.environ.get("AI_PROVIDER", "") self.model = model or os.environ.get("AI_MODEL", "gpt-4") self.base_url = base_url or os.environ.get("AI_BASE_URL", "") + self.max_retries = 3 # 最大重试次数 def generate(self, prompt: str, max_tokens: int = 1000) -> str: - """生成文本""" + """生成文本,支持重试""" if not self.api_key: raise ValueError("AI_API_KEY 未设置,请配置环境变量") - try: - if self.provider == "openai": - return self._call_openai(prompt, max_tokens) - elif self.provider == "zhipu": - return self._call_zhipu(prompt, max_tokens) - elif self.provider == "minimax": - return self._call_minimax(prompt, max_tokens) - else: - raise ValueError(f"不支持的 AI 提供商: {self.provider}") - except Exception as e: - print(f"[AI Client] 调用失败: {e}") - raise + last_error = None + for attempt in range(self.max_retries): + try: + if self.provider == "openai": + return self._call_openai(prompt, max_tokens) + elif self.provider == "zhipu": + return self._call_zhipu(prompt, max_tokens) + elif self.provider == "minimax": + return self._call_minimax(prompt, max_tokens) + else: + raise ValueError(f"不支持的 AI 提供商: {self.provider}") + except Exception as e: + last_error = e + if attempt < self.max_retries - 1: + print(f"[AI Client] 第 {attempt + 1} 次调用失败,{self.max_retries - attempt - 1} 次重试剩余: {e}") + continue + else: + print(f"[AI Client] 调用失败,已重试 {self.max_retries} 次: {e}") + raise def generate_json(self, prompt: str) -> Dict[str, Any]: """生成 JSON 格式响应""" @@ -103,7 +111,7 @@ class SimpleAIClient: "messages": [{"role": "user", "content": prompt}] } - response = requests.post(url, headers=headers, json=payload, timeout=30) + response = requests.post(url, headers=headers, json=payload, timeout=120) response.raise_for_status() return response.json()['choices'][0]['message']['content'] @@ -120,7 +128,7 @@ class SimpleAIClient: "max_tokens": max_tokens } - response = requests.post(url, headers=headers, json=payload, timeout=30) + response = requests.post(url, headers=headers, json=payload, timeout=120) response.raise_for_status() data = response.json() @@ -181,7 +189,7 @@ class SimpleAIClient: "messages": [{"role": "user", "content": prompt}] } - response = requests.post(url, headers=headers, json=payload, timeout=30) + response = requests.post(url, headers=headers, json=payload, timeout=120) response.raise_for_status() data = response.json() for item in data.get('content', []):