@dxos/plugin-assistant 0.8.4-main.9be5663bfe → 0.8.4-main.abd8ff62ef

package/src/containers/TracePanel/testing/snapshot-playback.ts

@@ -0,0 +1,45 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+import { type Dispatch, type SetStateAction, useEffect, useRef, useState } from 'react';
+
+/**
+ * Interval (ms) between auto-played trace messages in the snapshot playback story.
+ */
+export const PLAYBACK_INTERVAL_MS = 250;
+
+export const STEP_STORAGE_KEY = 'plugin-assistant.trace-panel.snapshot.step';
+
+/**
+ * `useState<number>` that hydrates from and persists to `localStorage`.
+ * The third tuple member is `true` iff a valid value was loaded on first render.
+ */
+export const useLocalStorageNumber = (
+  key: string,
+  initial: number,
+): [number, Dispatch<SetStateAction<number>>, boolean] => {
+  const initRef = useRef<{ value: number; hydrated: boolean } | null>(null);
+  if (initRef.current === null) {
+    let value = initial;
+    let hydrated = false;
+    if (typeof window !== 'undefined') {
+      const stored = window.localStorage.getItem(key);
+      const parsed = stored === null ? Number.NaN : Number(stored);
+      if (Number.isInteger(parsed) && parsed >= 0) {
+        value = parsed;
+        hydrated = true;
+      }
+    }
+    initRef.current = { value, hydrated };
+  }
+
+  const [value, setValue] = useState(initRef.current.value);
+  useEffect(() => {
+    if (typeof window !== 'undefined') {
+      window.localStorage.setItem(key, String(value));
+    }
+  }, [key, value]);
+
+  return [value, setValue, initRef.current.hydrated];
+};

package/src/containers/TriggerStatus/TriggerStatus.tsx

@@ -4,16 +4,16 @@
 
 import React, { useMemo } from 'react';
 
-import { useActiveSpace } from '@dxos/app-toolkit/ui';
+import { AppSurface } from '@dxos/app-toolkit/ui';
 import { type InvocationsState } from '@dxos/functions-runtime';
 import { useTriggerRuntimeControls } from '@dxos/plugin-automation/hooks';
-import { StatusBar } from '@dxos/plugin-status-bar';
-import { type Database } from '@dxos/react-client/echo';
-import { IconButton, Input, Popover, useTranslation } from '@dxos/react-ui';
+import { StatusBar } from '@dxos/plugin-status-bar/components';
+import { useObject } from '@dxos/react-client/echo';
+import { IconButton, Popover, useTranslation } from '@dxos/react-ui';
 
 import { meta } from '#meta';
 
-type TriggerStatusState = 'disabled' | 'idle' | 'running' | 'error';
+type TriggerStatusState = 'disabled' | 'idle' | 'running' | 'edge' | 'error';
 
 const getIcon = (state: TriggerStatusState): string => {
   switch (state) {
@@ -21,6 +21,8 @@ const getIcon = (state: TriggerStatusState): string => {
       return 'ph--lightning-slash--regular';
     case 'idle':
       return 'ph--lightning--regular';
+    case 'edge':
+      return 'ph--lightning--regular';
     case 'running':
       return 'ph--lightning--fill';
     case 'error':
@@ -39,24 +41,22 @@ const getIconClassNames = (state: TriggerStatusState): string | undefined => {
   }
 };
 
-export const TriggerStatus = () => {
-  const space = useActiveSpace();
-  const db = space?.db;
-  if (!db) {
-    return null;
-  }
-
-  return <SpaceStatusMain db={db} />;
-};
+export type SpaceStatusProps = AppSurface.SpaceArticleProps;
 
-const SpaceStatusMain = ({ db }: { db: Database.Database }) => {
+export const SpaceStatus = ({ space }: SpaceStatusProps) => {
   const { t } = useTranslation(meta.id);
-  const { state, start, stop } = useTriggerRuntimeControls(db);
-  const isRunning = state?.enabled ?? false;
+  const { state } = useTriggerRuntimeControls(space.db);
+  const isEnabled = state?.enabled ?? false;
+  const [properties] = useObject(space.properties);
+  const computeEnvironment = properties.computeEnvironment ?? 'local';
 
   // Determine the current trigger status state.
   const triggerState: TriggerStatusState = useMemo(() => {
-    if (!isRunning) {
+    if (computeEnvironment === 'edge') {
+      return 'edge';
+    }
+
+    if (!isEnabled) {
       return 'disabled';
     }
 
@@ -73,7 +73,7 @@ const SpaceStatusMain = ({ db }: { db: Database.Database }) => {
     }
 
     return 'idle';
-  }, [isRunning, state?.invocations]);
+  }, [computeEnvironment, isEnabled, state?.invocations]);
 
   return (
     <Popover.Root>
@@ -90,13 +90,11 @@ const SpaceStatusMain = ({ db }: { db: Database.Database }) => {
       <Popover.Portal>
         <Popover.Content side='left'>
           <TriggerStatusPopover
-            isRunning={isRunning}
             state={triggerState}
             currentFunctionName={
               state?.invocations.at(-1)?.function?.meta.name ?? state?.invocations.at(-1)?.function?.meta.key
             }
             lastInvocation={state?.invocations.at(-1)}
-            onToggle={isRunning ? stop : start}
           />
           <Popover.Arrow />
         </Popover.Content>
@@ -106,31 +104,20 @@ const SpaceStatusMain = ({ db }: { db: Database.Database }) => {
 };
 
 type TriggerStatusPopoverProps = {
-  isRunning: boolean;
   state: TriggerStatusState;
   currentFunctionName?: string;
   lastInvocation?: InvocationsState;
-  onToggle: () => void;
 };
 
 const TriggerStatusPopover = ({
-  isRunning,
   state,
   currentFunctionName,
   lastInvocation, // TODO(burdon): Show.
-  onToggle,
 }: TriggerStatusPopoverProps) => {
   const { t } = useTranslation(meta.id);
 
   return (
     <div className='flex flex-col gap-2 p-2 w-[240px]'>
-      <Input.Root>
-        <div className='flex items-center gap-2'>
-          <Input.Switch checked={isRunning} onCheckedChange={onToggle} />
-          <Input.Label>{t('trigger-runtime.label')}</Input.Label>
-        </div>
-      </Input.Root>
-
       <div className='flex flex-col gap-1'>
         <div className='text-sm'>{t(`trigger-status-${state}.label`)}</div>
         {currentFunctionName && state === 'running' && (
@@ -140,5 +127,3 @@ const TriggerStatusPopover = ({
     </div>
   );
 };
-
-export default TriggerStatus;

package/src/containers/TriggerStatus/index.ts

@@ -2,4 +2,4 @@
 // Copyright 2025 DXOS.org
 //
 
-export { TriggerStatus as default } from './TriggerStatus';
+export { SpaceStatus, SpaceStatus as default } from './TriggerStatus';

package/src/containers/index.ts

@@ -10,7 +10,7 @@ export const ChatContainer: ComponentType<any> = lazy(() => import('./ChatContai
 export const ChatDialog: ComponentType<any> = lazy(() => import('./ChatDialog'));
 export const AgentArticle: ComponentType<any> = lazy(() => import('./AgentArticle'));
 export const AgentProperties: ComponentType<any> = lazy(() => import('./AgentProperties'));
-export const PromptArticle: ComponentType<any> = lazy(() => import('./PromptArticle'));
-export const PromptList: ComponentType<any> = lazy(() => import('./PromptList'));
+export const RoutineArticle: ComponentType<any> = lazy(() => import('./RoutineArticle'));
+export const RoutineList: ComponentType<any> = lazy(() => import('./RoutineList'));
 export const TracePanel: ComponentType<any> = lazy(() => import('./TracePanel'));
 export const TriggerStatus: ComponentType<any> = lazy(() => import('./TriggerStatus'));

package/src/hooks/index.ts

@@ -3,11 +3,13 @@
 //
 
 export * from './useBlueprintRegistry';
+export * from './useChatKeymap';
 export * from './useChatProcessor';
 export * from './useChatServices';
 export * from './useChatToolbarActions';
 export * from './useContextBinder';
 export * from './useContextObjects';
+export { useDebug } from './useDebug';
 export * from './useFilteredTypes';
 export * from './useFlush';
 export * from './useOnline';

package/src/hooks/useBlueprintRegistry.ts

@@ -7,7 +7,7 @@ import { useCallback, useEffect, useMemo, useState } from 'react';
 import { useCapabilities } from '@dxos/app-framework/ui';
 import { AppCapabilities } from '@dxos/app-toolkit';
 import { type AiContextBinder } from '@dxos/assistant';
-import { Blueprint } from '@dxos/blueprints';
+import { Blueprint } from '@dxos/compute';
 import { type Database, Filter, Obj, Ref } from '@dxos/echo';
 import { useQuery } from '@dxos/react-client/echo';
 import { distinctBy } from '@dxos/util';

package/src/hooks/useChatKeymap.ts

@@ -0,0 +1,52 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import { type Extension, Prec } from '@codemirror/state';
+import { keymap } from '@codemirror/view';
+import { useMemo } from 'react';
+
+import { type Event } from '@dxos/async';
+import { isTruthy } from '@dxos/util';
+
+import { type ChatEvent } from '../components/Chat/events';
+
+/**
+ * CodeMirror keymap shared by the chat document (Thread) and the prompt editor — pressing
+ * Mod-Arrow keys when either editor is focused emits the corresponding `ChatEvent` on the
+ * shared event bus.
+ */
+export const useChatKeymapExtensions = ({ event }: { event: Event<ChatEvent> }): Extension[] => {
+  return useMemo<Extension[]>(() => {
+    return [
+      Prec.highest(
+        keymap.of([
+          {
+            key: 'Mod-ArrowUp',
+            preventDefault: true,
+            run: () => {
+              event.emit({ type: 'nav-previous' });
+              return true;
+            },
+            shift: () => {
+              event.emit({ type: 'thread-open' });
+              return true;
+            },
+          },
+          {
+            key: 'Mod-ArrowDown',
+            preventDefault: true,
+            run: () => {
+              event.emit({ type: 'nav-next' });
+              return true;
+            },
+            shift: () => {
+              event.emit({ type: 'thread-close' });
+              return true;
+            },
+          },
+        ]),
+      ),
+    ].filter(isTruthy);
+  }, [event]);
+};

package/src/hooks/useChatProcessor.ts

@@ -6,9 +6,9 @@ import { type Registry, RegistryContext } from '@effect-atom/atom-react';
 import * as Effect from 'effect/Effect';
 import { useContext, useMemo, useState } from 'react';
 
-import { AiConversation } from '@dxos/assistant';
+import { AiSession } from '@dxos/assistant';
 import { type Chat } from '@dxos/assistant-toolkit';
-import { type Blueprint } from '@dxos/blueprints';
+import { type Blueprint } from '@dxos/compute';
 import { Feed, Ref } from '@dxos/echo';
 import { createFeedServiceLayer } from '@dxos/echo-db';
 import { runAndForwardErrors } from '@dxos/effect';
@@ -43,7 +43,7 @@ export const useChatProcessor = ({
 }: UseChatProcessorProps): AiChatProcessor | undefined => {
   const observableRegistry = useContext(RegistryContext);
 
-  const [conversation, setConversation] = useState<AiConversation>();
+  const [session, setSession] = useState<AiSession>();
   useAsyncEffect(async () => {
     if (!space || !chat) {
       return;
@@ -57,34 +57,34 @@ export const useChatProcessor = ({
     const runtime = await runAndForwardErrors(
       Effect.runtime<Feed.FeedService>().pipe(Effect.provide(feedServiceLayer)),
     );
-    const conversation = new AiConversation({
+    const session = new AiSession({
       feed: feedTarget,
       runtime,
       registry: observableRegistry as Registry.Registry,
     });
-    await conversation.open();
-    setConversation(conversation);
+    await session.open();
+    setSession(session);
     return () => {
-      void conversation.close();
-      setConversation(undefined);
+      void session.close();
+      setSession(undefined);
     };
   }, [space, chat?.feed.target]);
 
   const feed = chat?.feed.target;
 
   const processor = useMemo(() => {
-    if (!runtime || !conversation || !chat || !feed) {
+    if (!runtime || !session || !chat || !feed) {
       return undefined;
     }
 
     log('creating processor', { preset, model: preset?.model, settings });
-    return new AiChatProcessor(conversation, runtime, feed, {
+    return new AiChatProcessor(session, runtime, feed, {
       chat: chat ? Ref.make(chat) : undefined,
       observableRegistry,
       blueprintRegistry,
       model: preset?.model,
     });
-  }, [runtime, conversation, blueprintRegistry, preset, feed]);
+  }, [runtime, session, blueprintRegistry, preset, feed]);
 
   return processor;
 };

package/src/hooks/useDebug.ts

@@ -0,0 +1,38 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import { useCallback } from 'react';
+
+import { trim } from '@dxos/util';
+
+import { type AiChatProcessor } from '../processor';
+
+/**
+ * Returns an async callback that logs the chat's current context, system prompt,
+ * and resolved tools to the browser console under a collapsible group.
+ * Used by the chat's debug-toggle event handler.
+ */
+export const useDebug = ({ processor }: { processor: AiChatProcessor }) => {
+  return useCallback(async () => {
+    const objects = processor.context.getObjects();
+    const blueprints = processor.context.getBlueprints();
+    const system = await processor.getSystemPrompt();
+    const tools = (await processor.getTools()) ?? {};
+    console.group('Chat', { objects, blueprints });
+    try {
+      console.log(trim`
+        System Prompt:
+        ${system}
+      `);
+      console.log(trim`
+        Tools:
+        ${Object.values(tools)
+          .map((tool) => JSON.stringify(tool, null, 2))
+          .join('\n')}
+      `);
+    } finally {
+      console.groupEnd();
+    }
+  }, [processor]);
+};

package/src/index.ts

@@ -2,11 +2,13 @@
 // Copyright 2023 DXOS.org
 //
 
-export * from './blueprints';
-export * from './components';
-export * from './hooks';
+import { Plugin } from '@dxos/app-framework';
+
+import { meta } from './meta';
+
+export const AssistantPlugin = Plugin.lazy(meta, () => import('#plugin'));
+
 export * from './meta';
-export * from './translations';
-export * from './types';
 
-export * from './AssistantPlugin';
+export * from './components';
+export * from './hooks';

package/src/operations/create-chat.ts

@@ -7,11 +7,10 @@ import * as Effect from 'effect/Effect';
 import { Capabilities, Capability } from '@dxos/app-framework';
 import { AiContextBinder } from '@dxos/assistant';
 import { Chat, DatabaseBlueprint, AgentWizardBlueprint } from '@dxos/assistant-toolkit';
-import { Blueprint } from '@dxos/blueprints';
+import { Blueprint, Operation } from '@dxos/compute';
 import { Feed, Filter, Obj, Ref } from '@dxos/echo';
 import { createFeedServiceLayer } from '@dxos/echo-db';
 import { invariant } from '@dxos/invariant';
-import { Operation } from '@dxos/operation';
 import { ClientCapabilities } from '@dxos/plugin-client/types';
 
 import { AssistantBlueprint } from '#blueprints';

package/src/operations/definitions.ts

@@ -7,10 +7,9 @@ import * as Schema from 'effect/Schema';
 import { AiService } from '@dxos/ai';
 import { Capability } from '@dxos/app-framework';
 import { Chat } from '@dxos/assistant-toolkit';
-import { Prompt } from '@dxos/blueprints';
 import { SpaceSchema } from '@dxos/client/echo';
+import { Routine, Operation } from '@dxos/compute';
 import { Collection, Database, DXN, Feed, Obj, Ref } from '@dxos/echo';
-import { Operation } from '@dxos/operation';
 
 import { meta } from '#meta';
 
@@ -68,8 +67,8 @@ export const RunPromptInNewChat = Operation.make({
     objects: Schema.optional(Schema.Array(Obj.Unknown)),
     /** Blueprint keys to look up and bind to the new chat. */
     blueprints: Schema.optional(Schema.Array(Schema.String)),
-    /** Raw instructions or an existing Prompt object reference (e.g. from the prompts surface). */
-    prompt: Schema.Union(Schema.String, Ref.Ref(Prompt.Prompt)),
+    /** Raw instructions or an existing Routine object reference. */
+    prompt: Schema.Union(Schema.String, Ref.Ref(Routine.Routine)),
     /**
      * When true, skips opening the chat: runs the Agent prompt operation against the new chat via the compute runtime (traced).
      */

package/src/operations/ensure-companion-chat.ts

@@ -6,8 +6,8 @@ import * as Effect from 'effect/Effect';
 
 import { Capabilities, Capability } from '@dxos/app-framework';
 import { Chat } from '@dxos/assistant-toolkit';
+import { Operation } from '@dxos/compute';
 import { Filter, Obj, Query } from '@dxos/echo';
-import { Operation } from '@dxos/operation';
 
 import { AssistantCapabilities } from '../types';
 import { CreateChat, EnsureCompanionChat, SetCurrentChat } from './definitions';

package/src/operations/index.ts

@@ -2,7 +2,7 @@
 // Copyright 2025 DXOS.org
 //
 
-import { OperationHandlerSet } from '@dxos/operation';
+import { OperationHandlerSet } from '@dxos/compute';
 
 export * as AssistantOperation from './definitions';
 

package/src/operations/on-create-space.ts

@@ -5,7 +5,7 @@
 import * as Effect from 'effect/Effect';
 
 import { AgentPrompt } from '@dxos/assistant-toolkit';
-import { Operation } from '@dxos/operation';
+import { Operation } from '@dxos/compute';
 
 import { CreateChat, OnCreateSpace } from './definitions';
 

package/src/operations/prompt.test.ts

@@ -0,0 +1,77 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+import * as Array from 'effect/Array';
+import * as Effect from 'effect/Effect';
+import { describe, test } from 'vitest';
+
+import { AgentPrompt, Chat } from '@dxos/assistant-toolkit';
+import { Routine, Operation } from '@dxos/compute';
+import { Database, Feed, Filter, Ref } from '@dxos/echo';
+import { runAndForwardErrors } from '@dxos/effect';
+import { ObjectId } from '@dxos/keys';
+import { AutomationPlugin } from '@dxos/plugin-automation';
+import { AutomationCapabilities } from '@dxos/plugin-automation/types';
+import { ClientPlugin } from '@dxos/plugin-client';
+import { initializeIdentity } from '@dxos/plugin-client/testing';
+import { ClientCapabilities } from '@dxos/plugin-client/types';
+import { createComposerTestApp } from '@dxos/plugin-testing/harness';
+import { Message } from '@dxos/types';
+
+import { AssistantPlugin } from '../cli';
+
+ObjectId.dangerouslyDisableRandomness();
+
+describe('Agent prompt (composer plugin harness)', () => {
+  // Hits AutomationPlugin compute runtime (plugin handlers, AiServiceLayer, blueprints).
+  // Requires reachable edge AI (see repo DX_EDGE_AI_SERVICE_URL); not memoized like AssistantTestLayer tests.
+  test(
+    'chat mode appends assistant messages to the chat queue',
+    { tags: ['llm'], timeout: 60_000 },
+    async ({ expect }) => {
+      await using harness = await createComposerTestApp({
+        plugins: [ClientPlugin({}), AssistantPlugin(), AutomationPlugin()],
+      });
+
+      const { personalSpace } = await runAndForwardErrors(initializeIdentity(harness.get(ClientCapabilities.Client)));
+      const computeProvider = await harness.waitForCapability(AutomationCapabilities.ComputeRuntime, {
+        timeout: 30_000,
+      });
+
+      await computeProvider.getRuntime(personalSpace.id).runPromise(
+        Effect.gen(function* () {
+          const feed = yield* Database.add(Feed.make());
+
+          const messageCountBefore = yield* Feed.runQuery(feed, Filter.type(Message.Message)).pipe(
+            Effect.map(Array.length),
+          );
+
+          const chat = yield* Database.add(Chat.make({ feed: Ref.make(feed) }));
+          const prompt = yield* Database.add(
+            Routine.make({
+              name: 'chat-mode-test',
+              instructions: 'Reply with a single word: ack.',
+              blueprints: [],
+              context: [],
+            }),
+          );
+          yield* Database.flush();
+
+          const result = yield* Operation.invoke(AgentPrompt, {
+            prompt: Ref.make(prompt),
+            input: {},
+            chat: Ref.make(chat),
+          });
+
+          const messageCountAfter = yield* Feed.runQuery(feed, Filter.type(Message.Message)).pipe(
+            Effect.map(Array.length),
+          );
+
+          expect(messageCountAfter).toBeGreaterThan(messageCountBefore);
+          expect(result).toBe('ack');
+        }),
+      );
+    },
+  );
+});

package/src/operations/resolve-navigation-targets.ts

@@ -6,7 +6,7 @@ import * as Effect from 'effect/Effect';
 
 import { Capability } from '@dxos/app-framework';
 import { AppCapabilities } from '@dxos/app-toolkit';
-import { Operation } from '@dxos/operation';
+import { Operation } from '@dxos/compute';
 
 import { ResolveNavigationTargets } from './definitions';
 

package/src/operations/run-prompt-in-new-chat.ts

@@ -8,12 +8,11 @@ import { Capabilities, Capability } from '@dxos/app-framework';
 import { getObjectPathFromObject, LayoutOperation } from '@dxos/app-toolkit';
 import { AiContextBinder } from '@dxos/assistant';
 import { AgentPrompt } from '@dxos/assistant-toolkit';
-import { Blueprint, Prompt, Template } from '@dxos/blueprints';
+import { Blueprint, Routine, Template, Operation } from '@dxos/compute';
 import { Database, Feed, Filter, Obj, Ref } from '@dxos/echo';
 import { createFeedServiceLayer } from '@dxos/echo-db';
 import { invariant } from '@dxos/invariant';
 import { log } from '@dxos/log';
-import { Operation } from '@dxos/operation';
 import { AutomationCapabilities } from '@dxos/plugin-automation/types';
 import { ClientCapabilities } from '@dxos/plugin-client/types';
 import { Text } from '@dxos/schema';
@@ -64,7 +63,7 @@ const handler: Operation.WithHandler<typeof RunPromptInNewChat> = RunPromptInNew
           const promptRef =
             typeof prompt === 'string'
               ? Ref.make(
-                  Prompt.make({
+                  Routine.make({
                     instructions: prompt,
                     blueprints: [],
                     context: [],

package/src/operations/set-current-chat.ts

@@ -5,8 +5,8 @@
 import * as Effect from 'effect/Effect';
 
 import { Capabilities, Capability } from '@dxos/app-framework';
+import { Operation } from '@dxos/compute';
 import { Obj } from '@dxos/echo';
-import { Operation } from '@dxos/operation';
 
 import { AssistantCapabilities } from '../types';
 import { CreateChat, SetCurrentChat } from './definitions';

package/src/operations/update-chat-name.ts

@@ -7,9 +7,9 @@ import * as Prompt from '@effect/ai/Prompt';
 import * as Effect from 'effect/Effect';
 
 import { AiPreprocessor, AiService } from '@dxos/ai';
+import { Operation } from '@dxos/compute';
 import { Database, Feed, Filter, Obj } from '@dxos/echo';
 import { log } from '@dxos/log';
-import { Operation } from '@dxos/operation';
 import { Message } from '@dxos/types';
 import { trim } from '@dxos/util';
 
@@ -48,7 +48,7 @@ const handler: Operation.WithHandler<typeof UpdateChatName> = UpdateChatName.pip
 
         const newName = response.text.replaceAll(/[^a-zA-Z0-9\s]/g, '').trim();
 
-        Obj.change(chat, (chat) => {
+        Obj.update(chat, (chat) => {
           chat.name = newName;
         });
         log.info('chat name updated', { chat, newName: chat.name });

package/src/processor/presets.ts

@@ -6,7 +6,7 @@ import * as Schema from 'effect/Schema';
 
 import { type ModelName } from '@dxos/ai';
 
-const ModelProviders = ['dxos-local', 'dxos-remote', 'lm-studio'] as const;
+const ModelProviders = ['dxos-local', 'dxos-remote', 'lm-studio', 'ollama'] as const;
 
 const ModelProvider = Schema.Literal(...ModelProviders);
 type ModelProvider = Schema.Schema.Type<typeof ModelProvider>;
@@ -37,6 +37,14 @@ export const AiServicePresets: AiServicePreset[] = [
     provider: 'dxos-remote' as const,
     model: '@anthropic/claude-3-5-haiku-20241022' as const,
   },
+  {
+    provider: 'ollama' as const,
+    model: 'gpt-oss:20b' as const,
+  },
+  {
+    provider: 'ollama' as const,
+    model: 'gemma4:latest' as const,
+  },
   {
     provider: 'lm-studio' as const,
     model: '@google/gemma-3-27b' as const,

package/src/processor/processor.test.ts

@@ -6,12 +6,12 @@ import { describe, it } from '@effect/vitest';
 import * as Effect from 'effect/Effect';
 import * as ManagedRuntime from 'effect/ManagedRuntime';
 
-import { AiConversation } from '@dxos/assistant';
+import { AiSession } from '@dxos/assistant';
 import { Chat } from '@dxos/assistant-toolkit';
-import { AssistantTestLayer } from '@dxos/assistant/testing';
 import { Database, Feed } from '@dxos/echo';
 import { acquireReleaseResource } from '@dxos/effect';
 import { TestHelpers } from '@dxos/effect/testing';
+import { AssistantTestLayer } from '@dxos/functions-runtime/testing';
 
 import { AiChatProcessor } from './processor';
 
@@ -25,11 +25,11 @@ describe('Chat processor', () => {
         const feed = Feed.make();
         yield* Database.add(feed);
         const runtime = yield* Effect.runtime<Feed.FeedService>();
-        const conversation = yield* acquireReleaseResource(() => new AiConversation({ feed, runtime }));
+        const session = yield* acquireReleaseResource(() => new AiSession({ feed, runtime }));
         const managedRuntime = ManagedRuntime.make(
           Effect.runSync(Effect.map(Effect.context<never>(), () => undefined as any)) as any,
         );
-        const processor = new AiChatProcessor(conversation, managedRuntime as any, feed);
+        const processor = new AiChatProcessor(session, managedRuntime as any, feed);
         expect(processor).toBeDefined();
         expect(processor.active).toBeDefined();
       },