@assistant-ui/core 0.2.4 → 0.2.5

package/dist/model-context/registry.js

@@ -1,108 +1,94 @@
-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();
-            },
-        };
-    }
-}
+import { mergeModelContexts } from "./types.js";
+//#region src/model-context/registry.ts
+var ModelContextRegistry = class {
+	_tools = /* @__PURE__ */ new Map();
+	_instructions = /* @__PURE__ */ new Map();
+	_providers = /* @__PURE__ */ new Map();
+	_subscribers = /* @__PURE__ */ new Set();
+	_providerUnsubscribes = /* @__PURE__ */ new Map();
+	getModelContext() {
+		const instructions = Array.from(this._instructions.values()).filter(Boolean);
+		const system = instructions.length > 0 ? instructions.join("\n\n") : void 0;
+		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 : void 0
+		};
+		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;
+		if (!(typeof config === "object" ? config.disabled : false)) {
+			this._instructions.set(id, instruction);
+			this.notifySubscribers();
+		}
+		return {
+			update: (newConfig) => {
+				const newInstruction = typeof newConfig === "string" ? newConfig : newConfig.instruction;
+				if (typeof newConfig === "object" ? newConfig.disabled : false) 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);
+			this._providerUnsubscribes.get(id)?.();
+			this._providerUnsubscribes.delete(id);
+			this.notifySubscribers();
+		} };
+	}
+};
+//#endregion
+export { ModelContextRegistry };
+
 //# sourceMappingURL=registry.js.map

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

@@ -1 +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"}
+{"version":3,"file":"registry.js","names":[],"sources":["../../src/model-context/registry.ts"],"sourcesContent":["import type { Tool } from \"assistant-stream\";\nimport {\n  type ModelContext,\n  type ModelContextProvider,\n  mergeModelContexts,\n  type AssistantToolProps,\n  type AssistantInstructionsConfig,\n} from \"./types\";\nimport type { Unsubscribe } from \"../types/unsubscribe\";\nimport type {\n  ModelContextRegistryToolHandle,\n  ModelContextRegistryInstructionHandle,\n  ModelContextRegistryProviderHandle,\n} from \"./registry-handles\";\n\nexport class ModelContextRegistry implements ModelContextProvider {\n  private _tools = new Map<symbol, AssistantToolProps<any, any>>();\n  private _instructions = new Map<symbol, string>();\n  private _providers = new Map<symbol, ModelContextProvider>();\n  private _subscribers = new Set<() => void>();\n  private _providerUnsubscribes = new Map<symbol, Unsubscribe | undefined>();\n\n  getModelContext(): ModelContext {\n    const instructions = Array.from(this._instructions.values()).filter(\n      Boolean,\n    );\n\n    const system =\n      instructions.length > 0 ? instructions.join(\"\\n\\n\") : undefined;\n\n    const tools: Record<string, Tool<any, any>> = {};\n    for (const toolProps of this._tools.values()) {\n      const { toolName, render, ...tool } = toolProps;\n      tools[toolName] = tool;\n    }\n\n    const providerContexts = mergeModelContexts(\n      new Set(this._providers.values()),\n    );\n\n    const context: ModelContext = {\n      system,\n      tools: Object.keys(tools).length > 0 ? tools : undefined,\n    };\n\n    if (providerContexts.system) {\n      context.system = context.system\n        ? `${context.system}\\n\\n${providerContexts.system}`\n        : providerContexts.system;\n    }\n\n    if (providerContexts.tools) {\n      context.tools = { ...(context.tools || {}), ...providerContexts.tools };\n    }\n\n    if (providerContexts.callSettings) {\n      context.callSettings = providerContexts.callSettings;\n    }\n\n    if (providerContexts.config) {\n      context.config = providerContexts.config;\n    }\n\n    return context;\n  }\n\n  subscribe(callback: () => void): Unsubscribe {\n    this._subscribers.add(callback);\n    return () => this._subscribers.delete(callback);\n  }\n\n  private notifySubscribers(): void {\n    for (const callback of this._subscribers) {\n      callback();\n    }\n  }\n\n  addTool<TArgs extends Record<string, unknown>, TResult>(\n    tool: AssistantToolProps<TArgs, TResult>,\n  ): ModelContextRegistryToolHandle<TArgs, TResult> {\n    const id = Symbol();\n\n    this._tools.set(id, tool);\n    this.notifySubscribers();\n\n    return {\n      update: (newTool: AssistantToolProps<TArgs, TResult>) => {\n        if (this._tools.has(id)) {\n          this._tools.set(id, newTool);\n          this.notifySubscribers();\n        }\n      },\n      remove: () => {\n        this._tools.delete(id);\n        this.notifySubscribers();\n      },\n    };\n  }\n\n  addInstruction(\n    config: string | AssistantInstructionsConfig,\n  ): ModelContextRegistryInstructionHandle {\n    const id = Symbol();\n\n    const instruction =\n      typeof config === \"string\" ? config : config.instruction;\n    const disabled = typeof config === \"object\" ? config.disabled : false;\n\n    if (!disabled) {\n      this._instructions.set(id, instruction);\n      this.notifySubscribers();\n    }\n\n    return {\n      update: (newConfig: string | AssistantInstructionsConfig) => {\n        const newInstruction =\n          typeof newConfig === \"string\" ? newConfig : newConfig.instruction;\n        const newDisabled =\n          typeof newConfig === \"object\" ? newConfig.disabled : false;\n\n        if (newDisabled) {\n          this._instructions.delete(id);\n        } else {\n          this._instructions.set(id, newInstruction);\n        }\n        this.notifySubscribers();\n      },\n      remove: () => {\n        this._instructions.delete(id);\n        this.notifySubscribers();\n      },\n    };\n  }\n\n  addProvider(\n    provider: ModelContextProvider,\n  ): ModelContextRegistryProviderHandle {\n    const id = Symbol();\n\n    this._providers.set(id, provider);\n\n    const unsubscribe = provider.subscribe?.(() => {\n      this.notifySubscribers();\n    });\n    this._providerUnsubscribes.set(id, unsubscribe);\n\n    this.notifySubscribers();\n\n    return {\n      remove: () => {\n        this._providers.delete(id);\n        const unsubscribe = this._providerUnsubscribes.get(id);\n        unsubscribe?.();\n        this._providerUnsubscribes.delete(id);\n        this.notifySubscribers();\n      },\n    };\n  }\n}\n"],"mappings":";;AAeA,IAAa,uBAAb,MAAkE;CAChE,yBAAiB,IAAI,IAA0C;CAC/D,gCAAwB,IAAI,IAAoB;CAChD,6BAAqB,IAAI,IAAkC;CAC3D,+BAAuB,IAAI,IAAgB;CAC3C,wCAAgC,IAAI,IAAqC;CAEzE,kBAAgC;EAC9B,MAAM,eAAe,MAAM,KAAK,KAAK,cAAc,OAAO,CAAC,EAAE,OAC3D,OACF;EAEA,MAAM,SACJ,aAAa,SAAS,IAAI,aAAa,KAAK,MAAM,IAAI,KAAA;EAExD,MAAM,QAAwC,CAAC;EAC/C,KAAK,MAAM,aAAa,KAAK,OAAO,OAAO,GAAG;GAC5C,MAAM,EAAE,UAAU,QAAQ,GAAG,SAAS;GACtC,MAAM,YAAY;EACpB;EAEA,MAAM,mBAAmB,mBACvB,IAAI,IAAI,KAAK,WAAW,OAAO,CAAC,CAClC;EAEA,MAAM,UAAwB;GAC5B;GACA,OAAO,OAAO,KAAK,KAAK,EAAE,SAAS,IAAI,QAAQ,KAAA;EACjD;EAEA,IAAI,iBAAiB,QACnB,QAAQ,SAAS,QAAQ,SACrB,GAAG,QAAQ,OAAO,MAAM,iBAAiB,WACzC,iBAAiB;EAGvB,IAAI,iBAAiB,OACnB,QAAQ,QAAQ;GAAE,GAAI,QAAQ,SAAS,CAAC;GAAI,GAAG,iBAAiB;EAAM;EAGxE,IAAI,iBAAiB,cACnB,QAAQ,eAAe,iBAAiB;EAG1C,IAAI,iBAAiB,QACnB,QAAQ,SAAS,iBAAiB;EAGpC,OAAO;CACT;CAEA,UAAU,UAAmC;EAC3C,KAAK,aAAa,IAAI,QAAQ;EAC9B,aAAa,KAAK,aAAa,OAAO,QAAQ;CAChD;CAEA,oBAAkC;EAChC,KAAK,MAAM,YAAY,KAAK,cAC1B,SAAS;CAEb;CAEA,QACE,MACgD;EAChD,MAAM,KAAK,OAAO;EAElB,KAAK,OAAO,IAAI,IAAI,IAAI;EACxB,KAAK,kBAAkB;EAEvB,OAAO;GACL,SAAS,YAAgD;IACvD,IAAI,KAAK,OAAO,IAAI,EAAE,GAAG;KACvB,KAAK,OAAO,IAAI,IAAI,OAAO;KAC3B,KAAK,kBAAkB;IACzB;GACF;GACA,cAAc;IACZ,KAAK,OAAO,OAAO,EAAE;IACrB,KAAK,kBAAkB;GACzB;EACF;CACF;CAEA,eACE,QACuC;EACvC,MAAM,KAAK,OAAO;EAElB,MAAM,cACJ,OAAO,WAAW,WAAW,SAAS,OAAO;EAG/C,IAAI,EAFa,OAAO,WAAW,WAAW,OAAO,WAAW,QAEjD;GACb,KAAK,cAAc,IAAI,IAAI,WAAW;GACtC,KAAK,kBAAkB;EACzB;EAEA,OAAO;GACL,SAAS,cAAoD;IAC3D,MAAM,iBACJ,OAAO,cAAc,WAAW,YAAY,UAAU;IAIxD,IAFE,OAAO,cAAc,WAAW,UAAU,WAAW,OAGrD,KAAK,cAAc,OAAO,EAAE;SAE5B,KAAK,cAAc,IAAI,IAAI,cAAc;IAE3C,KAAK,kBAAkB;GACzB;GACA,cAAc;IACZ,KAAK,cAAc,OAAO,EAAE;IAC5B,KAAK,kBAAkB;GACzB;EACF;CACF;CAEA,YACE,UACoC;EACpC,MAAM,KAAK,OAAO;EAElB,KAAK,WAAW,IAAI,IAAI,QAAQ;EAEhC,MAAM,cAAc,SAAS,kBAAkB;GAC7C,KAAK,kBAAkB;EACzB,CAAC;EACD,KAAK,sBAAsB,IAAI,IAAI,WAAW;EAE9C,KAAK,kBAAkB;EAEvB,OAAO,EACL,cAAc;GACZ,KAAK,WAAW,OAAO,EAAE;GAEzB,KADyB,sBAAsB,IAAI,EACzC,IAAI;GACd,KAAK,sBAAsB,OAAO,EAAE;GACpC,KAAK,kBAAkB;EACzB,EACF;CACF;AACF"}

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

@@ -1,4 +1,6 @@
-import type { Tool } from "assistant-stream";
+import { Tool } from "assistant-stream";
+
+//#region src/model-context/tool.d.ts
 /**
  * Defines a model tool with its argument schema, execution behavior, and
  * optional model-output conversion.
@@ -24,5 +26,7 @@ import type { Tool } from "assistant-stream";
  * });
  * ```
  */
-export declare function tool<TArgs extends Record<string, unknown>, TResult = any>(tool: Tool<TArgs, TResult>): Tool<TArgs, TResult>;
+declare function tool<TArgs extends Record<string, unknown>, TResult = any>(tool: Tool<TArgs, TResult>): Tool<TArgs, TResult>;
+//#endregion
+export { tool };
 //# sourceMappingURL=tool.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"tool.d.ts","sourceRoot":"","sources":["../../src/model-context/tool.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AAI7C;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,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"}
+{"version":3,"file":"tool.d.ts","names":[],"sources":["../../src/model-context/tool.ts"],"mappings":";;;;;AA6BA;;;;;;;;;;;;;;;;;;;;;;;iBAAgB,IAAA,eAAmB,MAAA,iCAAA,CACjC,IAAA,EAAM,IAAA,CAAK,KAAA,EAAO,OAAA,IACjB,IAAA,CAAK,KAAA,EAAO,OAAA"}

package/dist/model-context/tool.js

@@ -1,30 +1,33 @@
-// TODO re-add the inferrence of the parameters
+//#region src/model-context/tool.ts
 /**
- * Defines a model tool with its argument schema, execution behavior, and
- * optional model-output conversion.
- *
- * This helper keeps reusable tool definitions type-checked and convenient to
- * export for a {@link Toolkit}, {@link Tools}, or {@link useAssistantTool}.
- * Inference from parameter schemas is currently limited, so provide generic
- * arguments when you need precise args or result types.
- *
- * @param tool - Tool definition to expose to the assistant model.
- *
- * @example
- * ```ts
- * const getWeather = tool<{ city: string }, string>({
- *   type: "frontend",
- *   description: "Get the weather for a city.",
- *   parameters: {
- *     type: "object",
- *     properties: { city: { type: "string" } },
- *     required: ["city"],
- *   },
- *   execute: async ({ city }) => `Sunny in ${city}`,
- * });
- * ```
- */
-export function tool(tool) {
-    return tool;
+* Defines a model tool with its argument schema, execution behavior, and
+* optional model-output conversion.
+*
+* This helper keeps reusable tool definitions type-checked and convenient to
+* export for a {@link Toolkit}, {@link Tools}, or {@link useAssistantTool}.
+* Inference from parameter schemas is currently limited, so provide generic
+* arguments when you need precise args or result types.
+*
+* @param tool - Tool definition to expose to the assistant model.
+*
+* @example
+* ```ts
+* const getWeather = tool<{ city: string }, string>({
+*   type: "frontend",
+*   description: "Get the weather for a city.",
+*   parameters: {
+*     type: "object",
+*     properties: { city: { type: "string" } },
+*     required: ["city"],
+*   },
+*   execute: async ({ city }) => `Sunny in ${city}`,
+* });
+* ```
+*/
+function tool(tool) {
+	return tool;
 }
+//#endregion
+export { tool };
+
 //# sourceMappingURL=tool.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"tool.js","sourceRoot":"","sources":["../../src/model-context/tool.ts"],"names":[],"mappings":"AAEA,+CAA+C;AAE/C;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,MAAM,UAAU,IAAI,CAClB,IAA0B;IAE1B,OAAO,IAAI,CAAC;AACd,CAAC"}
+{"version":3,"file":"tool.js","names":[],"sources":["../../src/model-context/tool.ts"],"sourcesContent":["import type { Tool } from \"assistant-stream\";\n\n// TODO re-add the inferrence of the parameters\n\n/**\n * Defines a model tool with its argument schema, execution behavior, and\n * optional model-output conversion.\n *\n * This helper keeps reusable tool definitions type-checked and convenient to\n * export for a {@link Toolkit}, {@link Tools}, or {@link useAssistantTool}.\n * Inference from parameter schemas is currently limited, so provide generic\n * arguments when you need precise args or result types.\n *\n * @param tool - Tool definition to expose to the assistant model.\n *\n * @example\n * ```ts\n * const getWeather = tool<{ city: string }, string>({\n *   type: \"frontend\",\n *   description: \"Get the weather for a city.\",\n *   parameters: {\n *     type: \"object\",\n *     properties: { city: { type: \"string\" } },\n *     required: [\"city\"],\n *   },\n *   execute: async ({ city }) => `Sunny in ${city}`,\n * });\n * ```\n */\nexport function tool<TArgs extends Record<string, unknown>, TResult = any>(\n  tool: Tool<TArgs, TResult>,\n): Tool<TArgs, TResult> {\n  return tool;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AA6BA,SAAgB,KACd,MACsB;CACtB,OAAO;AACT"}

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

@@ -1,41 +1,45 @@
-import type { Unsubscribe } from "../types/unsubscribe.js";
-import type { Tool } from "assistant-stream";
-export type LanguageModelV1CallSettings = {
-    maxTokens?: number;
-    temperature?: number;
-    topP?: number;
-    presencePenalty?: number;
-    frequencyPenalty?: number;
-    seed?: number;
-    headers?: Record<string, string | undefined>;
+import { Unsubscribe } from "../types/unsubscribe.js";
+import { Tool } from "assistant-stream";
+
+//#region src/model-context/types.d.ts
+type LanguageModelV1CallSettings = {
+  maxTokens?: number;
+  temperature?: number;
+  topP?: number;
+  presencePenalty?: number;
+  frequencyPenalty?: number;
+  seed?: number;
+  headers?: Record<string, string | undefined>;
 };
-export type LanguageModelConfig = {
-    apiKey?: string;
-    baseUrl?: string;
-    modelName?: string;
+type LanguageModelConfig = {
+  apiKey?: string;
+  baseUrl?: string;
+  modelName?: string;
 };
-export type ModelContext = {
-    priority?: number | undefined;
-    system?: string | undefined;
-    tools?: Record<string, Tool<any, any>> | undefined;
-    callSettings?: LanguageModelV1CallSettings | undefined;
-    config?: LanguageModelConfig | undefined;
+type ModelContext = {
+  priority?: number | undefined;
+  system?: string | undefined;
+  tools?: Record<string, Tool<any, any>> | undefined;
+  callSettings?: LanguageModelV1CallSettings | undefined;
+  config?: LanguageModelConfig | undefined;
 };
-export type ModelContextProvider = {
-    getModelContext: () => ModelContext;
-    subscribe?: (callback: () => void) => Unsubscribe;
+type ModelContextProvider = {
+  getModelContext: () => ModelContext;
+  subscribe?: (callback: () => void) => Unsubscribe;
 };
-export type AssistantToolProps<TArgs extends Record<string, unknown>, TResult> = Tool<TArgs, TResult> & {
-    toolName: string;
-    render?: unknown;
+type AssistantToolProps<TArgs extends Record<string, unknown>, TResult> = Tool<TArgs, TResult> & {
+  toolName: string;
+  render?: unknown;
 };
-export type AssistantInstructionsConfig = {
-    disabled?: boolean | undefined;
-    instruction: string;
+type AssistantInstructionsConfig = {
+  disabled?: boolean | undefined;
+  instruction: string;
 };
-export type AssistantContextConfig = {
-    getContext: () => string;
-    disabled?: boolean | undefined;
+type AssistantContextConfig = {
+  getContext: () => string;
+  disabled?: boolean | undefined;
 };
-export declare const mergeModelContexts: (configSet: Set<ModelContextProvider>) => ModelContext;
+declare const mergeModelContexts: (configSet: Set<ModelContextProvider>) => ModelContext;
+//#endregion
+export { AssistantContextConfig, AssistantInstructionsConfig, AssistantToolProps, LanguageModelConfig, LanguageModelV1CallSettings, ModelContext, ModelContextProvider, mergeModelContexts };
 //# sourceMappingURL=types.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/model-context/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,gCAA6B;AACxD,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AAE7C,MAAM,MAAM,2BAA2B,GAAG;IACxC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC,CAAC;CAC9C,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG;IAChC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,YAAY,GAAG;IACzB,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,MAAM,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC5B,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,GAAG,SAAS,CAAC;IACnD,YAAY,CAAC,EAAE,2BAA2B,GAAG,SAAS,CAAC;IACvD,MAAM,CAAC,EAAE,mBAAmB,GAAG,SAAS,CAAC;CAC1C,CAAC;AAEF,MAAM,MAAM,oBAAoB,GAAG;IACjC,eAAe,EAAE,MAAM,YAAY,CAAC;IACpC,SAAS,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,IAAI,KAAK,WAAW,CAAC;CACnD,CAAC;AAEF,MAAM,MAAM,kBAAkB,CAC5B,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACrC,OAAO,IACL,IAAI,CAAC,KAAK,EAAE,OAAO,CAAC,GAAG;IACzB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,OAAO,CAAC;CAClB,CAAC;AAEF,MAAM,MAAM,2BAA2B,GAAG;IACxC,QAAQ,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IAC/B,WAAW,EAAE,MAAM,CAAC;CACrB,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG;IACnC,UAAU,EAAE,MAAM,MAAM,CAAC;IACzB,QAAQ,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;CAChC,CAAC;AAEF,eAAO,MAAM,kBAAkB,GAC7B,WAAW,GAAG,CAAC,oBAAoB,CAAC,KACnC,YAwCF,CAAC"}
+{"version":3,"file":"types.d.ts","names":[],"sources":["../../src/model-context/types.ts"],"mappings":";;;;KAGY,2BAAA;EACV,SAAA;EACA,WAAA;EACA,IAAA;EACA,eAAA;EACA,gBAAA;EACA,IAAA;EACA,OAAA,GAAU,MAAM;AAAA;AAAA,KAGN,mBAAA;EACV,MAAA;EACA,OAAA;EACA,SAAA;AAAA;AAAA,KAGU,YAAA;EACV,QAAA;EACA,MAAA;EACA,KAAA,GAAQ,MAAA,SAAe,IAAA;EACvB,YAAA,GAAe,2BAAA;EACf,MAAA,GAAS,mBAAA;AAAA;AAAA,KAGC,oBAAA;EACV,eAAA,QAAuB,YAAA;EACvB,SAAA,IAAa,QAAA,iBAAyB,WAAW;AAAA;AAAA,KAGvC,kBAAA,eACI,MAAA,8BAEZ,IAAA,CAAK,KAAA,EAAO,OAAA;EACd,QAAA;EACA,MAAA;AAAA;AAAA,KAGU,2BAAA;EACV,QAAA;EACA,WAAW;AAAA;AAAA,KAGD,sBAAA;EACV,UAAA;EACA,QAAQ;AAAA;AAAA,cAGG,kBAAA,GACX,SAAA,EAAW,GAAA,CAAI,oBAAA,MACd,YAAA"}

package/dist/model-context/types.js

@@ -1,40 +1,26 @@
-export const mergeModelContexts = (configSet) => {
-    const configs = Array.from(configSet)
-        .map((c) => c.getModelContext())
-        .sort((a, b) => (b.priority ?? 0) - (a.priority ?? 0));
-    return configs.reduce((acc, config) => {
-        if (config.system) {
-            if (acc.system) {
-                acc.system += `\n\n${config.system}`;
-            }
-            else {
-                acc.system = config.system;
-            }
-        }
-        if (config.tools) {
-            for (const [name, tool] of Object.entries(config.tools)) {
-                const existing = acc.tools?.[name];
-                if (existing && existing !== tool) {
-                    throw new Error(`You tried to define a tool with the name ${name}, but it already exists.`);
-                }
-                if (!acc.tools)
-                    acc.tools = {};
-                acc.tools[name] = tool;
-            }
-        }
-        if (config.config) {
-            acc.config = {
-                ...acc.config,
-                ...config.config,
-            };
-        }
-        if (config.callSettings) {
-            acc.callSettings = {
-                ...acc.callSettings,
-                ...config.callSettings,
-            };
-        }
-        return acc;
-    }, {});
+//#region src/model-context/types.ts
+const mergeModelContexts = (configSet) => {
+	return Array.from(configSet).map((c) => c.getModelContext()).sort((a, b) => (b.priority ?? 0) - (a.priority ?? 0)).reduce((acc, config) => {
+		if (config.system) if (acc.system) acc.system += `\n\n${config.system}`;
+		else acc.system = config.system;
+		if (config.tools) for (const [name, tool] of Object.entries(config.tools)) {
+			const existing = acc.tools?.[name];
+			if (existing && existing !== tool) throw new Error(`You tried to define a tool with the name ${name}, but it already exists.`);
+			if (!acc.tools) acc.tools = {};
+			acc.tools[name] = tool;
+		}
+		if (config.config) acc.config = {
+			...acc.config,
+			...config.config
+		};
+		if (config.callSettings) acc.callSettings = {
+			...acc.callSettings,
+			...config.callSettings
+		};
+		return acc;
+	}, {});
 };
+//#endregion
+export { mergeModelContexts };
+
 //# sourceMappingURL=types.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/model-context/types.ts"],"names":[],"mappings":"AAkDA,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAChC,SAAoC,EACtB,EAAE;IAChB,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC;SAClC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,eAAe,EAAE,CAAC;SAC/B,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,CAAC,CAAC;IAEzD,OAAO,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE;QACpC,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;gBACf,GAAG,CAAC,MAAM,IAAI,OAAO,MAAM,CAAC,MAAM,EAAE,CAAC;YACvC,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;YAC7B,CAAC;QACH,CAAC;QACD,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YACjB,KAAK,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;gBACxD,MAAM,QAAQ,GAAG,GAAG,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,CAAC;gBACnC,IAAI,QAAQ,IAAI,QAAQ,KAAK,IAAI,EAAE,CAAC;oBAClC,MAAM,IAAI,KAAK,CACb,4CAA4C,IAAI,0BAA0B,CAC3E,CAAC;gBACJ,CAAC;gBAED,IAAI,CAAC,GAAG,CAAC,KAAK;oBAAE,GAAG,CAAC,KAAK,GAAG,EAAE,CAAC;gBAC/B,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;YACzB,CAAC;QACH,CAAC;QACD,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,GAAG,CAAC,MAAM,GAAG;gBACX,GAAG,GAAG,CAAC,MAAM;gBACb,GAAG,MAAM,CAAC,MAAM;aACjB,CAAC;QACJ,CAAC;QACD,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;YACxB,GAAG,CAAC,YAAY,GAAG;gBACjB,GAAG,GAAG,CAAC,YAAY;gBACnB,GAAG,MAAM,CAAC,YAAY;aACvB,CAAC;QACJ,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC,EAAE,EAAkB,CAAC,CAAC;AACzB,CAAC,CAAC"}
+{"version":3,"file":"types.js","names":[],"sources":["../../src/model-context/types.ts"],"sourcesContent":["import type { Unsubscribe } from \"../types/unsubscribe\";\nimport type { Tool } from \"assistant-stream\";\n\nexport type LanguageModelV1CallSettings = {\n  maxTokens?: number;\n  temperature?: number;\n  topP?: number;\n  presencePenalty?: number;\n  frequencyPenalty?: number;\n  seed?: number;\n  headers?: Record<string, string | undefined>;\n};\n\nexport type LanguageModelConfig = {\n  apiKey?: string;\n  baseUrl?: string;\n  modelName?: string;\n};\n\nexport type ModelContext = {\n  priority?: number | undefined;\n  system?: string | undefined;\n  tools?: Record<string, Tool<any, any>> | undefined;\n  callSettings?: LanguageModelV1CallSettings | undefined;\n  config?: LanguageModelConfig | undefined;\n};\n\nexport type ModelContextProvider = {\n  getModelContext: () => ModelContext;\n  subscribe?: (callback: () => void) => Unsubscribe;\n};\n\nexport type AssistantToolProps<\n  TArgs extends Record<string, unknown>,\n  TResult,\n> = Tool<TArgs, TResult> & {\n  toolName: string;\n  render?: unknown;\n};\n\nexport type AssistantInstructionsConfig = {\n  disabled?: boolean | undefined;\n  instruction: string;\n};\n\nexport type AssistantContextConfig = {\n  getContext: () => string;\n  disabled?: boolean | undefined;\n};\n\nexport const mergeModelContexts = (\n  configSet: Set<ModelContextProvider>,\n): ModelContext => {\n  const configs = Array.from(configSet)\n    .map((c) => c.getModelContext())\n    .sort((a, b) => (b.priority ?? 0) - (a.priority ?? 0));\n\n  return configs.reduce((acc, config) => {\n    if (config.system) {\n      if (acc.system) {\n        acc.system += `\\n\\n${config.system}`;\n      } else {\n        acc.system = config.system;\n      }\n    }\n    if (config.tools) {\n      for (const [name, tool] of Object.entries(config.tools)) {\n        const existing = acc.tools?.[name];\n        if (existing && existing !== tool) {\n          throw new Error(\n            `You tried to define a tool with the name ${name}, but it already exists.`,\n          );\n        }\n\n        if (!acc.tools) acc.tools = {};\n        acc.tools[name] = tool;\n      }\n    }\n    if (config.config) {\n      acc.config = {\n        ...acc.config,\n        ...config.config,\n      };\n    }\n    if (config.callSettings) {\n      acc.callSettings = {\n        ...acc.callSettings,\n        ...config.callSettings,\n      };\n    }\n    return acc;\n  }, {} as ModelContext);\n};\n"],"mappings":";AAkDA,MAAa,sBACX,cACiB;CAKjB,OAJgB,MAAM,KAAK,SAAS,EACjC,KAAK,MAAM,EAAE,gBAAgB,CAAC,EAC9B,MAAM,GAAG,OAAO,EAAE,YAAY,MAAM,EAAE,YAAY,EAExC,EAAE,QAAQ,KAAK,WAAW;EACrC,IAAI,OAAO,QACT,IAAI,IAAI,QACN,IAAI,UAAU,OAAO,OAAO;OAE5B,IAAI,SAAS,OAAO;EAGxB,IAAI,OAAO,OACT,KAAK,MAAM,CAAC,MAAM,SAAS,OAAO,QAAQ,OAAO,KAAK,GAAG;GACvD,MAAM,WAAW,IAAI,QAAQ;GAC7B,IAAI,YAAY,aAAa,MAC3B,MAAM,IAAI,MACR,4CAA4C,KAAK,yBACnD;GAGF,IAAI,CAAC,IAAI,OAAO,IAAI,QAAQ,CAAC;GAC7B,IAAI,MAAM,QAAQ;EACpB;EAEF,IAAI,OAAO,QACT,IAAI,SAAS;GACX,GAAG,IAAI;GACP,GAAG,OAAO;EACZ;EAEF,IAAI,OAAO,cACT,IAAI,eAAe;GACjB,GAAG,IAAI;GACP,GAAG,OAAO;EACZ;EAEF,OAAO;CACT,GAAG,CAAC,CAAiB;AACvB"}

package/dist/react/AssistantProvider.d.ts

@@ -1,10 +1,14 @@
-import { type ComponentType, type FC, type PropsWithChildren } from "react";
-import { type AssistantClient } from "@assistant-ui/store";
-import type { AssistantRuntime } from "../runtime/api/assistant-runtime.js";
-export declare const getRenderComponent: (runtime: AssistantRuntime) => ComponentType | undefined;
-export type AssistantProviderBaseProps = PropsWithChildren<{
-    runtime: AssistantRuntime;
-    aui?: AssistantClient | null;
+import { AssistantRuntime } from "../runtime/api/assistant-runtime.js";
+import { ComponentType, FC, PropsWithChildren } from "react";
+import { AssistantClient } from "@assistant-ui/store";
+
+//#region src/react/AssistantProvider.d.ts
+declare const getRenderComponent: (runtime: AssistantRuntime) => ComponentType | undefined;
+type AssistantProviderBaseProps = PropsWithChildren<{
+  runtime: AssistantRuntime;
+  aui?: AssistantClient | null;
 }>;
-export declare const AssistantProviderBase: FC<AssistantProviderBaseProps>;
+declare const AssistantProviderBase: FC<AssistantProviderBaseProps>;
+//#endregion
+export { AssistantProviderBase, AssistantProviderBaseProps, getRenderComponent };
 //# sourceMappingURL=AssistantProvider.d.ts.map

package/dist/react/AssistantProvider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AssistantProvider.d.ts","sourceRoot":"","sources":["../../src/react/AssistantProvider.tsx"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,aAAa,EAClB,KAAK,EAAE,EACP,KAAK,iBAAiB,EACvB,MAAM,OAAO,CAAC;AACf,OAAO,EAAuB,KAAK,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAChF,OAAO,KAAK,EAAE,gBAAgB,EAAE,4CAAyC;AAIzE,eAAO,MAAM,kBAAkB,GAAI,SAAS,gBAAgB,KAEtD,aAAa,GACb,SACL,CAAC;AAEF,MAAM,MAAM,0BAA0B,GAAG,iBAAiB,CAAC;IACzD,OAAO,EAAE,gBAAgB,CAAC;IAC1B,GAAG,CAAC,EAAE,eAAe,GAAG,IAAI,CAAC;CAC9B,CAAC,CAAC;AAEH,eAAO,MAAM,qBAAqB,EAAE,EAAE,CAAC,0BAA0B,CAWhE,CAAC"}
+{"version":3,"file":"AssistantProvider.d.ts","names":[],"sources":["../../src/react/AssistantProvider.tsx"],"mappings":";;;;;cAWa,kBAAA,GAAsB,OAAA,EAAS,gBAAA,KAEtC,aAAa;AAAA,KAIP,0BAAA,GAA6B,iBAAA;EACvC,OAAA,EAAS,gBAAA;EACT,GAAA,GAAM,eAAA;AAAA;AAAA,cAGK,qBAAA,EAAuB,EAAE,CAAC,0BAAA"}

package/dist/react/AssistantProvider.js

@@ -1,13 +1,20 @@
-import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
-import { memo, } from "react";
-import { useAui, AuiProvider } from "@assistant-ui/store";
 import { RuntimeAdapter } from "./RuntimeAdapter.js";
-export const getRenderComponent = (runtime) => {
-    return runtime._core?.RenderComponent;
+import { memo } from "react";
+import { AuiProvider, useAui } from "@assistant-ui/store";
+import { jsx, jsxs } from "react/jsx-runtime";
+//#region src/react/AssistantProvider.tsx
+const getRenderComponent = (runtime) => {
+	return runtime._core?.RenderComponent;
 };
-export const AssistantProviderBase = memo(({ runtime, aui: parent = null, children }) => {
-    const aui = useAui({ threads: RuntimeAdapter(runtime) }, { parent });
-    const RenderComponent = getRenderComponent(runtime);
-    return (_jsxs(AuiProvider, { value: aui, children: [RenderComponent && _jsx(RenderComponent, {}), children] }));
+const AssistantProviderBase = memo(({ runtime, aui: parent = null, children }) => {
+	const aui = useAui({ threads: RuntimeAdapter(runtime) }, { parent });
+	const RenderComponent = getRenderComponent(runtime);
+	return /* @__PURE__ */ jsxs(AuiProvider, {
+		value: aui,
+		children: [RenderComponent && /* @__PURE__ */ jsx(RenderComponent, {}), children]
+	});
 });
+//#endregion
+export { AssistantProviderBase, getRenderComponent };
+
 //# sourceMappingURL=AssistantProvider.js.map

package/dist/react/AssistantProvider.js.map

@@ -1 +1 @@
-{"version":3,"file":"AssistantProvider.js","sourceRoot":"","sources":["../../src/react/AssistantProvider.tsx"],"names":[],"mappings":";AAAA,OAAO,EACL,IAAI,GAIL,MAAM,OAAO,CAAC;AACf,OAAO,EAAE,MAAM,EAAE,WAAW,EAAwB,MAAM,qBAAqB,CAAC;AAGhF,OAAO,EAAE,cAAc,EAAE,4BAAyB;AAElD,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,OAAyB,EAAE,EAAE;IAC9D,OAAQ,OAA4C,CAAC,KAAK,EAAE,eAE/C,CAAC;AAChB,CAAC,CAAC;AAOF,MAAM,CAAC,MAAM,qBAAqB,GAAmC,IAAI,CACvE,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,MAAM,GAAG,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE;IAC5C,MAAM,GAAG,GAAG,MAAM,CAAC,EAAE,OAAO,EAAE,cAAc,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;IACrE,MAAM,eAAe,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;IACpD,OAAO,CACL,MAAC,WAAW,IAAC,KAAK,EAAE,GAAG,aACpB,eAAe,IAAI,KAAC,eAAe,KAAG,EACtC,QAAQ,IACG,CACf,CAAC;AACJ,CAAC,CACF,CAAC"}
+{"version":3,"file":"AssistantProvider.js","names":[],"sources":["../../src/react/AssistantProvider.tsx"],"sourcesContent":["import {\n  memo,\n  type ComponentType,\n  type FC,\n  type PropsWithChildren,\n} from \"react\";\nimport { useAui, AuiProvider, type AssistantClient } from \"@assistant-ui/store\";\nimport type { AssistantRuntime } from \"../runtime/api/assistant-runtime\";\nimport type { AssistantRuntimeCore } from \"../runtime/interfaces/assistant-runtime-core\";\nimport { RuntimeAdapter } from \"./RuntimeAdapter\";\n\nexport const getRenderComponent = (runtime: AssistantRuntime) => {\n  return (runtime as { _core?: AssistantRuntimeCore })._core?.RenderComponent as\n    | ComponentType\n    | undefined;\n};\n\nexport type AssistantProviderBaseProps = PropsWithChildren<{\n  runtime: AssistantRuntime;\n  aui?: AssistantClient | null;\n}>;\n\nexport const AssistantProviderBase: FC<AssistantProviderBaseProps> = memo(\n  ({ runtime, aui: parent = null, children }) => {\n    const aui = useAui({ threads: RuntimeAdapter(runtime) }, { parent });\n    const RenderComponent = getRenderComponent(runtime);\n    return (\n      <AuiProvider value={aui}>\n        {RenderComponent && <RenderComponent />}\n        {children}\n      </AuiProvider>\n    );\n  },\n);\n"],"mappings":";;;;;AAWA,MAAa,sBAAsB,YAA8B;CAC/D,OAAQ,QAA6C,OAAO;AAG9D;AAOA,MAAa,wBAAwD,MAClE,EAAE,SAAS,KAAK,SAAS,MAAM,eAAe;CAC7C,MAAM,MAAM,OAAO,EAAE,SAAS,eAAe,OAAO,EAAE,GAAG,EAAE,OAAO,CAAC;CACnE,MAAM,kBAAkB,mBAAmB,OAAO;CAClD,OACE,qBAAC,aAAD;EAAa,OAAO;YAApB,CACG,mBAAmB,oBAAC,iBAAD,CAAkB,CAAA,GACrC,QACU;;AAEjB,CACF"}