Refactor tool handling: remove UI tool checks (#1846)

Fixes OPS-3293

Author: rSnapkoOpenOps

packages/react-ui/src/app/features/ai/lib/assistant-ui-chat-hook.ts

@@ -9,7 +9,12 @@ import { toast } from '@openops/components/ui';
 import { flowHelper, FlowVersion } from '@openops/shared';
 import { getFrontendToolDefinitions } from '@openops/ui-kit';
 import { useQuery, useQueryClient } from '@tanstack/react-query';
-import { DefaultChatTransport, ToolSet, UIMessage } from 'ai';
+import {
+  DefaultChatTransport,
+  lastAssistantMessageIsCompleteWithToolCalls,
+  ToolSet,
+  UIMessage,
+} from 'ai';
 import { t } from 'i18next';
 import { useCallback, useEffect, useMemo, useRef, useState } from 'react';
 import { aiChatApi } from '../../builder/ai-chat/lib/chat-api';
@@ -279,17 +284,7 @@ export const useAssistantChat = ({
         }
       }
     },
-    // send message automatically when there's a frontend tool call
-    sendAutomaticallyWhen: ({ messages }) => {
-      const lastMessage = messages[messages.length - 1];
-      const lastMessagePart =
-        lastMessage?.parts?.[lastMessage.parts.length - 1];
-      return (
-        lastMessagePart?.type?.includes('tool-ui') &&
-        'output' in lastMessagePart &&
-        !!lastMessagePart.output
-      );
-    },
+    sendAutomaticallyWhen: lastAssistantMessageIsCompleteWithToolCalls,
   });
 
   useEffect(() => {

packages/server/api/src/app/ai/chat/llm-stream-handler.ts

@@ -13,11 +13,7 @@ import {
   TextStreamPart,
   ToolSet,
 } from 'ai';
-import {
-  hasToolCall,
-  sanitizeMessages,
-  UI_TOOL_PREFIX,
-} from '../mcp/tool-utils';
+import { sanitizeMessages } from '../mcp/tool-utils';
 import { createVoidTool } from '../mcp/void-tool';
 import {
   addCacheControlToMessages,
@@ -92,10 +88,7 @@ export function getLLMAsyncStream(
     tools: toolsProxy,
     toolChoice,
     maxRetries: MAX_RETRIES,
-    stopWhen: [
-      stepCountIs(maxRecursionDepth),
-      hasToolCall((name) => name.startsWith(UI_TOOL_PREFIX)),
-    ],
+    stopWhen: [stepCountIs(maxRecursionDepth)],
     onStepFinish,
     onFinish: async (result) => {
       const outputText = result.text || '';