gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

API中可以通过POST参数控制查询是否only_need_context

Browse Source
This commit is contained in:
90houlaoheshang
2024-11-05 16:44:32 +08:00
parent 7d2641989b
commit 8d00f5a4cd
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/lightrag_api_openai_compatible_demo.py
View File

@@ -71,6 +71,7 @@ rag = LightRAG(
class QueryRequest(BaseModel): class QueryRequest(BaseModel):
query: str query: str
mode: str = "hybrid" mode: str = "hybrid"
only_need_context: bool = False
class InsertRequest(BaseModel): class InsertRequest(BaseModel):
@@ -95,7 +96,8 @@ async def query_endpoint(request: QueryRequest):
try: try:
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
result = await loop.run_in_executor( 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) return Response(status="success", data=result)
except Exception as e: except Exception as e: