Fix handling of messages from Anthropic models. AIMessageChunk content generated is a list of dict and message contains tool_calls

Author: StreetLamb

backend/app/core/graph/messages.py

@@ -44,15 +44,21 @@ def event_to_response(event: StreamEvent) -> ChatResponse | None:
     id = event["run_id"]
     if kind == "on_chat_model_stream":
         name = event["metadata"]["langgraph_node"]
-        message_chunk = event["data"]["chunk"]
-        content: str = message_chunk.content
+        message_chunk: AIMessageChunk = event["data"]["chunk"]
         type = get_message_type(message_chunk)
+        content: str = ""
+        if isinstance(message_chunk.content, list):
+            for c in message_chunk.content:
+                if isinstance(c, str):
+                    content += c
+                elif isinstance(c, dict):
+                    content += c.get("text", "")
+        else:
+            content = message_chunk.content
+        tool_calls = message_chunk.tool_calls
         if content and type:
             return ChatResponse(
-                type=type,
-                id=id,
-                name=name,
-                content=content,
+                type=type, id=id, name=name, content=content, tool_calls=tool_calls
             )
     elif kind == "on_chat_model_end":
         message: AIMessage = event["data"]["output"]

frontend/src/components/Teams/ChatTeam.tsx

@@ -97,7 +97,7 @@ const MessageBox = ({ message, onResume }: MessageBoxProps) => {
       <Container pt={2}>
         {content && <Markdown>{content}</Markdown>}
         {tool_calls?.map((tool_call, index) => (
-          <Box key={index}>
+          <Box key={index} mt={4}>
             <Tag colorScheme="purple" mb={2}>
               {tool_call.name}
             </Tag>
@@ -269,7 +269,7 @@ const ChatTeam = () => {
           // only content is streamable in chunks
           content: currentMessage.content
             ? currentMessage.content + response.content
-            : null,
+            : "",
           tool_output: response.tool_output,
         }
       } else {