浏览代码

增加模型名称参数

jiaqiang 4 天之前
父节点
当前提交
bca609637c
共有 2 个文件被更改,包括 14 次插入6 次删除
  1. 8 5
      llmops/complete_agent_flow_rule.py
  2. 6 1
      llmops/config.py

+ 8 - 5
llmops/complete_agent_flow_rule.py

@@ -48,7 +48,7 @@ from llmops.agents.rules_engine_metric_calculation_agent import RulesEngineMetri
 from llmops.agents.data_manager import DataManager
 import os
 from llmops.agents.data_classify_agent import data_classify
-from llmops.config import DEEPSEEK_API_KEY
+from llmops.config import DEEPSEEK_API_KEY,LLM_API_KEY, LLM_BASE_URL,LLM_MODEL_NAME
 
 class CompleteAgentFlow:
     """完整的智能体工作流"""
@@ -636,12 +636,13 @@ async def main():
     print("=" * 50)
 
     # 导入配置
-    import config
 
-    if not DEEPSEEK_API_KEY:
+
+    if not LLM_API_KEY:
         print("❌ 未找到API密钥")
         return
 
+
     # 行业
     industry = "农业"
 
@@ -662,8 +663,10 @@ async def main():
         industry = industry,
         data=test_data,
         file_name=file_name,
-        api_key=config.DEEPSEEK_API_KEY,
-        session_id="direct-test"
+        api_key=LLM_API_KEY,
+        session_id="direct-test",
+        base_url=LLM_BASE_URL,
+        model_name=LLM_MODEL_NAME
     )
 
     print(f"📋 结果: {'✅ 成功' if result.get('success') else '❌ 失败'}")

+ 6 - 1
llmops/config.py

@@ -115,4 +115,9 @@ deepseek_v3_model = {
     "name": "deepseek-chat",
     "api_key": DEEPSEEK_API_KEY,
     "base_url": DEEPSEEK_BASE_URL
-}
+}
+
+# LLM 运行配置
+LLM_API_KEY=deepseek_v3_model["api_key"]
+LLM_BASE_URL=deepseek_v3_model["base_url"]
+LLM_MODEL_NAME=deepseek_v3_model["name"]