sparkecoder 0.1.53 → 0.1.55

package/web/.next/standalone/web/src/components/chat-interface.tsx

@@ -1059,15 +1059,40 @@ export function ChatInterface({ session, isEmbed = false }: ChatInterfaceProps)
           setCurrentText('');
         }
         
-        // Create a new streaming tool call
-        const streamingToolCall: ToolCallInfo = {
-          toolCallId: event.toolCallId,
-          toolName: event.toolName,
-          input: {},
-          status: 'streaming' as const,
-          streamingArgsText: '',
-        };
-        toolCallsRef.current = [...toolCallsRef.current, streamingToolCall];
+        // Check if there's already a placeholder for this tool type
+        // (created from early tool-progress events before this SSE arrives)
+        {
+          const placeholderPrefix = `${event.toolName}_pending_`;
+          const existingPlaceholder = toolCallsRef.current.find(
+            tc => tc.toolName === event.toolName && tc.toolCallId.startsWith(placeholderPrefix)
+          );
+          const alreadyExists = toolCallsRef.current.some(tc => tc.toolCallId === event.toolCallId);
+          
+          if (existingPlaceholder) {
+            // Replace placeholder with real tool call, preserving accumulated data
+            toolCallsRef.current = toolCallsRef.current.map((tc) => {
+              if (tc.toolCallId !== existingPlaceholder.toolCallId) return tc;
+              return {
+                ...tc,
+                toolCallId: event.toolCallId,
+                toolName: event.toolName,
+                status: 'streaming' as const,
+                streamingArgsText: '',
+              };
+            });
+          } else if (!alreadyExists) {
+            // Create a new streaming tool call
+            const streamingToolCall: ToolCallInfo = {
+              toolCallId: event.toolCallId,
+              toolName: event.toolName,
+              input: {},
+              status: 'streaming' as const,
+              streamingArgsText: '',
+            };
+            toolCallsRef.current = [...toolCallsRef.current, streamingToolCall];
+          }
+          // If alreadyExists, do nothing - tool-input-available will update it
+        }
         setCurrentToolCalls([...toolCallsRef.current]);
         break;
 
@@ -1118,19 +1143,20 @@ export function ChatInterface({ session, isEmbed = false }: ChatInterfaceProps)
           setCurrentText('');
         }
         
-        // Create/update the tool call info, preserving live content from placeholder
+        // Create/update the tool call info, preserving live content from any source
+        // (placeholder, existing streaming entry, or tool-input-start entry)
         const toolCallInfo: ToolCallInfo = {
           toolCallId: event.toolCallId,
           toolName: event.toolName,
           input: event.input,
           status: 'running' as const,
           streamingArgsText: existingToolCall?.streamingArgsText,
-          // Preserve live content from placeholder if it exists
-          liveContent: placeholderWriteFile?.liveContent,
-          liveOldString: placeholderWriteFile?.liveOldString,
-          liveNewString: placeholderWriteFile?.liveNewString,
-          // Preserve explore steps from placeholder if it exists
-          exploreSteps: placeholderExplore?.exploreSteps,
+          // Preserve live content from whatever source has it
+          liveContent: placeholderWriteFile?.liveContent || existingToolCall?.liveContent,
+          liveOldString: placeholderWriteFile?.liveOldString || existingToolCall?.liveOldString,
+          liveNewString: placeholderWriteFile?.liveNewString || existingToolCall?.liveNewString,
+          // Preserve explore steps from whatever source has them
+          exploreSteps: placeholderExplore?.exploreSteps || existingToolCall?.exploreSteps,
         };
         
         // Update toolCallsRef