@dxos/assistant 0.8.4-main.21d9917 → 0.8.4-main.2244d791bb

package/dist/lib/{browser/chunk-7CDDG6LL.mjs → neutral/chunk-QBD6NQJK.mjs}

@@ -45,18 +45,6 @@ var AiModelError = class extends BaseError.extend("AiModelError", "AI model erro
 var AiAssistantError = class extends BaseError.extend("AiAssistantError", "AI assistant error") {
 };
 
-// src/session/session.ts
-import * as LanguageModel from "@effect/ai/LanguageModel";
-import * as Chunk from "effect/Chunk";
-import * as Effect3 from "effect/Effect";
-import * as Stream from "effect/Stream";
-import { AiParser, AiPreprocessor, callTool, getToolCalls } from "@dxos/ai";
-import { todo } from "@dxos/debug";
-import { Obj as Obj2 } from "@dxos/echo";
-import { TracingService } from "@dxos/functions";
-import { log as log2 } from "@dxos/log";
-import { Message as Message2 } from "@dxos/types";
-
 // src/session/format.ts
 import * as Effect2 from "effect/Effect";
 import * as Function2 from "effect/Function";
@@ -177,6 +165,16 @@ ${diff}` : ""}</changed-artifact>`).join("\n")}
 };
 
 // src/session/session.ts
+import * as LanguageModel from "@effect/ai/LanguageModel";
+import * as Chunk from "effect/Chunk";
+import * as Effect3 from "effect/Effect";
+import * as Stream from "effect/Stream";
+import { AiParser, AiPreprocessor, callTool, withoutToolCallParising } from "@dxos/ai";
+import { todo } from "@dxos/debug";
+import { Obj as Obj2 } from "@dxos/echo";
+import { TracingService } from "@dxos/functions";
+import { log as log2 } from "@dxos/log";
+import { Message as Message2 } from "@dxos/types";
 var __dxlog_file2 = "/__w/dxos/dxos/packages/core/assistant/src/session/session.ts";
 var AiSession = class {
   _options;
@@ -199,7 +197,7 @@ var AiSession = class {
   get toolCalls() {
     return this._toolCalls;
   }
-  run = ({ prompt, system: systemTemplate, history = [], objects = [], blueprints = [], toolkit, observer = GenerationObserver.noop() }) => Effect3.gen(this, function* () {
+  run = ({ prompt, system: systemTemplate, history = [], objects = [], blueprints = [], toolkit, observer = GenerationObserver.noop(), onOutput = () => Effect3.void }) => Effect3.gen(this, function* () {
     this._started = Date.now();
     this._history = [
       ...history
@@ -210,6 +208,7 @@ var AiSession = class {
       pending.push(message);
       yield* observer.onMessage(message);
       yield* TracingService.emitConverationMessage(message);
+      yield* onOutput(message);
       return message;
     });
     const promptMessage = yield* submitMessage(yield* formatUserPrompt({
@@ -233,7 +232,7 @@ var AiSession = class {
         objects: objects?.length ?? 0
       }, {
         F: __dxlog_file2,
-        L: 126,
+        L: 132,
         S: this,
         C: (f, a) => f(...a)
       });
@@ -241,13 +240,16 @@ var AiSession = class {
         ...this._history,
         ...this._pending
       ], {
-        system
+        system,
+        cacheControl: "ephemeral"
       });
-      const blocks = yield* LanguageModel.streamText({
+      const messages = yield* LanguageModel.streamText({
         prompt: prompt2,
         toolkit,
         disableToolCallResolution: true
       }).pipe(
+        // Disable tool call parsing so that we can handle parameter schema errors on a per-tool call basis.
+        withoutToolCallParising,
         // TOOD(dmaretskyi): Error mapping.
         AiParser.parseResponse({
           onBegin: () => observer.onBegin(),
@@ -255,27 +257,34 @@ var AiSession = class {
           onPart: (part) => observer.onPart(part),
           onEnd: (summary) => observer.onEnd(summary)
         }),
+        Stream.mapEffect((block) => Effect3.gen(function* () {
+          return yield* submitMessage(Obj2.make(Message2.Message, {
+            created: (/* @__PURE__ */ new Date()).toISOString(),
+            sender: {
+              role: "assistant"
+            },
+            blocks: [
+              block
+            ]
+          }));
+        })),
         Stream.runCollect,
         Effect3.map(Chunk.toArray)
       );
-      const response = yield* submitMessage(Obj2.make(Message2.Message, {
-        created: (/* @__PURE__ */ new Date()).toISOString(),
-        sender: {
-          role: "assistant"
-        },
-        blocks
-      }));
-      const toolCalls = getToolCalls(response);
+      const toolCalls = messages.flatMap((message) => message.blocks.filter((block) => block._tag === "toolCall" && block.providerExecuted === false).map((block) => ({
+        block,
+        message
+      })));
       if (toolCalls.length === 0) {
         break;
       } else if (!toolkit) {
         throw new Error("No toolkit provided");
       }
-      const toolResults = yield* Effect3.forEach(toolCalls, (toolCall) => {
-        return callTool(toolkit, toolCall).pipe(Effect3.provide(TracingService.layerSubframe((context) => ({
+      const toolResults = yield* Effect3.forEach(toolCalls, ({ block, message }) => {
+        return callTool(toolkit, block).pipe(Effect3.provide(TracingService.layerSubframe((context) => ({
           ...context,
-          parentMessage: response.id,
-          toolCallId: toolCall.toolCallId
+          parentMessage: message.id,
+          toolCallId: block.toolCallId
         }))));
       });
       yield* submitMessage(Obj2.make(Message2.Message, {
@@ -294,7 +303,7 @@ var AiSession = class {
       tools: this._toolCalls
     }, {
       F: __dxlog_file2,
-      L: 202,
+      L: 219,
       S: this,
       C: (f, a) => f(...a)
     });
@@ -360,11 +369,16 @@ var VersionPin = class extends VersionPinSchema {
 // src/session/GenericToolkit.ts
 var GenericToolkit_exports = {};
 __export(GenericToolkit_exports, {
+  Provider: () => Provider,
   TypeId: () => TypeId,
+  empty: () => empty3,
   make: () => make,
-  merge: () => merge3
+  merge: () => merge3,
+  providerEmpty: () => providerEmpty,
+  providerLayer: () => providerLayer
 });
 import * as Toolkit2 from "@effect/ai/Toolkit";
+import * as Context2 from "effect/Context";
 import * as Effect5 from "effect/Effect";
 import * as Layer from "effect/Layer";
 import * as Pipeable from "effect/Pipeable";
@@ -380,9 +394,16 @@ var make = (toolkit, layer) => {
     }
   };
 };
+var empty3 = make(Toolkit2.empty, Layer.empty);
 var merge3 = (...toolkits) => {
   return make(Toolkit2.merge(...toolkits.map((t) => t.toolkit)), Layer.mergeAll(...toolkits.map((t) => t.layer)));
 };
+var Provider = class extends Context2.Tag("@dxos/assistant/GenericToolkit.Provider")() {
+};
+var providerLayer = (toolkit) => Layer.succeed(Provider, {
+  getToolkit: () => toolkit
+});
+var providerEmpty = providerLayer(empty3);
 
 // src/types/artifact.ts
 import * as Schema2 from "effect/Schema";
@@ -396,15 +417,15 @@ var ArtifactId = class extends Schema2.String.annotations({
   // TODO(dmaretskyi): This section gets overriden.
   description: trim2`
     The ID of the referenced object. Formats accepted:
-    - DXN (dxn:echo:@:XXXXX). DXNs can be prepended with an @ symbol for compatibility with in-text references.
+    - 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 (XXXXX)
+    - Only object ID that is assumed to be in the current space (01KG7R1ZXWFMWQ4DA1Q6TN1DG4)
   `,
   examples: [
-    "dxn:echo:@:XXXXX",
-    "@dxn:echo:@:XXXXX",
-    "spaceID:objectID",
-    "XXXXX"
+    "dxn:echo:@:01KG7R1ZXWFMWQ4DA1Q6TN1DG4",
+    "@dxn:echo:@:01KG7R1ZXWFMWQ4DA1Q6TN1DG4",
+    "BM3FSHFOMJCHCG5QW7JTVKGYABD2GAA7G:01KG7R1ZXWFMWQ4DA1Q6TN1DG4",
+    "01KG7R1ZXWFMWQ4DA1Q6TN1DG4"
   ]
 }) {
   static toDXN(reference, owningSpaceId) {
@@ -429,56 +450,54 @@ var ArtifactId = class extends Schema2.String.annotations({
   */
   static resolve(schema, ref) {
     const dxn = ArtifactId.toDXN(ref);
-    return Database.Service.resolve(dxn, schema);
+    return Database.resolve(dxn, schema);
   }
 };
 var RefFromLLM = Schema2.transform(ArtifactId, Type.Ref(Type.Obj), {
   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
 import * as Tool from "@effect/ai/Tool";
-import * as Context2 from "effect/Context";
+import * as Context3 from "effect/Context";
 import * as Effect6 from "effect/Effect";
 import * as Layer2 from "effect/Layer";
 import * as Record from "effect/Record";
 import * as Schema3 from "effect/Schema";
 import { AiToolNotFoundError, ToolExecutionService as ToolExecutionService2, ToolResolverService as ToolResolverService2 } from "@dxos/ai";
 import { todo as todo2 } from "@dxos/debug";
-import { Query, Type as Type2 } from "@dxos/echo";
-import { Database as Database2 } from "@dxos/echo";
-import { Function as Function3, FunctionDefinition, FunctionInvocationService } from "@dxos/functions";
+import { Type as Type2 } from "@dxos/echo";
+import { FunctionInvocationService } from "@dxos/functions";
 import { invariant } from "@dxos/invariant";
 var __dxlog_file3 = "/__w/dxos/dxos/packages/core/assistant/src/functions/services.ts";
-var makeToolResolverFromFunctions = (functions, toolkit) => {
+var makeToolResolverFromFunctions = () => {
   return Layer2.effect(ToolResolverService2, Effect6.gen(function* () {
-    const dbService = yield* Database2.Service;
+    const toolkitProvider = yield* GenericToolkit_exports.Provider;
+    const functionInvocationService = yield* FunctionInvocationService;
     return {
       resolve: (id) => Effect6.gen(function* () {
-        const tool = toolkit.tools[id];
+        const toolkit = toolkitProvider.getToolkit();
+        const tool = toolkit.toolkit.tools[id];
         if (tool) {
           return tool;
         }
-        const [dbFunction] = yield* Database2.Service.runQuery(Query.type(Function3.Function, {
-          key: id
-        }));
-        const functionDef = dbFunction ? FunctionDefinition.deserialize(dbFunction) : functions.find((fn2) => fn2.key === id);
-        if (!functionDef) {
-          return yield* Effect6.fail(new AiToolNotFoundError(id));
-        }
-        return projectFunctionToTool(functionDef);
-      }).pipe(Effect6.provideService(Database2.Service, dbService))
+        return yield* functionInvocationService.resolveFunction(id).pipe(Effect6.map(projectFunctionToTool), Effect6.catchTag("FunctionNotFound", () => Effect6.fail(new AiToolNotFoundError(id))));
+      })
     };
   }));
 };
-var makeToolExecutionServiceFromFunctions = (toolkit, handlersLayer) => {
+var makeToolExecutionServiceFromFunctions = () => {
   return Layer2.effect(ToolExecutionService2, Effect6.gen(function* () {
-    const toolkitHandler = yield* toolkit.pipe(Effect6.provide(handlersLayer));
+    const toolkitProvider = yield* GenericToolkit_exports.Provider;
+    const toolkit = toolkitProvider.getToolkit();
+    const toolkitHandler = yield* toolkit.toolkit.pipe(Effect6.provide(toolkit.layer));
     invariant(isHandlerLike(toolkitHandler), void 0, {
       F: __dxlog_file3,
-      L: 77,
+      L: 76,
       S: this,
       A: [
         "isHandlerLike(toolkitHandler)",
@@ -497,7 +516,7 @@ var makeToolExecutionServiceFromFunctions = (toolkit, handlersLayer) => {
               const { result } = yield* toolkitHandler.handle(tool.name, input);
               return result;
             }
-            const { definition: functionDef } = Context2.get(FunctionToolAnnotation)(tool.annotations);
+            const { definition: functionDef } = Context3.get(FunctionToolAnnotation)(tool.annotations);
             return yield* functionInvocationService.invokeFunction(functionDef, input).pipe(Effect6.catchAllDefect((defect) => Effect6.fail(defect)));
           });
         };
@@ -506,7 +525,8 @@ var makeToolExecutionServiceFromFunctions = (toolkit, handlersLayer) => {
     };
   }));
 };
-var FunctionToolAnnotation = class extends Context2.Tag("@dxos/assistant/FunctionToolAnnotation")() {
+var ToolExecutionServices = Layer2.mergeAll(makeToolResolverFromFunctions(), makeToolExecutionServiceFromFunctions());
+var FunctionToolAnnotation = class extends Context3.Tag("@dxos/assistant/FunctionToolAnnotation")() {
 };
 var toolCache = /* @__PURE__ */ new WeakMap();
 var projectFunctionToTool = (fn2) => {
@@ -528,7 +548,7 @@ var makeToolName = (name) => {
   const toolName = name.toLowerCase().replace(/[^a-zA-Z0-9]/g, "-");
   invariant(toolName.match(/^[a-z_][a-z0-9-_]*$/), void 0, {
     F: __dxlog_file3,
-    L: 148,
+    L: 152,
     S: void 0,
     A: [
       "toolName.match(/^[a-z_][a-z0-9-_]*$/)",
@@ -552,7 +572,10 @@ var createStructFieldsFromSchema = (schema) => {
 };
 var mapSchemaTypeForLLM = (ast) => {
   if (Type2.Ref.isRefSchemaAST(ast)) {
-    return RefFromLLM.ast;
+    const description = ast.annotations.description ? ast.annotations.description + "\n" + RefFromLLM.ast.annotations.description : RefFromLLM.ast.annotations.description;
+    return RefFromLLM.annotations({
+      description
+    }).ast;
   }
   return ast;
 };
@@ -565,6 +588,8 @@ export {
   GenerationObserver,
   AiModelError,
   AiAssistantError,
+  formatSystemPrompt,
+  formatUserPrompt,
   AiSession,
   createToolkit,
   VersionPin,
@@ -573,6 +598,7 @@ export {
   ArtifactId,
   RefFromLLM,
   makeToolResolverFromFunctions,
-  makeToolExecutionServiceFromFunctions
+  makeToolExecutionServiceFromFunctions,
+  ToolExecutionServices
 };
-//# sourceMappingURL=chunk-7CDDG6LL.mjs.map
+//# sourceMappingURL=chunk-QBD6NQJK.mjs.map

package/dist/lib/neutral/chunk-QBD6NQJK.mjs.map

@@ -0,0 +1,7 @@
+{
+  "version": 3,
+  "sources": ["../../../src/session/artifact-diff.ts", "../../../src/session/observer.ts", "../../../src/errors.ts", "../../../src/session/format.ts", "../../../src/session/session.ts", "../../../src/session/toolkit.ts", "../../../src/session/version-pin.ts", "../../../src/session/GenericToolkit.ts", "../../../src/types/artifact.ts", "../../../src/functions/services.ts"],
+  "sourcesContent": ["//\n// Copyright 2025 DXOS.org\n//\n\nimport * as Context from 'effect/Context';\n\nimport { type ObjectVersion } from '@dxos/echo-db';\nimport { type ObjectId } from '@dxos/keys';\n\n/**\n * Resolves artifact ids to their versions.\n * Used to give the model a sense of the changes to the artifacts made by users during the conversation.\n * The artifacts versions are pinned in the history, and whenever the artifact changes in-between assistant's steps,\n * a diff is inserted into the conversation.\n *\n * Can be optionally provided to the session run call.\n */\n// TODO(dmaretskyi): Convert to Context.Reference\nexport class ArtifactDiffResolver extends Context.Tag('@dxos/assistant/ArtifactDiffResolver')<\n  ArtifactDiffResolver,\n  ArtifactDiffResolver.Service\n>() {}\n\nexport namespace ArtifactDiffResolver {\n  export type Service = {\n    resolve: (\n      artifacts: {\n        id: ObjectId;\n        lastVersion: ObjectVersion;\n      }[],\n    ) => Promise<\n      Map<\n        ObjectId,\n        {\n          version: ObjectVersion;\n          diff?: string;\n        }\n      >\n    >;\n  };\n}\n", "//\n// Copyright 2025 DXOS.org\n//\n\nimport type * as Tool from '@effect/ai/Tool';\nimport * as Effect from 'effect/Effect';\nimport * as Function from 'effect/Function';\n\nimport { type AiParser, type ConsolePrinter } from '@dxos/ai';\nimport { type Message } from '@dxos/types';\n\n/**\n * Live observer of the generation process.\n */\nexport interface GenerationObserver<Tools extends Record<string, Tool.Any> = any>\n  extends AiParser.ParseResponseCallbacks<Tools> {\n  /**\n   * Complete messages fired during the session, both from the model and from the user.\n   * This message will contain all message blocks emitted through the `onBlock` callback.\n   */\n  onMessage: (message: Message.Message) => Effect.Effect<void>;\n}\n\nexport const GenerationObserver = Object.freeze({\n  make: ({\n    onBegin = Function.constant(Effect.void),\n    onPart = Function.constant(Effect.void),\n    onBlock = Function.constant(Effect.void),\n    onEnd = Function.constant(Effect.void),\n    onMessage = Function.constant(Effect.void),\n  }: Partial<GenerationObserver> = {}): GenerationObserver => ({\n    onBegin,\n    onPart,\n    onBlock,\n    onEnd,\n    onMessage,\n  }),\n\n  noop: () => GenerationObserver.make(),\n\n  /**\n   * Debug printer to be used in unit-tests and browser devtools.\n   */\n  fromPrinter: (printer: ConsolePrinter) =>\n    GenerationObserver.make({\n      onBlock: (block) =>\n        Effect.sync(() => {\n          if (block.pending) {\n            return; // Only prints full blocks (better for unit-tests and browser devtools).\n          }\n          printer.printContentBlock(block);\n        }),\n      onMessage: (message) =>\n        Effect.sync(() => {\n          if (message.sender.role === 'assistant') {\n            return; // Skip assistant messages since they are printed in the `onBlock` callback.\n          }\n          printer.printMessage(message);\n        }),\n    }),\n});\n", "//\n// Copyright 2025 DXOS.org\n//\n\nimport { BaseError } from '@dxos/errors';\n\n/**\n * Generic error from AI model.\n */\nexport class AiModelError extends BaseError.extend('AiModelError', 'AI model error') {}\n\n/**\n * Generic error for AI agent execution.\n */\nexport class AiAssistantError extends BaseError.extend('AiAssistantError', 'AI assistant error') {}\n", "//\n// Copyright 2025 DXOS.org\n//\n\nimport * as Effect from 'effect/Effect';\nimport * as Function from 'effect/Function';\nimport * as Option from 'effect/Option';\n\nimport { Template } from '@dxos/blueprints';\nimport { Obj } from '@dxos/echo';\nimport { ObjectVersion } from '@dxos/echo-db';\nimport { type ObjectId } from '@dxos/keys';\nimport { log } from '@dxos/log';\nimport { type ContentBlock, Message } from '@dxos/types';\nimport { trim } from '@dxos/util';\n\nimport { AiAssistantError } from '../errors';\n\nimport { ArtifactDiffResolver } from './artifact-diff';\nimport { type AiSessionRunError, type AiSessionRunProps } from './session';\n\n/**\n * Formats the system prompt.\n */\n// TODO(burdon): Move to AiPreprocessor.\nexport const formatSystemPrompt = ({\n  system,\n  blueprints = [],\n  objects = [],\n}: Pick<AiSessionRunProps<any>, 'system' | 'blueprints' | 'objects'>) =>\n  Effect.gen(function* () {\n    const blueprintDefs = yield* Function.pipe(\n      blueprints,\n      Effect.forEach((blueprint) => Effect.succeed(blueprint.instructions)),\n      Effect.flatMap(\n        Effect.forEach((template) =>\n          Effect.gen(function* () {\n            return trim`\n            <blueprint>\n              ${yield* Template.processTemplate(template)}\n            </blueprint>\n          `;\n          }),\n        ),\n      ),\n      Effect.map((blueprints) =>\n        blueprints.length > 0 ? ['## Blueprints Definitions', ...blueprints].join('\\n\\n') : undefined,\n      ),\n    );\n\n    const objectDefs = yield* Function.pipe(\n      objects,\n      Effect.forEach((object) =>\n        Effect.succeed(trim`\n          <object>\n            <dxn>${Obj.getDXN(object)}</dxn>\n            <typename>${Obj.getTypename(object)}</typename>\n          </object>\n        `),\n      ),\n      Effect.map((objects) => (objects.length > 0 ? ['## Context Objects', ...objects].join('\\n\\n') : undefined)),\n    );\n\n    return yield* Function.pipe(\n      Effect.succeed([system, blueprintDefs, objectDefs].filter((def): def is string => def !== undefined)),\n      Effect.map((parts) => parts.join('\\n\\n')),\n    );\n  });\n\n/**\n * Formats the user prompt.\n */\n// TODO(burdon): Move to AiPreprocessor.\n// TODO(burdon): Convert util below to `Effect.fn` (to preserve stack info)\nexport const formatUserPrompt = ({\n  prompt,\n  history = [],\n}: Pick<AiSessionRunProps<any>, 'prompt' | 'history'>): Effect.Effect<Message.Message, AiSessionRunError> =>\n  Effect.gen(function* () {\n    const blocks: ContentBlock.Any[] = [];\n\n    // TODO(dmaretskyi): Evaluate other approaches as `serviceOption` isn't represented in the type system.\n    const artifactDiffResolver = yield* Effect.serviceOption(ArtifactDiffResolver);\n    if (Option.isSome(artifactDiffResolver)) {\n      const versions = gatherObjectVersions(history);\n      const artifactDiff = yield* Effect.tryPromise({\n        try: () =>\n          artifactDiffResolver.value.resolve(\n            [...versions.entries()].map(([id, version]) => ({ id, lastVersion: version })),\n          ),\n        catch: AiAssistantError.wrap({ message: 'Artifact diff resolution error' }),\n      });\n\n      log('version', { artifactDiff, versions });\n      for (const [id, { version }] of [...artifactDiff.entries()]) {\n        if (ObjectVersion.equals(version, versions.get(id)!)) {\n          artifactDiff.delete(id);\n          continue;\n        }\n\n        blocks.push({ _tag: 'anchor', objectId: id, version });\n      }\n\n      if (artifactDiff.size > 0) {\n        blocks.push(createArtifactUpdateBlock(artifactDiff));\n      }\n    }\n\n    return Obj.make(Message.Message, {\n      created: new Date().toISOString(),\n      sender: { role: 'user' },\n      blocks: [...blocks, { _tag: 'text', text: prompt }],\n    });\n  });\n\nconst gatherObjectVersions = (messages: Message.Message[]): Map<ObjectId, ObjectVersion> => {\n  const artifactIds = new Map<ObjectId, ObjectVersion>();\n  for (const message of messages) {\n    for (const block of message.blocks) {\n      if (block._tag === 'anchor') {\n        artifactIds.set(block.objectId, block.version as ObjectVersion);\n      }\n    }\n  }\n\n  return artifactIds;\n};\n\nconst createArtifactUpdateBlock = (\n  artifactDiff: Map<ObjectId, { version: ObjectVersion; diff?: string }>,\n): ContentBlock.Any => {\n  return {\n    _tag: 'text',\n    // TODO(dmaretskyi): Does this need to be a special content-block?\n    disposition: 'artifact-update',\n    text: trim`\n      The following artifacts have been updated since the last message:\n      ${[...artifactDiff.entries()]\n        .map(([id, { diff }]) => `<changed-artifact id=\"${id}\">${diff ? `\\n${diff}` : ''}</changed-artifact>`)\n        .join('\\n')}\n    `,\n  };\n};\n", "//\n// Copyright 2025 DXOS.org\n//\n\nimport type * as AiError from '@effect/ai/AiError';\nimport * as LanguageModel from '@effect/ai/LanguageModel';\nimport type * as Tool from '@effect/ai/Tool';\nimport type * as Toolkit from '@effect/ai/Toolkit';\nimport * as Chunk from 'effect/Chunk';\nimport * as Effect from 'effect/Effect';\nimport type * as Schema from 'effect/Schema';\nimport * as Stream from 'effect/Stream';\n\nimport {\n  AiParser,\n  AiPreprocessor,\n  type AiToolNotFoundError,\n  type PromptPreprocessingError,\n  type ToolExecutionService,\n  type ToolResolverService,\n  callTool,\n  withoutToolCallParising,\n} from '@dxos/ai';\nimport { type Blueprint } from '@dxos/blueprints';\nimport { todo } from '@dxos/debug';\nimport { Obj } from '@dxos/echo';\nimport { type FunctionInvocationService, TracingService } from '@dxos/functions';\nimport { log } from '@dxos/log';\nimport { type ContentBlock, Message } from '@dxos/types';\n\nimport { type AiAssistantError } from '../errors';\n\nimport { formatSystemPrompt, formatUserPrompt } from './format';\nimport { GenerationObserver } from './observer';\n\nexport type AiSessionRunError = AiError.AiError | PromptPreprocessingError | AiToolNotFoundError | AiAssistantError;\n\nexport type AiSessionRunRequirements =\n  | LanguageModel.LanguageModel\n  | ToolExecutionService\n  | ToolResolverService\n  | TracingService\n  | FunctionInvocationService;\n\nexport type AiSessionOptions = {};\n\nexport type AiSessionRunProps<Tools extends Record<string, Tool.Any>> = {\n  prompt: string;\n  // TODO(wittjosiah): Rename to systemPrompt.\n  system?: string;\n  history?: Message.Message[];\n  objects?: Obj.Unknown[];\n  blueprints?: readonly Blueprint.Blueprint[];\n  toolkit?: Toolkit.WithHandler<Tools>;\n  observer?: GenerationObserver<Tools>;\n  /**\n   * Callback for when a message is received from the user, model, or tool.\n   * This is useful for streaming the output to a queue.\n   */\n  // TODO(dmaretskyi): This is duplicated by generation observer. Better solution would be to convert the return type from effect to stream.\n  onOutput?: (message: Message.Message) => Effect.Effect<void, never, never>;\n};\n\n/**\n * Contains message history, tools, current context.\n * Current context means the state of the app, time of day, and other contextual information.\n * It makes requests to the model, its a state machine.\n * It keeps track of the current goal.\n * It manages the context window.\n * Tracks the success criteria of reaching the goal, exposing metrics (stretch).\n * Could be run locally in the app or remotely.\n * Could be personal or shared.\n */\nexport class AiSession {\n  /** Prevents concurrent execution of session. */\n  private readonly _semaphore = Effect.runSync(Effect.makeSemaphore(1));\n\n  /** Message history from queue. */\n  // TODO(burdon): Evolve into supporting a git-like graph of messages.\n  private _history: Message.Message[] = [];\n\n  /** Pending messages for this session (incl. the current prompt). */\n  private _pending: Message.Message[] = [];\n\n  private _started = 0;\n  private _ended = 0;\n  private _toolCalls = 0;\n\n  constructor(private readonly _options: AiSessionOptions = {}) {}\n\n  get duration(): number {\n    return this._ended - this._started;\n  }\n\n  get toolCalls(): number {\n    return this._toolCalls;\n  }\n\n  run = <Tools extends Record<string, Tool.Any>>({\n    prompt,\n    system: systemTemplate,\n    history = [],\n    objects = [],\n    blueprints = [],\n    toolkit,\n    observer = GenerationObserver.noop(),\n    onOutput = () => Effect.void,\n  }: AiSessionRunProps<Tools>): Effect.Effect<Message.Message[], AiSessionRunError, AiSessionRunRequirements> =>\n    Effect.gen(this, function* () {\n      this._started = Date.now();\n      this._history = [...history];\n      this._pending = [];\n      const pending = this._pending;\n\n      const submitMessage = Effect.fnUntraced(function* (message: Message.Message) {\n        pending.push(message);\n        yield* observer.onMessage(message);\n        yield* TracingService.emitConverationMessage(message);\n        yield* onOutput(message);\n        return message;\n      });\n\n      // Submit the prompt.\n      // TODO(burdon): Remove if cancelled?\n      const promptMessage = yield* submitMessage(yield* formatUserPrompt({ prompt, history }));\n\n      // Generate system and prompt messages.\n      const system = yield* formatSystemPrompt({ system: systemTemplate, blueprints, objects }).pipe(Effect.orDie);\n\n      // Tool call loop.\n      do {\n        log('request', {\n          prompt: promptMessage,\n          system: { snippet: createSnippet(system), length: system.length },\n          pending: this._pending.length,\n          history: this._history.length,\n          objects: objects?.length ?? 0,\n        });\n\n        const prompt = yield* AiPreprocessor.preprocessPrompt([...this._history, ...this._pending], {\n          system,\n          cacheControl: 'ephemeral',\n        });\n\n        // Execute the stream request.\n        const messages = yield* LanguageModel.streamText({\n          prompt,\n          toolkit,\n          disableToolCallResolution: true,\n        }).pipe(\n          // Disable tool call parsing so that we can handle parameter schema errors on a per-tool call basis.\n          withoutToolCallParising,\n          // TOOD(dmaretskyi): Error mapping.\n          AiParser.parseResponse({\n            onBegin: () => observer.onBegin(),\n            onBlock: (block) => observer.onBlock(block),\n            onPart: (part) => observer.onPart(part),\n            onEnd: (summary) => observer.onEnd(summary),\n          }),\n          Stream.mapEffect((block) =>\n            Effect.gen(function* () {\n              return yield* submitMessage(\n                Obj.make(Message.Message, {\n                  created: new Date().toISOString(),\n                  sender: { role: 'assistant' },\n                  blocks: [block],\n                }),\n              );\n            }),\n          ),\n          Stream.runCollect,\n          Effect.map(Chunk.toArray),\n        );\n\n        // Parse the response for tool calls.\n        const toolCalls = messages.flatMap((message) =>\n          message.blocks\n            .filter(\n              (block): block is ContentBlock.ToolCall => block._tag === 'toolCall' && block.providerExecuted === false,\n            )\n            .map((block) => ({ block, message })),\n        );\n        if (toolCalls.length === 0) {\n          break;\n        } else if (!toolkit) {\n          throw new Error('No toolkit provided'); // TODO(burdon): Throw user error?\n        }\n\n        // Execute the tool calls.\n        // TODO(burdon): Retry errors? Write result when each completes individually?\n        const toolResults = yield* Effect.forEach(toolCalls, ({ block, message }) => {\n          return callTool(toolkit, block).pipe(\n            Effect.provide(\n              TracingService.layerSubframe((context) => ({\n                ...context,\n                parentMessage: message.id,\n                toolCallId: block.toolCallId,\n              })),\n            ),\n          );\n        });\n\n        // Add to queue and continue loop.\n        // TODO(wittjosiah): Sometimes tool error results are added to the queue before the tool agent statuses.\n        //   This results in a broken execution graph.\n        // TODO(dmaretskyi): Stream tool results one by one.\n        yield* submitMessage(\n          Obj.make(Message.Message, {\n            created: new Date().toISOString(),\n            sender: { role: 'tool' },\n            blocks: toolResults,\n          }),\n        );\n\n        this._toolCalls++;\n      } while (true);\n\n      this._ended = Date.now();\n      log('done', { pending: this._pending.length, duration: this.duration, tools: this._toolCalls });\n      return this._pending;\n    }).pipe(this._semaphore.withPermits(1), Effect.withSpan('AiSession.run'));\n\n  /**\n   * @deprecated\n   */\n  // TODO(burdon): Implement or remove.\n  async runStructured<S extends Schema.Schema.AnyNoContext>(\n    _schema: S,\n    _options: AiSessionRunProps<any>,\n  ): Promise<Schema.Schema.Type<S>> {\n    return todo();\n    // const parser = structuredOutputParser(schema);\n    // const result = await this.run({\n    //   ...options,\n    //   executableTools: [...(options.executableTools ?? []), parser.tool],\n    // });\n    // return parser.getResult(result);\n  }\n}\n\nconst createSnippet = (text: string, len = 32) =>\n  text.length <= len * 2 ? text : [text.slice(0, len), '...', text.slice(-len)].join('');\n\n// TODO(dmaretskyi): Extract as a general util.\nconst logDump = (message: string, data: unknown) => {\n  // eslint-disable-next-line @typescript-eslint/no-require-imports\n  const { writeFileSync } = require('node:fs');\n  const path = `/tmp/log-data-${Date.now()}.json`;\n  writeFileSync(path, JSON.stringify(data, null, 2));\n  log.info(message, { path });\n};\n", "//\n// Copyright 2025 DXOS.org\n//\n\nimport type * as Tool from '@effect/ai/Tool';\nimport * as Toolkit from '@effect/ai/Toolkit';\nimport * as Effect from 'effect/Effect';\n\nimport { type AiToolNotFoundError, ToolExecutionService, ToolResolverService } from '@dxos/ai';\nimport { type Blueprint } from '@dxos/blueprints';\nimport { isTruthy } from '@dxos/util';\n\nexport type CreateToolkitProps = {\n  toolkit?: Toolkit.Any;\n  blueprints?: readonly Blueprint.Blueprint[];\n};\n\n/**\n * Build a combined toolkit from the blueprint tools and the provided toolkit.\n */\nexport const createToolkit = ({\n  toolkit: toolkitProp,\n  blueprints = [],\n}: CreateToolkitProps): Effect.Effect<\n  Toolkit.WithHandler<any>,\n  AiToolNotFoundError,\n  ToolResolverService | ToolExecutionService | Tool.HandlersFor<any>\n> =>\n  Effect.gen(function* () {\n    const blueprintToolkit = yield* ToolResolverService.resolveToolkit(blueprints.flatMap(({ tools }) => tools));\n    const blueprintToolHandler = yield* blueprintToolkit.toContext(ToolExecutionService.handlersFor(blueprintToolkit));\n\n    const toolkit = Toolkit.merge(...[toolkitProp, blueprintToolkit].filter(isTruthy));\n    return yield* toolkit.pipe(Effect.provide(blueprintToolHandler)) as any as Effect.Effect<\n      Toolkit.WithHandler<any>,\n      never,\n      Tool.HandlersFor<any>\n    >;\n  });\n", "//\n// Copyright 2025 DXOS.org\n//\n\nimport * as Schema from 'effect/Schema';\n\nimport { type Obj } from '@dxos/echo';\nimport { type ObjectVersion, getVersion } from '@dxos/echo-db';\nimport { ObjectId } from '@dxos/keys';\nimport { type ContentBlock } from '@dxos/types';\n\n// TODO(dmaretskyi): Extract.\nconst ObjectVersionSchema = Schema.Unknown as Schema.Schema<ObjectVersion>;\n\nconst VersionPinSchema = Schema.Struct({\n  // TODO(dmaretskyi): Use Ref when those support encoding.\n  objectId: ObjectId,\n  // TODO(dmaretskyi): Could be opaque.\n  version: ObjectVersionSchema,\n});\n\n/**\n * Used to pin a specific version of an object during the course of a conversation.\n * This allows us to detect when the object changes during the course of the conversation.\n */\nexport interface VersionPin extends Schema.Schema.Type<typeof VersionPinSchema> {}\n\nexport const VersionPin: typeof VersionPinSchema & {\n  DISPOSITION: 'version-pin';\n  fromObject: (object: Obj.Unknown) => VersionPin;\n  createBlock: (pin: VersionPin) => ContentBlock.Any;\n} = class extends VersionPinSchema {\n  static readonly DISPOSITION = 'version-pin';\n  static fromObject(object: Obj.Unknown): VersionPin {\n    return VersionPin.make({\n      objectId: object.id,\n      version: getVersion(object),\n    });\n  }\n\n  static createBlock(pin: VersionPin): ContentBlock.Any {\n    return {\n      _tag: 'json',\n      disposition: VersionPin.DISPOSITION,\n      data: JSON.stringify(pin),\n    };\n  }\n};\n", "//\n// Copyright 2025 DXOS.org\n//\n\nimport type * as Tool from '@effect/ai/Tool';\nimport * as Toolkit from '@effect/ai/Toolkit';\nimport * as Context from 'effect/Context';\nimport * as Effect from 'effect/Effect';\nimport * as Layer from 'effect/Layer';\nimport * as Pipeable from 'effect/Pipeable';\nimport type * as Schema from 'effect/Schema';\nimport type * as SchemaAST from 'effect/SchemaAST';\n\n/**\n * Unique identifier for generic toolkit instances.\n */\nexport const TypeId = '~@dxos/assistant/GenericToolkit';\n\n/**\n * Type-level representation of the generic toolkit identifier.\n */\nexport type TypeId = typeof TypeId;\n\n/**\n * Type-safe way to define toolkits where we don't know specific types of tools,\n * and we want to bundle definition and handlers together.\n *\n * Usefull for plugin-based systems, where tools are dynamically added.\n *\n * @param TR requirements type to the tool invocation.\n * @param E failure type to the handlers layer.\n * @param R requirements type to the handlers layer.\n */\nexport interface GenericToolkit<TR = never, E = never, R = never> extends Pipeable.Pipeable {\n  readonly [TypeId]: TypeId;\n\n  /**\n   * Toolkit definition.\n   */\n  readonly toolkit: Toolkit.Toolkit<GenericTools<TR>>;\n\n  /**\n   * Handlers layer.\n   */\n  readonly layer: Layer.Layer<unknown, E, R>;\n\n  /**\n   * Handlers effect.\n   */\n  readonly handlers: Effect.Effect<Toolkit.WithHandler<GenericTools<TR>>, E, R>;\n}\n\n/**\n * Generic constraint.\n *\n * NOTE: Only use in place of `T extends PortableToolkit.Any`. Not suitable for standalone use.\n */\nexport interface Any {\n  readonly [TypeId]: TypeId;\n  readonly toolkit: Toolkit.Toolkit<any>;\n  readonly layer: Layer.Layer<unknown, any, any>;\n}\n\nexport type InvocationRequirements<T extends Any> = T extends GenericToolkit<infer TR, infer _E, infer _R> ? TR : never;\nexport type Failure<T extends Any> = T extends GenericToolkit<infer _TR, infer E, infer _R> ? E : never;\nexport type Requirements<T extends Any> = T extends GenericToolkit<infer _TR, infer _E, infer R> ? R : never;\n\n/**\n * Creates a portable toolkit from toolkit definition and handlers layer.\n */\nexport const make = <Tools extends Record<string, Tool.Any>, E, R>(\n  toolkit: Toolkit.Toolkit<Tools>,\n  layer: Layer.Layer<Tool.HandlersFor<Tools>, E, R>,\n): GenericToolkit<E, R> => {\n  return {\n    [TypeId]: TypeId,\n    toolkit: toolkit as any,\n    layer: layer as any,\n    handlers: toolkit.pipe(Effect.provide(layer)) as any,\n    pipe() {\n      // eslint-disable-next-line prefer-rest-params\n      return Pipeable.pipeArguments(this, arguments);\n    },\n  };\n};\n\n/**\n * Empty generic toolkit.\n */\nexport const empty = make(Toolkit.empty, Layer.empty);\n\n/**\n * Merges multiple portable toolkits into a single portable toolkit.\n */\nexport const merge = <const Toolkits extends ReadonlyArray<Any>>(\n  /**\n   * The toolkits to merge together.\n   */\n  ...toolkits: Toolkits\n): GenericToolkit<\n  InvocationRequirements<Toolkits[number]>,\n  Failure<Toolkits[number]>,\n  Requirements<Toolkits[number]>\n> => {\n  return make(\n    Toolkit.merge(...toolkits.map((t) => t.toolkit)),\n    Layer.mergeAll(...(toolkits.map((t) => t.layer) as any)),\n  ) as any;\n};\n\n/**\n * Generic tools type where we cannot statically enumerate individual tools.\n */\nexport type GenericTools<R = never> = Record<\n  string,\n  Tool.Tool<\n    string,\n    {\n      readonly parameters: AnyStructSchemaNoContext;\n      readonly success: Schema.Schema.AnyNoContext;\n      readonly failure: typeof Schema.Never;\n      readonly failureMode: Tool.FailureMode;\n    },\n    R\n  >\n>;\n\nexport interface AnyStructSchemaNoContext extends Pipeable.Pipeable {\n  readonly [Schema.TypeId]: any;\n  readonly make: any;\n  readonly Type: any;\n  readonly Encoded: any;\n  readonly Context: never;\n  readonly ast: SchemaAST.AST;\n  readonly fields: Schema.Struct.Fields;\n  readonly annotations: any;\n}\n\n/**\n * Provides a generic toolkit to the agent.\n */\nexport class Provider extends Context.Tag('@dxos/assistant/GenericToolkit.Provider')<\n  Provider,\n  {\n    readonly getToolkit: () => GenericToolkit;\n  }\n>() {}\n\n/**\n * Layer for providing a generic toolkit to the agent.\n */\nexport const providerLayer = (toolkit: GenericToolkit) =>\n  Layer.succeed(Provider, {\n    getToolkit: () => toolkit,\n  });\n\n/**\n * Layer for providing an empty generic toolkit to the agent.\n */\nexport const providerEmpty = providerLayer(empty);\n", "//\n// Copyright 2025 DXOS.org\n//\n\nimport type * as Effect from 'effect/Effect';\nimport * as Schema from 'effect/Schema';\n\nimport { type Err, Type } from '@dxos/echo';\nimport { Database } from '@dxos/echo';\nimport { EncodedReference } from '@dxos/echo-protocol';\nimport { DXN, LOCAL_SPACE_TAG, type ObjectId, type SpaceId } from '@dxos/keys';\nimport { trim } from '@dxos/util';\n\n/**\n * @deprecated\n */\nexport const createArtifactElement = (id: ObjectId) => `<artifact id=${id} />`;\n\n// TODO(burdon): Rename RefFromLLM? -- yes -dm\n/**\n * A model-friendly way to reference an object.\n * Supports vairous formats that will be normalized to a DXN.\n *\n * @deprecated Use `Type.Ref(XXX)` instead.\n */\nexport const ArtifactId: Schema.Schema<string> & {\n  toDXN: (reference: ArtifactId, owningSpaceId?: SpaceId) => DXN;\n  resolve: <S extends Type.Entity.Any>(\n    schema: S,\n    ref: ArtifactId,\n  ) => Effect.Effect<Schema.Schema.Type<S>, Err.ObjectNotFoundError, Database.Service>;\n} = class extends Schema.String.annotations({\n  // TODO(dmaretskyi): This section gets overriden.\n  description: trim`\n    The ID of the referenced object. Formats accepted:\n    - DXN (dxn:echo:@:01KG7R1ZXWFMWQ4DA1Q6TN1DG4). DXNs can be prepended with an @ symbol for compatibility with in-text references.\n    - space ID, object ID tuple (spaceID:objectID)\n    - Only object ID that is assumed to be in the current space (01KG7R1ZXWFMWQ4DA1Q6TN1DG4)\n  `,\n  examples: [\n    'dxn:echo:@:01KG7R1ZXWFMWQ4DA1Q6TN1DG4',\n    '@dxn:echo:@:01KG7R1ZXWFMWQ4DA1Q6TN1DG4',\n    'BM3FSHFOMJCHCG5QW7JTVKGYABD2GAA7G:01KG7R1ZXWFMWQ4DA1Q6TN1DG4',\n    '01KG7R1ZXWFMWQ4DA1Q6TN1DG4',\n  ],\n}) {\n  static toDXN(reference: ArtifactId, owningSpaceId?: SpaceId): DXN {\n    // Allow @dxn: prefix for compatibility with in-text references.\n    if (reference.startsWith('@dxn:')) {\n      return DXN.parse(reference.slice(1));\n    } else if (reference.startsWith('dxn:')) {\n      return DXN.parse(reference);\n    } else if (/^[A-Z0-9]+:[A-Z0-9]+$/.test(reference)) {\n      const [spaceId, objectId] = reference.split(':');\n      // This is a workaround because the current Filter API doesn't work with fully qualified Echo DXNs.\n      // We check if the space ID is the same as the owning space and then use LOCAL_SPACE_TAG for local references.\n      // TODO(dmaretskyi): Fix this in the Echo and Filter API to properly handle fully qualified DXNs.\n      return new DXN(DXN.kind.ECHO, [spaceId === owningSpaceId ? LOCAL_SPACE_TAG : spaceId, objectId]);\n    } else if (/^[A-Z0-9]+$/.test(reference)) {\n      return DXN.fromLocalObjectId(reference);\n    } else {\n      throw new Error(`Unable to parse object reference: ${reference}`);\n    }\n  }\n\n  /**\n   * Resolves an artifact ID to an object.\n   */\n  static resolve<S extends Type.Entity.Any>(\n    schema: S,\n    ref: ArtifactId,\n  ): Effect.Effect<Schema.Schema.Type<S>, Err.ObjectNotFoundError, Database.Service> {\n    const dxn = ArtifactId.toDXN(ref);\n    return Database.resolve(dxn, schema);\n  }\n};\n\nexport type ArtifactId = Schema.Schema.Type<typeof ArtifactId>;\n\n/**\n * Schema that decodes ECHO reference object from an LLM-friendly input.\n */\nexport const RefFromLLM = Schema.transform(ArtifactId, Type.Ref(Type.Obj), {\n  decode: (fromA, fromI) => EncodedReference.fromDXN(ArtifactId.toDXN(fromA)),\n  encode: (toI, toA) => EncodedReference.toDXN(toI).toString(),\n  strict: false,\n}).annotations({\n  description: ArtifactId.ast.annotations.description as string,\n});\n", "//\n// Copyright 2025 DXOS.org\n//\n\nimport * as Tool from '@effect/ai/Tool';\nimport type * as Toolkit from '@effect/ai/Toolkit';\nimport * as Context from 'effect/Context';\nimport * as Effect from 'effect/Effect';\nimport * as Layer from 'effect/Layer';\nimport * as Record from 'effect/Record';\nimport * as Schema from 'effect/Schema';\nimport type * as SchemaAST from 'effect/SchemaAST';\n\nimport { AiToolNotFoundError, ToolExecutionService, ToolResolverService } from '@dxos/ai';\nimport { todo } from '@dxos/debug';\nimport { Type } from '@dxos/echo';\nimport { type FunctionDefinition, FunctionInvocationService } from '@dxos/functions';\nimport { invariant } from '@dxos/invariant';\n\nimport { GenericToolkit } from '../session';\nimport { RefFromLLM } from '../types';\n\n/**\n * Constructs a `ToolResolverService` whose `resolve(id)` looks up tools in the following order:\n *  1. Toolkit: return an existing tool from the provided `toolkit` if one is present under `id`.\n *  2. Functions in DB: query `FunctionType` by `key=id`; if found, deserialize and project to a tool\n *     (propagating any `deployedFunctionId` discovered in the object's metadata).\n *  3. Functions passed in: fall back to a matching `FunctionDefinition` from the `functions` array.\n *\n * If none of the above yield a match, the effect fails with `AiToolNotFoundError`.\n *\n * Requires `Database.Service` in the environment.\n */\nexport const makeToolResolverFromFunctions = (): Layer.Layer<\n  ToolResolverService,\n  never,\n  GenericToolkit.Provider | FunctionInvocationService\n> => {\n  return Layer.effect(\n    ToolResolverService,\n    Effect.gen(function* () {\n      const toolkitProvider = yield* GenericToolkit.Provider;\n      const functionInvocationService = yield* FunctionInvocationService;\n      return {\n        resolve: (id): Effect.Effect<Tool.Any, AiToolNotFoundError> =>\n          Effect.gen(function* () {\n            const toolkit = toolkitProvider.getToolkit();\n\n            const tool = toolkit.toolkit.tools[id];\n            if (tool) {\n              return tool;\n            }\n\n            return yield* functionInvocationService.resolveFunction(id).pipe(\n              Effect.map(projectFunctionToTool),\n              Effect.catchTag('FunctionNotFound', () => Effect.fail(new AiToolNotFoundError(id))),\n            );\n          }),\n      } satisfies Context.Tag.Service<ToolResolverService>;\n    }),\n  );\n};\n\nexport const makeToolExecutionServiceFromFunctions = (): Layer.Layer<\n  ToolExecutionService,\n  never,\n  FunctionInvocationService | GenericToolkit.Provider\n> => {\n  return Layer.effect(\n    ToolExecutionService,\n    Effect.gen(function* () {\n      const toolkitProvider = yield* GenericToolkit.Provider;\n      const toolkit = toolkitProvider.getToolkit();\n\n      const toolkitHandler = yield* toolkit.toolkit.pipe(Effect.provide(toolkit.layer));\n      invariant(isHandlerLike(toolkitHandler));\n      const functionInvocationService = yield* FunctionInvocationService;\n\n      return {\n        handlersFor: (toolkit) => {\n          const makeHandler = (tool: Tool.Any): ((params: unknown) => Effect.Effect<unknown, any, any>) => {\n            return Effect.fn(`toolFunctionHandler ${tool.name}`)(function* (input: any) {\n              if (toolkitHandler.tools[tool.name]) {\n                if (Tool.isProviderDefined(tool)) {\n                  throw new Error('Attempted to call a provider-defined tool');\n                }\n\n                // TODO(wittjosiah): Everything is `never` here.\n                const { result } = yield* (toolkitHandler.handle as any)(tool.name, input);\n                return result;\n              }\n\n              const { definition: functionDef } = Context.get(FunctionToolAnnotation)(tool.annotations as any);\n\n              return yield* functionInvocationService\n                .invokeFunction(functionDef, input as any)\n                .pipe(Effect.catchAllDefect((defect) => Effect.fail(defect)));\n            });\n          };\n\n          return toolkit.of(\n            Record.map(toolkit.tools, (tool, _name) => (Tool.isUserDefined(tool) ? makeHandler(tool) : null)) as any,\n          ) as any;\n        },\n      };\n    }),\n  );\n};\n\nexport const ToolExecutionServices = Layer.mergeAll(\n  makeToolResolverFromFunctions(),\n  makeToolExecutionServiceFromFunctions(),\n);\n\nclass FunctionToolAnnotation extends Context.Tag('@dxos/assistant/FunctionToolAnnotation')<\n  FunctionToolAnnotation,\n  { definition: FunctionDefinition.Any }\n>() {}\n\nconst toolCache = new WeakMap<FunctionDefinition.Any, Tool.Any>();\n\n/**\n * Projects a `FunctionDefinition` into an `AiTool`.\n *\n * @param fn The function definition to project into a tool.\n * @param meta Optional projection metadata.\n * @param meta.deployedFunctionId Backend deployment ID used for remote invocation when present.\n *    This is the EDGE service's function deployment identifier (not the ECHO object ID/DXN and not `FunctionDefinition.key`).\n */\nconst projectFunctionToTool = (fn: FunctionDefinition.Any): Tool.Any => {\n  if (toolCache.has(fn)) {\n    return toolCache.get(fn)!;\n  }\n\n  // TODO(burdon): Use and map function.key?\n  // TODO(dmaretskyi): Use function key instead.\n  const tool = Tool.make(makeToolName(fn.name), {\n    description: fn.description,\n    parameters: createStructFieldsFromSchema(fn.inputSchema),\n    // TODO(dmaretskyi): Include output schema.\n    failure: Schema.Any, // TODO(dmaretskyi): Better type for the failure?\n  }).annotate(FunctionToolAnnotation, { definition: fn });\n  toolCache.set(fn, tool);\n  return tool;\n};\n\n/**\n * @returns Tool name produced from function name by escaping invalid characters.\n */\nconst makeToolName = (name: string) => {\n  const toolName = name.toLowerCase().replace(/[^a-zA-Z0-9]/g, '-');\n  invariant(toolName.match(/^[a-z_][a-z0-9-_]*$/));\n  return toolName;\n};\n\n// TODO(dmaretskyi): Factor out.\nconst createStructFieldsFromSchema = (schema: Schema.Schema<any, any>): Record<string, Schema.Schema<any, any>> => {\n  switch (schema.ast._tag) {\n    case 'TypeLiteral':\n      return Object.fromEntries(\n        schema.ast.propertySignatures.map((prop) => [prop.name, Schema.make(mapSchemaTypeForLLM(prop.type))]),\n      );\n    case 'VoidKeyword':\n      return {};\n    default:\n      return todo(`Unsupported schema AST: ${schema.ast._tag}`);\n  }\n};\n\n/**\n * Picks an LLM-friendly schema type for the given schema AST.\n * The picked schema type decodes to the original schema type.\n */\nconst mapSchemaTypeForLLM = (ast: SchemaAST.AST): SchemaAST.AST => {\n  if (Type.Ref.isRefSchemaAST(ast)) {\n    const description = ast.annotations.description\n      ? ast.annotations.description + '\\n' + RefFromLLM.ast.annotations.description\n      : (RefFromLLM.ast.annotations.description as string);\n    return RefFromLLM.annotations({ description }).ast;\n  }\n\n  return ast;\n};\n\nconst isHandlerLike = (value: unknown): value is Toolkit.WithHandler<Record<string, Tool.Any>> => {\n  return typeof (value as any).tools === 'object' && typeof (value as any).handle === 'function';\n};\n"],
+  "mappings": ";;;;;AAIA,YAAYA,aAAa;AAclB,IAAMC,uBAAN,cAA2CC,YAAI,sCAAA,EAAA,EAAA;AAGjD;;;AChBL,YAAYC,YAAY;AACxB,YAAYC,cAAc;AAiBnB,IAAMC,qBAAqBC,OAAOC,OAAO;EAC9CC,MAAM,CAAC,EACLC,UAAmBC,kBAAgBC,WAAI,GACvCC,SAAkBF,kBAAgBC,WAAI,GACtCE,UAAmBH,kBAAgBC,WAAI,GACvCG,QAAiBJ,kBAAgBC,WAAI,GACrCI,YAAqBL,kBAAgBC,WAAI,EAAC,IACX,CAAC,OAA2B;IAC3DF;IACAG;IACAC;IACAC;IACAC;EACF;EAEAC,MAAM,MAAMX,mBAAmBG,KAAI;;;;EAKnCS,aAAa,CAACC,YACZb,mBAAmBG,KAAK;IACtBK,SAAS,CAACM,UACDC,YAAK,MAAA;AACV,UAAID,MAAME,SAAS;AACjB;MACF;AACAH,cAAQI,kBAAkBH,KAAAA;IAC5B,CAAA;IACFJ,WAAW,CAACQ,YACHH,YAAK,MAAA;AACV,UAAIG,QAAQC,OAAOC,SAAS,aAAa;AACvC;MACF;AACAP,cAAQQ,aAAaH,OAAAA;IACvB,CAAA;EACJ,CAAA;AACJ,CAAA;;;ACxDA,SAASI,iBAAiB;AAKnB,IAAMC,eAAN,cAA2BC,UAAUC,OAAO,gBAAgB,gBAAA,EAAA;AAAmB;AAK/E,IAAMC,mBAAN,cAA+BF,UAAUC,OAAO,oBAAoB,oBAAA,EAAA;AAAuB;;;ACVlG,YAAYE,aAAY;AACxB,YAAYC,eAAc;AAC1B,YAAYC,YAAY;AAExB,SAASC,gBAAgB;AACzB,SAASC,WAAW;AACpB,SAASC,qBAAqB;AAE9B,SAASC,WAAW;AACpB,SAA4BC,eAAe;AAC3C,SAASC,YAAY;;AAWd,IAAMC,qBAAqB,CAAC,EACjCC,QACAC,aAAa,CAAA,GACbC,UAAU,CAAA,EAAE,MAELC,YAAI,aAAA;AACT,QAAMC,gBAAgB,OAAgBC,eACpCJ,YACOK,gBAAQ,CAACC,cAAqBC,gBAAQD,UAAUE,YAAY,CAAA,GAC5DC,gBACEJ,gBAAQ,CAACK,aACPR,YAAI,aAAA;AACT,WAAOS;;gBAEH,OAAOC,SAASC,gBAAgBH,QAAAA,CAAAA;;;EAGtC,CAAA,CAAA,CAAA,GAGGI,YAAI,CAACd,gBACVA,YAAWe,SAAS,IAAI;IAAC;OAAgCf;IAAYgB,KAAK,MAAA,IAAUC,MAAAA,CAAAA;AAIxF,QAAMC,aAAa,OAAgBd,eACjCH,SACOI,gBAAQ,CAACc,WACPZ,gBAAQI;;mBAEJS,IAAIC,OAAOF,MAAAA,CAAAA;wBACNC,IAAIE,YAAYH,MAAAA,CAAAA;;SAE/B,CAAA,GAEIL,YAAI,CAACb,aAAaA,SAAQc,SAAS,IAAI;IAAC;OAAyBd;IAASe,KAAK,MAAA,IAAUC,MAAAA,CAAAA;AAGlG,SAAO,OAAgBb,eACdG,gBAAQ;IAACR;IAAQI;IAAee;IAAYK,OAAO,CAACC,QAAuBA,QAAQP,MAAAA,CAAAA,GACnFH,YAAI,CAACW,UAAUA,MAAMT,KAAK,MAAA,CAAA,CAAA;AAErC,CAAA;AAOK,IAAMU,mBAAmB,CAAC,EAC/BC,QACAC,UAAU,CAAA,EAAE,MAEL1B,YAAI,aAAA;AACT,QAAM2B,SAA6B,CAAA;AAGnC,QAAMC,uBAAuB,OAAcC,sBAAcC,oBAAAA;AACzD,MAAWC,cAAOH,oBAAAA,GAAuB;AACvC,UAAMI,WAAWC,qBAAqBP,OAAAA;AACtC,UAAMQ,eAAe,OAAcC,mBAAW;MAC5CC,KAAK,MACHR,qBAAqBS,MAAMC,QACzB;WAAIN,SAASO,QAAO;QAAI3B,IAAI,CAAC,CAAC4B,IAAIC,OAAAA,OAAc;QAAED;QAAIE,aAAaD;MAAQ,EAAA,CAAA;MAE/EE,OAAOC,iBAAiBC,KAAK;QAAEC,SAAS;MAAiC,CAAA;IAC3E,CAAA;AAEAC,QAAI,WAAW;MAAEb;MAAcF;IAAS,GAAA;;;;;;AACxC,eAAW,CAACQ,IAAI,EAAEC,QAAO,CAAE,KAAK;SAAIP,aAAaK,QAAO;OAAK;AAC3D,UAAIS,cAAcC,OAAOR,SAAST,SAASkB,IAAIV,EAAAA,CAAAA,GAAO;AACpDN,qBAAaiB,OAAOX,EAAAA;AACpB;MACF;AAEAb,aAAOyB,KAAK;QAAEC,MAAM;QAAUC,UAAUd;QAAIC;MAAQ,CAAA;IACtD;AAEA,QAAIP,aAAaqB,OAAO,GAAG;AACzB5B,aAAOyB,KAAKI,0BAA0BtB,YAAAA,CAAAA;IACxC;EACF;AAEA,SAAOhB,IAAIuC,KAAKC,QAAQA,SAAS;IAC/BC,UAAS,oBAAIC,KAAAA,GAAOC,YAAW;IAC/BC,QAAQ;MAAEC,MAAM;IAAO;IACvBpC,QAAQ;SAAIA;MAAQ;QAAE0B,MAAM;QAAQW,MAAMvC;MAAO;;EACnD,CAAA;AACF,CAAA;AAEF,IAAMQ,uBAAuB,CAACgC,aAAAA;AAC5B,QAAMC,cAAc,oBAAIC,IAAAA;AACxB,aAAWrB,WAAWmB,UAAU;AAC9B,eAAWG,SAAStB,QAAQnB,QAAQ;AAClC,UAAIyC,MAAMf,SAAS,UAAU;AAC3Ba,oBAAYG,IAAID,MAAMd,UAAUc,MAAM3B,OAAO;MAC/C;IACF;EACF;AAEA,SAAOyB;AACT;AAEA,IAAMV,4BAA4B,CAChCtB,iBAAAA;AAEA,SAAO;IACLmB,MAAM;;IAENiB,aAAa;IACbN,MAAMvD;;QAEF;SAAIyB,aAAaK,QAAO;MACvB3B,IAAI,CAAC,CAAC4B,IAAI,EAAE+B,KAAI,CAAE,MAAM,yBAAyB/B,EAAAA,KAAO+B,OAAO;EAAKA,IAAAA,KAAS,EAAA,qBAAuB,EACpGzD,KAAK,IAAA,CAAA;;EAEZ;AACF;;;ACzIA,YAAY0D,mBAAmB;AAG/B,YAAYC,WAAW;AACvB,YAAYC,aAAY;AAExB,YAAYC,YAAY;AAExB,SACEC,UACAC,gBAKAC,UACAC,+BACK;AAEP,SAASC,YAAY;AACrB,SAASC,OAAAA,YAAW;AACpB,SAAyCC,sBAAsB;AAC/D,SAASC,OAAAA,YAAW;AACpB,SAA4BC,WAAAA,gBAAe;;AA6CpC,IAAMC,YAAN,MAAMA;;;EAEMC,aAAoBC,gBAAeC,sBAAc,CAAA,CAAA;;;EAI1DC,WAA8B,CAAA;;EAG9BC,WAA8B,CAAA;EAE9BC,WAAW;EACXC,SAAS;EACTC,aAAa;EAErB,YAA6BC,WAA6B,CAAC,GAAG;SAAjCA,WAAAA;EAAkC;EAE/D,IAAIC,WAAmB;AACrB,WAAO,KAAKH,SAAS,KAAKD;EAC5B;EAEA,IAAIK,YAAoB;AACtB,WAAO,KAAKH;EACd;EAEAI,MAAM,CAAyC,EAC7CC,QACAC,QAAQC,gBACRC,UAAU,CAAA,GACVC,UAAU,CAAA,GACVC,aAAa,CAAA,GACbC,SACAC,WAAWC,mBAAmBC,KAAI,GAClCC,WAAW,MAAaC,aAAI,MAErBC,YAAI,MAAM,aAAA;AACf,SAAKnB,WAAWoB,KAAKC,IAAG;AACxB,SAAKvB,WAAW;SAAIY;;AACpB,SAAKX,WAAW,CAAA;AAChB,UAAMuB,UAAU,KAAKvB;AAErB,UAAMwB,gBAAuBC,mBAAW,WAAWC,SAAwB;AACzEH,cAAQI,KAAKD,OAAAA;AACb,aAAOX,SAASa,UAAUF,OAAAA;AAC1B,aAAOG,eAAeC,uBAAuBJ,OAAAA;AAC7C,aAAOR,SAASQ,OAAAA;AAChB,aAAOA;IACT,CAAA;AAIA,UAAMK,gBAAgB,OAAOP,cAAc,OAAOQ,iBAAiB;MAAExB;MAAQG;IAAQ,CAAA,CAAC;AAGtF,UAAMF,SAAS,OAAOwB,mBAAmB;MAAExB,QAAQC;MAAgBG;MAAYD;IAAQ,CAAA,EAAGsB,KAAYC,aAAK;AAG3G,OAAG;AACDC,MAAAA,KAAI,WAAW;QACb5B,QAAQuB;QACRtB,QAAQ;UAAE4B,SAASC,cAAc7B,MAAAA;UAAS8B,QAAQ9B,OAAO8B;QAAO;QAChEhB,SAAS,KAAKvB,SAASuC;QACvB5B,SAAS,KAAKZ,SAASwC;QACvB3B,SAASA,SAAS2B,UAAU;MAC9B,GAAA;;;;;;AAEA,YAAM/B,UAAS,OAAOgC,eAAeC,iBAAiB;WAAI,KAAK1C;WAAa,KAAKC;SAAW;QAC1FS;QACAiC,cAAc;MAChB,CAAA;AAGA,YAAMC,WAAW,OAAqBC,yBAAW;QAC/CpC,QAAAA;QACAM;QACA+B,2BAA2B;MAC7B,CAAA,EAAGX;;QAEDY;;QAEAC,SAASC,cAAc;UACrBC,SAAS,MAAMlC,SAASkC,QAAO;UAC/BC,SAAS,CAACC,UAAUpC,SAASmC,QAAQC,KAAAA;UACrCC,QAAQ,CAACC,SAAStC,SAASqC,OAAOC,IAAAA;UAClCC,OAAO,CAACC,YAAYxC,SAASuC,MAAMC,OAAAA;QACrC,CAAA;QACOC,iBAAU,CAACL,UACT/B,YAAI,aAAA;AACT,iBAAO,OAAOI,cACZiC,KAAIC,KAAKC,SAAQA,SAAS;YACxBC,UAAS,oBAAIvC,KAAAA,GAAOwC,YAAW;YAC/BC,QAAQ;cAAEC,MAAM;YAAY;YAC5BC,QAAQ;cAACb;;UACX,CAAA,CAAA;QAEJ,CAAA,CAAA;QAEKc;QACAC,YAAUC,aAAO;MAAA;AAI1B,YAAM7D,YAAYqC,SAASyB,QAAQ,CAAC1C,YAClCA,QAAQsC,OACLK,OACC,CAAClB,UAA0CA,MAAMmB,SAAS,cAAcnB,MAAMoB,qBAAqB,KAAA,EAEpGL,IAAI,CAACf,WAAW;QAAEA;QAAOzB;MAAQ,EAAA,CAAA;AAEtC,UAAIpB,UAAUiC,WAAW,GAAG;AAC1B;MACF,WAAW,CAACzB,SAAS;AACnB,cAAM,IAAI0D,MAAM,qBAAA;MAClB;AAIA,YAAMC,cAAc,OAAcC,gBAAQpE,WAAW,CAAC,EAAE6C,OAAOzB,QAAO,MAAE;AACtE,eAAOiD,SAAS7D,SAASqC,KAAAA,EAAOjB,KACvB0C,gBACL/C,eAAegD,cAAc,CAACC,aAAa;UACzC,GAAGA;UACHC,eAAerD,QAAQsD;UACvBC,YAAY9B,MAAM8B;QACpB,EAAA,CAAA,CAAA;MAGN,CAAA;AAMA,aAAOzD,cACLiC,KAAIC,KAAKC,SAAQA,SAAS;QACxBC,UAAS,oBAAIvC,KAAAA,GAAOwC,YAAW;QAC/BC,QAAQ;UAAEC,MAAM;QAAO;QACvBC,QAAQS;MACV,CAAA,CAAA;AAGF,WAAKtE;IACP,SAAS;AAET,SAAKD,SAASmB,KAAKC,IAAG;AACtBc,IAAAA,KAAI,QAAQ;MAAEb,SAAS,KAAKvB,SAASuC;MAAQlC,UAAU,KAAKA;MAAU6E,OAAO,KAAK/E;IAAW,GAAA;;;;;;AAC7F,WAAO,KAAKH;EACd,CAAA,EAAGkC,KAAK,KAAKtC,WAAWuF,YAAY,CAAA,GAAWC,iBAAS,eAAA,CAAA;;;;;EAM1D,MAAMC,cACJC,SACAlF,UACgC;AAChC,WAAOmF,KAAAA;EAOT;AACF;AAEA,IAAMjD,gBAAgB,CAACkD,MAAcC,MAAM,OACzCD,KAAKjD,UAAUkD,MAAM,IAAID,OAAO;EAACA,KAAKE,MAAM,GAAGD,GAAAA;EAAM;EAAOD,KAAKE,MAAM,CAACD,GAAAA;EAAME,KAAK,EAAA;;;AC5OrF,YAAYC,aAAa;AACzB,YAAYC,aAAY;AAExB,SAAmCC,sBAAsBC,2BAA2B;AAEpF,SAASC,gBAAgB;AAUlB,IAAMC,gBAAgB,CAAC,EAC5BC,SAASC,aACTC,aAAa,CAAA,EAAE,MAMRC,YAAI,aAAA;AACT,QAAMC,mBAAmB,OAAOC,oBAAoBC,eAAeJ,WAAWK,QAAQ,CAAC,EAAEC,MAAK,MAAOA,KAAAA,CAAAA;AACrG,QAAMC,uBAAuB,OAAOL,iBAAiBM,UAAUC,qBAAqBC,YAAYR,gBAAAA,CAAAA;AAEhG,QAAMJ,UAAkBa,cAAK,GAAI;IAACZ;IAAaG;IAAkBU,OAAOC,QAAAA,CAAAA;AACxE,SAAO,OAAOf,QAAQgB,KAAYC,gBAAQR,oBAAAA,CAAAA;AAK5C,CAAA;;;AClCF,YAAYS,YAAY;AAGxB,SAA6BC,kBAAkB;AAC/C,SAASC,gBAAgB;AAIzB,IAAMC,sBAA6BC;AAEnC,IAAMC,mBAA0BC,cAAO;;EAErCC,UAAUC;;EAEVC,SAASN;AACX,CAAA;AAQO,IAAMO,aAIT,cAAcL,iBAAAA;EAChB,OAAgBM,cAAc;EAC9B,OAAOC,WAAWC,QAAiC;AACjD,WAAOH,WAAWI,KAAK;MACrBP,UAAUM,OAAOE;MACjBN,SAASO,WAAWH,MAAAA;IACtB,CAAA;EACF;EAEA,OAAOI,YAAYC,KAAmC;AACpD,WAAO;MACLC,MAAM;MACNC,aAAaV,WAAWC;MACxBU,MAAMC,KAAKC,UAAUL,GAAAA;IACvB;EACF;AACF;;;AC/CA;;;;eAAAM;EAAA;eAAAC;EAAA;;;AAKA,YAAYC,cAAa;AACzB,YAAYC,cAAa;AACzB,YAAYC,aAAY;AACxB,YAAYC,WAAW;AACvB,YAAYC,cAAc;AAOnB,IAAMC,SAAS;AAsDf,IAAMC,OAAO,CAClBC,SACAC,UAAAA;AAEA,SAAO;IACL,CAACH,MAAAA,GAASA;IACVE;IACAC;IACAC,UAAUF,QAAQG,KAAYC,gBAAQH,KAAAA,CAAAA;IACtCE,OAAAA;AAEE,aAAgBE,uBAAc,MAAMC,SAAAA;IACtC;EACF;AACF;AAKO,IAAMC,SAAQR,KAAaQ,gBAAaA,WAAK;AAK7C,IAAMC,SAAQ,IAIhBC,aAAAA;AAMH,SAAOV,KACGS,eAAK,GAAIC,SAASC,IAAI,CAACC,MAAMA,EAAEX,OAAO,CAAA,GACxCY,eAAQ,GAAKH,SAASC,IAAI,CAACC,MAAMA,EAAEV,KAAK,CAAA,CAAA;AAElD;AAiCO,IAAMY,WAAN,cAA+BC,aAAI,yCAAA,EAAA,EAAA;AAKrC;AAKE,IAAMC,gBAAgB,CAACf,YACtBgB,cAAQH,UAAU;EACtBI,YAAY,MAAMjB;AACpB,CAAA;AAKK,IAAMkB,gBAAgBH,cAAcR,MAAAA;;;AC1J3C,YAAYY,aAAY;AAExB,SAAmBC,YAAY;AAC/B,SAASC,gBAAgB;AACzB,SAASC,wBAAwB;AACjC,SAASC,KAAKC,uBAAoD;AAClE,SAASC,QAAAA,aAAY;AAKd,IAAMC,wBAAwB,CAACC,OAAiB,gBAAgBA,EAAAA;AAShE,IAAMC,aAMT,cAAqBC,eAAOC,YAAY;;EAE1CC,aAAaC;;;;;;EAMbC,UAAU;IACR;IACA;IACA;IACA;;AAEJ,CAAA,EAAA;EACE,OAAOC,MAAMC,WAAuBC,eAA8B;AAEhE,QAAID,UAAUE,WAAW,OAAA,GAAU;AACjC,aAAOC,IAAIC,MAAMJ,UAAUK,MAAM,CAAA,CAAA;IACnC,WAAWL,UAAUE,WAAW,MAAA,GAAS;AACvC,aAAOC,IAAIC,MAAMJ,SAAAA;IACnB,WAAW,wBAAwBM,KAAKN,SAAAA,GAAY;AAClD,YAAM,CAACO,SAASC,QAAAA,IAAYR,UAAUS,MAAM,GAAA;AAI5C,aAAO,IAAIN,IAAIA,IAAIO,KAAKC,MAAM;QAACJ,YAAYN,gBAAgBW,kBAAkBL;QAASC;OAAS;IACjG,WAAW,cAAcF,KAAKN,SAAAA,GAAY;AACxC,aAAOG,IAAIU,kBAAkBb,SAAAA;IAC/B,OAAO;AACL,YAAM,IAAIc,MAAM,qCAAqCd,SAAAA,EAAW;IAClE;EACF;;;;EAKA,OAAOe,QACLC,QACAC,KACiF;AACjF,UAAMC,MAAMzB,WAAWM,MAAMkB,GAAAA;AAC7B,WAAOE,SAASJ,QAAQG,KAAKF,MAAAA;EAC/B;AACF;AAOO,IAAMI,aAAoBC,kBAAU5B,YAAY6B,KAAKC,IAAID,KAAKE,GAAG,GAAG;EACzEC,QAAQ,CAACC,OAAOC,UAAUC,iBAAiBC,QAAQpC,WAAWM,MAAM2B,KAAAA,CAAAA;EACpEI,QAAQ,CAACC,KAAKC,QAAQJ,iBAAiB7B,MAAMgC,GAAAA,EAAKE,SAAQ;EAC1DC,QAAQ;AACV,CAAA,EAAGvC,YAAY;EACbC,aAAaH,WAAW0C,IAAIxC,YAAYC;AAC1C,CAAA;;;ACpFA,YAAYwC,UAAU;AAEtB,YAAYC,cAAa;AACzB,YAAYC,aAAY;AACxB,YAAYC,YAAW;AACvB,YAAYC,YAAY;AACxB,YAAYC,aAAY;AAGxB,SAASC,qBAAqBC,wBAAAA,uBAAsBC,uBAAAA,4BAA2B;AAC/E,SAASC,QAAAA,aAAY;AACrB,SAASC,QAAAA,aAAY;AACrB,SAAkCC,iCAAiC;AACnE,SAASC,iBAAiB;;AAgBnB,IAAMC,gCAAgC,MAAA;AAK3C,SAAaC,cACXC,sBACOC,YAAI,aAAA;AACT,UAAMC,kBAAkB,OAAOC,uBAAeC;AAC9C,UAAMC,4BAA4B,OAAOC;AACzC,WAAO;MACLC,SAAS,CAACC,OACDP,YAAI,aAAA;AACT,cAAMQ,UAAUP,gBAAgBQ,WAAU;AAE1C,cAAMC,OAAOF,QAAQA,QAAQG,MAAMJ,EAAAA;AACnC,YAAIG,MAAM;AACR,iBAAOA;QACT;AAEA,eAAO,OAAON,0BAA0BQ,gBAAgBL,EAAAA,EAAIM,KACnDC,YAAIC,qBAAAA,GACJC,iBAAS,oBAAoB,MAAaC,aAAK,IAAIC,oBAAoBX,EAAAA,CAAAA,CAAAA,CAAAA;MAElF,CAAA;IACJ;EACF,CAAA,CAAA;AAEJ;AAEO,IAAMY,wCAAwC,MAAA;AAKnD,SAAarB,cACXsB,uBACOpB,YAAI,aAAA;AACT,UAAMC,kBAAkB,OAAOC,uBAAeC;AAC9C,UAAMK,UAAUP,gBAAgBQ,WAAU;AAE1C,UAAMY,iBAAiB,OAAOb,QAAQA,QAAQK,KAAYS,gBAAQd,QAAQe,KAAK,CAAA;AAC/EC,cAAUC,cAAcJ,cAAAA,GAAAA,QAAAA;;;;;;;;;AACxB,UAAMjB,4BAA4B,OAAOC;AAEzC,WAAO;MACLqB,aAAa,CAAClB,aAAAA;AACZ,cAAMmB,cAAc,CAACjB,SAAAA;AACnB,iBAAckB,WAAG,uBAAuBlB,KAAKmB,IAAI,EAAE,EAAE,WAAWC,OAAU;AACxE,gBAAIT,eAAeV,MAAMD,KAAKmB,IAAI,GAAG;AACnC,kBAASE,uBAAkBrB,IAAAA,GAAO;AAChC,sBAAM,IAAIsB,MAAM,2CAAA;cAClB;AAGA,oBAAM,EAAEC,OAAM,IAAK,OAAQZ,eAAea,OAAexB,KAAKmB,MAAMC,KAAAA;AACpE,qBAAOG;YACT;AAEA,kBAAM,EAAEE,YAAYC,YAAW,IAAaC,aAAIC,sBAAAA,EAAwB5B,KAAK6B,WAAW;AAExF,mBAAO,OAAOnC,0BACXoC,eAAeJ,aAAaN,KAAAA,EAC5BjB,KAAY4B,uBAAe,CAACC,WAAkBzB,aAAKyB,MAAAA,CAAAA,CAAAA;UACxD,CAAA;QACF;AAEA,eAAOlC,SAAQmC,GACN7B,WAAIN,SAAQG,OAAO,CAACD,MAAMkC,UAAgBC,mBAAcnC,IAAAA,IAAQiB,YAAYjB,IAAAA,IAAQ,IAAA,CAAA;MAE/F;IACF;EACF,CAAA,CAAA;AAEJ;AAEO,IAAMoC,wBAA8BC,gBACzClD,8BAAAA,GACAsB,sCAAAA,CAAAA;AAGF,IAAMmB,yBAAN,cAA6CU,aAAI,wCAAA,EAAA,EAAA;AAG5C;AAEL,IAAMC,YAAY,oBAAIC,QAAAA;AAUtB,IAAMnC,wBAAwB,CAACa,QAAAA;AAC7B,MAAIqB,UAAUE,IAAIvB,GAAAA,GAAK;AACrB,WAAOqB,UAAUZ,IAAIT,GAAAA;EACvB;AAIA,QAAMlB,OAAY0C,UAAKC,aAAazB,IAAGC,IAAI,GAAG;IAC5CyB,aAAa1B,IAAG0B;IAChBC,YAAYC,6BAA6B5B,IAAG6B,WAAW;;IAEvDC,SAAgBC;EAClB,CAAA,EAAGC,SAAStB,wBAAwB;IAAEH,YAAYP;EAAG,CAAA;AACrDqB,YAAUY,IAAIjC,KAAIlB,IAAAA;AAClB,SAAOA;AACT;AAKA,IAAM2C,eAAe,CAACxB,SAAAA;AACpB,QAAMiC,WAAWjC,KAAKkC,YAAW,EAAGC,QAAQ,iBAAiB,GAAA;AAC7DxC,YAAUsC,SAASG,MAAM,qBAAA,GAAA,QAAA;;;;;;;;;AACzB,SAAOH;AACT;AAGA,IAAMN,+BAA+B,CAACU,WAAAA;AACpC,UAAQA,OAAOC,IAAIC,MAAI;IACrB,KAAK;AACH,aAAOC,OAAOC,YACZJ,OAAOC,IAAII,mBAAmBzD,IAAI,CAAC0D,SAAS;QAACA,KAAK3C;QAAauB,aAAKqB,oBAAoBD,KAAKE,IAAI,CAAA;OAAG,CAAA;IAExG,KAAK;AACH,aAAO,CAAC;IACV;AACE,aAAOC,MAAK,2BAA2BT,OAAOC,IAAIC,IAAI,EAAE;EAC5D;AACF;AAMA,IAAMK,sBAAsB,CAACN,QAAAA;AAC3B,MAAIS,MAAKC,IAAIC,eAAeX,GAAAA,GAAM;AAChC,UAAMb,cAAca,IAAI5B,YAAYe,cAChCa,IAAI5B,YAAYe,cAAc,OAAOyB,WAAWZ,IAAI5B,YAAYe,cAC/DyB,WAAWZ,IAAI5B,YAAYe;AAChC,WAAOyB,WAAWxC,YAAY;MAAEe;IAAY,CAAA,EAAGa;EACjD;AAEA,SAAOA;AACT;AAEA,IAAM1C,gBAAgB,CAACuD,UAAAA;AACrB,SAAO,OAAQA,MAAcrE,UAAU,YAAY,OAAQqE,MAAc9C,WAAW;AACtF;",
+  "names": ["Context", "ArtifactDiffResolver", "Tag", "Effect", "Function", "GenerationObserver", "Object", "freeze", "make", "onBegin", "constant", "void", "onPart", "onBlock", "onEnd", "onMessage", "noop", "fromPrinter", "printer", "block", "sync", "pending", "printContentBlock", "message", "sender", "role", "printMessage", "BaseError", "AiModelError", "BaseError", "extend", "AiAssistantError", "Effect", "Function", "Option", "Template", "Obj", "ObjectVersion", "log", "Message", "trim", "formatSystemPrompt", "system", "blueprints", "objects", "gen", "blueprintDefs", "pipe", "forEach", "blueprint", "succeed", "instructions", "flatMap", "template", "trim", "Template", "processTemplate", "map", "length", "join", "undefined", "objectDefs", "object", "Obj", "getDXN", "getTypename", "filter", "def", "parts", "formatUserPrompt", "prompt", "history", "blocks", "artifactDiffResolver", "serviceOption", "ArtifactDiffResolver", "isSome", "versions", "gatherObjectVersions", "artifactDiff", "tryPromise", "try", "value", "resolve", "entries", "id", "version", "lastVersion", "catch", "AiAssistantError", "wrap", "message", "log", "ObjectVersion", "equals", "get", "delete", "push", "_tag", "objectId", "size", "createArtifactUpdateBlock", "make", "Message", "created", "Date", "toISOString", "sender", "role", "text", "messages", "artifactIds", "Map", "block", "set", "disposition", "diff", "LanguageModel", "Chunk", "Effect", "Stream", "AiParser", "AiPreprocessor", "callTool", "withoutToolCallParising", "todo", "Obj", "TracingService", "log", "Message", "AiSession", "_semaphore", "runSync", "makeSemaphore", "_history", "_pending", "_started", "_ended", "_toolCalls", "_options", "duration", "toolCalls", "run", "prompt", "system", "systemTemplate", "history", "objects", "blueprints", "toolkit", "observer", "GenerationObserver", "noop", "onOutput", "void", "gen", "Date", "now", "pending", "submitMessage", "fnUntraced", "message", "push", "onMessage", "TracingService", "emitConverationMessage", "promptMessage", "formatUserPrompt", "formatSystemPrompt", "pipe", "orDie", "log", "snippet", "createSnippet", "length", "AiPreprocessor", "preprocessPrompt", "cacheControl", "messages", "streamText", "disableToolCallResolution", "withoutToolCallParising", "AiParser", "parseResponse", "onBegin", "onBlock", "block", "onPart", "part", "onEnd", "summary", "mapEffect", "Obj", "make", "Message", "created", "toISOString", "sender", "role", "blocks", "runCollect", "map", "toArray", "flatMap", "filter", "_tag", "providerExecuted", "Error", "toolResults", "forEach", "callTool", "provide", "layerSubframe", "context", "parentMessage", "id", "toolCallId", "tools", "withPermits", "withSpan", "runStructured", "_schema", "todo", "text", "len", "slice", "join", "Toolkit", "Effect", "ToolExecutionService", "ToolResolverService", "isTruthy", "createToolkit", "toolkit", "toolkitProp", "blueprints", "gen", "blueprintToolkit", "ToolResolverService", "resolveToolkit", "flatMap", "tools", "blueprintToolHandler", "toContext", "ToolExecutionService", "handlersFor", "merge", "filter", "isTruthy", "pipe", "provide", "Schema", "getVersion", "ObjectId", "ObjectVersionSchema", "Unknown", "VersionPinSchema", "Struct", "objectId", "ObjectId", "version", "VersionPin", "DISPOSITION", "fromObject", "object", "make", "id", "getVersion", "createBlock", "pin", "_tag", "disposition", "data", "JSON", "stringify", "empty", "merge", "Toolkit", "Context", "Effect", "Layer", "Pipeable", "TypeId", "make", "toolkit", "layer", "handlers", "pipe", "provide", "pipeArguments", "arguments", "empty", "merge", "toolkits", "map", "t", "mergeAll", "Provider", "Tag", "providerLayer", "succeed", "getToolkit", "providerEmpty", "Schema", "Type", "Database", "EncodedReference", "DXN", "LOCAL_SPACE_TAG", "trim", "createArtifactElement", "id", "ArtifactId", "String", "annotations", "description", "trim", "examples", "toDXN", "reference", "owningSpaceId", "startsWith", "DXN", "parse", "slice", "test", "spaceId", "objectId", "split", "kind", "ECHO", "LOCAL_SPACE_TAG", "fromLocalObjectId", "Error", "resolve", "schema", "ref", "dxn", "Database", "RefFromLLM", "transform", "Type", "Ref", "Obj", "decode", "fromA", "fromI", "EncodedReference", "fromDXN", "encode", "toI", "toA", "toString", "strict", "ast", "Tool", "Context", "Effect", "Layer", "Record", "Schema", "AiToolNotFoundError", "ToolExecutionService", "ToolResolverService", "todo", "Type", "FunctionInvocationService", "invariant", "makeToolResolverFromFunctions", "effect", "ToolResolverService", "gen", "toolkitProvider", "GenericToolkit", "Provider", "functionInvocationService", "FunctionInvocationService", "resolve", "id", "toolkit", "getToolkit", "tool", "tools", "resolveFunction", "pipe", "map", "projectFunctionToTool", "catchTag", "fail", "AiToolNotFoundError", "makeToolExecutionServiceFromFunctions", "ToolExecutionService", "toolkitHandler", "provide", "layer", "invariant", "isHandlerLike", "handlersFor", "makeHandler", "fn", "name", "input", "isProviderDefined", "Error", "result", "handle", "definition", "functionDef", "get", "FunctionToolAnnotation", "annotations", "invokeFunction", "catchAllDefect", "defect", "of", "_name", "isUserDefined", "ToolExecutionServices", "mergeAll", "Tag", "toolCache", "WeakMap", "has", "make", "makeToolName", "description", "parameters", "createStructFieldsFromSchema", "inputSchema", "failure", "Any", "annotate", "set", "toolName", "toLowerCase", "replace", "match", "schema", "ast", "_tag", "Object", "fromEntries", "propertySignatures", "prop", "mapSchemaTypeForLLM", "type", "todo", "Type", "Ref", "isRefSchemaAST", "RefFromLLM", "value"]
+}