Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 9 additions & 3 deletions tee_gateway/controllers/chat_controller.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down Expand Up @@ -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(),
Expand All @@ -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


# ---------------------------------------------------------------------------
Expand Down
5 changes: 4 additions & 1 deletion tee_gateway/controllers/completions_controller.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Loading