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

package/dist/lib/browser/{chunk-NFVIZS3B.mjs → chunk-KGIACFAX.mjs}

@@ -1,19 +1,19 @@
 import {
   AssistantCapabilities
-} from "./chunk-VZ4W6SHE.mjs";
+} from "./chunk-SVUCJXGN.mjs";
 import {
   ServiceType,
   categoryIcons
-} from "./chunk-FRIKXDDQ.mjs";
+} from "./chunk-X6ALDUA5.mjs";
 import {
   ASSISTANT_PLUGIN
-} from "./chunk-NV7SVHMV.mjs";
+} from "./chunk-AF7VQAKS.mjs";
 
 // packages/plugins/experimental/plugin-assistant/src/components/Toolbox/Toolbox.tsx
 import React, { useState, useEffect, Fragment } from "react";
 import { Capabilities, useCapabilities } from "@dxos/app-framework";
 import { parseToolName } from "@dxos/artifact";
-import { FunctionType } from "@dxos/functions";
+import { FunctionType } from "@dxos/functions/types";
 import { log as log3 } from "@dxos/log";
 import { Filter, useQuery } from "@dxos/react-client/echo";
 import { mx } from "@dxos/react-ui-theme";
@@ -21,7 +21,7 @@ import { mx } from "@dxos/react-ui-theme";
 // packages/plugins/experimental/plugin-assistant/src/tools/function.ts
 import { defineTool, ToolResult } from "@dxos/artifact";
 import { toEffectSchema } from "@dxos/echo-schema";
-import { getInvocationUrl, getUserFunctionUrlInMetadata } from "@dxos/functions";
+import { getInvocationUrl, getUserFunctionUrlInMetadata } from "@dxos/functions/types";
 import { log } from "@dxos/log";
 import { getMeta } from "@dxos/react-client/echo";
 var __dxlog_file = "/home/runner/work/dxos/dxos/packages/plugins/experimental/plugin-assistant/src/tools/function.ts";
@@ -550,7 +550,7 @@ import { keyToFallback } from "@dxos/util";
 // packages/plugins/experimental/plugin-assistant/src/components/Thread/ThreadMessage.tsx
 import React3 from "react";
 import { invariant as invariant2 } from "@dxos/invariant";
-import { Button, ButtonGroup, Icon, IconButton } from "@dxos/react-ui";
+import { Button, Icon, IconButton } from "@dxos/react-ui";
 import { MarkdownViewer, ToggleContainer as NativeToggleContainer } from "@dxos/react-ui-components";
 import { Json as Json2 } from "@dxos/react-ui-syntax-highlighter";
 import { mx as mx2 } from "@dxos/react-ui-theme";
@@ -681,7 +681,7 @@ var ToolContainer = ({ classNames, items }) => {
 // packages/plugins/experimental/plugin-assistant/src/components/Thread/ThreadMessage.tsx
 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__ */ React3.createElement(NativeToggleContainer, {
     ...props,
@@ -706,7 +706,7 @@ var ThreadMessage = ({ classNames, space, message, tools, onPrompt }) => {
   const { role, content = [] } = message;
   if (isToolMessage(message)) {
     return /* @__PURE__ */ React3.createElement(MessageContainer, {
-      classNames
+      classNames: mx2(classNames, "animate-[fadeIn_0.5s]")
     }, /* @__PURE__ */ React3.createElement(ToolBlock, {
       space,
       classNames: panelClassNames,
@@ -721,7 +721,7 @@ var ThreadMessage = ({ classNames, space, message, tools, onPrompt }) => {
     const Component = components[block.type] ?? components.default;
     return /* @__PURE__ */ React3.createElement(MessageContainer, {
       key: idx,
-      classNames,
+      classNames: mx2(classNames, "animate-[fadeIn_0.5s]"),
       user: block.type === "text" && role === "user"
     }, /* @__PURE__ */ React3.createElement(Component, {
       space,
@@ -737,7 +737,7 @@ var components = {
   ["text"]: ({ block }) => {
     invariant2(block.type === "text", void 0, {
       F: __dxlog_file5,
-      L: 86,
+      L: 90,
       S: void 0,
       A: [
         "block.type === 'text'",
@@ -747,6 +747,7 @@ var components = {
     const title = block.disposition ? titles[block.disposition] : void 0;
     if (!title) {
       return /* @__PURE__ */ React3.createElement(MarkdownViewer, {
+        classNames: "[&>p]:animate-[fadeIn_0.5s]",
         content: block.text
       });
     }
@@ -773,7 +774,7 @@ var components = {
   ["json"]: ({ space, block, onPrompt }) => {
     invariant2(block.type === "json", void 0, {
       F: __dxlog_file5,
-      L: 124,
+      L: 128,
       S: void 0,
       A: [
         "block.type === 'json'",
@@ -800,7 +801,10 @@ var components = {
       }
       case "select": {
         const { options = [] } = safeParseJson(block.json ?? "{}") ?? {};
-        return /* @__PURE__ */ React3.createElement(ButtonGroup, null, options.map((option) => /* @__PURE__ */ React3.createElement(Button, {
+        return /* @__PURE__ */ React3.createElement("div", {
+          className: "flex flex-wrap gap-1"
+        }, options.map((option, idx) => /* @__PURE__ */ React3.createElement(Button, {
+          classNames: "animate-[fadeIn_0.5s] rounded-2xl text-sm",
           key: option,
           onClick: () => onPrompt?.(option)
         }, option)));
@@ -889,9 +893,10 @@ var messageReducer = ({ current, messages }, message) => {
 };
 
 // packages/plugins/experimental/plugin-assistant/src/components/Prompt/Prompt.tsx
+import { Prec as Prec2 } from "@codemirror/state";
 import React4, { forwardRef, useImperativeHandle } from "react";
 import { useThemeContext } from "@dxos/react-ui";
-import { createBasicExtensions, createThemeExtensions, keymap as keymap2, useTextEditor } from "@dxos/react-ui-editor";
+import { createBasicExtensions, createThemeExtensions, keymap as keymap3, useTextEditor } from "@dxos/react-ui-editor";
 import { mx as mx3 } from "@dxos/react-ui-theme";
 
 // packages/plugins/experimental/plugin-assistant/src/components/Prompt/autocomplete.ts
@@ -1048,10 +1053,127 @@ var InlineSuggestionWidget = class extends WidgetType {
   }
 };
 
+// packages/plugins/experimental/plugin-assistant/src/components/Prompt/references.ts
+import { autocompletion, completionKeymap } from "@codemirror/autocomplete";
+import { RangeSet } from "@codemirror/state";
+import { Decoration as Decoration2, EditorView as EditorView2, keymap as keymap2, ViewPlugin as ViewPlugin2, WidgetType as WidgetType2 } from "@codemirror/view";
+import { Mutex } from "@dxos/async";
+var promptReferences = ({ provider, debug = false, triggerCharacter = "@" }) => {
+  if (triggerCharacter.length !== 1) {
+    throw new Error("triggerCharacter must be a single character");
+  }
+  const decorationField = ViewPlugin2.fromClass(class ReferenceView {
+    constructor(view) {
+      this._mutex = new Mutex();
+      this.decorations = Decoration2.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(Decoration2.replace({
+            widget: new ReferenceWidget(data)
+          }).range(match.index, match.index + reference.length));
+        }
+      }
+      return Decoration2.set(decorations);
+    }
+  }, {
+    decorations: (v) => v.decorations,
+    provide: (plugin) => [
+      EditorView2.atomicRanges.of((view) => view.plugin(decorationField)?.decorations ?? RangeSet.empty)
+    ]
+  });
+  return [
+    decorationField,
+    EditorView2.theme({
+      ".cm-reference-pill": {
+        borderRadius: "0.25rem",
+        borderWidth: "1px",
+        marginRight: "0.25rem",
+        marginLeft: "0.25rem"
+      }
+    }),
+    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
+    }),
+    keymap2.of(completionKeymap)
+  ];
+};
+var ReferenceWidget = class extends WidgetType2 {
+  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;
+  }
+};
+
 // packages/plugins/experimental/plugin-assistant/src/components/Prompt/Prompt.tsx
-var Prompt = /* @__PURE__ */ forwardRef(({ classNames, autoFocus, lineWrapping = false, placeholder, onSubmit, onSuggest, onOpenChange }, forwardRef2) => {
+var Prompt = /* @__PURE__ */ forwardRef(({ classNames, autoFocus, lineWrapping = false, placeholder, onSubmit, onSuggest, onOpenChange, references }, forwardRef2) => {
   const { themeMode } = useThemeContext();
   const { parentRef, view } = useTextEditor({
+    debug: true,
     autoFocus,
     extensions: [
       createBasicExtensions({
@@ -1062,13 +1184,16 @@ var Prompt = /* @__PURE__ */ forwardRef(({ classNames, autoFocus, lineWrapping =
       createThemeExtensions({
         themeMode
       }),
+      references ? promptReferences({
+        provider: references
+      }) : [],
       createAutocompleteExtension({
         onSubmit,
         onSuggest
       }),
-      keymap2.of([
+      Prec2.highest(keymap3.of([
         {
-          key: "Alt-ArrowUp",
+          key: "cmd-ArrowUp",
           preventDefault: true,
           run: (view2) => {
             onOpenChange?.(true);
@@ -1076,14 +1201,14 @@ var Prompt = /* @__PURE__ */ forwardRef(({ classNames, autoFocus, lineWrapping =
           }
         },
         {
-          key: "Alt-ArrowDown",
+          key: "cmd-ArrowDown",
           preventDefault: true,
           run: (view2) => {
             onOpenChange?.(false);
             return true;
           }
         }
-      ])
+      ]))
     ]
   }, [
     themeMode,
@@ -1115,17 +1240,17 @@ var Prompt = /* @__PURE__ */ forwardRef(({ classNames, autoFocus, lineWrapping =
   ]);
   return /* @__PURE__ */ React4.createElement("div", {
     ref: parentRef,
-    className: mx3("w-full overflow-hidden", classNames)
+    className: mx3("w-full", classNames)
   });
 });
 
 // packages/plugins/experimental/plugin-assistant/src/components/Prompt/PromptBar.tsx
 import React5, { useRef as useRef2, useState as useState3 } from "react";
 import { useVoiceInput } from "@dxos/plugin-transcription";
-import { Icon as Icon2, IconButton as IconButton2, useTranslation } from "@dxos/react-ui";
+import { Icon as Icon2, IconButton as IconButton2, Tooltip, useTranslation } from "@dxos/react-ui";
 import { Spinner } from "@dxos/react-ui-sfx";
 import { errorText, mx as mx4 } from "@dxos/react-ui-theme";
-var PromptBar = ({ classNames, placeholder, processing, error, microphone, onCancel, ...props }) => {
+var PromptBar = ({ classNames, placeholder, processing, error, microphone, onCancel, references, ...props }) => {
   const { t } = useTranslation(ASSISTANT_PLUGIN);
   const promptRef = useRef2(null);
   const [active, setActive] = useState3(false);
@@ -1137,14 +1262,18 @@ var PromptBar = ({ classNames, placeholder, processing, error, microphone, onCan
     }
   });
   return /* @__PURE__ */ React5.createElement("div", {
-    className: mx4("flex shrink-0 w-full grid grid-cols-[var(--rail-action)_1fr_var(--rail-action)] overflow-hidden", classNames)
+    className: mx4("shrink-0 w-full grid grid-cols-[var(--rail-action)_1fr_var(--rail-action)] overflow-hidden", classNames)
   }, /* @__PURE__ */ React5.createElement("div", {
     className: "flex w-[--rail-action] h-[--rail-action] items-center justify-center"
-  }, error && /* @__PURE__ */ React5.createElement(Icon2, {
+  }, error && /* @__PURE__ */ React5.createElement(Tooltip.Root, {
+    delayDuration: 0
+  }, /* @__PURE__ */ React5.createElement(Tooltip.Trigger, null, /* @__PURE__ */ React5.createElement(Icon2, {
     icon: "ph--warning-circle--regular",
     classNames: errorText,
     size: 5
-  }) || /* @__PURE__ */ React5.createElement(Spinner, {
+  })), /* @__PURE__ */ React5.createElement(Tooltip.Portal, null, /* @__PURE__ */ React5.createElement(Tooltip.Content, null, /* @__PURE__ */ React5.createElement("div", {
+    className: "text-sm text-error-500"
+  }, error.message), /* @__PURE__ */ React5.createElement(Tooltip.Arrow, null)))) || /* @__PURE__ */ React5.createElement(Spinner, {
     active: processing
   })), /* @__PURE__ */ React5.createElement(Prompt, {
     ref: promptRef,
@@ -1152,6 +1281,7 @@ var PromptBar = ({ classNames, placeholder, processing, error, microphone, onCan
     classNames: "pbs-2",
     lineWrapping: true,
     placeholder: placeholder ?? t("prompt placeholder"),
+    references,
     ...props
   }), (onCancel || microphone) && /* @__PURE__ */ React5.createElement("div", {
     className: "flex w-[--rail-action] h-[--rail-action] items-center justify-center"
@@ -1179,7 +1309,7 @@ var PromptBar = ({ classNames, placeholder, processing, error, microphone, onCan
 };
 
 // packages/plugins/experimental/plugin-assistant/src/components/Thread/Thread.tsx
-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 = useRef3(null);
   const identity = useIdentity();
   const fallbackValue = keyToFallback(identity.identityKey);
@@ -1205,6 +1335,21 @@ var Thread = ({ classNames, space, messages, collapse = true, transcription, pro
     messages,
     collapse
   ]);
+  const references = useMemo2(() => {
+    if (!contextProvider) {
+      return void 0;
+    }
+    return {
+      getReferences: async ({ query }) => contextProvider.query({
+        query
+      }),
+      resolveReference: async ({ uri }) => contextProvider.resolveMetadata({
+        uri
+      })
+    };
+  }, [
+    contextProvider
+  ]);
   return /* @__PURE__ */ React6.createElement("div", {
     role: "none",
     className: mx5("flex flex-col grow overflow-hidden", classNames)
@@ -1229,38 +1374,36 @@ var Thread = ({ classNames, space, messages, collapse = true, transcription, pro
     error,
     onSubmit: handleSubmit,
     onCancel,
-    onOpenChange
+    onOpenChange,
+    references
   }));
 };
 
 // packages/plugins/experimental/plugin-assistant/src/components/Thread/ThreadContainer.tsx
 import React7, { useCallback as useCallback3 } from "react";
 import { invariant as invariant6 } from "@dxos/invariant";
-import { log as log9 } from "@dxos/log";
+import { log as log10 } from "@dxos/log";
 import { getSpace as getSpace2 } from "@dxos/react-client/echo";
 
 // packages/plugins/experimental/plugin-assistant/src/hooks/processor.ts
 import { batch, computed, signal } from "@preact/signals-core";
-import { Message } from "@dxos/artifact";
-import { isToolUse, runTools, MixedStreamParser, DEFAULT_LLM_MODEL } from "@dxos/assistant";
-import { createStatic } from "@dxos/echo-schema";
+import { AISession, DEFAULT_EDGE_MODEL } from "@dxos/assistant";
 import { invariant as invariant3 } from "@dxos/invariant";
 import { log as log5 } from "@dxos/log";
 var __dxlog_file6 = "/home/runner/work/dxos/dxos/packages/plugins/experimental/plugin-assistant/src/hooks/processor.ts";
 var defaultOptions = {
-  model: DEFAULT_LLM_MODEL,
+  model: 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 MixedStreamParser();
     this._pending = signal([]);
     this._block = signal(void 0);
-    this._history = [];
     this.streaming = computed(() => this._block.value !== void 0);
     this.error = signal(void 0);
     this.messages = computed(() => {
@@ -1271,7 +1414,7 @@ var ChatProcessor = class {
         const current = messages.pop();
         invariant3(current, void 0, {
           F: __dxlog_file6,
-          L: 85,
+          L: 70,
           S: this,
           A: [
             "current",
@@ -1290,20 +1433,6 @@ var ChatProcessor = class {
       }
       return messages;
     });
-    this._parser.message.on((message) => {
-      batch(() => {
-        this._pending.value = [
-          ...this._pending.value,
-          message
-        ];
-        this._block.value = void 0;
-      });
-    });
-    this._parser.update.on((block) => {
-      batch(() => {
-        this._block.value = block;
-      });
-    });
   }
   get tools() {
     return this._tools;
@@ -1318,23 +1447,70 @@ var ChatProcessor = class {
   * Make GPT request.
   */
   async request(message, options = {}) {
-    batch(() => {
-      this._history = options.history ?? [];
+    this._session = new AISession({
+      operationModel: "immediate"
+    });
+    this._session.message.on((message2) => {
+      batch(() => {
+        this._pending.value = [
+          ...this._pending.value,
+          message2
+        ];
+        this._block.value = void 0;
+      });
+    });
+    this._session.update.on((block) => {
+      batch(() => {
+        this._block.value = block;
+      });
+    });
+    this._session.userMessage.on((message2) => {
       this._pending.value = [
-        createStatic(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
+        }
+      });
+      log5.info("completed", {
+        messages
+      }, {
+        F: __dxlog_file6,
+        L: 137,
+        S: this,
+        C: (f, a) => f(...a)
+      });
+      options.onComplete?.(this._pending.value);
+    } catch (err) {
+      log5.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();
   }
   /**
@@ -1344,116 +1520,26 @@ var ChatProcessor = class {
   async cancel() {
     log5.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;
     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 {
-        log5("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();
-        log5("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);
-        invariant3(message, void 0, {
-          F: __dxlog_file6,
-          L: 200,
-          S: this,
-          A: [
-            "message",
-            ""
-          ]
-        });
-        if (isToolUse(message)) {
-          log5("tool request...", void 0, {
-            F: __dxlog_file6,
-            L: 202,
-            S: this,
-            C: (f, a) => f(...a)
-          });
-          const response = await runTools({
-            message: this._pending.value.at(-1),
-            tools: this._tools ?? [],
-            extensions: this._extensions
-          });
-          log5("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) {
-      log5.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";
   }
 };
 
@@ -1461,24 +1547,24 @@ var ChatProcessor = class {
 import { useEffect as useEffect3, useMemo as useMemo3, useState as useState4 } from "react";
 import { Capabilities as Capabilities2, useCapabilities as useCapabilities2, useCapability, useIntentDispatcher } from "@dxos/app-framework";
 import { createSystemPrompt } from "@dxos/artifact";
-import { DEFAULT_LLM_MODEL as DEFAULT_LLM_MODEL2 } from "@dxos/assistant";
-import { FunctionType as FunctionType2 } from "@dxos/functions";
+import { DEFAULT_EDGE_MODEL as DEFAULT_EDGE_MODEL2, DEFAULT_OLLAMA_MODEL } from "@dxos/assistant";
+import { FunctionType as FunctionType2 } from "@dxos/functions/types";
 import { log as log6 } from "@dxos/log";
 import { useConfig } from "@dxos/react-client";
-import { Filter as Filter2, useQuery as useQuery2 } from "@dxos/react-client/echo";
+import { Filter as Filter2, fullyQualifiedId, useQuery as useQuery2 } from "@dxos/react-client/echo";
 import { isNonNullable as isNonNullable2 } from "@dxos/util";
 
 // packages/plugins/experimental/plugin-assistant/src/capabilities/index.ts
 import { lazy } from "@dxos/app-framework";
-var AiClient = lazy(() => import("./ai-client-RTCGRKZE.mjs"));
-var AppGraphBuilder = lazy(() => import("./app-graph-builder-AXAIFOGV.mjs"));
-var IntentResolver = lazy(() => import("./intent-resolver-QRVRZL6K.mjs"));
-var ReactSurface = lazy(() => import("./react-surface-JLXNWOI6.mjs"));
-var AssistantSettings = lazy(() => import("./settings-JTT62IHD.mjs"));
+var AiClient = lazy(() => import("./ai-client-BAPVMSNX.mjs"));
+var AppGraphBuilder = lazy(() => import("./app-graph-builder-DTCUWBKB.mjs"));
+var IntentResolver = lazy(() => import("./intent-resolver-U57FXP3I.mjs"));
+var ReactSurface = lazy(() => import("./react-surface-YNN5NYJW.mjs"));
+var AssistantSettings = lazy(() => import("./settings-VAW6UWFL.mjs"));
 
 // packages/plugins/experimental/plugin-assistant/src/hooks/useChatProcessor.tsx
 var __dxlog_file7 = "/home/runner/work/dxos/dxos/packages/plugins/experimental/plugin-assistant/src/hooks/useChatProcessor.tsx";
-var useChatProcessor = (space, settings) => {
+var useChatProcessor = ({ chat, space, settings, part = "deck" }) => {
   const aiClient = useCapability(AssistantCapabilities.AiClient);
   const globalTools = useCapabilities2(Capabilities2.Tools);
   const artifactDefinitions = useCapabilities2(Capabilities2.ArtifactDefinition);
@@ -1488,7 +1574,7 @@ var useChatProcessor = (space, settings) => {
   useEffect3(() => {
     log6("creating service tools...", void 0, {
       F: __dxlog_file7,
-      L: 34,
+      L: 41,
       S: void 0,
       C: (f, a) => f(...a)
     });
@@ -1501,22 +1587,26 @@ var useChatProcessor = (space, settings) => {
   ]);
   const config = useConfig();
   const functions = useQuery2(space, Filter2.schema(FunctionType2));
+  const chatId = useMemo3(() => chat ? fullyQualifiedId(chat) : void 0, [
+    chat
+  ]);
   const [tools, extensions] = useMemo3(() => {
     log6("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(isNonNullable2)
     ];
     const extensions2 = {
       space,
-      dispatch
+      dispatch,
+      pivotId: chatId,
+      part
     };
     return [
       tools2,
@@ -1525,8 +1615,8 @@ var useChatProcessor = (space, settings) => {
   }, [
     dispatch,
     globalTools,
-    artifactDefinitions,
     space,
+    chatId,
     serviceTools,
     functions
   ]);
@@ -1535,37 +1625,36 @@ var useChatProcessor = (space, settings) => {
   }), [
     artifactDefinitions
   ]);
+  const model = settings?.llmProvider === "ollama" ? settings?.ollamaModel ?? DEFAULT_OLLAMA_MODEL : settings?.edgeModel ?? DEFAULT_EDGE_MODEL2;
   const processor = useMemo3(() => {
     log6("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 ?? DEFAULT_LLM_MODEL2,
-      // 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;
 };
 
 // packages/plugins/experimental/plugin-assistant/src/hooks/useLocalTriggerManager.ts
 import { useEffect as useEffect4, useState as useState5 } from "react";
-import { Mutex } from "@dxos/async";
+import { Mutex as Mutex2 } from "@dxos/async";
 import { Context } from "@dxos/context";
 import { createSubscriptionTrigger } from "@dxos/functions";
-import { FunctionTrigger } from "@dxos/functions";
+import { FunctionTrigger } from "@dxos/functions/types";
 import { invariant as invariant5 } from "@dxos/invariant";
 import { log as log8 } from "@dxos/log";
 import { useClient } from "@dxos/react-client";
@@ -1574,19 +1663,18 @@ import { Filter as Filter3, useQuery as useQuery3 } from "@dxos/react-client/ech
 // packages/plugins/experimental/plugin-assistant/src/hooks/invocation-handler.ts
 import { sleep } from "@dxos/async";
 import { getObjectCore, ResultFormat } from "@dxos/echo-db";
-import { FunctionType as FunctionType3 } from "@dxos/functions";
+import { FunctionType as FunctionType3, getUserFunctionUrlInMetadata as getUserFunctionUrlInMetadata2 } from "@dxos/functions/types";
 import { invariant as invariant4 } from "@dxos/invariant";
 import { DXN, LOCAL_SPACE_TAG } from "@dxos/keys";
 import { log as log7 } from "@dxos/log";
 
 // packages/plugins/experimental/plugin-assistant/src/hooks/useLocalTriggerManager.ts
-var registerTriggersMutex = new Mutex();
+var registerTriggersMutex = new Mutex2();
 
 // packages/plugins/experimental/plugin-assistant/src/hooks/useMessageQueue.ts
 import { useMemo as useMemo4 } from "react";
 import { DXN as DXN2, QueueSubspaceTags } from "@dxos/keys";
-import { getSpace } from "@dxos/react-client/echo";
-import { useEdgeClient, useQueue } from "@dxos/react-edge-client";
+import { getSpace, useQueue } from "@dxos/react-client/echo";
 var useMessageQueue = (chat) => {
   const space = getSpace(chat);
   const queueDxn = useMemo4(() => {
@@ -1600,8 +1688,7 @@ var useMessageQueue = (chat) => {
     space,
     chat?.queue.dxn
   ]);
-  const edgeClient = useEdgeClient();
-  return useQueue(edgeClient, queueDxn);
+  return useQueue(queueDxn);
 };
 
 // packages/plugins/experimental/plugin-assistant/src/hooks/useServices.ts
@@ -1609,10 +1696,10 @@ import { useEffect as useEffect5, useMemo as useMemo5, useState as useState6 } f
 
 // packages/plugins/experimental/plugin-assistant/src/testing/test-functions.ts
 import { createSystemPrompt as createSystemPrompt2 } from "@dxos/artifact";
-import { DEFAULT_LLM_MODEL as DEFAULT_LLM_MODEL3 } from "@dxos/assistant";
+import { DEFAULT_EDGE_MODEL as DEFAULT_EDGE_MODEL3 } from "@dxos/assistant";
 
 // packages/plugins/experimental/plugin-assistant/src/testing/test-services.ts
-import { createStatic as createStatic2 } from "@dxos/echo-schema";
+import { createStatic } from "@dxos/echo-schema";
 var MockServiceRegistry = class {
   async queryServices(query) {
     return TEST_SERVICES;
@@ -1634,7 +1721,7 @@ var VISUAL_CROSSING_CREDENTIALS = {
   }
 };
 var TEST_SERVICES = [
-  createStatic2(ServiceType, {
+  createStatic(ServiceType, {
     serviceId: "amadeus.com/service/FlightSearch",
     name: "Amadeus Flight Search",
     description: "Search for local and international flights.",
@@ -1647,7 +1734,7 @@ var TEST_SERVICES = [
       }
     ]
   }),
-  createStatic2(ServiceType, {
+  createStatic(ServiceType, {
     serviceId: "amadeus.com/service/HotelSearch",
     name: "Amadeus Hotel Search",
     description: "Search for local and international hotels.",
@@ -1660,7 +1747,7 @@ var TEST_SERVICES = [
       }
     ]
   }),
-  createStatic2(ServiceType, {
+  createStatic(ServiceType, {
     serviceId: "visualcrossing.com/service/Weather",
     name: "Visual Crossing Weather",
     description: "Search for global weather forecasts.",
@@ -1674,7 +1761,7 @@ var TEST_SERVICES = [
     ]
   }),
   // TODO(burdon): Needs auth.
-  createStatic2(ServiceType, {
+  createStatic(ServiceType, {
     serviceId: "abstractapi.com/service/GeoLocation",
     name: "Abstract GeoLocation",
     description: "Get the location of any IP address.",
@@ -1691,7 +1778,7 @@ var TEST_SERVICES = [
   //
   ...Array.from({
     length: 20
-  }, (_, i) => createStatic2(ServiceType, {
+  }, (_, i) => createStatic(ServiceType, {
     serviceId: `example.com/service/test-${i}`,
     name: `Test ${i}`,
     description: `Test ${i}`,
@@ -1725,11 +1812,67 @@ var useServices = (space, query) => {
 // packages/plugins/experimental/plugin-assistant/src/hooks/useTextInputEvents.ts
 import { useState as useState7, useCallback as useCallback2 } from "react";
 
+// packages/plugins/experimental/plugin-assistant/src/hooks/useContextProvider.ts
+import { useMemo as useMemo6 } from "react";
+import { Capabilities as Capabilities3, useCapabilities as useCapabilities3 } from "@dxos/app-framework";
+import { getDXN, getLabel, getSchema } from "@dxos/echo-schema";
+import { log as log9 } from "@dxos/log";
+import { Filter as Filter4 } from "@dxos/react-client/echo";
+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 = useCapabilities3(Capabilities3.ArtifactDefinition);
+  return useMemo6(() => {
+    if (!space) {
+      return void 0;
+    }
+    return {
+      query: async ({ query }) => {
+        const artifactSchemas = artifactDefinitions.map((artifact) => artifact.schema);
+        const { objects } = await space.db.query(Filter4.or(...artifactSchemas.map((schema) => Filter4.schema(schema)))).run();
+        return objects.map((object) => {
+          log9.info("object", {
+            object,
+            label: getLabel(getSchema(object), object)
+          }, {
+            F: __dxlog_file8,
+            L: 36,
+            S: void 0,
+            C: (f, a) => f(...a)
+          });
+          return object;
+        }).filter((object) => stringMatch(query, getLabel(getSchema(object), object) ?? "")).filter((object) => !!getDXN(object)).map((object) => ({
+          uri: getDXN(object).toString(),
+          label: getLabel(getSchema(object), object) ?? ""
+        }));
+      },
+      resolveMetadata: async ({ uri }) => {
+        const object = await space.db.query({
+          id: uri
+        }).first();
+        return {
+          uri,
+          label: getLabel(getSchema(object), object) ?? ""
+        };
+      }
+    };
+  }, [
+    space,
+    artifactDefinitions
+  ]);
+};
+
 // packages/plugins/experimental/plugin-assistant/src/components/Thread/ThreadContainer.tsx
-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_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 = getSpace2(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 ?? [],
@@ -1737,9 +1880,9 @@ var ThreadContainer = ({ classNames, chat, settings, onOpenChange, ...props }) =
   ];
   const handleSubmit = useCallback3((text) => {
     if (processor.streaming.value) {
-      log9.warn("ignoring submit; still processing.", void 0, {
-        F: __dxlog_file8,
-        L: 39,
+      log10.warn("ignoring submit; still processing.", void 0, {
+        F: __dxlog_file9,
+        L: 41,
         S: void 0,
         C: (f, a) => f(...a)
       });
@@ -1747,8 +1890,8 @@ var ThreadContainer = ({ classNames, chat, settings, onOpenChange, ...props }) =
     }
     onOpenChange?.(true);
     invariant6(messageQueue, void 0, {
-      F: __dxlog_file8,
-      L: 45,
+      F: __dxlog_file9,
+      L: 47,
       S: void 0,
       A: [
         "messageQueue",
@@ -1785,6 +1928,7 @@ var ThreadContainer = ({ classNames, chat, settings, onOpenChange, ...props }) =
     onCancel: handleCancel,
     onPrompt: handleSubmit,
     onOpenChange,
+    contextProvider,
     ...props
   });
 };
@@ -1801,4 +1945,4 @@ export {
   Thread,
   ThreadContainer
 };
-//# sourceMappingURL=chunk-NFVIZS3B.mjs.map
+//# sourceMappingURL=chunk-KGIACFAX.mjs.map