更新.gitignore以忽略临时文件,修改api_key文件,重构合并名词的逻辑,删除不再使用的脚本,优化对话到工单的处理流程,添加会话结果保存为JSON的功能,调整API调用参数,修复部分代码中的错误。
This commit is contained in:
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user