ruoyunbai/compliance
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix:kunlun deepseek

Browse Source
This commit is contained in:
gongwenxin 2025-08-13 10:29:19 +08:00
parent ed77da19b6
commit ee8c63dd63
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ddms_compliance_suite/llm_utils/llm_service.py
View File

@ -31,7 +31,7 @@ class LLMService:
封装与大语言模型 (LLM) API的交互，用于智能参数生成和验证。 封装与大语言模型 (LLM) API的交互，用于智能参数生成和验证。
目前针对通义千问兼容OpenAI的模式进行实现。 目前针对通义千问兼容OpenAI的模式进行实现。
""" """
def __init__(self, api_key: str, base_url: str, model_name: str = "qwen-plus"): def __init__(self, api_key: str, base_url: str, model_name: str = "deepseek-v3"):
""" """
初始化LLM服务。 初始化LLM服务。
@ -241,8 +241,9 @@ if __name__ == '__main__':
logging.basicConfig(level=logging.DEBUG, logging.basicConfig(level=logging.DEBUG,
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s') format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
DASH_API_KEY = os.environ.get("DASHSCOPE_API_KEY") or "sk-YOUR_DASHSCOPE_API_KEY" DASH_API_KEY = os.environ.get("DASHSCOPE_API_KEY") or "sk-lbGrsUPL1iby86h554FaE536C343435dAa9bA65967A840B2"
DASH_BASE_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1" # DASH_BASE_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1"
DASH_BASE_URL = "https://aiproxy.petrotech.cnpc/v1"
if "YOUR_DASHSCOPE_API_KEY" in DASH_API_KEY: if "YOUR_DASHSCOPE_API_KEY" in DASH_API_KEY:
logger.warning("请将 DASH_API_KEY 替换为您的有效API密钥，或设置 DASHSCOPE_API_KEY 环境变量。") logger.warning("请将 DASH_API_KEY 替换为您的有效API密钥，或设置 DASHSCOPE_API_KEY 环境变量。")