Browse Source

fix: raise error

Timothy J. Baek 10 months ago
parent
commit
96d7c3e99f
1 changed files with 3 additions and 6 deletions
  1. 3 6
      backend/main.py

+ 3 - 6
backend/main.py

@@ -330,11 +330,11 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
             body = await request.body()
             body_str = body.decode("utf-8")
             data = json.loads(body_str) if body_str else {}
+
             user = get_current_user(
                 request,
                 get_http_authorization_cred(request.headers.get("Authorization")),
             )
-
             # Flag to skip RAG completions if file_handler is present in tools/functions
             skip_files = False
 
@@ -346,8 +346,6 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
                 )
             model = app.state.MODELS[model_id]
 
-            print(":", data)
-
             # Check if the model has any filters
             for filter_id in model["info"]["meta"].get("filterIds", []):
                 filter = Functions.get_function_by_id(filter_id)
@@ -377,12 +375,11 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
                             )
                     except Exception as e:
                         print(f"Error: {e}")
-                        raise HTTPException(
+                        return JSONResponse(
                             status_code=status.HTTP_400_BAD_REQUEST,
-                            detail=e,
+                            content={"detail": str(e)},
                         )
 
-            print("Filtered:", data)
             # Set the task model
             task_model_id = data["model"]
             # Check if the user has a custom task model and use that model