diff --git a/app/api/search.py b/app/api/search.py index 10d93e8..cb3c4ce 100644 --- a/app/api/search.py +++ b/app/api/search.py @@ -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}") 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 = { "user_id": user_id,