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

fix

parent df71d7c6
......@@ -399,7 +399,7 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
# Get the signature of the function
sig = inspect.signature(inlet)
param = {"body": data}
params = {"body": data}
if "__user__" in sig.parameters:
__user__ = {
......@@ -424,15 +424,15 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
params = {**params, "__user__": __user__}
if "__id__" in sig.parameters:
param = {
**param,
params = {
**params,
"__id__": filter_id,
}
if inspect.iscoroutinefunction(inlet):
data = await inlet(**param)
data = await inlet(**params)
else:
data = inlet(**param)
data = inlet(**params)
except Exception as e:
print(f"Error: {e}")
......@@ -962,17 +962,17 @@ async def generate_chat_completions(form_data: dict, user=Depends(get_verified_u
try:
if inspect.iscoroutinefunction(pipe):
res = await pipe(**param)
res = await pipe(**params)
else:
res = pipe(**param)
res = pipe(**params)
except Exception as e:
print(f"Error: {e}")
return {"error": {"detail": str(e)}}
if inspect.iscoroutinefunction(pipe):
res = await pipe(**param)
res = await pipe(**params)
else:
res = pipe(**param)
res = pipe(**params)
if isinstance(res, dict):
return res
......@@ -1104,7 +1104,7 @@ async def chat_completed(form_data: dict, user=Depends(get_verified_user)):
# Get the signature of the function
sig = inspect.signature(outlet)
param = {"body": data}
params = {"body": data}
if "__user__" in sig.parameters:
__user__ = {
......@@ -1127,15 +1127,15 @@ async def chat_completed(form_data: dict, user=Depends(get_verified_user)):
params = {**params, "__user__": __user__}
if "__id__" in sig.parameters:
param = {
**param,
params = {
**params,
"__id__": filter_id,
}
if inspect.iscoroutinefunction(outlet):
data = await outlet(**param)
data = await outlet(**params)
else:
data = outlet(**param)
data = outlet(**params)
except Exception as e:
print(f"Error: {e}")
......
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