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"