fa7656ae0d
- 重构项目目录结构，添加src目录包含data/models/agent/llm模块 - 实现数据预处理、验证和分割功能 - 添加机器学习模型训练和预测功能 - 实现LLM服务集成DeepSeek API - 构建Agent框架实现工具调用和结果整合 - 添加Streamlit可视化应用 - 更新项目配置和依赖管理 - 删除无用文件和旧实现代码
34 lines
746 B
Python
34 lines
746 B
Python
from .agent import SpamClassificationAgent, AgentConfig, Message, agent, get_agent
|
|
from .tools import (
|
|
PredictSpamInput,
|
|
PredictSpamOutput,
|
|
predict_spam_tool,
|
|
ExplainPredictionInput,
|
|
ExplainPredictionOutput,
|
|
explain_prediction_tool,
|
|
TranslateTextInput,
|
|
TranslateTextOutput,
|
|
translate_text_tool,
|
|
TOOLS
|
|
)
|
|
|
|
__all__ = [
|
|
# agent.py
|
|
"SpamClassificationAgent",
|
|
"AgentConfig",
|
|
"Message",
|
|
"agent",
|
|
"get_agent",
|
|
|
|
# tools.py
|
|
"PredictSpamInput",
|
|
"PredictSpamOutput",
|
|
"predict_spam_tool",
|
|
"ExplainPredictionInput",
|
|
"ExplainPredictionOutput",
|
|
"explain_prediction_tool",
|
|
"TranslateTextInput",
|
|
"TranslateTextOutput",
|
|
"translate_text_tool",
|
|
"TOOLS"
|
|
] |