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