@assistant-ui/core 0.1.0

package/dist/model-context/frame/host.d.ts

@@ -0,0 +1,21 @@
+import { ModelContextProvider, ModelContext } from "../types.js";
+import type { Unsubscribe } from "../../types/index.js";
+export declare class AssistantFrameHost implements ModelContextProvider {
+    private _context;
+    private _subscribers;
+    private _pendingRequests;
+    private _requestCounter;
+    private _iframeWindow;
+    private _targetOrigin;
+    constructor(iframeWindow: Window, targetOrigin?: string);
+    private handleMessage;
+    private updateContext;
+    private callTool;
+    private sendRequest;
+    private requestContext;
+    private notifySubscribers;
+    getModelContext(): ModelContext;
+    subscribe(callback: () => void): Unsubscribe;
+    dispose(): void;
+}
+//# sourceMappingURL=host.d.ts.map

package/dist/model-context/frame/host.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"host.d.ts","sourceRoot":"","sources":["../../../src/model-context/frame/host.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,YAAY,EAAE,oBAAiB;AAC9D,OAAO,KAAK,EAAE,WAAW,EAAE,6BAAoB;AAyC/C,qBAAa,kBAAmB,YAAW,oBAAoB;IAC7D,OAAO,CAAC,QAAQ,CAAoB;IACpC,OAAO,CAAC,YAAY,CAAyB;IAC7C,OAAO,CAAC,gBAAgB,CAMpB;IACJ,OAAO,CAAC,eAAe,CAAK;IAC5B,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,aAAa,CAAS;gBAElB,YAAY,EAAE,MAAM,EAAE,YAAY,GAAE,MAAY;IAU5D,OAAO,CAAC,aAAa;IA6BrB,OAAO,CAAC,aAAa;IAmBrB,OAAO,CAAC,QAAQ;IAahB,OAAO,CAAC,WAAW;IAqCnB,OAAO,CAAC,cAAc;IAYtB,OAAO,CAAC,iBAAiB;IAIzB,eAAe,IAAI,YAAY;IAI/B,SAAS,CAAC,QAAQ,EAAE,MAAM,IAAI,GAAG,WAAW;IAK5C,OAAO;CAKR"}

package/dist/model-context/frame/host.js

@@ -0,0 +1,141 @@
+import { FRAME_MESSAGE_CHANNEL, } from "./types.js";
+/**
+ * Deserializes tools from JSON Schema format back to Tool objects
+ */
+const deserializeTool = (serializedTool) => ({
+    parameters: serializedTool.parameters,
+    ...(serializedTool.description && {
+        description: serializedTool.description,
+    }),
+    ...(serializedTool.disabled !== undefined && {
+        disabled: serializedTool.disabled,
+    }),
+    ...(serializedTool.type && { type: serializedTool.type }),
+});
+/**
+ * Deserializes a ModelContext from transmission format
+ */
+const deserializeModelContext = (serialized) => ({
+    ...(serialized.system !== undefined && { system: serialized.system }),
+    ...(serialized.tools && {
+        tools: Object.fromEntries(Object.entries(serialized.tools).map(([name, tool]) => [
+            name,
+            deserializeTool(tool),
+        ])),
+    }),
+});
+export class AssistantFrameHost {
+    _context = {};
+    _subscribers = new Set();
+    _pendingRequests = new Map();
+    _requestCounter = 0;
+    _iframeWindow;
+    _targetOrigin;
+    constructor(iframeWindow, targetOrigin = "*") {
+        this._iframeWindow = iframeWindow;
+        this._targetOrigin = targetOrigin;
+        this.handleMessage = this.handleMessage.bind(this);
+        window.addEventListener("message", this.handleMessage);
+        this.requestContext();
+    }
+    handleMessage(event) {
+        if (this._targetOrigin !== "*" && event.origin !== this._targetOrigin)
+            return;
+        if (event.source !== this._iframeWindow)
+            return;
+        if (event.data?.channel !== FRAME_MESSAGE_CHANNEL)
+            return;
+        const message = event.data.message;
+        switch (message.type) {
+            case "model-context-update": {
+                this.updateContext(message.context);
+                break;
+            }
+            case "tool-result": {
+                const pending = this._pendingRequests.get(message.id);
+                if (pending) {
+                    if (message.error) {
+                        pending.reject(new Error(message.error));
+                    }
+                    else {
+                        pending.resolve(message.result);
+                    }
+                    this._pendingRequests.delete(message.id);
+                }
+                break;
+            }
+        }
+    }
+    updateContext(serializedContext) {
+        const context = deserializeModelContext(serializedContext);
+        this._context = {
+            ...context,
+            tools: context.tools &&
+                Object.fromEntries(Object.entries(context.tools).map(([name, tool]) => [
+                    name,
+                    {
+                        ...tool,
+                        execute: (args) => this.callTool(name, args),
+                    },
+                ])),
+        };
+        this.notifySubscribers();
+    }
+    callTool(toolName, args) {
+        return this.sendRequest({
+            type: "tool-call",
+            id: `tool-${this._requestCounter++}`,
+            toolName,
+            args,
+        }, 30000, `Tool call "${toolName}" timed out`);
+    }
+    sendRequest(message, timeout = 30000, timeoutMessage = "Request timed out") {
+        return new Promise((resolve, reject) => {
+            this._pendingRequests.set(message.id, { resolve, reject });
+            this._iframeWindow.postMessage({ channel: FRAME_MESSAGE_CHANNEL, message }, this._targetOrigin);
+            const timeoutId = setTimeout(() => {
+                const pending = this._pendingRequests.get(message.id);
+                if (pending) {
+                    pending.reject(new Error(timeoutMessage));
+                    this._pendingRequests.delete(message.id);
+                }
+            }, timeout);
+            const originalResolve = this._pendingRequests.get(message.id).resolve;
+            const originalReject = this._pendingRequests.get(message.id).reject;
+            this._pendingRequests.set(message.id, {
+                resolve: (value) => {
+                    clearTimeout(timeoutId);
+                    originalResolve(value);
+                },
+                reject: (error) => {
+                    clearTimeout(timeoutId);
+                    originalReject(error);
+                },
+            });
+        });
+    }
+    requestContext() {
+        this._iframeWindow.postMessage({
+            channel: FRAME_MESSAGE_CHANNEL,
+            message: {
+                type: "model-context-request",
+            },
+        }, this._targetOrigin);
+    }
+    notifySubscribers() {
+        this._subscribers.forEach((callback) => callback());
+    }
+    getModelContext() {
+        return this._context;
+    }
+    subscribe(callback) {
+        this._subscribers.add(callback);
+        return () => this._subscribers.delete(callback);
+    }
+    dispose() {
+        window.removeEventListener("message", this.handleMessage);
+        this._subscribers.clear();
+        this._pendingRequests.clear();
+    }
+}
+//# sourceMappingURL=host.js.map

package/dist/model-context/frame/host.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"host.js","sourceRoot":"","sources":["../../../src/model-context/frame/host.ts"],"names":[],"mappings":"AAGA,OAAO,EAEL,qBAAqB,GAGtB,mBAAgB;AAEjB;;GAEG;AACH,MAAM,eAAe,GAAG,CAAC,cAA8B,EAAkB,EAAE,CACzE,CAAC;IACC,UAAU,EAAE,cAAc,CAAC,UAAU;IACrC,GAAG,CAAC,cAAc,CAAC,WAAW,IAAI;QAChC,WAAW,EAAE,cAAc,CAAC,WAAW;KACxC,CAAC;IACF,GAAG,CAAC,cAAc,CAAC,QAAQ,KAAK,SAAS,IAAI;QAC3C,QAAQ,EAAE,cAAc,CAAC,QAAQ;KAClC,CAAC;IACF,GAAG,CAAC,cAAc,CAAC,IAAI,IAAI,EAAE,IAAI,EAAE,cAAc,CAAC,IAAI,EAAE,CAAC;CAC1D,CAAmB,CAAC;AAEvB;;GAEG;AACH,MAAM,uBAAuB,GAAG,CAC9B,UAAkC,EACpB,EAAE,CAAC,CAAC;IAClB,GAAG,CAAC,UAAU,CAAC,MAAM,KAAK,SAAS,IAAI,EAAE,MAAM,EAAE,UAAU,CAAC,MAAM,EAAE,CAAC;IACrE,GAAG,CAAC,UAAU,CAAC,KAAK,IAAI;QACtB,KAAK,EAAE,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;YACrD,IAAI;YACJ,eAAe,CAAC,IAAI,CAAC;SACtB,CAAC,CACH;KACF,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,OAAO,kBAAkB;IACrB,QAAQ,GAAiB,EAAE,CAAC;IAC5B,YAAY,GAAG,IAAI,GAAG,EAAc,CAAC;IACrC,gBAAgB,GAAG,IAAI,GAAG,EAM/B,CAAC;IACI,eAAe,GAAG,CAAC,CAAC;IACpB,aAAa,CAAS;IACtB,aAAa,CAAS;IAE9B,YAAY,YAAoB,EAAE,eAAuB,GAAG;QAC1D,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;QAClC,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;QAElC,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnD,MAAM,CAAC,gBAAgB,CAAC,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;QAEvD,IAAI,CAAC,cAAc,EAAE,CAAC;IACxB,CAAC;IAEO,aAAa,CAAC,KAAmB;QACvC,IAAI,IAAI,CAAC,aAAa,KAAK,GAAG,IAAI,KAAK,CAAC,MAAM,KAAK,IAAI,CAAC,aAAa;YACnE,OAAO;QACT,IAAI,KAAK,CAAC,MAAM,KAAK,IAAI,CAAC,aAAa;YAAE,OAAO;QAChD,IAAI,KAAK,CAAC,IAAI,EAAE,OAAO,KAAK,qBAAqB;YAAE,OAAO;QAE1D,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,OAAuB,CAAC;QAEnD,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;YACrB,KAAK,sBAAsB,CAAC,CAAC,CAAC;gBAC5B,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;gBACpC,MAAM;YACR,CAAC;YAED,KAAK,aAAa,CAAC,CAAC,CAAC;gBACnB,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBACtD,IAAI,OAAO,EAAE,CAAC;oBACZ,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;wBAClB,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;oBAC3C,CAAC;yBAAM,CAAC;wBACN,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;oBAClC,CAAC;oBACD,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAC3C,CAAC;gBACD,MAAM;YACR,CAAC;QACH,CAAC;IACH,CAAC;IAEO,aAAa,CAAC,iBAAyC;QAC7D,MAAM,OAAO,GAAG,uBAAuB,CAAC,iBAAiB,CAAC,CAAC;QAC3D,IAAI,CAAC,QAAQ,GAAG;YACd,GAAG,OAAO;YACV,KAAK,EACH,OAAO,CAAC,KAAK;gBACb,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;oBAClD,IAAI;oBACJ;wBACE,GAAG,IAAI;wBACP,OAAO,EAAE,CAAC,IAAS,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC;qBAChC;iBACpB,CAAC,CACH;SACJ,CAAC;QACF,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC3B,CAAC;IAEO,QAAQ,CAAC,QAAgB,EAAE,IAAS;QAC1C,OAAO,IAAI,CAAC,WAAW,CACrB;YACE,IAAI,EAAE,WAAW;YACjB,EAAE,EAAE,QAAQ,IAAI,CAAC,eAAe,EAAE,EAAE;YACpC,QAAQ;YACR,IAAI;SACL,EACD,KAAK,EACL,cAAc,QAAQ,aAAa,CACpC,CAAC;IACJ,CAAC;IAEO,WAAW,CACjB,OAAU,EACV,OAAO,GAAG,KAAK,EACf,cAAc,GAAG,mBAAmB;QAEpC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;YAE3D,IAAI,CAAC,aAAa,CAAC,WAAW,CAC5B,EAAE,OAAO,EAAE,qBAAqB,EAAE,OAAO,EAAE,EAC3C,IAAI,CAAC,aAAa,CACnB,CAAC;YAEF,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;gBAChC,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBACtD,IAAI,OAAO,EAAE,CAAC;oBACZ,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC;oBAC1C,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAC3C,CAAC;YACH,CAAC,EAAE,OAAO,CAAC,CAAC;YAEZ,MAAM,eAAe,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAE,CAAC,OAAO,CAAC;YACvE,MAAM,cAAc,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAE,CAAC,MAAM,CAAC;YAErE,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE;gBACpC,OAAO,EAAE,CAAC,KAAU,EAAE,EAAE;oBACtB,YAAY,CAAC,SAAS,CAAC,CAAC;oBACxB,eAAe,CAAC,KAAK,CAAC,CAAC;gBACzB,CAAC;gBACD,MAAM,EAAE,CAAC,KAAU,EAAE,EAAE;oBACrB,YAAY,CAAC,SAAS,CAAC,CAAC;oBACxB,cAAc,CAAC,KAAK,CAAC,CAAC;gBACxB,CAAC;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,cAAc;QACpB,IAAI,CAAC,aAAa,CAAC,WAAW,CAC5B;YACE,OAAO,EAAE,qBAAqB;YAC9B,OAAO,EAAE;gBACP,IAAI,EAAE,uBAAuB;aACd;SAClB,EACD,IAAI,CAAC,aAAa,CACnB,CAAC;IACJ,CAAC;IAEO,iBAAiB;QACvB,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,EAAE,CAAC,CAAC;IACtD,CAAC;IAED,eAAe;QACb,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,SAAS,CAAC,QAAoB;QAC5B,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAChC,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAClD,CAAC;IAED,OAAO;QACL,MAAM,CAAC,mBAAmB,CAAC,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;QAC1D,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;QAC1B,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;IAChC,CAAC;CACF"}

package/dist/model-context/frame/index.d.ts

@@ -0,0 +1,5 @@
+export { AssistantFrameHost } from "./host.js";
+export { AssistantFrameProvider } from "./provider.js";
+export type { SerializedTool, SerializedModelContext, FrameMessageType, FrameMessage, } from "./types.js";
+export { FRAME_MESSAGE_CHANNEL } from "./types.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/model-context/frame/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/model-context/frame/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,kBAAe;AAC5C,OAAO,EAAE,sBAAsB,EAAE,sBAAmB;AAEpD,YAAY,EACV,cAAc,EACd,sBAAsB,EACtB,gBAAgB,EAChB,YAAY,GACb,mBAAgB;AACjB,OAAO,EAAE,qBAAqB,EAAE,mBAAgB"}

package/dist/model-context/frame/index.js

@@ -0,0 +1,4 @@
+export { AssistantFrameHost } from "./host.js";
+export { AssistantFrameProvider } from "./provider.js";
+export { FRAME_MESSAGE_CHANNEL } from "./types.js";
+//# sourceMappingURL=index.js.map

package/dist/model-context/frame/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/model-context/frame/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,kBAAe;AAC5C,OAAO,EAAE,sBAAsB,EAAE,sBAAmB;AAQpD,OAAO,EAAE,qBAAqB,EAAE,mBAAgB"}

package/dist/model-context/frame/provider.d.ts

@@ -0,0 +1,18 @@
+import { ModelContextProvider } from "../types.js";
+import type { Unsubscribe } from "../../types/index.js";
+export declare class AssistantFrameProvider {
+    private static _instance;
+    private _providers;
+    private _providerUnsubscribes;
+    private _targetOrigin;
+    private constructor();
+    private static getInstance;
+    private handleMessage;
+    private handleToolCall;
+    private sendMessage;
+    private getModelContext;
+    private broadcastUpdate;
+    static addModelContextProvider(provider: ModelContextProvider, targetOrigin?: string): Unsubscribe;
+    static dispose(): void;
+}
+//# sourceMappingURL=provider.d.ts.map

package/dist/model-context/frame/provider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.d.ts","sourceRoot":"","sources":["../../../src/model-context/frame/provider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAgB,oBAAiB;AAC9D,OAAO,KAAK,EAAE,WAAW,EAAE,6BAAoB;AAkC/C,qBAAa,sBAAsB;IACjC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAuC;IAE/D,OAAO,CAAC,UAAU,CAAmC;IACrD,OAAO,CAAC,qBAAqB,CAGzB;IACJ,OAAO,CAAC,aAAa,CAAS;IAE9B,OAAO;IAQP,OAAO,CAAC,MAAM,CAAC,WAAW;IAS1B,OAAO,CAAC,aAAa;YAqBP,cAAc;IAoC5B,OAAO,CAAC,WAAW;IAOnB,OAAO,CAAC,eAAe;IAkBvB,OAAO,CAAC,eAAe;IAcvB,MAAM,CAAC,uBAAuB,CAC5B,QAAQ,EAAE,oBAAoB,EAC9B,YAAY,CAAC,EAAE,MAAM,GACpB,WAAW;IAmBd,MAAM,CAAC,OAAO;CAYf"}

package/dist/model-context/frame/provider.js

@@ -0,0 +1,133 @@
+import { z } from "zod";
+import { FRAME_MESSAGE_CHANNEL, } from "./types.js";
+const serializeTool = (tool) => ({
+    ...(tool.description && { description: tool.description }),
+    parameters: tool.parameters instanceof z.ZodType
+        ? (z.toJSONSchema?.(tool.parameters) ?? tool.parameters)
+        : tool.parameters,
+    ...(tool.disabled !== undefined && { disabled: tool.disabled }),
+    ...(tool.type && { type: tool.type }),
+});
+const serializeModelContext = (context) => ({
+    ...(context.system !== undefined && { system: context.system }),
+    ...(context.tools && {
+        tools: Object.fromEntries(Object.entries(context.tools).map(([name, tool]) => [
+            name,
+            serializeTool(tool),
+        ])),
+    }),
+});
+export class AssistantFrameProvider {
+    static _instance = null;
+    _providers = new Set();
+    _providerUnsubscribes = new Map();
+    _targetOrigin;
+    constructor(targetOrigin = "*") {
+        this._targetOrigin = targetOrigin;
+        this.handleMessage = this.handleMessage.bind(this);
+        window.addEventListener("message", this.handleMessage);
+        setTimeout(() => this.broadcastUpdate(), 0);
+    }
+    static getInstance(targetOrigin) {
+        if (!AssistantFrameProvider._instance) {
+            AssistantFrameProvider._instance = new AssistantFrameProvider(targetOrigin);
+        }
+        return AssistantFrameProvider._instance;
+    }
+    handleMessage(event) {
+        if (this._targetOrigin !== "*" && event.origin !== this._targetOrigin)
+            return;
+        if (event.data?.channel !== FRAME_MESSAGE_CHANNEL)
+            return;
+        const message = event.data.message;
+        switch (message.type) {
+            case "model-context-request":
+                this.sendMessage(event, {
+                    type: "model-context-update",
+                    context: serializeModelContext(this.getModelContext()),
+                });
+                break;
+            case "tool-call":
+                this.handleToolCall(message, event);
+                break;
+        }
+    }
+    async handleToolCall(message, event) {
+        const tool = this.getModelContext().tools?.[message.toolName];
+        let result;
+        let error;
+        if (!tool) {
+            error = `Tool "${message.toolName}" not found`;
+        }
+        else {
+            try {
+                result = tool.execute
+                    ? await tool.execute(message.args, {
+                        toolCallId: message.id,
+                        abortSignal: new AbortController().signal,
+                        human: async () => {
+                            throw new Error("Tool human input is not supported in frame context");
+                        },
+                    })
+                    : undefined;
+            }
+            catch (e) {
+                error = e instanceof Error ? e.message : String(e);
+            }
+        }
+        this.sendMessage(event, {
+            type: "tool-result",
+            id: message.id,
+            ...(error ? { error } : { result }),
+        });
+    }
+    sendMessage(event, message) {
+        event.source?.postMessage({ channel: FRAME_MESSAGE_CHANNEL, message }, { targetOrigin: event.origin });
+    }
+    getModelContext() {
+        const contexts = Array.from(this._providers).map((p) => p.getModelContext());
+        return contexts.reduce((merged, context) => ({
+            system: context.system
+                ? merged.system
+                    ? `${merged.system}\n\n${context.system}`
+                    : context.system
+                : merged.system,
+            tools: { ...(merged.tools || {}), ...(context.tools || {}) },
+        }), {});
+    }
+    broadcastUpdate() {
+        if (window.parent && window.parent !== window) {
+            const updateMessage = {
+                type: "model-context-update",
+                context: serializeModelContext(this.getModelContext()),
+            };
+            window.parent.postMessage({ channel: FRAME_MESSAGE_CHANNEL, message: updateMessage }, this._targetOrigin);
+        }
+    }
+    static addModelContextProvider(provider, targetOrigin) {
+        const instance = AssistantFrameProvider.getInstance(targetOrigin);
+        instance._providers.add(provider);
+        const unsubscribe = provider.subscribe?.(() => instance.broadcastUpdate());
+        if (unsubscribe) {
+            instance._providerUnsubscribes.set(provider, unsubscribe);
+        }
+        instance.broadcastUpdate();
+        return () => {
+            instance._providers.delete(provider);
+            instance._providerUnsubscribes.get(provider)?.();
+            instance._providerUnsubscribes.delete(provider);
+            instance.broadcastUpdate();
+        };
+    }
+    static dispose() {
+        if (AssistantFrameProvider._instance) {
+            const instance = AssistantFrameProvider._instance;
+            window.removeEventListener("message", instance.handleMessage);
+            instance._providerUnsubscribes.forEach((unsubscribe) => unsubscribe?.());
+            instance._providerUnsubscribes.clear();
+            instance._providers.clear();
+            AssistantFrameProvider._instance = null;
+        }
+    }
+}
+//# sourceMappingURL=provider.js.map

package/dist/model-context/frame/provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.js","sourceRoot":"","sources":["../../../src/model-context/frame/provider.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAEL,qBAAqB,GAGtB,mBAAgB;AAEjB,MAAM,aAAa,GAAG,CAAC,IAAoB,EAAkB,EAAE,CAAC,CAAC;IAC/D,GAAG,CAAC,IAAI,CAAC,WAAW,IAAI,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC;IAC1D,UAAU,EACR,IAAI,CAAC,UAAU,YAAY,CAAC,CAAC,OAAO;QAClC,CAAC,CAAC,CAAE,CAAS,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC;QACjE,CAAC,CAAC,IAAI,CAAC,UAAU;IACrB,GAAG,CAAC,IAAI,CAAC,QAAQ,KAAK,SAAS,IAAI,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC;IAC/D,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;CACtC,CAAC,CAAC;AAEH,MAAM,qBAAqB,GAAG,CAC5B,OAAqB,EACG,EAAE,CAAC,CAAC;IAC5B,GAAG,CAAC,OAAO,CAAC,MAAM,KAAK,SAAS,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC;IAC/D,GAAG,CAAC,OAAO,CAAC,KAAK,IAAI;QACnB,KAAK,EAAE,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;YAClD,IAAI;YACJ,aAAa,CAAC,IAAI,CAAC;SACpB,CAAC,CACH;KACF,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,OAAO,sBAAsB;IACzB,MAAM,CAAC,SAAS,GAAkC,IAAI,CAAC;IAEvD,UAAU,GAAG,IAAI,GAAG,EAAwB,CAAC;IAC7C,qBAAqB,GAAG,IAAI,GAAG,EAGpC,CAAC;IACI,aAAa,CAAS;IAE9B,YAAoB,eAAuB,GAAG;QAC5C,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;QAClC,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnD,MAAM,CAAC,gBAAgB,CAAC,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;QAEvD,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC,CAAC;IAC9C,CAAC;IAEO,MAAM,CAAC,WAAW,CAAC,YAAqB;QAC9C,IAAI,CAAC,sBAAsB,CAAC,SAAS,EAAE,CAAC;YACtC,sBAAsB,CAAC,SAAS,GAAG,IAAI,sBAAsB,CAC3D,YAAY,CACb,CAAC;QACJ,CAAC;QACD,OAAO,sBAAsB,CAAC,SAAS,CAAC;IAC1C,CAAC;IAEO,aAAa,CAAC,KAAmB;QACvC,IAAI,IAAI,CAAC,aAAa,KAAK,GAAG,IAAI,KAAK,CAAC,MAAM,KAAK,IAAI,CAAC,aAAa;YACnE,OAAO;QACT,IAAI,KAAK,CAAC,IAAI,EAAE,OAAO,KAAK,qBAAqB;YAAE,OAAO;QAE1D,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,OAAuB,CAAC;QAEnD,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;YACrB,KAAK,uBAAuB;gBAC1B,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE;oBACtB,IAAI,EAAE,sBAAsB;oBAC5B,OAAO,EAAE,qBAAqB,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;iBACvD,CAAC,CAAC;gBACH,MAAM;YAER,KAAK,WAAW;gBACd,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;gBACpC,MAAM;QACV,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,OAAqD,EACrD,KAAmB;QAEnB,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAE9D,IAAI,MAAW,CAAC;QAChB,IAAI,KAAyB,CAAC;QAE9B,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,KAAK,GAAG,SAAS,OAAO,CAAC,QAAQ,aAAa,CAAC;QACjD,CAAC;aAAM,CAAC;YACN,IAAI,CAAC;gBACH,MAAM,GAAG,IAAI,CAAC,OAAO;oBACnB,CAAC,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE;wBAC/B,UAAU,EAAE,OAAO,CAAC,EAAE;wBACtB,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;wBACzC,KAAK,EAAE,KAAK,IAAI,EAAE;4BAChB,MAAM,IAAI,KAAK,CACb,oDAAoD,CACrD,CAAC;wBACJ,CAAC;qBACF,CAAC;oBACJ,CAAC,CAAC,SAAS,CAAC;YAChB,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,KAAK,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YACrD,CAAC;QACH,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE;YACtB,IAAI,EAAE,aAAa;YACnB,EAAE,EAAE,OAAO,CAAC,EAAE;YACd,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC;SACpC,CAAC,CAAC;IACL,CAAC;IAEO,WAAW,CAAC,KAAmB,EAAE,OAAqB;QAC5D,KAAK,CAAC,MAAM,EAAE,WAAW,CACvB,EAAE,OAAO,EAAE,qBAAqB,EAAE,OAAO,EAAE,EAC3C,EAAE,YAAY,EAAE,KAAK,CAAC,MAAM,EAAE,CAC/B,CAAC;IACJ,CAAC;IAEO,eAAe;QACrB,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CACrD,CAAC,CAAC,eAAe,EAAE,CACpB,CAAC;QAEF,OAAO,QAAQ,CAAC,MAAM,CACpB,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;YACpB,MAAM,EAAE,OAAO,CAAC,MAAM;gBACpB,CAAC,CAAC,MAAM,CAAC,MAAM;oBACb,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,OAAO,OAAO,CAAC,MAAM,EAAE;oBACzC,CAAC,CAAC,OAAO,CAAC,MAAM;gBAClB,CAAC,CAAC,MAAM,CAAC,MAAM;YACjB,KAAK,EAAE,EAAE,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,GAAG,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE;SAC7D,CAAC,EACF,EAAkB,CACnB,CAAC;IACJ,CAAC;IAEO,eAAe;QACrB,IAAI,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;YAC9C,MAAM,aAAa,GAAiB;gBAClC,IAAI,EAAE,sBAAsB;gBAC5B,OAAO,EAAE,qBAAqB,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;aACvD,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,WAAW,CACvB,EAAE,OAAO,EAAE,qBAAqB,EAAE,OAAO,EAAE,aAAa,EAAE,EAC1D,IAAI,CAAC,aAAa,CACnB,CAAC;QACJ,CAAC;IACH,CAAC;IAED,MAAM,CAAC,uBAAuB,CAC5B,QAA8B,EAC9B,YAAqB;QAErB,MAAM,QAAQ,GAAG,sBAAsB,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;QAClE,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAElC,MAAM,WAAW,GAAG,QAAQ,CAAC,SAAS,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,eAAe,EAAE,CAAC,CAAC;QAC3E,IAAI,WAAW,EAAE,CAAC;YAChB,QAAQ,CAAC,qBAAqB,CAAC,GAAG,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QAC5D,CAAC;QAED,QAAQ,CAAC,eAAe,EAAE,CAAC;QAE3B,OAAO,GAAG,EAAE;YACV,QAAQ,CAAC,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YACrC,QAAQ,CAAC,qBAAqB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjD,QAAQ,CAAC,qBAAqB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAChD,QAAQ,CAAC,eAAe,EAAE,CAAC;QAC7B,CAAC,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,OAAO;QACZ,IAAI,sBAAsB,CAAC,SAAS,EAAE,CAAC;YACrC,MAAM,QAAQ,GAAG,sBAAsB,CAAC,SAAS,CAAC;YAClD,MAAM,CAAC,mBAAmB,CAAC,SAAS,EAAE,QAAQ,CAAC,aAAa,CAAC,CAAC;YAE9D,QAAQ,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;YACzE,QAAQ,CAAC,qBAAqB,CAAC,KAAK,EAAE,CAAC;YACvC,QAAQ,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;YAE5B,sBAAsB,CAAC,SAAS,GAAG,IAAI,CAAC;QAC1C,CAAC;IACH,CAAC"}

package/dist/model-context/frame/types.d.ts

@@ -0,0 +1,29 @@
+export type SerializedTool = {
+    description?: string;
+    parameters: any;
+    disabled?: boolean;
+    type?: string;
+};
+export type SerializedModelContext = {
+    system?: string;
+    tools?: Record<string, SerializedTool>;
+};
+export type FrameMessageType = "model-context-request" | "model-context-update" | "tool-call" | "tool-result";
+export type FrameMessage = {
+    type: "model-context-request";
+} | {
+    type: "model-context-update";
+    context: SerializedModelContext;
+} | {
+    type: "tool-call";
+    id: string;
+    toolName: string;
+    args: unknown;
+} | {
+    type: "tool-result";
+    id: string;
+    result?: unknown;
+    error?: string;
+};
+export declare const FRAME_MESSAGE_CHANNEL = "assistant-ui-frame";
+//# sourceMappingURL=types.d.ts.map

package/dist/model-context/frame/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/model-context/frame/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,cAAc,GAAG;IAC3B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,GAAG,CAAC;IAChB,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG;IACnC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;CACxC,CAAC;AAEF,MAAM,MAAM,gBAAgB,GACxB,uBAAuB,GACvB,sBAAsB,GACtB,WAAW,GACX,aAAa,CAAC;AAElB,MAAM,MAAM,YAAY,GACpB;IACE,IAAI,EAAE,uBAAuB,CAAC;CAC/B,GACD;IACE,IAAI,EAAE,sBAAsB,CAAC;IAC7B,OAAO,EAAE,sBAAsB,CAAC;CACjC,GACD;IACE,IAAI,EAAE,WAAW,CAAC;IAClB,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,OAAO,CAAC;CACf,GACD;IACE,IAAI,EAAE,aAAa,CAAC;IACpB,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,CAAC;AAEN,eAAO,MAAM,qBAAqB,uBAAuB,CAAC"}

package/dist/model-context/frame/types.js

@@ -0,0 +1,2 @@
+export const FRAME_MESSAGE_CHANNEL = "assistant-ui-frame";
+//# sourceMappingURL=types.js.map

package/dist/model-context/frame/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/model-context/frame/types.ts"],"names":[],"mappings":"AAuCA,MAAM,CAAC,MAAM,qBAAqB,GAAG,oBAAoB,CAAC"}

package/dist/model-context/index.d.ts

@@ -0,0 +1,8 @@
+export type { LanguageModelV1CallSettings, LanguageModelConfig, ModelContext, ModelContextProvider, AssistantToolProps, AssistantInstructionsConfig, } from "./types.js";
+export { mergeModelContexts } from "./types.js";
+export { tool } from "./tool.js";
+export { ModelContextRegistry } from "./registry.js";
+export type { ModelContextRegistryToolHandle, ModelContextRegistryInstructionHandle, ModelContextRegistryProviderHandle, } from "./registry-handles.js";
+export { AssistantFrameHost, AssistantFrameProvider, FRAME_MESSAGE_CHANNEL, } from "./frame/index.js";
+export type { SerializedTool, SerializedModelContext, FrameMessageType, FrameMessage, } from "./frame/index.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/model-context/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/model-context/index.ts"],"names":[],"mappings":"AAAA,YAAY,EAEV,2BAA2B,EAC3B,mBAAmB,EAEnB,YAAY,EACZ,oBAAoB,EAEpB,kBAAkB,EAClB,2BAA2B,GAC5B,mBAAgB;AACjB,OAAO,EAAE,kBAAkB,EAAE,mBAAgB;AAE7C,OAAO,EAAE,IAAI,EAAE,kBAAe;AAE9B,OAAO,EAAE,oBAAoB,EAAE,sBAAmB;AAClD,YAAY,EACV,8BAA8B,EAC9B,qCAAqC,EACrC,kCAAkC,GACnC,8BAA2B;AAE5B,OAAO,EAEL,kBAAkB,EAClB,sBAAsB,EAEtB,qBAAqB,GACtB,yBAAgB;AACjB,YAAY,EACV,cAAc,EACd,sBAAsB,EACtB,gBAAgB,EAChB,YAAY,GACb,yBAAgB"}

package/dist/model-context/index.js

@@ -0,0 +1,9 @@
+export { mergeModelContexts } from "./types.js";
+export { tool } from "./tool.js";
+export { ModelContextRegistry } from "./registry.js";
+export { 
+// Classes
+AssistantFrameHost, AssistantFrameProvider, 
+// Constants
+FRAME_MESSAGE_CHANNEL, } from "./frame/index.js";
+//# sourceMappingURL=index.js.map

package/dist/model-context/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/model-context/index.ts"],"names":[],"mappings":"AAWA,OAAO,EAAE,kBAAkB,EAAE,mBAAgB;AAE7C,OAAO,EAAE,IAAI,EAAE,kBAAe;AAE9B,OAAO,EAAE,oBAAoB,EAAE,sBAAmB;AAOlD,OAAO;AACL,UAAU;AACV,kBAAkB,EAClB,sBAAsB;AACtB,YAAY;AACZ,qBAAqB,GACtB,yBAAgB"}

package/dist/model-context/registry-handles.d.ts

@@ -0,0 +1,13 @@
+import type { AssistantToolProps, AssistantInstructionsConfig } from "./types.js";
+export interface ModelContextRegistryToolHandle<TArgs extends Record<string, unknown> = any, TResult = any> {
+    update(tool: AssistantToolProps<TArgs, TResult>): void;
+    remove(): void;
+}
+export interface ModelContextRegistryInstructionHandle {
+    update(config: string | AssistantInstructionsConfig): void;
+    remove(): void;
+}
+export interface ModelContextRegistryProviderHandle {
+    remove(): void;
+}
+//# sourceMappingURL=registry-handles.d.ts.map

package/dist/model-context/registry-handles.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry-handles.d.ts","sourceRoot":"","sources":["../../src/model-context/registry-handles.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,2BAA2B,EAAE,mBAAgB;AAE/E,MAAM,WAAW,8BAA8B,CAC7C,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,GAAG,EAC3C,OAAO,GAAG,GAAG;IAEb,MAAM,CAAC,IAAI,EAAE,kBAAkB,CAAC,KAAK,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;IACvD,MAAM,IAAI,IAAI,CAAC;CAChB;AAED,MAAM,WAAW,qCAAqC;IACpD,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,2BAA2B,GAAG,IAAI,CAAC;IAC3D,MAAM,IAAI,IAAI,CAAC;CAChB;AAED,MAAM,WAAW,kCAAkC;IACjD,MAAM,IAAI,IAAI,CAAC;CAChB"}

package/dist/model-context/registry-handles.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=registry-handles.js.map

package/dist/model-context/registry-handles.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry-handles.js","sourceRoot":"","sources":["../../src/model-context/registry-handles.ts"],"names":[],"mappings":""}

package/dist/model-context/registry.d.ts

@@ -0,0 +1,17 @@
+import { ModelContext, ModelContextProvider, AssistantToolProps, AssistantInstructionsConfig } from "./types.js";
+import type { Unsubscribe } from "../types/index.js";
+import { ModelContextRegistryToolHandle, ModelContextRegistryInstructionHandle, ModelContextRegistryProviderHandle } from "./registry-handles.js";
+export declare class ModelContextRegistry implements ModelContextProvider {
+    private _tools;
+    private _instructions;
+    private _providers;
+    private _subscribers;
+    private _providerUnsubscribes;
+    getModelContext(): ModelContext;
+    subscribe(callback: () => void): Unsubscribe;
+    private notifySubscribers;
+    addTool<TArgs extends Record<string, unknown>, TResult>(tool: AssistantToolProps<TArgs, TResult>): ModelContextRegistryToolHandle<TArgs, TResult>;
+    addInstruction(config: string | AssistantInstructionsConfig): ModelContextRegistryInstructionHandle;
+    addProvider(provider: ModelContextProvider): ModelContextRegistryProviderHandle;
+}
+//# sourceMappingURL=registry.d.ts.map

package/dist/model-context/registry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry.d.ts","sourceRoot":"","sources":["../../src/model-context/registry.ts"],"names":[],"mappings":"AACA,OAAO,EACL,YAAY,EACZ,oBAAoB,EAEpB,kBAAkB,EAClB,2BAA2B,EAC5B,mBAAgB;AACjB,OAAO,KAAK,EAAE,WAAW,EAAE,0BAAiB;AAC5C,OAAO,EACL,8BAA8B,EAC9B,qCAAqC,EACrC,kCAAkC,EACnC,8BAA2B;AAE5B,qBAAa,oBAAqB,YAAW,oBAAoB;IAC/D,OAAO,CAAC,MAAM,CAAmD;IACjE,OAAO,CAAC,aAAa,CAA6B;IAClD,OAAO,CAAC,UAAU,CAA2C;IAC7D,OAAO,CAAC,YAAY,CAAyB;IAC7C,OAAO,CAAC,qBAAqB,CAA8C;IAE3E,eAAe,IAAI,YAAY;IA4C/B,SAAS,CAAC,QAAQ,EAAE,MAAM,IAAI,GAAG,WAAW;IAK5C,OAAO,CAAC,iBAAiB;IAMzB,OAAO,CAAC,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,OAAO,EACpD,IAAI,EAAE,kBAAkB,CAAC,KAAK,EAAE,OAAO,CAAC,GACvC,8BAA8B,CAAC,KAAK,EAAE,OAAO,CAAC;IAoBjD,cAAc,CACZ,MAAM,EAAE,MAAM,GAAG,2BAA2B,GAC3C,qCAAqC;IAiCxC,WAAW,CACT,QAAQ,EAAE,oBAAoB,GAC7B,kCAAkC;CAsBtC"}

package/dist/model-context/registry.js

@@ -0,0 +1,108 @@
+import { mergeModelContexts, } from "./types.js";
+export class ModelContextRegistry {
+    _tools = new Map();
+    _instructions = new Map();
+    _providers = new Map();
+    _subscribers = new Set();
+    _providerUnsubscribes = new Map();
+    getModelContext() {
+        const instructions = Array.from(this._instructions.values()).filter(Boolean);
+        const system = instructions.length > 0 ? instructions.join("\n\n") : undefined;
+        const tools = {};
+        for (const toolProps of this._tools.values()) {
+            const { toolName, render, ...tool } = toolProps;
+            tools[toolName] = tool;
+        }
+        const providerContexts = mergeModelContexts(new Set(this._providers.values()));
+        const context = {
+            system,
+            tools: Object.keys(tools).length > 0 ? tools : undefined,
+        };
+        if (providerContexts.system) {
+            context.system = context.system
+                ? `${context.system}\n\n${providerContexts.system}`
+                : providerContexts.system;
+        }
+        if (providerContexts.tools) {
+            context.tools = { ...(context.tools || {}), ...providerContexts.tools };
+        }
+        if (providerContexts.callSettings) {
+            context.callSettings = providerContexts.callSettings;
+        }
+        if (providerContexts.config) {
+            context.config = providerContexts.config;
+        }
+        return context;
+    }
+    subscribe(callback) {
+        this._subscribers.add(callback);
+        return () => this._subscribers.delete(callback);
+    }
+    notifySubscribers() {
+        for (const callback of this._subscribers) {
+            callback();
+        }
+    }
+    addTool(tool) {
+        const id = Symbol();
+        this._tools.set(id, tool);
+        this.notifySubscribers();
+        return {
+            update: (newTool) => {
+                if (this._tools.has(id)) {
+                    this._tools.set(id, newTool);
+                    this.notifySubscribers();
+                }
+            },
+            remove: () => {
+                this._tools.delete(id);
+                this.notifySubscribers();
+            },
+        };
+    }
+    addInstruction(config) {
+        const id = Symbol();
+        const instruction = typeof config === "string" ? config : config.instruction;
+        const disabled = typeof config === "object" ? config.disabled : false;
+        if (!disabled) {
+            this._instructions.set(id, instruction);
+            this.notifySubscribers();
+        }
+        return {
+            update: (newConfig) => {
+                const newInstruction = typeof newConfig === "string" ? newConfig : newConfig.instruction;
+                const newDisabled = typeof newConfig === "object" ? newConfig.disabled : false;
+                if (newDisabled) {
+                    this._instructions.delete(id);
+                }
+                else {
+                    this._instructions.set(id, newInstruction);
+                }
+                this.notifySubscribers();
+            },
+            remove: () => {
+                this._instructions.delete(id);
+                this.notifySubscribers();
+            },
+        };
+    }
+    addProvider(provider) {
+        const id = Symbol();
+        this._providers.set(id, provider);
+        const unsubscribe = provider.subscribe?.(() => {
+            this.notifySubscribers();
+        });
+        this._providerUnsubscribes.set(id, unsubscribe);
+        this.notifySubscribers();
+        return {
+            remove: () => {
+                this._providers.delete(id);
+                const unsubscribe = this._providerUnsubscribes.get(id);
+                unsubscribe?.();
+                this._providerUnsubscribes.delete(id);
+                this.notifySubscribers();
+            },
+        };
+    }
+}
+//# sourceMappingURL=registry.js.map

package/dist/model-context/registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry.js","sourceRoot":"","sources":["../../src/model-context/registry.ts"],"names":[],"mappings":"AACA,OAAO,EAGL,kBAAkB,GAGnB,mBAAgB;AAQjB,MAAM,OAAO,oBAAoB;IACvB,MAAM,GAAG,IAAI,GAAG,EAAwC,CAAC;IACzD,aAAa,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC1C,UAAU,GAAG,IAAI,GAAG,EAAgC,CAAC;IACrD,YAAY,GAAG,IAAI,GAAG,EAAc,CAAC;IACrC,qBAAqB,GAAG,IAAI,GAAG,EAAmC,CAAC;IAE3E,eAAe;QACb,MAAM,YAAY,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CACjE,OAAO,CACR,CAAC;QAEF,MAAM,MAAM,GACV,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAElE,MAAM,KAAK,GAAmC,EAAE,CAAC;QACjD,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;YAC7C,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,GAAG,SAAS,CAAC;YAChD,KAAK,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC;QACzB,CAAC;QAED,MAAM,gBAAgB,GAAG,kBAAkB,CACzC,IAAI,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,CAClC,CAAC;QAEF,MAAM,OAAO,GAAiB;YAC5B,MAAM;YACN,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;SACzD,CAAC;QAEF,IAAI,gBAAgB,CAAC,MAAM,EAAE,CAAC;YAC5B,OAAO,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM;gBAC7B,CAAC,CAAC,GAAG,OAAO,CAAC,MAAM,OAAO,gBAAgB,CAAC,MAAM,EAAE;gBACnD,CAAC,CAAC,gBAAgB,CAAC,MAAM,CAAC;QAC9B,CAAC;QAED,IAAI,gBAAgB,CAAC,KAAK,EAAE,CAAC;YAC3B,OAAO,CAAC,KAAK,GAAG,EAAE,GAAG,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,GAAG,gBAAgB,CAAC,KAAK,EAAE,CAAC;QAC1E,CAAC;QAED,IAAI,gBAAgB,CAAC,YAAY,EAAE,CAAC;YAClC,OAAO,CAAC,YAAY,GAAG,gBAAgB,CAAC,YAAY,CAAC;QACvD,CAAC;QAED,IAAI,gBAAgB,CAAC,MAAM,EAAE,CAAC;YAC5B,OAAO,CAAC,MAAM,GAAG,gBAAgB,CAAC,MAAM,CAAC;QAC3C,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,SAAS,CAAC,QAAoB;QAC5B,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAChC,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAClD,CAAC;IAEO,iBAAiB;QACvB,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACzC,QAAQ,EAAE,CAAC;QACb,CAAC;IACH,CAAC;IAED,OAAO,CACL,IAAwC;QAExC,MAAM,EAAE,GAAG,MAAM,EAAE,CAAC;QAEpB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;QAC1B,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAEzB,OAAO;YACL,MAAM,EAAE,CAAC,OAA2C,EAAE,EAAE;gBACtD,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;oBACxB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;oBAC7B,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAC3B,CAAC;YACH,CAAC;YACD,MAAM,EAAE,GAAG,EAAE;gBACX,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACvB,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,CAAC;SACF,CAAC;IACJ,CAAC;IAED,cAAc,CACZ,MAA4C;QAE5C,MAAM,EAAE,GAAG,MAAM,EAAE,CAAC;QAEpB,MAAM,WAAW,GACf,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC;QAC3D,MAAM,QAAQ,GAAG,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC;QAEtE,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;YACxC,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC3B,CAAC;QAED,OAAO;YACL,MAAM,EAAE,CAAC,SAA+C,EAAE,EAAE;gBAC1D,MAAM,cAAc,GAClB,OAAO,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC;gBACpE,MAAM,WAAW,GACf,OAAO,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC;gBAE7D,IAAI,WAAW,EAAE,CAAC;oBAChB,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBAChC,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;gBAC7C,CAAC;gBACD,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,CAAC;YACD,MAAM,EAAE,GAAG,EAAE;gBACX,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBAC9B,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,CAAC;SACF,CAAC;IACJ,CAAC;IAED,WAAW,CACT,QAA8B;QAE9B,MAAM,EAAE,GAAG,MAAM,EAAE,CAAC;QAEpB,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;QAElC,MAAM,WAAW,GAAG,QAAQ,CAAC,SAAS,EAAE,CAAC,GAAG,EAAE;YAC5C,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QAEhD,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAEzB,OAAO;YACL,MAAM,EAAE,GAAG,EAAE;gBACX,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBAC3B,MAAM,WAAW,GAAG,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACvD,WAAW,EAAE,EAAE,CAAC;gBAChB,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACtC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,CAAC;SACF,CAAC;IACJ,CAAC;CACF"}

package/dist/model-context/tool.d.ts

@@ -0,0 +1,3 @@
+import { Tool } from "assistant-stream";
+export declare function tool<TArgs extends Record<string, unknown>, TResult = any>(tool: Tool<TArgs, TResult>): Tool<TArgs, TResult>;
+//# sourceMappingURL=tool.d.ts.map

package/dist/model-context/tool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool.d.ts","sourceRoot":"","sources":["../../src/model-context/tool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AAIxC,wBAAgB,IAAI,CAAC,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,GAAG,EACvE,IAAI,EAAE,IAAI,CAAC,KAAK,EAAE,OAAO,CAAC,GACzB,IAAI,CAAC,KAAK,EAAE,OAAO,CAAC,CAEtB"}

package/dist/model-context/tool.js

@@ -0,0 +1,5 @@
+// TODO re-add the inferrence of the parameters
+export function tool(tool) {
+    return tool;
+}
+//# sourceMappingURL=tool.js.map

package/dist/model-context/tool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool.js","sourceRoot":"","sources":["../../src/model-context/tool.ts"],"names":[],"mappings":"AAEA,+CAA+C;AAE/C,MAAM,UAAU,IAAI,CAClB,IAA0B;IAE1B,OAAO,IAAI,CAAC;AACd,CAAC"}