@dxos/assistant 0.8.2-main.f11618f → 0.8.2-main.fbd8ed0

package/dist/lib/node/{chunk-7VH7NMIM.cjs → chunk-NOB7CIIC.cjs}

@@ -26,9 +26,11 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
   mod
 ));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-var chunk_7VH7NMIM_exports = {};
-__export(chunk_7VH7NMIM_exports, {
+var chunk_NOB7CIIC_exports = {};
+__export(chunk_NOB7CIIC_exports, {
   AIServiceEdgeClient: () => AIServiceEdgeClient,
+  AIServiecOverloadedError: () => AIServiecOverloadedError,
+  AISession: () => AISession,
   DEFAULT_EDGE_MODEL: () => DEFAULT_EDGE_MODEL,
   DEFAULT_EDGE_MODELS: () => DEFAULT_EDGE_MODELS,
   DEFAULT_OLLAMA_ENDPOINT: () => DEFAULT_OLLAMA_ENDPOINT,
@@ -44,6 +46,7 @@ __export(chunk_7VH7NMIM_exports, {
   OllamaClient: () => OllamaClient,
   StreamTransform: () => StreamTransform,
   ToolTypes: () => ToolTypes,
+  VersionPin: () => VersionPin,
   createArtifactElement: () => createArtifactElement,
   createGenerationStream: () => createGenerationStream,
   emitMessageAsEvents: () => emitMessageAsEvents,
@@ -60,7 +63,7 @@ __export(chunk_7VH7NMIM_exports, {
   runTools: () => runTools,
   trim: () => trim
 });
-module.exports = __toCommonJS(chunk_7VH7NMIM_exports);
+module.exports = __toCommonJS(chunk_NOB7CIIC_exports);
 var import_async = require("@dxos/async");
 var import_log = require("@dxos/log");
 var import_invariant = require("@dxos/invariant");
@@ -72,28 +75,41 @@ var import_echo_schema = require("@dxos/echo-schema");
 var import_invariant2 = require("@dxos/invariant");
 var import_log3 = require("@dxos/log");
 var import_effect2 = require("effect");
-var import_artifact3 = require("@dxos/artifact");
+var import_echo_db = require("@dxos/echo-db");
 var import_echo_schema2 = require("@dxos/echo-schema");
+var import_effect3 = require("effect");
+var import_artifact3 = require("@dxos/artifact");
+var import_async2 = require("@dxos/async");
+var import_echo_db2 = require("@dxos/echo-db");
+var import_echo_schema3 = require("@dxos/echo-schema");
 var import_invariant3 = require("@dxos/invariant");
 var import_log4 = require("@dxos/log");
-var P = __toESM(require("parsimmon"));
+var import_effect4 = require("effect");
 var import_artifact4 = require("@dxos/artifact");
-var import_async2 = require("@dxos/async");
-var import_echo_schema3 = require("@dxos/echo-schema");
+var import_echo_schema4 = require("@dxos/echo-schema");
 var import_invariant4 = require("@dxos/invariant");
 var import_log5 = require("@dxos/log");
+var import_artifact5 = require("@dxos/artifact");
+var import_async3 = require("@dxos/async");
+var import_echo_schema5 = require("@dxos/echo-schema");
+var import_invariant5 = require("@dxos/invariant");
+var import_log6 = require("@dxos/log");
 var import_util = require("@dxos/util");
+var P = __toESM(require("parsimmon"));
+var import_echo_schema6 = require("@dxos/echo-schema");
+var import_invariant6 = require("@dxos/invariant");
+var import_log7 = require("@dxos/log");
 var import_parjs = require("#parjs");
-var import_log6 = require("@dxos/log");
+var import_log8 = require("@dxos/log");
 var import_util2 = require("@dxos/util");
-var import_invariant5 = require("@dxos/invariant");
-var import_log7 = require("@dxos/log");
+var import_invariant7 = require("@dxos/invariant");
+var import_log9 = require("@dxos/log");
 var import_debug = require("@dxos/debug");
-var import_echo_schema4 = require("@dxos/echo-schema");
+var import_echo_schema7 = require("@dxos/echo-schema");
 var import_debug2 = require("@dxos/debug");
-var import_echo_db = require("@dxos/echo-db");
-var import_echo_schema5 = require("@dxos/echo-schema");
-var import_log8 = require("@dxos/log");
+var import_echo_db3 = require("@dxos/echo-db");
+var import_echo_schema8 = require("@dxos/echo-schema");
+var import_log10 = require("@dxos/log");
 var DEFAULT_EDGE_MODEL = "@anthropic/claude-3-5-sonnet-20241022";
 var DEFAULT_EDGE_MODELS = [
   // AI Gateway.
@@ -608,178 +624,12 @@ var GenerateResponse = import_effect.Schema.Struct({
   */
   tokenCount: import_effect.Schema.optional(import_effect.Schema.Number)
 });
-var __dxlog_file3 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/conversation/conversation.ts";
-var runLLM = async (params) => {
-  let conversationResult = null;
-  const history = params.history ?? [];
-  const generate = async () => {
-    (0, import_log3.log)("llm generate", {
-      tools: params.tools
-    }, {
-      F: __dxlog_file3,
-      L: 44,
-      S: void 0,
-      C: (f, a) => f(...a)
-    });
-    const beginTs = Date.now();
-    const stream = await params.client.execStream({
-      model: params.model,
-      history,
-      systemPrompt: params.system,
-      tools: params.tools
-    });
-    for await (const event of stream) {
-      params.logger?.(event);
-    }
-    await stream.complete();
-    const messages = [];
-    const message = messages.at(-1);
-    (0, import_invariant2.invariant)(message, void 0, {
-      F: __dxlog_file3,
-      L: 61,
-      S: void 0,
-      A: [
-        "message",
-        ""
-      ]
-    });
-    (0, import_log3.log)("llm result", {
-      time: Date.now() - beginTs,
-      message
-    }, {
-      F: __dxlog_file3,
-      L: 63,
-      S: void 0,
-      C: (f, a) => f(...a)
-    });
-    (0, import_invariant2.invariant)(message, void 0, {
-      F: __dxlog_file3,
-      L: 64,
-      S: void 0,
-      A: [
-        "message",
-        ""
-      ]
-    });
-    params.logger?.({
-      type: "message",
-      message
-    });
-    history.push(...messages.map((message2) => ({
-      ...message2,
-      content: message2.content.filter((block) => block.type !== "image")
-    })));
-    const isToolUse2 = message.content.at(-1)?.type === "tool_use";
-    if (isToolUse2) {
-      const toolCalls = message.content.filter((block) => block.type === "tool_use");
-      (0, import_invariant2.invariant)(toolCalls.length === 1, void 0, {
-        F: __dxlog_file3,
-        L: 75,
-        S: void 0,
-        A: [
-          "toolCalls.length === 1",
-          ""
-        ]
-      });
-      const toolCall = toolCalls[0];
-      const tool = params.tools.find((tool2) => tool2.name === toolCall.name);
-      if (!tool) {
-        throw new Error(`Tool not found: ${toolCall.name}`);
-      }
-      (0, import_invariant2.invariant)(tool.execute, void 0, {
-        F: __dxlog_file3,
-        L: 82,
-        S: void 0,
-        A: [
-          "tool.execute",
-          ""
-        ]
-      });
-      const toolResult = await tool.execute(toolCall.input);
-      switch (toolResult.kind) {
-        case "error": {
-          import_log3.log.warn("tool error", {
-            message: toolResult.message
-          }, {
-            F: __dxlog_file3,
-            L: 86,
-            S: void 0,
-            C: (f, a) => f(...a)
-          });
-          const resultMessage = {
-            id: import_echo_schema.ObjectId.random(),
-            role: "user",
-            content: [
-              {
-                type: "tool_result",
-                toolUseId: toolCall.id,
-                content: toolResult.message,
-                isError: true
-              }
-            ]
-          };
-          params.logger?.({
-            type: "message",
-            message: resultMessage
-          });
-          history.push(resultMessage);
-          return true;
-        }
-        case "success": {
-          (0, import_log3.log)("tool success", {
-            result: toolResult.result
-          }, {
-            F: __dxlog_file3,
-            L: 106,
-            S: void 0,
-            C: (f, a) => f(...a)
-          });
-          const resultMessage = {
-            id: import_echo_schema.ObjectId.random(),
-            role: "user",
-            content: [
-              {
-                type: "tool_result",
-                toolUseId: toolCall.id,
-                content: JSON.stringify(toolResult.result)
-              }
-            ]
-          };
-          params.logger?.({
-            type: "message",
-            message: resultMessage
-          });
-          history.push(resultMessage);
-          return true;
-        }
-        case "break": {
-          (0, import_log3.log)("tool break", {
-            result: toolResult.result
-          }, {
-            F: __dxlog_file3,
-            L: 125,
-            S: void 0,
-            C: (f, a) => f(...a)
-          });
-          conversationResult = toolResult.result;
-          return false;
-        }
-      }
-    }
-    return false;
-  };
-  while (await generate()) {
-  }
-  return {
-    result: conversationResult,
-    history
-  };
-};
+var __dxlog_file3 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/session/tools.ts";
 var isToolUse = (message, { onlyToolNames } = {}) => {
   const block = message.content.at(-1);
   (0, import_invariant2.invariant)(block, void 0, {
     F: __dxlog_file3,
-    L: 146,
+    L: 12,
     S: void 0,
     A: [
       "block",
@@ -792,7 +642,7 @@ var runTools = async ({ message, tools, extensions }) => {
   const toolCalls = message.content.filter((block) => block.type === "tool_use");
   (0, import_invariant2.invariant)(toolCalls.length === 1, void 0, {
     F: __dxlog_file3,
-    L: 168,
+    L: 34,
     S: void 0,
     A: [
       "toolCalls.length === 1",
@@ -822,7 +672,7 @@ var runTools = async ({ message, tools, extensions }) => {
   try {
     (0, import_invariant2.invariant)(tool.execute, void 0, {
       F: __dxlog_file3,
-      L: 193,
+      L: 59,
       S: void 0,
       A: [
         "tool.execute",
@@ -837,7 +687,7 @@ var runTools = async ({ message, tools, extensions }) => {
       error
     }, {
       F: __dxlog_file3,
-      L: 196,
+      L: 62,
       S: void 0,
       C: (f, a) => f(...a)
     });
@@ -849,7 +699,7 @@ var runTools = async ({ message, tools, extensions }) => {
         message: toolResult.message
       }, {
         F: __dxlog_file3,
-        L: 202,
+        L: 68,
         S: void 0,
         C: (f, a) => f(...a)
       });
@@ -874,7 +724,7 @@ var runTools = async ({ message, tools, extensions }) => {
         result: toolResult.result
       }, {
         F: __dxlog_file3,
-        L: 223,
+        L: 89,
         S: void 0,
         C: (f, a) => f(...a)
       });
@@ -899,7 +749,7 @@ var runTools = async ({ message, tools, extensions }) => {
         result: toolResult.result
       }, {
         F: __dxlog_file3,
-        L: 244,
+        L: 110,
         S: void 0,
         C: (f, a) => f(...a)
       });
@@ -910,6 +760,31 @@ var runTools = async ({ message, tools, extensions }) => {
     }
   }
 };
+var ObjectVersionSchema = import_effect2.Schema.Unknown;
+var VersionPinSchema = import_effect2.Schema.Struct({
+  // TODO(dmaretskyi): Use Ref when those support encoding.
+  objectId: import_echo_schema2.ObjectId,
+  // TODO(dmaretskyi): Could be opaque
+  version: ObjectVersionSchema
+});
+var VersionPin = class extends VersionPinSchema {
+  static {
+    this.DISPOSITION = "version-pin";
+  }
+  static fromObject(object) {
+    return VersionPin.make({
+      objectId: object.id,
+      version: (0, import_echo_db.getVersion)(object)
+    });
+  }
+  static createBlock(pin) {
+    return {
+      type: "json",
+      disposition: VersionPin.DISPOSITION,
+      json: JSON.stringify(pin)
+    };
+  }
+};
 var __dxlog_file4 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/ai-service/ollama-client.ts";
 var OllamaClient = class {
   /**
@@ -944,7 +819,7 @@ var OllamaClient = class {
     ];
   }
   async *_generateStream(request, signal) {
-    (0, import_invariant3.invariant)(request.prompt === void 0, "Prompt must be in history", {
+    (0, import_invariant4.invariant)(request.prompt === void 0, "Prompt must be in history", {
       F: __dxlog_file4,
       L: 88,
       S: this,
@@ -1033,7 +908,7 @@ var OllamaClient = class {
         }));
       }
     }
-    (0, import_log4.log)("generate request", {
+    (0, import_log5.log)("generate request", {
       history: request.history,
       prompt: request.prompt,
       ollamaRequest
@@ -1059,7 +934,7 @@ var OllamaClient = class {
     }
     const reader = response.body.pipeThrough(new TextDecoderStream()).pipeThrough(new OllamaDecoderStream()).getReader();
     try {
-      const messageId = import_echo_schema2.ObjectId.random();
+      const messageId = import_echo_schema4.ObjectId.random();
       yield {
         type: "message_start",
         message: {
@@ -1083,7 +958,7 @@ var OllamaClient = class {
         if (done) {
           break;
         }
-        (0, import_log4.log)("ollama data", {
+        (0, import_log5.log)("ollama data", {
           data
         }, {
           F: __dxlog_file4,
@@ -1103,7 +978,7 @@ var OllamaClient = class {
         }
         if (data.message?.tool_calls && data.message?.tool_calls?.length > 0) {
           if (data.message.tool_calls.length > 1) {
-            import_log4.log.warn("Ollama returned multiple tool calls. Only the first one will be used.", void 0, {
+            import_log5.log.warn("Ollama returned multiple tool calls. Only the first one will be used.", void 0, {
               F: __dxlog_file4,
               L: 261,
               S: this,
@@ -1119,7 +994,7 @@ var OllamaClient = class {
             index: currentBlockIndex,
             content: {
               type: "tool_use",
-              id: import_echo_schema2.ObjectId.random(),
+              id: import_echo_schema4.ObjectId.random(),
               name: data.message.tool_calls[0].function.name,
               input: sanitizeToolArguments(data.message.tool_calls[0].function.arguments)
             }
@@ -1208,19 +1083,19 @@ var sanitizeToolArguments = (args) => {
 };
 var SAMPLE_IMAGE_URL = "https://images.nightcafe.studio/jobs/BNmcRhHCM1JRKoUtqSei/BNmcRhHCM1JRKoUtqSei--1--5b9rv.jpg";
 var WELL_KNOWN_TOOLS = {
-  [ToolTypes.TextToImage]: (0, import_artifact3.defineTool)("system", {
+  [ToolTypes.TextToImage]: (0, import_artifact4.defineTool)("system", {
     name: "text-to-image",
     description: "Generate an image from a text prompt",
-    schema: import_effect2.Schema.Struct({
-      prompt: import_effect2.Schema.String.annotations({
+    schema: import_effect4.Schema.Struct({
+      prompt: import_effect4.Schema.String.annotations({
         description: "The text prompt describing the image to generate"
       })
     }),
     execute: async ({ prompt }) => {
       const image = await fetch(SAMPLE_IMAGE_URL).then(async (res) => Buffer.from(await res.arrayBuffer()));
       const imageBase64 = image.toString("base64");
-      const id = import_echo_schema2.ObjectId.random();
-      return import_artifact3.ToolResult.Success(`The generation was successful. Image ID: ${id} Prompt: ${prompt}`, [
+      const id = import_echo_schema4.ObjectId.random();
+      return import_artifact4.ToolResult.Success(`The generation was successful. Image ID: ${id} Prompt: ${prompt}`, [
         {
           type: "image",
           id,
@@ -1255,12 +1130,12 @@ var OllamaDecoderStream = class extends TransformStream {
     this._buffer = "";
   }
 };
-var ModelDoesNotSupportToolsError = class extends import_effect2.Schema.TaggedError()("ModelDoesNotSupportToolsError", {
-  model: import_effect2.Schema.String
+var ModelDoesNotSupportToolsError = class extends import_effect4.Schema.TaggedError()("ModelDoesNotSupportToolsError", {
+  model: import_effect4.Schema.String
 }) {
 };
-var ModelNotFoundError = class extends import_effect2.Schema.TaggedError()("ModelNotFoundError", {
-  model: import_effect2.Schema.String
+var ModelNotFoundError = class extends import_effect4.Schema.TaggedError()("ModelNotFoundError", {
+  model: import_effect4.Schema.String
 }) {
 };
 var parseOllamaError = (error) => {
@@ -1378,10 +1253,10 @@ var StreamTransform = class {
 var __dxlog_file5 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/ai-service/parser.ts";
 var MixedStreamParser = class {
   constructor() {
-    this.message = new import_async2.Event();
-    this.block = new import_async2.Event();
-    this.update = new import_async2.Event();
-    this.streamEvent = new import_async2.Event();
+    this.message = new import_async3.Event();
+    this.block = new import_async3.Event();
+    this.update = new import_async3.Event();
+    this.streamEvent = new import_async3.Event();
   }
   _emitBlock(contentBlock, streamBlock) {
     const messageBlock = streamBlock ? mergeMessageBlock(contentBlock, streamBlock) : contentBlock;
@@ -1389,7 +1264,7 @@ var MixedStreamParser = class {
       if (messageBlock.type === "text" && messageBlock.text.length === 0) {
         return;
       }
-      (0, import_invariant4.invariant)(this._message, void 0, {
+      (0, import_invariant5.invariant)(this._message, void 0, {
         F: __dxlog_file5,
         L: 49,
         S: this,
@@ -1419,7 +1294,7 @@ var MixedStreamParser = class {
     const stack = [];
     const messagesCollected = [];
     for await (const event of stream) {
-      (0, import_log5.log)("streamEvent", {
+      (0, import_log6.log)("streamEvent", {
         event
       }, {
         F: __dxlog_file5,
@@ -1434,14 +1309,14 @@ var MixedStreamParser = class {
         //
         case "message_start": {
           if (this._message) {
-            import_log5.log.warn("unexpected message_start", void 0, {
+            import_log6.log.warn("unexpected message_start", void 0, {
               F: __dxlog_file5,
               L: 94,
               S: this,
               C: (f, a) => f(...a)
             });
           }
-          this._message = (0, import_echo_schema3.create)(import_artifact4.Message, {
+          this._message = (0, import_echo_schema5.create)(import_artifact5.Message, {
             role: event.message.role,
             content: [
               ...event.message.content
@@ -1452,7 +1327,7 @@ var MixedStreamParser = class {
         }
         case "message_delta": {
           if (!this._message) {
-            import_log5.log.warn("unexpected message_delta", void 0, {
+            import_log6.log.warn("unexpected message_delta", void 0, {
               F: __dxlog_file5,
               L: 104,
               S: this,
@@ -1463,7 +1338,7 @@ var MixedStreamParser = class {
         }
         case "message_stop": {
           if (!this._message) {
-            import_log5.log.warn("unexpected message_stop", void 0, {
+            import_log6.log.warn("unexpected message_stop", void 0, {
               F: __dxlog_file5,
               L: 112,
               S: this,
@@ -1480,7 +1355,7 @@ var MixedStreamParser = class {
         //
         case "content_block_start": {
           if (contentBlock) {
-            import_log5.log.warn("unexpected content_block_start", {
+            import_log6.log.warn("unexpected content_block_start", {
               content: contentBlock
             }, {
               F: __dxlog_file5,
@@ -1494,7 +1369,7 @@ var MixedStreamParser = class {
         }
         case "content_block_delta": {
           if (!contentBlock) {
-            import_log5.log.warn("unexpected content_block_delta", void 0, {
+            import_log6.log.warn("unexpected content_block_delta", void 0, {
               F: __dxlog_file5,
               L: 136,
               S: this,
@@ -1509,7 +1384,7 @@ var MixedStreamParser = class {
             case "text_delta": {
               const chunks = transformer.transform(event.delta.text);
               for (const chunk of chunks) {
-                (0, import_log5.log)("text_delta", {
+                (0, import_log6.log)("text_delta", {
                   chunk
                 }, {
                   F: __dxlog_file5,
@@ -1528,7 +1403,7 @@ var MixedStreamParser = class {
                       } else if (chunk.closing) {
                         if (stack.length > 0) {
                           const top = stack.pop();
-                          (0, import_invariant4.invariant)(top && top.type === "tag", void 0, {
+                          (0, import_invariant5.invariant)(top && top.type === "tag", void 0, {
                             F: __dxlog_file5,
                             L: 160,
                             S: this,
@@ -1537,7 +1412,7 @@ var MixedStreamParser = class {
                               ""
                             ]
                           });
-                          (0, import_log5.log)("pop", {
+                          (0, import_log6.log)("pop", {
                             top
                           }, {
                             F: __dxlog_file5,
@@ -1560,7 +1435,7 @@ var MixedStreamParser = class {
                         streamBlock.content.push(chunk);
                       } else {
                         const last = streamBlock.content.at(-1);
-                        (0, import_invariant4.invariant)(last, void 0, {
+                        (0, import_invariant5.invariant)(last, void 0, {
                           F: __dxlog_file5,
                           L: 178,
                           S: this,
@@ -1614,7 +1489,7 @@ var MixedStreamParser = class {
             //
             case "input_json_delta": {
               if (!contentBlock) {
-                import_log5.log.warn("unexpected input_json_delta", {
+                import_log6.log.warn("unexpected input_json_delta", {
                   contentBlock
                 }, {
                   F: __dxlog_file5,
@@ -1639,7 +1514,7 @@ var MixedStreamParser = class {
         }
         case "content_block_stop": {
           if (!contentBlock) {
-            import_log5.log.warn("unexpected content_block_stop", void 0, {
+            import_log6.log.warn("unexpected content_block_stop", void 0, {
               F: __dxlog_file5,
               L: 245,
               S: this,
@@ -1661,7 +1536,7 @@ var MixedStreamParser = class {
   }
 };
 var mergeMessageBlock = (contentBlock, streamBlock) => {
-  (0, import_log5.log)("mergeMessageBlock", {
+  (0, import_log6.log)("mergeMessageBlock", {
     contentBlock,
     streamBlock
   }, {
@@ -1776,6 +1651,585 @@ var mergeMessageBlock = (contentBlock, streamBlock) => {
     }
   }
 };
+function _ts_decorate(decorators, target, key, desc) {
+  var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+  if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+  else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+  return c > 3 && r && Object.defineProperty(target, key, r), r;
+}
+var __dxlog_file6 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/session/session.ts";
+var AISession = class _AISession {
+  constructor(_options) {
+    this._options = _options;
+    this._parser = new MixedStreamParser();
+    this._pending = [];
+    this._history = [];
+    this.message = this._parser.message;
+    this.block = this._parser.block;
+    this.update = this._parser.update;
+    this.streamEvent = this._parser.streamEvent;
+    this.userMessage = new import_async2.Event();
+    this._parser.message.on((message) => {
+      this._pending.push(message);
+    });
+  }
+  async run(options) {
+    const systemTools = [];
+    switch (this._options.operationModel) {
+      case "planning":
+        systemTools.push(this._createQueryArtifactsTool(options.artifacts), this._createPlanningTool(options));
+        break;
+      case "importing":
+        systemTools.push(this._createQueryArtifactsTool(options.artifacts), this._createRequireTool(options.artifacts, (artifactDefinitionIds) => {
+          for (const artifactDefinitionId of artifactDefinitionIds) {
+            requiredArtifactIds.add(artifactDefinitionId);
+          }
+        }));
+        break;
+      case "configured":
+        break;
+      default:
+        throw new TypeError(`Invalid operation model: ${this._options.operationModel}`);
+    }
+    this._history = [
+      ...options.history
+    ];
+    this._pending = [
+      await this._formatUserPrompt(options.artifactDiffResolver, options.prompt, options.history)
+    ];
+    this.userMessage.emit(this._pending.at(-1));
+    this._stream = void 0;
+    let error;
+    const requiredArtifactIds = new Set(options.requiredArtifactIds ?? []);
+    try {
+      let more = false;
+      do {
+        const tools = [
+          ...systemTools,
+          ...options.tools,
+          ...options.artifacts.filter((artifact) => requiredArtifactIds.has(artifact.id)).flatMap((artifact) => artifact.tools)
+        ];
+        (0, import_log4.log)("request", {
+          pending: this._pending.length,
+          history: this._history.length,
+          tools: tools.map((tool) => tool.name)
+        }, {
+          F: __dxlog_file6,
+          L: 181,
+          S: this,
+          C: (f, a) => f(...a)
+        });
+        this._stream = await options.client.execStream({
+          ...options.generationOptions ?? {},
+          // TODO(burdon): Rename messages or separate history/message.
+          history: [
+            ...this._history,
+            ...this._pending
+          ],
+          tools,
+          systemPrompt: options.systemPrompt ?? createBaseInstructions({
+            availableArtifacts: Array.from(requiredArtifactIds),
+            operationModel: this._options.operationModel
+          })
+        });
+        await this._parser.parse(this._stream);
+        await this._stream.complete();
+        (0, import_log4.log)("response", {
+          pending: this._pending
+        }, {
+          F: __dxlog_file6,
+          L: 206,
+          S: this,
+          C: (f, a) => f(...a)
+        });
+        more = false;
+        const message = this._pending.at(-1);
+        (0, import_invariant3.invariant)(message, void 0, {
+          F: __dxlog_file6,
+          L: 211,
+          S: this,
+          A: [
+            "message",
+            ""
+          ]
+        });
+        if (isToolUse(message)) {
+          (0, import_log4.log)("tool request...", void 0, {
+            F: __dxlog_file6,
+            L: 213,
+            S: this,
+            C: (f, a) => f(...a)
+          });
+          const response = await runTools({
+            message: this._pending.at(-1),
+            tools,
+            extensions: options.extensions ?? {}
+          });
+          (0, import_log4.log)("tool response", {
+            response
+          }, {
+            F: __dxlog_file6,
+            L: 220,
+            S: this,
+            C: (f, a) => f(...a)
+          });
+          switch (response.type) {
+            case "continue": {
+              this._pending = [
+                ...this._pending,
+                response.message
+              ];
+              this.userMessage.emit(response.message);
+              more = true;
+              break;
+            }
+          }
+        }
+      } while (more);
+    } catch (err) {
+      import_log4.log.catch(err, void 0, {
+        F: __dxlog_file6,
+        L: 232,
+        S: this,
+        C: (f, a) => f(...a)
+      });
+      if (err instanceof Error && err.message.includes("Overloaded")) {
+        error = new AIServiecOverloadedError("AI service overloaded", {
+          cause: err
+        });
+      } else {
+        error = new Error("AI service error", {
+          cause: err
+        });
+      }
+    } finally {
+      this._stream = void 0;
+    }
+    if (error) {
+      throw error;
+    }
+    return this._pending;
+  }
+  async runStructured(schema, options) {
+    const parser = (0, import_artifact3.structuredOutputParser)(schema);
+    const result = await this.run({
+      ...options,
+      tools: [
+        ...options.tools,
+        parser.tool
+      ]
+    });
+    return parser.getResult(result);
+  }
+  async _formatUserPrompt(artifactDiffResolver, prompt, history) {
+    const prelude = [];
+    if (artifactDiffResolver) {
+      const versions = gatherObjectVersions(history);
+      const artifactDiff = await artifactDiffResolver(Array.from(versions.entries()).map(([id, version]) => ({
+        id,
+        lastVersion: version
+      })));
+      import_log4.log.info("vision", {
+        artifactDiff,
+        versions
+      }, {
+        F: __dxlog_file6,
+        L: 272,
+        S: this,
+        C: (f, a) => f(...a)
+      });
+      for (const [id, { version }] of [
+        ...artifactDiff.entries()
+      ]) {
+        if (import_echo_db2.ObjectVersion.equals(version, versions.get(id))) {
+          artifactDiff.delete(id);
+          continue;
+        }
+        prelude.push(VersionPin.createBlock(VersionPin.make({
+          objectId: id,
+          version
+        })));
+      }
+      if (artifactDiff.size > 0) {
+        prelude.push(createArtifactUpdateBlock(artifactDiff));
+      }
+    }
+    return (0, import_echo_schema3.create)(import_artifact3.Message, {
+      role: "user",
+      content: [
+        ...prelude,
+        {
+          type: "text",
+          text: prompt
+        }
+      ]
+    });
+  }
+  _createQueryArtifactsTool(artifacts) {
+    return (0, import_artifact3.defineTool)("system", {
+      name: "query_artifact_definitions",
+      description: "Query the available artifact definitions",
+      schema: import_effect3.Schema.Struct({}),
+      execute: async () => {
+        return import_artifact3.ToolResult.Success({
+          artifactDefinitions: artifacts.map((artifact) => ({
+            id: artifact.id,
+            name: artifact.name,
+            description: artifact.description
+          }))
+        });
+      }
+    });
+  }
+  _createRequireTool(artifacts, onRequire) {
+    return (0, import_artifact3.defineTool)("system", {
+      name: "require_artifact_definitions",
+      description: "Require the use of specific artifact definitions. This will allow the model to interact with artifact definitions and use their tools.",
+      schema: import_effect3.Schema.Struct({
+        artifactDefinitionIds: import_effect3.Schema.Array(import_effect3.Schema.String).annotations({
+          description: "The ids of the artifact definitions to require",
+          examples: [
+            [
+              "artifact:dxos.org/example/Test"
+            ]
+          ]
+        })
+      }),
+      execute: async ({ artifactDefinitionIds }) => {
+        const missingArtifactDefinitionIds = artifactDefinitionIds.filter((artifactId) => !artifacts.some((artifact) => artifact.id === artifactId));
+        if (missingArtifactDefinitionIds.length > 0) {
+          return import_artifact3.ToolResult.Error(`One or more artifact definition ids are invalid: ${missingArtifactDefinitionIds.join(", ")}`);
+        }
+        onRequire(artifactDefinitionIds);
+        return import_artifact3.ToolResult.Success({});
+      }
+    });
+  }
+  _createPlanningTool(options) {
+    return (0, import_artifact3.defineTool)("system", {
+      name: "create_plan",
+      description: "Create a plan. Make sure that each step is independent and can be executed solely on the data returned by the previous step. The steps only share the data that is specified in the plan.",
+      schema: import_effect3.Schema.Struct({
+        goal: import_effect3.Schema.String.annotations({
+          description: "The goal that the plan will achieve."
+        }),
+        steps: import_effect3.Schema.Array(import_effect3.Schema.Struct({
+          action: import_effect3.Schema.String.annotations({
+            description: "Complete, detailed action to perform."
+          }),
+          input: import_effect3.Schema.String.annotations({
+            description: "The required input data for this step. First step must not require any data."
+          }),
+          output: import_effect3.Schema.String.annotations({
+            description: "The output data from this step. This is passed to the next step as input."
+          }),
+          requiredArtifactIds: import_effect3.Schema.Array(import_effect3.Schema.String).annotations({
+            description: "The ids of the artifacts required to perform this step.",
+            examples: [
+              [
+                "artifact:dxos.org/example/Test"
+              ]
+            ]
+          })
+        })).annotations({
+          description: "Steps"
+        })
+      }),
+      execute: async ({ goal, steps }) => {
+        const missingArtifactIds = steps.flatMap((step) => step.requiredArtifactIds).filter((artifactId) => !options.artifacts.some((artifact) => artifact.id === artifactId));
+        if (missingArtifactIds.length > 0) {
+          return import_artifact3.ToolResult.Error(`One or more artifact ids are invalid: ${missingArtifactIds.join(", ")}`);
+        }
+        const stepResults = [];
+        for (const step of steps) {
+          (0, import_log4.log)("executing step", {
+            action: step.action
+          }, {
+            F: __dxlog_file6,
+            L: 379,
+            S: this,
+            C: (f, a) => f(...a)
+          });
+          const session = new _AISession({
+            operationModel: "importing"
+          });
+          session.message.on((ev) => this.message.emit(ev));
+          session.block.on((ev) => this.block.emit(ev));
+          session.update.on((ev) => this.update.emit(ev));
+          session.streamEvent.on((ev) => this.streamEvent.emit(ev));
+          session.userMessage.on((ev) => this.userMessage.emit(ev));
+          const messages = await session.run({
+            client: options.client,
+            artifacts: options.artifacts,
+            tools: options.tools,
+            history: options.history,
+            extensions: options.extensions,
+            generationOptions: options.generationOptions,
+            requiredArtifactIds: step.requiredArtifactIds.slice(),
+            artifactDiffResolver: options.artifactDiffResolver,
+            prompt: `
+              You are an agent that executes the subtask in a plan.
+              
+              While you know the plan's goal only focus on your subtask
+              Use the data from the previous step to complete your task.
+              Return all of the data from "output" in your last message.
+              
+              Current step (that you need to complete):
+              ${JSON.stringify({
+              goal,
+              input: step.input,
+              output: step.output,
+              yourTask: step.action
+            }, null, 2)}
+              
+              Previous step results:
+              ${JSON.stringify(stepResults.at(-1))}.
+            `
+          });
+          const result = messages.at(-1);
+          stepResults.push(result);
+        }
+        return import_artifact3.ToolResult.Success({
+          stepResults
+        });
+      }
+    });
+  }
+  abort() {
+    this._stream?.abort();
+  }
+};
+_ts_decorate([
+  import_async2.synchronized
+], AISession.prototype, "run", null);
+var createBaseInstructions = ({ availableArtifacts, operationModel }) => `
+  You are a friendly, advanced AI assistant capable of creating and managing artifacts from available data and tools. 
+  Your task is to process user commands and questions and decide how best to respond.
+  In some cases, you will need to create or reference data objects called artifacts.
+
+  Follow these guidelines carefully:
+
+  Decision-making:
+
+  - Analyze the structure and type of the content in the user's message.
+  
+  
+  ${operationModel === "planning" ? `
+      - Can you complete the task using the available artifacts?
+      - If you can't complete the task using the available artifacts, query the list of available artifacts using the appropriate tool.
+      - Identify which artifacts are relevant to the user's request.
+        - Break down the user's request into a step-by-step plan that you will use to execute the user's request, the plan items should include artifacts which will be used to perform the actions.
+      ` : ""}
+  ${operationModel === "importing" ? `
+      - Can you complete the task using the available artifacts?
+      - If you can't complete the task using the available artifacts, query the list of available artifacts using the appropriate tool.
+      - Identify which artifacts are relevant to the user's request.
+    - Are the required artifacts already available?
+    - If not, select which artifact(s) will be the most relevant and require them using the require_artifacts tool.
+    - The require'd artifact tools will be available for use after require.
+  ` : ""}
+  ${operationModel === "configured" ? `
+    - Select the most relevant artifact(s) to complete the task.
+    - Call the appropriate tool to use the artifact(s).
+  ` : ""}
+
+  ${availableArtifacts.length > 0 ? `Artifacts already in context: ${availableArtifacts.join("\n")}` : ""}
+`;
+var AIServiecOverloadedError = class extends import_effect3.Schema.TaggedError()("AIServiecOverloadedError", {}) {
+};
+var gatherObjectVersions = (messages) => {
+  const artifactIds = /* @__PURE__ */ new Map();
+  for (const message of messages) {
+    for (const block of message.content) {
+      if (block.type === "json" && block.disposition === VersionPin.DISPOSITION) {
+        const pin = VersionPin.pipe(import_effect3.Schema.decodeOption)(JSON.parse(block.json)).pipe(import_effect3.Option.getOrUndefined);
+        if (!pin) {
+          continue;
+        }
+        artifactIds.set(pin.objectId, pin.version);
+      }
+    }
+  }
+  return artifactIds;
+};
+var createArtifactUpdateBlock = (artifactDiff) => {
+  return {
+    type: "text",
+    disposition: "artifact-update",
+    text: `
+      The following artifacts have been updated since the last message:
+      ${Array.from(artifactDiff.entries()).map(([id, { diff }]) => `<changed-artifact id="${id}">${diff ? `
+${diff}` : ""}</changed-artifact>`).join("\n")}
+    `
+  };
+};
+var __dxlog_file7 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/experimental/conversation/conversation.ts";
+var runLLM = async (params) => {
+  let conversationResult = null;
+  const history = params.history ?? [];
+  const generate = async () => {
+    (0, import_log7.log)("llm generate", {
+      tools: params.tools
+    }, {
+      F: __dxlog_file7,
+      L: 44,
+      S: void 0,
+      C: (f, a) => f(...a)
+    });
+    const beginTs = Date.now();
+    const stream = await params.client.execStream({
+      model: params.model,
+      history,
+      systemPrompt: params.system,
+      tools: params.tools
+    });
+    for await (const event of stream) {
+      params.logger?.(event);
+    }
+    await stream.complete();
+    const messages = [];
+    const message = messages.at(-1);
+    (0, import_invariant6.invariant)(message, void 0, {
+      F: __dxlog_file7,
+      L: 61,
+      S: void 0,
+      A: [
+        "message",
+        ""
+      ]
+    });
+    (0, import_log7.log)("llm result", {
+      time: Date.now() - beginTs,
+      message
+    }, {
+      F: __dxlog_file7,
+      L: 63,
+      S: void 0,
+      C: (f, a) => f(...a)
+    });
+    (0, import_invariant6.invariant)(message, void 0, {
+      F: __dxlog_file7,
+      L: 64,
+      S: void 0,
+      A: [
+        "message",
+        ""
+      ]
+    });
+    params.logger?.({
+      type: "message",
+      message
+    });
+    history.push(...messages.map((message2) => ({
+      ...message2,
+      content: message2.content.filter((block) => block.type !== "image")
+    })));
+    const isToolUse2 = message.content.at(-1)?.type === "tool_use";
+    if (isToolUse2) {
+      const toolCalls = message.content.filter((block) => block.type === "tool_use");
+      (0, import_invariant6.invariant)(toolCalls.length === 1, void 0, {
+        F: __dxlog_file7,
+        L: 75,
+        S: void 0,
+        A: [
+          "toolCalls.length === 1",
+          ""
+        ]
+      });
+      const toolCall = toolCalls[0];
+      const tool = params.tools.find((tool2) => tool2.name === toolCall.name);
+      if (!tool) {
+        throw new Error(`Tool not found: ${toolCall.name}`);
+      }
+      (0, import_invariant6.invariant)(tool.execute, void 0, {
+        F: __dxlog_file7,
+        L: 82,
+        S: void 0,
+        A: [
+          "tool.execute",
+          ""
+        ]
+      });
+      const toolResult = await tool.execute(toolCall.input);
+      switch (toolResult.kind) {
+        case "error": {
+          import_log7.log.warn("tool error", {
+            message: toolResult.message
+          }, {
+            F: __dxlog_file7,
+            L: 86,
+            S: void 0,
+            C: (f, a) => f(...a)
+          });
+          const resultMessage = {
+            id: import_echo_schema6.ObjectId.random(),
+            role: "user",
+            content: [
+              {
+                type: "tool_result",
+                toolUseId: toolCall.id,
+                content: toolResult.message,
+                isError: true
+              }
+            ]
+          };
+          params.logger?.({
+            type: "message",
+            message: resultMessage
+          });
+          history.push(resultMessage);
+          return true;
+        }
+        case "success": {
+          (0, import_log7.log)("tool success", {
+            result: toolResult.result
+          }, {
+            F: __dxlog_file7,
+            L: 106,
+            S: void 0,
+            C: (f, a) => f(...a)
+          });
+          const resultMessage = {
+            id: import_echo_schema6.ObjectId.random(),
+            role: "user",
+            content: [
+              {
+                type: "tool_result",
+                toolUseId: toolCall.id,
+                content: JSON.stringify(toolResult.result)
+              }
+            ]
+          };
+          params.logger?.({
+            type: "message",
+            message: resultMessage
+          });
+          history.push(resultMessage);
+          return true;
+        }
+        case "break": {
+          (0, import_log7.log)("tool break", {
+            result: toolResult.result
+          }, {
+            F: __dxlog_file7,
+            L: 125,
+            S: void 0,
+            C: (f, a) => f(...a)
+          });
+          conversationResult = toolResult.result;
+          return false;
+        }
+      }
+    }
+    return false;
+  };
+  while (await generate()) {
+  }
+  return {
+    result: conversationResult,
+    history
+  };
+};
 var ws = import_parjs.parjs.whitespace();
 var nothing = import_parjs.parjs.string("").pipe(import_parjs.parjsCombinators.map(() => null));
 var optional = (parser) => parser.pipe(import_parjs.parjsCombinators.recover(() => ({
@@ -1858,7 +2312,7 @@ var parseCypherQuery = (query) => {
     throw new Error(`Failed to parse query: ${parseResult}`);
   }
 };
-var __dxlog_file6 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/cypher/execution-plan.ts";
+var __dxlog_file8 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/experimental/cypher/execution-plan.ts";
 var createExecutionPlan = (query) => {
   return new ExecutionPlanBuilder().build(query);
 };
@@ -1870,10 +2324,10 @@ var ExecutionPlanBuilder = class {
     return `anon_${this.counter++}`;
   }
   build(query) {
-    (0, import_log7.log)("Creating execution plan for query", {
+    (0, import_log9.log)("Creating execution plan for query", {
       query
     }, {
-      F: __dxlog_file6,
+      F: __dxlog_file8,
       L: 76,
       S: this,
       C: (f, a) => f(...a)
@@ -1890,8 +2344,8 @@ var ExecutionPlanBuilder = class {
     };
   }
   createStepsForMatchClause(match) {
-    (0, import_invariant5.invariant)(match.pattern.segments.length > 0, void 0, {
-      F: __dxlog_file6,
+    (0, import_invariant7.invariant)(match.pattern.segments.length > 0, void 0, {
+      F: __dxlog_file8,
       L: 90,
       S: this,
       A: [
@@ -1900,8 +2354,8 @@ var ExecutionPlanBuilder = class {
       ]
     });
     const firstSegment = match.pattern.segments[0];
-    (0, import_invariant5.invariant)(firstSegment.astKind === "NodePattern", void 0, {
-      F: __dxlog_file6,
+    (0, import_invariant7.invariant)(firstSegment.astKind === "NodePattern", void 0, {
+      F: __dxlog_file8,
       L: 92,
       S: this,
       A: [
@@ -1928,8 +2382,8 @@ var ExecutionPlanBuilder = class {
     }
     for (let i = 1; i < match.pattern.segments.length; i++) {
       const relationshipSegment = match.pattern.segments[i++];
-      (0, import_invariant5.invariant)(relationshipSegment.astKind === "RelationshipPattern", void 0, {
-        F: __dxlog_file6,
+      (0, import_invariant7.invariant)(relationshipSegment.astKind === "RelationshipPattern", void 0, {
+        F: __dxlog_file8,
         L: 115,
         S: this,
         A: [
@@ -1938,8 +2392,8 @@ var ExecutionPlanBuilder = class {
         ]
       });
       const nodeSegment = match.pattern.segments[i];
-      (0, import_invariant5.invariant)(nodeSegment.astKind === "NodePattern", void 0, {
-        F: __dxlog_file6,
+      (0, import_invariant7.invariant)(nodeSegment.astKind === "NodePattern", void 0, {
+        F: __dxlog_file8,
         L: 117,
         S: this,
         A: [
@@ -2020,15 +2474,15 @@ var getDirection = (left, right) => {
   }
   throw new Error(`Invalid direction: ()${left}[]${right}()`);
 };
-var __dxlog_file7 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/cypher/query-executor.ts";
+var __dxlog_file9 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/experimental/cypher/query-executor.ts";
 var executeQueryPlan = async (dataSource, plan) => {
   let buffer = [];
   const results = [];
   for (const step of plan.steps) {
-    (0, import_log6.log)("begin step", {
+    (0, import_log8.log)("begin step", {
       step
     }, {
-      F: __dxlog_file7,
+      F: __dxlog_file9,
       L: 41,
       S: void 0,
       C: (f, a) => f(...a)
@@ -2114,11 +2568,11 @@ var executeQueryPlan = async (dataSource, plan) => {
         break;
       }
     }
-    (0, import_log6.log)("finish step", {
+    (0, import_log8.log)("finish step", {
       buffer: buffer.map((tuple) => (0, import_util2.mapValues)(tuple, (x) => x.id)),
       results
     }, {
-      F: __dxlog_file7,
+      F: __dxlog_file9,
       L: 118,
       S: void 0,
       C: (f, a) => f(...a)
@@ -2201,14 +2655,14 @@ var formatInferredRelationshipLabel = (typename, property2) => {
   const nodeLabel = formatNodeLabel(typename);
   return `${nodeLabel}_${property2}`.toUpperCase();
 };
-var isReference = (schema) => schema.$id === import_echo_schema4.JSON_SCHEMA_ECHO_REF_ID || schema.type === "array" && !Array.isArray(schema.items) && typeof schema.items === "object" && "$id" in schema.items && schema.items.$id === import_echo_schema4.JSON_SCHEMA_ECHO_REF_ID;
-function _ts_decorate(decorators, target, key, desc) {
+var isReference = (schema) => schema.$id === import_echo_schema7.JSON_SCHEMA_ECHO_REF_ID || schema.type === "array" && !Array.isArray(schema.items) && typeof schema.items === "object" && "$id" in schema.items && schema.items.$id === import_echo_schema7.JSON_SCHEMA_ECHO_REF_ID;
+function _ts_decorate2(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
   else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
   return c > 3 && r && Object.defineProperty(target, key, r), r;
 }
-var __dxlog_file8 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/cypher/echo-data-source.ts";
+var __dxlog_file10 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/experimental/cypher/echo-data-source.ts";
 var EchoDataSource = class {
   constructor(_db) {
     this._db = _db;
@@ -2218,16 +2672,16 @@ var EchoDataSource = class {
       const { objects: objects2 } = await this._db.query().run();
       return objects2.map(this._objectToNode);
     }
-    const schema = (await this._getAllSchema()).find((s) => (0, import_echo_schema5.getSchemaTypename)(s)?.endsWith(label2));
+    const schema = (await this._getAllSchema()).find((schema2) => (0, import_echo_schema8.getSchemaTypename)(schema2)?.endsWith(label2));
     if (!schema) {
       return [];
     }
-    const { objects } = await this._db.query(import_echo_db.Filter.schema(schema)).run();
+    const { objects } = await this._db.query(import_echo_db3.Filter.type(schema)).run();
     return objects.map(this._objectToNode);
   }
   async getRelationships({ label: label2 }) {
-    const syntheticRefRelationships = (await this._getAllSchema()).filter((schema) => (0, import_echo_schema5.getSchemaTypename)(schema) !== import_echo_schema5.StoredSchema.typename).flatMap((schema) => {
-      const jsonSchema = (0, import_echo_schema5.toJsonSchema)(schema);
+    const syntheticRefRelationships = (await this._getAllSchema()).filter((schema) => (0, import_echo_schema8.getSchemaTypename)(schema) !== import_echo_schema8.StoredSchema.typename).flatMap((schema) => {
+      const jsonSchema = (0, import_echo_schema8.toJsonSchema)(schema);
       const relationships = Object.entries(jsonSchema.properties).filter(([key, prop]) => isReference(prop));
       return relationships.map(([key, value]) => {
         const relationshipLabel = formatInferredRelationshipLabel(jsonSchema.typename ?? (0, import_debug2.raise)(new Error("Schema must have typename")), key);
@@ -2242,22 +2696,22 @@ var EchoDataSource = class {
     if (!relationship) {
       return [];
     }
-    const { objects } = await this._db.query(import_echo_db.Filter.schema(relationship.schema)).run();
+    const { objects } = await this._db.query(import_echo_db3.Filter.type(relationship.schema)).run();
     return (await Promise.all(objects.map((object) => this._projectRefRelationship(object, relationship.property)))).flat();
   }
   async _getAllSchema() {
     return [
       ...await this._db.schemaRegistry.query().run(),
       // TODO(dmaretskyi): Remove once we can serialize recursive schema.
-      ...this._db.graph.schemaRegistry.schemas.filter((s) => (0, import_echo_schema5.getSchemaTypename)(s)?.startsWith("example.org"))
-    ].filter((schema) => (0, import_echo_schema5.getSchemaTypename)(schema) !== import_echo_schema5.StoredSchema.typename);
+      ...this._db.graph.schemaRegistry.schemas.filter((schema) => (0, import_echo_schema8.getSchemaTypename)(schema)?.startsWith("example.org"))
+    ].filter((schema) => (0, import_echo_schema8.getSchemaTypename)(schema) !== import_echo_schema8.StoredSchema.typename);
   }
   _objectToNode(object) {
     const { id, ...properties2 } = object;
     return {
       id,
       kind: "node",
-      label: formatNodeLabel((0, import_echo_schema5.getSchemaTypename)((0, import_echo_schema5.getSchema)(object))),
+      label: formatNodeLabel((0, import_echo_schema8.getSchemaTypename)((0, import_echo_schema8.getSchema)(object))),
       properties: properties2
     };
   }
@@ -2273,7 +2727,7 @@ var EchoDataSource = class {
       return {
         id: `${object.id}-${prop}-${target.id}`,
         kind: "relationship",
-        label: formatInferredRelationshipLabel((0, import_echo_schema5.getSchemaTypename)((0, import_echo_schema5.getSchema)(object)), prop),
+        label: formatInferredRelationshipLabel((0, import_echo_schema8.getSchemaTypename)((0, import_echo_schema8.getSchema)(object)), prop),
         properties: {},
         source: this._objectToNode(object),
         target: this._objectToNode(target)
@@ -2281,18 +2735,18 @@ var EchoDataSource = class {
     }));
   }
 };
-_ts_decorate([
-  import_log8.log.method(void 0, void 0, {
-    F: __dxlog_file8,
-    L: 16,
+_ts_decorate2([
+  import_log10.log.method(void 0, void 0, {
+    F: __dxlog_file10,
+    L: 18,
     S: void 0,
     C: (f, a) => f(...a)
   })
 ], EchoDataSource.prototype, "getNodes", null);
-_ts_decorate([
-  import_log8.log.method(void 0, void 0, {
-    F: __dxlog_file8,
-    L: 33,
+_ts_decorate2([
+  import_log10.log.method(void 0, void 0, {
+    F: __dxlog_file10,
+    L: 34,
     S: void 0,
     C: (f, a) => f(...a)
   })
@@ -2300,6 +2754,8 @@ _ts_decorate([
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   AIServiceEdgeClient,
+  AIServiecOverloadedError,
+  AISession,
   DEFAULT_EDGE_MODEL,
   DEFAULT_EDGE_MODELS,
   DEFAULT_OLLAMA_ENDPOINT,
@@ -2315,6 +2771,7 @@ _ts_decorate([
   OllamaClient,
   StreamTransform,
   ToolTypes,
+  VersionPin,
   createArtifactElement,
   createGenerationStream,
   emitMessageAsEvents,
@@ -2331,4 +2788,4 @@ _ts_decorate([
   runTools,
   trim
 });
-//# sourceMappingURL=chunk-7VH7NMIM.cjs.map
+//# sourceMappingURL=chunk-NOB7CIIC.cjs.map