fix(ui): show all tool calls and results when loading session data (#1227)

When multiple tool calls are made the UI correctly reflects the calls:

<img width="1677" height="1198" alt="image"
src="https://github.com/user-attachments/assets/53e49634-7e6d-4952-ae9b-7c624603bc12"
/>

However, when the chat is refreshed only 1 of the tool calls will be
displayed.

<img width="1684" height="795" alt="image"
src="https://github.com/user-attachments/assets/d2436ca1-2406-4a60-9453-f556d2b6824f"
/>

This PR fixes this issue ensuring that all tool calls are correctly
displayed when loading an existing session.

Signed-off-by: Brian Fox <[email protected]>

Author: onematchfox

ui/src/components/chat/ToolCallDisplay.tsx

@@ -72,20 +72,27 @@ const extractToolCallRequests = (message: Message): FunctionCall[] => {
 
   // Check for stored task format first (data parts)
   const dataParts = message.parts?.filter(part => part.kind === "data") || [];
+  const functionCalls: FunctionCall[] = [];
+  
   for (const part of dataParts) {
     if (part.metadata) {
       const partMetadata = part.metadata as ADKMetadata;
       if (partMetadata?.kagent_type === "function_call") {
         const data = part.data as unknown as FunctionCall;
-        return [{
+        functionCalls.push({
           id: data.id,
           name: data.name,
           args: data.args
-        }];
+        });
       }
     }
   }
 
+  // If we found function calls in data parts, return them
+  if (functionCalls.length > 0) {
+    return functionCalls;
+  }
+  
   // Try streaming format (metadata or text content)
   const textParts = message.parts?.filter(part => part.kind === "text") || [];
   const content = textParts.map(part => (part as TextPart).text).join("");
@@ -105,6 +112,8 @@ const extractToolCallResults = (message: Message): ProcessedToolResultData[] =>
 
   // Check for stored task format first (data parts)
   const dataParts = message.parts?.filter(part => part.kind === "data") || [];
+  const toolResults: ProcessedToolResultData[] = [];
+  
   for (const part of dataParts) {
     if (part.metadata) {
       const partMetadata = part.metadata as ADKMetadata;
@@ -113,16 +122,21 @@ const extractToolCallResults = (message: Message): ProcessedToolResultData[] =>
         // Extract normalized content from the result (supports string/object/array)
         const textContent = normalizeToolResultToText(data);
 
-        return [{
+        toolResults.push({
           call_id: data.id,
           name: data.name,
           content: textContent,
           is_error: data.response?.isError || false
-        }];
+        });
       }
     }
   }
 
+  // If we found tool results in data parts, return them
+  if (toolResults.length > 0) {
+    return toolResults;
+  }
+  
   // Try streaming format (metadata or text content)
   const textParts = message.parts?.filter(part => part.kind === "text") || [];
   // eslint-disable-next-line @typescript-eslint/no-explicit-any