yujunjun/LightRAG
1
0
Fork 0
mirror of https://github.com/HKUDS/LightRAG.git synced 2025-12-27 23:02:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixed linting

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-12-26 23:39:10 +01:00
parent 4d0184189f
commit 27a52ddf37
4 changed files with 43 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
lightrag/api/azure_openai_lightrag_server.py
View File

@ -309,7 +309,11 @@ def create_app(args):
try:
response = await rag.aquery(
request.query,
param=QueryParam(mode=request.mode, stream=False, only_need_context=request.only_need_context),
param=QueryParam(
mode=request.mode,
stream=False,
only_need_context=request.only_need_context,
),
)
return QueryResponse(response=response)
except Exception as e:
@ -320,7 +324,11 @@ def create_app(args):
try:
response = await rag.aquery(
request.query,
param=QueryParam(mode=request.mode, stream=True, only_need_context=request.only_need_context),
param=QueryParam(
mode=request.mode,
stream=True,
only_need_context=request.only_need_context,
),
)
if inspect.isasyncgen(response):

13
lightrag/api/lollms_lightrag_server.py
View File

@ -267,7 +267,11 @@ def create_app(args):
try:
response = await rag.aquery(
request.query,
param=QueryParam(mode=request.mode, stream=request.stream, only_need_context=request.only_need_context),
param=QueryParam(
mode=request.mode,
stream=request.stream,
only_need_context=request.only_need_context,
),
)
if request.stream:
@ -284,7 +288,12 @@ def create_app(args):
async def query_text_stream(request: QueryRequest):
try:
response = rag.query(
request.query, param=QueryParam(mode=request.mode, stream=True, only_need_context=request.only_need_context)
request.query,
param=QueryParam(
mode=request.mode,
stream=True,
only_need_context=request.only_need_context,
),
)
async def stream_generator():

13
lightrag/api/ollama_lightrag_server.py
View File

@ -267,7 +267,11 @@ def create_app(args):
try:
response = await rag.aquery(
request.query,
param=QueryParam(mode=request.mode, stream=request.stream, only_need_context=request.only_need_context),
param=QueryParam(
mode=request.mode,
stream=request.stream,
only_need_context=request.only_need_context,
),
)
if request.stream:
@ -284,7 +288,12 @@ def create_app(args):
async def query_text_stream(request: QueryRequest):
try:
response = rag.query(
request.query, param=QueryParam(mode=request.mode, stream=True, only_need_context=request.only_need_context)
request.query,
param=QueryParam(
mode=request.mode,
stream=True,
only_need_context=request.only_need_context,
),
)
async def stream_generator():

13
lightrag/api/openai_lightrag_server.py
View File

@ -271,7 +271,11 @@ def create_app(args):
try:
response = await rag.aquery(
request.query,
param=QueryParam(mode=request.mode, stream=request.stream, only_need_context=request.only_need_context),
param=QueryParam(
mode=request.mode,
stream=request.stream,
only_need_context=request.only_need_context,
),
)
if request.stream:
@ -288,7 +292,12 @@ def create_app(args):
async def query_text_stream(request: QueryRequest):
try:
response = rag.query(
request.query, param=QueryParam(mode=request.mode, stream=True, only_need_context=request.only_need_context)
request.query,
param=QueryParam(
mode=request.mode,
stream=True,
only_need_context=request.only_need_context,
),
)
async def stream_generator():