@assistant-ui/react 0.10.44 → 0.10.45

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

@@ -0,0 +1,117 @@
+// src/model-context/registry/ModelContextRegistry.ts
+import {
+  mergeModelContexts
+} from "../../model-context/ModelContextTypes.js";
+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}
+
+${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 unsubscribe2 = this._providerUnsubscribes.get(id);
+        unsubscribe2?.();
+        this._providerUnsubscribes.delete(id);
+        this.notifySubscribers();
+      }
+    };
+  }
+};
+export {
+  ModelContextRegistry
+};
+//# sourceMappingURL=ModelContextRegistry.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/model-context/registry/ModelContextRegistry.ts"],"sourcesContent":["import { Tool } from \"assistant-stream\";\nimport {\n  ModelContext,\n  ModelContextProvider,\n  mergeModelContexts,\n} from \"../../model-context/ModelContextTypes\";\nimport { Unsubscribe } from \"../../types/Unsubscribe\";\nimport {\n  ModelContextRegistryToolHandle,\n  ModelContextRegistryInstructionHandle,\n  ModelContextRegistryProviderHandle,\n} from \"./ModelContextRegistryHandles\";\nimport type { AssistantToolProps } from \"../../model-context/useAssistantTool\";\nimport type { AssistantInstructionsConfig } from \"../../model-context/useAssistantInstructions\";\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    // Merge instructions\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    // Collect tools\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    // Merge provider contexts\n    const providerContexts = mergeModelContexts(\n      new Set(this._providers.values()),\n    );\n\n    // Combine everything\n    const context: ModelContext = {\n      system,\n      tools: Object.keys(tools).length > 0 ? tools : undefined,\n    };\n\n    // Merge with provider contexts\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    // Subscribe to provider changes\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":";AACA;AAAA,EAGE;AAAA,OACK;AAUA,IAAM,uBAAN,MAA2D;AAAA,EACxD,SAAS,oBAAI,IAA0C;AAAA,EACvD,gBAAgB,oBAAI,IAAoB;AAAA,EACxC,aAAa,oBAAI,IAAkC;AAAA,EACnD,eAAe,oBAAI,IAAgB;AAAA,EACnC,wBAAwB,oBAAI,IAAqC;AAAA,EAEzE,kBAAgC;AAE9B,UAAM,eAAe,MAAM,KAAK,KAAK,cAAc,OAAO,CAAC,EAAE;AAAA,MAC3D;AAAA,IACF;AAEA,UAAM,SACJ,aAAa,SAAS,IAAI,aAAa,KAAK,MAAM,IAAI;AAGxD,UAAM,QAAwC,CAAC;AAC/C,eAAW,aAAa,KAAK,OAAO,OAAO,GAAG;AAC5C,YAAM,EAAE,UAAU,QAAQ,GAAG,KAAK,IAAI;AACtC,YAAM,QAAQ,IAAI;AAAA,IACpB;AAGA,UAAM,mBAAmB;AAAA,MACvB,IAAI,IAAI,KAAK,WAAW,OAAO,CAAC;AAAA,IAClC;AAGA,UAAM,UAAwB;AAAA,MAC5B;AAAA,MACA,OAAO,OAAO,KAAK,KAAK,EAAE,SAAS,IAAI,QAAQ;AAAA,IACjD;AAGA,QAAI,iBAAiB,QAAQ;AAC3B,cAAQ,SAAS,QAAQ,SACrB,GAAG,QAAQ,MAAM;AAAA;AAAA,EAAO,iBAAiB,MAAM,KAC/C,iBAAiB;AAAA,IACvB;AAEA,QAAI,iBAAiB,OAAO;AAC1B,cAAQ,QAAQ,EAAE,GAAI,QAAQ,SAAS,CAAC,GAAI,GAAG,iBAAiB,MAAM;AAAA,IACxE;AAEA,QAAI,iBAAiB,cAAc;AACjC,cAAQ,eAAe,iBAAiB;AAAA,IAC1C;AAEA,QAAI,iBAAiB,QAAQ;AAC3B,cAAQ,SAAS,iBAAiB;AAAA,IACpC;AAEA,WAAO;AAAA,EACT;AAAA,EAEA,UAAU,UAAmC;AAC3C,SAAK,aAAa,IAAI,QAAQ;AAC9B,WAAO,MAAM,KAAK,aAAa,OAAO,QAAQ;AAAA,EAChD;AAAA,EAEQ,oBAA0B;AAChC,eAAW,YAAY,KAAK,cAAc;AACxC,eAAS;AAAA,IACX;AAAA,EACF;AAAA,EAEA,QACE,MACgD;AAChD,UAAM,KAAK,OAAO;AAElB,SAAK,OAAO,IAAI,IAAI,IAAI;AACxB,SAAK,kBAAkB;AAEvB,WAAO;AAAA,MACL,QAAQ,CAAC,YAAgD;AACvD,YAAI,KAAK,OAAO,IAAI,EAAE,GAAG;AACvB,eAAK,OAAO,IAAI,IAAI,OAAO;AAC3B,eAAK,kBAAkB;AAAA,QACzB;AAAA,MACF;AAAA,MACA,QAAQ,MAAM;AACZ,aAAK,OAAO,OAAO,EAAE;AACrB,aAAK,kBAAkB;AAAA,MACzB;AAAA,IACF;AAAA,EACF;AAAA,EAEA,eACE,QACuC;AACvC,UAAM,KAAK,OAAO;AAElB,UAAM,cACJ,OAAO,WAAW,WAAW,SAAS,OAAO;AAC/C,UAAM,WAAW,OAAO,WAAW,WAAW,OAAO,WAAW;AAEhE,QAAI,CAAC,UAAU;AACb,WAAK,cAAc,IAAI,IAAI,WAAW;AACtC,WAAK,kBAAkB;AAAA,IACzB;AAEA,WAAO;AAAA,MACL,QAAQ,CAAC,cAAoD;AAC3D,cAAM,iBACJ,OAAO,cAAc,WAAW,YAAY,UAAU;AACxD,cAAM,cACJ,OAAO,cAAc,WAAW,UAAU,WAAW;AAEvD,YAAI,aAAa;AACf,eAAK,cAAc,OAAO,EAAE;AAAA,QAC9B,OAAO;AACL,eAAK,cAAc,IAAI,IAAI,cAAc;AAAA,QAC3C;AACA,aAAK,kBAAkB;AAAA,MACzB;AAAA,MACA,QAAQ,MAAM;AACZ,aAAK,cAAc,OAAO,EAAE;AAC5B,aAAK,kBAAkB;AAAA,MACzB;AAAA,IACF;AAAA,EACF;AAAA,EAEA,YACE,UACoC;AACpC,UAAM,KAAK,OAAO;AAElB,SAAK,WAAW,IAAI,IAAI,QAAQ;AAGhC,UAAM,cAAc,SAAS,YAAY,MAAM;AAC7C,WAAK,kBAAkB;AAAA,IACzB,CAAC;AACD,SAAK,sBAAsB,IAAI,IAAI,WAAW;AAE9C,SAAK,kBAAkB;AAEvB,WAAO;AAAA,MACL,QAAQ,MAAM;AACZ,aAAK,WAAW,OAAO,EAAE;AACzB,cAAMA,eAAc,KAAK,sBAAsB,IAAI,EAAE;AACrD,QAAAA,eAAc;AACd,aAAK,sBAAsB,OAAO,EAAE;AACpC,aAAK,kBAAkB;AAAA,MACzB;AAAA,IACF;AAAA,EACF;AACF;","names":["unsubscribe"]}

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

@@ -0,0 +1,14 @@
+import type { AssistantToolProps } from "../../model-context/useAssistantTool";
+import type { AssistantInstructionsConfig } from "../../model-context/useAssistantInstructions";
+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=ModelContextRegistryHandles.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ModelContextRegistryHandles.d.ts","sourceRoot":"","sources":["../../../src/model-context/registry/ModelContextRegistryHandles.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,sCAAsC,CAAC;AAC/E,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,8CAA8C,CAAC;AAEhG,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/ModelContextRegistryHandles.js

@@ -0,0 +1 @@
+//# sourceMappingURL=ModelContextRegistryHandles.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

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

@@ -0,0 +1,3 @@
+export * from "./ModelContextRegistry";
+export * from "./ModelContextRegistryHandles";
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/model-context/registry/index.ts"],"names":[],"mappings":"AAAA,cAAc,wBAAwB,CAAC;AACvC,cAAc,+BAA+B,CAAC"}

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

@@ -0,0 +1,4 @@
+// src/model-context/registry/index.ts
+export * from "./ModelContextRegistry.js";
+export * from "./ModelContextRegistryHandles.js";
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/model-context/registry/index.ts"],"sourcesContent":["export * from \"./ModelContextRegistry\";\nexport * from \"./ModelContextRegistryHandles\";\n"],"mappings":";AAAA,cAAc;AACd,cAAc;","names":[]}

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

@@ -1,7 +1,6 @@
-type AssistantInstructionsConfig = {
+export type AssistantInstructionsConfig = {
     disabled?: boolean | undefined;
     instruction: string;
 };
 export declare const useAssistantInstructions: (config: string | AssistantInstructionsConfig) => void;
-export {};
 //# sourceMappingURL=useAssistantInstructions.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"useAssistantInstructions.d.ts","sourceRoot":"","sources":["../../src/model-context/useAssistantInstructions.tsx"],"names":[],"mappings":"AAKA,KAAK,2BAA2B,GAAG;IACjC,QAAQ,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IAC/B,WAAW,EAAE,MAAM,CAAC;CACrB,CAAC;AASF,eAAO,MAAM,wBAAwB,GACnC,QAAQ,MAAM,GAAG,2BAA2B,SAe7C,CAAC"}
+{"version":3,"file":"useAssistantInstructions.d.ts","sourceRoot":"","sources":["../../src/model-context/useAssistantInstructions.tsx"],"names":[],"mappings":"AAKA,MAAM,MAAM,2BAA2B,GAAG;IACxC,QAAQ,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IAC/B,WAAW,EAAE,MAAM,CAAC;CACrB,CAAC;AASF,eAAO,MAAM,wBAAwB,GACnC,QAAQ,MAAM,GAAG,2BAA2B,SAe7C,CAAC"}

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

@@ -1 +1 @@
-{"version":3,"sources":["../../src/model-context/useAssistantInstructions.tsx"],"sourcesContent":["\"use client\";\n\nimport { useEffect } from \"react\";\nimport { useAssistantRuntime } from \"../context\";\n\ntype AssistantInstructionsConfig = {\n  disabled?: boolean | undefined;\n  instruction: string;\n};\n\nconst getInstructions = (\n  instruction: string | AssistantInstructionsConfig,\n): AssistantInstructionsConfig => {\n  if (typeof instruction === \"string\") return { instruction };\n  return instruction;\n};\n\nexport const useAssistantInstructions = (\n  config: string | AssistantInstructionsConfig,\n) => {\n  const { instruction, disabled = false } = getInstructions(config);\n  const assistantRuntime = useAssistantRuntime();\n\n  useEffect(() => {\n    if (disabled) return;\n\n    const config = {\n      system: instruction,\n    };\n    return assistantRuntime.registerModelContextProvider({\n      getModelContext: () => config,\n    });\n  }, [assistantRuntime, instruction, disabled]);\n};\n"],"mappings":";;;AAEA,SAAS,iBAAiB;AAC1B,SAAS,2BAA2B;AAOpC,IAAM,kBAAkB,CACtB,gBACgC;AAChC,MAAI,OAAO,gBAAgB,SAAU,QAAO,EAAE,YAAY;AAC1D,SAAO;AACT;AAEO,IAAM,2BAA2B,CACtC,WACG;AACH,QAAM,EAAE,aAAa,WAAW,MAAM,IAAI,gBAAgB,MAAM;AAChE,QAAM,mBAAmB,oBAAoB;AAE7C,YAAU,MAAM;AACd,QAAI,SAAU;AAEd,UAAMA,UAAS;AAAA,MACb,QAAQ;AAAA,IACV;AACA,WAAO,iBAAiB,6BAA6B;AAAA,MACnD,iBAAiB,MAAMA;AAAA,IACzB,CAAC;AAAA,EACH,GAAG,CAAC,kBAAkB,aAAa,QAAQ,CAAC;AAC9C;","names":["config"]}
+{"version":3,"sources":["../../src/model-context/useAssistantInstructions.tsx"],"sourcesContent":["\"use client\";\n\nimport { useEffect } from \"react\";\nimport { useAssistantRuntime } from \"../context\";\n\nexport type AssistantInstructionsConfig = {\n  disabled?: boolean | undefined;\n  instruction: string;\n};\n\nconst getInstructions = (\n  instruction: string | AssistantInstructionsConfig,\n): AssistantInstructionsConfig => {\n  if (typeof instruction === \"string\") return { instruction };\n  return instruction;\n};\n\nexport const useAssistantInstructions = (\n  config: string | AssistantInstructionsConfig,\n) => {\n  const { instruction, disabled = false } = getInstructions(config);\n  const assistantRuntime = useAssistantRuntime();\n\n  useEffect(() => {\n    if (disabled) return;\n\n    const config = {\n      system: instruction,\n    };\n    return assistantRuntime.registerModelContextProvider({\n      getModelContext: () => config,\n    });\n  }, [assistantRuntime, instruction, disabled]);\n};\n"],"mappings":";;;AAEA,SAAS,iBAAiB;AAC1B,SAAS,2BAA2B;AAOpC,IAAM,kBAAkB,CACtB,gBACgC;AAChC,MAAI,OAAO,gBAAgB,SAAU,QAAO,EAAE,YAAY;AAC1D,SAAO;AACT;AAEO,IAAM,2BAA2B,CACtC,WACG;AACH,QAAM,EAAE,aAAa,WAAW,MAAM,IAAI,gBAAgB,MAAM;AAChE,QAAM,mBAAmB,oBAAoB;AAE7C,YAAU,MAAM;AACd,QAAI,SAAU;AAEd,UAAMA,UAAS;AAAA,MACb,QAAQ;AAAA,IACV;AACA,WAAO,iBAAiB,6BAA6B;AAAA,MACnD,iBAAiB,MAAMA;AAAA,IACzB,CAAC;AAAA,EACH,GAAG,CAAC,kBAAkB,aAAa,QAAQ,CAAC;AAC9C;","names":["config"]}

package/package.json

@@ -28,7 +28,7 @@
     "conversational-ui",
     "conversational-ai"
   ],
-  "version": "0.10.44",
+  "version": "0.10.45",
   "license": "MIT",
   "type": "module",
   "exports": {
@@ -48,6 +48,7 @@
   ],
   "sideEffects": false,
   "dependencies": {
+    "assistant-cloud": "^0.1.1",
     "@radix-ui/primitive": "^1.1.3",
     "@radix-ui/react-compose-refs": "^1.1.2",
     "@radix-ui/react-context": "^1.1.2",
@@ -62,8 +63,7 @@
     "nanoid": "5.1.5",
     "react-textarea-autosize": "^8.5.9",
     "zod": "^4.0.17",
-    "zustand": "^5.0.7",
-    "assistant-cloud": "0.1.1"
+    "zustand": "^5.0.7"
   },
   "peerDependencies": {
     "@types/react": "*",

package/src/model-context/frame/AssistantFrame.test.ts

@@ -0,0 +1,353 @@
+/**
+ * @vitest-environment jsdom
+ */
+import { describe, it, expect, beforeEach, afterEach, vi } from "vitest";
+import { AssistantFrameProvider } from "./AssistantFrameProvider";
+import { AssistantFrameHost } from "./AssistantFrameHost";
+import { ModelContextRegistry } from "../registry/ModelContextRegistry";
+import z from "zod";
+
+describe("AssistantFrame Integration", () => {
+  let messageHandlers: Map<string, (event: MessageEvent) => void>;
+  let iframeWindow: Window;
+  let parentWindow: any;
+
+  beforeEach(() => {
+    messageHandlers = new Map();
+
+    // Create a mock parent window that the iframe can post back to
+    parentWindow = {
+      postMessage: vi.fn((data: any) => {
+        // When iframe posts to parent, deliver to parent handler
+        const parentHandler = messageHandlers.get("parent");
+        if (parentHandler) {
+          Promise.resolve().then(() => {
+            parentHandler({
+              data,
+              source: iframeWindow,
+              origin: "*",
+            } as MessageEvent);
+          });
+        }
+      }),
+    };
+
+    // Create mock iframe window with proper message routing
+    iframeWindow = {
+      postMessage: vi.fn((data: any) => {
+        // Route message to iframe handler (provider)
+        const iframeHandler = messageHandlers.get("iframe");
+        if (iframeHandler) {
+          Promise.resolve().then(() => {
+            iframeHandler({
+              data,
+              source: parentWindow, // parent window is the source for subscription
+              origin: "*",
+            } as MessageEvent);
+          });
+        }
+      }),
+    } as any;
+
+    // Mock window.parent for iframe to broadcast to
+    Object.defineProperty(window, "parent", {
+      value: parentWindow,
+      writable: true,
+      configurable: true,
+    });
+
+    // Mock window methods for message passing
+    vi.spyOn(window, "addEventListener").mockImplementation(
+      (event: string, handler: any) => {
+        if (event === "message") {
+          // Store both handlers - we'll determine which is which based on usage
+          if (!messageHandlers.has("iframe")) {
+            messageHandlers.set("iframe", handler); // First registration is provider
+          } else {
+            messageHandlers.set("parent", handler); // Second is host
+          }
+        }
+      },
+    );
+
+    vi.spyOn(window, "removeEventListener").mockImplementation(() => {});
+
+    vi.spyOn(window, "postMessage").mockImplementation(() => {
+      // This shouldn't be called in our test setup
+    });
+  });
+
+  afterEach(() => {
+    // Clean up
+    vi.restoreAllMocks();
+    AssistantFrameProvider.dispose();
+    messageHandlers.clear();
+  });
+
+  it("should establish connection between host and provider", async () => {
+    // Setup provider in iframe
+    const registry = new ModelContextRegistry();
+    const unsubscribe =
+      AssistantFrameProvider.addModelContextProvider(registry);
+
+    // Setup host in parent
+    const host = new AssistantFrameHost(iframeWindow);
+
+    // Wait for connection
+    await vi.waitFor(() => {
+      expect(iframeWindow.postMessage).toHaveBeenCalledWith(
+        expect.objectContaining({
+          channel: "assistant-ui-frame",
+          message: expect.objectContaining({
+            type: "model-context-request",
+          }),
+        }),
+        "*",
+      );
+    });
+
+    // Clean up
+    host.dispose();
+    unsubscribe();
+  });
+
+  it("should sync tools from provider to host", async () => {
+    // Setup provider with tools
+    const registry = new ModelContextRegistry();
+
+    const toolExecute = vi.fn().mockResolvedValue({ result: "search results" });
+    registry.addTool({
+      toolName: "search",
+      description: "Search the web",
+      parameters: z.object({ query: z.string() }),
+      execute: toolExecute,
+    });
+
+    const unsubscribe =
+      AssistantFrameProvider.addModelContextProvider(registry);
+
+    // Setup host
+    const host = new AssistantFrameHost(iframeWindow);
+
+    // Wait for connection and initial sync
+    await vi.waitFor(() => {
+      const context = host.getModelContext();
+      expect(context.tools).toBeDefined();
+      expect(context.tools?.search).toBeDefined();
+      expect(context.tools?.search.description).toBe("Search the web");
+    });
+
+    // Clean up
+    host.dispose();
+    unsubscribe();
+  });
+
+  it("should execute tools through the frame boundary", async () => {
+    // Setup provider with executable tool
+    const registry = new ModelContextRegistry();
+
+    const toolExecute = vi
+      .fn()
+      .mockResolvedValue({ results: ["result1", "result2"] });
+    registry.addTool({
+      toolName: "search",
+      description: "Search the web",
+      parameters: z.object({ query: z.string() }),
+      execute: toolExecute,
+    });
+
+    const unsubscribe =
+      AssistantFrameProvider.addModelContextProvider(registry);
+
+    // Setup host
+    const host = new AssistantFrameHost(iframeWindow);
+
+    // Wait for tools to be available
+    await vi.waitFor(() => {
+      const context = host.getModelContext();
+      expect(context.tools?.search).toBeDefined();
+    });
+
+    // Execute tool through host
+    const context = host.getModelContext();
+    const searchTool = context.tools?.search;
+
+    const resultPromise = searchTool!.execute!(
+      { query: "test query" },
+      {} as any,
+    );
+
+    // Wait for tool execution
+    await vi.waitFor(() => {
+      expect(toolExecute).toHaveBeenCalledWith(
+        { query: "test query" },
+        expect.objectContaining({
+          toolCallId: expect.any(String),
+          abortSignal: expect.any(AbortSignal),
+        }),
+      );
+    });
+
+    const result = await resultPromise;
+    expect(result).toEqual({ results: ["result1", "result2"] });
+
+    // Clean up
+    host.dispose();
+    unsubscribe();
+  });
+
+  it("should handle tool execution errors", async () => {
+    // Setup provider with failing tool
+    const registry = new ModelContextRegistry();
+
+    const toolExecute = vi
+      .fn()
+      .mockRejectedValue(new Error("Tool execution failed"));
+    registry.addTool({
+      toolName: "failingTool",
+      description: "A tool that fails",
+      parameters: z.object({ input: z.string() }),
+      execute: toolExecute,
+    });
+
+    const unsubscribe =
+      AssistantFrameProvider.addModelContextProvider(registry);
+
+    // Setup host
+    const host = new AssistantFrameHost(iframeWindow);
+
+    // Wait for tools to be available
+    await vi.waitFor(() => {
+      const context = host.getModelContext();
+      expect(context.tools?.failingTool).toBeDefined();
+    });
+
+    // Execute tool and expect error
+    const context = host.getModelContext();
+    const failingTool = context.tools?.failingTool;
+
+    await expect(
+      failingTool!.execute!({ input: "test" }, {} as any),
+    ).rejects.toThrow("Tool execution failed");
+
+    // Clean up
+    host.dispose();
+    unsubscribe();
+  });
+
+  it("should handle multiple providers", async () => {
+    // Setup multiple providers
+    const registry1 = new ModelContextRegistry();
+    const registry2 = new ModelContextRegistry();
+
+    registry1.addTool({
+      toolName: "tool1",
+      description: "First tool",
+      parameters: z.object({ input: z.string() }),
+      execute: async () => ({ from: "tool1" }),
+    });
+
+    registry2.addTool({
+      toolName: "tool2",
+      description: "Second tool",
+      parameters: z.object({ input: z.string() }),
+      execute: async () => ({ from: "tool2" }),
+    });
+
+    const unsub1 = AssistantFrameProvider.addModelContextProvider(registry1);
+    const unsub2 = AssistantFrameProvider.addModelContextProvider(registry2);
+
+    // Setup host
+    const host = new AssistantFrameHost(iframeWindow);
+
+    // Wait for both tools to be available
+    await vi.waitFor(() => {
+      const context = host.getModelContext();
+      expect(context.tools?.tool1).toBeDefined();
+      expect(context.tools?.tool2).toBeDefined();
+    });
+
+    // Clean up
+    host.dispose();
+    unsub1();
+    unsub2();
+  });
+
+  it("should merge system instructions from multiple providers", async () => {
+    // Setup providers with system instructions
+    const registry1 = new ModelContextRegistry();
+    const registry2 = new ModelContextRegistry();
+
+    registry1.addInstruction("You are a helpful assistant.");
+    registry2.addInstruction("Always be concise.");
+
+    const unsub1 = AssistantFrameProvider.addModelContextProvider(registry1);
+    const unsub2 = AssistantFrameProvider.addModelContextProvider(registry2);
+
+    // Setup host
+    const host = new AssistantFrameHost(iframeWindow);
+
+    // Wait for instructions to be synced
+    await vi.waitFor(() => {
+      const context = host.getModelContext();
+      expect(context.system).toBeDefined();
+      expect(context.system).toContain("You are a helpful assistant.");
+      expect(context.system).toContain("Always be concise.");
+    });
+
+    // Clean up
+    host.dispose();
+    unsub1();
+    unsub2();
+  });
+
+  it("should act as empty ModelContextProvider when iframe has no providers", async () => {
+    // Don't register any providers in the iframe
+    // This simulates an iframe that doesn't respond to model-context requests
+
+    // Setup host
+    const host = new AssistantFrameHost(iframeWindow);
+
+    // Host should immediately return empty context
+    const context = host.getModelContext();
+    expect(context).toEqual({});
+
+    // Wait a bit to ensure no errors occur
+    await new Promise((resolve) => setTimeout(resolve, 100));
+
+    // Context should still be empty
+    expect(host.getModelContext()).toEqual({});
+
+    // Clean up
+    host.dispose();
+  });
+
+  it("should clean up properly on dispose", async () => {
+    // Setup provider
+    const registry = new ModelContextRegistry();
+
+    const unsubscribe =
+      AssistantFrameProvider.addModelContextProvider(registry);
+
+    // Setup host
+    const host = new AssistantFrameHost(iframeWindow);
+
+    // Wait for connection
+    await vi.waitFor(() => {
+      expect(iframeWindow.postMessage).toHaveBeenCalled();
+    });
+
+    // Dispose host
+    host.dispose();
+
+    // Verify event listener was removed (no unsubscribe message in new design)
+    expect(window.removeEventListener).toHaveBeenCalledWith(
+      "message",
+      expect.any(Function),
+    );
+
+    // Clean up provider
+    unsubscribe();
+    AssistantFrameProvider.dispose();
+  });
+});