@dxos/assistant 0.8.4-main.3eb6e50203 → 0.8.4-main.3fbcb4aa9b

package/dist/lib/neutral/index.mjs

@@ -1,48 +1,29 @@
-import {
-  AiAssistantError,
-  AiModelError,
-  AiSession,
-  ArtifactDiffResolver,
-  ArtifactId,
-  GenerationObserver,
-  GenericToolkit_exports,
-  RefFromLLM,
-  ToolExecutionServices,
-  VersionPin,
-  createArtifactElement,
-  createToolkit,
-  formatSystemPrompt,
-  formatUserPrompt,
-  makeToolExecutionServiceFromFunctions,
-  makeToolResolverFromFunctions
-} from "./chunk-QBD6NQJK.mjs";
-import "./chunk-J5LGTIGS.mjs";
-
 // src/conversation/context.ts
 import { Atom, Registry } from "@effect-atom/atom-react";
 import * as EArray from "effect/Array";
 import * as Context from "effect/Context";
 import * as Effect from "effect/Effect";
 import * as Function from "effect/Function";
+import * as Runtime from "effect/Runtime";
 import * as Schema from "effect/Schema";
-import { Blueprint } from "@dxos/blueprints";
+import { Blueprint } from "@dxos/compute";
 import { Resource } from "@dxos/context";
-import { DXN, Obj, Query, Type } from "@dxos/echo";
+import { DXN, Feed, Obj, Query, Ref, Type } from "@dxos/echo";
 import { assertArgument } from "@dxos/invariant";
 import { log } from "@dxos/log";
-import { ComplexSet, isTruthy } from "@dxos/util";
-var __dxlog_file = "/__w/dxos/dxos/packages/core/assistant/src/conversation/context.ts";
+import { ComplexSet, isNonNullable } from "@dxos/util";
+var __dxlog_file = "/__w/dxos/dxos/packages/core/compute/assistant/src/conversation/context.ts";
 var ContextBinding = Schema.Struct({
   blueprints: Schema.Struct({
-    added: Schema.Array(Type.Ref(Blueprint.Blueprint)),
-    removed: Schema.Array(Type.Ref(Blueprint.Blueprint))
+    added: Schema.Array(Ref.Ref(Blueprint.Blueprint)),
+    removed: Schema.Array(Ref.Ref(Blueprint.Blueprint))
   }),
   objects: Schema.Struct({
-    added: Schema.Array(Type.Ref(Type.Obj)),
-    removed: Schema.Array(Type.Ref(Type.Obj))
+    added: Schema.Array(Ref.Ref(Obj.Unknown)),
+    removed: Schema.Array(Ref.Ref(Obj.Unknown))
   })
 }).pipe(Type.object({
-  typename: "dxos.org/type/ContextBinding",
+  typename: "org.dxos.type.contextBinding",
   version: "0.1.0"
 }));
 var Bindings = class {
@@ -60,11 +41,15 @@ var AiContextBinder = class extends Resource {
   _blueprints = Atom.make([]).pipe(Atom.keepAlive);
   _objects = Atom.make([]).pipe(Atom.keepAlive);
   _registry;
-  _queue;
+  _feed;
+  _runtime;
+  #bindingsQuery;
   constructor(options) {
     super();
-    assertArgument(options.queue, "options.queue", "Queue is required");
-    this._queue = options.queue;
+    assertArgument(options.feed, "options.feed", "Feed is required");
+    assertArgument(options.runtime, "options.runtime", "Feed runtime is required");
+    this._feed = options.feed;
+    this._runtime = options.runtime;
     this._registry = options.registry ?? Registry.make();
   }
   /**
@@ -104,33 +89,67 @@ var AiContextBinder = class extends Resource {
     return this._registry.subscribe(this._objects, () => cb(this._registry.get(this._objects)));
   }
   async _open() {
-    const query = this._queue.query(Query.type(ContextBinding));
-    const initialResults = await query.run();
+    this.#bindingsQuery = await Runtime.runPromise(this._runtime)(Feed.query(this._feed, Query.type(ContextBinding)));
+    const initialResults = await this.#bindingsQuery.run();
     await this._updateBindings(initialResults);
-    this._ctx.onDispose(query.subscribe(async () => {
-      await this._updateBindings(query.results);
+    this._ctx.onDispose(this.#bindingsQuery.subscribe(async () => {
+      await this._updateBindings(this.#bindingsQuery.results);
     }));
   }
+  /**
+  * Re-reads bindings from the feed to pick up changes made by other processes.
+  */
+  async sync() {
+    if (this.#bindingsQuery) {
+      const results = await this.#bindingsQuery.run();
+      log("sync", {
+        bindingItems: results.length
+      }, { "~LogMeta": "~LogMeta", F: __dxlog_file, L: 107, S: this });
+      await this._updateBindings(results);
+      log("sync complete", {
+        blueprints: this._registry.get(this._blueprints).length,
+        blueprintKeys: this._registry.get(this._blueprints).map((bp) => bp.key)
+      }, { "~LogMeta": "~LogMeta", F: __dxlog_file, L: 111, S: this });
+    }
+  }
   async _updateBindings(items) {
     if (items.length === 0) {
       return;
     }
     const bindings = this._reduce(items);
+    log("_updateBindings", {
+      items: items.length,
+      blueprintRefs: [
+        ...bindings.blueprints
+      ].map((ref) => ({
+        dxn: ref.dxn.toString(),
+        available: ref.isAvailable
+      }))
+    }, { "~LogMeta": "~LogMeta", F: __dxlog_file, L: 123, S: this });
     const currentBlueprints = this._registry.get(this._blueprints);
     const currentObjects = this._registry.get(this._objects);
-    const newBlueprints = await this._resolve(bindings.blueprints, currentBlueprints);
-    const newObjects = await this._resolve(bindings.objects, currentObjects);
-    this._registry.set(this._blueprints, newBlueprints);
-    this._registry.set(this._objects, newObjects);
+    const resolvedBlueprints = await this._resolve(bindings.blueprints, currentBlueprints);
+    const resolvedObjects = await this._resolve(bindings.objects, currentObjects);
+    log("_updateBindings resolved", {
+      resolvedBlueprints: resolvedBlueprints.length,
+      resolvedBlueprintKeys: resolvedBlueprints.map((bp) => bp.key)
+    }, { "~LogMeta": "~LogMeta", F: __dxlog_file, L: 137, S: this });
+    const reducedBlueprintDxns = new ComplexSet(DXN.hash, [
+      ...bindings.blueprints
+    ].map((ref) => ref.dxn));
+    const reducedObjectDxns = new ComplexSet(DXN.hash, [
+      ...bindings.objects
+    ].map((ref) => ref.dxn));
+    const filteredBlueprints = currentBlueprints.filter((obj) => reducedBlueprintDxns.has(Obj.getDXN(obj)));
+    const filteredObjects = currentObjects.filter((obj) => reducedObjectDxns.has(Obj.getDXN(obj)));
+    const mergedBlueprints = this._mergeInto(filteredBlueprints, resolvedBlueprints);
+    const mergedObjects = this._mergeInto(filteredObjects, resolvedObjects);
+    this._registry.set(this._blueprints, mergedBlueprints);
+    this._registry.set(this._objects, mergedObjects);
     log("updated bindings", {
-      blueprints: newBlueprints.length,
-      objects: newObjects.length
-    }, {
-      F: __dxlog_file,
-      L: 159,
-      S: this,
-      C: (f, a) => f(...a)
-    });
+      blueprints: mergedBlueprints.length,
+      objects: mergedObjects.length
+    }, { "~LogMeta": "~LogMeta", F: __dxlog_file, L: 155, S: this });
   }
   async _close() {
     this._registry.set(this._blueprints, []);
@@ -149,13 +168,8 @@ var AiContextBinder = class extends Resource {
     log("bind", {
       blueprints: addedBlueprints.length,
       objects: addedObjects.length
-    }, {
-      F: __dxlog_file,
-      L: 185,
-      S: this,
-      C: (f, a) => f(...a)
-    });
-    await this._queue.append([
+    }, { "~LogMeta": "~LogMeta", F: __dxlog_file, L: 176, S: this });
+    await Runtime.runPromise(this._runtime)(Feed.append(this._feed, [
       Obj.make(ContextBinding, {
         blueprints: {
           added: addedBlueprints,
@@ -166,22 +180,27 @@ var AiContextBinder = class extends Resource {
           removed: []
         }
       })
-    ]);
+    ]));
   }
   async unbind({ blueprints, objects }) {
     if (!blueprints?.length && !objects?.length) {
       return;
     }
+    const removedBlueprintDxns = (blueprints ?? []).map((ref) => ref.dxn);
+    const removedObjectDxns = (objects ?? []).map((ref) => ref.dxn);
+    if (removedBlueprintDxns.length > 0) {
+      const current = this._registry.get(this._blueprints);
+      this._registry.set(this._blueprints, current.filter((obj) => !removedBlueprintDxns.some((dxn) => DXN.equalsEchoId(Obj.getDXN(obj), dxn))));
+    }
+    if (removedObjectDxns.length > 0) {
+      const current = this._registry.get(this._objects);
+      this._registry.set(this._objects, current.filter((obj) => !removedObjectDxns.some((dxn) => DXN.equalsEchoId(Obj.getDXN(obj), dxn))));
+    }
     log("unbind", {
       blueprints: blueprints?.length,
       objects: objects?.length
-    }, {
-      F: __dxlog_file,
-      L: 205,
-      S: this,
-      C: (f, a) => f(...a)
-    });
-    await this._queue.append([
+    }, { "~LogMeta": "~LogMeta", F: __dxlog_file, L: 208, S: this });
+    await Runtime.runPromise(this._runtime)(Feed.append(this._feed, [
       Obj.make(ContextBinding, {
         blueprints: {
           added: [],
@@ -192,7 +211,7 @@ var AiContextBinder = class extends Resource {
           removed: objects ?? []
         }
       })
-    ]);
+    ]));
   }
   /**
   * Process bindings to filter duplicates and determine next state.
@@ -246,6 +265,23 @@ var AiContextBinder = class extends Resource {
     }));
   }
   /**
+  * Merge resolved items into the current set, adding only items not already present (by DXN).
+  */
+  _mergeInto(current, resolved) {
+    const seen = new Set(current.map((obj) => Obj.getDXN(obj).toString()));
+    const merged = [
+      ...current
+    ];
+    for (const obj of resolved) {
+      const dxn = Obj.getDXN(obj).toString();
+      if (!seen.has(dxn)) {
+        seen.add(dxn);
+        merged.push(obj);
+      }
+    }
+    return merged;
+  }
+  /**
   * Resolve references to objects, loading them first if needed and falling back to existing objects.
   */
   async _resolve(refs, current) {
@@ -259,7 +295,7 @@ var AiContextBinder = class extends Resource {
         target = ref.target;
       }
       return target ?? current.find((obj) => Obj.getDXN(obj).toString() === ref.dxn.toString());
-    }).filter(isTruthy);
+    }).filter(isNonNullable);
   }
 };
 var AiContextService = class _AiContextService extends Context.Tag("@dxos/assistant/AiContextService")() {
@@ -276,165 +312,926 @@ var AiContextService = class _AiContextService extends Context.Tag("@dxos/assist
       return binder.getObjects().filter(Obj.instanceOf(type));
     });
   };
-  /**
-  * Glorified type cast until we figure out function service typing.
-  * For context, the AiContextService is provided to functions called from tool calls, but it's not implemented in types yet.
-  * @deprecated
-  */
-  static fixFunctionHandlerType = (eff) => eff;
 };
 
-// src/conversation/conversation.ts
+// src/conversation/session.ts
+import * as Array4 from "effect/Array";
+import * as Context4 from "effect/Context";
+import * as Effect7 from "effect/Effect";
+import * as Either from "effect/Either";
+import { pipe as pipe4 } from "effect/Function";
+import * as Layer3 from "effect/Layer";
+import * as Record2 from "effect/Record";
+import * as Runtime2 from "effect/Runtime";
+import { Operation as Operation2, Trace as Trace3 } from "@dxos/compute";
+import { Resource as Resource2 } from "@dxos/context";
+import { Database as Database2, Feed as Feed2, Filter, Obj as Obj6 } from "@dxos/echo";
+import { acquireReleaseResource } from "@dxos/effect";
+import { invariant as invariant2 } from "@dxos/invariant";
+import { log as log4 } from "@dxos/log";
+import { McpToolkit } from "@dxos/mcp-client";
+import { Message as Message3 } from "@dxos/types";
+
+// src/functions/services.ts
+import * as Tool from "@effect/ai/Tool";
 import * as Context2 from "effect/Context";
 import * as Effect2 from "effect/Effect";
 import * as Layer from "effect/Layer";
-import { Resource as Resource2 } from "@dxos/context";
-import { Obj as Obj2 } from "@dxos/echo";
-import { acquireReleaseResource } from "@dxos/effect";
-import { QueueService } from "@dxos/functions";
+import * as Option from "effect/Option";
+import * as Record from "effect/Record";
+import * as Schema3 from "effect/Schema";
+import * as SchemaAST from "effect/SchemaAST";
+import { AiToolNotFoundError, ToolExecutionService, ToolResolverService } from "@dxos/ai";
+import { OpaqueToolkit } from "@dxos/ai";
+import { Operation, OperationRegistry } from "@dxos/compute";
+import { todo } from "@dxos/debug";
+import { Ref as Ref3 } from "@dxos/echo";
 import { invariant } from "@dxos/invariant";
+
+// src/types/artifact.ts
+import * as Schema2 from "effect/Schema";
+import { Obj as Obj2, Ref as Ref2 } from "@dxos/echo";
+import { Database } from "@dxos/echo";
+import { EncodedReference } from "@dxos/echo-protocol";
+import { DXN as DXN2, LOCAL_SPACE_TAG } from "@dxos/keys";
+import { trim } from "@dxos/util";
+var createArtifactElement = (id) => `<artifact id=${id} />`;
+var ArtifactId = class extends Schema2.String.annotations({
+  // TODO(dmaretskyi): This section gets overriden.
+  description: trim`
+    The ID of the referenced object. Formats accepted:
+    - DXN (dxn:echo:@:01KG7R1ZXWFMWQ4DA1Q6TN1DG4). DXNs can be prepended with an @ symbol for compatibility with in-text references.
+    - space ID, object ID tuple (spaceID:objectID)
+    - Only object ID that is assumed to be in the current space (01KG7R1ZXWFMWQ4DA1Q6TN1DG4)
+  `,
+  examples: [
+    "dxn:echo:@:01KG7R1ZXWFMWQ4DA1Q6TN1DG4",
+    "@dxn:echo:@:01KG7R1ZXWFMWQ4DA1Q6TN1DG4",
+    "BM3FSHFOMJCHCG5QW7JTVKGYABD2GAA7G:01KG7R1ZXWFMWQ4DA1Q6TN1DG4",
+    "01KG7R1ZXWFMWQ4DA1Q6TN1DG4"
+  ]
+}) {
+  static toDXN(reference, owningSpaceId) {
+    if (reference.startsWith("@dxn:")) {
+      return DXN2.parse(reference.slice(1));
+    } else if (reference.startsWith("dxn:")) {
+      return DXN2.parse(reference);
+    } else if (/^[A-Z0-9]+:[A-Z0-9]+$/.test(reference)) {
+      const [spaceId, objectId] = reference.split(":");
+      return new DXN2(DXN2.kind.ECHO, [
+        spaceId === owningSpaceId ? LOCAL_SPACE_TAG : spaceId,
+        objectId
+      ]);
+    } else if (/^[A-Z0-9]+$/.test(reference)) {
+      return DXN2.fromLocalObjectId(reference);
+    } else {
+      throw new Error(`Unable to parse object reference: ${reference}`);
+    }
+  }
+  /**
+  * Resolves an artifact ID to an object.
+  */
+  static resolve(schema, ref) {
+    const dxn = ArtifactId.toDXN(ref);
+    return Database.resolve(dxn, schema);
+  }
+};
+var RefFromLLM = Schema2.transform(ArtifactId, Ref2.Ref(Obj2.Unknown), {
+  decode: (fromA, fromI) => EncodedReference.fromDXN(ArtifactId.toDXN(fromA)),
+  encode: (toI, toA) => EncodedReference.toDXN(toI).toString(),
+  strict: false
+}).annotations({
+  description: ArtifactId.ast.annotations.description
+});
+
+// src/functions/services.ts
+var __dxlog_file2 = "/__w/dxos/dxos/packages/core/compute/assistant/src/functions/services.ts";
+var makeToolResolverFromOperations = ({ toolkit: extraToolkit = OpaqueToolkit.empty } = {}) => {
+  return Layer.effect(ToolResolverService, Effect2.gen(function* () {
+    const toolkitProvider = yield* OpaqueToolkit.OpaqueToolkitProvider;
+    const operationRegistry = yield* OperationRegistry.Service;
+    return {
+      resolve: (id) => Effect2.gen(function* () {
+        const toolkit = OpaqueToolkit.merge(extraToolkit, toolkitProvider.getToolkit());
+        const tool = toolkit.toolkit.tools[id];
+        if (tool) {
+          return tool;
+        }
+        return yield* operationRegistry.resolve(id).pipe(Effect2.flatMap(Option.match({
+          onSome: (_) => Effect2.succeed(projectFunctionToTool(_)),
+          onNone: () => Effect2.fail(new AiToolNotFoundError(id))
+        })));
+      })
+    };
+  }));
+};
+var makeToolExecutionService = (opts) => Layer.effect(ToolExecutionService, Effect2.gen(function* () {
+  const toolkitProvider = yield* OpaqueToolkit.OpaqueToolkitProvider;
+  const toolkit = toolkitProvider.getToolkit();
+  const toolkitHandler = yield* toolkit.toolkit.pipe(Effect2.provide(toolkit.layer));
+  invariant(isHandlerLike(toolkitHandler), void 0, { "~LogMeta": "~LogMeta", F: __dxlog_file2, L: 42, S: this, A: ["isHandlerLike(toolkitHandler)", ""] });
+  return {
+    handlersFor: (toolkit2) => {
+      const makeHandler = (tool) => {
+        return Effect2.fn(`toolFunctionHandler ${tool.name}`)(function* (input) {
+          if (toolkitHandler.tools[tool.name]) {
+            if (Tool.isProviderDefined(tool)) {
+              throw new Error("Attempted to call a provider-defined tool");
+            }
+            const { result } = yield* toolkitHandler.handle(tool.name, input);
+            return result;
+          }
+          return yield* opts.invoke(tool, input).pipe(Effect2.catchAllDefect((defect) => Effect2.fail(defect)));
+        });
+      };
+      return toolkit2.of(Record.map(toolkit2.tools, (tool, _name) => Tool.isUserDefined(tool) ? makeHandler(tool) : null));
+    }
+  };
+}));
+var makeToolExecutionServiceFromOperationInvoker = () => {
+  return Layer.unwrapEffect(Effect2.gen(function* () {
+    const operationInvoker = yield* Operation.Service;
+    return makeToolExecutionService({
+      invoke: (tool, input) => Effect2.gen(function* () {
+        const operationDef = getOperationFromTool(tool).pipe(Option.getOrThrow);
+        return yield* operationInvoker.invoke(operationDef, input).pipe(Effect2.orDie);
+      })
+    });
+  }));
+};
+var ToolExecutionServices = Layer.mergeAll(makeToolResolverFromOperations(), makeToolExecutionServiceFromOperationInvoker());
+var FunctionToolAnnotation = class extends Context2.Tag("@dxos/assistant/FunctionToolAnnotation")() {
+};
+var getOperationFromTool = (tool) => {
+  return Context2.getOption(FunctionToolAnnotation)(tool.annotations).pipe(Option.map(({ definition }) => definition));
+};
+var toolCache = /* @__PURE__ */ new WeakMap();
+var projectFunctionToTool = (fn2) => {
+  if (toolCache.has(fn2)) {
+    return toolCache.get(fn2);
+  }
+  const tool = Tool.make(makeToolName(fn2.meta.name ?? fn2.meta.key), {
+    description: fn2.meta.description,
+    parameters: createStructFieldsFromSchema(fn2.input),
+    failure: Schema3.Any
+  }).annotate(FunctionToolAnnotation, {
+    definition: fn2
+  });
+  toolCache.set(fn2, tool);
+  return tool;
+};
+var makeToolName = (name) => {
+  const toolName = name.toLowerCase().replace(/[^a-zA-Z0-9]/g, "-");
+  invariant(toolName.match(/^[a-z_][a-z0-9-_]*$/), void 0, { "~LogMeta": "~LogMeta", F: __dxlog_file2, L: 100, S: void 0, A: ["toolName.match(/^[a-z_][a-z0-9-_]*$/)", ""] });
+  return toolName;
+};
+var createStructFieldsFromSchema = (schema) => {
+  switch (schema.ast._tag) {
+    case "TypeLiteral":
+      return Object.fromEntries(schema.ast.propertySignatures.map((prop) => [
+        prop.name,
+        Schema3.make(mapSchemaTypeForLLM(prop.type))
+      ]));
+    case "VoidKeyword":
+      return {};
+    default:
+      return todo(`Unsupported schema AST: ${schema.ast._tag}`);
+  }
+};
+var mapSchemaTypeForLLM = (ast) => {
+  if (Ref3.isRefType(ast)) {
+    const description = ast.annotations.description ? ast.annotations.description + "\n" + RefFromLLM.ast.annotations.description : RefFromLLM.ast.annotations.description;
+    return RefFromLLM.annotations({
+      description
+    }).ast;
+  } else if (SchemaAST.isTupleType(ast)) {
+    return new SchemaAST.TupleType(ast.elements.map((t) => new SchemaAST.OptionalType(mapSchemaTypeForLLM(t.type), t.isOptional, t.annotations)), ast.rest.map((t) => new SchemaAST.Type(mapSchemaTypeForLLM(t.type), t.annotations)), ast.isReadonly, ast.annotations);
+  } else if (SchemaAST.isTypeLiteral(ast)) {
+    return new SchemaAST.TypeLiteral(ast.propertySignatures.map((p) => new SchemaAST.PropertySignature(p.name, mapSchemaTypeForLLM(p.type), p.isOptional, p.isReadonly, p.annotations)), ast.indexSignatures, ast.annotations);
+  }
+  return ast;
+};
+var isHandlerLike = (value) => {
+  return typeof value.tools === "object" && typeof value.handle === "function";
+};
+
+// src/session/artifact-diff.ts
+import * as Context3 from "effect/Context";
+var ArtifactDiffResolver = class extends Context3.Tag("@dxos/assistant/ArtifactDiffResolver")() {
+};
+
+// src/session/observer.ts
+import * as Effect3 from "effect/Effect";
+import * as Function2 from "effect/Function";
+var GenerationObserver = Object.freeze({
+  make: ({ onBegin = Function2.constant(Effect3.void), onPart = Function2.constant(Effect3.void), onBlock = Function2.constant(Effect3.void), onEnd = Function2.constant(Effect3.void), onMessage = Function2.constant(Effect3.void) } = {}) => ({
+    onBegin,
+    onPart,
+    onBlock,
+    onEnd,
+    onMessage
+  }),
+  noop: () => GenerationObserver.make(),
+  /**
+  * Debug printer to be used in unit-tests and browser devtools.
+  */
+  fromPrinter: (printer) => GenerationObserver.make({
+    onBlock: (block) => Effect3.sync(() => {
+      if (block.pending) {
+        return;
+      }
+      printer.printContentBlock(block);
+    }),
+    onMessage: (message) => Effect3.sync(() => {
+      if (message.sender.role === "assistant") {
+        return;
+      }
+      printer.printMessage(message);
+    })
+  })
+});
+
+// src/session/request.ts
+import * as LanguageModel from "@effect/ai/LanguageModel";
+import * as Array3 from "effect/Array";
+import * as Chunk from "effect/Chunk";
+import * as Effect5 from "effect/Effect";
+import { pipe as pipe3 } from "effect/Function";
+import * as Option3 from "effect/Option";
+import * as Stream from "effect/Stream";
+import { AiParser, AiPreprocessor, AiSummarizer, callTool, withoutToolCallParising } from "@dxos/ai";
+import { Trace as Trace2 } from "@dxos/compute";
+import { Obj as Obj5 } from "@dxos/echo";
+import { log as log3 } from "@dxos/log";
+import { ContentBlock as ContentBlock2, Message as Message2 } from "@dxos/types";
+
+// src/tracing.ts
+import * as Schema4 from "effect/Schema";
+import { Trace } from "@dxos/compute";
+import { Obj as Obj3 } from "@dxos/echo";
+import { ContentBlock, Actor } from "@dxos/types";
+var PartialBlock = Trace.EventType("assistant.partialBlock", {
+  schema: Schema4.Struct({
+    messageId: Obj3.ID,
+    role: Actor.Role,
+    block: ContentBlock.Any
+  }),
+  isEphemeral: true
+});
+var CompleteBlock = Trace.EventType("assistant.completeBlock", {
+  schema: Schema4.Struct({
+    messageId: Obj3.ID,
+    role: Actor.Role,
+    block: ContentBlock.Any
+  }),
+  isEphemeral: false
+});
+var AgentRequestBegin = Trace.EventType("assistant.agentRequestBegin", {
+  schema: Schema4.Struct({}),
+  isEphemeral: false
+});
+var AgentRequestEnd = Trace.EventType("assistant.agentRequestEnd", {
+  schema: Schema4.Struct({}),
+  isEphemeral: false
+});
+var McpServerError = Trace.EventType("assistant.mcpServerError", {
+  schema: Schema4.Struct({
+    url: Schema4.String,
+    protocol: Schema4.Literal("sse", "http"),
+    message: Schema4.String
+  }),
+  isEphemeral: true
+});
+
+// src/session/format.ts
+import * as Effect4 from "effect/Effect";
+import * as Function3 from "effect/Function";
+import * as Option2 from "effect/Option";
+import { Template } from "@dxos/compute";
+import { Obj as Obj4 } from "@dxos/echo";
+import { ObjectVersion } from "@dxos/echo-db";
 import { log as log2 } from "@dxos/log";
 import { Message } from "@dxos/types";
-var __dxlog_file2 = "/__w/dxos/dxos/packages/core/assistant/src/conversation/conversation.ts";
-var AiConversation = class extends Resource2 {
+import { trim as trim2 } from "@dxos/util";
+
+// src/errors.ts
+import { BaseError } from "@dxos/errors";
+var AiModelError = class extends BaseError.extend("AiModelError", "AI model error") {
+};
+var AiAssistantError = class extends BaseError.extend("AiAssistantError", "AI assistant error") {
+};
+
+// src/session/format.ts
+var __dxlog_file3 = "/__w/dxos/dxos/packages/core/compute/assistant/src/session/format.ts";
+var formatSystemPrompt = ({ system, blueprints = [], objects = [] }) => Effect4.gen(function* () {
+  const blueprintDefs = yield* Function3.pipe(blueprints, Effect4.forEach((blueprint) => Effect4.succeed(blueprint.instructions)), Effect4.flatMap(Effect4.forEach((template) => Effect4.gen(function* () {
+    return trim2`
+            <blueprint>
+              ${yield* Template.processTemplate(template)}
+            </blueprint>
+          `;
+  }))), Effect4.map((blueprints2) => blueprints2.length > 0 ? [
+    "## Blueprints Definitions",
+    ...blueprints2
+  ].join("\n\n") : void 0));
+  const objectDefs = yield* Function3.pipe(objects, Effect4.forEach((object) => Effect4.succeed(trim2`
+          <object>
+            <dxn>${Obj4.getDXN(object)}</dxn>
+            <typename>${Obj4.getTypename(object)}</typename>
+          </object>
+        `)), Effect4.map((objects2) => objects2.length > 0 ? [
+    "## Context Objects",
+    ...objects2
+  ].join("\n\n") : void 0));
+  return yield* Function3.pipe(Effect4.succeed([
+    system,
+    blueprintDefs,
+    objectDefs
+  ].filter((def) => def !== void 0)), Effect4.map((parts) => parts.join("\n\n")));
+}).pipe(Effect4.withSpan("formatSystemPrompt"));
+var formatUserPrompt = ({ prompt, history = [] }) => Effect4.gen(function* () {
+  const blocks = [];
+  const artifactDiffResolver = yield* Effect4.serviceOption(ArtifactDiffResolver);
+  if (Option2.isSome(artifactDiffResolver)) {
+    const versions = gatherObjectVersions(history);
+    const artifactDiff = yield* Effect4.tryPromise({
+      try: () => artifactDiffResolver.value.resolve([
+        ...versions.entries()
+      ].map(([id, version]) => ({
+        id,
+        lastVersion: version
+      }))),
+      catch: AiAssistantError.wrap({
+        message: "Artifact diff resolution error"
+      })
+    });
+    log2("version", {
+      artifactDiff,
+      versions
+    }, { "~LogMeta": "~LogMeta", F: __dxlog_file3, L: 65, S: this });
+    for (const [id, { version }] of [
+      ...artifactDiff.entries()
+    ]) {
+      if (ObjectVersion.equals(version, versions.get(id))) {
+        artifactDiff.delete(id);
+        continue;
+      }
+      blocks.push({
+        _tag: "anchor",
+        objectId: id,
+        version
+      });
+    }
+    if (artifactDiff.size > 0) {
+      blocks.push(createArtifactUpdateBlock(artifactDiff));
+    }
+  }
+  return Obj4.make(Message.Message, {
+    created: (/* @__PURE__ */ new Date()).toISOString(),
+    sender: {
+      role: "user"
+    },
+    blocks: typeof prompt === "string" ? [
+      ...blocks,
+      {
+        _tag: "text",
+        text: prompt
+      }
+    ] : [
+      ...blocks,
+      ...prompt
+    ]
+  });
+}).pipe(Effect4.withSpan("formatUserPrompt"));
+var gatherObjectVersions = (messages) => {
+  const artifactIds = /* @__PURE__ */ new Map();
+  for (const message of messages) {
+    for (const block of message.blocks) {
+      if (block._tag === "anchor") {
+        artifactIds.set(block.objectId, block.version);
+      }
+    }
+  }
+  return artifactIds;
+};
+var createArtifactUpdateBlock = (artifactDiff) => {
+  return {
+    _tag: "text",
+    // TODO(dmaretskyi): Does this need to be a special content-block?
+    disposition: "synthetic",
+    text: trim2`
+      The following artifacts have been updated since the last message:
+      ${[
+      ...artifactDiff.entries()
+    ].map(([id, { diff }]) => `<changed-artifact id="${id}">${diff ? `
+${diff}` : ""}</changed-artifact>`).join("\n")}
+    `
+  };
+};
+
+// src/session/request.ts
+var __dxlog_file4 = "/__w/dxos/dxos/packages/core/compute/assistant/src/session/request.ts";
+var AiRequest = class {
+  _options;
+  /** Prevents concurrent execution of session. */
+  _semaphore = Effect5.runSync(Effect5.makeSemaphore(1));
+  _observer;
+  _onOutput;
+  /** Message history from queue. */
+  // TODO(burdon): Evolve into supporting a git-like graph of messages.
+  _history = [];
+  /** Pending messages for this session (incl. the current prompt). */
+  _pending = [];
+  _started = 0;
+  _ended = 0;
+  _toolCalls = 0;
+  constructor(_options = {}) {
+    this._options = _options;
+    this._observer = _options.observer ?? GenerationObserver.noop();
+    this._onOutput = _options.onOutput ?? (() => Effect5.void);
+  }
+  get duration() {
+    return this._ended - this._started;
+  }
+  get toolCalls() {
+    return this._toolCalls;
+  }
+  get pending() {
+    return this._pending;
+  }
+  _submitMessage = (message) => Effect5.gen(this, function* () {
+    this._pending.push(message);
+    yield* this._observer.onMessage(message);
+    for (const block of message.blocks) {
+      log3("write complete block", {
+        messageId: message.id,
+        role: message.sender.role,
+        block: JSON.stringify(block)
+      }, { "~LogMeta": "~LogMeta", F: __dxlog_file4, L: 57, S: this });
+      yield* Trace2.write(CompleteBlock, {
+        messageId: message.id,
+        role: message.sender.role,
+        block
+      });
+    }
+    yield* this._onOutput(message);
+    return message;
+  });
+  getToolCalls = () => pipe3([
+    ...this._history,
+    ...this._pending
+  ], Array3.reverse, Array3.takeWhile((_) => _.sender.role === "assistant"), Array3.flatMap((_) => _.blocks.filter(ContentBlock2.is("toolCall")).map((block) => ({
+    block,
+    message: _
+  }))), Array3.filter((_) => !_.block.providerExecuted), Array3.reverse);
   /**
-  * Blueprints and objects bound to the conversation.
+  * Initialize a session: set up history, perform summarization if needed, and submit the user prompt.
+  * Must be called before `runTurn()`.
+  */
+  begin = ({ prompt, system, history = [], blueprints = [], objects = [] }) => Effect5.gen(this, function* () {
+    this._started = Date.now();
+    this._history = [
+      ...history
+    ];
+    this._pending = [];
+    const systemPrompt = yield* formatSystemPrompt({
+      system,
+      blueprints,
+      objects
+    }).pipe(Effect5.orDie);
+    if (this._options.summarizationThreshold !== void 0) {
+      const tokenCount = yield* AiPreprocessor.estimateTokens(yield* AiPreprocessor.preprocessPrompt([
+        ...this._history
+      ], {
+        system: systemPrompt
+      }));
+      if (tokenCount > this._options.summarizationThreshold) {
+        const summary = yield* AiSummarizer.summarize([
+          ...this._history
+        ]);
+        yield* this._submitMessage(summary);
+      }
+    }
+    yield* this._submitMessage(yield* formatUserPrompt({
+      prompt,
+      history
+    }));
+  }).pipe(Effect5.withSpan("AiRequest.begin"));
+  /**
+  * Execute a single turn: one LLM generation followed by tool execution.
+  * The toolkit and system prompt can be updated between turns to reflect context changes (e.g. dynamically enabled blueprints).
+  */
+  runAgentTurn = ({ system, toolkit: opaqueToolkit }) => Effect5.gen(this, function* () {
+    log3.info("request", {
+      system: {
+        snippet: createSnippet(system),
+        length: system.length
+      },
+      pending: this._pending.length,
+      history: this._history.length
+    }, { "~LogMeta": "~LogMeta", F: __dxlog_file4, L: 114, S: this });
+    const prompt = yield* AiPreprocessor.preprocessPrompt([
+      ...this._history,
+      ...this._pending
+    ], {
+      system,
+      cacheControl: "ephemeral"
+    });
+    const toolkit = opaqueToolkit ? yield* opaqueToolkit.handlers : void 0;
+    const observer = this._observer;
+    let currentMessageId = null;
+    const messages = yield* LanguageModel.streamText({
+      prompt,
+      toolkit,
+      disableToolCallResolution: true
+    }).pipe(withoutToolCallParising, AiParser.parseResponse({
+      emitPartial: true,
+      onBegin: () => observer.onBegin(),
+      onBlock: (block) => observer.onBlock(block),
+      onPart: (part) => observer.onPart(part),
+      onEnd: (summary) => observer.onEnd(summary)
+    }), Stream.mapEffect((block) => Effect5.gen(this, function* () {
+      if (block.pending) {
+        currentMessageId ??= Obj5.ID.random();
+        log3("emit ephemeral message", {
+          id: currentMessageId,
+          type: block._tag
+        }, { "~LogMeta": "~LogMeta", F: __dxlog_file4, L: 145, S: this });
+        yield* Trace2.write(PartialBlock, {
+          messageId: currentMessageId,
+          role: "assistant",
+          block
+        });
+        return Option3.none();
+      } else {
+        currentMessageId ??= Obj5.ID.random();
+        const id = currentMessageId;
+        currentMessageId = null;
+        log3("emit complete message", {
+          id,
+          type: block._tag
+        }, { "~LogMeta": "~LogMeta", F: __dxlog_file4, L: 159, S: this });
+        const message = Obj5.make(Message2.Message, {
+          id,
+          created: (/* @__PURE__ */ new Date()).toISOString(),
+          sender: {
+            role: "assistant"
+          },
+          blocks: [
+            block
+          ]
+        });
+        return Option3.some(yield* this._submitMessage(message));
+      }
+    }), {
+      concurrency: 1,
+      unordered: false
+    }), Stream.filterMap((_) => _), Stream.runCollect, Effect5.map(Chunk.toArray));
+    log3.info("messages", {
+      messages
+    }, { "~LogMeta": "~LogMeta", F: __dxlog_file4, L: 179, S: this });
+    const toolCalls = this.getToolCalls();
+    if (toolCalls.length === 0) {
+      this._ended = Date.now();
+      return {
+        messages,
+        done: true
+      };
+    } else if (!toolkit) {
+      throw new Error("No toolkit provided");
+    }
+    return {
+      messages,
+      done: false
+    };
+  }).pipe(Effect5.withSpan("AiRequest.runAgentTurn"));
+  runTools = ({ toolkit: opaqueToolkit }) => Effect5.gen(this, function* () {
+    const toolkit = opaqueToolkit ? yield* opaqueToolkit.handlers : void 0;
+    const toolCalls = this.getToolCalls();
+    const toolResults = yield* Effect5.forEach(toolCalls, ({ block, message }) => {
+      if (!toolkit) {
+        throw new Error("No toolkit provided");
+      }
+      return callTool(toolkit, block);
+    });
+    yield* this._submitMessage(Obj5.make(Message2.Message, {
+      created: (/* @__PURE__ */ new Date()).toISOString(),
+      sender: {
+        role: "tool"
+      },
+      blocks: toolResults
+    }));
+    this._toolCalls += toolResults.length;
+  }).pipe(Effect5.withSpan("AiRequest.runTools"));
+  /**
+  * Run a full conversation turn loop. Equivalent to calling `begin()` then `runTurn()` in a loop.
+  */
+  run = ({ prompt, system: systemTemplate, history = [], objects = [], blueprints = [], toolkit }) => Effect5.gen(this, function* () {
+    yield* this.begin({
+      prompt,
+      system: systemTemplate,
+      history,
+      objects,
+      blueprints
+    });
+    const system = yield* formatSystemPrompt({
+      system: systemTemplate,
+      blueprints,
+      objects
+    }).pipe(Effect5.orDie);
+    do {
+      const { done } = yield* this.runAgentTurn({
+        system,
+        toolkit
+      });
+      if (done) {
+        break;
+      }
+      yield* this.runTools({
+        toolkit
+      });
+    } while (true);
+    log3("done", {
+      pending: this._pending.length,
+      duration: this.duration,
+      tools: this._toolCalls
+    }, { "~LogMeta": "~LogMeta", F: __dxlog_file4, L: 242, S: this });
+    return this._pending;
+  }).pipe(this._semaphore.withPermits(1), Effect5.withSpan("AiRequest.run"));
+};
+var createSnippet = (text, len = 32) => text.length <= len * 2 ? text : [
+  text.slice(0, len),
+  "...",
+  text.slice(-len)
+].join("");
+
+// src/session/toolkit.ts
+import * as Toolkit from "@effect/ai/Toolkit";
+import * as Effect6 from "effect/Effect";
+import * as Layer2 from "effect/Layer";
+import { OpaqueToolkit as OpaqueToolkit2, ToolExecutionService as ToolExecutionService2, ToolResolverService as ToolResolverService2 } from "@dxos/ai";
+import { isTruthy } from "@dxos/util";
+var createToolkit = ({ toolkit: toolkitProp, blueprints = [], opaqueToolkits = [] }) => Effect6.gen(function* () {
+  const blueprintToolkit = yield* ToolResolverService2.resolveToolkit(blueprints.flatMap(({ tools }) => tools));
+  const blueprintToolHandler = yield* blueprintToolkit.toContext(ToolExecutionService2.handlersFor(blueprintToolkit));
+  const opaqueToolkit = OpaqueToolkit2.merge(...opaqueToolkits);
+  const toolkitDefs = [
+    toolkitProp?.toolkit,
+    blueprintToolkit,
+    opaqueToolkit.toolkit
+  ].filter(isTruthy);
+  const mergedToolkit = Toolkit.merge(...toolkitDefs);
+  const combinedHandlerLayer = Layer2.mergeAll(Layer2.succeedContext(blueprintToolHandler), toolkitProp?.layer ?? OpaqueToolkit2.empty.layer, opaqueToolkit.layer);
+  return OpaqueToolkit2.make(mergedToolkit, combinedHandlerLayer);
+});
+
+// src/session/version-pin.ts
+import * as Schema5 from "effect/Schema";
+import { getVersion } from "@dxos/echo-db";
+import { ObjectId } from "@dxos/keys";
+var ObjectVersionSchema = Schema5.Unknown;
+var VersionPinSchema = Schema5.Struct({
+  // TODO(dmaretskyi): Use Ref when those support encoding.
+  objectId: ObjectId,
+  // TODO(dmaretskyi): Could be opaque.
+  version: ObjectVersionSchema
+});
+var VersionPin = class extends VersionPinSchema {
+  static DISPOSITION = "version-pin";
+  static fromObject(object) {
+    return VersionPin.make({
+      objectId: object.id,
+      version: getVersion(object)
+    });
+  }
+  static createBlock(pin) {
+    return {
+      _tag: "json",
+      disposition: VersionPin.DISPOSITION,
+      data: JSON.stringify(pin)
+    };
+  }
+};
+
+// src/conversation/session.ts
+var __dxlog_file5 = "/__w/dxos/dxos/packages/core/compute/assistant/src/conversation/session.ts";
+var SUMMARY_THRESHOLD = 8e4;
+var AiSession = class extends Resource2 {
+  /**
+  * Blueprints and objects bound to the session.
   */
   _binder;
-  _queue;
-  _toolkit;
+  _feed;
+  _runtime;
   constructor(options) {
     super();
-    this._queue = options.queue;
-    this._toolkit = options.toolkit;
-    invariant(this._queue, void 0, {
-      F: __dxlog_file2,
-      L: 60,
-      S: this,
-      A: [
-        "this._queue",
-        ""
-      ]
-    });
+    this._feed = options.feed;
+    this._runtime = options.runtime;
+    invariant2(this._feed, void 0, { "~LogMeta": "~LogMeta", F: __dxlog_file5, L: 41, S: this, A: ["this._feed", ""] });
+    invariant2(this._runtime, void 0, { "~LogMeta": "~LogMeta", F: __dxlog_file5, L: 42, S: this, A: ["this._runtime", ""] });
     this._binder = new AiContextBinder({
-      queue: this._queue,
+      feed: this._feed,
+      runtime: this._runtime,
       registry: options.registry
     });
   }
   async _open() {
     await this._binder.open(this._ctx);
   }
-  get queue() {
-    return this._queue;
+  get feed() {
+    return this._feed;
   }
   get context() {
     return this._binder;
   }
-  get toolkit() {
-    return this._toolkit;
-  }
   async getHistory() {
-    const queueItems = await this._queue.queryObjects();
-    return queueItems.filter(Obj2.instanceOf(Message.Message));
+    const queryResult = await Runtime2.runPromise(this._runtime)(Feed2.query(this._feed, Filter.type(Message3.Message)));
+    const items = await queryResult.run();
+    return items.filter(Obj6.instanceOf(Message3.Message));
   }
   getTools() {
-    return Effect2.gen(this, function* () {
-      const blueprints = this.context.getBlueprints();
-      const tookit = yield* createToolkit({
-        toolkit: this._toolkit,
-        blueprints
+    return Effect7.gen(this, function* () {
+      const toolkit = yield* createToolkit({
+        blueprints: this.context.getBlueprints()
       });
-      return tookit.tools;
-    }).pipe(Effect2.orDie);
+      return toolkit.toolkit.tools;
+    }).pipe(Effect7.orDie);
+  }
+  /**
+  * Not provided by default, since users might want to override them.
+  */
+  makeToolExecutionServices() {
+    return ToolExecutionServices.pipe(Layer3.provide(Operation2.withInvocationOptions({
+      conversation: Obj6.getDXN(this._feed).toString()
+    })));
   }
   /**
   * Creates a new cancelable request effect.
   */
   createRequest(params) {
-    return Effect2.gen(this, function* () {
-      const history = yield* Effect2.promise(() => this.getHistory());
+    return Effect7.gen(this, function* () {
+      const history = yield* Effect7.promise(() => this.getHistory());
       const blueprints = this.context.getBlueprints();
-      const toolkit = yield* createToolkit({
-        toolkit: this._toolkit,
-        blueprints
-      });
       const objects = this.context.getObjects();
-      log2("run", {
+      log4("run", {
         history: history.length,
         blueprints: blueprints.length,
-        tools: Object.keys(toolkit.tools).length,
         objects: objects.length
-      }, {
-        F: __dxlog_file2,
-        L: 112,
-        S: this,
-        C: (f, a) => f(...a)
+      }, { "~LogMeta": "~LogMeta", F: __dxlog_file5, L: 85, S: this });
+      const request = new AiRequest({
+        summarizationThreshold: SUMMARY_THRESHOLD,
+        observer: params.observer,
+        onOutput: (message) => Effect7.promise(() => Runtime2.runPromise(this._runtime)(Feed2.append(this._feed, [
+          message
+        ])))
       });
-      const session = new AiSession();
-      const messages = yield* session.run({
+      yield* request.begin({
         history,
         blueprints,
-        toolkit,
         objects,
-        ...params,
-        onOutput: (message) => Effect2.promise(() => this._queue.append([
-          message
-        ]))
-      }).pipe(Effect2.provide(Layer.mergeAll(Layer.succeed(AiContextService, {
-        binder: this.context
-      }), Layer.succeed(AiConversationService, this))));
-      log2("result", {
-        messages: messages.length,
-        duration: session.duration,
-        toolCalls: session.toolCalls
-      }, {
-        F: __dxlog_file2,
-        L: 141,
-        S: this,
-        C: (f, a) => f(...a)
+        prompt: params.prompt,
+        system: params.system
       });
-      return messages;
-    });
+      do {
+        yield* Effect7.promise(() => this.context.sync());
+        const currentBlueprints = this.context.getBlueprints();
+        const mcps = yield* connectMcpServers(currentBlueprints, params.mcpServers);
+        const toolkit = yield* createToolkit({
+          toolkit: params.toolkit,
+          blueprints: currentBlueprints,
+          opaqueToolkits: mcps
+        });
+        log4("toolkit", {
+          tools: Record2.keys(toolkit.toolkit.tools)
+        }, { "~LogMeta": "~LogMeta", F: __dxlog_file5, L: 114, S: this });
+        const system = yield* formatSystemPrompt({
+          system: params.system,
+          blueprints: currentBlueprints,
+          objects: this.context.getObjects()
+        }).pipe(Effect7.orDie);
+        const { done } = yield* request.runAgentTurn({
+          system,
+          toolkit
+        });
+        if (done) {
+          break;
+        }
+        yield* request.runTools({
+          toolkit
+        });
+      } while (true);
+      log4("result", {
+        messages: request.pending.length,
+        duration: request.duration,
+        toolCalls: request.toolCalls
+      }, { "~LogMeta": "~LogMeta", F: __dxlog_file5, L: 133, S: this });
+      return [
+        ...request.pending
+      ];
+    }).pipe(Effect7.provide(Layer3.mergeAll(Layer3.succeed(AiContextService, {
+      binder: this.context
+    }), Layer3.succeed(AiSessionService, this), Operation2.withInvocationOptions({
+      conversation: Obj6.getDXN(this._feed).toString()
+    }))), Effect7.withSpan("AiSession.createRequest"));
   }
 };
-var AiConversationService = class _AiConversationService extends Context2.Tag("@dxos/assistant/AiConversationService")() {
+var AiSessionService = class _AiSessionService extends Context4.Tag("@dxos/assistant/AiSessionService")() {
   /**
-  * Create a new conversation layer from options.
+  * Create a new session layer from options.
   */
-  static layer = (options) => aiContextFromConversation.pipe(Layer.provideMerge(Layer.scoped(_AiConversationService, Effect2.gen(function* () {
-    const conversation = yield* acquireReleaseResource(() => new AiConversation(options));
-    return conversation;
+  static layer = (options) => aiContextFromSession.pipe(Layer3.provideMerge(Layer3.scoped(_AiSessionService, Effect7.gen(function* () {
+    const session = yield* acquireReleaseResource(() => new AiSession(options));
+    return session;
   }))));
   /**
-  * Create a new conversation with a new queue.
+  * Create a new session with a new feed.
   */
-  static layerNewQueue = (options) => Layer.unwrapScoped(Effect2.gen(function* () {
-    return _AiConversationService.layer({
+  static layerNewFeed = (options) => Layer3.unwrapScoped(Effect7.gen(function* () {
+    const feed = yield* Database2.add(Feed2.make());
+    const runtime2 = yield* Effect7.runtime();
+    return _AiSessionService.layer({
       ...options,
-      queue: yield* QueueService.createQueue()
+      feed,
+      runtime: runtime2
     });
   }));
   /**
-  * Run a prompt in the current conversation.
+  * Run a prompt in the current session.
   */
-  static run = (params) => Effect2.gen(function* () {
-    const conversation = yield* _AiConversationService;
-    return yield* conversation.createRequest(params);
+  static run = (params) => Effect7.gen(function* () {
+    const session = yield* _AiSessionService;
+    return yield* session.createRequest(params);
   });
 };
-var aiContextFromConversation = Layer.effect(AiContextService, Effect2.gen(function* () {
-  const conversation = yield* AiConversationService;
+var aiContextFromSession = Layer3.effect(AiContextService, Effect7.gen(function* () {
+  const session = yield* AiSessionService;
   return {
-    binder: conversation.context
+    binder: session.context
   };
 }));
+var connectMcpServers = (blueprints, spaceMcpServers = []) => {
+  const blueprintServers = pipe4(blueprints, Array4.flatMap((_) => _.mcpServers ?? []), Array4.map(({ url, protocol, apiKey }) => ({
+    url,
+    protocol,
+    apiKey
+  })));
+  const spaceServers = spaceMcpServers.map(({ url, protocol, apiKey }) => ({
+    url,
+    protocol,
+    apiKey
+  }));
+  const allServers = [
+    ...blueprintServers,
+    ...spaceServers
+  ];
+  return pipe4(allServers, Effect7.forEach((options) => McpToolkit.make(options).pipe(
+    // NOTE: Type-inference fails here without explicit void return.
+    Effect7.tap((toolkit) => log4.info("Connected to MCP server", {
+      url: options.url,
+      tools: Object.keys(toolkit.toolkit.tools).length
+    }, { "~LogMeta": "~LogMeta", F: __dxlog_file5, L: 197, S: void 0 })),
+    // Surface typed connection failures via ephemeral trace + warn log, then drop the server.
+    Effect7.tapError((error) => Effect7.gen(function* () {
+      log4.warn("Failed to connect to MCP server", {
+        url: error.url,
+        protocol: error.protocol,
+        message: error.message
+      }, { "~LogMeta": "~LogMeta", F: __dxlog_file5, L: 202, S: this });
+      yield* Trace3.write(McpServerError, {
+        url: error.url,
+        protocol: error.protocol,
+        message: error.message
+      });
+    })),
+    // Catch unexpected defects too (e.g. malformed tool schemas) so a single broken
+    // server can never abort the whole turn — surface them through the same channel.
+    Effect7.catchAllDefect((defect) => Effect7.gen(function* () {
+      const message = defect instanceof Error ? defect.message : String(defect);
+      log4.warn("Unexpected MCP defect", {
+        url: options.url,
+        message
+      }, { "~LogMeta": "~LogMeta", F: __dxlog_file5, L: 216, S: this });
+      yield* Trace3.write(McpServerError, {
+        url: options.url,
+        protocol: options.protocol,
+        message: `Unexpected MCP failure: ${message}`
+      });
+      return yield* Effect7.fail(new McpToolkit.McpConnectionError({
+        url: options.url,
+        protocol: options.protocol,
+        message
+      }));
+    })),
+    Effect7.either
+  )), Effect7.map(Array4.filterMap((_) => Either.getRight(_))));
+};
 
 // src/diff.ts
 import { next as A } from "@automerge/automerge";
 import { DocAccessor, toCursorRange } from "@dxos/echo-db";
-import { log as log3 } from "@dxos/log";
-import { isNonNullable } from "@dxos/util";
-var __dxlog_file3 = "/__w/dxos/dxos/packages/core/assistant/src/diff.ts";
+import { log as log5 } from "@dxos/log";
+import { isNonNullable as isNonNullable2 } from "@dxos/util";
+var __dxlog_file6 = "/__w/dxos/dxos/packages/core/compute/assistant/src/diff.ts";
 var reduceDiffs = (diffs) => {
   return diffs.reduce((acc, diff) => {
     const current = acc[acc.length - 1];
@@ -478,7 +1275,7 @@ var computeDiffsWithCursors = (accessor, diffs) => {
         text: diff.replace
       };
     }
-  }).filter(isNonNullable);
+  }).filter(isNonNullable2);
 };
 var applyDiffs = (accessor, diffs, { errorOnNotFound = false } = {}) => {
   for (const diff of reduceDiffs(diffs)) {
@@ -491,12 +1288,7 @@ var applyDiffs = (accessor, diffs, { errorOnNotFound = false } = {}) => {
         if (errorOnNotFound) {
           throw new Error(`Diff not found: ${diff.match}`);
         } else {
-          log3.warn("diff not found", diff, {
-            F: __dxlog_file3,
-            L: 86,
-            S: void 0,
-            C: (f, a) => f(...a)
-          });
+          log5.warn("diff not found", diff, { "~LogMeta": "~LogMeta", F: __dxlog_file6, L: 71, S: void 0 });
         }
       }
     });
@@ -506,11 +1298,11 @@ var applyDiffs = (accessor, diffs, { errorOnNotFound = false } = {}) => {
 
 // src/execution-graph.ts
 import { AgentStatus } from "@dxos/ai";
-import { Obj as Obj3 } from "@dxos/echo";
+import { Obj as Obj7 } from "@dxos/echo";
 import { MESSAGE_PROPERTY_TOOL_CALL_ID } from "@dxos/functions";
 import { LogLevel } from "@dxos/log";
-import { ContentBlock, Message as Message2 } from "@dxos/types";
-import { isNonNullable as isNonNullable2 } from "@dxos/util";
+import { ContentBlock as ContentBlock3, Message as Message4 } from "@dxos/types";
+import { isNonNullable as isNonNullable3 } from "@dxos/util";
 var SKIP_BLOCKS = [
   "text"
 ];
@@ -533,9 +1325,9 @@ var ExecutionGraph = class {
   addEvents(events) {
     const sortedEvents = this.sortObjectsByCreated(events);
     for (const event of sortedEvents) {
-      if (Obj3.instanceOf(Message2.Message, event)) {
+      if (Obj7.instanceOf(Message4.Message, event)) {
         this._processMessage(event);
-      } else if (Obj3.instanceOf(AgentStatus, event)) {
+      } else if (Obj7.instanceOf(AgentStatus, event)) {
         this._processAgentStatus(event);
       }
     }
@@ -637,11 +1429,17 @@ var ExecutionGraph = class {
   */
   _updatePendingToolResults(toolCallId, agentStatusId) {
     const toolResultId = this._pendingToolResults.get(toolCallId);
-    if (!toolResultId) return;
+    if (!toolResultId) {
+      return;
+    }
     const toolResultCommit = this._commits.find((c) => c.id === toolResultId);
-    if (!toolResultCommit) return;
+    if (!toolResultCommit) {
+      return;
+    }
     const toolCallCommitId = this._toolCallCommitIds.get(toolCallId);
-    if (!toolCallCommitId) return;
+    if (!toolCallCommitId) {
+      return;
+    }
     const toolCallIndex = toolResultCommit.parents.indexOf(toolCallCommitId);
     if (toolCallIndex !== -1) {
       toolResultCommit.parents[toolCallIndex] = agentStatusId;
@@ -653,7 +1451,9 @@ var ExecutionGraph = class {
   */
   _handleLateToolResultUpdate(toolCallId, toolResultId) {
     const toolCallCommitId = this._toolCallCommitIds.get(toolCallId);
-    if (!toolCallCommitId) return;
+    if (!toolCallCommitId) {
+      return;
+    }
     const messageId = toolCallCommitId.split("_toolCall_")[0];
     const toolCallBranch = `${messageId}_${toolCallId}`;
     const agentStatusCommitId = this._lastCommitByBranch.get(toolCallBranch);
@@ -722,7 +1522,7 @@ var messageToCommits = (message, lastBlockId, toolCallIds, lastCommitByBranch, s
       previousBlockId = commit.id;
     }
     return commit;
-  }).filter(isNonNullable2);
+  }).filter(isNonNullable3);
 };
 var getBlockParents = (block, previousBlockId, message, toolCallIds, lastCommitByBranch) => {
   const parents = [];
@@ -756,7 +1556,9 @@ var createBlockCommit = (block, message, branch, parents, idx) => {
   const timestamp = new Date(message.created);
   switch (block._tag) {
     case "text":
-      if (!block.text.trim().length) return null;
+      if (!block.text.trim().length) {
+        return null;
+      }
       return {
         id: getGenericBlockId(message.id, idx),
         branch,
@@ -769,7 +1571,7 @@ var createBlockCommit = (block, message, branch, parents, idx) => {
           ],
           message: "Processing request..."
         } : {
-          icon: "ph--robot--regular",
+          icon: "ph--drone--regular",
           message: `Response (${block.text.split(" ").length} words)`
         }
       };
@@ -793,7 +1595,7 @@ var createBlockCommit = (block, message, branch, parents, idx) => {
         level: block.error ? LogLevel.ERROR : LogLevel.INFO,
         message: block.error ? "Tool error: " + block.error : "Tool call succeeded"
       };
-    case "summary":
+    case "stats":
       return {
         id: getGenericBlockId(message.id, idx),
         branch,
@@ -801,7 +1603,7 @@ var createBlockCommit = (block, message, branch, parents, idx) => {
         timestamp,
         icon: "ph--rocket--regular",
         level: LogLevel.INFO,
-        message: ContentBlock.createSummaryMessage(block)
+        message: ContentBlock3.createStatsMessage(block)
       };
     case "status":
       return {
@@ -871,48 +1673,49 @@ var stringifyRef = (ref) => {
   return ref.dxn.asEchoDXN()?.echoId ?? ref.dxn.asQueueDXN()?.objectId ?? "";
 };
 var stringifyObject = (obj) => {
-  return Obj3.getLabel(obj) ?? Obj3.getTypename(obj) ?? obj.id;
+  return Obj7.getLabel(obj) ?? Obj7.getTypename(obj) ?? obj.id;
 };
 
 // src/templates/system.ts
 import defaultsDeep from "lodash.defaultsdeep";
-import { Template } from "@dxos/blueprints";
+import { Template as Template2 } from "@dxos/compute";
 
-// raw-loader:/__w/dxos/dxos/packages/core/assistant/src/templates/instructions/format.tpl?raw
+// raw-loader:/__w/dxos/dxos/packages/core/compute/assistant/src/templates/instructions/format.tpl?raw
 var format_default = "{{! System Core }}\n\n## Input Formats\n\n### Blueprints\n\nYou may be provided with Blueprint definitions that include specific instructions and tools for a given set of tasks.\nBlue print instructions will be enclosed in a <blueprint> tag.\n\n### Context objects\n\nYou may be provided with references to objects that are relevant to the current prompt.\nObjects will be in the form of a <object> tag with a <dxn> element that is the unique key of the object required by many tools.\nThe <typename> element is the type of the object.\n\n<object>\n  <dxn>object-id</dxn>\n  <typename>example.com/type/TypeName</typename>\n</object>\n\n## Response Formats\n\nIt is very important to respond in the correct format.\nYou response must be valid Markdown.\nIn some cases you should respond with well-formed XML tags along with the content.\nCode snippets should use 2-space indents.\n\n{{#if cot}}\nBefore responding, explain your reasoning and include your detailed chain-of-thought enclosed in a <cot> tag.\nThe <cot> tag should be the first thing in your response.\n{{/if}}\n\n## Block references\n\nYou can reply with a block reference for objects if you have the DXN. For example:\n\n<object>\n  <dxn>dxn:queue:data:B6INSIBY3CBEF4M5VZRYBCMAHQMPYK5AJ:01K24XMVHSZHS97SG1VTVQDM5Z:01K24XPK464FSCKVQJAB2H662M</dxn>\n</object>\n<object>\n  <dxn>dxn:echo:@:01K24XPK464FSCKVQJAB2H662M</dxn>\n</object>\n\n### Suggestions\n\nEach suggestion must in the form of a proposed action enclosed in a <suggestion> tag.\nExample: <suggestion>Show on a map</suggestion>\nDo not use bullets when listing suggestions.\n\n### Multiple choice questions\n\nYou may ask multiple choice questions by wrapping consise options inside <option> tags inside a well formed <select> tag.\nExample: <select><option>Yes</option><option>No</option></select>\n\n## Context\n\nThe current date and time is {{DATETIME}}.\n";
 
-// raw-loader:/__w/dxos/dxos/packages/core/assistant/src/templates/instructions/system.tpl?raw
-var system_default = "{{! System Prompt }}\n\n- You are a helpful assistant called Kai. \n- Your name is Kai and you are powered by different language models.\n- You were created by DXOS and operate inside of Composer and on the DXOS EDGE network.\n- In your initial greeting state your name and make some suggestions based on the current context objects.\n- You are an advanced AI assistant capable of creating and managing artifacts from provided data and tools.\n- Your task is to process user commands and questions and decide how best to respond.\n- In general be concise and direct.\n- Follow all instructions carefully.\n\n## Planning\n\n- Analyze the structure and type of the content in the user's message.\n- Determine if you complete the task using the available blueprint definitions?\n- If you can't complete the task using the available blueprint definitions, query the list of available blueprint definitions using the appropriate tool.\n- Identify which blueprint definitions are relevant to the user's request.\n- Evaluate the potential benefits of creating an artifact vs. normal processing for each identified element.\n- Make a final decision on whether to create an artifact and explain your reasoning.\n- Are the required blueprint definitions already available?\n- If not, select which blueprint definition(s) will be the most relevant and require them using the require_artifact_definitions tool.\n- The require'd artifact tools will be available for use after require.\n- If creating an artifact, outline how you will structure it within the response.\n- Decide if the artifact needs to be shown to the user.\n- Call the show tool to show the artifact to the user.\n- Your reasoning must include: whether to use artifacts or not, to create one or query, whether to show the artifact to the user, and how to structure the response.\n\n## Blueprints and Artifacts\n\n- Determine if the interaction involves an artifact. Prefer artifacts for tables, lists, spreadsheets, kanbans, games, images, and other structured data.\n- Determine if the user is explicitly talking about creating a new artifact, or wants to use an existing artifact.\n- If it is ambiguous, query for existing artifacts first and then decide.\n- If you decide to create an artifact, call the associated tool to create the artifact.\n- Artifacts are stored in the database. Tools are used to create and query artifacts.\n- If you are unsure about creating an artifact ask the user for clarification.\n- Artifacts are mutable objects that can change over the course of the conversation.\n- Always re-query the artifact using the tool (like query or inspect) to get the latest state of the artifact before answering the user.\n- You must never generate the id of the artifact; only recall the ids that are already in the history.\n- Artifacts are created by requiring the specific artifact using the require_artifact tool and creating it by calling the associated tool.\n- You can add suggested actions at the end of your response.\n- Suggested actions should be very concise and start with a verb and be phrased as a command to an agent -- not a question to the user.\n- Suggested actions must be in the form of a user instruction that you can follow.\n- Suggested actions could include actions that create artifacts.\n- After creating an artifact use a tool to add it to the chat context.\n\n## Context\n\nToday is {{DATETIME}}.\n";
+// raw-loader:/__w/dxos/dxos/packages/core/compute/assistant/src/templates/instructions/system.tpl?raw
+var system_default = "{{! System Prompt }}\n\n- You are a helpful assistant called Kai. \n- Your name is Kai and you are powered by different language models.\n- You were created by DXOS and operate inside of Composer and on the DXOS EDGE network.\n- In your initial greeting state your name and make some suggestions based on the current context objects.\n- You are an advanced AI assistant capable of creating and managing artifacts from provided data and tools.\n- Your task is to process user commands and questions and decide how best to respond.\n- In general be concise and direct.\n- Follow all instructions carefully.\n\n## Planning\n\n- Analyze the structure and type of the content in the user's message.\n- Determine if you complete the task using the available blueprint definitions?\n- If you can't complete the task using the available blueprint definitions, query the list of available blueprint definitions using the appropriate tool.\n- Identify which blueprint definitions are relevant to the user's request.\n- Evaluate the potential benefits of creating an artifact vs. normal processing for each identified element.\n- Make a final decision on whether to create an artifact and explain your reasoning.\n- Are the required blueprint definitions already available?\n- If not, select which blueprint definition(s) will be the most relevant and require them using the require_artifact_definitions tool.\n- The require'd artifact tools will be available for use after require.\n- If creating an artifact, outline how you will structure it within the response.\n- Decide if the artifact needs to be shown to the user.\n- Call the show tool to show the artifact to the user.\n- Your reasoning must include: whether to use artifacts or not, to create one or query, whether to show the artifact to the user, and how to structure the response.\n\n## Blueprints and Artifacts\n\n- Determine if the interaction involves an artifact. Prefer artifacts for tables, lists, spreadsheets, kanbans, games, images, and other structured data.\n- Determine if the user is explicitly talking about creating a new artifact, or wants to use an existing artifact.\n- If it is ambiguous, query for existing artifacts first and then decide.\n- If you decide to create an artifact, call the associated tool to create the artifact.\n- Artifacts are stored in the database. Tools are used to create and query artifacts.\n- If you are unsure about creating an artifact ask the user for clarification.\n- Artifacts are mutable objects that can change over the course of the conversation.\n- Always re-query the artifact using the tool (like query or inspect) to get the latest state of the artifact before answering the user.\n- You must never generate the id of the artifact; only recall the ids that are already in the history.\n- Artifacts are created by requiring the specific artifact using the require_artifact tool and creating it by calling the associated tool.\n- You can add suggested actions at the end of your response.\n- Suggested actions should be very concise and start with a verb and be phrased as a command to an agent -- not a question to the user.\n- Suggested actions must be in the form of a user instruction that you can follow.\n- Suggested actions could include actions that create artifacts.\n- After creating an artifact use a tool to add it to the chat context.\n\n## Navigation\n\n- When the user asks to open, go to, navigate to, or show a specific document, page, or object, use the navigation tools instead of loading content inline.\n- First, call the Resolve navigation targets tool to find the navigation path for the requested item.\n  - Without arguments, it returns available pages that can be navigated to.\n  - With a DXN in the query, it resolves that specific object to a navigation path.\n- If you know the object's DXN (e.g. from a database query or context objects), pass it in the query to get the exact navigation path.\n- Then, call the Open tool with the resolved navigation path(s) in the subject array to navigate the user to that item.\n- Do not respond with the document's content when the user asks to open it \u2014 open it in the main content area instead.\n- If multiple targets match, present the options to the user and let them choose which to open.\n\n## Response format\n\n- NEVER output narration text outside `<status>` tags or the final user-facing answer. Any progress update, transition sentence, or \"let me\u2026\" lead-in MUST be wrapped in `<status>...</status>`. Violating this is an error.\n- When updating the user about the progress of the work you are doing, put the update in a <status> XML tag.\n- Only when you need to show the result to the user or ask a question use a text block without status tags.\n- After a `</status>` tag, the next output must be either another `<status>` tag, a tool call, or the final user-facing answer \u2014 never a transition sentence.\n\n<example>\n// reasoning...\n<status>Looking for emails from Depot</status>\n// tool call..\n<status>Found 2 emails from Depot.</status>\n<status>Loading email details</status>\n// tool call\n<status>Located mailbox</status>\nI found **2 emails from Depot**:\n</example>\n\nDo not do (no status tags):\n\n<incorrect_example>\n// reasoning...\n// tool call..\nI found 2 emails from Depot. Let me load their details:\n// tool call\nI found **2 emails from Depot**:\n</incorrect_example>\n\nDo not do (transition sentence after a status tag, before tool calls):\n\n<incorrect_example>\n<status>Doing X</status>\nLet me start by doing X:\n// tool call\n</incorrect_example>\n\n- NEVER write a lead-in sentence before a tool call. Use a `<status>` tag instead.\n- Avoid using plain text to tell the user what you are about to do; instead, use status blocks.\n- WRONG: Let me get...\n- CORRECT: <status>Getting...</status>";
 
 // src/templates/system.ts
-var createSystemPrompt = (options = {}) => Template.process(format_default, defaultsDeep(options, {
+var createSystemPrompt = (options = {}) => Template2.process(format_default, defaultsDeep(options, {
   DATETIME: (/* @__PURE__ */ new Date()).toISOString()
 }));
+var SYSTEM_ID = "01K2J2FZWH39F55CS7B26GP1Q3";
 var templates = {
-  /** Editable system prompt.*/
-  system: Template.make({
-    /**
-    * NOTE: We are forbidden from accessing randomness at model level in cloudflare, so we have to have pre-generated ids.
-    */
-    id: "01K2J2FZWH39F55CS7B26GP1Q3",
+  system: Template2.make({
+    id: SYSTEM_ID,
     source: system_default
   })
 };
 export {
+  AgentRequestBegin,
+  AgentRequestEnd,
   AiAssistantError,
   AiContextBinder,
   AiContextService,
-  AiConversation,
-  AiConversationService,
   AiModelError,
+  AiRequest,
   AiSession,
+  AiSessionService,
   ArtifactDiffResolver,
   ArtifactId,
   Bindings,
+  CompleteBlock,
   ContextBinding,
   ExecutionGraph,
   GenerationObserver,
-  GenericToolkit_exports as GenericToolkit,
+  McpServerError,
+  PartialBlock,
   RefFromLLM,
   ToolExecutionServices,
   VersionPin,
@@ -923,8 +1726,10 @@ export {
   createToolkit,
   formatSystemPrompt,
   formatUserPrompt,
-  makeToolExecutionServiceFromFunctions,
-  makeToolResolverFromFunctions,
+  getOperationFromTool,
+  makeToolExecutionService,
+  makeToolExecutionServiceFromOperationInvoker,
+  makeToolResolverFromOperations,
   reduceDiffs,
   templates
 };