@dxos/assistant 0.7.4-df37807 → 0.7.5-feature-compute.4d9d99a

package/dist/lib/node/{chunk-G6CGWEMO.cjs → chunk-SHOSAIOD.cjs}

@@ -16,12 +16,13 @@ var __copyProps = (to, from, except, desc) => {
   return to;
 };
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-var chunk_G6CGWEMO_exports = {};
-__export(chunk_G6CGWEMO_exports, {
+var chunk_SHOSAIOD_exports = {};
+__export(chunk_SHOSAIOD_exports, {
   AIServiceClientImpl: () => AIServiceClientImpl,
   EchoDataSource: () => EchoDataSource,
   GenerateRequest: () => GenerateRequest,
   GenerationStream: () => GenerationStream,
+  ImageContentBlock: () => ImageContentBlock,
   ImageSource: () => ImageSource,
   LLMModel: () => LLMModel,
   LLMTool: () => LLMTool,
@@ -29,18 +30,15 @@ __export(chunk_G6CGWEMO_exports, {
   LLMToolResult: () => LLMToolResult,
   Message: () => Message,
   MessageContentBlock: () => MessageContentBlock,
-  MessageFromDb: () => MessageFromDb,
-  MessageImageContentBlock: () => MessageImageContentBlock,
   MessageRole: () => MessageRole,
-  MessageTextContentBlock: () => MessageTextContentBlock,
-  MessageToolResultContentBlock: () => MessageToolResultContentBlock,
-  MessageToolUseContentBlock: () => MessageToolUseContentBlock,
-  ObjectId: () => ObjectId,
-  ObjectIdBrand: () => ObjectIdBrand,
   Space: () => Space,
   SpaceIdSchema: () => SpaceIdSchema,
+  TagContentBlock: () => TagContentBlock,
+  TextContentBlock: () => TextContentBlock,
   Thread: () => Thread,
+  ToolResultContentBlock: () => ToolResultContentBlock,
   ToolTypes: () => ToolTypes,
+  ToolUseContentBlock: () => ToolUseContentBlock,
   createUserMessage: () => createUserMessage,
   defineTool: () => defineTool,
   executeQuery: () => executeQuery,
@@ -49,152 +47,41 @@ __export(chunk_G6CGWEMO_exports, {
   formatJsonSchemaForLLM: () => formatJsonSchemaForLLM,
   formatNodeLabel: () => formatNodeLabel,
   isReference: () => isReference,
+  isToolUse: () => isToolUse,
   parseCypherQuery: () => parseCypherQuery,
   runLLM: () => runLLM,
+  runTools: () => runTools,
   trim: () => trim
 });
-module.exports = __toCommonJS(chunk_G6CGWEMO_exports);
+module.exports = __toCommonJS(chunk_SHOSAIOD_exports);
+var import_async = require("@dxos/async");
+var import_invariant = require("@dxos/invariant");
+var import_log = require("@dxos/log");
 var import_schema = require("@effect/schema");
-var import_ulidx = require("ulidx");
 var import_echo_schema = require("@dxos/echo-schema");
 var import_keys = require("@dxos/keys");
 var import_schema2 = require("@effect/schema");
-var import_async = require("@dxos/async");
-var import_invariant = require("@dxos/invariant");
-var import_parjs = require("#parjs");
-var import_log = require("@dxos/log");
-var import_util = require("@dxos/util");
+var import_echo_schema2 = require("@dxos/echo-schema");
+var import_schema3 = require("@effect/schema");
 var import_invariant2 = require("@dxos/invariant");
+var import_parjs = require("#parjs");
 var import_log2 = require("@dxos/log");
+var import_util = require("@dxos/util");
+var import_invariant3 = require("@dxos/invariant");
+var import_log3 = require("@dxos/log");
 var import_debug = require("@dxos/debug");
-var import_echo_schema2 = require("@dxos/echo-schema");
+var import_echo_schema3 = require("@dxos/echo-schema");
 var import_debug2 = require("@dxos/debug");
 var import_echo_db = require("@dxos/echo-db");
-var import_echo_schema3 = require("@dxos/echo-schema");
+var import_echo_schema4 = require("@dxos/echo-schema");
 var import_live_object = require("@dxos/live-object");
-var import_log3 = require("@dxos/log");
-var import_invariant3 = require("@dxos/invariant");
 var import_log4 = require("@dxos/log");
-var import_schema3 = require("@effect/schema");
-var import_echo_schema4 = require("@dxos/echo-schema");
+var import_schema4 = require("@effect/schema");
 var import_echo_schema5 = require("@dxos/echo-schema");
 var import_log5 = require("@dxos/log");
-var SpaceIdSchema = import_schema.Schema.String.pipe(import_schema.Schema.filter(import_keys.SpaceId.isValid));
-var ObjectIdBrand = Symbol("@dxos/echo/ObjectId");
-var ObjectIdSchema = import_schema.Schema.ULID.pipe(import_schema.Schema.brand(ObjectIdBrand));
-var ObjectId = class extends ObjectIdSchema {
-  static random() {
-    return (0, import_ulidx.ulid)();
-  }
-};
-var Space = import_schema.Schema.Struct({
-  id: SpaceIdSchema
-});
-var Thread = import_schema.Schema.Struct({
-  id: ObjectId,
-  spaceId: SpaceIdSchema
-});
-var MessageRole = import_schema.Schema.String.pipe(import_schema.Schema.filter((role) => role === "user" || role === "assistant"));
-var MessageTextContentBlock = import_schema.Schema.Struct({
-  type: import_schema.Schema.Literal("text"),
-  text: import_schema.Schema.String
-}).pipe(import_schema.Schema.mutable);
-var ImageSource = import_schema.Schema.Struct({
-  type: import_schema.Schema.Literal("base64"),
-  media_type: import_schema.Schema.Literal("image/jpeg", "image/png", "image/gif", "image/webp"),
-  data: import_schema.Schema.String
-}).pipe(import_schema.Schema.mutable);
-var MessageImageContentBlock = import_schema.Schema.Struct({
-  type: import_schema.Schema.Literal("image"),
-  id: import_schema.Schema.optional(import_schema.Schema.String),
-  source: import_schema.Schema.optional(ImageSource)
-}).pipe(import_schema.Schema.mutable);
-var MessageToolUseContentBlock = import_schema.Schema.Struct({
-  type: import_schema.Schema.Literal("tool_use"),
-  /**
-  * Opaque service-defined ID of the tool invocation.
-  * Used to match the tool result block.
-  */
-  id: import_schema.Schema.String,
-  /**
-  * Tool name.
-  */
-  name: import_schema.Schema.String,
-  input: import_schema.Schema.Unknown,
-  /**
-  * Used to accumulate the partial tool input JSON in streaming mode.
-  */
-  inputJson: import_schema.Schema.optional(import_schema.Schema.String)
-}).pipe(import_schema.Schema.mutable);
-var MessageToolResultContentBlock = import_schema.Schema.Struct({
-  type: import_schema.Schema.Literal("tool_result"),
-  toolUseId: import_schema.Schema.String,
-  content: import_schema.Schema.String,
-  isError: import_schema.Schema.optional(import_schema.Schema.Boolean)
-});
-var MessageContentBlock = import_schema.Schema.Union(MessageTextContentBlock, MessageImageContentBlock, MessageToolUseContentBlock, MessageToolResultContentBlock);
-var Message = import_schema.Schema.Struct({
-  id: ObjectId,
-  threadId: import_schema.Schema.optional(ObjectId),
-  spaceId: import_schema.Schema.optional(SpaceIdSchema),
-  role: MessageRole,
-  content: import_schema.Schema.Array(MessageContentBlock).pipe(import_schema.Schema.mutable),
-  /**
-  * ID of the message from the foreign provider.
-  */
-  // TODO(dmaretskyi): Should be in meta/keys.
-  foreignId: import_schema.Schema.optional(import_schema.Schema.String)
-});
-var MessageFromDb = import_schema.Schema.Struct({
-  ...Message.fields,
-  content: import_schema.Schema.propertySignature(Message.fields.content.pipe((schema) => import_schema.Schema.parseJson(schema))).pipe(import_schema.Schema.fromKey("contentJson"))
-});
-var LLMModel = import_schema.Schema.Literal("@hf/nousresearch/hermes-2-pro-mistral-7b", "@anthropic/claude-3-5-sonnet-20241022", "@anthropic/claude-3-5-haiku-20241022", "@ollama/llama-3-2-3b", "@ollama/llama-3-1-nemotron-70b-instruct", "@ollama/llama-3-1-nemotron-mini-4b-instruct");
-var ToolTypes = Object.freeze({
-  // TODO(dmaretskyi): Not implemented yet.
-  // DatabaseQuery: 'database_query',
-  TextToImage: "text_to_image"
-});
-var LLMTool = import_schema.Schema.Struct({
-  name: import_schema.Schema.String,
-  /**
-  * If the tool is implemented by the service a type should be provided.
-  * For user-implemented tools, this field should be omitted.
-  * See {@link ToolTypes} for the list of supported types.
-  */
-  type: import_schema.Schema.optional(import_schema.Schema.String),
-  /**
-  * Required for user-implemented tools.
-  */
-  description: import_schema.Schema.optional(import_schema.Schema.String),
-  /**
-  * Input schema for the tool in the JSON Schema format.
-  * Required for user-implemented tools.
-  */
-  parameters: import_schema.Schema.optional(import_echo_schema.JsonSchemaType),
-  /**
-  * Tool-specific options.
-  */
-  options: import_schema.Schema.optional(import_schema.Schema.Any),
-  /**
-  * Javascript function to execute the tool.
-  */
-  execute: import_schema.Schema.optional(import_schema.Schema.Any)
-});
-var GenerateRequest = import_schema.Schema.Struct({
-  model: LLMModel,
-  spaceId: SpaceIdSchema,
-  threadId: ObjectId,
-  /**
-  * System instructions to the LLM.
-  */
-  // TODO(dmaretskyi): Should this be part of the thread?.
-  systemPrompt: import_schema.Schema.optional(import_schema.Schema.String),
-  /**
-  * Tools available for the LLM.
-  */
-  tools: import_schema.Schema.Array(LLMTool).pipe(import_schema.Schema.mutable)
-});
+var import_echo_schema6 = require("@dxos/echo-schema");
+var import_invariant4 = require("@dxos/invariant");
+var import_log6 = require("@dxos/log");
 async function* iterSSEMessages(response, controller) {
   if (!response.body) {
     controller.abort();
@@ -429,83 +316,44 @@ var LineDecoder = class _LineDecoder {
     return lines;
   }
 };
-var __dxlog_file = "/Users/yarolegovich/Documents/Workspace/dxos/packages/experimental/assistant/src/ai-service/client.ts";
-var AIServiceClientImpl = class {
-  constructor({ endpoint }) {
-    this._endpoint = endpoint;
-  }
-  async getSpace(spaceId) {
-    throw new Error("Not implemented");
-  }
-  async getThread(spaceId, threadId) {
-    throw new Error("Not implemented");
-  }
-  async getMessagesInThread(spaceId, threadId) {
-    const res = await fetch(`${this._endpoint}/space/${spaceId}/thread/${threadId}/message`);
-    return import_schema2.Schema.decodePromise(import_schema2.Schema.Array(Message).pipe(import_schema2.Schema.mutable))((await res.json()).data.results);
-  }
-  async insertMessages(messages) {
-    const res = await fetch(`${this._endpoint}/message`, {
-      method: "POST",
-      headers: {
-        "Content-Type": "application/json"
-      },
-      body: JSON.stringify({
-        messages: await import_schema2.Schema.encodePromise(import_schema2.Schema.Array(Message))(messages)
-      })
-    });
-    if (!res.ok) {
-      throw new Error(await res.text());
-    }
-  }
-  // TODO(dmaretskyi): Accept partial message schema.
-  async updateMessage(spaceId, threadId, messageId, message) {
-    throw new Error("Not implemented");
-  }
-  async generate(request) {
-    const response = await fetch(`${this._endpoint}/generate`, {
-      method: "POST",
-      headers: {
-        "Content-Type": "application/json"
-      },
-      body: JSON.stringify(request)
-    });
-    (0, import_invariant.invariant)(response.body instanceof ReadableStream, void 0, {
-      F: __dxlog_file,
-      L: 81,
-      S: this,
-      A: [
-        "response.body instanceof ReadableStream",
-        ""
-      ]
-    });
-    return GenerationStream.fromSSEResponse({
-      spaceId: request.spaceId,
-      threadId: request.threadId
-    }, response);
-  }
-};
+var __dxlog_file = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/ai-service/stream.ts";
 var GenerationStream = class _GenerationStream {
-  static fromSSEResponse(params, response) {
-    const controller = new AbortController();
+  /**
+  * 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 (e) {
-            console.error("Could not parse message into JSON:", sse.data);
-            console.error("From chunk:", sse.raw);
-            throw e;
+          } catch (err) {
+            import_log.log.error("could not parse message into JSON:", {
+              data: sse.data,
+              raw: sse.raw
+            }, {
+              F: __dxlog_file,
+              L: 30,
+              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 (e) {
-            console.error("Could not parse message into JSON:", sse.data);
-            console.error("From chunk:", sse.raw);
-            throw e;
+          } catch (err) {
+            import_log.log.error("could not parse message into JSON:", {
+              data: sse.data,
+              raw: sse.raw
+            }, {
+              F: __dxlog_file,
+              L: 46,
+              S: this,
+              C: (f, a) => f(...a)
+            });
+            throw err;
           }
         }
         if (sse.event === "ping") {
@@ -516,22 +364,20 @@ var GenerationStream = class _GenerationStream {
         }
       }
     };
-    return new _GenerationStream(params, iterator);
+    return new _GenerationStream(controller, iterator, params);
   }
-  constructor(_params, _getIterator) {
-    this._params = _params;
+  constructor(_controller, _getIterator, _params) {
+    this._controller = _controller;
     this._getIterator = _getIterator;
+    this._params = _params;
+    this._done = new import_async.Trigger();
     this._previousMessages = [];
     this._accumulatedMessage = void 0;
-    this._done = new import_async.Trigger();
     this._iterator = void 0;
   }
   [Symbol.asyncIterator]() {
     return this._createIterator();
   }
-  cancel() {
-    throw new Error("Not implemented");
-  }
   get accumulatedMessages() {
     return [
       ...this._previousMessages,
@@ -541,6 +387,12 @@ var GenerationStream = class _GenerationStream {
     ];
   }
   /**
+  * Aborts the stream.
+  */
+  abort() {
+    this._controller.abort();
+  }
+  /**
   * Returns the complete message list generated by the AI service.
   */
   async complete() {
@@ -549,12 +401,20 @@ var GenerationStream = class _GenerationStream {
   }
   _createIterator() {
     const self = this;
-    return this._iterator ??= async function* () {
-      for await (const event of self._getIterator()) {
-        self._processEvent(event);
-        yield event;
-      }
-    }();
+    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) {
@@ -585,7 +445,7 @@ var GenerationStream = class _GenerationStream {
       case "content_block_start": {
         (0, import_invariant.invariant)(this._accumulatedMessage, void 0, {
           F: __dxlog_file,
-          L: 212,
+          L: 148,
           S: this,
           A: [
             "this._accumulatedMessage",
@@ -598,7 +458,7 @@ var GenerationStream = class _GenerationStream {
       case "content_block_delta": {
         (0, import_invariant.invariant)(this._accumulatedMessage, void 0, {
           F: __dxlog_file,
-          L: 217,
+          L: 154,
           S: this,
           A: [
             "this._accumulatedMessage",
@@ -617,7 +477,7 @@ var GenerationStream = class _GenerationStream {
       case "content_block_stop": {
         (0, import_invariant.invariant)(this._accumulatedMessage, void 0, {
           F: __dxlog_file,
-          L: 233,
+          L: 171,
           S: this,
           A: [
             "this._accumulatedMessage",
@@ -626,12 +486,194 @@ var GenerationStream = class _GenerationStream {
         });
         const lastBlock = this._accumulatedMessage.content.at(-1);
         if (lastBlock?.type === "tool_use") {
-          lastBlock.input = JSON.parse(lastBlock.inputJson ?? "");
+          lastBlock.input = lastBlock.inputJson == null || lastBlock.inputJson === "" ? {} : JSON.parse(lastBlock.inputJson ?? "");
         }
         break;
       }
     }
-    this._done.wake();
+  }
+};
+var SpaceIdSchema = import_schema.Schema.String.pipe(import_schema.Schema.filter(import_keys.SpaceId.isValid));
+var Space = import_schema.Schema.Struct({
+  id: SpaceIdSchema
+});
+var Thread = import_schema.Schema.Struct({
+  id: import_echo_schema.ObjectId,
+  spaceId: SpaceIdSchema
+});
+var MessageRole = import_schema.Schema.String.pipe(import_schema.Schema.filter((role) => role === "user" || role === "assistant"));
+var TagContentBlock = import_schema.Schema.Struct({
+  type: import_schema.Schema.Literal("tag"),
+  tag: import_schema.Schema.String,
+  text: import_schema.Schema.String
+}).pipe(import_schema.Schema.mutable);
+var TextContentBlock = import_schema.Schema.Struct({
+  type: import_schema.Schema.Literal("text"),
+  text: import_schema.Schema.String
+}).pipe(import_schema.Schema.mutable);
+var ImageSource = import_schema.Schema.Struct({
+  type: import_schema.Schema.Literal("base64"),
+  media_type: import_schema.Schema.Literal("image/jpeg", "image/png", "image/gif", "image/webp"),
+  data: import_schema.Schema.String
+}).pipe(import_schema.Schema.mutable);
+var ImageContentBlock = import_schema.Schema.Struct({
+  type: import_schema.Schema.Literal("image"),
+  id: import_schema.Schema.optional(import_schema.Schema.String),
+  source: import_schema.Schema.optional(ImageSource)
+}).pipe(import_schema.Schema.mutable);
+var ToolUseContentBlock = import_schema.Schema.Struct({
+  type: import_schema.Schema.Literal("tool_use"),
+  /**
+  * Opaque service-defined ID of the tool invocation.
+  * Used to match the tool result block.
+  */
+  id: import_schema.Schema.String,
+  /**
+  * Tool name.
+  */
+  name: import_schema.Schema.String,
+  input: import_schema.Schema.Unknown,
+  /**
+  * Used to accumulate the partial tool input JSON in streaming mode.
+  */
+  // TODO(burdon): Remove?
+  inputJson: import_schema.Schema.optional(import_schema.Schema.String)
+}).pipe(import_schema.Schema.mutable);
+var ToolResultContentBlock = import_schema.Schema.Struct({
+  type: import_schema.Schema.Literal("tool_result"),
+  toolUseId: import_schema.Schema.String,
+  content: import_schema.Schema.String,
+  isError: import_schema.Schema.optional(import_schema.Schema.Boolean)
+});
+var MessageContentBlock = import_schema.Schema.Union(TagContentBlock, TextContentBlock, ImageContentBlock, ToolUseContentBlock, ToolResultContentBlock);
+var MessageSchema = import_schema.Schema.Struct({
+  id: import_echo_schema.ObjectId,
+  threadId: import_schema.Schema.optional(import_echo_schema.ObjectId),
+  spaceId: import_schema.Schema.optional(SpaceIdSchema),
+  /**
+  * ID of the message from the foreign provider.
+  */
+  // TODO(dmaretskyi): Should be in meta/keys.
+  foreignId: import_schema.Schema.optional(import_schema.Schema.String),
+  // TODO(dmaretskyi): Figure out how to deal with those.
+  // created: S.optional(S.DateFromString),
+  // updated: S.optional(S.DateFromString),
+  role: MessageRole,
+  // TODO(burdon): Rename blocks?
+  content: import_schema.Schema.Array(MessageContentBlock).pipe(import_schema.Schema.mutable)
+});
+var Message = MessageSchema.pipe((0, import_echo_schema.EchoObject)("dxos.org/type/Message", "0.1.0"));
+var LLMModel = import_schema2.Schema.Literal("@hf/nousresearch/hermes-2-pro-mistral-7b", "@anthropic/claude-3-5-sonnet-20241022", "@anthropic/claude-3-5-haiku-20241022", "@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 LLMTool = import_schema2.Schema.Struct({
+  name: import_schema2.Schema.String,
+  /**
+  * If the tool is implemented by the service a type should be provided.
+  * For user-implemented tools, this field should be omitted.
+  * See {@link ToolTypes} for the list of supported types.
+  */
+  type: import_schema2.Schema.optional(import_schema2.Schema.String),
+  /**
+  * Required for user-implemented tools.
+  */
+  description: import_schema2.Schema.optional(import_schema2.Schema.String),
+  /**
+  * Input schema for the tool in the JSON Schema format.
+  * Required for user-implemented tools.
+  */
+  parameters: import_schema2.Schema.optional(import_echo_schema2.JsonSchemaType),
+  /**
+  * Tool-specific options.
+  */
+  options: import_schema2.Schema.optional(import_schema2.Schema.Any),
+  /**
+  * Javascript function to execute the tool.
+  */
+  execute: import_schema2.Schema.optional(import_schema2.Schema.Any)
+});
+var GenerateRequest = import_schema2.Schema.Struct({
+  model: LLMModel,
+  spaceId: import_schema2.Schema.optional(SpaceIdSchema),
+  threadId: import_schema2.Schema.optional(import_echo_schema2.ObjectId),
+  /**
+  * History of messages to include in the context window.
+  */
+  history: import_schema2.Schema.optional(import_schema2.Schema.Array(Message)),
+  /**
+  * System instructions to the LLM.
+  */
+  // TODO(dmaretskyi): Should this be part of the thread?.
+  systemPrompt: import_schema2.Schema.optional(import_schema2.Schema.String),
+  /**
+  * Tools available for the LLM.
+  */
+  tools: import_schema2.Schema.optional(import_schema2.Schema.Array(LLMTool).pipe(import_schema2.Schema.mutable))
+});
+var __dxlog_file2 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/ai-service/client.ts";
+var AIServiceClientImpl = class {
+  constructor({ endpoint }) {
+    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 import_schema3.Schema.decodePromise(import_schema3.Schema.Array(Message).pipe(import_schema3.Schema.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 import_schema3.Schema.encodePromise(import_schema3.Schema.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
+    });
+    (0, import_invariant2.invariant)(response.body instanceof ReadableStream, void 0, {
+      F: __dxlog_file2,
+      L: 85,
+      S: this,
+      A: [
+        "response.body instanceof ReadableStream",
+        ""
+      ]
+    });
+    return GenerationStream.fromSSEResponse({
+      spaceId: request.spaceId,
+      threadId: request.threadId
+    }, response, controller);
   }
 };
 var ws = import_parjs.parjs.whitespace();
@@ -716,7 +758,7 @@ var parseCypherQuery = (query) => {
     throw new Error(`Failed to parse query: ${parseResult}`);
   }
 };
-var __dxlog_file2 = "/Users/yarolegovich/Documents/Workspace/dxos/packages/experimental/assistant/src/cypher/execution-plan.ts";
+var __dxlog_file3 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/cypher/execution-plan.ts";
 var createExecutionPlan = (query) => {
   return new ExecutionPlanBuilder().build(query);
 };
@@ -728,10 +770,10 @@ var ExecutionPlanBuilder = class {
     return `anon_${this.counter++}`;
   }
   build(query) {
-    (0, import_log2.log)("Creating execution plan for query", {
+    (0, import_log3.log)("Creating execution plan for query", {
       query
     }, {
-      F: __dxlog_file2,
+      F: __dxlog_file3,
       L: 76,
       S: this,
       C: (f, a) => f(...a)
@@ -748,8 +790,8 @@ var ExecutionPlanBuilder = class {
     };
   }
   createStepsForMatchClause(match) {
-    (0, import_invariant2.invariant)(match.pattern.segments.length > 0, void 0, {
-      F: __dxlog_file2,
+    (0, import_invariant3.invariant)(match.pattern.segments.length > 0, void 0, {
+      F: __dxlog_file3,
       L: 90,
       S: this,
       A: [
@@ -758,8 +800,8 @@ var ExecutionPlanBuilder = class {
       ]
     });
     const firstSegment = match.pattern.segments[0];
-    (0, import_invariant2.invariant)(firstSegment.astKind === "NodePattern", void 0, {
-      F: __dxlog_file2,
+    (0, import_invariant3.invariant)(firstSegment.astKind === "NodePattern", void 0, {
+      F: __dxlog_file3,
       L: 92,
       S: this,
       A: [
@@ -786,8 +828,8 @@ var ExecutionPlanBuilder = class {
     }
     for (let i = 1; i < match.pattern.segments.length; i++) {
       const relationshipSegment = match.pattern.segments[i++];
-      (0, import_invariant2.invariant)(relationshipSegment.astKind === "RelationshipPattern", void 0, {
-        F: __dxlog_file2,
+      (0, import_invariant3.invariant)(relationshipSegment.astKind === "RelationshipPattern", void 0, {
+        F: __dxlog_file3,
         L: 115,
         S: this,
         A: [
@@ -796,8 +838,8 @@ var ExecutionPlanBuilder = class {
         ]
       });
       const nodeSegment = match.pattern.segments[i];
-      (0, import_invariant2.invariant)(nodeSegment.astKind === "NodePattern", void 0, {
-        F: __dxlog_file2,
+      (0, import_invariant3.invariant)(nodeSegment.astKind === "NodePattern", void 0, {
+        F: __dxlog_file3,
         L: 117,
         S: this,
         A: [
@@ -878,15 +920,15 @@ var getDirection = (left, right) => {
   }
   throw new Error(`Invalid direction: ()${left}[]${right}()`);
 };
-var __dxlog_file3 = "/Users/yarolegovich/Documents/Workspace/dxos/packages/experimental/assistant/src/cypher/query-executor.ts";
+var __dxlog_file4 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/cypher/query-executor.ts";
 var executeQueryPlan = async (dataSource, plan) => {
   let buffer = [];
   const results = [];
   for (const step of plan.steps) {
-    (0, import_log.log)("begin step", {
+    (0, import_log2.log)("begin step", {
       step
     }, {
-      F: __dxlog_file3,
+      F: __dxlog_file4,
       L: 41,
       S: void 0,
       C: (f, a) => f(...a)
@@ -972,11 +1014,11 @@ var executeQueryPlan = async (dataSource, plan) => {
         break;
       }
     }
-    (0, import_log.log)("finish step", {
+    (0, import_log2.log)("finish step", {
       buffer: buffer.map((tuple) => (0, import_util.mapValues)(tuple, (x) => x.id)),
       results
     }, {
-      F: __dxlog_file3,
+      F: __dxlog_file4,
       L: 118,
       S: void 0,
       C: (f, a) => f(...a)
@@ -1059,14 +1101,14 @@ var formatInferredRelationshipLabel = (typename, property2) => {
   const nodeLabel = formatNodeLabel(typename);
   return `${nodeLabel}_${property2}`.toUpperCase();
 };
-var isReference = (schema) => schema.$id === import_echo_schema2.JSON_SCHEMA_ECHO_REF_ID || schema.type === "array" && schema.items?.$id === import_echo_schema2.JSON_SCHEMA_ECHO_REF_ID;
+var isReference = (schema) => schema.$id === import_echo_schema3.JSON_SCHEMA_ECHO_REF_ID || schema.type === "array" && schema.items?.$id === import_echo_schema3.JSON_SCHEMA_ECHO_REF_ID;
 function _ts_decorate(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
   else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
   return c > 3 && r && Object.defineProperty(target, key, r), r;
 }
-var __dxlog_file4 = "/Users/yarolegovich/Documents/Workspace/dxos/packages/experimental/assistant/src/cypher/echo-data-source.ts";
+var __dxlog_file5 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/cypher/echo-data-source.ts";
 var EchoDataSource = class {
   constructor(_db) {
     this._db = _db;
@@ -1076,7 +1118,7 @@ var EchoDataSource = class {
       const { objects: objects2 } = await this._db.query().run();
       return objects2.map(this._objectToNode);
     }
-    const schema = (await this._getAllSchema()).find((s) => (0, import_echo_schema3.getSchemaTypename)(s)?.endsWith(label2));
+    const schema = (await this._getAllSchema()).find((s) => (0, import_echo_schema4.getSchemaTypename)(s)?.endsWith(label2));
     if (!schema) {
       return [];
     }
@@ -1084,8 +1126,8 @@ var EchoDataSource = class {
     return objects.map(this._objectToNode);
   }
   async getRelationships({ label: label2 }) {
-    const syntheticRefRelationships = (await this._getAllSchema()).filter((schema) => (0, import_echo_schema3.getSchemaTypename)(schema) !== import_echo_schema3.StoredSchema.typename).flatMap((schema) => {
-      const jsonSchema = (0, import_echo_schema3.toJsonSchema)(schema);
+    const syntheticRefRelationships = (await this._getAllSchema()).filter((schema) => (0, import_echo_schema4.getSchemaTypename)(schema) !== import_echo_schema4.StoredSchema.typename).flatMap((schema) => {
+      const jsonSchema = (0, import_echo_schema4.toJsonSchema)(schema);
       const relationships = Object.entries(jsonSchema.properties).filter(([key, prop]) => isReference(prop));
       return relationships.map(([key, value]) => {
         const relationshipLabel = formatInferredRelationshipLabel(jsonSchema.typename ?? (0, import_debug2.raise)(new Error("Schema must have typename")), key);
@@ -1107,15 +1149,15 @@ var EchoDataSource = class {
     return [
       ...await this._db.schemaRegistry.query().run(),
       // TODO(dmaretskyi): Remove once we can serialize recursive schema.
-      ...this._db.graph.schemaRegistry.schemas.filter((s) => (0, import_echo_schema3.getSchemaTypename)(s)?.startsWith("example.org"))
-    ].filter((schema) => (0, import_echo_schema3.getSchemaTypename)(schema) !== import_echo_schema3.StoredSchema.typename);
+      ...this._db.graph.schemaRegistry.schemas.filter((s) => (0, import_echo_schema4.getSchemaTypename)(s)?.startsWith("example.org"))
+    ].filter((schema) => (0, import_echo_schema4.getSchemaTypename)(schema) !== import_echo_schema4.StoredSchema.typename);
   }
   _objectToNode(object) {
     const { id, ...properties2 } = object;
     return {
       id,
       kind: "node",
-      label: formatNodeLabel((0, import_echo_schema3.getSchemaTypename)((0, import_live_object.getSchema)(object))),
+      label: formatNodeLabel((0, import_echo_schema4.getSchemaTypename)((0, import_live_object.getSchema)(object))),
       properties: properties2
     };
   }
@@ -1131,7 +1173,7 @@ var EchoDataSource = class {
       return {
         id: `${object.id}-${prop}-${target.id}`,
         kind: "relationship",
-        label: formatInferredRelationshipLabel((0, import_echo_schema3.getSchemaTypename)((0, import_live_object.getSchema)(object)), prop),
+        label: formatInferredRelationshipLabel((0, import_echo_schema4.getSchemaTypename)((0, import_live_object.getSchema)(object)), prop),
         properties: {},
         source: this._objectToNode(object),
         target: this._objectToNode(target)
@@ -1140,58 +1182,133 @@ var EchoDataSource = class {
   }
 };
 _ts_decorate([
-  import_log3.log.method(void 0, void 0, {
-    F: __dxlog_file4,
+  import_log4.log.method(void 0, void 0, {
+    F: __dxlog_file5,
     L: 17,
     S: void 0,
     C: (f, a) => f(...a)
   })
 ], EchoDataSource.prototype, "getNodes", null);
 _ts_decorate([
-  import_log3.log.method(void 0, void 0, {
-    F: __dxlog_file4,
+  import_log4.log.method(void 0, void 0, {
+    F: __dxlog_file5,
     L: 34,
     S: void 0,
     C: (f, a) => f(...a)
   })
 ], EchoDataSource.prototype, "getRelationships", null);
-var __dxlog_file5 = "/Users/yarolegovich/Documents/Workspace/dxos/packages/experimental/assistant/src/conversation/conversation.ts";
+var __dxlog_file6 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/conversation/types.ts";
+var createUserMessage = (spaceId, threadId, text) => ({
+  id: import_echo_schema5.ObjectId.random(),
+  spaceId,
+  threadId,
+  role: "user",
+  content: [
+    {
+      type: "text",
+      text
+    }
+  ]
+});
+var LLMToolDefinition = import_schema4.Schema.Struct({
+  name: import_schema4.Schema.String,
+  description: import_schema4.Schema.String,
+  parameters: import_echo_schema5.JsonSchemaType,
+  execute: import_schema4.Schema.Any
+});
+var LLMToolResult = Object.freeze({
+  /**
+  * The tool execution was successful.
+  * Gives the result back to the LLM.
+  */
+  Success: (result, extractContentBlocks) => ({
+    kind: "success",
+    result,
+    extractContentBlocks
+  }),
+  Error: (message) => ({
+    kind: "error",
+    message
+  }),
+  /**
+  * Stop the conversation and return the result.
+  */
+  Break: (result) => ({
+    kind: "break",
+    result
+  })
+});
+var defineTool = ({ name, description, schema, execute }) => {
+  return {
+    name,
+    description,
+    parameters: toFunctionParameterSchema(schema),
+    execute: (params, context) => {
+      const sanitized = import_schema4.Schema.decodeSync(schema)(params);
+      return execute(sanitized, context);
+    }
+  };
+};
+var toFunctionParameterSchema = (schema) => {
+  const jsonSchema = (0, import_echo_schema5.toJsonSchema)(schema);
+  (0, import_log5.log)("tool schema", {
+    jsonSchema
+  }, {
+    F: __dxlog_file6,
+    L: 120,
+    S: void 0,
+    C: (f, a) => f(...a)
+  });
+  delete jsonSchema.anyOf;
+  delete jsonSchema.$id;
+  jsonSchema.type = "object";
+  jsonSchema.properties ??= {};
+  for (const key in jsonSchema.properties) {
+    if (typeof jsonSchema.properties[key].description !== "string") {
+      throw new Error(`Property missing description: ${key}`);
+    }
+  }
+  return jsonSchema;
+};
+var __dxlog_file7 = "/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", {
+    (0, import_log6.log)("llm generate", {
       tools: params.tools
     }, {
-      F: __dxlog_file5,
-      L: 42,
+      F: __dxlog_file7,
+      L: 45,
       S: void 0,
       C: (f, a) => f(...a)
     });
     const beginTs = Date.now();
-    const result = await params.client.generate({
+    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 result) {
+    for await (const event of stream) {
       params.logger?.(event);
     }
-    const messages = await result.complete();
+    const messages = await stream.complete();
     const message = messages.at(-1);
-    (0, import_log4.log)("llm result", {
+    (0, import_log6.log)("llm result", {
       time: Date.now() - beginTs,
       message
     }, {
-      F: __dxlog_file5,
-      L: 58,
+      F: __dxlog_file7,
+      L: 62,
       S: void 0,
       C: (f, a) => f(...a)
     });
-    (0, import_invariant3.invariant)(message, void 0, {
-      F: __dxlog_file5,
-      L: 59,
+    (0, import_invariant4.invariant)(message, void 0, {
+      F: __dxlog_file7,
+      L: 63,
       S: void 0,
       A: [
         "message",
@@ -1202,16 +1319,16 @@ var runLLM = async (params) => {
       type: "message",
       message
     });
-    await params.client.insertMessages(messages.map((msg) => ({
-      ...msg,
-      content: msg.content.filter((c) => c.type !== "image")
+    history.push(...messages.map((message2) => ({
+      ...message2,
+      content: message2.content.filter((block) => block.type !== "image")
     })));
-    const isToolUse = message.content.at(-1)?.type === "tool_use";
-    if (isToolUse) {
+    const isToolUse2 = message.content.at(-1)?.type === "tool_use";
+    if (isToolUse2) {
       const toolCalls = message.content.filter((c) => c.type === "tool_use");
-      (0, import_invariant3.invariant)(toolCalls.length === 1, void 0, {
-        F: __dxlog_file5,
-        L: 68,
+      (0, import_invariant4.invariant)(toolCalls.length === 1, void 0, {
+        F: __dxlog_file7,
+        L: 74,
         S: void 0,
         A: [
           "toolCalls.length === 1",
@@ -1219,23 +1336,23 @@ var runLLM = async (params) => {
         ]
       });
       const toolCall = toolCalls[0];
-      const tool = params.tools.find((t) => t.name === toolCall.name);
+      const tool = params.tools.find((tool2) => tool2.name === toolCall.name);
       if (!tool) {
         throw new Error(`Tool not found: ${toolCall.name}`);
       }
       const toolResult = await tool.execute(toolCall.input, {});
       switch (toolResult.kind) {
         case "error": {
-          (0, import_log4.log)("tool error", {
+          import_log6.log.warn("tool error", {
             message: toolResult.message
           }, {
-            F: __dxlog_file5,
-            L: 78,
+            F: __dxlog_file7,
+            L: 84,
             S: void 0,
             C: (f, a) => f(...a)
           });
           const resultMessage = {
-            id: ObjectId.random(),
+            id: import_echo_schema6.ObjectId.random(),
             spaceId: params.spaceId,
             threadId: params.threadId,
             role: "user",
@@ -1252,22 +1369,20 @@ var runLLM = async (params) => {
             type: "message",
             message: resultMessage
           });
-          await params.client.insertMessages([
-            resultMessage
-          ]);
+          history.push(resultMessage);
           return true;
         }
         case "success": {
-          (0, import_log4.log)("tool success", {
+          (0, import_log6.log)("tool success", {
             result: toolResult.result
           }, {
-            F: __dxlog_file5,
-            L: 99,
+            F: __dxlog_file7,
+            L: 106,
             S: void 0,
             C: (f, a) => f(...a)
           });
           const resultMessage = {
-            id: ObjectId.random(),
+            id: import_echo_schema6.ObjectId.random(),
             spaceId: params.spaceId,
             threadId: params.threadId,
             role: "user",
@@ -1283,17 +1398,15 @@ var runLLM = async (params) => {
             type: "message",
             message: resultMessage
           });
-          await params.client.insertMessages([
-            resultMessage
-          ]);
+          history.push(resultMessage);
           return true;
         }
         case "break": {
-          (0, import_log4.log)("tool break", {
+          (0, import_log6.log)("tool break", {
             result: toolResult.result
           }, {
-            F: __dxlog_file5,
-            L: 119,
+            F: __dxlog_file7,
+            L: 127,
             S: void 0,
             C: (f, a) => f(...a)
           });
@@ -1307,76 +1420,108 @@ var runLLM = async (params) => {
   while (await generate()) {
   }
   return {
-    result: conversationResult
+    result: conversationResult,
+    history
   };
 };
-var __dxlog_file6 = "/Users/yarolegovich/Documents/Workspace/dxos/packages/experimental/assistant/src/conversation/types.ts";
-var createUserMessage = (spaceId, threadId, text) => ({
-  id: ObjectId.random(),
-  spaceId,
-  threadId,
-  role: "user",
-  content: [
-    {
-      type: "text",
-      text
-    }
-  ]
-});
-var LLMToolDefinition = import_schema3.Schema.Struct({
-  name: import_schema3.Schema.String,
-  description: import_schema3.Schema.String,
-  parameters: import_echo_schema5.JsonSchemaType,
-  execute: import_schema3.Schema.Any
-});
-var LLMToolResult = Object.freeze({
-  /**
-  * The tool execution was successful.
-  * Gives the result back to the LLM.
-  */
-  Success: (result) => ({
-    kind: "success",
-    result
-  }),
-  Error: (message) => ({
-    kind: "error",
-    message
-  }),
-  /**
-  * Stop the conversation and return the result.
-  */
-  Break: (result) => ({
-    kind: "break",
-    result
-  })
-});
-var defineTool = ({ name, description, schema, execute }) => {
-  return {
-    name,
-    description,
-    parameters: toFunctionParameterSchema(schema),
-    execute: (params, context) => {
-      const sanitized = import_schema3.Schema.decodeSync(schema)(params);
-      return execute(sanitized, context);
-    }
-  };
-};
-var toFunctionParameterSchema = (schema) => {
-  const jsonSchema = (0, import_echo_schema4.toJsonSchema)(schema);
-  (0, import_log5.log)("tool schema", {
-    jsonSchema
-  }, {
-    F: __dxlog_file6,
-    L: 83,
+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");
+  (0, import_invariant4.invariant)(toolCalls.length === 1, void 0, {
+    F: __dxlog_file7,
+    L: 165,
     S: void 0,
-    C: (f, a) => f(...a)
+    A: [
+      "toolCalls.length === 1",
+      ""
+    ]
   });
-  for (const key in jsonSchema.properties) {
-    if (typeof jsonSchema.properties[key].description !== "string") {
-      throw new Error(`Property missing description: ${key}`);
+  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 {
+    toolResult = await tool.execute(toolCall.input, {
+      extensions
+    });
+  } catch (error) {
+    (0, import_log6.log)("tool error", {
+      error
+    }, {
+      F: __dxlog_file7,
+      L: 176,
+      S: void 0,
+      C: (f, a) => f(...a)
+    });
+    toolResult = LLMToolResult.Error(error.message);
+  }
+  switch (toolResult.kind) {
+    case "error": {
+      (0, import_log6.log)("tool error", {
+        message: toolResult.message
+      }, {
+        F: __dxlog_file7,
+        L: 182,
+        S: void 0,
+        C: (f, a) => f(...a)
+      });
+      const resultMessage = (0, import_echo_schema6.createStatic)(Message, {
+        role: "user",
+        content: [
+          {
+            type: "tool_result",
+            toolUseId: toolCall.id,
+            content: toolResult.message,
+            isError: true
+          }
+        ]
+      });
+      return {
+        type: "continue",
+        message: resultMessage
+      };
+    }
+    case "success": {
+      (0, import_log6.log)("tool success", {
+        result: toolResult.result
+      }, {
+        F: __dxlog_file7,
+        L: 203,
+        S: void 0,
+        C: (f, a) => f(...a)
+      });
+      const resultMessage = (0, import_echo_schema6.createStatic)(Message, {
+        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": {
+      (0, import_log6.log)("tool break", {
+        result: toolResult.result
+      }, {
+        F: __dxlog_file7,
+        L: 222,
+        S: void 0,
+        C: (f, a) => f(...a)
+      });
+      return {
+        type: "break",
+        result: toolResult.result
+      };
     }
   }
-  return jsonSchema;
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
@@ -1384,6 +1529,7 @@ var toFunctionParameterSchema = (schema) => {
   EchoDataSource,
   GenerateRequest,
   GenerationStream,
+  ImageContentBlock,
   ImageSource,
   LLMModel,
   LLMTool,
@@ -1391,18 +1537,15 @@ var toFunctionParameterSchema = (schema) => {
   LLMToolResult,
   Message,
   MessageContentBlock,
-  MessageFromDb,
-  MessageImageContentBlock,
   MessageRole,
-  MessageTextContentBlock,
-  MessageToolResultContentBlock,
-  MessageToolUseContentBlock,
-  ObjectId,
-  ObjectIdBrand,
   Space,
   SpaceIdSchema,
+  TagContentBlock,
+  TextContentBlock,
   Thread,
+  ToolResultContentBlock,
   ToolTypes,
+  ToolUseContentBlock,
   createUserMessage,
   defineTool,
   executeQuery,
@@ -1411,8 +1554,10 @@ var toFunctionParameterSchema = (schema) => {
   formatJsonSchemaForLLM,
   formatNodeLabel,
   isReference,
+  isToolUse,
   parseCypherQuery,
   runLLM,
+  runTools,
   trim
 });
-//# sourceMappingURL=chunk-G6CGWEMO.cjs.map
+//# sourceMappingURL=chunk-SHOSAIOD.cjs.map