This commit is contained in:
parent
8bb35d12ea
commit
25c83fc7fb
|
|
@ -61,6 +61,34 @@ def search_docs(request: QueryRequest, user_id: str = Query(..., description="
|
||||||
logger.info(f"Retrieving top {settings.TOP_K} results for query: {request.query}")
|
logger.info(f"Retrieving top {settings.TOP_K} results for query: {request.query}")
|
||||||
nodes = retriever.retrieve(request.query)
|
nodes = retriever.retrieve(request.query)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
# 打印出每个结果的向量和文本
|
||||||
|
for i, node in enumerate(nodes):
|
||||||
|
# 打印文本
|
||||||
|
logger.info(f"Result {i+1}:")
|
||||||
|
logger.info(f" Text: {node.get_content()}") # 打印文本
|
||||||
|
|
||||||
|
# 打印向量及其长度
|
||||||
|
embedding = node.embedding
|
||||||
|
logger.info(f" Embedding (Vector): {embedding}") # 打印向量
|
||||||
|
logger.info(f" Embedding Length: {len(embedding)}") # 打印向量的长度(即向量的维度)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
# 返回检索结果
|
# 返回检索结果
|
||||||
result = {
|
result = {
|
||||||
"user_id": user_id,
|
"user_id": user_id,
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue