@illuma-ai/agents 1.1.21 → 1.1.23

package/src/agents/__tests__/resolveStructuredOutputMode.test.js

@@ -1,117 +0,0 @@
-// src/agents/__tests__/resolveStructuredOutputMode.test.ts
-import { AgentContext } from '../AgentContext';
-import { Providers } from '@/common';
-describe('resolveStructuredOutputMode', () => {
-    const createContext = (provider, mode) => {
-        return AgentContext.fromConfig({
-            agentId: 'test-agent',
-            provider,
-            instructions: 'Test',
-            structuredOutput: {
-                schema: { type: 'object', properties: { name: { type: 'string' } } },
-                mode,
-            },
-        });
-    };
-    describe('auto mode resolution', () => {
-        it('resolves to jsonSchema for Anthropic', () => {
-            const ctx = createContext(Providers.ANTHROPIC, 'auto');
-            const { method, warnings } = ctx.resolveStructuredOutputMode();
-            expect(method).toBe('jsonSchema');
-            expect(warnings).toEqual([]);
-        });
-        it('resolves to jsonSchema for OpenAI', () => {
-            const ctx = createContext(Providers.OPENAI, 'auto');
-            const { method, warnings } = ctx.resolveStructuredOutputMode();
-            expect(method).toBe('jsonSchema');
-            expect(warnings).toEqual([]);
-        });
-        it('resolves to jsonSchema for Azure', () => {
-            const ctx = createContext(Providers.AZURE, 'auto');
-            const { method, warnings } = ctx.resolveStructuredOutputMode();
-            expect(method).toBe('jsonSchema');
-            expect(warnings).toEqual([]);
-        });
-        it('resolves to undefined (LangChain default) for Bedrock', () => {
-            const ctx = createContext(Providers.BEDROCK, 'auto');
-            const { method } = ctx.resolveStructuredOutputMode();
-            expect(method).toBeUndefined();
-        });
-        it('resolves to undefined for Google', () => {
-            const ctx = createContext(Providers.GOOGLE, 'auto');
-            const { method } = ctx.resolveStructuredOutputMode();
-            expect(method).toBeUndefined();
-        });
-        it('resolves to undefined for OpenRouter', () => {
-            const ctx = createContext(Providers.OPENROUTER, 'auto');
-            const { method } = ctx.resolveStructuredOutputMode();
-            expect(method).toBeUndefined();
-        });
-        it('defaults to auto when mode is undefined', () => {
-            const ctx = createContext(Providers.OPENAI, undefined);
-            const { method } = ctx.resolveStructuredOutputMode();
-            expect(method).toBe('jsonSchema');
-        });
-    });
-    describe('native mode resolution', () => {
-        it('resolves to jsonSchema for Anthropic', () => {
-            const ctx = createContext(Providers.ANTHROPIC, 'native');
-            const { method, warnings } = ctx.resolveStructuredOutputMode();
-            expect(method).toBe('jsonSchema');
-            expect(warnings).toEqual([]);
-        });
-        it('resolves to jsonSchema for OpenAI', () => {
-            const ctx = createContext(Providers.OPENAI, 'native');
-            const { method } = ctx.resolveStructuredOutputMode();
-            expect(method).toBe('jsonSchema');
-        });
-        it('falls back to functionCalling for Bedrock with warning', () => {
-            const ctx = createContext(Providers.BEDROCK, 'native');
-            const { method, warnings } = ctx.resolveStructuredOutputMode();
-            expect(method).toBe('functionCalling');
-            expect(warnings.length).toBe(1);
-            expect(warnings[0]).toContain('not supported');
-            expect(warnings[0]).toContain('Falling back');
-        });
-        it('falls back to functionCalling for Google with warning', () => {
-            const ctx = createContext(Providers.GOOGLE, 'native');
-            const { method, warnings } = ctx.resolveStructuredOutputMode();
-            expect(method).toBe('functionCalling');
-            expect(warnings.length).toBe(1);
-        });
-    });
-    describe('tool mode resolution', () => {
-        it('always resolves to functionCalling regardless of provider', () => {
-            const providers = [
-                Providers.ANTHROPIC,
-                Providers.OPENAI,
-                Providers.AZURE,
-                Providers.BEDROCK,
-                Providers.GOOGLE,
-            ];
-            for (const provider of providers) {
-                const ctx = createContext(provider, 'tool');
-                const { method } = ctx.resolveStructuredOutputMode();
-                expect(method).toBe('functionCalling');
-            }
-        });
-    });
-    describe('provider mode resolution', () => {
-        it('resolves to jsonMode for Anthropic', () => {
-            const ctx = createContext(Providers.ANTHROPIC, 'provider');
-            const { method } = ctx.resolveStructuredOutputMode();
-            expect(method).toBe('jsonMode');
-        });
-        it('resolves to jsonMode for OpenAI', () => {
-            const ctx = createContext(Providers.OPENAI, 'provider');
-            const { method } = ctx.resolveStructuredOutputMode();
-            expect(method).toBe('jsonMode');
-        });
-        it('resolves to functionCalling for Bedrock (no jsonMode support)', () => {
-            const ctx = createContext(Providers.BEDROCK, 'provider');
-            const { method } = ctx.resolveStructuredOutputMode();
-            expect(method).toBe('functionCalling');
-        });
-    });
-});
-//# sourceMappingURL=resolveStructuredOutputMode.test.js.map

package/src/common/enum.js

@@ -1,192 +0,0 @@
-/**
- * Enum representing the various event types emitted during the execution of runnables.
- * These events provide real-time information about the progress and state of different components.
- *
- * @enum {string}
- */
-export var GraphEvents;
-(function (GraphEvents) {
-    /* Custom Events */
-    /** [Custom] Agent update event in multi-agent graph/workflow */
-    GraphEvents["ON_AGENT_UPDATE"] = "on_agent_update";
-    /** [Custom] Delta event for run steps (message creation and tool calls) */
-    GraphEvents["ON_RUN_STEP"] = "on_run_step";
-    /** [Custom] Delta event for run steps (tool calls) */
-    GraphEvents["ON_RUN_STEP_DELTA"] = "on_run_step_delta";
-    /** [Custom] Completed event for run steps (tool calls) */
-    GraphEvents["ON_RUN_STEP_COMPLETED"] = "on_run_step_completed";
-    /** [Custom] Delta events for messages */
-    GraphEvents["ON_MESSAGE_DELTA"] = "on_message_delta";
-    /** [Custom] Reasoning Delta events for messages */
-    GraphEvents["ON_REASONING_DELTA"] = "on_reasoning_delta";
-    /** [Custom] Context analytics event for traces */
-    GraphEvents["ON_CONTEXT_ANALYTICS"] = "on_context_analytics";
-    /** [Custom] Structured output event - emitted when agent returns structured JSON */
-    GraphEvents["ON_STRUCTURED_OUTPUT"] = "on_structured_output";
-    /** [Custom] Request to execute tools - dispatched by ToolNode, handled by host */
-    GraphEvents["ON_TOOL_EXECUTE"] = "on_tool_execute";
-    /** [Custom] Context pressure event for monitoring compaction triggers */
-    GraphEvents["ON_CONTEXT_PRESSURE"] = "on_context_pressure";
-    /** [Custom] Tool requires human approval before execution - dispatched by ToolNode HITL */
-    GraphEvents["ON_TOOL_APPROVAL_REQUIRED"] = "on_tool_approval_required";
-    /* Official Events */
-    /** Custom event, emitted by system */
-    GraphEvents["ON_CUSTOM_EVENT"] = "on_custom_event";
-    /** Emitted when a chat model starts processing. */
-    GraphEvents["CHAT_MODEL_START"] = "on_chat_model_start";
-    /** Emitted when a chat model streams a chunk of its response. */
-    GraphEvents["CHAT_MODEL_STREAM"] = "on_chat_model_stream";
-    /** Emitted when a chat model completes its processing. */
-    GraphEvents["CHAT_MODEL_END"] = "on_chat_model_end";
-    /** Emitted when a language model starts processing. */
-    GraphEvents["LLM_START"] = "on_llm_start";
-    /** Emitted when a language model streams a chunk of its response. */
-    GraphEvents["LLM_STREAM"] = "on_llm_stream";
-    /** Emitted when a language model completes its processing. */
-    GraphEvents["LLM_END"] = "on_llm_end";
-    /** Emitted when a chain starts processing. */
-    GraphEvents["CHAIN_START"] = "on_chain_start";
-    /** Emitted when a chain streams a chunk of its output. */
-    GraphEvents["CHAIN_STREAM"] = "on_chain_stream";
-    /** Emitted when a chain completes its processing. */
-    GraphEvents["CHAIN_END"] = "on_chain_end";
-    /** Emitted when a tool starts its operation. */
-    GraphEvents["TOOL_START"] = "on_tool_start";
-    /** Emitted when a tool completes its operation. */
-    GraphEvents["TOOL_END"] = "on_tool_end";
-    /** Emitted when a retriever starts its operation. */
-    GraphEvents["RETRIEVER_START"] = "on_retriever_start";
-    /** Emitted when a retriever completes its operation. */
-    GraphEvents["RETRIEVER_END"] = "on_retriever_end";
-    /** Emitted when a prompt starts processing. */
-    GraphEvents["PROMPT_START"] = "on_prompt_start";
-    /** Emitted when a prompt completes its processing. */
-    GraphEvents["PROMPT_END"] = "on_prompt_end";
-})(GraphEvents || (GraphEvents = {}));
-export var Providers;
-(function (Providers) {
-    Providers["OPENAI"] = "openAI";
-    Providers["VERTEXAI"] = "vertexai";
-    Providers["BEDROCK"] = "bedrock";
-    Providers["ANTHROPIC"] = "anthropic";
-    Providers["MISTRALAI"] = "mistralai";
-    Providers["MISTRAL"] = "mistral";
-    Providers["GOOGLE"] = "google";
-    Providers["AZURE"] = "azureOpenAI";
-    Providers["DEEPSEEK"] = "deepseek";
-    Providers["OPENROUTER"] = "openrouter";
-    Providers["XAI"] = "xai";
-    Providers["MOONSHOT"] = "moonshot";
-})(Providers || (Providers = {}));
-export var GraphNodeKeys;
-(function (GraphNodeKeys) {
-    GraphNodeKeys["TOOLS"] = "tools=";
-    GraphNodeKeys["AGENT"] = "agent=";
-    GraphNodeKeys["ROUTER"] = "router";
-    GraphNodeKeys["PRE_TOOLS"] = "pre_tools";
-    GraphNodeKeys["POST_TOOLS"] = "post_tools";
-})(GraphNodeKeys || (GraphNodeKeys = {}));
-export var GraphNodeActions;
-(function (GraphNodeActions) {
-    GraphNodeActions["TOOL_NODE"] = "tool_node";
-    GraphNodeActions["CALL_MODEL"] = "call_model";
-    GraphNodeActions["ROUTE_MESSAGE"] = "route_message";
-})(GraphNodeActions || (GraphNodeActions = {}));
-export var CommonEvents;
-(function (CommonEvents) {
-    CommonEvents["LANGGRAPH"] = "LangGraph";
-})(CommonEvents || (CommonEvents = {}));
-export var StepTypes;
-(function (StepTypes) {
-    StepTypes["TOOL_CALLS"] = "tool_calls";
-    StepTypes["MESSAGE_CREATION"] = "message_creation";
-})(StepTypes || (StepTypes = {}));
-export var ContentTypes;
-(function (ContentTypes) {
-    ContentTypes["TEXT"] = "text";
-    ContentTypes["ERROR"] = "error";
-    ContentTypes["THINK"] = "think";
-    ContentTypes["TOOL_CALL"] = "tool_call";
-    ContentTypes["IMAGE_URL"] = "image_url";
-    ContentTypes["IMAGE_FILE"] = "image_file";
-    /** Anthropic */
-    ContentTypes["THINKING"] = "thinking";
-    /** Vertex AI / Google Common */
-    ContentTypes["REASONING"] = "reasoning";
-    /** Multi-Agent Switch */
-    ContentTypes["AGENT_UPDATE"] = "agent_update";
-    /** Bedrock */
-    ContentTypes["REASONING_CONTENT"] = "reasoning_content";
-})(ContentTypes || (ContentTypes = {}));
-export var ToolCallTypes;
-(function (ToolCallTypes) {
-    ToolCallTypes["FUNCTION"] = "function";
-    ToolCallTypes["RETRIEVAL"] = "retrieval";
-    ToolCallTypes["FILE_SEARCH"] = "file_search";
-    ToolCallTypes["CODE_INTERPRETER"] = "code_interpreter";
-    /* Agents Tool Call */
-    ToolCallTypes["TOOL_CALL"] = "tool_call";
-})(ToolCallTypes || (ToolCallTypes = {}));
-export var Callback;
-(function (Callback) {
-    Callback["TOOL_ERROR"] = "handleToolError";
-    Callback["TOOL_START"] = "handleToolStart";
-    Callback["TOOL_END"] = "handleToolEnd";
-    Callback["CUSTOM_EVENT"] = "handleCustomEvent";
-    /*
-    LLM_START = 'handleLLMStart',
-    LLM_NEW_TOKEN = 'handleLLMNewToken',
-    LLM_ERROR = 'handleLLMError',
-    LLM_END = 'handleLLMEnd',
-    CHAT_MODEL_START = 'handleChatModelStart',
-    CHAIN_START = 'handleChainStart',
-    CHAIN_ERROR = 'handleChainError',
-    CHAIN_END = 'handleChainEnd',
-    TEXT = 'handleText',
-    AGENT_ACTION = 'handleAgentAction',
-    AGENT_END = 'handleAgentEnd',
-    RETRIEVER_START = 'handleRetrieverStart',
-    RETRIEVER_END = 'handleRetrieverEnd',
-    RETRIEVER_ERROR = 'handleRetrieverError',
-    */
-})(Callback || (Callback = {}));
-export var Constants;
-(function (Constants) {
-    Constants["OFFICIAL_CODE_BASEURL"] = "https://api.illuma.ai/v1";
-    Constants["EXECUTE_CODE"] = "execute_code";
-    Constants["TOOL_SEARCH"] = "tool_search";
-    Constants["PROGRAMMATIC_TOOL_CALLING"] = "run_tools_with_code";
-    Constants["WEB_SEARCH"] = "web_search";
-    Constants["CONTENT_AND_ARTIFACT"] = "content_and_artifact";
-    Constants["LC_TRANSFER_TO_"] = "lc_transfer_to_";
-    /** Delimiter for MCP tools: toolName_mcp_serverName */
-    Constants["MCP_DELIMITER"] = "_mcp_";
-})(Constants || (Constants = {}));
-export var TitleMethod;
-(function (TitleMethod) {
-    TitleMethod["STRUCTURED"] = "structured";
-    TitleMethod["FUNCTIONS"] = "functions";
-    TitleMethod["COMPLETION"] = "completion";
-})(TitleMethod || (TitleMethod = {}));
-export var EnvVar;
-(function (EnvVar) {
-    EnvVar["CODE_API_KEY"] = "CODE_EXECUTOR_API_KEY";
-    EnvVar["CODE_BASEURL"] = "CODE_EXECUTOR_BASEURL";
-})(EnvVar || (EnvVar = {}));
-/**
- * Message type identifiers used by LangChain's BaseMessage.getType().
- * Use these constants instead of instanceof checks to avoid module mismatch issues
- * when different copies of @langchain/core/messages are loaded.
- */
-export var MessageTypes;
-(function (MessageTypes) {
-    MessageTypes["HUMAN"] = "human";
-    MessageTypes["AI"] = "ai";
-    MessageTypes["SYSTEM"] = "system";
-    MessageTypes["TOOL"] = "tool";
-    MessageTypes["FUNCTION"] = "function";
-    MessageTypes["GENERIC"] = "generic";
-    MessageTypes["DEVELOPER"] = "developer";
-    MessageTypes["REMOVE"] = "remove";
-})(MessageTypes || (MessageTypes = {}));
-//# sourceMappingURL=enum.js.map

package/src/common/index.js

@@ -1,3 +0,0 @@
-// src/common/index.ts
-export * from './enum';
-//# sourceMappingURL=index.js.map

package/src/events.js

@@ -1,166 +0,0 @@
-import { handleToolCalls } from '@/tools/handlers';
-import { Constants, Providers } from '@/common';
-export class HandlerRegistry {
-    handlers = new Map();
-    register(eventType, handler) {
-        this.handlers.set(eventType, handler);
-    }
-    getHandler(eventType) {
-        return this.handlers.get(eventType);
-    }
-}
-export class ModelEndHandler {
-    collectedUsage;
-    constructor(collectedUsage) {
-        if (collectedUsage && !Array.isArray(collectedUsage)) {
-            throw new Error('collectedUsage must be an array');
-        }
-        this.collectedUsage = collectedUsage;
-    }
-    async handle(event, data, metadata, graph) {
-        if (!graph || !metadata) {
-            console.warn(`Graph or metadata not found in ${event} event`);
-            return;
-        }
-        const usage = data?.output?.usage_metadata;
-        if (usage != null && this.collectedUsage != null) {
-            this.collectedUsage.push(usage);
-        }
-        if (metadata.ls_provider === 'FakeListChatModel') {
-            return handleToolCalls(data?.output?.tool_calls, metadata, graph);
-        }
-        console.log(`====== ${event.toUpperCase()} ======`);
-        console.dir({
-            usage,
-        }, { depth: null });
-        const agentContext = graph.getAgentContext(metadata);
-        if (agentContext.provider !== Providers.GOOGLE &&
-            agentContext.provider !== Providers.BEDROCK) {
-            return;
-        }
-        await handleToolCalls(data?.output?.tool_calls, metadata, graph);
-    }
-}
-export class ToolEndHandler {
-    callback;
-    logger;
-    constructor(callback, logger) {
-        this.callback = callback;
-        this.logger = logger;
-    }
-    /**
-     * Handles on_tool_end events from the for-await stream consumer.
-     *
-     * This handler is now purely a consumer callback — tool completion
-     * (ON_RUN_STEP_COMPLETED dispatch + session context storage) is handled
-     * in graph context by ToolNode directly, eliminating the race between
-     * the stream consumer and graph execution.
-     */
-    async handle(event, data, metadata, graph) {
-        try {
-            if (!graph || !metadata) {
-                if (this.logger) {
-                    this.logger.warn(`Graph or metadata not found in ${event} event`);
-                }
-                else {
-                    console.warn(`Graph or metadata not found in ${event} event`);
-                }
-                return;
-            }
-            const toolEndData = data;
-            if (!toolEndData?.output) {
-                if (this.logger) {
-                    this.logger.warn('No output found in tool_end event');
-                }
-                else {
-                    console.warn('No output found in tool_end event');
-                }
-                return;
-            }
-            if (metadata[Constants.PROGRAMMATIC_TOOL_CALLING] === true) {
-                return;
-            }
-            if (this.callback) {
-                await this.callback(toolEndData, metadata);
-            }
-        }
-        catch (error) {
-            if (this.logger) {
-                this.logger.error('Error handling tool_end event:', error);
-            }
-            else {
-                console.error('Error handling tool_end event:', error);
-            }
-        }
-    }
-}
-export class TestLLMStreamHandler {
-    handle(event, data) {
-        const chunk = data?.chunk;
-        const isMessageChunk = !!(chunk && 'message' in chunk);
-        const msg = isMessageChunk ? chunk.message : undefined;
-        if (msg && msg.tool_call_chunks && msg.tool_call_chunks.length > 0) {
-            console.log(msg.tool_call_chunks);
-        }
-        else if (msg && msg.content) {
-            if (typeof msg.content === 'string') {
-                process.stdout.write(msg.content);
-            }
-        }
-    }
-}
-export class TestChatStreamHandler {
-    handle(event, data) {
-        const chunk = data?.chunk;
-        const isContentChunk = !!(chunk && 'content' in chunk);
-        const content = isContentChunk && chunk.content;
-        if (!content || !isContentChunk) {
-            return;
-        }
-        if (chunk.tool_call_chunks && chunk.tool_call_chunks.length > 0) {
-            console.dir(chunk.tool_call_chunks, { depth: null });
-        }
-        if (typeof content === 'string') {
-            process.stdout.write(content);
-        }
-        else {
-            console.dir(content, { depth: null });
-        }
-    }
-}
-export class LLMStreamHandler {
-    handle(event, data, metadata) {
-        const chunk = data?.chunk;
-        const isMessageChunk = !!(chunk && 'message' in chunk);
-        const msg = isMessageChunk && chunk.message;
-        if (metadata) {
-            console.log(metadata);
-        }
-        if (msg && msg.tool_call_chunks && msg.tool_call_chunks.length > 0) {
-            console.log(msg.tool_call_chunks);
-        }
-        else if (msg && msg.content) {
-            if (typeof msg.content === 'string') {
-                // const text_delta = msg.content;
-                // dispatchCustomEvent(GraphEvents.CHAT_MODEL_STREAM, { chunk }, config);
-                process.stdout.write(msg.content);
-            }
-        }
-    }
-}
-export const createMetadataAggregator = (_collected) => {
-    const collected = _collected || [];
-    const handleLLMEnd = (output) => {
-        const { generations } = output;
-        const lastMessageOutput = generations[generations.length - 1]?.[0];
-        if (!lastMessageOutput) {
-            return;
-        }
-        const { message } = lastMessageOutput;
-        if (message?.response_metadata) {
-            collected.push(message.response_metadata);
-        }
-    };
-    return { handleLLMEnd, collected };
-};
-//# sourceMappingURL=events.js.map