This commit is contained in:
hailin 2025-05-11 14:08:18 +08:00
parent 552c36f445
commit a534980b8b
1 changed files with 41 additions and 9 deletions

View File

@ -101,15 +101,47 @@ def search_docs(request: QueryRequest, user_id: str = Query(..., description="
# # 返回检索结果
# result = {
# "user_id": user_id,
# "query": request.query,
# "results": [
# {"score": float(node.score or 0), "text": node.get_content()} # 确保从 Node 中获取文本
# for node in response
# ]
# }
# 确保 response 是一个有效的 JSON 数据列表
if response and hasattr(response, 'json'):
response_data = response.json() # 提取 JSON 数据
# 检查 response_data 是否为列表且不为空
if isinstance(response_data, list) and len(response_data) > 0:
# 返回检索结果
result = {
"user_id": user_id,
"query": request.query,
"results": [
{"score": float(node.score or 0), "text": node.get_content()} # 确保从 Node 中获取文本
for node in response
for node in response_data
]
}
else:
logger.warning("No valid results found in response_data or response_data is not a list.")
result = {
"user_id": user_id,
"query": request.query,
"results": []
}
else:
logger.warning("Response is not iterable or invalid.")
result = {
"user_id": user_id,
"query": request.query,
"results": []
}
logger.info(f"Search results for user {user_id}: {result}")
return result