Skip to content

GitLab

Commit 5621025c authored by Timothy J. Baek's avatar Timothy J. Baek
Browse files

feat: async filter support

parent 6bb2f418
Show whitespace changes
Inline Side-by-side
Showing with 45 additions and 18 deletions
backend/main.py
View file @ 5621025c
......@@ -384,7 +384,10 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
try:
if hasattr(function_module, "inlet"):
data = function_module.inlet(
inlet = function_module.inlet
if inspect.iscoroutinefunction(inlet):
data = await inlet(
data,
{
"id": user.id,
......@@ -393,6 +396,17 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
"role": user.role,
},
)
else:
data = inlet(
data,
{
"id": user.id,
"email": user.email,
"name": user.name,
"role": user.role,
},
)
except Exception as e:
print(f"Error: {e}")
return JSONResponse(
......@@ -1007,7 +1021,9 @@ async def chat_completed(form_data: dict, user=Depends(get_verified_user)):
try:
if hasattr(function_module, "outlet"):
data = function_module.outlet(
outlet = function_module.outlet
if inspect.iscoroutinefunction(outlet):
data = await outlet(
data,
{
"id": user.id,
......@@ -1016,6 +1032,17 @@ async def chat_completed(form_data: dict, user=Depends(get_verified_user)):
"role": user.role,
},
)
else:
data = outlet(
data,
{
"id": user.id,
"email": user.email,
"name": user.name,
"role": user.role,
},
)
except Exception as e:
print(f"Error: {e}")
return JSONResponse(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment