fix: fix typo in the response format for the POST SSE

letta-ai · Nov 14, 2024 · effb7b2 · effb7b2
1 parent b7f2d14
commit effb7b2
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/letta/server/rest_api/routers/v1/agents.py b/letta/server/rest_api/routers/v1/agents.py
@@ -376,8 +376,14 @@ def update_message(
         200: {
             "description": "Successful response",
             "content": {
-                "application/json": {"$ref": "#/components/schemas/LettaResponse"},  # Use model_json_schema() instead of model directly
-                "text/event-stream": {"description": "Server-Sent Events stream"},
+                "application/json": {
+                    "schema": {
+                        "$ref": "#/components/schemas/LettaResponse",
+                    },
+                },
+                "text/event-stream": {
+                    "description": "Server-Sent Events stream",
+                },
             },
         }
     },