Function execute

worker/src/components/ai/ai-agent.ts:542–770

({ inputs, params }, context)

    },
  },
  async execute({ inputs, params }, context) {
    const { userInput, conversationState, chatModel, modelApiKey } = inputs;
    const { systemPrompt, temperature, maxTokens, memorySize, stepLimit } = params;

    const agentRunId = `${context.runId}:${context.componentRef}:${randomUUID()}`;

    const agentStream = new AgentStreamRecorder(context, agentRunId);
    const { connectedToolNodeIds, organizationId } = context.metadata;
    const aiSdkOverrides = (context.metadata as { aiSdkOverrides?: AiSdkOverrides }).aiSdkOverrides;
    const createMCPClientImpl = aiSdkOverrides?.createMCPClient ?? createMCPClient;
    const ToolLoopAgentImpl = aiSdkOverrides?.ToolLoopAgent ?? ToolLoopAgent;
    const stepCountIsImpl = aiSdkOverrides?.stepCountIs ?? stepCountIs;
    const createOpenAIImpl = aiSdkOverrides?.createOpenAI ?? createOpenAI;
    const createGoogleGenerativeAIImpl =
      aiSdkOverrides?.createGoogleGenerativeAI ?? createGoogleGenerativeAI;

    let discoveredTools: ToolSet = {};
    let closeDiscovery: (() => Promise<void>) | undefined;

    if (connectedToolNodeIds && connectedToolNodeIds.length > 0) {
      context.logger.info(
        `Discovering tools from gateway for nodes: ${connectedToolNodeIds.join(', ')}`,
      );
      try {
        const sessionToken = await getGatewaySessionToken(
          context.runId,
          organizationId ?? null,
          connectedToolNodeIds,
        );
        const discoveryResult = await registerGatewayTools({
          gatewayUrl: DEFAULT_GATEWAY_URL,
          sessionToken,
          createClient: createMCPClientImpl,
        });
        discoveredTools = discoveryResult.tools;
        closeDiscovery = discoveryResult.close;
      } catch (error) {
        context.logger.error(`Failed to discover tools from gateway: ${error}`);
      }
    }

    try {
      agentStream.emitMessageStart();
      context.emitProgress({
        level: 'info',
        message: 'AI agent session started',
        data: {
          agentRunId,
          agentStatus: 'started',
        },
      });

      const trimmedInput = userInput.trim();

      if (!trimmedInput) {
        throw new ValidationError('AI Agent requires a non-empty user input.', {
          fieldErrors: { userInput: ['Input cannot be empty'] },
        });