|
|
|
@ -96,6 +96,7 @@ async def reply(session_id: str = Form(...), prompt: str = Form(...)):
|
|
|
|
|
results = collection_manager.search(
|
|
|
|
|
data=current_embedding, # 输入向量
|
|
|
|
|
search_params=search_params, # 搜索参数
|
|
|
|
|
expr=f"session_id == '{session_id}'", # 按 session_id 过滤
|
|
|
|
|
limit=5 # 返回 5 条结果
|
|
|
|
|
)
|
|
|
|
|
end_time = time.time()
|
|
|
|
|