This commit is contained in:
parent
552c36f445
commit
a534980b8b
|
|
@ -101,15 +101,47 @@ def search_docs(request: QueryRequest, user_id: str = Query(..., description="
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
# 返回检索结果
|
# # 返回检索结果
|
||||||
result = {
|
# result = {
|
||||||
"user_id": user_id,
|
# "user_id": user_id,
|
||||||
"query": request.query,
|
# "query": request.query,
|
||||||
"results": [
|
# "results": [
|
||||||
{"score": float(node.score or 0), "text": node.get_content()} # 确保从 Node 中获取文本
|
# {"score": float(node.score or 0), "text": node.get_content()} # 确保从 Node 中获取文本
|
||||||
for node in response
|
# 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_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}")
|
logger.info(f"Search results for user {user_id}: {result}")
|
||||||
return result
|
return result
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue