@gravity-ui/aikit 2.0.5 → 2.0.6-beta.86aa50f22bec20e0cae460b5d3be7112f9d4efd2.0

package/build/cjs/hooks/index.d.ts

@@ -5,3 +5,4 @@ export * from "./useScrollPreservation.js";
 export * from "./useAutoCollapseOnSuccess.js";
 export * from "./useAutoCollapseOnCancelled.js";
 export * from "./useFileUploadStore.js";
+export * from "./useToolset.js";

package/build/cjs/hooks/index.js

@@ -8,4 +8,5 @@ tslib_1.__exportStar(require("./useScrollPreservation.js"), exports);
 tslib_1.__exportStar(require("./useAutoCollapseOnSuccess.js"), exports);
 tslib_1.__exportStar(require("./useAutoCollapseOnCancelled.js"), exports);
 tslib_1.__exportStar(require("./useFileUploadStore.js"), exports);
+tslib_1.__exportStar(require("./useToolset.js"), exports);
 //# sourceMappingURL=index.js.map

package/build/cjs/hooks/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"../../../src","sources":["hooks/index.ts"],"names":[],"mappings":";;;AAAA,sEAAmC;AACnC,8DAAiC;AACjC,8DAAiC;AACjC,qEAAwC;AACxC,wEAA2C;AAC3C,0EAA6C;AAC7C,kEAAqC","sourcesContent":["export * from './useDateFormatter';\nexport * from './useToolMessage';\nexport * from './useSmartScroll';\nexport * from './useScrollPreservation';\nexport * from './useAutoCollapseOnSuccess';\nexport * from './useAutoCollapseOnCancelled';\nexport * from './useFileUploadStore';\n"]}
+{"version":3,"file":"index.js","sourceRoot":"../../../src","sources":["hooks/index.ts"],"names":[],"mappings":";;;AAAA,sEAAmC;AACnC,8DAAiC;AACjC,8DAAiC;AACjC,qEAAwC;AACxC,wEAA2C;AAC3C,0EAA6C;AAC7C,kEAAqC;AACrC,0DAA6B","sourcesContent":["export * from './useDateFormatter';\nexport * from './useToolMessage';\nexport * from './useSmartScroll';\nexport * from './useScrollPreservation';\nexport * from './useAutoCollapseOnSuccess';\nexport * from './useAutoCollapseOnCancelled';\nexport * from './useFileUploadStore';\nexport * from './useToolset';\n"]}

package/build/cjs/hooks/useToolset.d.ts

@@ -0,0 +1,26 @@
+import type { Dispatch, SetStateAction } from 'react';
+import type { TChatMessage, TMessageContent } from "../types/messages.js";
+import type { MessageRendererRegistry } from "../utils/messageTypeRegistry.js";
+import { type Toolset, type ToolsetResultEvent } from "../utils/toolset.js";
+export type UseToolsetOptions<TCustom extends TMessageContent = never> = {
+    /**
+     * Called after the tool result has been merged into history.
+     * Typical use: forward the updated transcript to the model.
+     */
+    onAfterResult?: (messages: TChatMessage<TCustom>[]) => void;
+    /** Existing registry to extend instead of starting from a fresh one. */
+    registry?: MessageRendererRegistry;
+};
+export type UseToolsetReturn = {
+    messageRendererRegistry: MessageRendererRegistry;
+    handleToolResult: (event: ToolsetResultEvent) => void;
+};
+/**
+ * Wire a toolset into the chat: returns a `MessageRendererRegistry` that
+ * renders tool parts via the toolset and a `handleToolResult` callback that
+ * merges results into history and notifies via `onAfterResult`.
+ *
+ * Pass the same `toolset` reference across renders (e.g. via `useMemo`) so
+ * the registry stays stable.
+ */
+export declare function useToolset<TCustom extends TMessageContent = never>(toolset: Toolset, setMessages: Dispatch<SetStateAction<TChatMessage<TCustom>[]>>, options?: UseToolsetOptions<TCustom>): UseToolsetReturn;

package/build/cjs/hooks/useToolset.js

@@ -0,0 +1,34 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.useToolset = useToolset;
+const react_1 = require("react");
+const toolset_1 = require("../utils/toolset.js");
+/**
+ * Wire a toolset into the chat: returns a `MessageRendererRegistry` that
+ * renders tool parts via the toolset and a `handleToolResult` callback that
+ * merges results into history and notifies via `onAfterResult`.
+ *
+ * Pass the same `toolset` reference across renders (e.g. via `useMemo`) so
+ * the registry stays stable.
+ */
+function useToolset(toolset, setMessages, options) {
+    const onAfterResultRef = (0, react_1.useRef)(options === null || options === void 0 ? void 0 : options.onAfterResult);
+    (0, react_1.useEffect)(() => {
+        onAfterResultRef.current = options === null || options === void 0 ? void 0 : options.onAfterResult;
+    });
+    const handleToolResult = (0, react_1.useCallback)((event) => {
+        setMessages((prev) => {
+            const updated = (0, toolset_1.applyToolResult)(prev, event);
+            if (updated !== prev) {
+                queueMicrotask(() => { var _a; return (_a = onAfterResultRef.current) === null || _a === void 0 ? void 0 : _a.call(onAfterResultRef, updated); });
+            }
+            return updated;
+        });
+    }, [setMessages]);
+    const messageRendererRegistry = (0, react_1.useMemo)(() => (0, toolset_1.createToolsetRenderer)(toolset, {
+        onToolResult: handleToolResult,
+        registry: options === null || options === void 0 ? void 0 : options.registry,
+    }), [toolset, handleToolResult, options === null || options === void 0 ? void 0 : options.registry]);
+    return { messageRendererRegistry, handleToolResult };
+}
+//# sourceMappingURL=useToolset.js.map

package/build/cjs/hooks/useToolset.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useToolset.js","sourceRoot":"../../../src","sources":["hooks/useToolset.ts"],"names":[],"mappings":";;AAmCA,gCAiCC;AApED,iCAA8D;AAK9D,iDAK0B;AAiB1B;;;;;;;GAOG;AACH,SAAgB,UAAU,CACtB,OAAgB,EAChB,WAA8D,EAC9D,OAAoC;IAEpC,MAAM,gBAAgB,GAAG,IAAA,cAAM,EAAC,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,aAAa,CAAC,CAAC;IACxD,IAAA,iBAAS,EAAC,GAAG,EAAE;QACX,gBAAgB,CAAC,OAAO,GAAG,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,aAAa,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,MAAM,gBAAgB,GAAG,IAAA,mBAAW,EAChC,CAAC,KAAyB,EAAE,EAAE;QAC1B,WAAW,CAAC,CAAC,IAAI,EAAE,EAAE;YACjB,MAAM,OAAO,GAAG,IAAA,yBAAe,EAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAC7C,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;gBACnB,cAAc,CAAC,GAAG,EAAE,WAAC,OAAA,MAAA,gBAAgB,CAAC,OAAO,iEAAG,OAAO,CAAC,CAAA,EAAA,CAAC,CAAC;YAC9D,CAAC;YACD,OAAO,OAAO,CAAC;QACnB,CAAC,CAAC,CAAC;IACP,CAAC,EACD,CAAC,WAAW,CAAC,CAChB,CAAC;IAEF,MAAM,uBAAuB,GAAG,IAAA,eAAO,EACnC,GAAG,EAAE,CACD,IAAA,+BAAqB,EAAC,OAAO,EAAE;QAC3B,YAAY,EAAE,gBAAgB;QAC9B,QAAQ,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,QAAQ;KAC9B,CAAC,EACN,CAAC,OAAO,EAAE,gBAAgB,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,QAAQ,CAAC,CACjD,CAAC;IAEF,OAAO,EAAC,uBAAuB,EAAE,gBAAgB,EAAC,CAAC;AACvD,CAAC","sourcesContent":["import {useCallback, useEffect, useMemo, useRef} from 'react';\nimport type {Dispatch, SetStateAction} from 'react';\n\nimport type {TChatMessage, TMessageContent} from '../types/messages';\nimport type {MessageRendererRegistry} from '../utils/messageTypeRegistry';\nimport {\n    type Toolset,\n    type ToolsetResultEvent,\n    applyToolResult,\n    createToolsetRenderer,\n} from '../utils/toolset';\n\nexport type UseToolsetOptions<TCustom extends TMessageContent = never> = {\n    /**\n     * Called after the tool result has been merged into history.\n     * Typical use: forward the updated transcript to the model.\n     */\n    onAfterResult?: (messages: TChatMessage<TCustom>[]) => void;\n    /** Existing registry to extend instead of starting from a fresh one. */\n    registry?: MessageRendererRegistry;\n};\n\nexport type UseToolsetReturn = {\n    messageRendererRegistry: MessageRendererRegistry;\n    handleToolResult: (event: ToolsetResultEvent) => void;\n};\n\n/**\n * Wire a toolset into the chat: returns a `MessageRendererRegistry` that\n * renders tool parts via the toolset and a `handleToolResult` callback that\n * merges results into history and notifies via `onAfterResult`.\n *\n * Pass the same `toolset` reference across renders (e.g. via `useMemo`) so\n * the registry stays stable.\n */\nexport function useToolset<TCustom extends TMessageContent = never>(\n    toolset: Toolset,\n    setMessages: Dispatch<SetStateAction<TChatMessage<TCustom>[]>>,\n    options?: UseToolsetOptions<TCustom>,\n): UseToolsetReturn {\n    const onAfterResultRef = useRef(options?.onAfterResult);\n    useEffect(() => {\n        onAfterResultRef.current = options?.onAfterResult;\n    });\n\n    const handleToolResult = useCallback(\n        (event: ToolsetResultEvent) => {\n            setMessages((prev) => {\n                const updated = applyToolResult(prev, event);\n                if (updated !== prev) {\n                    queueMicrotask(() => onAfterResultRef.current?.(updated));\n                }\n                return updated;\n            });\n        },\n        [setMessages],\n    );\n\n    const messageRendererRegistry = useMemo(\n        () =>\n            createToolsetRenderer(toolset, {\n                onToolResult: handleToolResult,\n                registry: options?.registry,\n            }),\n        [toolset, handleToolResult, options?.registry],\n    );\n\n    return {messageRendererRegistry, handleToolResult};\n}\n"]}

package/build/cjs/package.json

@@ -1 +1 @@
-{"version":"2.0.5","type":"commonjs","sideEffects":["*.css","*.scss"]}
+{"version":"2.0.6-beta.86aa50f22bec20e0cae460b5d3be7112f9d4efd2.0","type":"commonjs","sideEffects":["*.css","*.scss"]}

package/build/cjs/utils/index.d.ts

@@ -3,3 +3,4 @@ export * from "./messageUtils.js";
 export * from "./validation.js";
 export * from "./messageTypeRegistry.js";
 export * from "./clipboardUtils.js";
+export * from "./toolset.js";

package/build/cjs/utils/index.js

@@ -7,4 +7,5 @@ tslib_1.__exportStar(require("./messageUtils.js"), exports);
 tslib_1.__exportStar(require("./validation.js"), exports);
 tslib_1.__exportStar(require("./messageTypeRegistry.js"), exports);
 tslib_1.__exportStar(require("./clipboardUtils.js"), exports);
+tslib_1.__exportStar(require("./toolset.js"), exports);
 //# sourceMappingURL=index.js.map

package/build/cjs/utils/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"../../../src","sources":["utils/index.ts"],"names":[],"mappings":";;;AAAA,mBAAmB;AACnB,yDAA4B;AAC5B,4DAA+B;AAC/B,0DAA6B;AAC7B,mEAAsC;AACtC,8DAAiC","sourcesContent":["// Export utilities\nexport * from './chatUtils';\nexport * from './messageUtils';\nexport * from './validation';\nexport * from './messageTypeRegistry';\nexport * from './clipboardUtils';\n"]}
+{"version":3,"file":"index.js","sourceRoot":"../../../src","sources":["utils/index.ts"],"names":[],"mappings":";;;AAAA,mBAAmB;AACnB,yDAA4B;AAC5B,4DAA+B;AAC/B,0DAA6B;AAC7B,mEAAsC;AACtC,8DAAiC;AACjC,uDAA0B","sourcesContent":["// Export utilities\nexport * from './chatUtils';\nexport * from './messageUtils';\nexport * from './validation';\nexport * from './messageTypeRegistry';\nexport * from './clipboardUtils';\nexport * from './toolset';\n"]}

package/build/cjs/utils/toolset.d.ts

@@ -0,0 +1,85 @@
+import type { ComponentType, ReactNode } from 'react';
+import type { TChatMessage, TMessageContent, ToolMessageContentData } from "../types/messages.js";
+import { type MessageRendererRegistry } from "./messageTypeRegistry.js";
+export type ToolSchemaResult<TArgs> = {
+    success: true;
+    data: TArgs;
+} | {
+    success: false;
+    error: {
+        message: string;
+    };
+};
+export type ToolSchema<TArgs> = {
+    validate: (input: unknown) => ToolSchemaResult<TArgs>;
+};
+export type ToolComponentProps<TArgs, TResult> = {
+    args: TArgs;
+    result?: TResult;
+    submitResult: (result: TResult) => void;
+};
+export type ToolDefinition<TArgs, TResult> = {
+    name: string;
+    description: string;
+    parameters: Record<string, unknown>;
+    schema: ToolSchema<TArgs>;
+    component: ComponentType<ToolComponentProps<TArgs, TResult>>;
+    execute?: (params: {
+        args: TArgs;
+        result: TResult;
+        toolCallId: string;
+    }) => TResult | Promise<TResult>;
+};
+export type RuntimeToolDefinition = {
+    name: string;
+    description: string;
+    parameters: Record<string, unknown>;
+    validate: (input: unknown) => ToolSchemaResult<unknown>;
+    render: (props: {
+        args: unknown;
+        result?: unknown;
+        submitResult: (result: unknown) => void;
+    }) => ReactNode;
+    execute: (params: {
+        args: unknown;
+        result: unknown;
+        toolCallId: string;
+    }) => unknown | Promise<unknown>;
+};
+export type Toolset = Record<string, RuntimeToolDefinition>;
+export type ToolPartContentData<TArgs = unknown, TResult = unknown> = ToolMessageContentData & {
+    toolCallId: string;
+    args?: TArgs;
+    result?: TResult;
+};
+export type ToolPartContent<TArgs = unknown, TResult = unknown> = TMessageContent<'tool', ToolPartContentData<TArgs, TResult>>;
+export type ToolsetResultEvent = {
+    toolCallId: string;
+    toolName: string;
+    result: unknown;
+};
+/**
+ * Wrap a typed tool definition into an erased runtime entry that the
+ * toolset renderer can dispatch by `toolName`.
+ */
+export declare function defineTool<TArgs, TResult>(definition: ToolDefinition<TArgs, TResult>): RuntimeToolDefinition;
+export type CreateToolsetRendererOptions = {
+    onToolResult: (event: ToolsetResultEvent) => void;
+    /**
+     * Existing registry to extend instead of starting from a fresh one.
+     * The `tool` renderer is overridden; other types are preserved.
+     */
+    registry?: MessageRendererRegistry;
+};
+/**
+ * Build a `MessageRendererRegistry` whose `tool` renderer dispatches
+ * by `toolName` into the provided toolset. Unknown tools and invalid
+ * args fall back to a generic `<ToolMessage status="error" />`.
+ */
+export declare function createToolsetRenderer(toolset: Toolset, options: CreateToolsetRendererOptions): MessageRendererRegistry;
+/**
+ * Merge a tool result into the matching `tool` part of the chat history.
+ * Returns the original array reference when nothing matched, so React state
+ * setters can skip needless updates.
+ */
+export declare function applyToolResult<TCustom extends TMessageContent = never>(messages: TChatMessage<TCustom>[], event: ToolsetResultEvent): TChatMessage<TCustom>[];

package/build/cjs/utils/toolset.js

@@ -0,0 +1,104 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.defineTool = defineTool;
+exports.createToolsetRenderer = createToolsetRenderer;
+exports.applyToolResult = applyToolResult;
+const jsx_runtime_1 = require("react/jsx-runtime");
+const ToolMessage_1 = require("../components/organisms/ToolMessage/index.js");
+const messageTypeRegistry_1 = require("./messageTypeRegistry.js");
+/**
+ * Wrap a typed tool definition into an erased runtime entry that the
+ * toolset renderer can dispatch by `toolName`.
+ */
+function defineTool(definition) {
+    const Component = definition.component;
+    return {
+        name: definition.name,
+        description: definition.description,
+        parameters: definition.parameters,
+        validate: definition.schema.validate,
+        render: ({ args, result, submitResult }) => ((0, jsx_runtime_1.jsx)(Component, { args: args, result: result, submitResult: submitResult })),
+        execute: ({ args, result, toolCallId }) => definition.execute
+            ? definition.execute({
+                args: args,
+                result: result,
+                toolCallId,
+            })
+            : result,
+    };
+}
+/**
+ * Build a `MessageRendererRegistry` whose `tool` renderer dispatches
+ * by `toolName` into the provided toolset. Unknown tools and invalid
+ * args fall back to a generic `<ToolMessage status="error" />`.
+ */
+function createToolsetRenderer(toolset, options) {
+    const { onToolResult, registry = (0, messageTypeRegistry_1.createMessageRendererRegistry)() } = options;
+    (0, messageTypeRegistry_1.registerMessageRenderer)(registry, 'tool', {
+        component: ({ part }) => {
+            const toolPart = part.data;
+            const toolDef = toolset[toolPart.toolName];
+            if (!toolDef) {
+                return ((0, jsx_runtime_1.jsx)(ToolMessage_1.ToolMessage, Object.assign({}, toolPart, { status: "error", expandable: true, initialExpanded: true, bodyContent: `Unknown tool: ${toolPart.toolName}` })));
+            }
+            const validation = toolDef.validate(toolPart.args);
+            if (!validation.success) {
+                return ((0, jsx_runtime_1.jsx)(ToolMessage_1.ToolMessage, Object.assign({}, toolPart, { status: "error", expandable: true, initialExpanded: true, bodyContent: validation.error.message })));
+            }
+            const submitResult = (result) => {
+                Promise.resolve(toolDef.execute({
+                    args: validation.data,
+                    result,
+                    toolCallId: toolPart.toolCallId,
+                })).then((finalResult) => {
+                    onToolResult({
+                        toolCallId: toolPart.toolCallId,
+                        toolName: toolPart.toolName,
+                        result: finalResult,
+                    });
+                });
+            };
+            return toolDef.render({
+                args: validation.data,
+                result: toolPart.result,
+                submitResult,
+            });
+        },
+    });
+    return registry;
+}
+function isToolPartContent(part) {
+    return part.type === 'tool' && typeof part.data === 'object' && part.data !== null;
+}
+function toContentParts(content) {
+    if (typeof content === 'string') {
+        return content ? [{ type: 'text', data: { text: content } }] : [];
+    }
+    return Array.isArray(content) ? content : [content];
+}
+/**
+ * Merge a tool result into the matching `tool` part of the chat history.
+ * Returns the original array reference when nothing matched, so React state
+ * setters can skip needless updates.
+ */
+function applyToolResult(messages, event) {
+    let changed = false;
+    const next = messages.map((msg) => {
+        if (msg.role !== 'assistant')
+            return msg;
+        const parts = toContentParts(msg.content);
+        const hasMatch = parts.some((part) => isToolPartContent(part) && part.data.toolCallId === event.toolCallId);
+        if (!hasMatch)
+            return msg;
+        changed = true;
+        const updatedParts = parts.map((part) => {
+            if (!isToolPartContent(part) || part.data.toolCallId !== event.toolCallId) {
+                return part;
+            }
+            return Object.assign(Object.assign({}, part), { data: Object.assign(Object.assign({}, part.data), { status: 'success', result: event.result }) });
+        });
+        return Object.assign(Object.assign({}, msg), { content: updatedParts });
+    });
+    return changed ? next : messages;
+}
+//# sourceMappingURL=toolset.js.map

package/build/cjs/utils/toolset.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"toolset.js","sourceRoot":"../../../src","sources":["utils/toolset.tsx"],"names":[],"mappings":";;AAmFA,gCA0BC;AAgBD,sDA6DC;AAoBD,0CAqCC;;AAjPD,8EAAgE;AAQhE,kEAI+B;AAiE/B;;;GAGG;AACH,SAAgB,UAAU,CACtB,UAA0C;IAE1C,MAAM,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;IAEvC,OAAO;QACH,IAAI,EAAE,UAAU,CAAC,IAAI;QACrB,WAAW,EAAE,UAAU,CAAC,WAAW;QACnC,UAAU,EAAE,UAAU,CAAC,UAAU;QACjC,QAAQ,EAAE,UAAU,CAAC,MAAM,CAAC,QAAyD;QACrF,MAAM,EAAE,CAAC,EAAC,IAAI,EAAE,MAAM,EAAE,YAAY,EAAC,EAAE,EAAE,CAAC,CACtC,uBAAC,SAAS,IACN,IAAI,EAAE,IAAa,EACnB,MAAM,EAAE,MAA6B,EACrC,YAAY,EAAE,YAAyC,GACzD,CACL;QACD,OAAO,EAAE,CAAC,EAAC,IAAI,EAAE,MAAM,EAAE,UAAU,EAAC,EAAE,EAAE,CACpC,UAAU,CAAC,OAAO;YACd,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC;gBACf,IAAI,EAAE,IAAa;gBACnB,MAAM,EAAE,MAAiB;gBACzB,UAAU;aACb,CAAC;YACJ,CAAC,CAAC,MAAM;KACnB,CAAC;AACN,CAAC;AAWD;;;;GAIG;AACH,SAAgB,qBAAqB,CACjC,OAAgB,EAChB,OAAqC;IAErC,MAAM,EAAC,YAAY,EAAE,QAAQ,GAAG,IAAA,mDAA6B,GAAE,EAAC,GAAG,OAAO,CAAC;IAE3E,IAAA,6CAAuB,EAAkB,QAAQ,EAAE,MAAM,EAAE;QACvD,SAAS,EAAE,CAAC,EAAC,IAAI,EAAC,EAAE,EAAE;YAClB,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC;YAC3B,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YAE3C,IAAI,CAAC,OAAO,EAAE,CAAC;gBACX,OAAO,CACH,uBAAC,yBAAW,oBACJ,QAAQ,IACZ,MAAM,EAAC,OAAO,EACd,UAAU,QACV,eAAe,QACf,WAAW,EAAE,iBAAiB,QAAQ,CAAC,QAAQ,EAAE,IACnD,CACL,CAAC;YACN,CAAC;YAED,MAAM,UAAU,GAAG,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YACnD,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;gBACtB,OAAO,CACH,uBAAC,yBAAW,oBACJ,QAAQ,IACZ,MAAM,EAAC,OAAO,EACd,UAAU,QACV,eAAe,QACf,WAAW,EAAE,UAAU,CAAC,KAAK,CAAC,OAAO,IACvC,CACL,CAAC;YACN,CAAC;YAED,MAAM,YAAY,GAAG,CAAC,MAAe,EAAE,EAAE;gBACrC,OAAO,CAAC,OAAO,CACX,OAAO,CAAC,OAAO,CAAC;oBACZ,IAAI,EAAE,UAAU,CAAC,IAAI;oBACrB,MAAM;oBACN,UAAU,EAAE,QAAQ,CAAC,UAAU;iBAClC,CAAC,CACL,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,EAAE;oBACnB,YAAY,CAAC;wBACT,UAAU,EAAE,QAAQ,CAAC,UAAU;wBAC/B,QAAQ,EAAE,QAAQ,CAAC,QAAQ;wBAC3B,MAAM,EAAE,WAAW;qBACtB,CAAC,CAAC;gBACP,CAAC,CAAC,CAAC;YACP,CAAC,CAAC;YAEF,OAAO,OAAO,CAAC,MAAM,CAAC;gBAClB,IAAI,EAAE,UAAU,CAAC,IAAI;gBACrB,MAAM,EAAE,QAAQ,CAAC,MAAM;gBACvB,YAAY;aACf,CAAC,CAAC;QACP,CAAC;KACJ,CAAC,CAAC;IAEH,OAAO,QAAQ,CAAC;AACpB,CAAC;AAED,SAAS,iBAAiB,CAAC,IAAqB;IAC5C,OAAO,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC;AACvF,CAAC;AAED,SAAS,cAAc,CACnB,OAA8C;IAE9C,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC,EAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAC,IAAI,EAAE,OAAO,EAAC,EAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAClE,CAAC;IACD,OAAO,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAE,OAA6B,CAAC,CAAC,CAAC,CAAC,OAA0B,CAAC,CAAC;AAClG,CAAC;AAED;;;;GAIG;AACH,SAAgB,eAAe,CAC3B,QAAiC,EACjC,KAAyB;IAEzB,IAAI,OAAO,GAAG,KAAK,CAAC;IAEpB,MAAM,IAAI,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAyB,EAAE;QACrD,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW;YAAE,OAAO,GAAG,CAAC;QAEzC,MAAM,KAAK,GAAG,cAAc,CAAU,GAAG,CAAC,OAAO,CAAC,CAAC;QACnD,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CACvB,CAAC,IAAI,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,UAAU,KAAK,KAAK,CAAC,UAAU,CACjF,CAAC;QACF,IAAI,CAAC,QAAQ;YAAE,OAAO,GAAG,CAAC;QAE1B,OAAO,GAAG,IAAI,CAAC;QACf,MAAM,YAAY,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YACpC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,UAAU,KAAK,KAAK,CAAC,UAAU,EAAE,CAAC;gBACxE,OAAO,IAAI,CAAC;YAChB,CAAC;YACD,uCACO,IAAI,KACP,IAAI,kCACG,IAAI,CAAC,IAAI,KACZ,MAAM,EAAE,SAAkB,EAC1B,MAAM,EAAE,KAAK,CAAC,MAAM,OAE1B;QACN,CAAC,CAAC,CAAC;QAEH,uCACO,GAAG,KACN,OAAO,EAAE,YAAqD,IAChE;IACN,CAAC,CAAC,CAAC;IAEH,OAAO,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC;AACrC,CAAC","sourcesContent":["import type {ComponentType, ReactNode} from 'react';\n\nimport {ToolMessage} from '../components/organisms/ToolMessage';\nimport type {\n    TAssistantMessage,\n    TChatMessage,\n    TMessageContent,\n    ToolMessageContentData,\n} from '../types/messages';\n\nimport {\n    type MessageRendererRegistry,\n    createMessageRendererRegistry,\n    registerMessageRenderer,\n} from './messageTypeRegistry';\n\nexport type ToolSchemaResult<TArgs> =\n    | {success: true; data: TArgs}\n    | {success: false; error: {message: string}};\n\nexport type ToolSchema<TArgs> = {\n    validate: (input: unknown) => ToolSchemaResult<TArgs>;\n};\n\nexport type ToolComponentProps<TArgs, TResult> = {\n    args: TArgs;\n    result?: TResult;\n    submitResult: (result: TResult) => void;\n};\n\nexport type ToolDefinition<TArgs, TResult> = {\n    name: string;\n    description: string;\n    parameters: Record<string, unknown>;\n    schema: ToolSchema<TArgs>;\n    component: ComponentType<ToolComponentProps<TArgs, TResult>>;\n    execute?: (params: {\n        args: TArgs;\n        result: TResult;\n        toolCallId: string;\n    }) => TResult | Promise<TResult>;\n};\n\nexport type RuntimeToolDefinition = {\n    name: string;\n    description: string;\n    parameters: Record<string, unknown>;\n    validate: (input: unknown) => ToolSchemaResult<unknown>;\n    render: (props: {\n        args: unknown;\n        result?: unknown;\n        submitResult: (result: unknown) => void;\n    }) => ReactNode;\n    execute: (params: {\n        args: unknown;\n        result: unknown;\n        toolCallId: string;\n    }) => unknown | Promise<unknown>;\n};\n\nexport type Toolset = Record<string, RuntimeToolDefinition>;\n\nexport type ToolPartContentData<TArgs = unknown, TResult = unknown> = ToolMessageContentData & {\n    toolCallId: string;\n    args?: TArgs;\n    result?: TResult;\n};\n\nexport type ToolPartContent<TArgs = unknown, TResult = unknown> = TMessageContent<\n    'tool',\n    ToolPartContentData<TArgs, TResult>\n>;\n\nexport type ToolsetResultEvent = {\n    toolCallId: string;\n    toolName: string;\n    result: unknown;\n};\n\n/**\n * Wrap a typed tool definition into an erased runtime entry that the\n * toolset renderer can dispatch by `toolName`.\n */\nexport function defineTool<TArgs, TResult>(\n    definition: ToolDefinition<TArgs, TResult>,\n): RuntimeToolDefinition {\n    const Component = definition.component;\n\n    return {\n        name: definition.name,\n        description: definition.description,\n        parameters: definition.parameters,\n        validate: definition.schema.validate as (input: unknown) => ToolSchemaResult<unknown>,\n        render: ({args, result, submitResult}) => (\n            <Component\n                args={args as TArgs}\n                result={result as TResult | undefined}\n                submitResult={submitResult as (result: TResult) => void}\n            />\n        ),\n        execute: ({args, result, toolCallId}) =>\n            definition.execute\n                ? definition.execute({\n                      args: args as TArgs,\n                      result: result as TResult,\n                      toolCallId,\n                  })\n                : result,\n    };\n}\n\nexport type CreateToolsetRendererOptions = {\n    onToolResult: (event: ToolsetResultEvent) => void;\n    /**\n     * Existing registry to extend instead of starting from a fresh one.\n     * The `tool` renderer is overridden; other types are preserved.\n     */\n    registry?: MessageRendererRegistry;\n};\n\n/**\n * Build a `MessageRendererRegistry` whose `tool` renderer dispatches\n * by `toolName` into the provided toolset. Unknown tools and invalid\n * args fall back to a generic `<ToolMessage status=\"error\" />`.\n */\nexport function createToolsetRenderer(\n    toolset: Toolset,\n    options: CreateToolsetRendererOptions,\n): MessageRendererRegistry {\n    const {onToolResult, registry = createMessageRendererRegistry()} = options;\n\n    registerMessageRenderer<ToolPartContent>(registry, 'tool', {\n        component: ({part}) => {\n            const toolPart = part.data;\n            const toolDef = toolset[toolPart.toolName];\n\n            if (!toolDef) {\n                return (\n                    <ToolMessage\n                        {...toolPart}\n                        status=\"error\"\n                        expandable\n                        initialExpanded\n                        bodyContent={`Unknown tool: ${toolPart.toolName}`}\n                    />\n                );\n            }\n\n            const validation = toolDef.validate(toolPart.args);\n            if (!validation.success) {\n                return (\n                    <ToolMessage\n                        {...toolPart}\n                        status=\"error\"\n                        expandable\n                        initialExpanded\n                        bodyContent={validation.error.message}\n                    />\n                );\n            }\n\n            const submitResult = (result: unknown) => {\n                Promise.resolve(\n                    toolDef.execute({\n                        args: validation.data,\n                        result,\n                        toolCallId: toolPart.toolCallId,\n                    }),\n                ).then((finalResult) => {\n                    onToolResult({\n                        toolCallId: toolPart.toolCallId,\n                        toolName: toolPart.toolName,\n                        result: finalResult,\n                    });\n                });\n            };\n\n            return toolDef.render({\n                args: validation.data,\n                result: toolPart.result,\n                submitResult,\n            });\n        },\n    });\n\n    return registry;\n}\n\nfunction isToolPartContent(part: TMessageContent): part is ToolPartContent {\n    return part.type === 'tool' && typeof part.data === 'object' && part.data !== null;\n}\n\nfunction toContentParts<TCustom extends TMessageContent>(\n    content: TAssistantMessage<TCustom>['content'],\n): TMessageContent[] {\n    if (typeof content === 'string') {\n        return content ? [{type: 'text', data: {text: content}}] : [];\n    }\n    return Array.isArray(content) ? (content as TMessageContent[]) : [content as TMessageContent];\n}\n\n/**\n * Merge a tool result into the matching `tool` part of the chat history.\n * Returns the original array reference when nothing matched, so React state\n * setters can skip needless updates.\n */\nexport function applyToolResult<TCustom extends TMessageContent = never>(\n    messages: TChatMessage<TCustom>[],\n    event: ToolsetResultEvent,\n): TChatMessage<TCustom>[] {\n    let changed = false;\n\n    const next = messages.map((msg): TChatMessage<TCustom> => {\n        if (msg.role !== 'assistant') return msg;\n\n        const parts = toContentParts<TCustom>(msg.content);\n        const hasMatch = parts.some(\n            (part) => isToolPartContent(part) && part.data.toolCallId === event.toolCallId,\n        );\n        if (!hasMatch) return msg;\n\n        changed = true;\n        const updatedParts = parts.map((part) => {\n            if (!isToolPartContent(part) || part.data.toolCallId !== event.toolCallId) {\n                return part;\n            }\n            return {\n                ...part,\n                data: {\n                    ...part.data,\n                    status: 'success' as const,\n                    result: event.result,\n                },\n            };\n        });\n\n        return {\n            ...msg,\n            content: updatedParts as TAssistantMessage<TCustom>['content'],\n        };\n    });\n\n    return changed ? next : messages;\n}\n"]}

package/build/esm/hooks/index.d.ts

@@ -5,3 +5,4 @@ export * from "./useScrollPreservation.js";
 export * from "./useAutoCollapseOnSuccess.js";
 export * from "./useAutoCollapseOnCancelled.js";
 export * from "./useFileUploadStore.js";
+export * from "./useToolset.js";

package/build/esm/hooks/index.js

@@ -5,4 +5,5 @@ export * from "./useScrollPreservation.js";
 export * from "./useAutoCollapseOnSuccess.js";
 export * from "./useAutoCollapseOnCancelled.js";
 export * from "./useFileUploadStore.js";
+export * from "./useToolset.js";
 //# sourceMappingURL=index.js.map

package/build/esm/hooks/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"../../../src","sources":["hooks/index.ts"],"names":[],"mappings":"AAAA,4CAAmC;AACnC,oCAAiC;AACjC,oCAAiC;AACjC,2CAAwC;AACxC,8CAA2C;AAC3C,gDAA6C;AAC7C,wCAAqC","sourcesContent":["export * from './useDateFormatter';\nexport * from './useToolMessage';\nexport * from './useSmartScroll';\nexport * from './useScrollPreservation';\nexport * from './useAutoCollapseOnSuccess';\nexport * from './useAutoCollapseOnCancelled';\nexport * from './useFileUploadStore';\n"]}
+{"version":3,"file":"index.js","sourceRoot":"../../../src","sources":["hooks/index.ts"],"names":[],"mappings":"AAAA,4CAAmC;AACnC,oCAAiC;AACjC,oCAAiC;AACjC,2CAAwC;AACxC,8CAA2C;AAC3C,gDAA6C;AAC7C,wCAAqC;AACrC,gCAA6B","sourcesContent":["export * from './useDateFormatter';\nexport * from './useToolMessage';\nexport * from './useSmartScroll';\nexport * from './useScrollPreservation';\nexport * from './useAutoCollapseOnSuccess';\nexport * from './useAutoCollapseOnCancelled';\nexport * from './useFileUploadStore';\nexport * from './useToolset';\n"]}

package/build/esm/hooks/useToolset.d.ts

@@ -0,0 +1,26 @@
+import type { Dispatch, SetStateAction } from 'react';
+import type { TChatMessage, TMessageContent } from "../types/messages.js";
+import type { MessageRendererRegistry } from "../utils/messageTypeRegistry.js";
+import { type Toolset, type ToolsetResultEvent } from "../utils/toolset.js";
+export type UseToolsetOptions<TCustom extends TMessageContent = never> = {
+    /**
+     * Called after the tool result has been merged into history.
+     * Typical use: forward the updated transcript to the model.
+     */
+    onAfterResult?: (messages: TChatMessage<TCustom>[]) => void;
+    /** Existing registry to extend instead of starting from a fresh one. */
+    registry?: MessageRendererRegistry;
+};
+export type UseToolsetReturn = {
+    messageRendererRegistry: MessageRendererRegistry;
+    handleToolResult: (event: ToolsetResultEvent) => void;
+};
+/**
+ * Wire a toolset into the chat: returns a `MessageRendererRegistry` that
+ * renders tool parts via the toolset and a `handleToolResult` callback that
+ * merges results into history and notifies via `onAfterResult`.
+ *
+ * Pass the same `toolset` reference across renders (e.g. via `useMemo`) so
+ * the registry stays stable.
+ */
+export declare function useToolset<TCustom extends TMessageContent = never>(toolset: Toolset, setMessages: Dispatch<SetStateAction<TChatMessage<TCustom>[]>>, options?: UseToolsetOptions<TCustom>): UseToolsetReturn;

package/build/esm/hooks/useToolset.js

@@ -0,0 +1,31 @@
+import { useCallback, useEffect, useMemo, useRef } from 'react';
+import { applyToolResult, createToolsetRenderer, } from "../utils/toolset.js";
+/**
+ * Wire a toolset into the chat: returns a `MessageRendererRegistry` that
+ * renders tool parts via the toolset and a `handleToolResult` callback that
+ * merges results into history and notifies via `onAfterResult`.
+ *
+ * Pass the same `toolset` reference across renders (e.g. via `useMemo`) so
+ * the registry stays stable.
+ */
+export function useToolset(toolset, setMessages, options) {
+    const onAfterResultRef = useRef(options === null || options === void 0 ? void 0 : options.onAfterResult);
+    useEffect(() => {
+        onAfterResultRef.current = options === null || options === void 0 ? void 0 : options.onAfterResult;
+    });
+    const handleToolResult = useCallback((event) => {
+        setMessages((prev) => {
+            const updated = applyToolResult(prev, event);
+            if (updated !== prev) {
+                queueMicrotask(() => { var _a; return (_a = onAfterResultRef.current) === null || _a === void 0 ? void 0 : _a.call(onAfterResultRef, updated); });
+            }
+            return updated;
+        });
+    }, [setMessages]);
+    const messageRendererRegistry = useMemo(() => createToolsetRenderer(toolset, {
+        onToolResult: handleToolResult,
+        registry: options === null || options === void 0 ? void 0 : options.registry,
+    }), [toolset, handleToolResult, options === null || options === void 0 ? void 0 : options.registry]);
+    return { messageRendererRegistry, handleToolResult };
+}
+//# sourceMappingURL=useToolset.js.map

package/build/esm/hooks/useToolset.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useToolset.js","sourceRoot":"../../../src","sources":["hooks/useToolset.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAC,MAAM,OAAO,CAAC;AAK9D,OAAO,EAGH,eAAe,EACf,qBAAqB,GACxB,4BAAyB;AAiB1B;;;;;;;GAOG;AACH,MAAM,UAAU,UAAU,CACtB,OAAgB,EAChB,WAA8D,EAC9D,OAAoC;IAEpC,MAAM,gBAAgB,GAAG,MAAM,CAAC,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,aAAa,CAAC,CAAC;IACxD,SAAS,CAAC,GAAG,EAAE;QACX,gBAAgB,CAAC,OAAO,GAAG,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,aAAa,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,MAAM,gBAAgB,GAAG,WAAW,CAChC,CAAC,KAAyB,EAAE,EAAE;QAC1B,WAAW,CAAC,CAAC,IAAI,EAAE,EAAE;YACjB,MAAM,OAAO,GAAG,eAAe,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAC7C,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;gBACnB,cAAc,CAAC,GAAG,EAAE,WAAC,OAAA,MAAA,gBAAgB,CAAC,OAAO,iEAAG,OAAO,CAAC,CAAA,EAAA,CAAC,CAAC;YAC9D,CAAC;YACD,OAAO,OAAO,CAAC;QACnB,CAAC,CAAC,CAAC;IACP,CAAC,EACD,CAAC,WAAW,CAAC,CAChB,CAAC;IAEF,MAAM,uBAAuB,GAAG,OAAO,CACnC,GAAG,EAAE,CACD,qBAAqB,CAAC,OAAO,EAAE;QAC3B,YAAY,EAAE,gBAAgB;QAC9B,QAAQ,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,QAAQ;KAC9B,CAAC,EACN,CAAC,OAAO,EAAE,gBAAgB,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,QAAQ,CAAC,CACjD,CAAC;IAEF,OAAO,EAAC,uBAAuB,EAAE,gBAAgB,EAAC,CAAC;AACvD,CAAC","sourcesContent":["import {useCallback, useEffect, useMemo, useRef} from 'react';\nimport type {Dispatch, SetStateAction} from 'react';\n\nimport type {TChatMessage, TMessageContent} from '../types/messages';\nimport type {MessageRendererRegistry} from '../utils/messageTypeRegistry';\nimport {\n    type Toolset,\n    type ToolsetResultEvent,\n    applyToolResult,\n    createToolsetRenderer,\n} from '../utils/toolset';\n\nexport type UseToolsetOptions<TCustom extends TMessageContent = never> = {\n    /**\n     * Called after the tool result has been merged into history.\n     * Typical use: forward the updated transcript to the model.\n     */\n    onAfterResult?: (messages: TChatMessage<TCustom>[]) => void;\n    /** Existing registry to extend instead of starting from a fresh one. */\n    registry?: MessageRendererRegistry;\n};\n\nexport type UseToolsetReturn = {\n    messageRendererRegistry: MessageRendererRegistry;\n    handleToolResult: (event: ToolsetResultEvent) => void;\n};\n\n/**\n * Wire a toolset into the chat: returns a `MessageRendererRegistry` that\n * renders tool parts via the toolset and a `handleToolResult` callback that\n * merges results into history and notifies via `onAfterResult`.\n *\n * Pass the same `toolset` reference across renders (e.g. via `useMemo`) so\n * the registry stays stable.\n */\nexport function useToolset<TCustom extends TMessageContent = never>(\n    toolset: Toolset,\n    setMessages: Dispatch<SetStateAction<TChatMessage<TCustom>[]>>,\n    options?: UseToolsetOptions<TCustom>,\n): UseToolsetReturn {\n    const onAfterResultRef = useRef(options?.onAfterResult);\n    useEffect(() => {\n        onAfterResultRef.current = options?.onAfterResult;\n    });\n\n    const handleToolResult = useCallback(\n        (event: ToolsetResultEvent) => {\n            setMessages((prev) => {\n                const updated = applyToolResult(prev, event);\n                if (updated !== prev) {\n                    queueMicrotask(() => onAfterResultRef.current?.(updated));\n                }\n                return updated;\n            });\n        },\n        [setMessages],\n    );\n\n    const messageRendererRegistry = useMemo(\n        () =>\n            createToolsetRenderer(toolset, {\n                onToolResult: handleToolResult,\n                registry: options?.registry,\n            }),\n        [toolset, handleToolResult, options?.registry],\n    );\n\n    return {messageRendererRegistry, handleToolResult};\n}\n"]}

package/build/esm/package.json

@@ -1 +1 @@
-{"version":"2.0.5","type":"module","sideEffects":["*.css","*.scss"]}
+{"version":"2.0.6-beta.86aa50f22bec20e0cae460b5d3be7112f9d4efd2.0","type":"module","sideEffects":["*.css","*.scss"]}

package/build/esm/utils/index.d.ts

@@ -3,3 +3,4 @@ export * from "./messageUtils.js";
 export * from "./validation.js";
 export * from "./messageTypeRegistry.js";
 export * from "./clipboardUtils.js";
+export * from "./toolset.js";

package/build/esm/utils/index.js

@@ -4,4 +4,5 @@ export * from "./messageUtils.js";
 export * from "./validation.js";
 export * from "./messageTypeRegistry.js";
 export * from "./clipboardUtils.js";
+export * from "./toolset.js";
 //# sourceMappingURL=index.js.map

package/build/esm/utils/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"../../../src","sources":["utils/index.ts"],"names":[],"mappings":"AAAA,mBAAmB;AACnB,+BAA4B;AAC5B,kCAA+B;AAC/B,gCAA6B;AAC7B,yCAAsC;AACtC,oCAAiC","sourcesContent":["// Export utilities\nexport * from './chatUtils';\nexport * from './messageUtils';\nexport * from './validation';\nexport * from './messageTypeRegistry';\nexport * from './clipboardUtils';\n"]}
+{"version":3,"file":"index.js","sourceRoot":"../../../src","sources":["utils/index.ts"],"names":[],"mappings":"AAAA,mBAAmB;AACnB,+BAA4B;AAC5B,kCAA+B;AAC/B,gCAA6B;AAC7B,yCAAsC;AACtC,oCAAiC;AACjC,6BAA0B","sourcesContent":["// Export utilities\nexport * from './chatUtils';\nexport * from './messageUtils';\nexport * from './validation';\nexport * from './messageTypeRegistry';\nexport * from './clipboardUtils';\nexport * from './toolset';\n"]}

package/build/esm/utils/toolset.d.ts

@@ -0,0 +1,85 @@
+import type { ComponentType, ReactNode } from 'react';
+import type { TChatMessage, TMessageContent, ToolMessageContentData } from "../types/messages.js";
+import { type MessageRendererRegistry } from "./messageTypeRegistry.js";
+export type ToolSchemaResult<TArgs> = {
+    success: true;
+    data: TArgs;
+} | {
+    success: false;
+    error: {
+        message: string;
+    };
+};
+export type ToolSchema<TArgs> = {
+    validate: (input: unknown) => ToolSchemaResult<TArgs>;
+};
+export type ToolComponentProps<TArgs, TResult> = {
+    args: TArgs;
+    result?: TResult;
+    submitResult: (result: TResult) => void;
+};
+export type ToolDefinition<TArgs, TResult> = {
+    name: string;
+    description: string;
+    parameters: Record<string, unknown>;
+    schema: ToolSchema<TArgs>;
+    component: ComponentType<ToolComponentProps<TArgs, TResult>>;
+    execute?: (params: {
+        args: TArgs;
+        result: TResult;
+        toolCallId: string;
+    }) => TResult | Promise<TResult>;
+};
+export type RuntimeToolDefinition = {
+    name: string;
+    description: string;
+    parameters: Record<string, unknown>;
+    validate: (input: unknown) => ToolSchemaResult<unknown>;
+    render: (props: {
+        args: unknown;
+        result?: unknown;
+        submitResult: (result: unknown) => void;
+    }) => ReactNode;
+    execute: (params: {
+        args: unknown;
+        result: unknown;
+        toolCallId: string;
+    }) => unknown | Promise<unknown>;
+};
+export type Toolset = Record<string, RuntimeToolDefinition>;
+export type ToolPartContentData<TArgs = unknown, TResult = unknown> = ToolMessageContentData & {
+    toolCallId: string;
+    args?: TArgs;
+    result?: TResult;
+};
+export type ToolPartContent<TArgs = unknown, TResult = unknown> = TMessageContent<'tool', ToolPartContentData<TArgs, TResult>>;
+export type ToolsetResultEvent = {
+    toolCallId: string;
+    toolName: string;
+    result: unknown;
+};
+/**
+ * Wrap a typed tool definition into an erased runtime entry that the
+ * toolset renderer can dispatch by `toolName`.
+ */
+export declare function defineTool<TArgs, TResult>(definition: ToolDefinition<TArgs, TResult>): RuntimeToolDefinition;
+export type CreateToolsetRendererOptions = {
+    onToolResult: (event: ToolsetResultEvent) => void;
+    /**
+     * Existing registry to extend instead of starting from a fresh one.
+     * The `tool` renderer is overridden; other types are preserved.
+     */
+    registry?: MessageRendererRegistry;
+};
+/**
+ * Build a `MessageRendererRegistry` whose `tool` renderer dispatches
+ * by `toolName` into the provided toolset. Unknown tools and invalid
+ * args fall back to a generic `<ToolMessage status="error" />`.
+ */
+export declare function createToolsetRenderer(toolset: Toolset, options: CreateToolsetRendererOptions): MessageRendererRegistry;
+/**
+ * Merge a tool result into the matching `tool` part of the chat history.
+ * Returns the original array reference when nothing matched, so React state
+ * setters can skip needless updates.
+ */
+export declare function applyToolResult<TCustom extends TMessageContent = never>(messages: TChatMessage<TCustom>[], event: ToolsetResultEvent): TChatMessage<TCustom>[];

package/build/esm/utils/toolset.js

@@ -0,0 +1,99 @@
+import { jsx as _jsx } from "react/jsx-runtime";
+import { ToolMessage } from "../components/organisms/ToolMessage/index.js";
+import { createMessageRendererRegistry, registerMessageRenderer, } from "./messageTypeRegistry.js";
+/**
+ * Wrap a typed tool definition into an erased runtime entry that the
+ * toolset renderer can dispatch by `toolName`.
+ */
+export function defineTool(definition) {
+    const Component = definition.component;
+    return {
+        name: definition.name,
+        description: definition.description,
+        parameters: definition.parameters,
+        validate: definition.schema.validate,
+        render: ({ args, result, submitResult }) => (_jsx(Component, { args: args, result: result, submitResult: submitResult })),
+        execute: ({ args, result, toolCallId }) => definition.execute
+            ? definition.execute({
+                args: args,
+                result: result,
+                toolCallId,
+            })
+            : result,
+    };
+}
+/**
+ * Build a `MessageRendererRegistry` whose `tool` renderer dispatches
+ * by `toolName` into the provided toolset. Unknown tools and invalid
+ * args fall back to a generic `<ToolMessage status="error" />`.
+ */
+export function createToolsetRenderer(toolset, options) {
+    const { onToolResult, registry = createMessageRendererRegistry() } = options;
+    registerMessageRenderer(registry, 'tool', {
+        component: ({ part }) => {
+            const toolPart = part.data;
+            const toolDef = toolset[toolPart.toolName];
+            if (!toolDef) {
+                return (_jsx(ToolMessage, Object.assign({}, toolPart, { status: "error", expandable: true, initialExpanded: true, bodyContent: `Unknown tool: ${toolPart.toolName}` })));
+            }
+            const validation = toolDef.validate(toolPart.args);
+            if (!validation.success) {
+                return (_jsx(ToolMessage, Object.assign({}, toolPart, { status: "error", expandable: true, initialExpanded: true, bodyContent: validation.error.message })));
+            }
+            const submitResult = (result) => {
+                Promise.resolve(toolDef.execute({
+                    args: validation.data,
+                    result,
+                    toolCallId: toolPart.toolCallId,
+                })).then((finalResult) => {
+                    onToolResult({
+                        toolCallId: toolPart.toolCallId,
+                        toolName: toolPart.toolName,
+                        result: finalResult,
+                    });
+                });
+            };
+            return toolDef.render({
+                args: validation.data,
+                result: toolPart.result,
+                submitResult,
+            });
+        },
+    });
+    return registry;
+}
+function isToolPartContent(part) {
+    return part.type === 'tool' && typeof part.data === 'object' && part.data !== null;
+}
+function toContentParts(content) {
+    if (typeof content === 'string') {
+        return content ? [{ type: 'text', data: { text: content } }] : [];
+    }
+    return Array.isArray(content) ? content : [content];
+}
+/**
+ * Merge a tool result into the matching `tool` part of the chat history.
+ * Returns the original array reference when nothing matched, so React state
+ * setters can skip needless updates.
+ */
+export function applyToolResult(messages, event) {
+    let changed = false;
+    const next = messages.map((msg) => {
+        if (msg.role !== 'assistant')
+            return msg;
+        const parts = toContentParts(msg.content);
+        const hasMatch = parts.some((part) => isToolPartContent(part) && part.data.toolCallId === event.toolCallId);
+        if (!hasMatch)
+            return msg;
+        changed = true;
+        const updatedParts = parts.map((part) => {
+            if (!isToolPartContent(part) || part.data.toolCallId !== event.toolCallId) {
+                return part;
+            }
+            return Object.assign(Object.assign({}, part), { data: Object.assign(Object.assign({}, part.data), { status: 'success', result: event.result }) });
+        });
+        return Object.assign(Object.assign({}, msg), { content: updatedParts });
+    });
+    return changed ? next : messages;
+}
+//# sourceMappingURL=toolset.js.map