fixed linting

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

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):

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():

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():

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():