@bubblelab/bubble-core 0.1.8 → 0.1.10

package/dist/bubbles/service-bubble/ai-agent.d.ts

@@ -2,70 +2,24 @@ import { z } from 'zod';
 import { ServiceBubble } from '../../types/service-bubble-class.js';
 import type { BubbleContext } from '../../types/bubble.js';
 import { CredentialType } from '@bubblelab/shared-schemas';
-import type { BubbleName } from '@bubblelab/shared-schemas';
-export type StreamingEvent = {
-    type: 'start';
-    data: {
-        message: string;
-        maxIterations: number;
-        timestamp: string;
-    };
-} | {
-    type: 'llm_start';
-    data: {
-        model: string;
-        temperature: number;
-    };
-} | {
-    type: 'token';
-    data: {
-        content: string;
-        messageId: string;
-    };
-} | {
-    type: 'llm_complete';
-    data: {
-        messageId: string;
-        totalTokens?: number;
-    };
-} | {
-    type: 'tool_start';
-    data: {
-        tool: string;
-        input: unknown;
-        callId: string;
-    };
-} | {
-    type: 'tool_complete';
-    data: {
-        callId: string;
-        output: unknown;
-        duration: number;
-    };
-} | {
-    type: 'iteration_start';
-    data: {
-        iteration: number;
-    };
-} | {
-    type: 'iteration_complete';
-    data: {
-        iteration: number;
-        hasToolCalls: boolean;
-    };
-} | {
-    type: 'error';
-    data: {
-        error: string;
-        recoverable: boolean;
-    };
-} | {
-    type: 'complete';
-    data: {
-        result: AIAgentResult;
-        totalDuration: number;
-    };
+import type { BaseMessage } from '@langchain/core/messages';
+import { type AvailableTool } from '../../types/available-tools.js';
+import type { BubbleName, BubbleResult } from '@bubblelab/shared-schemas';
+import type { StreamingEvent } from '@bubblelab/shared-schemas';
+export type ToolHookContext = {
+    toolName: AvailableTool;
+    toolInput: unknown;
+    toolOutput?: BubbleResult<unknown>;
+    messages: BaseMessage[];
 };
+export type ToolHookAfter = (context: ToolHookContext) => Promise<{
+    messages: BaseMessage[];
+    shouldStop?: boolean;
+}>;
+export type ToolHookBefore = (context: ToolHookContext) => Promise<{
+    messages: BaseMessage[];
+    toolInput: Record<string, any>;
+}>;
 export type StreamingCallback = (event: StreamingEvent) => Promise<void> | void;
 declare const AIAgentParamsSchema: z.ZodObject<{
     message: z.ZodString;
@@ -75,8 +29,8 @@ declare const AIAgentParamsSchema: z.ZodObject<{
         mimeType: z.ZodDefault<z.ZodString>;
         description: z.ZodOptional<z.ZodString>;
     }, "strip", z.ZodTypeAny, {
-        data: string;
         type: "base64";
+        data: string;
         mimeType: string;
         description?: string | undefined;
     }, {
@@ -100,48 +54,64 @@ declare const AIAgentParamsSchema: z.ZodObject<{
     systemPrompt: z.ZodDefault<z.ZodString>;
     name: z.ZodOptional<z.ZodDefault<z.ZodString>>;
     model: z.ZodDefault<z.ZodObject<{
-        model: z.ZodDefault<z.ZodEnum<["openai/gpt-5", "openai/gpt-5-mini", "openai/gpt-o4-mini", "openai/gpt-4o", "google/gemini-2.5-pro", "google/gemini-2.5-flash", "google/gemini-2.5-flash-lite", "google/gemini-2.5-flash-image-preview", "anthropic/claude-sonnet-4-5-20250929"]>>;
+        model: z.ZodDefault<z.ZodEnum<["openai/gpt-5", "openai/gpt-5-mini", "google/gemini-2.5-pro", "google/gemini-2.5-flash", "google/gemini-2.5-flash-lite", "google/gemini-2.5-flash-image-preview", "anthropic/claude-sonnet-4-5", "anthropic/claude-haiku-4-5", "openrouter/x-ai/grok-code-fast-1", "openrouter/z-ai/glm-4.6"]>>;
         temperature: z.ZodDefault<z.ZodNumber>;
         maxTokens: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
         jsonMode: z.ZodDefault<z.ZodBoolean>;
     }, "strip", z.ZodTypeAny, {
-        model: "openai/gpt-5" | "openai/gpt-5-mini" | "openai/gpt-o4-mini" | "openai/gpt-4o" | "google/gemini-2.5-pro" | "google/gemini-2.5-flash" | "google/gemini-2.5-flash-lite" | "google/gemini-2.5-flash-image-preview" | "anthropic/claude-sonnet-4-5-20250929";
+        model: "openai/gpt-5" | "openai/gpt-5-mini" | "google/gemini-2.5-pro" | "google/gemini-2.5-flash" | "google/gemini-2.5-flash-lite" | "google/gemini-2.5-flash-image-preview" | "anthropic/claude-sonnet-4-5" | "anthropic/claude-haiku-4-5" | "openrouter/x-ai/grok-code-fast-1" | "openrouter/z-ai/glm-4.6";
         temperature: number;
         maxTokens: number;
         jsonMode: boolean;
     }, {
-        model?: "openai/gpt-5" | "openai/gpt-5-mini" | "openai/gpt-o4-mini" | "openai/gpt-4o" | "google/gemini-2.5-pro" | "google/gemini-2.5-flash" | "google/gemini-2.5-flash-lite" | "google/gemini-2.5-flash-image-preview" | "anthropic/claude-sonnet-4-5-20250929" | undefined;
+        model?: "openai/gpt-5" | "openai/gpt-5-mini" | "google/gemini-2.5-pro" | "google/gemini-2.5-flash" | "google/gemini-2.5-flash-lite" | "google/gemini-2.5-flash-image-preview" | "anthropic/claude-sonnet-4-5" | "anthropic/claude-haiku-4-5" | "openrouter/x-ai/grok-code-fast-1" | "openrouter/z-ai/glm-4.6" | undefined;
         temperature?: number | undefined;
         maxTokens?: number | undefined;
         jsonMode?: boolean | undefined;
     }>>;
     tools: z.ZodDefault<z.ZodArray<z.ZodObject<{
-        name: z.ZodEnum<["web-search-tool", "web-scrape-tool", "web-crawl-tool", "web-extract-tool", "research-agent-tool", "reddit-scrape-tool", "list-bubbles-tool", "get-bubble-details-tool", "bubbleflow-validation-tool", "chart-js-tool", "sql-query-tool"]>;
+        name: z.ZodEnum<["web-search-tool", "web-scrape-tool", "web-crawl-tool", "web-extract-tool", "research-agent-tool", "reddit-scrape-tool", "instagram-tool", "list-bubbles-tool", "get-bubble-details-tool", "bubbleflow-validation-tool", "chart-js-tool", "sql-query-tool"]>;
         credentials: z.ZodOptional<z.ZodDefault<z.ZodRecord<z.ZodNativeEnum<typeof CredentialType>, z.ZodString>>>;
         config: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
     }, "strip", z.ZodTypeAny, {
-        name: "get-bubble-details-tool" | "list-bubbles-tool" | "sql-query-tool" | "chart-js-tool" | "web-search-tool" | "web-scrape-tool" | "web-crawl-tool" | "web-extract-tool" | "research-agent-tool" | "reddit-scrape-tool" | "bubbleflow-validation-tool";
+        name: "get-bubble-details-tool" | "list-bubbles-tool" | "sql-query-tool" | "chart-js-tool" | "web-search-tool" | "web-scrape-tool" | "web-crawl-tool" | "web-extract-tool" | "research-agent-tool" | "reddit-scrape-tool" | "bubbleflow-validation-tool" | "instagram-tool";
         credentials?: Partial<Record<CredentialType, string>> | undefined;
         config?: Record<string, unknown> | undefined;
     }, {
-        name: "get-bubble-details-tool" | "list-bubbles-tool" | "sql-query-tool" | "chart-js-tool" | "web-search-tool" | "web-scrape-tool" | "web-crawl-tool" | "web-extract-tool" | "research-agent-tool" | "reddit-scrape-tool" | "bubbleflow-validation-tool";
+        name: "get-bubble-details-tool" | "list-bubbles-tool" | "sql-query-tool" | "chart-js-tool" | "web-search-tool" | "web-scrape-tool" | "web-crawl-tool" | "web-extract-tool" | "research-agent-tool" | "reddit-scrape-tool" | "bubbleflow-validation-tool" | "instagram-tool";
         credentials?: Partial<Record<CredentialType, string>> | undefined;
         config?: Record<string, unknown> | undefined;
     }>, "many">>;
+    customTools: z.ZodOptional<z.ZodDefault<z.ZodArray<z.ZodObject<{
+        name: z.ZodString;
+        description: z.ZodString;
+        schema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
+        func: z.ZodFunction<z.ZodTuple<[z.ZodRecord<z.ZodString, z.ZodUnknown>], z.ZodUnknown>, z.ZodPromise<z.ZodUnknown>>;
+    }, "strip", z.ZodTypeAny, {
+        description: string;
+        name: string;
+        schema: Record<string, unknown>;
+        func: (args_0: Record<string, unknown>, ...args: unknown[]) => Promise<unknown>;
+    }, {
+        description: string;
+        name: string;
+        schema: Record<string, unknown>;
+        func: (args_0: Record<string, unknown>, ...args: unknown[]) => Promise<unknown>;
+    }>, "many">>>;
     maxIterations: z.ZodDefault<z.ZodNumber>;
     credentials: z.ZodOptional<z.ZodRecord<z.ZodNativeEnum<typeof CredentialType>, z.ZodString>>;
     streaming: z.ZodDefault<z.ZodBoolean>;
 }, "strip", z.ZodTypeAny, {
     message: string;
     model: {
-        model: "openai/gpt-5" | "openai/gpt-5-mini" | "openai/gpt-o4-mini" | "openai/gpt-4o" | "google/gemini-2.5-pro" | "google/gemini-2.5-flash" | "google/gemini-2.5-flash-lite" | "google/gemini-2.5-flash-image-preview" | "anthropic/claude-sonnet-4-5-20250929";
+        model: "openai/gpt-5" | "openai/gpt-5-mini" | "google/gemini-2.5-pro" | "google/gemini-2.5-flash" | "google/gemini-2.5-flash-lite" | "google/gemini-2.5-flash-image-preview" | "anthropic/claude-sonnet-4-5" | "anthropic/claude-haiku-4-5" | "openrouter/x-ai/grok-code-fast-1" | "openrouter/z-ai/glm-4.6";
         temperature: number;
         maxTokens: number;
         jsonMode: boolean;
     };
     images: ({
-        data: string;
         type: "base64";
+        data: string;
         mimeType: string;
         description?: string | undefined;
     } | {
@@ -151,7 +121,7 @@ declare const AIAgentParamsSchema: z.ZodObject<{
     })[];
     systemPrompt: string;
     tools: {
-        name: "get-bubble-details-tool" | "list-bubbles-tool" | "sql-query-tool" | "chart-js-tool" | "web-search-tool" | "web-scrape-tool" | "web-crawl-tool" | "web-extract-tool" | "research-agent-tool" | "reddit-scrape-tool" | "bubbleflow-validation-tool";
+        name: "get-bubble-details-tool" | "list-bubbles-tool" | "sql-query-tool" | "chart-js-tool" | "web-search-tool" | "web-scrape-tool" | "web-crawl-tool" | "web-extract-tool" | "research-agent-tool" | "reddit-scrape-tool" | "bubbleflow-validation-tool" | "instagram-tool";
         credentials?: Partial<Record<CredentialType, string>> | undefined;
         config?: Record<string, unknown> | undefined;
     }[];
@@ -159,12 +129,18 @@ declare const AIAgentParamsSchema: z.ZodObject<{
     streaming: boolean;
     name?: string | undefined;
     credentials?: Partial<Record<CredentialType, string>> | undefined;
+    customTools?: {
+        description: string;
+        name: string;
+        schema: Record<string, unknown>;
+        func: (args_0: Record<string, unknown>, ...args: unknown[]) => Promise<unknown>;
+    }[] | undefined;
 }, {
     message: string;
     name?: string | undefined;
     credentials?: Partial<Record<CredentialType, string>> | undefined;
     model?: {
-        model?: "openai/gpt-5" | "openai/gpt-5-mini" | "openai/gpt-o4-mini" | "openai/gpt-4o" | "google/gemini-2.5-pro" | "google/gemini-2.5-flash" | "google/gemini-2.5-flash-lite" | "google/gemini-2.5-flash-image-preview" | "anthropic/claude-sonnet-4-5-20250929" | undefined;
+        model?: "openai/gpt-5" | "openai/gpt-5-mini" | "google/gemini-2.5-pro" | "google/gemini-2.5-flash" | "google/gemini-2.5-flash-lite" | "google/gemini-2.5-flash-image-preview" | "anthropic/claude-sonnet-4-5" | "anthropic/claude-haiku-4-5" | "openrouter/x-ai/grok-code-fast-1" | "openrouter/z-ai/glm-4.6" | undefined;
         temperature?: number | undefined;
         maxTokens?: number | undefined;
         jsonMode?: boolean | undefined;
@@ -181,10 +157,16 @@ declare const AIAgentParamsSchema: z.ZodObject<{
     })[] | undefined;
     systemPrompt?: string | undefined;
     tools?: {
-        name: "get-bubble-details-tool" | "list-bubbles-tool" | "sql-query-tool" | "chart-js-tool" | "web-search-tool" | "web-scrape-tool" | "web-crawl-tool" | "web-extract-tool" | "research-agent-tool" | "reddit-scrape-tool" | "bubbleflow-validation-tool";
+        name: "get-bubble-details-tool" | "list-bubbles-tool" | "sql-query-tool" | "chart-js-tool" | "web-search-tool" | "web-scrape-tool" | "web-crawl-tool" | "web-extract-tool" | "research-agent-tool" | "reddit-scrape-tool" | "bubbleflow-validation-tool" | "instagram-tool";
         credentials?: Partial<Record<CredentialType, string>> | undefined;
         config?: Record<string, unknown> | undefined;
     }[] | undefined;
+    customTools?: {
+        description: string;
+        name: string;
+        schema: Record<string, unknown>;
+        func: (args_0: Record<string, unknown>, ...args: unknown[]) => Promise<unknown>;
+    }[] | undefined;
     maxIterations?: number | undefined;
     streaming?: boolean | undefined;
 }>;
@@ -227,8 +209,16 @@ declare const AIAgentResultSchema: z.ZodObject<{
     }[];
     iterations: number;
 }>;
-type AIAgentParams = z.input<typeof AIAgentParamsSchema>;
-type AIAgentParamsParsed = z.output<typeof AIAgentParamsSchema>;
+type AIAgentParams = z.input<typeof AIAgentParamsSchema> & {
+    beforeToolCall?: ToolHookBefore;
+    afterToolCall?: ToolHookAfter;
+    streamingCallback?: StreamingCallback;
+};
+type AIAgentParamsParsed = z.output<typeof AIAgentParamsSchema> & {
+    beforeToolCall?: ToolHookBefore;
+    afterToolCall?: ToolHookAfter;
+    streamingCallback?: StreamingCallback;
+};
 type AIAgentResult = z.output<typeof AIAgentResultSchema>;
 export declare class AIAgentBubble extends ServiceBubble<AIAgentParamsParsed, AIAgentResult> {
     static readonly type: "service";
@@ -243,8 +233,8 @@ export declare class AIAgentBubble extends ServiceBubble<AIAgentParamsParsed, AI
             mimeType: z.ZodDefault<z.ZodString>;
             description: z.ZodOptional<z.ZodString>;
         }, "strip", z.ZodTypeAny, {
-            data: string;
             type: "base64";
+            data: string;
             mimeType: string;
             description?: string | undefined;
         }, {
@@ -268,48 +258,64 @@ export declare class AIAgentBubble extends ServiceBubble<AIAgentParamsParsed, AI
         systemPrompt: z.ZodDefault<z.ZodString>;
         name: z.ZodOptional<z.ZodDefault<z.ZodString>>;
         model: z.ZodDefault<z.ZodObject<{
-            model: z.ZodDefault<z.ZodEnum<["openai/gpt-5", "openai/gpt-5-mini", "openai/gpt-o4-mini", "openai/gpt-4o", "google/gemini-2.5-pro", "google/gemini-2.5-flash", "google/gemini-2.5-flash-lite", "google/gemini-2.5-flash-image-preview", "anthropic/claude-sonnet-4-5-20250929"]>>;
+            model: z.ZodDefault<z.ZodEnum<["openai/gpt-5", "openai/gpt-5-mini", "google/gemini-2.5-pro", "google/gemini-2.5-flash", "google/gemini-2.5-flash-lite", "google/gemini-2.5-flash-image-preview", "anthropic/claude-sonnet-4-5", "anthropic/claude-haiku-4-5", "openrouter/x-ai/grok-code-fast-1", "openrouter/z-ai/glm-4.6"]>>;
             temperature: z.ZodDefault<z.ZodNumber>;
             maxTokens: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
             jsonMode: z.ZodDefault<z.ZodBoolean>;
         }, "strip", z.ZodTypeAny, {
-            model: "openai/gpt-5" | "openai/gpt-5-mini" | "openai/gpt-o4-mini" | "openai/gpt-4o" | "google/gemini-2.5-pro" | "google/gemini-2.5-flash" | "google/gemini-2.5-flash-lite" | "google/gemini-2.5-flash-image-preview" | "anthropic/claude-sonnet-4-5-20250929";
+            model: "openai/gpt-5" | "openai/gpt-5-mini" | "google/gemini-2.5-pro" | "google/gemini-2.5-flash" | "google/gemini-2.5-flash-lite" | "google/gemini-2.5-flash-image-preview" | "anthropic/claude-sonnet-4-5" | "anthropic/claude-haiku-4-5" | "openrouter/x-ai/grok-code-fast-1" | "openrouter/z-ai/glm-4.6";
             temperature: number;
             maxTokens: number;
             jsonMode: boolean;
         }, {
-            model?: "openai/gpt-5" | "openai/gpt-5-mini" | "openai/gpt-o4-mini" | "openai/gpt-4o" | "google/gemini-2.5-pro" | "google/gemini-2.5-flash" | "google/gemini-2.5-flash-lite" | "google/gemini-2.5-flash-image-preview" | "anthropic/claude-sonnet-4-5-20250929" | undefined;
+            model?: "openai/gpt-5" | "openai/gpt-5-mini" | "google/gemini-2.5-pro" | "google/gemini-2.5-flash" | "google/gemini-2.5-flash-lite" | "google/gemini-2.5-flash-image-preview" | "anthropic/claude-sonnet-4-5" | "anthropic/claude-haiku-4-5" | "openrouter/x-ai/grok-code-fast-1" | "openrouter/z-ai/glm-4.6" | undefined;
             temperature?: number | undefined;
             maxTokens?: number | undefined;
             jsonMode?: boolean | undefined;
         }>>;
         tools: z.ZodDefault<z.ZodArray<z.ZodObject<{
-            name: z.ZodEnum<["web-search-tool", "web-scrape-tool", "web-crawl-tool", "web-extract-tool", "research-agent-tool", "reddit-scrape-tool", "list-bubbles-tool", "get-bubble-details-tool", "bubbleflow-validation-tool", "chart-js-tool", "sql-query-tool"]>;
+            name: z.ZodEnum<["web-search-tool", "web-scrape-tool", "web-crawl-tool", "web-extract-tool", "research-agent-tool", "reddit-scrape-tool", "instagram-tool", "list-bubbles-tool", "get-bubble-details-tool", "bubbleflow-validation-tool", "chart-js-tool", "sql-query-tool"]>;
             credentials: z.ZodOptional<z.ZodDefault<z.ZodRecord<z.ZodNativeEnum<typeof CredentialType>, z.ZodString>>>;
             config: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
         }, "strip", z.ZodTypeAny, {
-            name: "get-bubble-details-tool" | "list-bubbles-tool" | "sql-query-tool" | "chart-js-tool" | "web-search-tool" | "web-scrape-tool" | "web-crawl-tool" | "web-extract-tool" | "research-agent-tool" | "reddit-scrape-tool" | "bubbleflow-validation-tool";
+            name: "get-bubble-details-tool" | "list-bubbles-tool" | "sql-query-tool" | "chart-js-tool" | "web-search-tool" | "web-scrape-tool" | "web-crawl-tool" | "web-extract-tool" | "research-agent-tool" | "reddit-scrape-tool" | "bubbleflow-validation-tool" | "instagram-tool";
             credentials?: Partial<Record<CredentialType, string>> | undefined;
             config?: Record<string, unknown> | undefined;
         }, {
-            name: "get-bubble-details-tool" | "list-bubbles-tool" | "sql-query-tool" | "chart-js-tool" | "web-search-tool" | "web-scrape-tool" | "web-crawl-tool" | "web-extract-tool" | "research-agent-tool" | "reddit-scrape-tool" | "bubbleflow-validation-tool";
+            name: "get-bubble-details-tool" | "list-bubbles-tool" | "sql-query-tool" | "chart-js-tool" | "web-search-tool" | "web-scrape-tool" | "web-crawl-tool" | "web-extract-tool" | "research-agent-tool" | "reddit-scrape-tool" | "bubbleflow-validation-tool" | "instagram-tool";
             credentials?: Partial<Record<CredentialType, string>> | undefined;
             config?: Record<string, unknown> | undefined;
         }>, "many">>;
+        customTools: z.ZodOptional<z.ZodDefault<z.ZodArray<z.ZodObject<{
+            name: z.ZodString;
+            description: z.ZodString;
+            schema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
+            func: z.ZodFunction<z.ZodTuple<[z.ZodRecord<z.ZodString, z.ZodUnknown>], z.ZodUnknown>, z.ZodPromise<z.ZodUnknown>>;
+        }, "strip", z.ZodTypeAny, {
+            description: string;
+            name: string;
+            schema: Record<string, unknown>;
+            func: (args_0: Record<string, unknown>, ...args: unknown[]) => Promise<unknown>;
+        }, {
+            description: string;
+            name: string;
+            schema: Record<string, unknown>;
+            func: (args_0: Record<string, unknown>, ...args: unknown[]) => Promise<unknown>;
+        }>, "many">>>;
         maxIterations: z.ZodDefault<z.ZodNumber>;
         credentials: z.ZodOptional<z.ZodRecord<z.ZodNativeEnum<typeof CredentialType>, z.ZodString>>;
         streaming: z.ZodDefault<z.ZodBoolean>;
     }, "strip", z.ZodTypeAny, {
         message: string;
         model: {
-            model: "openai/gpt-5" | "openai/gpt-5-mini" | "openai/gpt-o4-mini" | "openai/gpt-4o" | "google/gemini-2.5-pro" | "google/gemini-2.5-flash" | "google/gemini-2.5-flash-lite" | "google/gemini-2.5-flash-image-preview" | "anthropic/claude-sonnet-4-5-20250929";
+            model: "openai/gpt-5" | "openai/gpt-5-mini" | "google/gemini-2.5-pro" | "google/gemini-2.5-flash" | "google/gemini-2.5-flash-lite" | "google/gemini-2.5-flash-image-preview" | "anthropic/claude-sonnet-4-5" | "anthropic/claude-haiku-4-5" | "openrouter/x-ai/grok-code-fast-1" | "openrouter/z-ai/glm-4.6";
             temperature: number;
             maxTokens: number;
             jsonMode: boolean;
         };
         images: ({
-            data: string;
             type: "base64";
+            data: string;
             mimeType: string;
             description?: string | undefined;
         } | {
@@ -319,7 +325,7 @@ export declare class AIAgentBubble extends ServiceBubble<AIAgentParamsParsed, AI
         })[];
         systemPrompt: string;
         tools: {
-            name: "get-bubble-details-tool" | "list-bubbles-tool" | "sql-query-tool" | "chart-js-tool" | "web-search-tool" | "web-scrape-tool" | "web-crawl-tool" | "web-extract-tool" | "research-agent-tool" | "reddit-scrape-tool" | "bubbleflow-validation-tool";
+            name: "get-bubble-details-tool" | "list-bubbles-tool" | "sql-query-tool" | "chart-js-tool" | "web-search-tool" | "web-scrape-tool" | "web-crawl-tool" | "web-extract-tool" | "research-agent-tool" | "reddit-scrape-tool" | "bubbleflow-validation-tool" | "instagram-tool";
             credentials?: Partial<Record<CredentialType, string>> | undefined;
             config?: Record<string, unknown> | undefined;
         }[];
@@ -327,12 +333,18 @@ export declare class AIAgentBubble extends ServiceBubble<AIAgentParamsParsed, AI
         streaming: boolean;
         name?: string | undefined;
         credentials?: Partial<Record<CredentialType, string>> | undefined;
+        customTools?: {
+            description: string;
+            name: string;
+            schema: Record<string, unknown>;
+            func: (args_0: Record<string, unknown>, ...args: unknown[]) => Promise<unknown>;
+        }[] | undefined;
     }, {
         message: string;
         name?: string | undefined;
         credentials?: Partial<Record<CredentialType, string>> | undefined;
         model?: {
-            model?: "openai/gpt-5" | "openai/gpt-5-mini" | "openai/gpt-o4-mini" | "openai/gpt-4o" | "google/gemini-2.5-pro" | "google/gemini-2.5-flash" | "google/gemini-2.5-flash-lite" | "google/gemini-2.5-flash-image-preview" | "anthropic/claude-sonnet-4-5-20250929" | undefined;
+            model?: "openai/gpt-5" | "openai/gpt-5-mini" | "google/gemini-2.5-pro" | "google/gemini-2.5-flash" | "google/gemini-2.5-flash-lite" | "google/gemini-2.5-flash-image-preview" | "anthropic/claude-sonnet-4-5" | "anthropic/claude-haiku-4-5" | "openrouter/x-ai/grok-code-fast-1" | "openrouter/z-ai/glm-4.6" | undefined;
             temperature?: number | undefined;
             maxTokens?: number | undefined;
             jsonMode?: boolean | undefined;
@@ -349,10 +361,16 @@ export declare class AIAgentBubble extends ServiceBubble<AIAgentParamsParsed, AI
         })[] | undefined;
         systemPrompt?: string | undefined;
         tools?: {
-            name: "get-bubble-details-tool" | "list-bubbles-tool" | "sql-query-tool" | "chart-js-tool" | "web-search-tool" | "web-scrape-tool" | "web-crawl-tool" | "web-extract-tool" | "research-agent-tool" | "reddit-scrape-tool" | "bubbleflow-validation-tool";
+            name: "get-bubble-details-tool" | "list-bubbles-tool" | "sql-query-tool" | "chart-js-tool" | "web-search-tool" | "web-scrape-tool" | "web-crawl-tool" | "web-extract-tool" | "research-agent-tool" | "reddit-scrape-tool" | "bubbleflow-validation-tool" | "instagram-tool";
             credentials?: Partial<Record<CredentialType, string>> | undefined;
             config?: Record<string, unknown> | undefined;
         }[] | undefined;
+        customTools?: {
+            description: string;
+            name: string;
+            schema: Record<string, unknown>;
+            func: (args_0: Record<string, unknown>, ...args: unknown[]) => Promise<unknown>;
+        }[] | undefined;
         maxIterations?: number | undefined;
         streaming?: boolean | undefined;
     }>;
@@ -399,6 +417,10 @@ export declare class AIAgentBubble extends ServiceBubble<AIAgentParamsParsed, AI
     static readonly longDescription = "\n    An AI agent powered by LangGraph that can use any tool bubble to answer questions.\n    Use cases:\n    - Add tools to enhance the AI agent's capabilities (web-search-tool, web-scrape-tool)\n    - Multi-step reasoning with tool assistance\n    - Tool-augmented conversations with any registered tool\n    - JSON mode for structured output (strips markdown formatting)\n  ";
     static readonly alias = "agent";
     private factory;
+    private beforeToolCallHook;
+    private afterToolCallHook;
+    private streamingCallback;
+    private shouldStopAfterTools;
     constructor(params?: AIAgentParams, context?: BubbleContext);
     testCredential(): Promise<boolean>;
     protected performAction(context?: BubbleContext): Promise<AIAgentResult>;
@@ -407,16 +429,12 @@ export declare class AIAgentBubble extends ServiceBubble<AIAgentParamsParsed, AI
      */
     actionWithStreaming(streamingCallback: StreamingCallback, context?: BubbleContext): Promise<AIAgentResult>;
     protected chooseCredential(): string | undefined;
-    /**
-     * Format final response with special handling for Gemini image models and JSON mode
-     */
-    private formatFinalResponse;
-    /**
-     * Convert Gemini's inlineData format to LangChain-compatible data URI format
-     */
-    private formatGeminiImageResponse;
     private initializeModel;
     private initializeTools;
+    /**
+     * Custom tool execution node that supports hooks
+     */
+    private executeToolsWithHooks;
     private createAgentGraph;
     private executeAgent;
     /**

package/dist/bubbles/service-bubble/ai-agent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ai-agent.d.ts","sourceRoot":"","sources":["../../../src/bubbles/service-bubble/ai-agent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,aAAa,EAAE,MAAM,qCAAqC,CAAC;AACpE,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAC3D,OAAO,EACL,cAAc,EAEf,MAAM,2BAA2B,CAAC;AAYnC,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAW5D,MAAM,MAAM,cAAc,GACtB;IACE,IAAI,EAAE,OAAO,CAAC;IACd,IAAI,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,aAAa,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC;CACrE,GACD;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,IAAI,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,GACnE;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,IAAI,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,GAC/D;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,IAAI,EAAE;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,GAC3E;IACE,IAAI,EAAE,YAAY,CAAC;IACnB,IAAI,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,OAAO,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC;CACxD,GACD;IACE,IAAI,EAAE,eAAe,CAAC;IACtB,IAAI,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,OAAO,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC;CAC7D,GACD;IAAE,IAAI,EAAE,iBAAiB,CAAC;IAAC,IAAI,EAAE;QAAE,SAAS,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,GACxD;IACE,IAAI,EAAE,oBAAoB,CAAC;IAC3B,IAAI,EAAE;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,YAAY,EAAE,OAAO,CAAA;KAAE,CAAC;CACpD,GACD;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,IAAI,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,OAAO,CAAA;KAAE,CAAA;CAAE,GAChE;IACE,IAAI,EAAE,UAAU,CAAC;IACjB,IAAI,EAAE;QAAE,MAAM,EAAE,aAAa,CAAC;QAAC,aAAa,EAAE,MAAM,CAAA;KAAE,CAAC;CACxD,CAAC;AAGN,MAAM,MAAM,iBAAiB,GAAG,CAAC,KAAK,EAAE,cAAc,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;AA8EhF,QAAA,MAAM,mBAAmB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA4DvB,CAAC;AACH,QAAA,MAAM,mBAAmB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAwBvB,CAAC;AAEH,KAAK,aAAa,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,mBAAmB,CAAC,CAAC;AACzD,KAAK,mBAAmB,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,mBAAmB,CAAC,CAAC;AAEhE,KAAK,aAAa,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,mBAAmB,CAAC,CAAC;AAE1D,qBAAa,aAAc,SAAQ,aAAa,CAC9C,mBAAmB,EACnB,aAAa,CACd;IACC,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAG,SAAS,CAAU;IAC1C,MAAM,CAAC,QAAQ,CAAC,OAAO,cAAc;IACrC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAG,QAAQ,CAAU;IAC7C,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAc;IACpD,MAAM,CAAC,QAAQ,CAAC,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAAuB;IAC7C,MAAM,CAAC,QAAQ,CAAC,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAAuB;IACnD,MAAM,CAAC,QAAQ,CAAC,gBAAgB,+FAC8D;IAC9F,MAAM,CAAC,QAAQ,CAAC,eAAe,+XAO7B;IACF,MAAM,CAAC,QAAQ,CAAC,KAAK,WAAW;IAEhC,OAAO,CAAC,OAAO,CAAgB;gBAG7B,MAAM,GAAE,aAGP,EACD,OAAO,CAAC,EAAE,aAAa;IAMZ,cAAc,IAAI,OAAO,CAAC,OAAO,CAAC;cAW/B,aAAa,CAC3B,OAAO,CAAC,EAAE,aAAa,GACtB,OAAO,CAAC,aAAa,CAAC;IA0CzB;;OAEG;IACU,mBAAmB,CAC9B,iBAAiB,EAAE,iBAAiB,EACpC,OAAO,CAAC,EAAE,aAAa,GACtB,OAAO,CAAC,aAAa,CAAC;IAsFzB,SAAS,CAAC,gBAAgB,IAAI,MAAM,GAAG,SAAS;IA2BhD;;OAEG;YACW,mBAAmB;IA+BjC;;OAEG;IACH,OAAO,CAAC,yBAAyB;IA6CjC,OAAO,CAAC,eAAe;YA+CT,eAAe;YA4Ef,gBAAgB;YAqDhB,YAAY;IAwQ1B;;OAEG;YACW,yBAAyB;CA0TxC"}
+{"version":3,"file":"ai-agent.d.ts","sourceRoot":"","sources":["../../../src/bubbles/service-bubble/ai-agent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,aAAa,EAAE,MAAM,qCAAqC,CAAC;AACpE,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAC3D,OAAO,EACL,cAAc,EAEf,MAAM,2BAA2B,CAAC;AAWnC,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AAG5D,OAAO,EAEL,KAAK,aAAa,EACnB,MAAM,gCAAgC,CAAC;AAExC,OAAO,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAC1E,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAQhE,MAAM,MAAM,eAAe,GAAG;IAC5B,QAAQ,EAAE,aAAa,CAAC;IACxB,SAAS,EAAE,OAAO,CAAC;IACnB,UAAU,CAAC,EAAE,YAAY,CAAC,OAAO,CAAC,CAAC;IACnC,QAAQ,EAAE,WAAW,EAAE,CAAC;CACzB,CAAC;AAGF,MAAM,MAAM,aAAa,GAAG,CAC1B,OAAO,EAAE,eAAe,KACrB,OAAO,CAAC;IAAE,QAAQ,EAAE,WAAW,EAAE,CAAC;IAAC,UAAU,CAAC,EAAE,OAAO,CAAA;CAAE,CAAC,CAAC;AAEhE,MAAM,MAAM,cAAc,GAAG,CAC3B,OAAO,EAAE,eAAe,KACrB,OAAO,CAAC;IAAE,QAAQ,EAAE,WAAW,EAAE,CAAC;IAAC,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;CAAE,CAAC,CAAC;AAG1E,MAAM,MAAM,iBAAiB,GAAG,CAAC,KAAK,EAAE,cAAc,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;AA0GhF,QAAA,MAAM,mBAAmB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAwEvB,CAAC;AACH,QAAA,MAAM,mBAAmB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAwBvB,CAAC;AAEH,KAAK,aAAa,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,mBAAmB,CAAC,GAAG;IAEzD,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;CACvC,CAAC;AACF,KAAK,mBAAmB,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,mBAAmB,CAAC,GAAG;IAChE,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;CACvC,CAAC;AAEF,KAAK,aAAa,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,mBAAmB,CAAC,CAAC;AAE1D,qBAAa,aAAc,SAAQ,aAAa,CAC9C,mBAAmB,EACnB,aAAa,CACd;IACC,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAG,SAAS,CAAU;IAC1C,MAAM,CAAC,QAAQ,CAAC,OAAO,cAAc;IACrC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAG,QAAQ,CAAU;IAC7C,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAc;IACpD,MAAM,CAAC,QAAQ,CAAC,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAAuB;IAC7C,MAAM,CAAC,QAAQ,CAAC,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAAuB;IACnD,MAAM,CAAC,QAAQ,CAAC,gBAAgB,+FAC8D;IAC9F,MAAM,CAAC,QAAQ,CAAC,eAAe,+XAO7B;IACF,MAAM,CAAC,QAAQ,CAAC,KAAK,WAAW;IAEhC,OAAO,CAAC,OAAO,CAAgB;IAC/B,OAAO,CAAC,kBAAkB,CAA6B;IACvD,OAAO,CAAC,iBAAiB,CAA4B;IACrD,OAAO,CAAC,iBAAiB,CAAgC;IACzD,OAAO,CAAC,oBAAoB,CAAS;gBAGnC,MAAM,GAAE,aAGP,EACD,OAAO,CAAC,EAAE,aAAa;IASZ,cAAc,IAAI,OAAO,CAAC,OAAO,CAAC;cAW/B,aAAa,CAC3B,OAAO,CAAC,EAAE,aAAa,GACtB,OAAO,CAAC,aAAa,CAAC;IAiDzB;;OAEG;IACU,mBAAmB,CAC9B,iBAAiB,EAAE,iBAAiB,EACpC,OAAO,CAAC,EAAE,aAAa,GACtB,OAAO,CAAC,aAAa,CAAC;IA6FzB,SAAS,CAAC,gBAAgB,IAAI,MAAM,GAAG,SAAS;IA2BhD,OAAO,CAAC,eAAe;YA+DT,eAAe;IA8G7B;;OAEG;YACW,qBAAqB;YAsHrB,gBAAgB;YA+GhB,YAAY;IAuR1B;;OAEG;YACW,yBAAyB;CA6TxC"}