@dxos/plugin-assistant 0.8.1-main.ae460ac → 0.8.1-staging.31c3ee1

package/src/AssistantPlugin.tsx

@@ -2,14 +2,21 @@
 // Copyright 2023 DXOS.org
 //
 
-import { Capabilities, contributes, createIntent, defineModule, definePlugin, Events } from '@dxos/app-framework';
+import {
+  allOf,
+  Capabilities,
+  contributes,
+  createIntent,
+  defineModule,
+  definePlugin,
+  Events,
+} from '@dxos/app-framework';
 import { ClientCapabilities, ClientEvents } from '@dxos/plugin-client';
-import { DeckCapabilities, DeckEvents } from '@dxos/plugin-deck';
 import { SpaceCapabilities } from '@dxos/plugin-space';
 import { defineObjectForm } from '@dxos/plugin-space/types';
 
 import { AiClient, AppGraphBuilder, IntentResolver, ReactSurface, AssistantSettings } from './capabilities';
-import { ASSISTANT_PLUGIN, meta } from './meta';
+import { meta } from './meta';
 import translations from './translations';
 import { AssistantAction, AIChatType, ServiceType, TemplateType } from './types';
 
@@ -32,14 +39,12 @@ export const AssistantPlugin = () =>
         contributes(Capabilities.Metadata, {
           id: TemplateType.typename,
           metadata: {
-            placeholder: ['template title placeholder', { ns: ASSISTANT_PLUGIN }],
             icon: 'ph--code-block--regular',
           },
         }),
         contributes(Capabilities.Metadata, {
           id: AIChatType.typename,
           metadata: {
-            placeholder: ['chat title placeholder', { ns: ASSISTANT_PLUGIN }],
             icon: 'ph--atom--regular',
           },
         }),
@@ -60,6 +65,7 @@ export const AssistantPlugin = () =>
           SpaceCapabilities.ObjectForm,
           defineObjectForm({
             objectSchema: TemplateType,
+            hidden: true,
             getIntent: () => createIntent(AssistantAction.CreateTemplate),
           }),
         ),
@@ -70,23 +76,6 @@ export const AssistantPlugin = () =>
       activatesOn: ClientEvents.SetupSchema,
       activate: () => contributes(ClientCapabilities.Schema, [ServiceType, TemplateType]),
     }),
-    defineModule({
-      id: `${meta.id}/module/complementary-panels`,
-      activatesOn: DeckEvents.SetupComplementaryPanels,
-      activate: () =>
-        contributes(DeckCapabilities.ComplementaryPanel, {
-          id: 'service-registry',
-          label: ['service registry label', { ns: ASSISTANT_PLUGIN }],
-          icon: 'ph--plugs--regular',
-        }),
-    }),
-    defineModule({
-      id: `${meta.id}/module/react-surface`,
-      activatesOn: Events.SetupReactSurface,
-      // TODO(wittjosiah): Should occur before the chat is loaded when surfaces activation is more granular.
-      activatesBefore: [Events.SetupArtifactDefinition],
-      activate: ReactSurface,
-    }),
     defineModule({
       id: `${meta.id}/module/app-graph-builder`,
       activatesOn: Events.SetupAppGraph,
@@ -97,9 +86,16 @@ export const AssistantPlugin = () =>
       activatesOn: Events.SetupIntentResolver,
       activate: IntentResolver,
     }),
+    defineModule({
+      id: `${meta.id}/module/react-surface`,
+      activatesOn: Events.SetupReactSurface,
+      // TODO(wittjosiah): Should occur before the chat is loaded when surfaces activation is more granular.
+      activatesBefore: [Events.SetupArtifactDefinition],
+      activate: ReactSurface,
+    }),
     defineModule({
       id: `${meta.id}/module/ai-client`,
-      activatesOn: ClientEvents.ClientReady,
+      activatesOn: allOf(ClientEvents.ClientReady, Events.SettingsReady),
       activate: AiClient,
     }),
   ]);

package/src/capabilities/ai-client.ts

@@ -2,12 +2,15 @@
 // Copyright 2025 DXOS.org
 //
 
-import { contributes, type PluginsContext } from '@dxos/app-framework';
-import { AIServiceEdgeClient } from '@dxos/assistant';
-import { log } from '@dxos/log';
+import { effect, signal } from '@preact/signals-core';
+
+import { Capabilities, contributes, type PluginsContext } from '@dxos/app-framework';
+import { type AIServiceClient, AIServiceEdgeClient, OllamaClient } from '@dxos/assistant';
 import { ClientCapabilities } from '@dxos/plugin-client';
 
 import { AssistantCapabilities } from './capabilities';
+import { ASSISTANT_PLUGIN } from '../meta';
+import { type AssistantSettingsProps } from '../types';
 
 // TODO(wittjosiah): Factor out.
 const DEFAULT_AI_SERVICE_URL = 'http://localhost:8788';
@@ -15,8 +18,21 @@ const DEFAULT_AI_SERVICE_URL = 'http://localhost:8788';
 export default (context: PluginsContext) => {
   const client = context.requestCapability(ClientCapabilities.Client);
   const endpoint = client.config.values.runtime?.services?.ai?.server ?? DEFAULT_AI_SERVICE_URL;
-  // TODO(burdon): Use prefs.
-  const ai = new AIServiceEdgeClient({ endpoint });
-  log.info('ai', { endpoint, context });
-  return contributes(AssistantCapabilities.AiClient, ai);
+
+  const ai = signal<AIServiceClient>(new AIServiceEdgeClient({ endpoint }));
+
+  const unsubscribe = effect(() => {
+    // TODO(burdon): Could be undefined.
+    const settings = context
+      .requestCapability(Capabilities.SettingsStore)
+      .getStore<AssistantSettingsProps>(ASSISTANT_PLUGIN)?.value;
+
+    if (settings?.llmProvider === 'ollama') {
+      ai.value = new OllamaClient();
+    } else {
+      ai.value = new AIServiceEdgeClient({ endpoint });
+    }
+  });
+
+  return contributes(AssistantCapabilities.AiClient, ai, () => unsubscribe());
 };

package/src/capabilities/app-graph-builder.ts

@@ -11,14 +11,23 @@ import {
   type PluginsContext,
 } from '@dxos/app-framework';
 import { invariant } from '@dxos/invariant';
-import { log } from '@dxos/log';
 import { ClientCapabilities } from '@dxos/plugin-client';
+import { PLANK_COMPANION_TYPE, ATTENDABLE_PATH_SEPARATOR, DECK_COMPANION_TYPE } from '@dxos/plugin-deck/types';
 import { createExtension, type Node, ROOT_ID } from '@dxos/plugin-graph';
+import { memoizeQuery } from '@dxos/plugin-space';
 import { SpaceAction } from '@dxos/plugin-space/types';
-import { type Space, Filter, getSpace } from '@dxos/react-client/echo';
+import {
+  type Space,
+  Filter,
+  fullyQualifiedId,
+  getSpace,
+  isSpace,
+  isReactiveObject,
+  parseId,
+} from '@dxos/react-client/echo';
 
 import { ASSISTANT_DIALOG, ASSISTANT_PLUGIN } from '../meta';
-import { AIChatType, AssistantAction } from '../types';
+import { AIChatType, AssistantAction, TemplateType } from '../types';
 
 export default (context: PluginsContext) =>
   contributes(Capabilities.AppGraphBuilder, [
@@ -51,7 +60,6 @@ export default (context: PluginsContext) =>
             }
 
             if (!chat) {
-              log.warn('no chat found');
               return;
             }
 
@@ -71,7 +79,7 @@ export default (context: PluginsContext) =>
           },
           properties: {
             label: ['open assistant label', { ns: ASSISTANT_PLUGIN }],
-            icon: 'ph--chat-centered-text--regular',
+            icon: 'ph--star-four--regular',
             disposition: 'pin-end',
             position: 'hoist',
             keyBinding: {
@@ -82,6 +90,93 @@ export default (context: PluginsContext) =>
         },
       ],
     }),
+
+    createExtension({
+      id: `${ASSISTANT_PLUGIN}/service-registry`,
+      filter: (node): node is Node<null> => node.id === 'root',
+      connector: ({ node }) => {
+        const layout = context.requestCapability(Capabilities.Layout);
+        const client = context.requestCapability(ClientCapabilities.Client);
+        const { spaceId } = parseId(layout.workspace);
+        const space = spaceId ? client.spaces.get(spaceId) : null;
+
+        return [
+          {
+            id: [node.id, 'service-registry'].join(ATTENDABLE_PATH_SEPARATOR),
+            type: DECK_COMPANION_TYPE,
+            data: space,
+            properties: {
+              label: ['service registry label', { ns: ASSISTANT_PLUGIN }],
+              icon: 'ph--plugs--regular',
+              disposition: 'hidden',
+            },
+          },
+        ];
+      },
+    }),
+
+    createExtension({
+      id: `${ASSISTANT_PLUGIN}/object-chat-companion`,
+      filter: (node): node is Node<AIChatType> =>
+        isReactiveObject(node.data) && node.data.assistantChatQueue && node.data.type !== AIChatType.typename,
+      connector: ({ node }) => [
+        {
+          id: [node.id, 'assistant-chat'].join(ATTENDABLE_PATH_SEPARATOR),
+          type: PLANK_COMPANION_TYPE,
+          data: 'assistant-chat',
+          properties: {
+            label: ['assistant chat label', { ns: ASSISTANT_PLUGIN }],
+            icon: 'ph--star-four--regular',
+            position: 'hoist',
+          },
+        },
+      ],
+    }),
+
+    createExtension({
+      id: `${ASSISTANT_PLUGIN}/root`,
+      filter: (node): node is Node<Space> => isSpace(node.data),
+      connector: ({ node }) => {
+        const templates = memoizeQuery(node.data, Filter.schema(TemplateType));
+        return templates.length > 0
+          ? [
+              {
+                id: `${ASSISTANT_PLUGIN}/templates`,
+                type: `${ASSISTANT_PLUGIN}/templates`,
+                data: null,
+                properties: {
+                  label: ['templates label', { ns: ASSISTANT_PLUGIN }],
+                  icon: 'ph--file-code--regular',
+                  space: node.data,
+                },
+              },
+            ]
+          : [];
+      },
+    }),
+
+    createExtension({
+      id: `${ASSISTANT_PLUGIN}/templates`,
+      filter: (node): node is Node<null, { space: Space }> => node.id === `${ASSISTANT_PLUGIN}/templates`,
+      connector: ({ node }) => {
+        const templates = memoizeQuery(node.properties.space, Filter.schema(TemplateType));
+        return templates
+          .toSorted((a, b) => {
+            const nameA = a.name ?? '';
+            const nameB = b.name ?? '';
+            return nameA.localeCompare(nameB);
+          })
+          .map((template) => ({
+            id: fullyQualifiedId(template),
+            type: `${ASSISTANT_PLUGIN}/template`,
+            data: template,
+            properties: {
+              label: template.name ?? ['template title placeholder', { ns: ASSISTANT_PLUGIN }],
+              icon: 'ph--file-code--regular',
+            },
+          }));
+      },
+    }),
   ]);
 
 // TODO(burdon): Factor out.

package/src/capabilities/capabilities.ts

@@ -2,11 +2,13 @@
 // Copyright 2025 DXOS.org
 //
 
+import { type ReadonlySignal } from '@preact/signals-core';
+
 import { defineCapability } from '@dxos/app-framework';
 import { type AIServiceClient } from '@dxos/assistant';
 
 import { ASSISTANT_PLUGIN } from '../meta';
 
 export namespace AssistantCapabilities {
-  export const AiClient = defineCapability<AIServiceClient>(`${ASSISTANT_PLUGIN}/capability/ai-client`);
+  export const AiClient = defineCapability<ReadonlySignal<AIServiceClient>>(`${ASSISTANT_PLUGIN}/capability/ai-client`);
 }

package/src/capabilities/intent-resolver.ts

@@ -17,8 +17,9 @@ export default () => [
       resolve: ({ spaceId, name }) => ({
         data: {
           object: create(AIChatType, {
-            name,
-            queue: refFromDXN(new DXN(DXN.kind.QUEUE, [QueueSubspaceTags.DATA, spaceId, ObjectId.random()])),
+            assistantChatQueue: refFromDXN(
+              new DXN(DXN.kind.QUEUE, [QueueSubspaceTags.DATA, spaceId, ObjectId.random()]),
+            ),
           }),
         },
       }),
@@ -30,7 +31,7 @@ export default () => [
       intent: AssistantAction.CreateTemplate,
       resolve: ({ name }) => ({
         data: {
-          object: create(TemplateType, { name, source: '{{! Template }}' }),
+          object: create(TemplateType, { name, kind: { include: 'manual' }, source: '{{! Template }}' }),
         },
       }),
     }),

package/src/capabilities/react-surface.tsx

@@ -2,11 +2,13 @@
 // Copyright 2025 DXOS.org
 //
 
-import React from 'react';
+import React, { useMemo } from 'react';
 
 import { Capabilities, contributes, createSurface } from '@dxos/app-framework';
+import { isInstanceOf } from '@dxos/echo-schema';
+import { getTypename } from '@dxos/live-object';
 import { SettingsStore } from '@dxos/local-storage';
-import { getSpace, isSpace } from '@dxos/react-client/echo';
+import { fullyQualifiedId, isReactiveObject, isSpace, type Space } from '@dxos/react-client/echo';
 
 import { AssistantDialog, AssistantSettings, ChatContainer, ServiceRegistry, TemplateContainer } from '../components';
 import { ASSISTANT_PLUGIN, ASSISTANT_DIALOG } from '../meta';
@@ -15,7 +17,7 @@ import { AIChatType, type AssistantSettingsProps, TemplateType } from '../types'
 export default () =>
   contributes(Capabilities.ReactSurface, [
     createSurface({
-      id: `${ASSISTANT_PLUGIN}/settings`,
+      id: `${ASSISTANT_PLUGIN}/plugin-settings`,
       role: 'article',
       filter: (data): data is { subject: SettingsStore<AssistantSettingsProps> } =>
         data.subject instanceof SettingsStore && data.subject.prefix === ASSISTANT_PLUGIN,
@@ -30,20 +32,36 @@ export default () =>
     createSurface({
       id: `${ASSISTANT_PLUGIN}/chat`,
       role: 'article',
-      filter: (data): data is { subject: AIChatType } => data.subject instanceof AIChatType,
+      filter: (data): data is { subject: AIChatType; variant: undefined } =>
+        isInstanceOf(AIChatType, data.subject) && data.variant !== 'assistant-chat',
       component: ({ data, role }) => <ChatContainer role={role} chat={data.subject} />,
     }),
+    createSurface({
+      id: `${ASSISTANT_PLUGIN}/object-chat`,
+      role: 'article',
+      filter: (data): data is { companionTo: AIChatType; subject: 'assistant-chat' } =>
+        isReactiveObject(data.companionTo) && data.companionTo.assistantChatQueue && data.subject === 'assistant-chat',
+      component: ({ data, role }) => {
+        const associatedArtifact = useMemo(
+          () => ({
+            id: fullyQualifiedId(data.companionTo),
+            typename: getTypename(data.companionTo) ?? 'unknown',
+          }),
+          [data.companionTo],
+        );
+        return <ChatContainer role={role} chat={data.companionTo} associatedArtifact={associatedArtifact} />;
+      },
+    }),
     createSurface({
       id: `${ASSISTANT_PLUGIN}/template`,
       role: 'article',
-      filter: (data): data is { subject: TemplateType } => data.subject instanceof TemplateType,
+      filter: (data): data is { subject: TemplateType } => isInstanceOf(TemplateType, data.subject),
       component: ({ data, role }) => <TemplateContainer role={role} template={data.subject} />,
     }),
     createSurface({
       id: `${ASSISTANT_PLUGIN}/service-registry`,
-      role: 'complementary--service-registry',
-      component: ({ data }) => (
-        <ServiceRegistry space={isSpace(data.subject) ? data.subject : getSpace(data.subject)!} />
-      ),
+      role: 'deck-companion--service-registry',
+      filter: (data): data is { subject: Space } => isSpace(data.subject),
+      component: ({ data }) => <ServiceRegistry space={data.subject} />,
     }),
   ]);

package/src/components/AmbientDialog/AmbientDialog.stories.tsx

@@ -65,7 +65,9 @@ const meta: Meta<typeof AmbientDialog> = {
       createIdentity: true,
       createSpace: true,
     }),
-    withPluginManager({ plugins: [IntentPlugin()] }),
+    withPluginManager({
+      plugins: [IntentPlugin()],
+    }),
     withTheme,
     withLayout({ fullscreen: true, tooltips: true }),
   ],

package/src/components/AmbientDialog/AmbientDialog.tsx

@@ -12,21 +12,21 @@ const preventDefault = (event: Event) => event.preventDefault();
 
 const minSize = 5;
 
-// TODO(burdon): Factor out.
-export const AmbientDialog = ({
-  children,
-  open: _open,
-  title,
-  onOpenChange,
-}: PropsWithChildren<{ open?: boolean; onOpenChange?: (open: boolean) => void; title?: string }>) => {
+export type AmbientDialogProps = PropsWithChildren<{
+  open?: boolean;
+  title?: string;
+  onOpenChange?: (open: boolean) => void;
+}>;
+
+export const AmbientDialog = ({ children, open: controlledOpen, title, onOpenChange }: AmbientDialogProps) => {
   const [resizeKey, setReizeKey] = useState(0);
   const [size, setSize] = useState<Size>('min-content');
-  const [open, setOpen] = useState(_open);
+  const [open, setOpen] = useState(controlledOpen);
 
   // Update controlled value.
   useEffect(() => {
-    setOpen(_open);
-  }, [_open]);
+    setOpen(controlledOpen);
+  }, [controlledOpen]);
 
   // Update size and key.
   useEffect(() => {
@@ -52,7 +52,7 @@ export const AmbientDialog = ({
         {...resizeAttributes}
         style={{
           ...sizeStyle(size, 'vertical', true),
-          maxBlockSize: 'calc(100dvh - env(safe-area-inset-bottom) - env(safe-area-inset-top) - 8rem)',
+          maxBlockSize: 'calc(100dvh - env(safe-area-inset-bottom) - env(safe-area-inset-top) - 9rem)',
         }}
         onInteractOutside={preventDefault}
       >

package/src/components/AssistantDialog.tsx

@@ -15,7 +15,7 @@ import { type AssistantSettingsProps, type AIChatType } from '../types';
 
 export const AssistantDialog: FC<{ chat?: AIChatType }> = ({ chat }) => {
   const { t } = useTranslation(ASSISTANT_PLUGIN);
-  const transcription = useCapabilities(TranscriptionCapabilities.Transcription).length > 0;
+  const transcription = useCapabilities(TranscriptionCapabilities.Transcriber).length > 0;
   const settings = useCapability(Capabilities.SettingsStore).getStore<AssistantSettingsProps>(ASSISTANT_PLUGIN)?.value;
 
   // TODO(burdon): Refocus when open.
@@ -23,7 +23,13 @@ export const AssistantDialog: FC<{ chat?: AIChatType }> = ({ chat }) => {
 
   return (
     <AmbientDialog open={open} onOpenChange={setOpen} title={t('assistant dialog title')}>
-      <ThreadContainer chat={chat} onOpenChange={setOpen} settings={settings} transcription={transcription} />
+      <ThreadContainer
+        chat={chat}
+        onOpenChange={setOpen}
+        settings={settings}
+        part={'dialog'}
+        transcription={transcription}
+      />
     </AmbientDialog>
   );
 };

package/src/components/AssistantSettings/AssistantSettings.tsx

@@ -4,13 +4,16 @@
 
 import React from 'react';
 
-import { DEFAULT_LLM_MODELS } from '@dxos/assistant';
+import { DEFAULT_EDGE_MODELS, DEFAULT_OLLAMA_MODELS } from '@dxos/assistant';
 import { Input, Select, useTranslation } from '@dxos/react-ui';
 import { DeprecatedFormContainer, DeprecatedFormInput } from '@dxos/react-ui-form';
 
 import { ASSISTANT_PLUGIN } from '../../meta';
 import { type AssistantSettingsProps } from '../../types';
 
+// TODO(burdon): Factor out.
+const DEFAULT_VALUE = '__default';
+
 export const AssistantSettings = ({ settings }: { settings: AssistantSettingsProps }) => {
   const { t } = useTranslation(ASSISTANT_PLUGIN);
 
@@ -23,18 +26,49 @@ export const AssistantSettings = ({ settings }: { settings: AssistantSettingsPro
         />
       </DeprecatedFormInput>
 
-      <DeprecatedFormInput label={t('settings llm model label')}>
+      <DeprecatedFormInput label={t('settings llm provider label')}>
+        <Input.Switch
+          checked={settings.llmProvider === 'ollama'}
+          onCheckedChange={(checked) => (settings.llmProvider = checked ? 'ollama' : 'edge')}
+        />
+      </DeprecatedFormInput>
+
+      <DeprecatedFormInput label={t('settings edge llm model label')}>
+        <Select.Root
+          value={settings.edgeModel ?? DEFAULT_VALUE}
+          onValueChange={(value) => {
+            settings.edgeModel = value === DEFAULT_VALUE ? undefined : value;
+          }}
+        >
+          <Select.TriggerButton placeholder={t('settings default llm model label')} />
+          <Select.Portal>
+            <Select.Content>
+              <Select.Viewport>
+                <Select.Option value={DEFAULT_VALUE}>{t('settings default label')}</Select.Option>
+                {DEFAULT_EDGE_MODELS.map((model) => (
+                  <Select.Option key={model} value={model}>
+                    {model}
+                  </Select.Option>
+                ))}
+              </Select.Viewport>
+            </Select.Content>
+          </Select.Portal>
+        </Select.Root>
+      </DeprecatedFormInput>
+
+      <DeprecatedFormInput label={t('settings ollama llm model label')}>
         <Select.Root
-          value={settings.llmModel ?? 'default'}
+          value={settings.ollamaModel ?? DEFAULT_VALUE}
           onValueChange={(value) => {
-            settings.llmModel = value;
+            settings.ollamaModel = value === DEFAULT_VALUE ? undefined : value;
           }}
         >
           <Select.TriggerButton placeholder={t('settings default llm model label')} />
           <Select.Portal>
             <Select.Content>
               <Select.Viewport>
-                {DEFAULT_LLM_MODELS.map((model) => (
+                <Select.Option value={DEFAULT_VALUE}>{t('settings default label')}</Select.Option>
+                {DEFAULT_OLLAMA_MODELS.map((model) => (
                   <Select.Option key={model} value={model}>
                     {model}
                   </Select.Option>

package/src/components/ChatContainer.tsx

@@ -5,6 +5,7 @@
 import React from 'react';
 
 import { Capabilities, useCapabilities, useCapability } from '@dxos/app-framework';
+import { type AssociatedArtifact } from '@dxos/artifact';
 import { TranscriptionCapabilities } from '@dxos/plugin-transcription';
 import { StackItem } from '@dxos/react-ui-stack';
 
@@ -13,13 +14,26 @@ import { ASSISTANT_PLUGIN } from '../meta';
 import { type AssistantSettingsProps, type AIChatType } from '../types';
 
 // TODO(burdon): Attention.
-export const ChatContainer = ({ chat, role }: { chat: AIChatType; role: string }) => {
-  const transcription = useCapabilities(TranscriptionCapabilities.Transcription).length > 0;
+export const ChatContainer = ({
+  role,
+  chat,
+  associatedArtifact,
+}: {
+  role: string;
+  chat: AIChatType;
+  associatedArtifact?: AssociatedArtifact;
+}) => {
+  const transcription = useCapabilities(TranscriptionCapabilities.Transcriber).length > 0;
   const settings = useCapability(Capabilities.SettingsStore).getStore<AssistantSettingsProps>(ASSISTANT_PLUGIN)?.value;
 
   return (
-    <StackItem.Content toolbar={false} role={role} classNames='mli-auto w-full max-w-[50rem]'>
-      <ThreadContainer chat={chat} settings={settings} transcription={transcription} />
+    <StackItem.Content role={role} classNames='container-max-width'>
+      <ThreadContainer
+        chat={chat}
+        settings={settings}
+        transcription={transcription}
+        associatedArtifact={associatedArtifact}
+      />
     </StackItem.Content>
   );
 };

package/src/components/Prompt/Prompt.stories.tsx

@@ -11,6 +11,7 @@ import { withTheme, withLayout } from '@dxos/storybook-utils';
 
 import { Prompt } from './Prompt';
 import { PromptBar } from './PromptBar';
+import type { ReferenceData } from './references';
 import translations from '../../translations';
 
 const meta: Meta<typeof Prompt> = {
@@ -77,3 +78,36 @@ export const Toolbar: Story = {
     );
   },
 };
+
+export const Includes: Story = {
+  args: {
+    classNames: 'w-96 p-4 rounded outline outline-gray-200',
+    references: {
+      getReferences: async ({ query }) => {
+        const res = references.filter((i) => i.label.toLowerCase().startsWith(query.toLowerCase()));
+        console.log('getReferences', { query, res });
+        return res;
+      },
+      resolveReference: async ({ uri }) => {
+        const res = references.find((i) => i.uri === uri);
+        console.log('resolveReference', { uri, res });
+        return res ?? null;
+      },
+    },
+  },
+};
+
+const references: ReferenceData[] = [
+  {
+    uri: 'dxn:echo:@:AAAAAAAA',
+    label: 'Meeting Notes',
+  },
+  {
+    uri: 'dxn:echo:@:BBBBBBBB',
+    label: 'Project Plan',
+  },
+  {
+    uri: 'dxn:echo:@:CCCCCCCC',
+    label: 'Meeting Plan',
+  },
+];