@dxos/assistant 0.8.0 → 0.8.1-main.303c73a

package/dist/lib/node/index.cjs

@@ -28,18 +28,21 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var node_exports = {};
 __export(node_exports, {
-  AIServiceClientImpl: () => AIServiceClientImpl,
+  AIServiceEdgeClient: () => AIServiceEdgeClient,
   DEFAULT_LLM_MODEL: () => DEFAULT_LLM_MODEL,
   DEFAULT_LLM_MODELS: () => DEFAULT_LLM_MODELS,
   EchoDataSource: () => import_chunk_QPHGQINW.EchoDataSource,
   GenerateRequest: () => GenerateRequest,
-  GenerationStream: () => GenerationStream,
+  GenerationStreamImpl: () => GenerationStreamImpl,
   LLMModel: () => LLMModel,
+  MessageCollector: () => MessageCollector,
   MixedStreamParser: () => MixedStreamParser,
+  OllamaClient: () => OllamaClient,
   StreamTransform: () => StreamTransform,
   ToolTypes: () => ToolTypes,
   createArtifactElement: () => createArtifactElement,
   createGenerationStream: () => createGenerationStream,
+  emitMessageAsEvents: () => emitMessageAsEvents,
   executeQuery: () => import_chunk_QPHGQINW.executeQuery,
   executeQueryPlan: () => import_chunk_QPHGQINW.executeQueryPlan,
   formatInferredRelationshipLabel: () => import_chunk_QPHGQINW.formatInferredRelationshipLabel,
@@ -54,26 +57,45 @@ __export(node_exports, {
 });
 module.exports = __toCommonJS(node_exports);
 var import_chunk_QPHGQINW = require("./chunk-QPHGQINW.cjs");
-var import_schema = require("@effect/schema");
+var import_effect = require("effect");
 var import_artifact = require("@dxos/artifact");
 var import_invariant = require("@dxos/invariant");
 var import_log = require("@dxos/log");
 var import_async = require("@dxos/async");
 var import_log2 = require("@dxos/log");
 var import_artifact2 = require("@dxos/artifact");
-var import_async2 = require("@dxos/async");
 var import_echo_schema = require("@dxos/echo-schema");
 var import_invariant2 = require("@dxos/invariant");
 var import_log3 = require("@dxos/log");
-var import_util = require("@dxos/util");
-var P = __toESM(require("parsimmon"));
-var import_schema2 = require("@effect/schema");
+var import_effect2 = require("effect");
 var import_artifact3 = require("@dxos/artifact");
 var import_echo_schema2 = require("@dxos/echo-schema");
 var import_artifact4 = require("@dxos/artifact");
 var import_echo_schema3 = require("@dxos/echo-schema");
 var import_invariant3 = require("@dxos/invariant");
 var import_log4 = require("@dxos/log");
+var import_artifact5 = require("@dxos/artifact");
+var import_async2 = require("@dxos/async");
+var import_echo_schema4 = require("@dxos/echo-schema");
+var import_invariant4 = require("@dxos/invariant");
+var import_log5 = require("@dxos/log");
+var import_util = require("@dxos/util");
+var P = __toESM(require("parsimmon"));
+var DEFAULT_LLM_MODEL = "@anthropic/claude-3-5-sonnet-20241022";
+var DEFAULT_LLM_MODELS = [
+  // AI Gateway.
+  // https://developers.cloudflare.com/ai-gateway
+  "@anthropic/claude-3-5-haiku-20241022",
+  "@anthropic/claude-3-5-sonnet-20241022",
+  "@anthropic/claude-3-7-sonnet-20250219",
+  // Workers AI.
+  // https://developers.cloudflare.com/workers-ai/models
+  "@cf/deepseek-ai/deepseek-r1-distill-qwen-32b",
+  "@hf/nousresearch/hermes-2-pro-mistral-7b",
+  "@ollama/llama-3-1-nemotron-70b-instruct",
+  "@ollama/llama-3-1-nemotron-mini-4b-instruct",
+  "@ollama/llama-3-2-3b"
+];
 async function* iterSSEMessages(response, controller) {
   if (!response.body) {
     controller.abort();
@@ -321,7 +343,7 @@ var createGenerationStream = (response, controller = new AbortController()) => {
             raw: sse.raw
           }, {
             F: __dxlog_file,
-            L: 21,
+            L: 22,
             S: this,
             C: (f, a) => f(...a)
           });
@@ -337,7 +359,7 @@ var createGenerationStream = (response, controller = new AbortController()) => {
             raw: sse.raw
           }, {
             F: __dxlog_file,
-            L: 37,
+            L: 38,
             S: this,
             C: (f, a) => f(...a)
           });
@@ -352,9 +374,9 @@ var createGenerationStream = (response, controller = new AbortController()) => {
       }
     }
   };
-  return new GenerationStream(controller, iterator);
+  return new GenerationStreamImpl(controller, iterator);
 };
-var GenerationStream = class {
+var GenerationStreamImpl = class {
   constructor(_controller, _getIterator) {
     this._controller = _controller;
     this._getIterator = _getIterator;
@@ -393,18 +415,19 @@ var GenerationStream = class {
           this._done.wake();
         } catch (err) {
           this._done.throw(err);
+          throw err;
         }
       };
       return generator.call(this);
     })();
   }
 };
-var __dxlog_file2 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/ai-service/client.ts";
-var AIServiceClientImpl = class {
+var __dxlog_file2 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/ai-service/edge-client.ts";
+var AIServiceEdgeClient = class {
   constructor({ endpoint }) {
     (0, import_invariant.invariant)(endpoint, "endpoint is required", {
       F: __dxlog_file2,
-      L: 34,
+      L: 23,
       S: this,
       A: [
         "endpoint",
@@ -413,6 +436,10 @@ var AIServiceClientImpl = class {
     });
     this._endpoint = endpoint;
   }
+  /**
+  * @deprecated
+  */
+  // TODO(burdon): Remove.
   async appendMessages(messages) {
     const url = `${this._endpoint}/message`;
     const res = await fetch(url, {
@@ -421,7 +448,7 @@ var AIServiceClientImpl = class {
         "Content-Type": "application/json"
       },
       body: JSON.stringify({
-        messages: await import_schema.Schema.encodePromise(import_schema.Schema.Array(import_artifact.Message))(messages)
+        messages: await import_effect.Schema.encodePromise(import_effect.Schema.Array(import_artifact.Message))(messages)
       })
     });
     if (!res.ok) {
@@ -432,15 +459,11 @@ var AIServiceClientImpl = class {
   * Process request and open message stream.
   */
   async exec(request) {
-    request = {
-      tools: request.tools ?? [],
-      ...request
-    };
     import_log.log.info("requesting", {
       endpoint: this._endpoint
     }, {
       F: __dxlog_file2,
-      L: 63,
+      L: 50,
       S: this,
       C: (f, a) => f(...a)
     });
@@ -451,11 +474,15 @@ var AIServiceClientImpl = class {
       headers: {
         "Content-Type": "application/json"
       },
-      body: JSON.stringify(request)
+      body: JSON.stringify({
+        // TODO(dmaretskyi): Errors if tools are not provided.
+        tools: request.tools ?? [],
+        ...request
+      })
     });
     (0, import_invariant.invariant)(response.body instanceof ReadableStream, void 0, {
       F: __dxlog_file2,
-      L: 72,
+      L: 63,
       S: this,
       A: [
         "response.body instanceof ReadableStream",
@@ -465,807 +492,1217 @@ var AIServiceClientImpl = class {
     return createGenerationStream(response, controller);
   }
 };
-var DEFAULT_LLM_MODEL = "@anthropic/claude-3-5-sonnet-20241022";
-var DEFAULT_LLM_MODELS = [
-  // AI Gateway.
-  // https://developers.cloudflare.com/ai-gateway
-  "@anthropic/claude-3-5-haiku-20241022",
-  "@anthropic/claude-3-5-sonnet-20241022",
-  "@anthropic/claude-3-7-sonnet-20250219",
-  // Workers AI.
-  // https://developers.cloudflare.com/workers-ai/models
-  "@cf/deepseek-ai/deepseek-r1-distill-qwen-32b",
-  "@hf/nousresearch/hermes-2-pro-mistral-7b",
-  "@ollama/llama-3-1-nemotron-70b-instruct",
-  "@ollama/llama-3-1-nemotron-mini-4b-instruct",
-  "@ollama/llama-3-2-3b"
-];
-var textChunk = P.regexp(/[^<]+/).map((content) => ({
-  type: "text",
-  content
-}));
-var attribute = P.seq(P.regexp(/[a-zA-Z_][a-zA-Z0-9_-]*/), P.string("=").trim(P.optWhitespace), P.alt(P.regexp(/"[^"]*"/).map((s) => s.slice(1, -1)), P.regexp(/'[^']*'/).map((s) => s.slice(1, -1)))).map(([name, _, value]) => [
-  name,
-  value
-]);
-var tagName = P.regexp(/[a-zA-Z_][a-zA-Z0-9_-]*/);
-var selfClosingTag = P.seqMap(
-  //
-  P.string("<"),
-  tagName,
-  P.optWhitespace.then(attribute).many(),
-  P.optWhitespace.then(P.string("/>")),
-  P.optWhitespace,
-  (_, tag, attributes, __) => ({
-    type: "tag",
-    tag,
-    attributes: Object.fromEntries(attributes),
-    content: [],
-    selfClosing: true
-  })
-);
-var openTag = P.seqMap(
-  //
-  P.string("<"),
-  tagName,
-  P.optWhitespace.then(attribute).many(),
-  P.string(">"),
-  P.regexp(/[^<]*/),
-  (_, tag, attributes, __, content) => ({
-    type: "tag",
-    tag,
-    attributes: Object.fromEntries(attributes),
-    content: content.length ? [
-      {
-        type: "text",
-        content
-      }
-    ] : []
-  })
-);
-var closeTag = P.seqMap(
-  //
-  P.string("</"),
-  tagName,
-  P.string(">"),
-  P.optWhitespace,
-  (_, tag) => ({
-    type: "tag",
-    tag,
-    content: [],
-    closing: true
-  })
-);
-var mixedChunk = P.alt(selfClosingTag, openTag, closeTag, textChunk);
-var StreamTransform = class {
+var MessageCollector = class {
   constructor() {
-    this._buffer = "";
+    this._messages = [];
+    this._pendingMessage = void 0;
   }
-  transform(chunk) {
-    this._buffer += chunk;
-    const results = [];
-    const parser = mixedChunk.many().skip(P.optWhitespace);
-    const result = parser.parse(this._buffer);
-    if (result.status) {
-      for (const chunk2 of result.value) {
-        results.push(chunk2);
-      }
-      this._buffer = "";
+  get messages() {
+    return this._messages;
+  }
+  get pendingMessage() {
+    return this._pendingMessage;
+  }
+  push(event) {
+    switch (event.type) {
+      case "message_start":
+        this._pendingMessage = event.message;
+        break;
+      case "message_stop":
+        if (this._pendingMessage) {
+          this._messages.push(this._pendingMessage);
+          this._pendingMessage = void 0;
+        }
+        break;
+      case "content_block_start":
+        if (this._pendingMessage) {
+          this._pendingMessage.content = this._pendingMessage.content || [];
+          this._pendingMessage.content[event.index] = event.content;
+        }
+        break;
+      case "content_block_delta":
+        if (this._pendingMessage && this._pendingMessage.content) {
+          const block = this._pendingMessage.content[event.index];
+          if (block && event.delta.type === "text_delta" && block.type === "text") {
+            block.text = (block.text || "") + event.delta.text;
+          }
+        }
+        break;
+      case "content_block_stop":
+        if (this._pendingMessage && this._pendingMessage.content) {
+          const block = this._pendingMessage.content[event.index];
+          if (block) {
+            block.pending = false;
+          }
+        }
+        break;
+      case "message_delta":
+        if (this._pendingMessage && event.delta) {
+          Object.assign(this._pendingMessage, event.delta);
+        }
+        break;
     }
-    return results;
   }
-  // TODO(burdon): Process remaining buffer?
-  flush() {
-    const remaining = this._buffer;
-    this._buffer = "";
-    return remaining.length > 0 ? [
-      {
-        type: "text",
-        content: remaining
-      }
-    ] : [];
+  pushMessage(message) {
+    this._messages.push(message);
   }
 };
-var __dxlog_file3 = "/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();
-  }
-  _emitBlock(contentBlock, streamBlock) {
-    const messageBlock = streamBlock ? mergeMessageBlock(contentBlock, streamBlock) : contentBlock;
-    if (messageBlock) {
-      if (messageBlock.type === "text" && messageBlock.text.length === 0) {
-        return;
+function* emitMessageAsEvents(message) {
+  yield {
+    type: "message_start",
+    message
+  };
+  yield {
+    type: "message_stop"
+  };
+}
+var createArtifactElement = (id) => `<artifact id=${id} />`;
+var LLMModel = import_effect2.Schema.Literal(...DEFAULT_LLM_MODELS);
+var ToolTypes = Object.freeze({
+  // TODO(dmaretskyi): Not implemented yet.
+  // DatabaseQuery: 'database_query',
+  TextToImage: "text_to_image"
+});
+var GenerateRequest = import_effect2.Schema.Struct({
+  spaceId: import_effect2.Schema.optional(import_artifact3.SpaceIdSchema),
+  threadId: import_effect2.Schema.optional(import_echo_schema2.ObjectId),
+  /**
+  * Preferred model or system default.
+  */
+  model: import_effect2.Schema.optional(LLMModel),
+  /**
+  * Tools available for the LLM.
+  */
+  tools: import_effect2.Schema.optional(import_effect2.Schema.Array(import_artifact3.Tool).pipe(import_effect2.Schema.mutable)),
+  /**
+  * System instructions to the LLM.
+  */
+  systemPrompt: import_effect2.Schema.optional(import_effect2.Schema.String),
+  /**
+  * History of messages to include in the context window.
+  */
+  // TODO(burdon): Rename messages.
+  history: import_effect2.Schema.optional(import_effect2.Schema.Array(import_artifact3.Message)),
+  /**
+  * Current request.
+  */
+  prompt: import_effect2.Schema.optional(import_artifact3.Message)
+});
+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_log4.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.exec({
+      model: params.model,
+      spaceId: params.spaceId,
+      threadId: params.threadId,
+      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_invariant3.invariant)(message, void 0, {
+      F: __dxlog_file3,
+      L: 63,
+      S: void 0,
+      A: [
+        "message",
+        ""
+      ]
+    });
+    (0, import_log4.log)("llm result", {
+      time: Date.now() - beginTs,
+      message
+    }, {
+      F: __dxlog_file3,
+      L: 65,
+      S: void 0,
+      C: (f, a) => f(...a)
+    });
+    (0, import_invariant3.invariant)(message, void 0, {
+      F: __dxlog_file3,
+      L: 66,
+      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_invariant3.invariant)(toolCalls.length === 1, void 0, {
+        F: __dxlog_file3,
+        L: 77,
+        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)(this._message, void 0, {
+      (0, import_invariant3.invariant)(tool.execute, void 0, {
         F: __dxlog_file3,
-        L: 46,
-        S: this,
+        L: 84,
+        S: void 0,
         A: [
-          "this._message",
+          "tool.execute",
           ""
         ]
       });
-      this._message.content.push(messageBlock);
-      this.block.emit(messageBlock);
-    }
-  }
-  _emitUpdate(contentBlock, streamBlock) {
-    const messageBlock = mergeMessageBlock(contentBlock, streamBlock);
-    if (messageBlock) {
-      messageBlock.pending = true;
-      this.update.emit(messageBlock);
-    }
-  }
-  /**
-  * Parse stream until end.
-  */
-  async parse(stream) {
-    const transformer = new StreamTransform();
-    let contentBlock;
-    let streamBlock;
-    const stack = [];
-    const messagesCollected = [];
-    for await (const event of stream) {
-      switch (event.type) {
-        //
-        // Messages.
-        //
-        case "message_start": {
-          if (this._message) {
-            import_log3.log.warn("unexpected message_start", void 0, {
-              F: __dxlog_file3,
-              L: 88,
-              S: this,
-              C: (f, a) => f(...a)
-            });
-          }
-          this._message = (0, import_echo_schema.createStatic)(import_artifact2.Message, {
-            role: "assistant",
-            content: []
+      const toolResult = await tool.execute(toolCall.input);
+      switch (toolResult.kind) {
+        case "error": {
+          import_log4.log.warn("tool error", {
+            message: toolResult.message
+          }, {
+            F: __dxlog_file3,
+            L: 88,
+            S: void 0,
+            C: (f, a) => f(...a)
           });
-          this.message.emit(this._message);
-          break;
-        }
-        case "message_delta": {
-          if (!this._message) {
-            import_log3.log.warn("unexpected message_delta", void 0, {
-              F: __dxlog_file3,
-              L: 98,
-              S: this,
-              C: (f, a) => f(...a)
-            });
-          }
-          break;
-        }
-        case "message_stop": {
-          if (!this._message) {
-            import_log3.log.warn("unexpected message_stop", void 0, {
-              F: __dxlog_file3,
-              L: 106,
-              S: this,
-              C: (f, a) => f(...a)
-            });
-          } else {
-            messagesCollected.push(this._message);
-            this._message = void 0;
-          }
-          break;
-        }
-        //
-        // Content blocks.
-        //
-        case "content_block_start": {
-          if (contentBlock) {
-            import_log3.log.warn("unexpected content_block_start", {
-              content: contentBlock
-            }, {
-              F: __dxlog_file3,
-              L: 120,
-              S: this,
-              C: (f, a) => f(...a)
-            });
-          }
-          contentBlock = event.content;
-          break;
-        }
-        case "content_block_delta": {
-          if (!contentBlock) {
-            import_log3.log.warn("unexpected content_block_delta", void 0, {
-              F: __dxlog_file3,
-              L: 130,
-              S: this,
-              C: (f, a) => f(...a)
-            });
-            break;
-          }
-          switch (event.delta.type) {
-            //
-            // Text
-            //
-            case "text_delta": {
-              const chunks = transformer.transform(event.delta.text);
-              for (const chunk of chunks) {
-                switch (streamBlock?.type) {
-                  //
-                  // XML Fragment.
-                  //
-                  case "tag": {
-                    if (chunk.type === "tag") {
-                      if (chunk.selfClosing) {
-                        streamBlock.content.push(chunk);
-                      } else if (chunk.closing) {
-                        if (stack.length > 0) {
-                          const top = stack.pop();
-                          (0, import_invariant2.invariant)(top && top.type === "tag", void 0, {
-                            F: __dxlog_file3,
-                            L: 154,
-                            S: this,
-                            A: [
-                              "top && top.type === 'tag'",
-                              ""
-                            ]
-                          });
-                          top.content.push(streamBlock);
-                          streamBlock = top;
-                        } else {
-                          this._emitBlock(contentBlock, streamBlock);
-                          streamBlock = void 0;
-                        }
-                      } else {
-                        stack.push(streamBlock);
-                        streamBlock = chunk;
-                      }
-                    } else {
-                      if (streamBlock.content.length === 0) {
-                        streamBlock.content.push(chunk);
-                      } else {
-                        const last = streamBlock.content.at(-1);
-                        (0, import_invariant2.invariant)(last, void 0, {
-                          F: __dxlog_file3,
-                          L: 171,
-                          S: this,
-                          A: [
-                            "last",
-                            ""
-                          ]
-                        });
-                        if (last.type === "text") {
-                          last.content += chunk.content;
-                        } else {
-                          streamBlock.content.push(chunk);
-                        }
-                      }
-                    }
-                    break;
-                  }
-                  //
-                  // Text Fragment.
-                  //
-                  case "text": {
-                    if (chunk.type === "tag") {
-                      this._emitBlock(contentBlock, streamBlock);
-                      if (chunk.selfClosing) {
-                        this._emitBlock(contentBlock, chunk);
-                        streamBlock = void 0;
-                      } else {
-                        streamBlock = chunk;
-                      }
-                    } else {
-                      streamBlock.content += chunk.content;
-                    }
-                    break;
-                  }
-                  //
-                  // No current chunk.
-                  //
-                  default: {
-                    if (chunk.type === "tag" && chunk.selfClosing) {
-                      this._emitBlock(contentBlock, chunk);
-                    } else {
-                      streamBlock = chunk;
-                    }
-                  }
-                }
-              }
-              break;
-            }
-            //
-            // JSON
-            //
-            case "input_json_delta": {
-              if (!contentBlock) {
-                import_log3.log.warn("unexpected input_json_delta", {
-                  contentBlock
-                }, {
-                  F: __dxlog_file3,
-                  L: 221,
-                  S: this,
-                  C: (f, a) => f(...a)
-                });
-                break;
+          const resultMessage = {
+            id: import_echo_schema3.ObjectId.random(),
+            spaceId: params.spaceId,
+            threadId: params.threadId,
+            role: "user",
+            content: [
+              {
+                type: "tool_result",
+                toolUseId: toolCall.id,
+                content: toolResult.message,
+                isError: true
               }
-              if (streamBlock?.type === "json") {
-                streamBlock.content += event.delta.partial_json;
-              } else {
-                streamBlock = {
-                  type: "json",
-                  content: event.delta.partial_json
-                };
+            ]
+          };
+          params.logger?.({
+            type: "message",
+            message: resultMessage
+          });
+          history.push(resultMessage);
+          return true;
+        }
+        case "success": {
+          (0, import_log4.log)("tool success", {
+            result: toolResult.result
+          }, {
+            F: __dxlog_file3,
+            L: 110,
+            S: void 0,
+            C: (f, a) => f(...a)
+          });
+          const resultMessage = {
+            id: import_echo_schema3.ObjectId.random(),
+            spaceId: params.spaceId,
+            threadId: params.threadId,
+            role: "user",
+            content: [
+              {
+                type: "tool_result",
+                toolUseId: toolCall.id,
+                content: JSON.stringify(toolResult.result)
               }
-              break;
-            }
-          }
-          break;
+            ]
+          };
+          params.logger?.({
+            type: "message",
+            message: resultMessage
+          });
+          history.push(resultMessage);
+          return true;
         }
-        case "content_block_stop": {
-          if (!contentBlock) {
-            import_log3.log.warn("unexpected content_block_stop", void 0, {
-              F: __dxlog_file3,
-              L: 238,
-              S: this,
-              C: (f, a) => f(...a)
-            });
-            break;
-          }
-          this._emitBlock(contentBlock, streamBlock);
-          contentBlock = void 0;
-          streamBlock = void 0;
-          break;
+        case "break": {
+          (0, import_log4.log)("tool break", {
+            result: toolResult.result
+          }, {
+            F: __dxlog_file3,
+            L: 131,
+            S: void 0,
+            C: (f, a) => f(...a)
+          });
+          conversationResult = toolResult.result;
+          return false;
         }
       }
-      if (contentBlock && streamBlock) {
-        this._emitUpdate(contentBlock, streamBlock);
-      }
     }
-    return messagesCollected;
+    return false;
+  };
+  while (await generate()) {
   }
+  return {
+    result: conversationResult,
+    history
+  };
 };
-var mergeMessageBlock = (contentBlock, streamBlock) => {
-  (0, import_log3.log)("mergeMessageBlock", {
-    contentBlock,
-    streamBlock
-  }, {
+var isToolUse = (message, { onlyToolNames } = {}) => {
+  const block = message.content.at(-1);
+  (0, import_invariant3.invariant)(block, void 0, {
     F: __dxlog_file3,
-    L: 265,
+    L: 152,
     S: void 0,
-    C: (f, a) => f(...a)
+    A: [
+      "block",
+      ""
+    ]
   });
-  switch (streamBlock.type) {
-    //
-    // Text
-    //
-    case "text": {
+  return block.type === "tool_use" && (!onlyToolNames || onlyToolNames.includes(block.name));
+};
+var runTools = async ({ message, tools, extensions }) => {
+  const toolCalls = message.content.filter((block) => block.type === "tool_use");
+  (0, import_invariant3.invariant)(toolCalls.length === 1, void 0, {
+    F: __dxlog_file3,
+    L: 174,
+    S: void 0,
+    A: [
+      "toolCalls.length === 1",
+      ""
+    ]
+  });
+  const toolCall = toolCalls[0];
+  const tool = tools.find((tool2) => tool2.name === toolCall.name);
+  if (!tool) {
+    throw new Error(`Tool not found: ${toolCall.name}`);
+  }
+  let toolResult;
+  try {
+    (0, import_invariant3.invariant)(tool.execute, void 0, {
+      F: __dxlog_file3,
+      L: 183,
+      S: void 0,
+      A: [
+        "tool.execute",
+        ""
+      ]
+    });
+    toolResult = await tool.execute(toolCall.input, {
+      extensions
+    });
+  } catch (error) {
+    (0, import_log4.log)("tool error", {
+      error
+    }, {
+      F: __dxlog_file3,
+      L: 186,
+      S: void 0,
+      C: (f, a) => f(...a)
+    });
+    toolResult = import_artifact4.ToolResult.Error(error.message);
+  }
+  switch (toolResult.kind) {
+    case "error": {
+      (0, import_log4.log)("tool error", {
+        message: toolResult.message
+      }, {
+        F: __dxlog_file3,
+        L: 192,
+        S: void 0,
+        C: (f, a) => f(...a)
+      });
+      const resultMessage = (0, import_echo_schema3.createStatic)(import_artifact4.Message, {
+        role: "user",
+        content: [
+          {
+            type: "tool_result",
+            toolUseId: toolCall.id,
+            content: toolResult.message,
+            isError: true
+          }
+        ]
+      });
       return {
-        ...contentBlock,
-        type: "text",
-        text: streamBlock.content
+        type: "continue",
+        message: resultMessage
       };
     }
-    //
-    // XML
-    //
-    case "tag": {
-      switch (streamBlock.tag) {
-        case "cot": {
-          const content = streamBlock.content.map((block) => {
-            switch (block.type) {
-              case "text":
-                return block.content;
-              default:
-                return null;
-            }
-          }).filter(import_util.isNotFalsy).join("\n");
-          return {
-            ...contentBlock,
-            type: "text",
-            disposition: "cot",
-            text: content
-          };
-        }
-        case "artifact": {
-          const { attributes } = streamBlock;
-          return {
-            type: "json",
-            disposition: "artifact",
-            json: JSON.stringify(attributes)
-          };
-        }
-        case "suggest": {
-          if (streamBlock.content.length === 1 && streamBlock.content[0].type === "text") {
-            return {
-              type: "json",
-              disposition: "suggest",
-              json: JSON.stringify({
-                text: streamBlock.content[0].content
-              })
-            };
-          }
-          break;
-        }
-        case "select": {
-          return {
-            type: "json",
-            disposition: "select",
-            json: JSON.stringify({
-              options: streamBlock.content.flatMap((content) => content.type === "tag" && content.content.length === 1 && content.content[0].type === "text" ? [
-                content.content[0].content
-              ] : [])
-            })
-          };
-        }
-        case "tool-list": {
-          return {
-            type: "json",
-            disposition: "tool_list",
-            json: JSON.stringify({})
-          };
-        }
-      }
-      break;
+    case "success": {
+      (0, import_log4.log)("tool success", {
+        result: toolResult.result
+      }, {
+        F: __dxlog_file3,
+        L: 213,
+        S: void 0,
+        C: (f, a) => f(...a)
+      });
+      const resultMessage = (0, import_echo_schema3.createStatic)(import_artifact4.Message, {
+        role: "user",
+        content: [
+          {
+            type: "tool_result",
+            toolUseId: toolCall.id,
+            content: typeof toolResult.result === "string" ? toolResult.result : JSON.stringify(toolResult.result) ?? ""
+          },
+          ...toolResult.extractContentBlocks ?? []
+        ]
+      });
+      return {
+        type: "continue",
+        message: resultMessage
+      };
     }
-    //
-    // JSON
-    //
-    case "json": {
-      switch (contentBlock.type) {
-        case "tool_use": {
-          return {
-            ...contentBlock,
-            input: (0, import_util.safeParseJson)(streamBlock.content) ?? {}
-          };
-        }
-      }
+    case "break": {
+      (0, import_log4.log)("tool break", {
+        result: toolResult.result
+      }, {
+        F: __dxlog_file3,
+        L: 234,
+        S: void 0,
+        C: (f, a) => f(...a)
+      });
       return {
-        ...contentBlock,
-        type: "json",
-        disposition: "artifact",
-        json: streamBlock.content
+        type: "break",
+        result: toolResult.result
       };
     }
   }
 };
-var createArtifactElement = (id) => `<artifact id=${id} />`;
-var LLMModel = import_schema2.Schema.Literal(...DEFAULT_LLM_MODELS);
-var ToolTypes = Object.freeze({
-  // TODO(dmaretskyi): Not implemented yet.
-  // DatabaseQuery: 'database_query',
-  TextToImage: "text_to_image"
-});
-var GenerateRequest = import_schema2.Schema.Struct({
-  spaceId: import_schema2.Schema.optional(import_artifact3.SpaceIdSchema),
-  threadId: import_schema2.Schema.optional(import_echo_schema2.ObjectId),
-  /**
-  * Preferred model or system default.
-  */
-  model: import_schema2.Schema.optional(LLMModel),
-  /**
-  * Tools available for the LLM.
-  */
-  tools: import_schema2.Schema.optional(import_schema2.Schema.Array(import_artifact3.Tool).pipe(import_schema2.Schema.mutable)),
+var __dxlog_file4 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/ai-service/ollama-client.ts";
+var DEFAULT_OLLAMA_URL = "http://localhost:11434";
+var DEFAULT_OLLAMA_MODEL = "llama3.2:1b";
+var OllamaClient = class _OllamaClient {
   /**
-  * System instructions to the LLM.
-  */
-  systemPrompt: import_schema2.Schema.optional(import_schema2.Schema.String),
-  /**
-  * History of messages to include in the context window.
+  * Create a test client with small local model and no temperature for predictable results.
   */
-  // TODO(burdon): Rename messages.
-  history: import_schema2.Schema.optional(import_schema2.Schema.Array(import_artifact3.Message)),
+  static createClient(options) {
+    return new _OllamaClient({
+      tools: options?.tools,
+      overrides: {
+        model: DEFAULT_OLLAMA_MODEL,
+        temperature: 0
+      }
+    });
+  }
   /**
-  * Current request.
+  * Check if Ollama server is running and accessible.
+  * @returns Promise that resolves to true if Ollama is running, false otherwise.
   */
-  prompt: import_schema2.Schema.optional(import_artifact3.Message)
-});
-var __dxlog_file4 = "/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_log4.log)("llm generate", {
-      tools: params.tools
-    }, {
-      F: __dxlog_file4,
-      L: 44,
-      S: void 0,
-      C: (f, a) => f(...a)
-    });
-    const beginTs = Date.now();
-    const stream = await params.client.exec({
-      model: params.model,
-      spaceId: params.spaceId,
-      threadId: params.threadId,
-      history,
-      systemPrompt: params.system,
-      tools: params.tools
-    });
-    for await (const event of stream) {
-      params.logger?.(event);
+  static async isRunning() {
+    try {
+      const response = await fetch(`${DEFAULT_OLLAMA_URL}/api/version`, {
+        method: "GET",
+        headers: {
+          "Content-Type": "application/json"
+        },
+        signal: AbortSignal.timeout(100)
+      });
+      return response.ok;
+    } catch (error) {
+      return false;
     }
-    await stream.complete();
-    const messages = [];
-    const message = messages.at(-1);
-    (0, import_invariant3.invariant)(message, void 0, {
+  }
+  constructor({ endpoint, tools, overrides, maxToolInvocations }) {
+    this._endpoint = endpoint ?? DEFAULT_OLLAMA_URL;
+    this._tools = tools ?? [];
+    this._modelOverride = overrides?.model;
+    this._temperatureOverride = overrides?.temperature;
+    this._maxToolInvocations = maxToolInvocations ?? 3;
+  }
+  _getEmbededTools(clientTools) {
+    return [
+      ...this._tools,
+      ...clientTools.filter((tool) => tool.type !== void 0 && WELL_KNOWN_TOOLS[tool.type]).map((tool) => WELL_KNOWN_TOOLS[tool.type])
+    ];
+  }
+  async *_generateStream(request, signal) {
+    (0, import_invariant2.invariant)(request.prompt === void 0, "Prompt must be in history", {
       F: __dxlog_file4,
-      L: 63,
-      S: void 0,
+      L: 99,
+      S: this,
       A: [
-        "message",
-        ""
+        "request.prompt === undefined",
+        "'Prompt must be in history'"
       ]
     });
-    (0, import_log4.log)("llm result", {
-      time: Date.now() - beginTs,
-      message
+    const ollamaRequest = {
+      model: this._modelOverride ?? request.model ?? "llama3",
+      messages: [],
+      stream: true,
+      options: {
+        temperature: this._temperatureOverride ?? 0.7
+      }
+    };
+    if (request.systemPrompt) {
+      ollamaRequest.messages.push({
+        role: "system",
+        content: request.systemPrompt
+      });
+    }
+    if (request.history?.length) {
+      for (const message of request.history) {
+        const role = message.role === "assistant" ? "assistant" : "user";
+        if (message.content.length === 0) {
+          continue;
+        }
+        for (const content of message.content) {
+          switch (content.type) {
+            case "text": {
+              ollamaRequest.messages.push({
+                role,
+                content: content.text
+              });
+              break;
+            }
+            case "tool_use": {
+              ollamaRequest.messages.push({
+                role: "assistant",
+                content: "",
+                tool_calls: [
+                  {
+                    function: {
+                      name: content.name,
+                      arguments: content.input
+                    }
+                  }
+                ]
+              });
+              break;
+            }
+            case "tool_result": {
+              ollamaRequest.messages.push({
+                role: "tool",
+                content: JSON.stringify({
+                  type: "tool_result",
+                  toolUseId: content.toolUseId,
+                  content: content.content
+                })
+              });
+              break;
+            }
+          }
+        }
+      }
+    }
+    if (request.tools?.length || this._tools.length) {
+      const allTools = [
+        ...request.tools ?? [],
+        ...this._tools
+      ];
+      const withWellKnownTools = allTools.map((tool) => tool.type !== void 0 && WELL_KNOWN_TOOLS[tool.type] ? WELL_KNOWN_TOOLS[tool.type] : tool);
+      if (withWellKnownTools.length > 0) {
+        ollamaRequest.tools = withWellKnownTools.map((tool) => ({
+          type: "function",
+          function: {
+            name: tool.name,
+            description: tool.description,
+            parameters: {
+              type: "object",
+              properties: tool.parameters?.properties,
+              required: tool.parameters?.required
+            }
+          }
+        }));
+      }
+    }
+    import_log3.log.info("generate request", {
+      history: request.history,
+      prompt: request.prompt,
+      ollamaRequest
     }, {
       F: __dxlog_file4,
-      L: 65,
-      S: void 0,
+      L: 193,
+      S: this,
       C: (f, a) => f(...a)
     });
-    (0, import_invariant3.invariant)(message, void 0, {
-      F: __dxlog_file4,
-      L: 66,
-      S: void 0,
-      A: [
-        "message",
-        ""
-      ]
+    const response = await fetch(`${this._endpoint}/api/chat`, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json"
+      },
+      body: JSON.stringify(ollamaRequest),
+      signal
     });
-    params.logger?.({
-      type: "message",
-      message
+    if (!response.ok) {
+      throw new Error(`Ollama API error: ${response.status} ${response.statusText}`);
+    }
+    if (!response.body) {
+      throw new Error("Ollama API returned no response body");
+    }
+    const reader = response.body.pipeThrough(new TextDecoderStream()).pipeThrough(new OllamaDecoderStream()).getReader();
+    try {
+      const messageId = import_echo_schema.ObjectId.random();
+      yield {
+        type: "message_start",
+        message: {
+          id: messageId,
+          role: "assistant",
+          content: []
+        }
+      };
+      const textBlock = {
+        type: "text",
+        text: ""
+      };
+      let currentBlockIndex = 0;
+      yield {
+        type: "content_block_start",
+        index: currentBlockIndex,
+        content: textBlock
+      };
+      while (true) {
+        const { done, value: data } = await reader.read();
+        if (done) {
+          break;
+        }
+        import_log3.log.info("ollama data", {
+          data
+        }, {
+          F: __dxlog_file4,
+          L: 256,
+          S: this,
+          C: (f, a) => f(...a)
+        });
+        if (data.message?.content) {
+          yield {
+            type: "content_block_delta",
+            index: 0,
+            delta: {
+              type: "text_delta",
+              text: data.message.content
+            }
+          };
+        }
+        if (data.message?.tool_calls && data.message?.tool_calls?.length > 0) {
+          if (data.message.tool_calls.length > 1) {
+            import_log3.log.warn("Ollama returned multiple tool calls. Only the first one will be used.", void 0, {
+              F: __dxlog_file4,
+              L: 273,
+              S: this,
+              C: (f, a) => f(...a)
+            });
+          }
+          yield {
+            type: "content_block_stop",
+            index: currentBlockIndex++
+          };
+          yield {
+            type: "content_block_start",
+            index: currentBlockIndex,
+            content: {
+              type: "tool_use",
+              id: import_echo_schema.ObjectId.random(),
+              name: data.message.tool_calls[0].function.name,
+              input: sanitizeToolArguments(data.message.tool_calls[0].function.arguments)
+            }
+          };
+          yield {
+            type: "message_delta",
+            delta: {
+              stopReason: "tool_use"
+            }
+          };
+        }
+      }
+      yield {
+        type: "content_block_stop",
+        index: currentBlockIndex
+      };
+      yield {
+        type: "message_stop"
+      };
+    } finally {
+      reader.releaseLock();
+    }
+  }
+  async exec(request) {
+    const controller = new AbortController();
+    try {
+      return new GenerationStreamImpl(controller, async function* () {
+        const collector = new MessageCollector();
+        let toolInvocations = 0;
+        while (true) {
+          for await (const event of this._generateStream({
+            ...request,
+            history: [
+              ...request.history ?? [],
+              ...request.prompt ? [
+                request.prompt
+              ] : [],
+              ...collector.messages
+            ],
+            prompt: void 0
+          }, controller.signal)) {
+            collector.push(event);
+            yield event;
+          }
+          if (collector.messages.length > 0 && !isToolUse(collector.messages.at(-1), {
+            onlyToolNames: this._getEmbededTools(request.tools ?? []).map((tool) => tool.name)
+          })) {
+            break;
+          }
+          if (++toolInvocations > this._maxToolInvocations) {
+            throw new Error("Maximum number of tool invocations reached");
+          }
+          const result = await runTools({
+            message: collector.messages.at(-1),
+            tools: this._getEmbededTools(request.tools ?? [])
+          });
+          switch (result.type) {
+            case "continue": {
+              collector.pushMessage(result.message);
+              yield* emitMessageAsEvents(result.message);
+              break;
+            }
+            case "break": {
+              throw new Error("Tool use breaks not allowed");
+            }
+          }
+        }
+      }.bind(this));
+    } catch (error) {
+      controller.abort();
+      throw error;
+    }
+  }
+};
+var sanitizeToolArguments = (args) => {
+  if (args.type === "object" && typeof args.properties === "object" && Array.isArray(args.required)) {
+    return args.properties;
+  }
+  return args;
+};
+var SAMPLE_IMAGE_URL = "https://images.nightcafe.studio/jobs/BNmcRhHCM1JRKoUtqSei/BNmcRhHCM1JRKoUtqSei--1--5b9rv.jpg";
+var WELL_KNOWN_TOOLS = {
+  [ToolTypes.TextToImage]: (0, import_artifact2.defineTool)("system", {
+    name: "text-to-image",
+    description: "Generate an image from a text prompt",
+    schema: import_echo_schema.S.Struct({
+      prompt: import_echo_schema.S.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_schema.ObjectId.random();
+      return import_artifact2.ToolResult.Success(`The generation was successful. Image ID: ${id} Prompt: ${prompt}`, [
+        {
+          type: "image",
+          id,
+          source: {
+            type: "base64",
+            mediaType: "image/jpeg",
+            data: imageBase64
+          }
+        }
+      ]);
+    }
+  })
+};
+var OllamaDecoderStream = class extends TransformStream {
+  constructor() {
+    super({
+      transform: (chunk, controller) => {
+        this._buffer += chunk;
+        const lines = this._buffer.split("\n");
+        this._buffer = lines.pop() || "";
+        for (const line of lines) {
+          controller.enqueue(JSON.parse(line.trim()));
+        }
+      },
+      flush: (controller) => {
+        this._buffer = this._buffer.trim();
+        if (this._buffer.length > 0) {
+          controller.enqueue(JSON.parse(this._buffer));
+        }
+      }
     });
-    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_invariant3.invariant)(toolCalls.length === 1, void 0, {
-        F: __dxlog_file4,
-        L: 77,
-        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}`);
+    this._buffer = "";
+  }
+};
+var textChunk = P.regexp(/[^<]+/).map((content) => ({
+  type: "text",
+  content
+}));
+var attribute = P.seq(P.regexp(/[a-zA-Z_][a-zA-Z0-9_-]*/), P.string("=").trim(P.optWhitespace), P.alt(P.regexp(/"[^"]*"/).map((s) => s.slice(1, -1)), P.regexp(/'[^']*'/).map((s) => s.slice(1, -1)))).map(([name, _, value]) => [
+  name,
+  value
+]);
+var tagName = P.regexp(/[a-zA-Z_][a-zA-Z0-9_-]*/);
+var selfClosingTag = P.seqMap(
+  //
+  P.string("<"),
+  tagName,
+  P.optWhitespace.then(attribute).many(),
+  P.optWhitespace.then(P.string("/>")),
+  P.optWhitespace,
+  (_, tag, attributes, __) => ({
+    type: "tag",
+    tag,
+    attributes: Object.fromEntries(attributes),
+    content: [],
+    selfClosing: true
+  })
+);
+var openTag = P.seqMap(
+  //
+  P.string("<"),
+  tagName,
+  P.optWhitespace.then(attribute).many(),
+  P.string(">"),
+  P.regexp(/[^<]*/),
+  (_, tag, attributes, __, content) => ({
+    type: "tag",
+    tag,
+    attributes: Object.fromEntries(attributes),
+    content: content.length ? [
+      {
+        type: "text",
+        content
+      }
+    ] : []
+  })
+);
+var closeTag = P.seqMap(
+  //
+  P.string("</"),
+  tagName,
+  P.string(">"),
+  P.optWhitespace,
+  (_, tag) => ({
+    type: "tag",
+    tag,
+    content: [],
+    closing: true
+  })
+);
+var mixedChunk = P.alt(selfClosingTag, openTag, closeTag, textChunk);
+var StreamTransform = class {
+  constructor() {
+    this._buffer = "";
+  }
+  transform(chunk) {
+    this._buffer += chunk;
+    const results = [];
+    const parser = mixedChunk.many().skip(P.optWhitespace);
+    const result = parser.parse(this._buffer);
+    if (result.status) {
+      for (const chunk2 of result.value) {
+        results.push(chunk2);
       }
-      (0, import_invariant3.invariant)(tool.execute, void 0, {
-        F: __dxlog_file4,
-        L: 84,
-        S: void 0,
+      this._buffer = "";
+    }
+    return results;
+  }
+  // TODO(burdon): Process remaining buffer?
+  flush() {
+    const remaining = this._buffer;
+    this._buffer = "";
+    return remaining.length > 0 ? [
+      {
+        type: "text",
+        content: remaining
+      }
+    ] : [];
+  }
+};
+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();
+  }
+  _emitBlock(contentBlock, streamBlock) {
+    const messageBlock = streamBlock ? mergeMessageBlock(contentBlock, streamBlock) : contentBlock;
+    if (messageBlock) {
+      if (messageBlock.type === "text" && messageBlock.text.length === 0) {
+        return;
+      }
+      (0, import_invariant4.invariant)(this._message, void 0, {
+        F: __dxlog_file5,
+        L: 49,
+        S: this,
         A: [
-          "tool.execute",
+          "this._message",
           ""
         ]
       });
-      const toolResult = await tool.execute(toolCall.input);
-      switch (toolResult.kind) {
-        case "error": {
-          import_log4.log.warn("tool error", {
-            message: toolResult.message
-          }, {
-            F: __dxlog_file4,
-            L: 88,
-            S: void 0,
-            C: (f, a) => f(...a)
-          });
-          const resultMessage = {
-            id: import_echo_schema3.ObjectId.random(),
-            spaceId: params.spaceId,
-            threadId: params.threadId,
-            role: "user",
+      this._message.content.push(messageBlock);
+      this.block.emit(messageBlock);
+    }
+  }
+  _emitUpdate(contentBlock, streamBlock) {
+    const messageBlock = mergeMessageBlock(contentBlock, streamBlock);
+    if (messageBlock) {
+      messageBlock.pending = true;
+      this.update.emit(messageBlock);
+    }
+  }
+  /**
+  * Parse stream until end.
+  */
+  async parse(stream) {
+    const transformer = new StreamTransform();
+    let contentBlock;
+    let streamBlock;
+    const stack = [];
+    const messagesCollected = [];
+    for await (const event of stream) {
+      this.streamEvent.emit(event);
+      switch (event.type) {
+        //
+        // Messages.
+        //
+        case "message_start": {
+          if (this._message) {
+            import_log5.log.warn("unexpected message_start", void 0, {
+              F: __dxlog_file5,
+              L: 93,
+              S: this,
+              C: (f, a) => f(...a)
+            });
+          }
+          this._message = (0, import_echo_schema4.createStatic)(import_artifact5.Message, {
+            role: event.message.role,
             content: [
-              {
-                type: "tool_result",
-                toolUseId: toolCall.id,
-                content: toolResult.message,
-                isError: true
-              }
+              ...event.message.content
             ]
-          };
-          params.logger?.({
-            type: "message",
-            message: resultMessage
           });
-          history.push(resultMessage);
-          return true;
+          this.message.emit(this._message);
+          break;
         }
-        case "success": {
-          (0, import_log4.log)("tool success", {
-            result: toolResult.result
-          }, {
-            F: __dxlog_file4,
-            L: 110,
-            S: void 0,
-            C: (f, a) => f(...a)
-          });
-          const resultMessage = {
-            id: import_echo_schema3.ObjectId.random(),
-            spaceId: params.spaceId,
-            threadId: params.threadId,
-            role: "user",
-            content: [
-              {
-                type: "tool_result",
-                toolUseId: toolCall.id,
-                content: JSON.stringify(toolResult.result)
+        case "message_delta": {
+          if (!this._message) {
+            import_log5.log.warn("unexpected message_delta", void 0, {
+              F: __dxlog_file5,
+              L: 103,
+              S: this,
+              C: (f, a) => f(...a)
+            });
+          }
+          break;
+        }
+        case "message_stop": {
+          if (!this._message) {
+            import_log5.log.warn("unexpected message_stop", void 0, {
+              F: __dxlog_file5,
+              L: 111,
+              S: this,
+              C: (f, a) => f(...a)
+            });
+          } else {
+            messagesCollected.push(this._message);
+            this._message = void 0;
+          }
+          break;
+        }
+        //
+        // Content blocks.
+        //
+        case "content_block_start": {
+          if (contentBlock) {
+            import_log5.log.warn("unexpected content_block_start", {
+              content: contentBlock
+            }, {
+              F: __dxlog_file5,
+              L: 125,
+              S: this,
+              C: (f, a) => f(...a)
+            });
+          }
+          contentBlock = event.content;
+          break;
+        }
+        case "content_block_delta": {
+          if (!contentBlock) {
+            import_log5.log.warn("unexpected content_block_delta", void 0, {
+              F: __dxlog_file5,
+              L: 135,
+              S: this,
+              C: (f, a) => f(...a)
+            });
+            break;
+          }
+          switch (event.delta.type) {
+            //
+            // Text
+            //
+            case "text_delta": {
+              const chunks = transformer.transform(event.delta.text);
+              for (const chunk of chunks) {
+                switch (streamBlock?.type) {
+                  //
+                  // XML Fragment.
+                  //
+                  case "tag": {
+                    if (chunk.type === "tag") {
+                      if (chunk.selfClosing) {
+                        streamBlock.content.push(chunk);
+                      } else if (chunk.closing) {
+                        if (stack.length > 0) {
+                          const top = stack.pop();
+                          (0, import_invariant4.invariant)(top && top.type === "tag", void 0, {
+                            F: __dxlog_file5,
+                            L: 159,
+                            S: this,
+                            A: [
+                              "top && top.type === 'tag'",
+                              ""
+                            ]
+                          });
+                          top.content.push(streamBlock);
+                          streamBlock = top;
+                        } else {
+                          this._emitBlock(contentBlock, streamBlock);
+                          streamBlock = void 0;
+                        }
+                      } else {
+                        stack.push(streamBlock);
+                        streamBlock = chunk;
+                      }
+                    } else {
+                      if (streamBlock.content.length === 0) {
+                        streamBlock.content.push(chunk);
+                      } else {
+                        const last = streamBlock.content.at(-1);
+                        (0, import_invariant4.invariant)(last, void 0, {
+                          F: __dxlog_file5,
+                          L: 176,
+                          S: this,
+                          A: [
+                            "last",
+                            ""
+                          ]
+                        });
+                        if (last.type === "text") {
+                          last.content += chunk.content;
+                        } else {
+                          streamBlock.content.push(chunk);
+                        }
+                      }
+                    }
+                    break;
+                  }
+                  //
+                  // Text Fragment.
+                  //
+                  case "text": {
+                    if (chunk.type === "tag") {
+                      this._emitBlock(contentBlock, streamBlock);
+                      if (chunk.selfClosing) {
+                        this._emitBlock(contentBlock, chunk);
+                        streamBlock = void 0;
+                      } else {
+                        streamBlock = chunk;
+                      }
+                    } else {
+                      streamBlock.content += chunk.content;
+                    }
+                    break;
+                  }
+                  //
+                  // No current chunk.
+                  //
+                  default: {
+                    if (chunk.type === "tag" && chunk.selfClosing) {
+                      this._emitBlock(contentBlock, chunk);
+                    } else {
+                      streamBlock = chunk;
+                    }
+                  }
+                }
               }
-            ]
-          };
-          params.logger?.({
-            type: "message",
-            message: resultMessage
-          });
-          history.push(resultMessage);
-          return true;
+              break;
+            }
+            //
+            // JSON
+            //
+            case "input_json_delta": {
+              if (!contentBlock) {
+                import_log5.log.warn("unexpected input_json_delta", {
+                  contentBlock
+                }, {
+                  F: __dxlog_file5,
+                  L: 226,
+                  S: this,
+                  C: (f, a) => f(...a)
+                });
+                break;
+              }
+              if (streamBlock?.type === "json") {
+                streamBlock.content += event.delta.partial_json;
+              } else {
+                streamBlock = {
+                  type: "json",
+                  content: event.delta.partial_json
+                };
+              }
+              break;
+            }
+          }
+          break;
         }
-        case "break": {
-          (0, import_log4.log)("tool break", {
-            result: toolResult.result
-          }, {
-            F: __dxlog_file4,
-            L: 131,
-            S: void 0,
-            C: (f, a) => f(...a)
-          });
-          conversationResult = toolResult.result;
-          return false;
+        case "content_block_stop": {
+          if (!contentBlock) {
+            import_log5.log.warn("unexpected content_block_stop", void 0, {
+              F: __dxlog_file5,
+              L: 243,
+              S: this,
+              C: (f, a) => f(...a)
+            });
+            break;
+          }
+          this._emitBlock(contentBlock, streamBlock);
+          contentBlock = void 0;
+          streamBlock = void 0;
+          break;
         }
       }
+      if (contentBlock && streamBlock) {
+        this._emitUpdate(contentBlock, streamBlock);
+      }
     }
-    return false;
-  };
-  while (await generate()) {
+    return messagesCollected;
   }
-  return {
-    result: conversationResult,
-    history
-  };
-};
-var isToolUse = (message) => {
-  const block = message.content.at(-1);
-  (0, import_invariant3.invariant)(block, void 0, {
-    F: __dxlog_file4,
-    L: 152,
-    S: void 0,
-    A: [
-      "block",
-      ""
-    ]
-  });
-  return block.type === "tool_use";
 };
-var runTools = async ({ message, tools, extensions }) => {
-  const toolCalls = message.content.filter((block) => block.type === "tool_use");
-  (0, import_invariant3.invariant)(toolCalls.length === 1, void 0, {
-    F: __dxlog_file4,
-    L: 174,
+var mergeMessageBlock = (contentBlock, streamBlock) => {
+  (0, import_log5.log)("mergeMessageBlock", {
+    contentBlock,
+    streamBlock
+  }, {
+    F: __dxlog_file5,
+    L: 270,
     S: void 0,
-    A: [
-      "toolCalls.length === 1",
-      ""
-    ]
+    C: (f, a) => f(...a)
   });
-  const toolCall = toolCalls[0];
-  const tool = tools.find((tool2) => tool2.name === toolCall.name);
-  if (!tool) {
-    throw new Error(`Tool not found: ${toolCall.name}`);
-  }
-  let toolResult;
-  try {
-    (0, import_invariant3.invariant)(tool.execute, void 0, {
-      F: __dxlog_file4,
-      L: 183,
-      S: void 0,
-      A: [
-        "tool.execute",
-        ""
-      ]
-    });
-    toolResult = await tool.execute(toolCall.input, {
-      extensions
-    });
-  } catch (error) {
-    (0, import_log4.log)("tool error", {
-      error
-    }, {
-      F: __dxlog_file4,
-      L: 186,
-      S: void 0,
-      C: (f, a) => f(...a)
-    });
-    toolResult = import_artifact4.ToolResult.Error(error.message);
-  }
-  switch (toolResult.kind) {
-    case "error": {
-      (0, import_log4.log)("tool error", {
-        message: toolResult.message
-      }, {
-        F: __dxlog_file4,
-        L: 192,
-        S: void 0,
-        C: (f, a) => f(...a)
-      });
-      const resultMessage = (0, import_echo_schema3.createStatic)(import_artifact4.Message, {
-        role: "user",
-        content: [
-          {
-            type: "tool_result",
-            toolUseId: toolCall.id,
-            content: toolResult.message,
-            isError: true
-          }
-        ]
-      });
+  switch (streamBlock.type) {
+    //
+    // Text
+    //
+    case "text": {
       return {
-        type: "continue",
-        message: resultMessage
+        ...contentBlock,
+        type: "text",
+        text: streamBlock.content
       };
     }
-    case "success": {
-      (0, import_log4.log)("tool success", {
-        result: toolResult.result
-      }, {
-        F: __dxlog_file4,
-        L: 213,
-        S: void 0,
-        C: (f, a) => f(...a)
-      });
-      const resultMessage = (0, import_echo_schema3.createStatic)(import_artifact4.Message, {
-        role: "user",
-        content: [
-          {
-            type: "tool_result",
-            toolUseId: toolCall.id,
-            content: typeof toolResult.result === "string" ? toolResult.result : JSON.stringify(toolResult.result) ?? ""
+    //
+    // XML
+    //
+    case "tag": {
+      switch (streamBlock.tag) {
+        case "cot": {
+          const content = streamBlock.content.map((block) => {
+            switch (block.type) {
+              case "text":
+                return block.content;
+              default:
+                return null;
+            }
+          }).filter(import_util.isNotFalsy).join("\n");
+          return {
+            ...contentBlock,
+            type: "text",
+            disposition: "cot",
+            text: content
+          };
+        }
+        case "artifact": {
+          const { attributes } = streamBlock;
+          return {
+            type: "json",
+            disposition: "artifact",
+            json: JSON.stringify(attributes)
+          };
+        }
+        case "suggest": {
+          if (streamBlock.content.length === 1 && streamBlock.content[0].type === "text") {
+            return {
+              type: "json",
+              disposition: "suggest",
+              json: JSON.stringify({
+                text: streamBlock.content[0].content
+              })
+            };
           }
-        ]
-      });
-      return {
-        type: "continue",
-        message: resultMessage
-      };
+          break;
+        }
+        case "select": {
+          return {
+            type: "json",
+            disposition: "select",
+            json: JSON.stringify({
+              options: streamBlock.content.flatMap((content) => content.type === "tag" && content.content.length === 1 && content.content[0].type === "text" ? [
+                content.content[0].content
+              ] : [])
+            })
+          };
+        }
+        case "tool-list": {
+          return {
+            type: "json",
+            disposition: "tool_list",
+            json: JSON.stringify({})
+          };
+        }
+      }
+      break;
     }
-    case "break": {
-      (0, import_log4.log)("tool break", {
-        result: toolResult.result
-      }, {
-        F: __dxlog_file4,
-        L: 233,
-        S: void 0,
-        C: (f, a) => f(...a)
-      });
+    //
+    // JSON
+    //
+    case "json": {
+      switch (contentBlock.type) {
+        case "tool_use": {
+          return {
+            ...contentBlock,
+            input: (0, import_util.safeParseJson)(streamBlock.content) ?? {}
+          };
+        }
+      }
       return {
-        type: "break",
-        result: toolResult.result
+        ...contentBlock,
+        type: "json",
+        disposition: "artifact",
+        json: streamBlock.content
       };
     }
   }
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  AIServiceClientImpl,
+  AIServiceEdgeClient,
   DEFAULT_LLM_MODEL,
   DEFAULT_LLM_MODELS,
   EchoDataSource,
   GenerateRequest,
-  GenerationStream,
+  GenerationStreamImpl,
   LLMModel,
+  MessageCollector,
   MixedStreamParser,
+  OllamaClient,
   StreamTransform,
   ToolTypes,
   createArtifactElement,
   createGenerationStream,
+  emitMessageAsEvents,
   executeQuery,
   executeQueryPlan,
   formatInferredRelationshipLabel,