MachineLearning2025/sms-castle-walls
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 修复base_url缺少协议前缀的问题

Browse Source 
为agent和llm_service模块添加base_url协议前缀验证逻辑，当缺少http/https前缀时自动添加https://
This commit is contained in:
萧 2026-01-15 14:51:38 +08:00
parent 48013e93bd
commit 72ffdf1647
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/agent/agent.py
View File

@ -50,6 +50,11 @@ class SpamClassificationAgent:
self.api_key = env_vars.get("DEEPSEEK_API_KEY") or os.getenv("DEEPSEEK_API_KEY") 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") 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 self.client = None

5
src/llm/llm_service.py
View File

@ -29,6 +29,11 @@ class LLMService:
self.api_key = env_vars.get("DEEPSEEK_API_KEY") or os.getenv("DEEPSEEK_API_KEY") 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") 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" self.default_model = "deepseek-chat"