优化DifyCompareTest类,添加DifyExporter实例以支持词条检索,更新DifyQueryRetrieval_api.py中的topk参数,增强DifyExporter类以从HTTP服务获取查询类型和点踩原因,简化构造函数,移除不必要的查询日志加载逻辑。

This commit is contained in:
2025-07-30 17:30:24 +08:00
parent 57369059eb
commit 728262cc65
3 changed files with 95 additions and 54 deletions
+3 -2
View File
@@ -5,7 +5,7 @@ import os
from fastapi import FastAPI, HTTPException, Request
from fastapi.responses import JSONResponse
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel, Field
from pydantic import BaseModel, Field, ConfigDict
from typing import Dict, List, Any, Optional
import asyncio
@@ -43,6 +43,7 @@ class RetrieveRequest(BaseModel):
query_list: str
data_set_list: str
query_expand_dict: dict | str = Field(default="{}")
topk: int = Field(default=4)
# 创建FastAPI应用
app = FastAPI(
@@ -102,7 +103,7 @@ async def retrieve(request: RetrieveRequest):
query_list,
data_set_list,
query_expand_dict=query_expand_dict,
top_k=4
top_k=request.topk
)
end_time = time.time()