优化DifyCompareTest类,添加DifyExporter实例以支持词条检索,更新DifyQueryRetrieval_api.py中的topk参数,增强DifyExporter类以从HTTP服务获取查询类型和点踩原因,简化构造函数,移除不必要的查询日志加载逻辑。
This commit is contained in:
@@ -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()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user