@dxos/plugin-assistant 0.7.5-staging.e6154f3 → 0.8.1-main.013e445

package/dist/lib/node/{chunk-XUTDR7HI.cjs → chunk-JAB6KLPP.cjs}

@@ -26,8 +26,8 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
   mod
 ));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-var chunk_XUTDR7HI_exports = {};
-__export(chunk_XUTDR7HI_exports, {
+var chunk_JAB6KLPP_exports = {};
+__export(chunk_JAB6KLPP_exports, {
   AiClient: () => AiClient,
   AppGraphBuilder: () => AppGraphBuilder,
   AssistantSettings: () => AssistantSettings,
@@ -39,20 +39,20 @@ __export(chunk_XUTDR7HI_exports, {
   ToolboxContainer: () => ToolboxContainer,
   useServices: () => useServices
 });
-module.exports = __toCommonJS(chunk_XUTDR7HI_exports);
-var import_chunk_ZGH6F5YA = require("./chunk-ZGH6F5YA.cjs");
-var import_chunk_37GI4NYH = require("./chunk-37GI4NYH.cjs");
-var import_chunk_GNPXCHFT = require("./chunk-GNPXCHFT.cjs");
+module.exports = __toCommonJS(chunk_JAB6KLPP_exports);
+var import_chunk_VRXFIS4X = require("./chunk-VRXFIS4X.cjs");
+var import_chunk_KLSNCP34 = require("./chunk-KLSNCP34.cjs");
+var import_chunk_U6J2GO7I = require("./chunk-U6J2GO7I.cjs");
 var import_react = __toESM(require("react"));
 var import_app_framework = require("@dxos/app-framework");
 var import_artifact = require("@dxos/artifact");
-var import_functions = require("@dxos/functions");
+var import_types = require("@dxos/functions/types");
 var import_log = require("@dxos/log");
 var import_echo = require("@dxos/react-client/echo");
 var import_react_ui_theme = require("@dxos/react-ui-theme");
 var import_artifact2 = require("@dxos/artifact");
 var import_echo_schema = require("@dxos/echo-schema");
-var import_functions2 = require("@dxos/functions");
+var import_types2 = require("@dxos/functions/types");
 var import_log2 = require("@dxos/log");
 var import_echo2 = require("@dxos/react-client/echo");
 var import_jsonpointer = __toESM(require("jsonpointer"));
@@ -78,12 +78,17 @@ var import_log4 = require("@dxos/log");
 var import_react_ui_components3 = require("@dxos/react-ui-components");
 var import_react_ui_syntax_highlighter2 = require("@dxos/react-ui-syntax-highlighter");
 var import_util4 = require("@dxos/util");
+var import_state = require("@codemirror/state");
 var import_react5 = __toESM(require("react"));
 var import_react_ui2 = require("@dxos/react-ui");
 var import_react_ui_editor = require("@dxos/react-ui-editor");
 var import_react_ui_theme4 = require("@dxos/react-ui-theme");
-var import_state = require("@codemirror/state");
+var import_state2 = require("@codemirror/state");
 var import_view = require("@codemirror/view");
+var import_autocomplete = require("@codemirror/autocomplete");
+var import_state3 = require("@codemirror/state");
+var import_view2 = require("@codemirror/view");
+var import_async = require("@dxos/async");
 var import_react6 = __toESM(require("react"));
 var import_plugin_transcription = require("@dxos/plugin-transcription");
 var import_react_ui3 = require("@dxos/react-ui");
@@ -94,55 +99,57 @@ var import_invariant3 = require("@dxos/invariant");
 var import_log5 = require("@dxos/log");
 var import_echo3 = require("@dxos/react-client/echo");
 var import_signals_core = require("@preact/signals-core");
-var import_artifact4 = require("@dxos/artifact");
 var import_assistant = require("@dxos/assistant");
-var import_echo_schema3 = require("@dxos/echo-schema");
 var import_invariant4 = require("@dxos/invariant");
 var import_log6 = require("@dxos/log");
 var import_react8 = require("react");
 var import_app_framework2 = require("@dxos/app-framework");
-var import_artifact5 = require("@dxos/artifact");
+var import_artifact4 = require("@dxos/artifact");
 var import_assistant2 = require("@dxos/assistant");
-var import_functions3 = require("@dxos/functions");
+var import_types3 = require("@dxos/functions/types");
 var import_log7 = require("@dxos/log");
 var import_react_client = require("@dxos/react-client");
 var import_echo4 = require("@dxos/react-client/echo");
 var import_util5 = require("@dxos/util");
 var import_app_framework3 = require("@dxos/app-framework");
 var import_react9 = require("react");
-var import_async = require("@dxos/async");
+var import_async2 = require("@dxos/async");
 var import_context = require("@dxos/context");
-var import_functions4 = require("@dxos/functions");
-var import_functions5 = require("@dxos/functions");
+var import_functions = require("@dxos/functions");
+var import_types4 = require("@dxos/functions/types");
 var import_invariant5 = require("@dxos/invariant");
 var import_log8 = require("@dxos/log");
 var import_react_client2 = require("@dxos/react-client");
 var import_echo5 = require("@dxos/react-client/echo");
-var import_async2 = require("@dxos/async");
+var import_async3 = require("@dxos/async");
 var import_echo_db = require("@dxos/echo-db");
-var import_functions6 = require("@dxos/functions");
+var import_types5 = require("@dxos/functions/types");
 var import_invariant6 = require("@dxos/invariant");
 var import_keys = require("@dxos/keys");
 var import_log9 = require("@dxos/log");
 var import_react10 = require("react");
 var import_keys2 = require("@dxos/keys");
 var import_echo6 = require("@dxos/react-client/echo");
-var import_react_edge_client = require("@dxos/react-edge-client");
 var import_react11 = require("react");
-var import_artifact6 = require("@dxos/artifact");
+var import_artifact5 = require("@dxos/artifact");
 var import_assistant3 = require("@dxos/assistant");
-var import_echo_schema4 = require("@dxos/echo-schema");
+var import_echo_schema3 = require("@dxos/echo-schema");
 var import_react12 = require("react");
+var import_react13 = require("react");
+var import_app_framework4 = require("@dxos/app-framework");
+var import_echo_schema4 = require("@dxos/echo-schema");
+var import_log10 = require("@dxos/log");
+var import_echo7 = require("@dxos/react-client/echo");
 var __dxlog_file = "/home/runner/work/dxos/dxos/packages/plugins/experimental/plugin-assistant/src/tools/function.ts";
 var covertFunctionToTool = (fn, edgeUrl, spaceId) => {
   if (!fn.description || !fn.inputSchema) {
     return void 0;
   }
-  const existingFunctionUrl = (0, import_functions2.getUserFunctionUrlInMetadata)((0, import_echo2.getMeta)(fn));
+  const existingFunctionUrl = (0, import_types2.getUserFunctionUrlInMetadata)((0, import_echo2.getMeta)(fn));
   if (!existingFunctionUrl) {
     return void 0;
   }
-  const url = (0, import_functions2.getInvocationUrl)(existingFunctionUrl, edgeUrl, {
+  const url = (0, import_types2.getInvocationUrl)(existingFunctionUrl, edgeUrl, {
     spaceId
   });
   return (0, import_artifact2.defineTool)("user-function", {
@@ -609,7 +616,7 @@ var Section = ({ title, items, striped }) => {
 };
 var ToolboxContainer = ({ classNames, space }) => {
   const artifactDefinitions = (0, import_app_framework.useCapabilities)(import_app_framework.Capabilities.ArtifactDefinition);
-  const services = (0, import_echo.useQuery)(space, import_echo.Filter.schema(import_chunk_37GI4NYH.ServiceType));
+  const services = (0, import_echo.useQuery)(space, import_echo.Filter.schema(import_chunk_KLSNCP34.ServiceType));
   const [serviceTools, setServiceTools] = (0, import_react.useState)([]);
   (0, import_react.useEffect)(() => {
     (0, import_log.log)("creating service tools...", {
@@ -630,7 +637,7 @@ var ToolboxContainer = ({ classNames, space }) => {
   }, [
     services
   ]);
-  const functions = (0, import_echo.useQuery)(space, import_echo.Filter.schema(import_functions.FunctionType));
+  const functions = (0, import_echo.useQuery)(space, import_echo.Filter.schema(import_types.FunctionType));
   return /* @__PURE__ */ import_react.default.createElement(Toolbox, {
     classNames,
     artifacts: artifactDefinitions,
@@ -755,7 +762,7 @@ var ToolContainer = ({ classNames, items }) => {
 };
 var __dxlog_file5 = "/home/runner/work/dxos/dxos/packages/plugins/experimental/plugin-assistant/src/components/Thread/ThreadMessage.tsx";
 var panelClassNames = "flex flex-col w-full px-2 bg-groupSurface rounded-md";
-var userClassNames = "bg-[--user-fill]";
+var userClassNames = "bg-[--user-fill] text-inverse";
 var ToggleContainer2 = (props) => {
   return /* @__PURE__ */ import_react3.default.createElement(import_react_ui_components2.ToggleContainer, {
     ...props,
@@ -780,7 +787,7 @@ var ThreadMessage = ({ classNames, space, message, tools, onPrompt }) => {
   const { role, content = [] } = message;
   if (isToolMessage(message)) {
     return /* @__PURE__ */ import_react3.default.createElement(MessageContainer, {
-      classNames
+      classNames: (0, import_react_ui_theme3.mx)(classNames, "animate-[fadeIn_0.5s]")
     }, /* @__PURE__ */ import_react3.default.createElement(ToolBlock, {
       space,
       classNames: panelClassNames,
@@ -795,7 +802,7 @@ var ThreadMessage = ({ classNames, space, message, tools, onPrompt }) => {
     const Component = components[block.type] ?? components.default;
     return /* @__PURE__ */ import_react3.default.createElement(MessageContainer, {
       key: idx,
-      classNames,
+      classNames: (0, import_react_ui_theme3.mx)(classNames, "animate-[fadeIn_0.5s]"),
       user: block.type === "text" && role === "user"
     }, /* @__PURE__ */ import_react3.default.createElement(Component, {
       space,
@@ -811,7 +818,7 @@ var components = {
   ["text"]: ({ block }) => {
     (0, import_invariant2.invariant)(block.type === "text", void 0, {
       F: __dxlog_file5,
-      L: 86,
+      L: 90,
       S: void 0,
       A: [
         "block.type === 'text'",
@@ -821,6 +828,7 @@ var components = {
     const title = block.disposition ? titles[block.disposition] : void 0;
     if (!title) {
       return /* @__PURE__ */ import_react3.default.createElement(import_react_ui_components2.MarkdownViewer, {
+        classNames: "[&>p]:animate-[fadeIn_0.5s]",
         content: block.text
       });
     }
@@ -847,7 +855,7 @@ var components = {
   ["json"]: ({ space, block, onPrompt }) => {
     (0, import_invariant2.invariant)(block.type === "json", void 0, {
       F: __dxlog_file5,
-      L: 124,
+      L: 128,
       S: void 0,
       A: [
         "block.type === 'json'",
@@ -874,7 +882,10 @@ var components = {
       }
       case "select": {
         const { options = [] } = (0, import_util3.safeParseJson)(block.json ?? "{}") ?? {};
-        return /* @__PURE__ */ import_react3.default.createElement(import_react_ui.ButtonGroup, null, options.map((option) => /* @__PURE__ */ import_react3.default.createElement(import_react_ui.Button, {
+        return /* @__PURE__ */ import_react3.default.createElement("div", {
+          className: "flex flex-wrap gap-1"
+        }, options.map((option, idx) => /* @__PURE__ */ import_react3.default.createElement(import_react_ui.Button, {
+          classNames: "animate-[fadeIn_0.5s] rounded-2xl text-sm",
           key: option,
           onClick: () => onPrompt?.(option)
         }, option)));
@@ -1020,7 +1031,7 @@ var createAutocompleteExtension = ({ onSubmit, onSuggest }) => {
       }
     }),
     // Accept the current suggestion.
-    import_state.Prec.highest(import_view.keymap.of([
+    import_state2.Prec.highest(import_view.keymap.of([
       {
         key: "Tab",
         preventDefault: true,
@@ -1109,9 +1120,120 @@ var InlineSuggestionWidget = class extends import_view.WidgetType {
     return other.suffix === this.suffix;
   }
 };
-var Prompt = /* @__PURE__ */ (0, import_react5.forwardRef)(({ classNames, autoFocus, lineWrapping = false, placeholder, onSubmit, onSuggest, onOpenChange }, forwardRef2) => {
+var promptReferences = ({ provider, debug = false, triggerCharacter = "@" }) => {
+  if (triggerCharacter.length !== 1) {
+    throw new Error("triggerCharacter must be a single character");
+  }
+  const decorationField = import_view2.ViewPlugin.fromClass(class ReferenceView {
+    constructor(view) {
+      this._mutex = new import_async.Mutex();
+      this.decorations = import_view2.Decoration.set([]);
+      queueMicrotask(async () => {
+        const guard = await this._mutex.acquire();
+        try {
+          this.decorations = await this._computeDecorations(view);
+        } finally {
+          guard.release();
+        }
+      });
+    }
+    update(update) {
+      if (update.docChanged) {
+        queueMicrotask(async () => {
+          const guard = await this._mutex.acquire();
+          try {
+            this.decorations = await this._computeDecorations(update.view);
+          } finally {
+            guard.release();
+          }
+        });
+      }
+    }
+    async _computeDecorations(view) {
+      const text = view.state.doc.toString();
+      const references = text.matchAll(new RegExp(`${triggerCharacter}[a-zA-Z0-9@:]+\\s`, "g"));
+      const decorations = [];
+      for (const match of references) {
+        const reference = match[0];
+        const uri = reference.slice(1, -1);
+        const data = await provider.resolveReference({
+          uri
+        });
+        if (data) {
+          decorations.push(import_view2.Decoration.replace({
+            widget: new ReferenceWidget(data)
+          }).range(match.index, match.index + reference.length));
+        }
+      }
+      return import_view2.Decoration.set(decorations);
+    }
+  }, {
+    decorations: (v) => v.decorations,
+    provide: (plugin) => [
+      import_view2.EditorView.atomicRanges.of((view) => view.plugin(decorationField)?.decorations ?? import_state3.RangeSet.empty)
+    ]
+  });
+  return [
+    decorationField,
+    import_view2.EditorView.theme({
+      ".cm-reference-pill": {
+        borderRadius: "0.25rem",
+        borderWidth: "1px",
+        marginRight: "0.25rem",
+        marginLeft: "0.25rem"
+      }
+    }),
+    (0, import_autocomplete.autocompletion)({
+      activateOnTyping: true,
+      override: [
+        async (context) => {
+          const match = context.matchBefore(new RegExp(`${triggerCharacter}[a-zA-Z0-9]+`));
+          if (!match || match?.to === match?.from) {
+            return null;
+          }
+          const query = match.text.slice(1);
+          const references = await provider.getReferences({
+            query
+          });
+          return {
+            from: match.from,
+            filter: false,
+            options: references.map((reference) => ({
+              label: reference.label,
+              apply: `${triggerCharacter}${reference.uri} `
+            }))
+          };
+        }
+      ],
+      closeOnBlur: !debug,
+      tooltipClass: () => "shadow rounded",
+      aboveCursor: true
+    }),
+    import_view2.keymap.of(import_autocomplete.completionKeymap)
+  ];
+};
+var ReferenceWidget = class extends import_view2.WidgetType {
+  constructor(data) {
+    super();
+    this.data = data;
+  }
+  toDOM() {
+    const span = document.createElement("span");
+    span.textContent = `@ ${this.data.label}`;
+    span.className = "cm-reference-pill";
+    return span;
+  }
+  eq(other) {
+    return other.data.uri === this.data.uri;
+  }
+  ignoreEvent() {
+    return true;
+  }
+};
+var Prompt = /* @__PURE__ */ (0, import_react5.forwardRef)(({ classNames, autoFocus, lineWrapping = false, placeholder, onSubmit, onSuggest, onOpenChange, references }, forwardRef2) => {
   const { themeMode } = (0, import_react_ui2.useThemeContext)();
   const { parentRef, view } = (0, import_react_ui_editor.useTextEditor)({
+    debug: true,
     autoFocus,
     extensions: [
       (0, import_react_ui_editor.createBasicExtensions)({
@@ -1122,13 +1244,16 @@ var Prompt = /* @__PURE__ */ (0, import_react5.forwardRef)(({ classNames, autoFo
       (0, import_react_ui_editor.createThemeExtensions)({
         themeMode
       }),
+      references ? promptReferences({
+        provider: references
+      }) : [],
       createAutocompleteExtension({
         onSubmit,
         onSuggest
       }),
-      import_react_ui_editor.keymap.of([
+      import_state.Prec.highest(import_react_ui_editor.keymap.of([
         {
-          key: "Alt-ArrowUp",
+          key: "cmd-ArrowUp",
           preventDefault: true,
           run: (view2) => {
             onOpenChange?.(true);
@@ -1136,14 +1261,14 @@ var Prompt = /* @__PURE__ */ (0, import_react5.forwardRef)(({ classNames, autoFo
           }
         },
         {
-          key: "Alt-ArrowDown",
+          key: "cmd-ArrowDown",
           preventDefault: true,
           run: (view2) => {
             onOpenChange?.(false);
             return true;
           }
         }
-      ])
+      ]))
     ]
   }, [
     themeMode,
@@ -1175,11 +1300,11 @@ var Prompt = /* @__PURE__ */ (0, import_react5.forwardRef)(({ classNames, autoFo
   ]);
   return /* @__PURE__ */ import_react5.default.createElement("div", {
     ref: parentRef,
-    className: (0, import_react_ui_theme4.mx)("w-full overflow-hidden", classNames)
+    className: (0, import_react_ui_theme4.mx)("w-full", classNames)
   });
 });
-var PromptBar = ({ classNames, placeholder, processing, error, microphone, onCancel, ...props }) => {
-  const { t } = (0, import_react_ui3.useTranslation)(import_chunk_GNPXCHFT.ASSISTANT_PLUGIN);
+var PromptBar = ({ classNames, placeholder, processing, error, microphone, onCancel, references, ...props }) => {
+  const { t } = (0, import_react_ui3.useTranslation)(import_chunk_U6J2GO7I.ASSISTANT_PLUGIN);
   const promptRef = (0, import_react6.useRef)(null);
   const [active, setActive] = (0, import_react6.useState)(false);
   const { recording } = (0, import_plugin_transcription.useVoiceInput)({
@@ -1190,14 +1315,18 @@ var PromptBar = ({ classNames, placeholder, processing, error, microphone, onCan
     }
   });
   return /* @__PURE__ */ import_react6.default.createElement("div", {
-    className: (0, import_react_ui_theme5.mx)("flex shrink-0 w-full grid grid-cols-[var(--rail-action)_1fr_var(--rail-action)] overflow-hidden", classNames)
+    className: (0, import_react_ui_theme5.mx)("shrink-0 w-full grid grid-cols-[var(--rail-action)_1fr_var(--rail-action)] overflow-hidden", classNames)
   }, /* @__PURE__ */ import_react6.default.createElement("div", {
     className: "flex w-[--rail-action] h-[--rail-action] items-center justify-center"
-  }, error && /* @__PURE__ */ import_react6.default.createElement(import_react_ui3.Icon, {
+  }, error && /* @__PURE__ */ import_react6.default.createElement(import_react_ui3.Tooltip.Root, {
+    delayDuration: 0
+  }, /* @__PURE__ */ import_react6.default.createElement(import_react_ui3.Tooltip.Trigger, null, /* @__PURE__ */ import_react6.default.createElement(import_react_ui3.Icon, {
     icon: "ph--warning-circle--regular",
     classNames: import_react_ui_theme5.errorText,
     size: 5
-  }) || /* @__PURE__ */ import_react6.default.createElement(import_react_ui_sfx.Spinner, {
+  })), /* @__PURE__ */ import_react6.default.createElement(import_react_ui3.Tooltip.Portal, null, /* @__PURE__ */ import_react6.default.createElement(import_react_ui3.Tooltip.Content, null, /* @__PURE__ */ import_react6.default.createElement("div", {
+    className: "text-sm text-error-500"
+  }, error.message), /* @__PURE__ */ import_react6.default.createElement(import_react_ui3.Tooltip.Arrow, null)))) || /* @__PURE__ */ import_react6.default.createElement(import_react_ui_sfx.Spinner, {
     active: processing
   })), /* @__PURE__ */ import_react6.default.createElement(Prompt, {
     ref: promptRef,
@@ -1205,6 +1334,7 @@ var PromptBar = ({ classNames, placeholder, processing, error, microphone, onCan
     classNames: "pbs-2",
     lineWrapping: true,
     placeholder: placeholder ?? t("prompt placeholder"),
+    references,
     ...props
   }), (onCancel || microphone) && /* @__PURE__ */ import_react6.default.createElement("div", {
     className: "flex w-[--rail-action] h-[--rail-action] items-center justify-center"
@@ -1230,7 +1360,7 @@ var PromptBar = ({ classNames, placeholder, processing, error, microphone, onCan
     onTouchEnd: () => setActive(false)
   })));
 };
-var Thread = ({ classNames, space, messages, collapse = true, transcription, processing, error, onSubmit, onCancel, onOpenChange, ...props }) => {
+var Thread = ({ classNames, space, messages, collapse = true, transcription, processing, error, onSubmit, onCancel, onOpenChange, contextProvider, ...props }) => {
   const scroller = (0, import_react2.useRef)(null);
   const identity = (0, import_halo.useIdentity)();
   const fallbackValue = (0, import_util2.keyToFallback)(identity.identityKey);
@@ -1256,6 +1386,21 @@ var Thread = ({ classNames, space, messages, collapse = true, transcription, pro
     messages,
     collapse
   ]);
+  const references = (0, import_react2.useMemo)(() => {
+    if (!contextProvider) {
+      return void 0;
+    }
+    return {
+      getReferences: async ({ query }) => contextProvider.query({
+        query
+      }),
+      resolveReference: async ({ uri }) => contextProvider.resolveMetadata({
+        uri
+      })
+    };
+  }, [
+    contextProvider
+  ]);
   return /* @__PURE__ */ import_react2.default.createElement("div", {
     role: "none",
     className: (0, import_react_ui_theme2.mx)("flex flex-col grow overflow-hidden", classNames)
@@ -1280,24 +1425,24 @@ var Thread = ({ classNames, space, messages, collapse = true, transcription, pro
     error,
     onSubmit: handleSubmit,
     onCancel,
-    onOpenChange
+    onOpenChange,
+    references
   }));
 };
 var __dxlog_file6 = "/home/runner/work/dxos/dxos/packages/plugins/experimental/plugin-assistant/src/hooks/processor.ts";
 var defaultOptions = {
-  model: import_assistant.DEFAULT_LLM_MODEL,
+  model: import_assistant.DEFAULT_EDGE_MODEL,
   systemPrompt: "you are a helpful assistant"
 };
 var ChatProcessor = class {
-  constructor(_service, _tools, _extensions, _options = defaultOptions) {
-    this._service = _service;
+  constructor(_ai, _tools, _artifacts, _extensions, _options = defaultOptions) {
+    this._ai = _ai;
     this._tools = _tools;
+    this._artifacts = _artifacts;
     this._extensions = _extensions;
     this._options = _options;
-    this._parser = new import_assistant.MixedStreamParser();
     this._pending = (0, import_signals_core.signal)([]);
     this._block = (0, import_signals_core.signal)(void 0);
-    this._history = [];
     this.streaming = (0, import_signals_core.computed)(() => this._block.value !== void 0);
     this.error = (0, import_signals_core.signal)(void 0);
     this.messages = (0, import_signals_core.computed)(() => {
@@ -1308,7 +1453,7 @@ var ChatProcessor = class {
         const current = messages.pop();
         (0, import_invariant4.invariant)(current, void 0, {
           F: __dxlog_file6,
-          L: 85,
+          L: 70,
           S: this,
           A: [
             "current",
@@ -1327,20 +1472,6 @@ var ChatProcessor = class {
       }
       return messages;
     });
-    this._parser.message.on((message) => {
-      (0, import_signals_core.batch)(() => {
-        this._pending.value = [
-          ...this._pending.value,
-          message
-        ];
-        this._block.value = void 0;
-      });
-    });
-    this._parser.update.on((block) => {
-      (0, import_signals_core.batch)(() => {
-        this._block.value = block;
-      });
-    });
   }
   get tools() {
     return this._tools;
@@ -1355,23 +1486,70 @@ var ChatProcessor = class {
   * Make GPT request.
   */
   async request(message, options = {}) {
-    (0, import_signals_core.batch)(() => {
-      this._history = options.history ?? [];
+    this._session = new import_assistant.AISession({
+      operationModel: "immediate"
+    });
+    this._session.message.on((message2) => {
+      (0, import_signals_core.batch)(() => {
+        this._pending.value = [
+          ...this._pending.value,
+          message2
+        ];
+        this._block.value = void 0;
+      });
+    });
+    this._session.update.on((block) => {
+      (0, import_signals_core.batch)(() => {
+        this._block.value = block;
+      });
+    });
+    this._session.userMessage.on((message2) => {
       this._pending.value = [
-        (0, import_echo_schema3.createStatic)(import_artifact4.Message, {
-          role: "user",
-          content: [
-            {
-              type: "text",
-              text: message
-            }
-          ]
-        })
+        ...this._pending.value,
+        message2
       ];
-      this._block.value = void 0;
     });
-    await this._request();
-    options.onComplete?.(this._pending.value);
+    try {
+      const messages = await this._session.run({
+        client: this._ai,
+        history: options.history ?? [],
+        artifacts: this._artifacts ?? [],
+        tools: this._tools ?? [],
+        prompt: message,
+        extensions: this._extensions,
+        generationOptions: {
+          model: this._options.model,
+          systemPrompt: this._options.systemPrompt
+        }
+      });
+      import_log6.log.info("completed", {
+        messages
+      }, {
+        F: __dxlog_file6,
+        L: 137,
+        S: this,
+        C: (f, a) => f(...a)
+      });
+      options.onComplete?.(this._pending.value);
+    } catch (err) {
+      import_log6.log.catch(err, void 0, {
+        F: __dxlog_file6,
+        L: 141,
+        S: this,
+        C: (f, a) => f(...a)
+      });
+      if (err instanceof Error && err.message.includes("Overloaded")) {
+        this.error.value = new AIServiceOverloadedError("AI service overloaded", {
+          cause: err
+        });
+      } else {
+        this.error.value = new Error("AI service error", {
+          cause: err
+        });
+      }
+    } finally {
+      this._session = void 0;
+    }
     return this._reset();
   }
   /**
@@ -1381,135 +1559,45 @@ var ChatProcessor = class {
   async cancel() {
     import_log6.log.info("cancelling...", void 0, {
       F: __dxlog_file6,
-      L: 152,
+      L: 159,
       S: this,
       C: (f, a) => f(...a)
     });
-    this._stream?.abort();
+    this._session?.abort();
     return this._reset();
   }
   async _reset() {
     const messages = this._pending.value;
     (0, import_signals_core.batch)(() => {
-      this._history = [];
       this._pending.value = [];
       this._block.value = void 0;
     });
     return messages;
   }
-  /**
-  * Generate a response from the AI service.
-  * Iterates over tool requests.
-  */
-  async _request() {
-    try {
-      let more = false;
-      do {
-        (0, import_log6.log)("request", {
-          pending: this._pending.value.length,
-          history: this._history.length,
-          tools: this._tools?.map((tool) => tool.name)
-        }, {
-          F: __dxlog_file6,
-          L: 176,
-          S: this,
-          C: (f, a) => f(...a)
-        });
-        this._stream = await this._service.exec({
-          ...this._options,
-          // TODO(burdon): Rename messages or separate history/message.
-          history: [
-            ...this._history,
-            ...this._pending.value
-          ],
-          tools: this._tools
-        });
-        await this._parser.parse(this._stream);
-        await this._stream.complete();
-        (0, import_log6.log)("response", {
-          pending: this._pending.value
-        }, {
-          F: __dxlog_file6,
-          L: 195,
-          S: this,
-          C: (f, a) => f(...a)
-        });
-        more = false;
-        const message = this._pending.value.at(-1);
-        (0, import_invariant4.invariant)(message, void 0, {
-          F: __dxlog_file6,
-          L: 200,
-          S: this,
-          A: [
-            "message",
-            ""
-          ]
-        });
-        if ((0, import_assistant.isToolUse)(message)) {
-          (0, import_log6.log)("tool request...", void 0, {
-            F: __dxlog_file6,
-            L: 202,
-            S: this,
-            C: (f, a) => f(...a)
-          });
-          const response = await (0, import_assistant.runTools)({
-            message: this._pending.value.at(-1),
-            tools: this._tools ?? [],
-            extensions: this._extensions
-          });
-          (0, import_log6.log)("tool response", {
-            response
-          }, {
-            F: __dxlog_file6,
-            L: 209,
-            S: this,
-            C: (f, a) => f(...a)
-          });
-          switch (response.type) {
-            case "continue": {
-              this._pending.value = [
-                ...this._pending.value,
-                response.message
-              ];
-              more = true;
-              break;
-            }
-          }
-        }
-      } while (more);
-      this.error.value = void 0;
-    } catch (err) {
-      import_log6.log.catch(err, void 0, {
-        F: __dxlog_file6,
-        L: 222,
-        S: this,
-        C: (f, a) => f(...a)
-      });
-      this.error.value = new Error("AI service error", {
-        cause: err
-      });
-    } finally {
-      this._stream = void 0;
-    }
+};
+var AIServiceOverloadedError = class extends Error {
+  constructor() {
+    super(...arguments);
+    this.code = "AI_SERVICE_OVERLOADED";
   }
 };
-var AiClient = (0, import_app_framework3.lazy)(() => import("./ai-client-YANJEPO3.cjs"));
-var AppGraphBuilder = (0, import_app_framework3.lazy)(() => import("./app-graph-builder-D7SHQTZS.cjs"));
-var IntentResolver = (0, import_app_framework3.lazy)(() => import("./intent-resolver-YMMAFVOB.cjs"));
-var ReactSurface = (0, import_app_framework3.lazy)(() => import("./react-surface-BSUZQ3HZ.cjs"));
-var AssistantSettings = (0, import_app_framework3.lazy)(() => import("./settings-4YEO7KXF.cjs"));
+var AiClient = (0, import_app_framework3.lazy)(() => import("./ai-client-5ESLYXAV.cjs"));
+var AppGraphBuilder = (0, import_app_framework3.lazy)(() => import("./app-graph-builder-3P6WSON2.cjs"));
+var IntentResolver = (0, import_app_framework3.lazy)(() => import("./intent-resolver-YIFAMM3B.cjs"));
+var ReactSurface = (0, import_app_framework3.lazy)(() => import("./react-surface-IDGIN55C.cjs"));
+var AssistantSettings = (0, import_app_framework3.lazy)(() => import("./settings-2FEYGLYU.cjs"));
 var __dxlog_file7 = "/home/runner/work/dxos/dxos/packages/plugins/experimental/plugin-assistant/src/hooks/useChatProcessor.tsx";
-var useChatProcessor = (space, settings) => {
-  const aiClient = (0, import_app_framework2.useCapability)(import_chunk_ZGH6F5YA.AssistantCapabilities.AiClient);
+var useChatProcessor = ({ chat, space, settings, part = "deck" }) => {
+  const aiClient = (0, import_app_framework2.useCapability)(import_chunk_VRXFIS4X.AssistantCapabilities.AiClient);
   const globalTools = (0, import_app_framework2.useCapabilities)(import_app_framework2.Capabilities.Tools);
   const artifactDefinitions = (0, import_app_framework2.useCapabilities)(import_app_framework2.Capabilities.ArtifactDefinition);
   const { dispatchPromise: dispatch } = (0, import_app_framework2.useIntentDispatcher)();
-  const services = (0, import_echo4.useQuery)(space, import_echo4.Filter.schema(import_chunk_37GI4NYH.ServiceType));
+  const services = (0, import_echo4.useQuery)(space, import_echo4.Filter.schema(import_chunk_KLSNCP34.ServiceType));
   const [serviceTools, setServiceTools] = (0, import_react8.useState)([]);
   (0, import_react8.useEffect)(() => {
     (0, import_log7.log)("creating service tools...", void 0, {
       F: __dxlog_file7,
-      L: 34,
+      L: 41,
       S: void 0,
       C: (f, a) => f(...a)
     });
@@ -1521,23 +1609,27 @@ var useChatProcessor = (space, settings) => {
     services
   ]);
   const config = (0, import_react_client.useConfig)();
-  const functions = (0, import_echo4.useQuery)(space, import_echo4.Filter.schema(import_functions3.FunctionType));
+  const functions = (0, import_echo4.useQuery)(space, import_echo4.Filter.schema(import_types3.FunctionType));
+  const chatId = (0, import_react8.useMemo)(() => chat ? (0, import_echo4.fullyQualifiedId)(chat) : void 0, [
+    chat
+  ]);
   const [tools, extensions] = (0, import_react8.useMemo)(() => {
     (0, import_log7.log)("creating tools...", void 0, {
       F: __dxlog_file7,
-      L: 45,
+      L: 53,
       S: void 0,
       C: (f, a) => f(...a)
     });
     const tools2 = [
       ...globalTools.flat(),
-      ...artifactDefinitions.flatMap((definition) => definition.tools),
       ...serviceTools,
       ...functions.map((fn) => covertFunctionToTool(fn, config.values.runtime?.services?.edge?.url ?? "", space?.id)).filter(import_util5.isNonNullable)
     ];
     const extensions2 = {
       space,
-      dispatch
+      dispatch,
+      pivotId: chatId,
+      part
     };
     return [
       tools2,
@@ -1546,41 +1638,40 @@ var useChatProcessor = (space, settings) => {
   }, [
     dispatch,
     globalTools,
-    artifactDefinitions,
     space,
+    chatId,
     serviceTools,
     functions
   ]);
-  const systemPrompt = (0, import_react8.useMemo)(() => (0, import_artifact5.createSystemPrompt)({
+  const systemPrompt = (0, import_react8.useMemo)(() => (0, import_artifact4.createSystemPrompt)({
     artifacts: artifactDefinitions.map((definition) => definition.instructions)
   }), [
     artifactDefinitions
   ]);
+  const model = settings?.llmProvider === "ollama" ? settings?.ollamaModel ?? import_assistant2.DEFAULT_OLLAMA_MODEL : settings?.edgeModel ?? import_assistant2.DEFAULT_EDGE_MODEL;
   const processor = (0, import_react8.useMemo)(() => {
     (0, import_log7.log)("creating processor...", {
       settings
     }, {
       F: __dxlog_file7,
-      L: 70,
+      L: 83,
       S: void 0,
       C: (f, a) => f(...a)
     });
-    return new ChatProcessor(aiClient, tools, extensions, {
-      // TODO(burdon): Remove defualt (let backend decide if not specified).
-      model: settings?.llmModel ?? import_assistant2.DEFAULT_LLM_MODEL,
-      // TOOD(burdon): Query.
+    return new ChatProcessor(aiClient.value, tools, artifactDefinitions, extensions, {
+      model,
       systemPrompt
     });
   }, [
-    aiClient,
+    aiClient.value,
     tools,
     extensions,
-    systemPrompt,
-    settings?.llmModel
+    model,
+    systemPrompt
   ]);
   return processor;
 };
-var registerTriggersMutex = new import_async.Mutex();
+var registerTriggersMutex = new import_async2.Mutex();
 var useMessageQueue = (chat) => {
   const space = (0, import_echo6.getSpace)(chat);
   const queueDxn = (0, import_react10.useMemo)(() => {
@@ -1594,8 +1685,7 @@ var useMessageQueue = (chat) => {
     space,
     chat?.queue.dxn
   ]);
-  const edgeClient = (0, import_react_edge_client.useEdgeClient)();
-  return (0, import_react_edge_client.useQueue)(edgeClient, queueDxn);
+  return (0, import_echo6.useQueue)(queueDxn);
 };
 var MockServiceRegistry = class {
   async queryServices(query) {
@@ -1618,7 +1708,7 @@ var VISUAL_CROSSING_CREDENTIALS = {
   }
 };
 var TEST_SERVICES = [
-  (0, import_echo_schema4.createStatic)(import_chunk_37GI4NYH.ServiceType, {
+  (0, import_echo_schema3.createStatic)(import_chunk_KLSNCP34.ServiceType, {
     serviceId: "amadeus.com/service/FlightSearch",
     name: "Amadeus Flight Search",
     description: "Search for local and international flights.",
@@ -1631,7 +1721,7 @@ var TEST_SERVICES = [
       }
     ]
   }),
-  (0, import_echo_schema4.createStatic)(import_chunk_37GI4NYH.ServiceType, {
+  (0, import_echo_schema3.createStatic)(import_chunk_KLSNCP34.ServiceType, {
     serviceId: "amadeus.com/service/HotelSearch",
     name: "Amadeus Hotel Search",
     description: "Search for local and international hotels.",
@@ -1644,7 +1734,7 @@ var TEST_SERVICES = [
       }
     ]
   }),
-  (0, import_echo_schema4.createStatic)(import_chunk_37GI4NYH.ServiceType, {
+  (0, import_echo_schema3.createStatic)(import_chunk_KLSNCP34.ServiceType, {
     serviceId: "visualcrossing.com/service/Weather",
     name: "Visual Crossing Weather",
     description: "Search for global weather forecasts.",
@@ -1658,7 +1748,7 @@ var TEST_SERVICES = [
     ]
   }),
   // TODO(burdon): Needs auth.
-  (0, import_echo_schema4.createStatic)(import_chunk_37GI4NYH.ServiceType, {
+  (0, import_echo_schema3.createStatic)(import_chunk_KLSNCP34.ServiceType, {
     serviceId: "abstractapi.com/service/GeoLocation",
     name: "Abstract GeoLocation",
     description: "Get the location of any IP address.",
@@ -1675,11 +1765,11 @@ var TEST_SERVICES = [
   //
   ...Array.from({
     length: 20
-  }, (_, i) => (0, import_echo_schema4.createStatic)(import_chunk_37GI4NYH.ServiceType, {
+  }, (_, i) => (0, import_echo_schema3.createStatic)(import_chunk_KLSNCP34.ServiceType, {
     serviceId: `example.com/service/test-${i}`,
     name: `Test ${i}`,
     description: `Test ${i}`,
-    category: Object.keys(import_chunk_37GI4NYH.categoryIcons)[Math.floor(Math.random() * Object.keys(import_chunk_37GI4NYH.categoryIcons).length)],
+    category: Object.keys(import_chunk_KLSNCP34.categoryIcons)[Math.floor(Math.random() * Object.keys(import_chunk_KLSNCP34.categoryIcons).length)],
     interfaces: [
       {
         kind: "api",
@@ -1703,10 +1793,59 @@ var useServices = (space, query) => {
   ]);
   return services;
 };
-var __dxlog_file8 = "/home/runner/work/dxos/dxos/packages/plugins/experimental/plugin-assistant/src/components/Thread/ThreadContainer.tsx";
-var ThreadContainer = ({ classNames, chat, settings, onOpenChange, ...props }) => {
+var __dxlog_file8 = "/home/runner/work/dxos/dxos/packages/plugins/experimental/plugin-assistant/src/hooks/useContextProvider.ts";
+var stringMatch = (query, label) => label.toLowerCase().startsWith(query.toLowerCase());
+var useContextProvider = (space) => {
+  const artifactDefinitions = (0, import_app_framework4.useCapabilities)(import_app_framework4.Capabilities.ArtifactDefinition);
+  return (0, import_react13.useMemo)(() => {
+    if (!space) {
+      return void 0;
+    }
+    return {
+      query: async ({ query }) => {
+        const artifactSchemas = artifactDefinitions.map((artifact) => artifact.schema);
+        const { objects } = await space.db.query(import_echo7.Filter.or(...artifactSchemas.map((schema) => import_echo7.Filter.schema(schema)))).run();
+        return objects.map((object) => {
+          import_log10.log.info("object", {
+            object,
+            label: (0, import_echo_schema4.getLabel)((0, import_echo_schema4.getSchema)(object), object)
+          }, {
+            F: __dxlog_file8,
+            L: 36,
+            S: void 0,
+            C: (f, a) => f(...a)
+          });
+          return object;
+        }).filter((object) => stringMatch(query, (0, import_echo_schema4.getLabel)((0, import_echo_schema4.getSchema)(object), object) ?? "")).filter((object) => !!(0, import_echo_schema4.getDXN)(object)).map((object) => ({
+          uri: (0, import_echo_schema4.getDXN)(object).toString(),
+          label: (0, import_echo_schema4.getLabel)((0, import_echo_schema4.getSchema)(object), object) ?? ""
+        }));
+      },
+      resolveMetadata: async ({ uri }) => {
+        const object = await space.db.query({
+          id: uri
+        }).first();
+        return {
+          uri,
+          label: (0, import_echo_schema4.getLabel)((0, import_echo_schema4.getSchema)(object), object) ?? ""
+        };
+      }
+    };
+  }, [
+    space,
+    artifactDefinitions
+  ]);
+};
+var __dxlog_file9 = "/home/runner/work/dxos/dxos/packages/plugins/experimental/plugin-assistant/src/components/Thread/ThreadContainer.tsx";
+var ThreadContainer = ({ classNames, chat, settings, part, onOpenChange, ...props }) => {
   const space = (0, import_echo3.getSpace)(chat);
-  const processor = useChatProcessor(space, settings);
+  const contextProvider = useContextProvider(space);
+  const processor = useChatProcessor({
+    chat,
+    space,
+    settings,
+    part
+  });
   const messageQueue = useMessageQueue(chat);
   const messages = [
     ...messageQueue?.items ?? [],
@@ -1715,8 +1854,8 @@ var ThreadContainer = ({ classNames, chat, settings, onOpenChange, ...props }) =
   const handleSubmit = (0, import_react7.useCallback)((text) => {
     if (processor.streaming.value) {
       import_log5.log.warn("ignoring submit; still processing.", void 0, {
-        F: __dxlog_file8,
-        L: 39,
+        F: __dxlog_file9,
+        L: 41,
         S: void 0,
         C: (f, a) => f(...a)
       });
@@ -1724,8 +1863,8 @@ var ThreadContainer = ({ classNames, chat, settings, onOpenChange, ...props }) =
     }
     onOpenChange?.(true);
     (0, import_invariant3.invariant)(messageQueue, void 0, {
-      F: __dxlog_file8,
-      L: 45,
+      F: __dxlog_file9,
+      L: 47,
       S: void 0,
       A: [
         "messageQueue",
@@ -1762,6 +1901,7 @@ var ThreadContainer = ({ classNames, chat, settings, onOpenChange, ...props }) =
     onCancel: handleCancel,
     onPrompt: handleSubmit,
     onOpenChange,
+    contextProvider,
     ...props
   });
 };
@@ -1778,4 +1918,4 @@ var ThreadContainer = ({ classNames, chat, settings, onOpenChange, ...props }) =
   ToolboxContainer,
   useServices
 });
-//# sourceMappingURL=chunk-XUTDR7HI.cjs.map
+//# sourceMappingURL=chunk-JAB6KLPP.cjs.map