@dxos/assistant 0.8.4-main.f9ba587 → 0.8.4-main.fcc0d83b33

package/dist/lib/neutral/index.mjs

@@ -0,0 +1,1693 @@
+// 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/compute";
+import { Resource } from "@dxos/context";
+import { DXN, Feed, Obj, Query, Ref, Type } from "@dxos/echo";
+import { assertArgument } from "@dxos/invariant";
+import { log } from "@dxos/log";
+import { ComplexSet, isNonNullable } from "@dxos/util";
+var __dxlog_file = "/__w/dxos/dxos/packages/core/assistant/src/conversation/context.ts";
+var ContextBinding = Schema.Struct({
+  blueprints: Schema.Struct({
+    added: Schema.Array(Ref.Ref(Blueprint.Blueprint)),
+    removed: Schema.Array(Ref.Ref(Blueprint.Blueprint))
+  }),
+  objects: Schema.Struct({
+    added: Schema.Array(Ref.Ref(Obj.Unknown)),
+    removed: Schema.Array(Ref.Ref(Obj.Unknown))
+  })
+}).pipe(Type.object({
+  typename: "org.dxos.type.contextBinding",
+  version: "0.1.0"
+}));
+var Bindings = class {
+  blueprints = new ComplexSet((ref) => ref.dxn.toString());
+  // TODO(burdon): Some DXNs have the Space prefix so only compare the object ID.
+  objects = new ComplexSet((ref) => ref.dxn.asEchoDXN()?.echoId);
+  toJSON() {
+    return {
+      blueprints: EArray.fromIterable(this.blueprints).map((ref) => ref.dxn.toString()),
+      objects: EArray.fromIterable(this.objects).map((ref) => ref.dxn.toString())
+    };
+  }
+};
+var AiContextBinder = class extends Resource {
+  _blueprints = Atom.make([]).pipe(Atom.keepAlive);
+  _objects = Atom.make([]).pipe(Atom.keepAlive);
+  _registry;
+  _feed;
+  _runtime;
+  #bindingsQuery;
+  constructor(options) {
+    super();
+    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();
+  }
+  /**
+  * Returns the blueprints atom for subscription.
+  */
+  get blueprints() {
+    return this._blueprints;
+  }
+  /**
+  * Returns the objects atom for subscription.
+  */
+  get objects() {
+    return this._objects;
+  }
+  /**
+  * Gets the current blueprints value.
+  */
+  getBlueprints() {
+    return this._registry.get(this._blueprints);
+  }
+  /**
+  * Gets the current objects value.
+  */
+  getObjects() {
+    return this._registry.get(this._objects);
+  }
+  /**
+  * Subscribe to changes in blueprints.
+  */
+  subscribeBlueprints(cb) {
+    return this._registry.subscribe(this._blueprints, () => cb(this._registry.get(this._blueprints)));
+  }
+  /**
+  * Subscribe to changes in objects.
+  */
+  subscribeObjects(cb) {
+    return this._registry.subscribe(this._objects, () => cb(this._registry.get(this._objects)));
+  }
+  async _open() {
+    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(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 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: mergedBlueprints.length,
+      objects: mergedObjects.length
+    }, { "~LogMeta": "~LogMeta", F: __dxlog_file, L: 155, S: this });
+  }
+  async _close() {
+    this._registry.set(this._blueprints, []);
+    this._registry.set(this._objects, []);
+  }
+  async bind({ blueprints, objects }) {
+    const currentBlueprints = this._registry.get(this._blueprints);
+    const currentObjects = this._registry.get(this._objects);
+    const { added: addedBlueprints, next: nextBlueprints } = this._processBindings(blueprints, currentBlueprints);
+    const { added: addedObjects, next: nextObjects } = this._processBindings(objects, currentObjects);
+    if (!addedBlueprints.length && !addedObjects.length) {
+      return;
+    }
+    this._registry.set(this._blueprints, nextBlueprints);
+    this._registry.set(this._objects, nextObjects);
+    log("bind", {
+      blueprints: addedBlueprints.length,
+      objects: addedObjects.length
+    }, { "~LogMeta": "~LogMeta", F: __dxlog_file, L: 176, S: this });
+    await Runtime.runPromise(this._runtime)(Feed.append(this._feed, [
+      Obj.make(ContextBinding, {
+        blueprints: {
+          added: addedBlueprints,
+          removed: []
+        },
+        objects: {
+          added: addedObjects,
+          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
+    }, { "~LogMeta": "~LogMeta", F: __dxlog_file, L: 208, S: this });
+    await Runtime.runPromise(this._runtime)(Feed.append(this._feed, [
+      Obj.make(ContextBinding, {
+        blueprints: {
+          added: [],
+          removed: blueprints ?? []
+        },
+        objects: {
+          added: [],
+          removed: objects ?? []
+        }
+      })
+    ]));
+  }
+  /**
+  * Process bindings to filter duplicates and determine next state.
+  */
+  _processBindings(refs, current) {
+    const next = [
+      ...current
+    ];
+    const added = [];
+    if (!refs?.length) {
+      return {
+        added,
+        next
+      };
+    }
+    const seen = new Set(current.map((obj) => Obj.getDXN(obj).toString()));
+    for (const ref of refs) {
+      const dxn = ref.dxn.toString();
+      if (!seen.has(dxn)) {
+        seen.add(dxn);
+        added.push(ref);
+        if (ref.isAvailable) {
+          const target = ref.target;
+          if (target) {
+            next.push(target);
+          }
+        }
+      }
+    }
+    return {
+      added,
+      next
+    };
+  }
+  /**
+  * Reduce results into sets of blueprints and objects.
+  */
+  _reduce(items) {
+    return Function.pipe(items, EArray.reduce(new Bindings(), (context, { blueprints, objects }) => {
+      blueprints.added.forEach((ref) => context.blueprints.add(ref));
+      blueprints.removed.forEach((ref) => context.blueprints.delete(ref));
+      objects.added.forEach((ref) => context.objects.add(ref));
+      objects.removed.forEach((ref) => {
+        for (const obj of context.objects) {
+          if (DXN.equalsEchoId(obj.dxn, ref.dxn)) {
+            context.objects.delete(obj);
+          }
+        }
+      });
+      return context;
+    }));
+  }
+  /**
+  * 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) {
+    const refArray = [
+      ...refs
+    ];
+    await Promise.all(refArray.map((ref) => ref.tryLoad()));
+    return refArray.map((ref) => {
+      let target;
+      if (ref.isAvailable) {
+        target = ref.target;
+      }
+      return target ?? current.find((obj) => Obj.getDXN(obj).toString() === ref.dxn.toString());
+    }).filter(isNonNullable);
+  }
+};
+var AiContextService = class _AiContextService extends Context.Tag("@dxos/assistant/AiContextService")() {
+  static bindContext = ({ blueprints, objects }) => Effect.gen(function* () {
+    const { binder } = yield* _AiContextService;
+    yield* Effect.promise(() => binder.bind({
+      blueprints,
+      objects
+    }));
+  });
+  static findObjects = (type) => {
+    return Effect.gen(function* () {
+      const { binder } = yield* _AiContextService;
+      return binder.getObjects().filter(Obj.instanceOf(type));
+    });
+  };
+};
+
+// 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 } 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 * 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/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
+});
+
+// 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";
+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/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/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) {
+      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);
+  /**
+  * 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: 109, 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: 140, 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: 154, 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: 174, 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: 237, 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/assistant/src/conversation/session.ts";
+var SUMMARY_THRESHOLD = 8e4;
+var AiSession = class extends Resource2 {
+  /**
+  * Blueprints and objects bound to the session.
+  */
+  _binder;
+  _feed;
+  _runtime;
+  constructor(options) {
+    super();
+    this._feed = options.feed;
+    this._runtime = options.runtime;
+    invariant2(this._feed, void 0, { "~LogMeta": "~LogMeta", F: __dxlog_file5, L: 40, S: this, A: ["this._feed", ""] });
+    invariant2(this._runtime, void 0, { "~LogMeta": "~LogMeta", F: __dxlog_file5, L: 41, S: this, A: ["this._runtime", ""] });
+    this._binder = new AiContextBinder({
+      feed: this._feed,
+      runtime: this._runtime,
+      registry: options.registry
+    });
+  }
+  async _open() {
+    await this._binder.open(this._ctx);
+  }
+  get feed() {
+    return this._feed;
+  }
+  get context() {
+    return this._binder;
+  }
+  async getHistory() {
+    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 Effect7.gen(this, function* () {
+      const toolkit = yield* createToolkit({
+        blueprints: this.context.getBlueprints()
+      });
+      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 Effect7.gen(this, function* () {
+      const history = yield* Effect7.promise(() => this.getHistory());
+      const blueprints = this.context.getBlueprints();
+      const objects = this.context.getObjects();
+      log4("run", {
+        history: history.length,
+        blueprints: blueprints.length,
+        objects: objects.length
+      }, { "~LogMeta": "~LogMeta", F: __dxlog_file5, L: 84, 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
+        ])))
+      });
+      yield* request.begin({
+        history,
+        blueprints,
+        objects,
+        prompt: params.prompt,
+        system: params.system
+      });
+      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: 113, 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: 132, 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 AiSessionService = class _AiSessionService extends Context4.Tag("@dxos/assistant/AiSessionService")() {
+  /**
+  * Create a new session layer from options.
+  */
+  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 session with a new feed.
+  */
+  static layerNewFeed = (options) => Layer3.unwrapScoped(Effect7.gen(function* () {
+    const feed = Feed2.make();
+    yield* Database2.add(feed);
+    const runtime2 = yield* Effect7.runtime();
+    return _AiSessionService.layer({
+      ...options,
+      feed,
+      runtime: runtime2
+    });
+  }));
+  /**
+  * Run a prompt in the current session.
+  */
+  static run = (params) => Effect7.gen(function* () {
+    const session = yield* _AiSessionService;
+    return yield* session.createRequest(params);
+  });
+};
+var aiContextFromSession = Layer3.effect(AiContextService, Effect7.gen(function* () {
+  const session = yield* AiSessionService;
+  return {
+    binder: session.context
+  };
+}));
+var connectMcpServers = (blueprints, spaceMcpServers = []) => {
+  const blueprintServers = pipe4(blueprints, Array4.flatMap((_) => _.mcpServers ?? []), Array4.map(({ url, protocol }) => ({
+    url,
+    kind: protocol
+  })));
+  const spaceServers = spaceMcpServers.map(({ url, protocol, apiKey }) => ({
+    url,
+    kind: 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: 196, S: void 0 })),
+    Effect7.tapDefect((error) => Effect7.sync(() => log4.warn("Failed to connect to MCP server", {
+      error
+    }, { "~LogMeta": "~LogMeta", F: __dxlog_file5, L: 199, S: void 0 }))),
+    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 log5 } from "@dxos/log";
+import { isNonNullable as isNonNullable2 } from "@dxos/util";
+var __dxlog_file6 = "/__w/dxos/dxos/packages/core/assistant/src/diff.ts";
+var reduceDiffs = (diffs) => {
+  return diffs.reduce((acc, diff) => {
+    const current = acc[acc.length - 1];
+    const [op, ...rest] = diff.split(" ");
+    switch (op) {
+      case "-": {
+        if (current && !current.match) {
+          current.match += rest.join(" ");
+        } else {
+          acc.push({
+            match: rest.join(" "),
+            replace: ""
+          });
+        }
+        return acc;
+      }
+      case "+": {
+        if (current && !current.replace) {
+          current.replace += rest.join(" ");
+        } else {
+          acc.push({
+            match: "",
+            replace: rest.join(" ")
+          });
+        }
+        return acc;
+      }
+      default: {
+        throw new Error(`Invalid diff format: each line must start with "- " or "+ ", got: ${JSON.stringify(diff)}`);
+      }
+    }
+  }, []);
+};
+var computeDiffsWithCursors = (accessor, diffs) => {
+  return reduceDiffs(diffs).map((diff) => {
+    const text = DocAccessor.getValue(accessor);
+    const idx = text.indexOf(diff.match);
+    if (idx !== -1) {
+      return {
+        cursor: toCursorRange(accessor, idx, idx + diff.match.length),
+        text: diff.replace
+      };
+    }
+  }).filter(isNonNullable2);
+};
+var applyDiffs = (accessor, diffs, { errorOnNotFound = false } = {}) => {
+  for (const diff of reduceDiffs(diffs)) {
+    accessor.handle.change((doc) => {
+      const text = DocAccessor.getValue(accessor);
+      const idx = text.indexOf(diff.match);
+      if (idx !== -1) {
+        A.splice(doc, accessor.path, idx, diff.match.length, diff.replace);
+      } else {
+        if (errorOnNotFound) {
+          throw new Error(`Diff not found: ${diff.match}`);
+        } else {
+          log5.warn("diff not found", diff, { "~LogMeta": "~LogMeta", F: __dxlog_file6, L: 71, S: void 0 });
+        }
+      }
+    });
+  }
+  return DocAccessor.getValue(accessor);
+};
+
+// src/execution-graph.ts
+import { AgentStatus } from "@dxos/ai";
+import { MESSAGE_PROPERTY_TOOL_CALL_ID } from "@dxos/compute";
+import { Obj as Obj7 } from "@dxos/echo";
+import { LogLevel } from "@dxos/log";
+import { ContentBlock as ContentBlock3, Message as Message4 } from "@dxos/types";
+import { isNonNullable as isNonNullable3 } from "@dxos/util";
+var SKIP_BLOCKS = [
+  "text"
+];
+var ExecutionGraph = class {
+  _skipBlocks;
+  // TODO(dmaretskyi): Evolve the internal state to support chats, functions, circuit workflows, etc.
+  _commits = [];
+  _branchNames = /* @__PURE__ */ new Set();
+  _lastBlockId;
+  // Tool call tracking.
+  _toolCallCommitIds = /* @__PURE__ */ new Map();
+  _lastCommitByBranch = /* @__PURE__ */ new Map();
+  _pendingToolResults = /* @__PURE__ */ new Map();
+  constructor(_skipBlocks = SKIP_BLOCKS) {
+    this._skipBlocks = _skipBlocks;
+  }
+  /**
+  * Adds events to the graph.
+  */
+  addEvents(events) {
+    const sortedEvents = this.sortObjectsByCreated(events);
+    for (const event of sortedEvents) {
+      if (Obj7.instanceOf(Message4.Message, event)) {
+        this._processMessage(event);
+      } else if (Obj7.instanceOf(AgentStatus, event)) {
+        this._processAgentStatus(event);
+      }
+    }
+  }
+  /**
+  * Processes a message event and creates commits for its blocks.
+  */
+  _processMessage(message) {
+    const messageCommits = messageToCommits(message, this._lastBlockId, this._toolCallCommitIds, this._lastCommitByBranch, this._skipBlocks);
+    this._commits.push(...messageCommits);
+    messageCommits.forEach((commit) => this._branchNames.add(commit.branch));
+    this._trackToolCalls(messageCommits);
+    if (messageCommits.length > 0) {
+      this._lastBlockId = messageCommits[messageCommits.length - 1].id;
+    }
+  }
+  /**
+  * Processes an AgentStatus event.
+  */
+  _processAgentStatus(event) {
+    const branch = this._getToolCallBranch(event.parentMessage, event.toolCallId);
+    this._branchNames.add(branch);
+    const agentStatusCommit = {
+      id: event.id,
+      branch,
+      message: event.message,
+      icon: "ph--flag--regular",
+      parents: this._getAgentStatusParents(event),
+      timestamp: new Date(event.created)
+    };
+    this._commits.push(agentStatusCommit);
+    this._lastBlockId = event.id;
+    if (event.toolCallId) {
+      this._lastCommitByBranch.set(branch, event.id);
+      this._updatePendingToolResults(event.toolCallId, event.id);
+    }
+  }
+  /**
+  * Stable sort: objects with 'created' field are sorted by it, others remain in place.
+  */
+  sortObjectsByCreated(objects) {
+    const sortedObjects = objects.slice();
+    const createdObjects = [];
+    sortedObjects.forEach((obj, idx) => {
+      if (obj && typeof obj.created === "string") {
+        createdObjects.push({
+          idx,
+          obj
+        });
+      }
+    });
+    createdObjects.sort((a, b) => {
+      const aDate = new Date(a.obj.created).getTime();
+      const bDate = new Date(b.obj.created).getTime();
+      return aDate - bDate;
+    });
+    let createdIdx = 0;
+    return sortedObjects.map((obj, _idx) => {
+      if (obj && typeof obj.created === "string") {
+        return createdObjects[createdIdx++].obj;
+      }
+      return obj;
+    });
+  }
+  /**
+  * Returns the current state of the graph.
+  */
+  getGraph(lastRequest = false) {
+    const idx = lastRequest ? this._commits.findLastIndex((c) => c.tags?.includes("user")) : -1;
+    const commits = idx === -1 ? this._commits : this._commits.slice(idx);
+    return {
+      branches: Array.from(this._branchNames),
+      commits
+    };
+  }
+  /**
+  * Tracks tool calls and handles pending tool result updates.
+  */
+  _trackToolCalls(commits) {
+    commits.forEach((commit) => {
+      if (this._isToolCallCommit(commit)) {
+        const toolCallId = this._extractToolCallId(commit.id);
+        if (toolCallId) {
+          this._toolCallCommitIds.set(toolCallId, commit.id);
+        }
+      }
+      if (this._isToolResultCommit(commit)) {
+        const toolCallId = this._extractToolCallId(commit.id);
+        if (toolCallId) {
+          this._pendingToolResults.set(toolCallId, commit.id);
+          this._handleLateToolResultUpdate(toolCallId, commit.id);
+        }
+      }
+      this._lastCommitByBranch.set(commit.branch, commit.id);
+    });
+  }
+  /**
+  * Updates pending tool results to point to an AgentStatus event.
+  */
+  _updatePendingToolResults(toolCallId, agentStatusId) {
+    const toolResultId = this._pendingToolResults.get(toolCallId);
+    if (!toolResultId) {
+      return;
+    }
+    const toolResultCommit = this._commits.find((c) => c.id === toolResultId);
+    if (!toolResultCommit) {
+      return;
+    }
+    const toolCallCommitId = this._toolCallCommitIds.get(toolCallId);
+    if (!toolCallCommitId) {
+      return;
+    }
+    const toolCallIndex = toolResultCommit.parents.indexOf(toolCallCommitId);
+    if (toolCallIndex !== -1) {
+      toolResultCommit.parents[toolCallIndex] = agentStatusId;
+    }
+    this._pendingToolResults.delete(toolCallId);
+  }
+  /**
+  * Handles late updates when AgentStatus is processed before tool result.
+  */
+  _handleLateToolResultUpdate(toolCallId, toolResultId) {
+    const toolCallCommitId = this._toolCallCommitIds.get(toolCallId);
+    if (!toolCallCommitId) {
+      return;
+    }
+    const messageId = toolCallCommitId.split("_toolCall_")[0];
+    const toolCallBranch = `${messageId}_${toolCallId}`;
+    const agentStatusCommitId = this._lastCommitByBranch.get(toolCallBranch);
+    if (agentStatusCommitId && agentStatusCommitId !== toolCallCommitId) {
+      const toolResultCommit = this._commits.find((c) => c.id === toolResultId);
+      if (toolResultCommit) {
+        const toolCallIndex = toolResultCommit.parents.indexOf(toolCallCommitId);
+        if (toolCallIndex !== -1) {
+          toolResultCommit.parents[toolCallIndex] = agentStatusCommitId;
+        }
+      }
+      this._pendingToolResults.delete(toolCallId);
+    }
+  }
+  /**
+  * Gets the branch name for a tool call.
+  */
+  _getToolCallBranch(parentMessage, toolCallId) {
+    if (toolCallId) {
+      return toolCallId;
+    }
+    return parentMessage || "main";
+  }
+  /**
+  * Gets the parents for an AgentStatus commit.
+  */
+  _getAgentStatusParents(event) {
+    if (event.parentMessage && event.toolCallId) {
+      return [
+        getToolCallId(event.parentMessage, event.toolCallId)
+      ];
+    }
+    return [];
+  }
+  /**
+  * Checks if a commit is a tool call commit.
+  */
+  _isToolCallCommit(commit) {
+    return commit.message.includes("Calling tool");
+  }
+  /**
+  * Checks if a commit is a tool result commit.
+  */
+  _isToolResultCommit(commit) {
+    return commit.message.includes("Tool call succeeded") || commit.message.includes("Tool error:");
+  }
+  /**
+  * Extracts tool call ID from a commit ID.
+  */
+  _extractToolCallId(commitId) {
+    const toolCallMatch = commitId.match(/_toolCall_(.+)$/);
+    const toolResultMatch = commitId.match(/_toolResult_(.+)$/);
+    return toolCallMatch?.[1] || toolResultMatch?.[1];
+  }
+};
+var messageToCommits = (message, lastBlockId, toolCallIds, lastCommitByBranch, skipBlocks) => {
+  let previousBlockId = lastBlockId;
+  return message.blocks.map((block, idx) => {
+    const branch = getMessageBranch(message);
+    const parents = getBlockParents(block, previousBlockId, message, toolCallIds, lastCommitByBranch);
+    if (skipBlocks?.includes(block._tag)) {
+      return null;
+    }
+    const commit = createBlockCommit(block, message, branch, parents, idx);
+    if (commit) {
+      previousBlockId = commit.id;
+    }
+    return commit;
+  }).filter(isNonNullable3);
+};
+var getBlockParents = (block, previousBlockId, message, toolCallIds, lastCommitByBranch) => {
+  const parents = [];
+  if (block._tag === "toolResult") {
+    if (previousBlockId) {
+      parents.push(previousBlockId);
+    }
+    const toolCallCommitId = toolCallIds?.get(block.toolCallId);
+    if (toolCallCommitId) {
+      const toolCallBranch = getToolCallBranchFromCommitId(toolCallCommitId, block.toolCallId);
+      const lastBlockFromToolCallBranch = lastCommitByBranch?.get(toolCallBranch);
+      if (lastBlockFromToolCallBranch) {
+        parents.push(lastBlockFromToolCallBranch);
+      } else {
+        parents.push(toolCallCommitId);
+      }
+    }
+  } else {
+    if (previousBlockId) {
+      parents.push(previousBlockId);
+    } else {
+      const toolCallParent = getParentId(message);
+      if (toolCallParent) {
+        parents.push(toolCallParent);
+      }
+    }
+  }
+  return parents;
+};
+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;
+      }
+      return {
+        id: getGenericBlockId(message.id, idx),
+        branch,
+        parents,
+        timestamp,
+        ...message.sender.role === "user" ? {
+          icon: "ph--user--regular",
+          tags: [
+            "user"
+          ],
+          message: "Processing request..."
+        } : {
+          icon: "ph--drone--regular",
+          message: `Response (${block.text.split(" ").length} words)`
+        }
+      };
+    case "toolCall":
+      return {
+        id: getToolCallId(message.id, block.toolCallId),
+        branch,
+        parents,
+        timestamp,
+        icon: "ph--wrench--regular",
+        level: LogLevel.INFO,
+        message: `Calling tool (${block.name})`
+      };
+    case "toolResult":
+      return {
+        id: getToolResultId(message.id, block.toolCallId),
+        branch,
+        parents,
+        timestamp,
+        icon: block.error ? "ph--x-circle--regular" : "ph--check-circle--regular",
+        level: block.error ? LogLevel.ERROR : LogLevel.INFO,
+        message: block.error ? "Tool error: " + block.error : "Tool call succeeded"
+      };
+    case "stats":
+      return {
+        id: getGenericBlockId(message.id, idx),
+        branch,
+        parents,
+        timestamp,
+        icon: "ph--rocket--regular",
+        level: LogLevel.INFO,
+        message: ContentBlock3.createStatsMessage(block)
+      };
+    case "status":
+      return {
+        id: getGenericBlockId(message.id, idx),
+        branch,
+        parents,
+        timestamp,
+        message: block.statusText,
+        level: LogLevel.INFO,
+        icon: "ph--flag--regular"
+      };
+    case "reasoning":
+      return {
+        id: getGenericBlockId(message.id, idx),
+        branch,
+        parents,
+        timestamp,
+        message: block.reasoningText ?? "Thinking...",
+        icon: "ph--brain--regular"
+      };
+    case "reference":
+      return {
+        id: getGenericBlockId(message.id, idx),
+        branch,
+        parents,
+        timestamp,
+        icon: "ph--link--regular",
+        message: stringifyRef(block.reference)
+      };
+    default:
+      return null;
+  }
+};
+var getToolCallBranchFromCommitId = (toolCallCommitId, toolCallId) => {
+  const messageId = toolCallCommitId.split("_toolCall_")[0];
+  return `${messageId}_${toolCallId}`;
+};
+var getToolCallId = (messageId, toolCallId) => `${messageId}_toolCall_${toolCallId}`;
+var getToolResultId = (messageId, toolCallId) => `${messageId}_toolResult_${toolCallId}`;
+var getGenericBlockId = (messageId, idx) => `${messageId}_block_${idx}`;
+var getBranchName = (options) => {
+  if (options.toolCallId) {
+    return options.toolCallId;
+  } else if (options.parentMessage) {
+    return options.parentMessage;
+  } else {
+    return "main";
+  }
+};
+var getMessageBranch = (message) => {
+  return getBranchName({
+    parentMessage: message.parentMessage,
+    toolCallId: message.properties?.[MESSAGE_PROPERTY_TOOL_CALL_ID]
+  });
+};
+var getParentId = (message) => {
+  if (message.parentMessage && message.properties?.[MESSAGE_PROPERTY_TOOL_CALL_ID]) {
+    return getToolCallId(message.parentMessage, message.properties[MESSAGE_PROPERTY_TOOL_CALL_ID]);
+  } else {
+    return void 0;
+  }
+};
+var stringifyRef = (ref) => {
+  if (ref.target) {
+    return stringifyObject(ref.target);
+  }
+  return ref.dxn.asEchoDXN()?.echoId ?? ref.dxn.asQueueDXN()?.objectId ?? "";
+};
+var stringifyObject = (obj) => {
+  return Obj7.getLabel(obj) ?? Obj7.getTypename(obj) ?? obj.id;
+};
+
+// src/templates/system.ts
+import defaultsDeep from "lodash.defaultsdeep";
+import { Template as Template2 } from "@dxos/compute";
+
+// raw-loader:/__w/dxos/dxos/packages/core/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## 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- 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\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\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 = {}) => Template2.process(format_default, defaultsDeep(options, {
+  DATETIME: (/* @__PURE__ */ new Date()).toISOString()
+}));
+var SYSTEM_ID = "01K2J2FZWH39F55CS7B26GP1Q3";
+var templates = {
+  system: Template2.make({
+    id: SYSTEM_ID,
+    source: system_default
+  })
+};
+export {
+  AgentRequestBegin,
+  AgentRequestEnd,
+  AiAssistantError,
+  AiContextBinder,
+  AiContextService,
+  AiModelError,
+  AiRequest,
+  AiSession,
+  AiSessionService,
+  ArtifactDiffResolver,
+  ArtifactId,
+  Bindings,
+  CompleteBlock,
+  ContextBinding,
+  ExecutionGraph,
+  GenerationObserver,
+  PartialBlock,
+  RefFromLLM,
+  ToolExecutionServices,
+  VersionPin,
+  applyDiffs,
+  computeDiffsWithCursors,
+  createArtifactElement,
+  createSystemPrompt,
+  createToolkit,
+  formatSystemPrompt,
+  formatUserPrompt,
+  getOperationFromTool,
+  makeToolExecutionService,
+  makeToolExecutionServiceFromOperationInvoker,
+  makeToolResolverFromOperations,
+  reduceDiffs,
+  templates
+};
+//# sourceMappingURL=index.mjs.map