From f95384b68cd7ea0afda5efafe73587ac3b0f33a9 Mon Sep 17 00:00:00 2001 From: WinstonCHEN1 <1281838223@qq.com> Date: Wed, 30 Oct 2024 15:36:44 -0700 Subject: [PATCH] fix:Step_3.py context --- reproduce/Step_3.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/reproduce/Step_3.py b/reproduce/Step_3.py index a56190fc..2c5d699c 100644 --- a/reproduce/Step_3.py +++ b/reproduce/Step_3.py @@ -18,8 +18,8 @@ def extract_queries(file_path): async def process_query(query_text, rag_instance, query_param): try: - result, context = await rag_instance.aquery(query_text, param=query_param) - return {"query": query_text, "result": result, "context": context}, None + result = await rag_instance.aquery(query_text, param=query_param) + return {"query": query_text, "result": result}, None except Exception as e: return None, {"query": query_text, "error": str(e)}