diff --git a/tee_gateway/controllers/chat_controller.py b/tee_gateway/controllers/chat_controller.py index 49fb406..e6d9337 100644 --- a/tee_gateway/controllers/chat_controller.py +++ b/tee_gateway/controllers/chat_controller.py @@ -272,7 +272,10 @@ def _create_non_streaming_response(chat_request: CreateChatCompletionRequest): except Exception as e: logger.error(f"Chat completion error: {str(e)}", exc_info=True) - return {"error": "Request processing failed"}, 500 + return { + "error": "Request processing failed", + "exception_type": type(e).__name__, + }, 500 def _create_streaming_response(chat_request: CreateChatCompletionRequest): @@ -598,7 +601,7 @@ def generate(): except Exception as e: logger.error(f"Streaming error: {str(e)}", exc_info=True) - yield f"data: {json.dumps({'error': 'Stream processing failed'})}\n\n" + yield f"data: {json.dumps({'error': 'Stream processing failed', 'exception_type': type(e).__name__})}\n\n" return Response( generate(), @@ -611,7 +614,10 @@ def generate(): except Exception as e: logger.error(f"Stream setup error: {str(e)}", exc_info=True) - return {"error": "Stream setup failed"}, 500 + return { + "error": "Stream setup failed", + "exception_type": type(e).__name__, + }, 500 # --------------------------------------------------------------------------- diff --git a/tee_gateway/controllers/completions_controller.py b/tee_gateway/controllers/completions_controller.py index ba941fe..78b9314 100644 --- a/tee_gateway/controllers/completions_controller.py +++ b/tee_gateway/controllers/completions_controller.py @@ -79,4 +79,7 @@ def create_completion(body): except Exception as e: logger.error(f"Completion error: {str(e)}", exc_info=True) - return {"error": "Request processing failed"}, 500 + return { + "error": "Request processing failed", + "exception_type": type(e).__name__, + }, 500