feat: making some tool messages not use the streaming to avoid weird behavior

Author: wladimiiir

packages/common/src/tools.ts

@@ -63,6 +63,16 @@ export const MEMORY_TOOL_DELETE = 'delete_memory';
 export const MEMORY_TOOL_LIST = 'list_memories';
 export const MEMORY_TOOL_UPDATE = 'update_memory';
 
+export const STREAMING_DISABLED_TOOLS: ReadonlySet<string> = new Set([
+  `${POWER_TOOL_GROUP_NAME}${TOOL_GROUP_NAME_SEPARATOR}${POWER_TOOL_FILE_READ}`,
+  `${POWER_TOOL_GROUP_NAME}${TOOL_GROUP_NAME_SEPARATOR}${POWER_TOOL_GREP}`,
+  `${POWER_TOOL_GROUP_NAME}${TOOL_GROUP_NAME_SEPARATOR}${POWER_TOOL_GLOB}`,
+  `${POWER_TOOL_GROUP_NAME}${TOOL_GROUP_NAME_SEPARATOR}${POWER_TOOL_SEMANTIC_SEARCH}`,
+  `${MEMORY_TOOL_GROUP_NAME}${TOOL_GROUP_NAME_SEPARATOR}${MEMORY_TOOL_RETRIEVE}`,
+  `${MEMORY_TOOL_GROUP_NAME}${TOOL_GROUP_NAME_SEPARATOR}${MEMORY_TOOL_DELETE}`,
+  `${MEMORY_TOOL_GROUP_NAME}${TOOL_GROUP_NAME_SEPARATOR}${MEMORY_TOOL_LIST}`,
+]);
+
 export const MEMORY_TOOL_DESCRIPTIONS = {
   [MEMORY_TOOL_STORE]: 'Stores important information, patterns, or preferences into memory for future tasks',
   [MEMORY_TOOL_RETRIEVE]: `Searches and retrieves relevant memories using semantic vector search.

src/renderer/src/components/message/MessageBlock.tsx

@@ -21,6 +21,8 @@ import {
   POWER_TOOL_SEMANTIC_SEARCH,
   SKILLS_TOOL_ACTIVATE_SKILL,
   SKILLS_TOOL_GROUP_NAME,
+  STREAMING_DISABLED_TOOLS,
+  TOOL_GROUP_NAME_SEPARATOR,
   SUBAGENTS_TOOL_GROUP_NAME,
   SUBAGENTS_TOOL_RUN_TASK,
   TASKS_TOOL_GROUP_NAME,
@@ -170,6 +172,19 @@ const MessageBlockComponent = ({
   if (isToolMessage(message)) {
     const toolMessage = message as ToolMessage;
 
+    if (toolMessage.isStreaming === true && STREAMING_DISABLED_TOOLS.has(`${toolMessage.serverName}${TOOL_GROUP_NAME_SEPARATOR}${toolMessage.toolName}`)) {
+      return (
+        <ToolMessageBlock
+          message={toolMessage}
+          onRemove={remove}
+          compact={compact}
+          hideMessageBar={hideMessageBar}
+          onFork={onFork}
+          onRemoveUpTo={onRemoveUpTo}
+        />
+      );
+    }
+
     switch (toolMessage.serverName) {
       case POWER_TOOL_GROUP_NAME:
         switch (toolMessage.toolName) {

src/renderer/src/components/message/ToolMessageBlock.tsx

@@ -45,6 +45,10 @@ export const ToolMessageBlock = ({ message, onRemove, compact = false, onFork, o
   const getToolLabel = (message: ToolMessage): ReactNode => {
     const defaultLabel = () => t('toolMessage.toolLabel', { server: formatName(message.serverName), tool: formatName(message.toolName) });
 
+    if (isStreaming) {
+      return defaultLabel();
+    }
+
     switch (message.serverName) {
       case AIDER_TOOL_GROUP_NAME:
         switch (message.toolName) {
@@ -177,20 +181,15 @@ export const ToolMessageBlock = ({ message, onRemove, compact = false, onFork, o
 
   const content = (
     <div className="text-2xs whitespace-pre-wrap text-text-tertiary bg-bg-secondary relative p-3">
-      {Object.keys(message.args).length > 0 && (
+      {!isStreaming && Object.keys(message.args).length > 0 && (
         <div className="mb-3">
-          <div className="font-semibold mb-1 text-text-secondary">
-            {t('toolMessage.arguments')}
-            {isStreaming && <span className="ml-2 text-text-muted-light italic font-normal text-2xs">{t('toolMessage.preparing')}</span>}
-          </div>
-          <pre
-            className={`whitespace-pre-wrap max-h-[400px] overflow-y-auto scrollbar-thin scrollbar-track-bg-primary-light scrollbar-thumb-bg-secondary-light hover:scrollbar-thumb-bg-fourth bg-bg-primary-light p-2 rounded text-text-secondary text-2xs${isStreaming ? ' animate-pulse' : ''}`}
-          >
+          <div className="font-semibold mb-1 text-text-secondary">{t('toolMessage.arguments')}</div>
+          <pre className="whitespace-pre-wrap max-h-[400px] overflow-y-auto scrollbar-thin scrollbar-track-bg-primary-light scrollbar-thumb-bg-secondary-light hover:scrollbar-thumb-bg-fourth bg-bg-primary-light p-2 rounded text-text-secondary text-2xs">
             {JSON.stringify(message.args, null, 2)}
           </pre>
         </div>
       )}
-      {isStreaming && Object.keys(message.args).length === 0 ? (
+      {isStreaming ? (
         <div className="text-xs italic text-text-muted-light">{t('toolMessage.preparing')}</div>
       ) : isExecuting ? (
         <div className="text-xs italic text-text-muted-light">{t('toolMessage.executing')}</div>
@@ -215,7 +214,7 @@ export const ToolMessageBlock = ({ message, onRemove, compact = false, onFork, o
         </div>
       </div>
       {/* Tool Specific Content */}
-      {renderToolSpecificContent()}
+      {!isStreaming && renderToolSpecificContent()}
     </div>
   );