@dxos/agent-runtime 0.14.0 → 0.16.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dxos/agent-runtime",
-  "version": "0.14.0",
+  "version": "0.16.0",
   "description": "Local agent process runtime (agent-service) and assistant-coupled test helpers.",
   "homepage": "https://dxos.org",
   "bugs": "https://github.com/dxos/dxos/issues",
@@ -38,30 +38,30 @@
     "src"
   ],
   "dependencies": {
-    "effect": "4.0.0",
-    "@dxos/ai": "0.14.0",
-    "@dxos/compute": "0.14.0",
-    "@dxos/compute-runtime": "0.14.0",
-    "@dxos/echo": "0.14.0",
-    "@dxos/echo-client": "0.14.0",
-    "@dxos/effect": "0.14.0",
-    "@dxos/keys": "0.14.0",
-    "@dxos/link": "0.14.0",
-    "@dxos/log": "0.14.0",
-    "@dxos/mcp-client": "0.14.0",
-    "@dxos/schema": "0.14.0",
-    "@dxos/types": "0.14.0",
-    "@dxos/util": "0.14.0"
+    "effect": "4.0.4",
+    "@dxos/ai": "0.16.0",
+    "@dxos/compute": "0.16.0",
+    "@dxos/compute-runtime": "0.16.0",
+    "@dxos/echo": "0.16.0",
+    "@dxos/echo-client": "0.16.0",
+    "@dxos/effect": "0.16.0",
+    "@dxos/keys": "0.16.0",
+    "@dxos/link": "0.16.0",
+    "@dxos/log": "0.16.0",
+    "@dxos/mcp-client": "0.16.0",
+    "@dxos/schema": "0.16.0",
+    "@dxos/types": "0.16.0",
+    "@dxos/util": "0.16.0"
   },
   "devDependencies": {
-    "@effect/vitest": "4.0.0",
+    "@effect/vitest": "4.0.4",
     "vitest": "4.1.10",
-    "@dxos/assistant": "0.14.0",
-    "@dxos/errors": "0.14.0"
+    "@dxos/assistant": "0.16.0",
+    "@dxos/errors": "0.16.0"
   },
   "peerDependencies": {
-    "effect": "4.0.0",
-    "@dxos/assistant": "^0.14.0"
+    "effect": "4.0.4",
+    "@dxos/assistant": "^0.16.0"
   },
   "publishConfig": {
     "access": "public"

package/src/agent-service/AgentService.test.ts

@@ -891,6 +891,47 @@ describe('Agent Service (control plane)', () => {
       TestHelpers.provideTestContext,
     ),
   );
+  it.effect(
+    'session reuse tracks the harness selected on the chat',
+    Effect.fnUntraced(
+      function* (_) {
+        const processManager = yield* ProcessManager.ProcessManagerService;
+
+        const feed = yield* Database.add(Feed.make());
+        const chat = yield* Database.add(Chat.make({ feed: Ref.make(feed) }));
+        yield* Database.flush();
+        const target = Obj.getURI(chat);
+
+        const isActive = (state: Process.State) =>
+          state !== Process.State.SUCCEEDED && state !== Process.State.FAILED && state !== Process.State.TERMINATED;
+        const activePids = Effect.gen(function* () {
+          const processes = yield* processManager.list({ target, key: AGENT_PROCESS_KEY });
+          return processes.filter((process) => isActive(process.status.state)).map((process) => String(process.pid));
+        });
+
+        // Opening a chat spawns its process on the default harness before the user picks one.
+        const sessionA = yield* ComputeAgentService.getSession(chat);
+        const [pidA] = yield* activePids;
+
+        // The turn producer is chosen when the process starts, so picking a harness must respawn it.
+        Obj.update(chat, (chat) => {
+          chat.session = { harness: 'claude-code-edge' };
+        });
+        yield* Database.flush();
+        const sessionB = yield* ComputeAgentService.getSession(chat);
+        const [pidB] = yield* activePids;
+        expect(sessionB).not.toBe(sessionA);
+        expect(pidB).not.toBe(pidA);
+        expect(yield* activePids).toEqual([pidB]);
+
+        expect(yield* ComputeAgentService.getSession(chat)).toBe(sessionB);
+
+        yield* sessionB.terminate();
+      },
+      Effect.provide(TestLayer()),
+      TestHelpers.provideTestContext,
+    ),
+  );
   it.effect(
     'traces a turn run through the agent process',
     Effect.fnUntraced(

package/src/agent-service/AgentService.ts

@@ -10,6 +10,7 @@ import * as Layer from 'effect/Layer';
 import * as Atom from 'effect/reactivity/Atom';
 import * as Semaphore from 'effect/Semaphore';
 
+import { SessionConfig } from '@dxos/ai';
 import { AiContext, type HarnessControlRpcs } from '@dxos/assistant';
 import * as Chat from '@dxos/assistant/Chat';
 import * as AgentService from '@dxos/compute/AgentService';
@@ -145,13 +146,14 @@ export const layer = (opts?: Options): Layer.Layer<AgentService.AgentService, ne
         return { kind: 'edge', space: spaceId };
       };
 
-      // The agent's model and steering instructions are bound to its process at spawn time, so the
-      // cache tracks what each session was created with. A chat repointed at a different model or
-      // instructions ref tears down the old process and spawns a fresh one (see below).
+      // The agent's model, harness and steering instructions are bound to its process at spawn time, so
+      // the cache tracks what each session was created with. A chat repointed at a different model,
+      // harness or instructions ref tears down the old process and spawns a fresh one (see below).
       const sessionCache = new Map<
         string,
         {
           model: string | undefined;
+          harness: string;
           provider: DXN.DXN | undefined;
           instructions: string | undefined;
           location: AgentService.AgentLocation;
@@ -249,6 +251,7 @@ export const layer = (opts?: Options): Layer.Layer<AgentService.AgentService, ne
                 // Read off the chat rather than passed in: the process is bound to the chat, so its
                 // model and steering are whatever the chat points at when the process is spawned.
                 const model = chat.session?.model;
+                const harness = SessionConfig.harnessOf(chat.session);
                 const instructions = chat.instructions?.uri;
                 const location: AgentService.AgentLocation = options?.location ?? 'local';
                 const executable = executableFor(chat, provider);
@@ -262,6 +265,7 @@ export const layer = (opts?: Options): Layer.Layer<AgentService.AgentService, ne
                 if (cached) {
                   if (
                     cached.model === model &&
+                    cached.harness === harness &&
                     cached.provider === provider &&
                     cached.instructions === instructions &&
                     cached.location === location &&
@@ -272,7 +276,7 @@ export const layer = (opts?: Options): Layer.Layer<AgentService.AgentService, ne
                   }
 
                   if (!Process.isTerminal(cached.handle.status.state)) {
-                    // Model, provider, steering instructions, location or process changed (e.g. the user
+                    // Model, harness, provider, steering instructions, location or process changed (e.g. the user
                     // toggled online/offline, or moved the chat to the cloud): terminate the
                     // existing process so the conversation continues on a fresh process bound to the new
                     // configuration. Conversation history is preserved via the feed, which the new
@@ -363,6 +367,7 @@ export const layer = (opts?: Options): Layer.Layer<AgentService.AgentService, ne
                 const session = makeSession(handle, chat, feed, releaseSession, isFinished, resubmit);
                 sessionCache.set(chat.id, {
                   model,
+                  harness,
                   provider,
                   instructions,
                   location,