From 72ffdf1647a4df75ee3154ec92cb1aca77b3850a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=90=A7?= Date: Thu, 15 Jan 2026 14:51:38 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8Dbase=5Furl=E7=BC=BA?= =?UTF-8?q?=E5=B0=91=E5=8D=8F=E8=AE=AE=E5=89=8D=E7=BC=80=E7=9A=84=E9=97=AE?= =?UTF-8?q?=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 为agent和llm_service模块添加base_url协议前缀验证逻辑，当缺少http/https前缀时自动添加https:// --- src/agent/agent.py | 5 +++++ src/llm/llm_service.py | 5 +++++ 2 files changed, 10 insertions(+) diff --git a/src/agent/agent.py b/src/agent/agent.py index 47280c8..0f5cd0b 100644 --- a/src/agent/agent.py +++ b/src/agent/agent.py @@ -50,6 +50,11 @@ class SpamClassificationAgent: self.api_key = env_vars.get("DEEPSEEK_API_KEY") or os.getenv("DEEPSEEK_API_KEY") self.base_url = env_vars.get("DEEPSEEK_BASE_URL") or os.getenv("DEEPSEEK_BASE_URL", "https://api.deepseek.com") + # 验证并修复base_url，确保包含协议前缀 + if self.base_url and not (self.base_url.startswith("http://") or self.base_url.startswith("https://")): + print(f"警告: base_url '{self.base_url}' 缺少协议前缀，将添加 'https://'") + self.base_url = f"https://{self.base_url}" + # 延迟创建客户端，直到实际需要时 self.client = None diff --git a/src/llm/llm_service.py b/src/llm/llm_service.py index ced1695..067c563 100644 --- a/src/llm/llm_service.py +++ b/src/llm/llm_service.py @@ -29,6 +29,11 @@ class LLMService: self.api_key = env_vars.get("DEEPSEEK_API_KEY") or os.getenv("DEEPSEEK_API_KEY") self.base_url = env_vars.get("DEEPSEEK_BASE_URL") or os.getenv("DEEPSEEK_BASE_URL", "https://api.deepseek.com") + # 验证并修复base_url，确保包含协议前缀 + if self.base_url and not (self.base_url.startswith("http://") or self.base_url.startswith("https://")): + print(f"警告: base_url '{self.base_url}' 缺少协议前缀，将添加 'https://'") + self.base_url = f"https://{self.base_url}" + # 默认模型 self.default_model = "deepseek-chat"