@dxos/plugin-assistant 0.8.0 → 0.8.1-main.81238a8

package/dist/lib/node-esm/{chunk-7JENJTLB.mjs → chunk-AMTHOYNB.mjs}

@@ -1,20 +1,20 @@
 import { createRequire } from 'node:module';const require = createRequire(import.meta.url);
 import {
   AssistantCapabilities
-} from "./chunk-77ARTFBA.mjs";
+} from "./chunk-2CIYX3SD.mjs";
 import {
   ServiceType,
   categoryIcons
-} from "./chunk-CJ4Y3QW5.mjs";
+} from "./chunk-NMMRHHAR.mjs";
 import {
   ASSISTANT_PLUGIN
-} from "./chunk-7SV6X6XU.mjs";
+} from "./chunk-N6BVC2C2.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";
@@ -22,7 +22,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";
@@ -551,7 +551,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";
@@ -682,7 +682,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,
@@ -707,7 +707,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,
@@ -722,7 +722,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,
@@ -738,7 +738,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'",
@@ -748,6 +748,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
       });
     }
@@ -774,7 +775,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'",
@@ -801,7 +802,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)));
@@ -892,7 +896,7 @@ var messageReducer = ({ current, messages }, message) => {
 // packages/plugins/experimental/plugin-assistant/src/components/Prompt/Prompt.tsx
 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
@@ -1049,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({
@@ -1063,11 +1184,14 @@ var Prompt = /* @__PURE__ */ forwardRef(({ classNames, autoFocus, lineWrapping =
       createThemeExtensions({
         themeMode
       }),
+      references ? promptReferences({
+        provider: references
+      }) : [],
       createAutocompleteExtension({
         onSubmit,
         onSuggest
       }),
-      keymap2.of([
+      keymap3.of([
         {
           key: "Alt-ArrowUp",
           preventDefault: true,
@@ -1116,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);
@@ -1138,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,
@@ -1153,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"
@@ -1180,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);
@@ -1206,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)
@@ -1230,31 +1374,32 @@ 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 { isToolUse, runTools, MixedStreamParser, DEFAULT_EDGE_MODEL } from "@dxos/assistant";
 import { createStatic } from "@dxos/echo-schema";
 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, _extensions, _options = defaultOptions) {
+    this._ai = _ai;
     this._tools = _tools;
     this._extensions = _extensions;
     this._options = _options;
@@ -1379,7 +1524,7 @@ var ChatProcessor = class {
           S: this,
           C: (f, a) => f(...a)
         });
-        this._stream = await this._service.exec({
+        this._stream = await this._ai.exec({
           ...this._options,
           // TODO(burdon): Rename messages or separate history/message.
           history: [
@@ -1449,37 +1594,49 @@ var ChatProcessor = class {
         S: this,
         C: (f, a) => f(...a)
       });
-      this.error.value = new Error("AI service error", {
-        cause: err
-      });
+      if (err instanceof Error && err.message.includes("Overloaded")) {
+        this.error.value = new AIServiecOverloadedError("AI service overloaded", {
+          cause: err
+        });
+      } else {
+        this.error.value = new Error("AI service error", {
+          cause: err
+        });
+      }
     } finally {
       this._stream = void 0;
     }
   }
 };
+var AIServiecOverloadedError = class extends Error {
+  constructor() {
+    super(...arguments);
+    this.code = "AI_SERVICE_OVERLOADED";
+  }
+};
 
 // packages/plugins/experimental/plugin-assistant/src/hooks/useChatProcessor.tsx
 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-66IBZVCX.mjs"));
-var AppGraphBuilder = lazy(() => import("./app-graph-builder-H2GC2AZA.mjs"));
-var IntentResolver = lazy(() => import("./intent-resolver-MR7BOKEW.mjs"));
-var ReactSurface = lazy(() => import("./react-surface-IGVYAOGL.mjs"));
-var AssistantSettings = lazy(() => import("./settings-S7P5RWQI.mjs"));
+var AiClient = lazy(() => import("./ai-client-XYZ5N7CR.mjs"));
+var AppGraphBuilder = lazy(() => import("./app-graph-builder-PMAQLTTN.mjs"));
+var IntentResolver = lazy(() => import("./intent-resolver-SQ4HLL5L.mjs"));
+var ReactSurface = lazy(() => import("./react-surface-LE57AGPI.mjs"));
+var AssistantSettings = lazy(() => import("./settings-VHR5KT4J.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);
@@ -1489,7 +1646,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)
     });
@@ -1502,10 +1659,13 @@ 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)
     });
@@ -1517,7 +1677,9 @@ var useChatProcessor = (space, settings) => {
     ];
     const extensions2 = {
       space,
-      dispatch
+      dispatch,
+      pivotId: chatId,
+      part
     };
     return [
       tools2,
@@ -1528,6 +1690,7 @@ var useChatProcessor = (space, settings) => {
     globalTools,
     artifactDefinitions,
     space,
+    chatId,
     serviceTools,
     functions
   ]);
@@ -1536,37 +1699,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: 84,
       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, 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";
@@ -1575,19 +1737,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(() => {
@@ -1601,8 +1762,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
@@ -1610,7 +1770,7 @@ 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";
@@ -1726,11 +1886,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 ?? [],
@@ -1738,9 +1954,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)
       });
@@ -1748,8 +1964,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",
@@ -1786,6 +2002,7 @@ var ThreadContainer = ({ classNames, chat, settings, onOpenChange, ...props }) =
     onCancel: handleCancel,
     onPrompt: handleSubmit,
     onOpenChange,
+    contextProvider,
     ...props
   });
 };
@@ -1802,4 +2019,4 @@ export {
   Thread,
   ThreadContainer
 };
-//# sourceMappingURL=chunk-7JENJTLB.mjs.map
+//# sourceMappingURL=chunk-AMTHOYNB.mjs.map