更新.gitignore以忽略临时文件,修改api_key文件,重构合并名词的逻辑,删除不再使用的脚本,优化对话到工单的处理流程,添加会话结果保存为JSON的功能,调整API调用参数,修复部分代码中的错误。

This commit is contained in:
2025-07-25 09:53:47 +08:00
parent 4d7ef54ae7
commit 2cbdc23fc0
13 changed files with 1205 additions and 27522 deletions
+7 -2
View File
@@ -39,6 +39,7 @@ class RetrieveRequest(BaseModel):
original_query: str
query_list: str
data_set_list: str
query_expand_dict: dict | str = Field(default="{}")
# 创建FastAPI应用
app = FastAPI(
@@ -87,13 +88,17 @@ async def retrieve(request: RetrieveRequest):
# 解析查询列表和数据集列表
query_list = request.query_list.split("<sub_query>")
data_set_list = request.data_set_list.split("<dataset>")
if isinstance(request.query_expand_dict, str):
query_expand_dict = json.loads(request.query_expand_dict)
else:
query_expand_dict = request.query_expand_dict
# 调用异步检索方法
start_time = time.time()
results = await dify_query_retrieval.retrieve_api_async(
request.original_query,
query_list,
data_set_list,
query_expand_dict=query_expand_dict,
top_k=5
)
end_time = time.time()
@@ -107,7 +112,7 @@ async def retrieve(request: RetrieveRequest):
if __name__ == "__main__":
# 使用Uvicorn运行FastAPI应用
import uvicorn
uvicorn.run("rag2_0.dify.DifyQueryRetrieval_api:app", host="0.0.0.0", port=8002, reload=False, workers=10, log_level="info")
uvicorn.run("rag2_0.dify.DifyQueryRetrieval_api:app", host="0.0.0.0", port=8002, reload=False, workers=1, log_level="info")
# # 使用uvicorn启动服务
# import uvicorn
# uvicorn.run(