diff --git a/templates/types/streaming/fastapi/app/api/routers/vercel_response.py b/templates/types/streaming/fastapi/app/api/routers/vercel_response.py
index 339ca99598db5910a20b138f82617ef5cb35be1f..83952b21e95a79111de8bc1ef94196ed5a517bc9 100644
--- a/templates/types/streaming/fastapi/app/api/routers/vercel_response.py
+++ b/templates/types/streaming/fastapi/app/api/routers/vercel_response.py
@@ -51,9 +51,7 @@ class VercelStreamResponse(StreamingResponse):
                     event_response = event.to_response()
                     yield self.convert_data(event_response)
                 else:
-                    yield self.convert_data(
-                        {"type": "agent", "data": event.model_dump()}
-                    )
+                    yield self.convert_data(event.model_dump())
 
         except asyncio.CancelledError:
             logger.warning("Client cancelled the request!")
diff --git a/templates/types/streaming/nextjs/package.json b/templates/types/streaming/nextjs/package.json
index 48c43c122f756b092a2518b04e387f5cc657e273..355fd4dfa75590be73016857669dd1e2ae94a2f9 100644
--- a/templates/types/streaming/nextjs/package.json
+++ b/templates/types/streaming/nextjs/package.json
@@ -17,7 +17,7 @@
     "@radix-ui/react-select": "^2.1.1",
     "@radix-ui/react-slot": "^1.0.2",
     "@radix-ui/react-tabs": "^1.1.0",
-    "@llamaindex/chat-ui": "0.0.14",
+    "@llamaindex/chat-ui": "0.1.0",
     "ai": "^4.0.3",
     "ajv": "^8.12.0",
     "class-variance-authority": "^0.7.1",