Reformat the error to match OpenAI's

copilot_proxy/app.py

@@ -2,12 +2,14 @@ import logging
 import os
 
 import uvicorn
-from fastapi import FastAPI, Response, HTTPException
+from fastapi import FastAPI, Request, Response
+from fastapi.responses import JSONResponse
 from sse_starlette.sse import EventSourceResponse
 
 from config.log_config import uvicorn_logger
 from models import OpenAIinput
 from utils.codegen import CodeGenProxy
+from utils.errors import FauxPilotException
 
 logging.config.dictConfig(uvicorn_logger)
 
@@ -25,6 +27,12 @@ app = FastAPI(
     swagger_ui_parameters={"defaultModelsExpandDepth": -1}
 )
 
+@app.exception_handler(FauxPilotException)
+async def fauxpilot_handler(request: Request, exc: FauxPilotException):
+    return JSONResponse(
+        status_code=400,
+        content=exc.json()
+    )
 
 @app.post("/v1/engines/codegen/completions")
 @app.post("/v1/completions")
@@ -33,9 +41,11 @@ async def completions(data: OpenAIinput):
     try:
         content = codegen(data=data)
     except codegen.TokensExceedsMaximum as E:
-        raise HTTPException(
-            status_code=400,
-            detail=str(E)
+        raise FauxPilotException(
+            message=str(E),
+            type="invalid_request_error",
+            param=None,
+            code=None,
         )
 
     if data.get("stream") is not None: