自动同步: 2026-03-22 02:10

skills/kb_workflow/deep_analyzer/__init__.py

@@ -37,20 +37,25 @@ class AnalysisResult:
 class SimpleAIClient:
     """AI 客户端"""
 
-    def __init__(self, api_key: str = None, provider: str = None, model: str = None, base_url: str = None):
+    def __init__(self, api_key: str = None, provider: str = None, model: str = None, base_url: str = None, disable_reasoning: bool = False):
         self.api_key = api_key or os.environ.get("AI_API_KEY", "")
         self.provider = provider or os.environ.get("AI_PROVIDER", "")
         self.model = model or os.environ.get("AI_MODEL", "glm-5")
         self.base_url = base_url or os.environ.get("AI_BASE_URL", "")
         self.max_retries = 3
+        self.disable_reasoning = disable_reasoning
 
-    def generate(self, prompt: str, max_tokens: int = 16000) -> str:
+    def generate(self, prompt: str, max_tokens: int = 16000, disable_reasoning: bool = None) -> str:
         """生成文本"""
         api_key = os.environ.get("AI_API_KEY", self.api_key)
         provider = os.environ.get("AI_PROVIDER", self.provider)
         model = os.environ.get("AI_MODEL", self.model)
         base_url = os.environ.get("AI_BASE_URL", self.base_url)
 
+        # 确定 disable_reasoning 的值：优先使用参数，否则使用实例变量
+        if disable_reasoning is None:
+            disable_reasoning = self.disable_reasoning
+
         if not api_key:
             raise ValueError("AI_API_KEY 未设置")
 
@@ -63,7 +68,7 @@ class SimpleAIClient:
                 if provider == "openai":
                     result = self._call_openai(prompt, max_tokens, api_key, model, base_url)
                 elif provider == "zhipu":
-                    result = self._call_zhipu(prompt, max_tokens, api_key, model, base_url)
+                    result = self._call_zhipu(prompt, max_tokens, api_key, model, base_url, disable_reasoning)
                 elif provider == "minimax":
                     result = self._call_minimax(prompt, max_tokens, api_key, model, base_url)
                 else:
@@ -88,35 +93,51 @@ class SimpleAIClient:
         response.raise_for_status()
         return response.json()['choices'][0]['message']['content']
 
-    def _call_zhipu(self, prompt: str, max_tokens: int, api_key: str, model: str, base_url: str) -> str:
+    def _call_zhipu(self, prompt: str, max_tokens: int, api_key: str, model: str, base_url: str, disable_reasoning: bool = False) -> str:
         import requests
         import sys
 
         base_url = base_url or "https://open.bigmodel.cn/api/coding/paas/v4"
         url = f"{base_url}/chat/completions"
         headers = {"Authorization": f"Bearer {api_key}"}
-        # GLM-4.5/4.6/4.7/5 默认启用thinking mode，使用 thinking.type 参数禁用
+        # GLM-4.5/4.6/4.7/5 默认启用thinking mode
+        # 通过 disable_reasoning 参数控制是否禁用思考模式
         payload = {
             "model": model,
             "messages": [{"role": "user", "content": prompt}],
-            "max_tokens": max_tokens,
-            "thinking": {
-                "type": "disabled"  # 禁用思考模式，获取直接响应
-            }
+            "max_tokens": max_tokens
         }
+
+        # 如果明确要求禁用思考模式，添加 thinking 参数
+        if disable_reasoning:
+            payload["thinking"] = {"type": "disabled"}
         response = requests.post(url, headers=headers, json=payload, timeout=600)
         response.raise_for_status()
         data = response.json()
         message = data.get('choices', [{}])[0].get('message', {})
+
+        # 调试：打印完整的 API 响应结构
+        print(f"[DEBUG] disable_reasoning={disable_reasoning}", file=sys.stderr)
+        print(f"[DEBUG] message keys: {list(message.keys())}", file=sys.stderr)
+        if 'content' in message:
+            print(f"[DEBUG] content 长度: {len(message['content'])}, 前200字符: {message['content'][:200]}", file=sys.stderr)
+        if 'reasoning_content' in message:
+            print(f"[DEBUG] reasoning_content 长度: {len(message['reasoning_content'])}, 前200字符: {message['reasoning_content'][:200]}", file=sys.stderr)
+
+        # 返回 content 字段
+        # 当禁用思考模式时，content 包含直接答案
+        # 当启用思考模式时，content 可能包含思考过程和分析
         content = message.get('content', '')
 
         if not content:
-            # 如果仍然为空，检查是否有reasoning_content（用于调试）
+            # 如果 content 为空，检查是否有 reasoning_content
             reasoning = message.get('reasoning_content', '')
             if reasoning:
-                print(f"[DEBUG] Zhipu API返回空的content，但有reasoning_content。", file=sys.stderr)
-                print(f"[DEBUG] 这可能表示thinking.type参数未生效或模型不支持。", file=sys.stderr)
+                print(f"[DEBUG] Zhipu API返回空的content，使用reasoning_content。", file=sys.stderr)
                 print(f"[DEBUG] reasoning_content前200字符: {reasoning[:200]}", file=sys.stderr)
+                content = reasoning
+
+        if not content:
             raise ValueError(f"Zhipu API返回空内容。请检查API配置和模型参数。响应数据: {data}")
 
         return content