This commit is contained in:
parent
552c36f445
commit
a534980b8b
|
|
@ -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
|
||||
]
|
||||
}
|
||||
# # 返回检索结果
|
||||
# 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_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
|
||||
|
|
|
|||
Loading…
Reference in New Issue