Upload ai_advisor.py

modules/ai_advisor.py

@@ -128,46 +128,36 @@ class AIAdvisor:
             logger.debug(f"Calling Hugging Face API with model {self.model_id}")
             start_time = time.time()
             
-            # 直接调用 Hugging Face Inference API
-            api_url = f"https://api-inference.huggingface.co/models/{self.model_id}"
-            headers = {"Authorization": f"Bearer {self.hf_token}"}
-            payload = {
-                "inputs": prompt,
-                "parameters": {
-                    "max_new_tokens": 300,
-                    "temperature": 0.7,
-                    "top_p": 0.95,
-                }
-            }
+            if not self.client:
+                return "【宗师未苏醒】客户端初始化失败，请检查配置。"
             
-            response = requests.post(api_url, headers=headers, json=payload, timeout=self.timeout)
-            elapsed = time.time() - start_time
-            
-            if response.status_code == 200:
-                result = response.json()
-                if isinstance(result, list) and len(result) > 0:
-                    insight = result[0].get('generated_text', '').strip()
-                    if insight.startswith(prompt):
-                        insight = insight[len(prompt):].strip()
-                    logger.info(f"AI Advisor response received in {elapsed:.2f}s: {insight[:100]}")
-                    return insight if insight else "【宗师沉默】模型返回了空信息。"
-                else:
-                    logger.error(f"Unexpected API response format: {result}")
-                    return "【宗师解析失败】模型返回格式异常，无法解读。"
-            else:
-                logger.error(f"HF API returned {response.status_code}: {response.text[:200]}")
-                if response.status_code == 503:
-                    return f"【宗师正在入定】模型系统正在唤醒加载中，请等待约半分钟后重新点击「以此起课」。"
-                elif response.status_code == 401 or response.status_code == 403:
-                    return f"【宗师传音受阻】云端认证失败 (HTTP {response.status_code})。请检查 HF_TOKEN 是否正确或已过期。"
+            # 使用官方 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 "【宗师沉默】模型返回了空信息。"
+                
+            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"【宗师占卜受阻】API 调用异常 (HTTP {response.status_code})，请查看系统日志。"
-            
-        except requests.exceptions.Timeout:
-            logger.error(f"AI Advisor API call timed out after {self.timeout}s")
-            return f"【宗师传音超时】本次跨越时空通讯超过 {self.timeout} 秒未能接通，请稍后重试。"
-        except Exception as e:
-            logger.error(f"AI Advisor API call failed: {type(e).__name__}: {str(e)}")
-            import traceback
-            logger.error(traceback.format_exc())
-            return f"【宗师连接中断】系统内部通信故障：{type(e).__name__}。"
+                    return f"【宗师连接中断】(HTTP: {type(e).__name__}) 请稍后重试。"