feat: 三个工具对应的接口(优化)

This commit is contained in:
martsforever
2025-09-24 16:59:47 +08:00
parent c93805e159
commit 432f4b7095
+18 -3
View File
@@ -22,8 +22,14 @@ def add_api_route(app: FastAPI):
@app.get('/api/project/analysis') @app.get('/api/project/analysis')
async def project_analysis(project_name=Query(..., description="项目名称")): async def project_analysis(project_name=Query(..., description="项目名称")):
tool_result_string = await tool_project_analysis.ainvoke({"project_name": project_name}) tool_result_string = await tool_project_analysis.ainvoke({"project_name": project_name})
try:
tool_result_data = json.loads(tool_result_string) tool_result_data = json.loads(tool_result_string)
return tool_result_data[0] result = tool_result_data[0]
except:
result = tool_result_string
return {
"result": result
}
# 项目日报总结 # 项目日报总结
@app.get('/api/project/report') @app.get('/api/project/report')
@@ -37,9 +43,18 @@ def add_api_route(app: FastAPI):
"start_time": start_time, "start_time": start_time,
"end_time": end_time, "end_time": end_time,
}) })
return json.loads(tool_result_string) try:
tool_result_data = json.loads(tool_result_string)
result = tool_result_data[0]
except:
result = tool_result_string
return {
"result": result
}
# 检索企业内部文档工具 # 检索企业内部文档工具
@app.get('/api/document/retrieve') @app.get('/api/document/retrieve')
async def document_retrieve(question=Query(..., description="用户的问题")): async def document_retrieve(question=Query(..., description="用户的问题")):
return await tool_retrieve_documents.ainvoke({"question": question}) return {
"result": await tool_retrieve_documents.ainvoke({"question": question})
}