Update chat.py

chat.py

@@ -118,17 +118,25 @@ async def post_message(
 
     async def stream_generator():
         full_response = ""
-        # Pass messages list as the first positional argument to .stream()
+        # Pass messages list as positional 'input' to .stream()
         for chunk in llm.stream(messages):
-            # adapt to ChatGroq response structure
-            content = (
-                chunk.get("content")
-                or chunk.get("choices", [{}])[0].get("delta", {}).get("content")
-            )
-            if content:
-                yield content
-                full_response += content
-        # save final AI message
+            # 1) Try AIMessageChunk.content
+            content = getattr(chunk, "content", None)
+            # 2) Fallback to dict-based chunk
+            if content is None and isinstance(chunk, dict):
+                content = (
+                    chunk.get("content")
+                    or chunk.get("choices", [{}])[0]
+                             .get("delta", {})
+                             .get("content")
+                )
+            if not content:
+                continue
+            # Yield and accumulate
+            yield content
+            full_response += content
+
+        # Save final AI message
         history.add_ai_message(full_response)
 
     return StreamingResponse(stream_generator(), media_type="text/plain")