Upload ai_advisor.py

modules/ai_advisor.py

@@ -107,6 +107,8 @@ class AIAdvisor:
 
 【宗师点拨】
 """
+        return prompt
+
     def get_master_insight(self, divination_results: Dict[str, Any]) -> Optional[str]:
         """调用 Hugging Face Inference API 获取宗师解读"""
         if not self.hf_token:
@@ -114,8 +116,6 @@ class AIAdvisor:
             return None
         
         try:
-            import requests
-            
             prompt = self._build_master_prompt(
                 divination_results.get('lunar_data', {}),
                 divination_results.get('bazi', {}),
@@ -133,31 +133,31 @@ class AIAdvisor:
             
             # 使用官方 InferenceClient 自带的智能路由功能 (支持 Together/HF Pro 等多种后端引擎)
             messages = [{"role": "user", "content": prompt}]
-            try:
-                response = self.client.chat_completion(
-                    messages=messages,
-                    model=self.model_id,
-                    max_tokens=300,
-                    stream=False,
-                    temperature=0.7,
-                    top_p=0.95
-                )
-                elapsed = time.time() - start_time
-                insight = response.choices[0].message.content.strip()
-                logger.info(f"AI Advisor response received in {elapsed:.2f}s: {insight[:100]}")
-                return insight if insight else "【宗师沉默】模型返回了空信息。"
+            
+            response = self.client.chat_completion(
+                messages=messages,
+                model=self.model_id,
+                max_tokens=300,
+                stream=False,
+                temperature=0.7,
+                top_p=0.95
+            )
+            elapsed = time.time() - start_time
+            insight = response.choices[0].message.content.strip()
+            logger.info(f"AI Advisor response received in {elapsed:.2f}s: {insight[:100]}")
+            return insight if insight else "【宗师沉默】模型返回了空信息。"
                 
-            except Exception as e:
-                # 捕获 HF Hub 相关错误
-                err_msg = str(e)
-                logger.error(f"HF API call failed: {type(e).__name__} - {err_msg}")
-                if "401" in err_msg or "Unauthorized" in err_msg:
-                    return f"【宗师传音受阻】云端认证失败 (HTTP 401)。请确保 HF_TOKEN 具有 Inference 权限并且未过期。"
-                elif "503" in err_msg or "loading" in err_msg.lower():
-                    return f"【宗师正在入定】模型节点正在冷启动唤醒中，大约需要半分钟，请稍后重新起课。"
-                elif "404" in err_msg or "not found" in err_msg.lower():
-                    return f"【时空道标丢失】找不到该模型 ({self.model_id}) 或其不支持此种对话模式，请尝试更换模型ID。"
-                elif "timeout" in err_msg.lower():
-                    return f"【宗师传音超时】跨越时空通讯未能按时接通，请稍后重试。"
-                else:
-                    return f"【宗师连接中断】(HTTP: {type(e).__name__}) 请稍后重试。"
+        except Exception as e:
+            # 捕获 HF Hub 相关错误及普通错误
+            err_msg = str(e)
+            logger.error(f"HF API call failed: {type(e).__name__} - {err_msg}")
+            if "401" in err_msg or "Unauthorized" in err_msg:
+                return f"【宗师传音受阻】云端认证失败 (HTTP 401)。请确保 HF_TOKEN 具有 Inference 权限并且未过期。"
+            elif "503" in err_msg or "loading" in err_msg.lower():
+                return f"【宗师正在入定】模型节点正在冷启动唤醒中，大约需要半分钟，请稍后重新起课。"
+            elif "404" in err_msg or "not found" in err_msg.lower():
+                return f"【时空道标丢失】找不到该模型 ({self.model_id}) 或其不支持此种对话模式，请尝试更换模型ID。"
+            elif "timeout" in err_msg.lower():
+                return f"【宗师传音超时】跨越时空通讯未能按时接通，请稍后重试。"
+            else:
+                return f"【宗师连接中断】({type(e).__name__}) 请稍后重试。"