@dxos/assistant 0.7.4 → 0.7.5-labs.5f04cf6

package/dist/lib/browser/index.mjs

@@ -1,71 +1,960 @@
 import {
-  AIServiceClientImpl,
   EchoDataSource,
-  GenerateRequest,
-  GenerationStream,
-  ImageSource,
-  LLMModel,
-  LLMTool,
-  LLMToolDefinition,
-  LLMToolResult,
-  Message,
-  MessageContentBlock,
-  MessageFromDb,
-  MessageImageContentBlock,
-  MessageRole,
-  MessageTextContentBlock,
-  MessageToolResultContentBlock,
-  MessageToolUseContentBlock,
-  ObjectId,
-  ObjectIdBrand,
-  Space,
-  SpaceIdSchema,
-  Thread,
-  ToolTypes,
-  createUserMessage,
-  defineTool,
   executeQuery,
   executeQueryPlan,
   formatInferredRelationshipLabel,
   formatJsonSchemaForLLM,
   formatNodeLabel,
   isReference,
-  parseCypherQuery,
-  runLLM
-} from "./chunk-AKP45LQH.mjs";
+  parseCypherQuery
+} from "./chunk-37SDA65J.mjs";
+
+// packages/experimental/assistant/src/ai-service/client.ts
+import { Schema as S } from "@effect/schema";
+import { Message } from "@dxos/artifact";
+import { invariant as invariant2 } from "@dxos/invariant";
+
+// packages/experimental/assistant/src/ai-service/stream.ts
+import { Trigger } from "@dxos/async";
+import { invariant } from "@dxos/invariant";
+import { log } from "@dxos/log";
+
+// packages/experimental/assistant/src/ai-service/util/sse.ts
+async function* iterSSEMessages(response, controller) {
+  if (!response.body) {
+    controller.abort();
+    throw new Error("Attempted to iterate over a response with no body");
+  }
+  const sseDecoder = new SSEDecoder();
+  const lineDecoder = new LineDecoder();
+  const iter = readableStreamAsyncIterable(response.body);
+  for await (const sseChunk of iterSSEChunks(iter)) {
+    for (const line of lineDecoder.decode(sseChunk)) {
+      const sse = sseDecoder.decode(line);
+      if (sse) {
+        yield sse;
+      }
+    }
+  }
+  for (const line of lineDecoder.flush()) {
+    const sse = sseDecoder.decode(line);
+    if (sse) {
+      yield sse;
+    }
+  }
+}
+async function* iterSSEChunks(iterator) {
+  let data = new Uint8Array();
+  for await (const chunk of iterator) {
+    if (chunk == null) {
+      continue;
+    }
+    const binaryChunk = chunk instanceof ArrayBuffer ? new Uint8Array(chunk) : typeof chunk === "string" ? new TextEncoder().encode(chunk) : chunk;
+    const newData = new Uint8Array(data.length + binaryChunk.length);
+    newData.set(data);
+    newData.set(binaryChunk, data.length);
+    data = newData;
+    let patternIndex;
+    while ((patternIndex = findDoubleNewlineIndex(data)) !== -1) {
+      yield data.slice(0, patternIndex);
+      data = data.slice(patternIndex);
+    }
+  }
+  if (data.length > 0) {
+    yield data;
+  }
+}
+var findDoubleNewlineIndex = (buffer) => {
+  const newline = 10;
+  const carriage = 13;
+  for (let i = 0; i < buffer.length - 2; i++) {
+    if (buffer[i] === newline && buffer[i + 1] === newline) {
+      return i + 2;
+    }
+    if (buffer[i] === carriage && buffer[i + 1] === carriage) {
+      return i + 2;
+    }
+    if (buffer[i] === carriage && buffer[i + 1] === newline && i + 3 < buffer.length && buffer[i + 2] === carriage && buffer[i + 3] === newline) {
+      return i + 4;
+    }
+  }
+  return -1;
+};
+var SSEDecoder = class {
+  constructor() {
+    this.event = null;
+    this.data = [];
+    this.chunks = [];
+  }
+  decode(line) {
+    if (line.endsWith("\r")) {
+      line = line.substring(0, line.length - 1);
+    }
+    if (!line) {
+      if (!this.event && !this.data.length) {
+        return null;
+      }
+      const sse = {
+        event: this.event,
+        data: this.data.join("\n"),
+        raw: this.chunks
+      };
+      this.event = null;
+      this.data = [];
+      this.chunks = [];
+      return sse;
+    }
+    this.chunks.push(line);
+    if (line.startsWith(":")) {
+      return null;
+    }
+    let [fieldname, _, value] = partition(line, ":");
+    if (value.startsWith(" ")) {
+      value = value.substring(1);
+    }
+    if (fieldname === "event") {
+      this.event = value;
+    } else if (fieldname === "data") {
+      this.data.push(value);
+    }
+    return null;
+  }
+};
+var partition = (str, delimiter) => {
+  const index = str.indexOf(delimiter);
+  if (index !== -1) {
+    return [
+      str.substring(0, index),
+      delimiter,
+      str.substring(index + delimiter.length)
+    ];
+  }
+  return [
+    str,
+    "",
+    ""
+  ];
+};
+function readableStreamAsyncIterable(stream) {
+  if (stream[Symbol.asyncIterator]) {
+    return stream;
+  }
+  const reader = stream.getReader();
+  return {
+    next: async () => {
+      try {
+        const result = await reader.read();
+        if (result?.done) {
+          reader.releaseLock();
+        }
+        return result;
+      } catch (e) {
+        reader.releaseLock();
+        throw e;
+      }
+    },
+    return: async () => {
+      const cancelPromise = reader.cancel();
+      reader.releaseLock();
+      await cancelPromise;
+      return {
+        done: true,
+        value: void 0
+      };
+    },
+    [Symbol.asyncIterator]() {
+      return this;
+    }
+  };
+}
+var LineDecoder = class _LineDecoder {
+  static {
+    // prettier-ignore
+    this.NEWLINE_CHARS = /* @__PURE__ */ new Set([
+      "\n",
+      "\r"
+    ]);
+  }
+  static {
+    this.NEWLINE_REGEXP = /\r\n|[\n\r]/g;
+  }
+  constructor() {
+    this.buffer = [];
+    this.trailingCR = false;
+  }
+  decode(chunk) {
+    let text = this.decodeText(chunk);
+    if (this.trailingCR) {
+      text = "\r" + text;
+      this.trailingCR = false;
+    }
+    if (text.endsWith("\r")) {
+      this.trailingCR = true;
+      text = text.slice(0, -1);
+    }
+    if (!text) {
+      return [];
+    }
+    const trailingNewline = _LineDecoder.NEWLINE_CHARS.has(text[text.length - 1] || "");
+    let lines = text.split(_LineDecoder.NEWLINE_REGEXP);
+    if (trailingNewline) {
+      lines.pop();
+    }
+    if (lines.length === 1 && !trailingNewline) {
+      this.buffer.push(lines[0]);
+      return [];
+    }
+    if (this.buffer.length > 0) {
+      lines = [
+        this.buffer.join("") + lines[0],
+        ...lines.slice(1)
+      ];
+      this.buffer = [];
+    }
+    if (!trailingNewline) {
+      this.buffer = [
+        lines.pop() || ""
+      ];
+    }
+    return lines;
+  }
+  decodeText(bytes) {
+    if (bytes == null) {
+      return "";
+    }
+    if (typeof bytes === "string") {
+      return bytes;
+    }
+    if (typeof Buffer !== "undefined") {
+      if (bytes instanceof Buffer) {
+        return bytes.toString();
+      }
+      if (bytes instanceof Uint8Array) {
+        return Buffer.from(bytes).toString();
+      }
+      throw new Error(`Unexpected: received non-Uint8Array (${bytes.constructor.name}) stream chunk in an environment with a global "Buffer" defined, which this library assumes to be Node. Please report this error.`);
+    }
+    if (typeof TextDecoder !== "undefined") {
+      if (bytes instanceof Uint8Array || bytes instanceof ArrayBuffer) {
+        this.textDecoder ??= new TextDecoder("utf8");
+        return this.textDecoder.decode(bytes);
+      }
+      throw new Error(`Unexpected: received non-Uint8Array/ArrayBuffer (${bytes.constructor.name}) in a web platform. Please report this error.`);
+    }
+    throw new Error("Unexpected: neither Buffer nor TextDecoder are available as globals. Please report this error.");
+  }
+  flush() {
+    if (!this.buffer.length && !this.trailingCR) {
+      return [];
+    }
+    const lines = [
+      this.buffer.join("")
+    ];
+    this.buffer = [];
+    this.trailingCR = false;
+    return lines;
+  }
+};
+
+// packages/experimental/assistant/src/ai-service/stream.ts
+var __dxlog_file = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/ai-service/stream.ts";
+var GenerationStream = class _GenerationStream {
+  /**
+  * Creates a stream from an SSE response.
+  */
+  static fromSSEResponse(params, response, controller = new AbortController()) {
+    const iterator = async function* () {
+      for await (const sse of iterSSEMessages(response, controller)) {
+        if (sse.event === "completion") {
+          try {
+            yield JSON.parse(sse.data);
+          } catch (err) {
+            log.error("could not parse message into JSON:", {
+              data: sse.data,
+              raw: sse.raw
+            }, {
+              F: __dxlog_file,
+              L: 31,
+              S: this,
+              C: (f, a) => f(...a)
+            });
+            throw err;
+          }
+        }
+        if (sse.event === "message_start" || sse.event === "message_delta" || sse.event === "message_stop" || sse.event === "content_block_start" || sse.event === "content_block_delta" || sse.event === "content_block_stop") {
+          try {
+            yield JSON.parse(sse.data);
+          } catch (err) {
+            log.error("could not parse message into JSON:", {
+              data: sse.data,
+              raw: sse.raw
+            }, {
+              F: __dxlog_file,
+              L: 47,
+              S: this,
+              C: (f, a) => f(...a)
+            });
+            throw err;
+          }
+        }
+        if (sse.event === "ping") {
+          continue;
+        }
+        if (sse.event === "error") {
+          throw new Error(`Message generation error: ${sse.data}`);
+        }
+      }
+    };
+    return new _GenerationStream(controller, iterator, params);
+  }
+  constructor(_controller, _getIterator, _params) {
+    this._controller = _controller;
+    this._getIterator = _getIterator;
+    this._params = _params;
+    this._done = new Trigger();
+    this._previousMessages = [];
+    this._accumulatedMessage = void 0;
+    this._iterator = void 0;
+  }
+  [Symbol.asyncIterator]() {
+    return this._createIterator();
+  }
+  get accumulatedMessages() {
+    return [
+      ...this._previousMessages,
+      ...this._accumulatedMessage == null ? [] : [
+        this._accumulatedMessage
+      ]
+    ];
+  }
+  /**
+  * Aborts the stream.
+  */
+  abort() {
+    this._controller.abort();
+  }
+  /**
+  * Returns the complete message list generated by the AI service.
+  */
+  async complete() {
+    await this._done.wait();
+    return this.accumulatedMessages;
+  }
+  _createIterator() {
+    const self = this;
+    return this._iterator ??= (() => {
+      const generator = async function* () {
+        try {
+          for await (const event of self._getIterator()) {
+            self._processEvent(event);
+            yield event;
+          }
+          this._done.wake();
+        } catch (err) {
+          this._done.throw(err);
+        }
+      };
+      return generator.call(this);
+    })();
+  }
+  _processEvent(event) {
+    switch (event.type) {
+      case "message_start": {
+        if (this._accumulatedMessage) {
+          this._previousMessages.push(this._accumulatedMessage);
+          this._accumulatedMessage = void 0;
+        }
+        this._accumulatedMessage = {
+          ...event.message,
+          threadId: this._params.threadId,
+          spaceId: this._params.spaceId
+        };
+        break;
+      }
+      case "message_delta": {
+        if (this._accumulatedMessage === void 0) {
+          throw new Error("Received message delta without a message start");
+        }
+        break;
+      }
+      case "message_stop": {
+        if (this._accumulatedMessage === void 0) {
+          throw new Error("Received message stop without a message start");
+        }
+        break;
+      }
+      case "content_block_start": {
+        invariant(this._accumulatedMessage, void 0, {
+          F: __dxlog_file,
+          L: 149,
+          S: this,
+          A: [
+            "this._accumulatedMessage",
+            ""
+          ]
+        });
+        this._accumulatedMessage.content.push(event.content);
+        break;
+      }
+      case "content_block_delta": {
+        invariant(this._accumulatedMessage, void 0, {
+          F: __dxlog_file,
+          L: 155,
+          S: this,
+          A: [
+            "this._accumulatedMessage",
+            ""
+          ]
+        });
+        const snapshotContent = this._accumulatedMessage.content.at(event.index);
+        if (snapshotContent?.type === "text" && event.delta.type === "text_delta") {
+          snapshotContent.text += event.delta.text;
+        } else if (snapshotContent?.type === "tool_use" && event.delta.type === "input_json_delta") {
+          snapshotContent.inputJson ??= "";
+          snapshotContent.inputJson += event.delta.partial_json;
+        }
+        break;
+      }
+      case "content_block_stop": {
+        invariant(this._accumulatedMessage, void 0, {
+          F: __dxlog_file,
+          L: 172,
+          S: this,
+          A: [
+            "this._accumulatedMessage",
+            ""
+          ]
+        });
+        const lastBlock = this._accumulatedMessage.content.at(-1);
+        if (lastBlock?.type === "tool_use") {
+          lastBlock.input = lastBlock.inputJson == null || lastBlock.inputJson === "" ? {} : JSON.parse(lastBlock.inputJson ?? "");
+        }
+        break;
+      }
+    }
+  }
+};
+
+// packages/experimental/assistant/src/ai-service/client.ts
+var __dxlog_file2 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/ai-service/client.ts";
+var AIServiceClientImpl = class {
+  constructor({ endpoint }) {
+    invariant2(endpoint, "endpoint is required", {
+      F: __dxlog_file2,
+      L: 36,
+      S: this,
+      A: [
+        "endpoint",
+        "'endpoint is required'"
+      ]
+    });
+    this._endpoint = endpoint;
+  }
+  async getSpace(spaceId) {
+    throw new Error("Not implemented");
+  }
+  async getThread(spaceId, threadId) {
+    throw new Error("Not implemented");
+  }
+  async getMessages(spaceId, threadId) {
+    const res = await fetch(`${this._endpoint}/space/${spaceId}/thread/${threadId}/message`);
+    return S.decodePromise(S.Array(Message).pipe(S.mutable))((await res.json()).data.results);
+  }
+  async appendMessages(messages) {
+    const url = `${this._endpoint}/message`;
+    const res = await fetch(url, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json"
+      },
+      body: JSON.stringify({
+        messages: await S.encodePromise(S.Array(Message))(messages)
+      })
+    });
+    if (!res.ok) {
+      throw new Error(`${await res.text()} [${url}]`);
+    }
+  }
+  // TODO(dmaretskyi): Accept partial message schema.
+  async updateMessage(spaceId, threadId, messageId, message) {
+    throw new Error("Not implemented");
+  }
+  /**
+  * Open message stream.
+  */
+  async generate(request) {
+    const controller = new AbortController();
+    const signal = controller.signal;
+    const response = await fetch(`${this._endpoint}/generate`, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json"
+      },
+      body: JSON.stringify(request),
+      signal
+    });
+    invariant2(response.body instanceof ReadableStream, void 0, {
+      F: __dxlog_file2,
+      L: 87,
+      S: this,
+      A: [
+        "response.body instanceof ReadableStream",
+        ""
+      ]
+    });
+    return GenerationStream.fromSSEResponse({
+      spaceId: request.spaceId,
+      threadId: request.threadId
+    }, response, controller);
+  }
+};
+
+// packages/experimental/assistant/src/ai-service/defs.ts
+import { Schema as S2 } from "@effect/schema";
+import { Tool, Message as Message2, SpaceIdSchema } from "@dxos/artifact";
+import { ObjectId } from "@dxos/echo-schema";
+var createArtifactElement = (id) => `<artifact id=${id} />`;
+var LLMModel = S2.Literal("@anthropic/claude-3-5-haiku-20241022", "@anthropic/claude-3-5-sonnet-20241022", "@cf/deepseek-ai/deepseek-r1-distill-qwen-32b", "@hf/nousresearch/hermes-2-pro-mistral-7b", "@ollama/llama-3-2-3b", "@ollama/llama-3-1-nemotron-70b-instruct", "@ollama/llama-3-1-nemotron-mini-4b-instruct", "@cf/deepseek-ai/deepseek-r1-distill-qwen-32b");
+var ToolTypes = Object.freeze({
+  // TODO(dmaretskyi): Not implemented yet.
+  // DatabaseQuery: 'database_query',
+  TextToImage: "text_to_image"
+});
+var GenerateRequest = S2.Struct({
+  spaceId: S2.optional(SpaceIdSchema),
+  threadId: S2.optional(ObjectId),
+  model: LLMModel,
+  /**
+  * History of messages to include in the context window.
+  */
+  history: S2.optional(S2.Array(Message2)),
+  /**
+  * System instructions to the LLM.
+  */
+  // TODO(dmaretskyi): Should this be part of the thread?.
+  systemPrompt: S2.optional(S2.String),
+  /**
+  * Tools available for the LLM.
+  */
+  tools: S2.optional(S2.Array(Tool).pipe(S2.mutable))
+});
+
+// packages/experimental/assistant/src/ai-service/parser.ts
+var StreamingParser = class {
+  constructor(onBlock) {
+    this.onBlock = onBlock;
+    this.buffer = "";
+  }
+  /**
+  * Write a chunk of text to the parser.
+  */
+  write(chunk) {
+    this.buffer += chunk;
+    this.processBuffer();
+  }
+  /**
+  * Signal that no more input will be written.
+  */
+  end() {
+    if (this.buffer.length > 0) {
+      this.emitTextBlock(this.buffer);
+      this.buffer = "";
+    }
+  }
+  processBuffer() {
+    while (this.buffer.length > 0) {
+      if (this.tryParseCodeBlock()) {
+        continue;
+      }
+      const tagStartIndex = this.buffer.indexOf("<");
+      if (tagStartIndex === -1) {
+        return;
+      }
+      if (tagStartIndex > 0) {
+        this.emitTextBlock(this.buffer.slice(0, tagStartIndex));
+        this.buffer = this.buffer.slice(tagStartIndex);
+        continue;
+      }
+      const tagEndIndex = this.buffer.indexOf(">", tagStartIndex);
+      if (tagEndIndex === -1) {
+        return;
+      }
+      const isSelfClosing = this.buffer[tagEndIndex - 1] === "/";
+      const actualTagEndIndex = isSelfClosing ? tagEndIndex - 1 : tagEndIndex;
+      const tagContent = this.buffer.slice(tagStartIndex + 1, actualTagEndIndex).trim();
+      const [tagName, ...attrParts] = tagContent.split(/\s+/);
+      const attributes = this.parseAttributes(attrParts.join(" "));
+      if (isSelfClosing) {
+        this.emitTagBlock(tagName, "", attributes, true);
+        this.buffer = this.buffer.slice(tagEndIndex + 1);
+        continue;
+      }
+      const closingTagIndex = this.buffer.indexOf("</" + tagName + ">", tagEndIndex);
+      if (closingTagIndex === -1) {
+        return;
+      }
+      const content = this.buffer.slice(tagEndIndex + 1, closingTagIndex);
+      this.emitTagBlock(tagName, content, attributes, false);
+      this.buffer = this.buffer.slice(closingTagIndex + tagName.length + 3);
+    }
+  }
+  tryParseCodeBlock() {
+    const codeBlockStart = this.buffer.indexOf("```");
+    if (codeBlockStart === -1) {
+      return false;
+    }
+    if (codeBlockStart > 0) {
+      this.emitTextBlock(this.buffer.slice(0, codeBlockStart));
+      this.buffer = this.buffer.slice(codeBlockStart);
+      return true;
+    }
+    const nextNewline = this.buffer.indexOf("\n", codeBlockStart);
+    if (nextNewline === -1) {
+      return false;
+    }
+    const codeBlockEnd = this.buffer.indexOf("\n```", nextNewline);
+    if (codeBlockEnd === -1) {
+      return false;
+    }
+    const languageLine = this.buffer.slice(3, nextNewline).trim();
+    const content = this.buffer.slice(nextNewline + 1, codeBlockEnd);
+    this.emitCodeBlock(languageLine, content);
+    this.buffer = this.buffer.slice(codeBlockEnd + 4);
+    return true;
+  }
+  parseAttributes(attrString) {
+    const attributes = {};
+    if (!attrString) {
+      return attributes;
+    }
+    const attrPattern = /(\w+)\s*=\s*(?:"([^"]*)"|'([^']*)'|(\S+))/g;
+    let match;
+    while ((match = attrPattern.exec(attrString)) !== null) {
+      const [, name, doubleQuoted, singleQuoted, unquoted] = match;
+      attributes[name] = doubleQuoted ?? singleQuoted ?? unquoted;
+    }
+    return attributes;
+  }
+  emitTextBlock(text) {
+    if (text.trim().length > 0) {
+      this.onBlock({
+        type: "text",
+        content: text
+      });
+    }
+  }
+  emitTagBlock(name, content, attributes, selfClosing) {
+    this.onBlock({
+      type: "tag",
+      name,
+      content: content.trim(),
+      attributes,
+      selfClosing
+    });
+  }
+  emitCodeBlock(language, content) {
+    this.onBlock({
+      type: "code",
+      language,
+      content: content.trim()
+    });
+  }
+};
+
+// packages/experimental/assistant/src/conversation/conversation.ts
+import { ToolResult, Message as Message3 } from "@dxos/artifact";
+import { createStatic, ObjectId as ObjectId2 } from "@dxos/echo-schema";
+import { invariant as invariant3 } from "@dxos/invariant";
+import { log as log2 } from "@dxos/log";
+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 () => {
+    log2("llm generate", {
+      tools: params.tools
+    }, {
+      F: __dxlog_file3,
+      L: 43,
+      S: void 0,
+      C: (f, a) => f(...a)
+    });
+    const beginTs = Date.now();
+    const stream = await params.client.generate({
+      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);
+    }
+    const messages = await stream.complete();
+    const message = messages.at(-1);
+    log2("llm result", {
+      time: Date.now() - beginTs,
+      message
+    }, {
+      F: __dxlog_file3,
+      L: 60,
+      S: void 0,
+      C: (f, a) => f(...a)
+    });
+    invariant3(message, void 0, {
+      F: __dxlog_file3,
+      L: 61,
+      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");
+      invariant3(toolCalls.length === 1, void 0, {
+        F: __dxlog_file3,
+        L: 72,
+        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}`);
+      }
+      invariant3(tool.execute, void 0, {
+        F: __dxlog_file3,
+        L: 79,
+        S: void 0,
+        A: [
+          "tool.execute",
+          ""
+        ]
+      });
+      const toolResult = await tool.execute(toolCall.input);
+      switch (toolResult.kind) {
+        case "error": {
+          log2.warn("tool error", {
+            message: toolResult.message
+          }, {
+            F: __dxlog_file3,
+            L: 83,
+            S: void 0,
+            C: (f, a) => f(...a)
+          });
+          const resultMessage = {
+            id: ObjectId2.random(),
+            spaceId: params.spaceId,
+            threadId: params.threadId,
+            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": {
+          log2("tool success", {
+            result: toolResult.result
+          }, {
+            F: __dxlog_file3,
+            L: 105,
+            S: void 0,
+            C: (f, a) => f(...a)
+          });
+          const resultMessage = {
+            id: ObjectId2.random(),
+            spaceId: params.spaceId,
+            threadId: params.threadId,
+            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": {
+          log2("tool break", {
+            result: toolResult.result
+          }, {
+            F: __dxlog_file3,
+            L: 126,
+            S: void 0,
+            C: (f, a) => f(...a)
+          });
+          conversationResult = toolResult.result;
+          return false;
+        }
+      }
+    }
+    return false;
+  };
+  while (await generate()) {
+  }
+  return {
+    result: conversationResult,
+    history
+  };
+};
+var isToolUse = (message) => message.content.at(-1)?.type === "tool_use";
+var runTools = async ({ message, tools, extensions }) => {
+  const toolCalls = message.content.filter((block) => block.type === "tool_use");
+  invariant3(toolCalls.length === 1, void 0, {
+    F: __dxlog_file3,
+    L: 164,
+    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 {
+    invariant3(tool.execute, void 0, {
+      F: __dxlog_file3,
+      L: 173,
+      S: void 0,
+      A: [
+        "tool.execute",
+        ""
+      ]
+    });
+    toolResult = await tool.execute(toolCall.input, {
+      extensions
+    });
+  } catch (error) {
+    log2("tool error", {
+      error
+    }, {
+      F: __dxlog_file3,
+      L: 176,
+      S: void 0,
+      C: (f, a) => f(...a)
+    });
+    toolResult = ToolResult.Error(error.message);
+  }
+  switch (toolResult.kind) {
+    case "error": {
+      log2("tool error", {
+        message: toolResult.message
+      }, {
+        F: __dxlog_file3,
+        L: 182,
+        S: void 0,
+        C: (f, a) => f(...a)
+      });
+      const resultMessage = createStatic(Message3, {
+        role: "user",
+        content: [
+          {
+            type: "tool_result",
+            toolUseId: toolCall.id,
+            content: toolResult.message,
+            isError: true
+          }
+        ]
+      });
+      return {
+        type: "continue",
+        message: resultMessage
+      };
+    }
+    case "success": {
+      log2("tool success", {
+        result: toolResult.result
+      }, {
+        F: __dxlog_file3,
+        L: 203,
+        S: void 0,
+        C: (f, a) => f(...a)
+      });
+      const resultMessage = createStatic(Message3, {
+        role: "user",
+        content: [
+          {
+            type: "tool_result",
+            toolUseId: toolCall.id,
+            content: typeof toolResult.result === "string" ? toolResult.result : JSON.stringify(toolResult.result)
+          }
+        ]
+      });
+      return {
+        type: "continue",
+        message: resultMessage
+      };
+    }
+    case "break": {
+      log2("tool break", {
+        result: toolResult.result
+      }, {
+        F: __dxlog_file3,
+        L: 222,
+        S: void 0,
+        C: (f, a) => f(...a)
+      });
+      return {
+        type: "break",
+        result: toolResult.result
+      };
+    }
+  }
+};
 export {
   AIServiceClientImpl,
   EchoDataSource,
   GenerateRequest,
   GenerationStream,
-  ImageSource,
   LLMModel,
-  LLMTool,
-  LLMToolDefinition,
-  LLMToolResult,
-  Message,
-  MessageContentBlock,
-  MessageFromDb,
-  MessageImageContentBlock,
-  MessageRole,
-  MessageTextContentBlock,
-  MessageToolResultContentBlock,
-  MessageToolUseContentBlock,
-  ObjectId,
-  ObjectIdBrand,
-  Space,
-  SpaceIdSchema,
-  Thread,
+  StreamingParser,
   ToolTypes,
-  createUserMessage,
-  defineTool,
+  createArtifactElement,
   executeQuery,
   executeQueryPlan,
   formatInferredRelationshipLabel,
   formatJsonSchemaForLLM,
   formatNodeLabel,
   isReference,
+  isToolUse,
   parseCypherQuery,
-  runLLM
+  runLLM,
+  runTools
 };
 //# sourceMappingURL=index.mjs.map