From 8d00f5a4cd1c10aa6e5699ac7bc68f40e15d5911 Mon Sep 17 00:00:00 2001
From: 90houlaoheshang <907333918@qq.com>
Date: Tue, 5 Nov 2024 16:44:32 +0800
Subject: [PATCH] =?UTF-8?q?API=E4=B8=AD=E5=8F=AF=E4=BB=A5=E9=80=9A?=
 =?UTF-8?q?=E8=BF=87POST=E5=8F=82=E6=95=B0=E6=8E=A7=E5=88=B6=E6=9F=A5?=
 =?UTF-8?q?=E8=AF=A2=E6=98=AF=E5=90=A6only=5Fneed=5Fcontext?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 examples/lightrag_api_openai_compatible_demo.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/examples/lightrag_api_openai_compatible_demo.py b/examples/lightrag_api_openai_compatible_demo.py
index 7e1f608a..8a7286b7 100644
--- a/examples/lightrag_api_openai_compatible_demo.py
+++ b/examples/lightrag_api_openai_compatible_demo.py
@@ -71,6 +71,7 @@ rag = LightRAG(
 class QueryRequest(BaseModel):
     query: str
     mode: str = "hybrid"
+    only_need_context: bool = False
 
 
 class InsertRequest(BaseModel):
@@ -95,7 +96,8 @@ async def query_endpoint(request: QueryRequest):
     try:
         loop = asyncio.get_event_loop()
         result = await loop.run_in_executor(
-            None, lambda: rag.query(request.query, param=QueryParam(mode=request.mode))
+            None, lambda: rag.query(request.query,
+                                    param=QueryParam(mode=request.mode, only_need_context=request.only_need_context))
         )
         return Response(status="success", data=result)
     except Exception as e: