From 0af81776c7e57dfed676ccd3e2d609aa0cce0501 Mon Sep 17 00:00:00 2001
From: Dalton Flanagan <6599399+dltn@users.noreply.github.com>
Date: Tue, 3 Sep 2024 13:11:40 -0400
Subject: [PATCH] fix for incomplete SSE type generation

---
 llama_toolchain/agentic_system/api/endpoints.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/llama_toolchain/agentic_system/api/endpoints.py b/llama_toolchain/agentic_system/api/endpoints.py
index 1f6bdcc9d..06a7323ea 100644
--- a/llama_toolchain/agentic_system/api/endpoints.py
+++ b/llama_toolchain/agentic_system/api/endpoints.py
@@ -50,9 +50,7 @@ class AgenticSystemTurnCreateRequest(BaseModel):
     override_config: Optional[AgenticSystemInstanceConfig] = None
 
 
-@json_schema_type(
-    schema={"description": "Server side event (SSE) stream of these events"}
-)
+@json_schema_type
 class AgenticSystemTurnResponseStreamChunk(BaseModel):
     event: AgenticSystemTurnResponseEvent