@dxos/assistant 0.8.2-main.f11618f → 0.8.2-staging.7ac8446

package/dist/lib/node-esm/{chunk-7JI4MDGL.mjs → chunk-DCQJOLSR.mjs}

@@ -361,6 +361,8 @@ var GenerationStreamImpl = class {
 };
 
 // packages/experimental/assistant/src/ai-service/edge-client.ts
+import { Schema as S } from "effect";
+import { Message } from "@dxos/artifact";
 import { assertArgument, invariant } from "@dxos/invariant";
 import { log as log2 } from "@dxos/log";
 var __dxlog_file2 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/ai-service/edge-client.ts";
@@ -368,7 +370,7 @@ var AIServiceEdgeClient = class {
   constructor({ endpoint }) {
     invariant(endpoint, "endpoint is required", {
       F: __dxlog_file2,
-      L: 20,
+      L: 23,
       S: this,
       A: [
         "endpoint",
@@ -378,21 +380,34 @@ var AIServiceEdgeClient = class {
     this._endpoint = endpoint;
   }
   /**
-  * Generate non-streaming response.
+  * @deprecated
   */
-  async exec(request) {
-    throw new Error("Not implemented");
+  // TODO(burdon): Remove.
+  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}]`);
+    }
   }
   /**
   * Process request and open message stream.
   */
-  async execStream(request) {
+  async exec(request) {
     assertArgument(typeof request.model === "string", "model is required");
     log2("requesting", {
       endpoint: this._endpoint
     }, {
       F: __dxlog_file2,
-      L: 36,
+      L: 51,
       S: this,
       C: (f, a) => f(...a)
     });
@@ -409,12 +424,9 @@ var AIServiceEdgeClient = class {
         ...request
       })
     });
-    if (!response.ok) {
-      throw await parseErrorResponse(response);
-    }
     invariant(response.body instanceof ReadableStream, void 0, {
       F: __dxlog_file2,
-      L: 53,
+      L: 64,
       S: this,
       A: [
         "response.body instanceof ReadableStream",
@@ -424,9 +436,6 @@ var AIServiceEdgeClient = class {
     return createGenerationStream(response, controller);
   }
 };
-var parseErrorResponse = async (response) => {
-  throw new Error(`AI Service error: ${response.status} ${response.statusText} ${await response.text()}`);
-};
 
 // packages/experimental/assistant/src/ai-service/message-collector.ts
 var MessageCollector = class {
@@ -495,49 +504,45 @@ function* emitMessageAsEvents(message) {
 }
 
 // packages/experimental/assistant/src/ai-service/types.ts
-import { Schema as S } from "effect";
-import { Tool, Message } from "@dxos/artifact";
+import { Schema as S2 } from "effect";
+import { Tool, Message as Message2, SpaceIdSchema } from "@dxos/artifact";
+import { ObjectId } from "@dxos/echo-schema";
 var createArtifactElement = (id) => `<artifact id=${id} />`;
-var LLMModel = S.Literal(...DEFAULT_EDGE_MODELS, ...DEFAULT_OLLAMA_MODELS);
+var LLMModel = S2.Literal(...DEFAULT_EDGE_MODELS, ...DEFAULT_OLLAMA_MODELS);
 var ToolTypes = Object.freeze({
   // TODO(dmaretskyi): Not implemented yet.
   // DatabaseQuery: 'database_query',
   TextToImage: "text_to_image"
 });
-var GenerateRequest = S.Struct({
+var GenerateRequest = S2.Struct({
+  spaceId: S2.optional(SpaceIdSchema),
+  threadId: S2.optional(ObjectId),
   /**
   * Preferred model or system default.
   */
-  model: S.optional(LLMModel),
+  model: S2.optional(LLMModel),
   /**
   * Tools available for the LLM.
   */
-  tools: S.optional(S.Array(Tool).pipe(S.mutable)),
+  tools: S2.optional(S2.Array(Tool).pipe(S2.mutable)),
   /**
   * System instructions to the LLM.
   */
-  systemPrompt: S.optional(S.String),
+  systemPrompt: S2.optional(S2.String),
   /**
   * History of messages to include in the context window.
   */
   // TODO(burdon): Rename messages.
-  history: S.optional(S.Array(Message)),
+  history: S2.optional(S2.Array(Message2)),
   /**
   * Current request.
   */
-  prompt: S.optional(Message)
-});
-var GenerateResponse = S.Struct({
-  messages: S.Array(Message),
-  /**
-  * Number of tokens used in the response.
-  */
-  tokenCount: S.optional(S.Number)
+  prompt: S2.optional(Message2)
 });
 
 // packages/experimental/assistant/src/conversation/conversation.ts
-import { ToolResult, Message as Message2 } from "@dxos/artifact";
-import { create, ObjectId } from "@dxos/echo-schema";
+import { ToolResult, Message as Message3 } from "@dxos/artifact";
+import { createStatic, ObjectId as ObjectId2 } from "@dxos/echo-schema";
 import { invariant as invariant2 } from "@dxos/invariant";
 import { log as log3 } from "@dxos/log";
 var __dxlog_file3 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/conversation/conversation.ts";
@@ -554,8 +559,10 @@ var runLLM = async (params) => {
       C: (f, a) => f(...a)
     });
     const beginTs = Date.now();
-    const stream = await params.client.execStream({
+    const stream = await params.client.exec({
       model: params.model,
+      spaceId: params.spaceId,
+      threadId: params.threadId,
       history,
       systemPrompt: params.system,
       tools: params.tools
@@ -568,7 +575,7 @@ var runLLM = async (params) => {
     const message = messages.at(-1);
     invariant2(message, void 0, {
       F: __dxlog_file3,
-      L: 61,
+      L: 63,
       S: void 0,
       A: [
         "message",
@@ -580,13 +587,13 @@ var runLLM = async (params) => {
       message
     }, {
       F: __dxlog_file3,
-      L: 63,
+      L: 65,
       S: void 0,
       C: (f, a) => f(...a)
     });
     invariant2(message, void 0, {
       F: __dxlog_file3,
-      L: 64,
+      L: 66,
       S: void 0,
       A: [
         "message",
@@ -606,7 +613,7 @@ var runLLM = async (params) => {
       const toolCalls = message.content.filter((block) => block.type === "tool_use");
       invariant2(toolCalls.length === 1, void 0, {
         F: __dxlog_file3,
-        L: 75,
+        L: 77,
         S: void 0,
         A: [
           "toolCalls.length === 1",
@@ -620,7 +627,7 @@ var runLLM = async (params) => {
       }
       invariant2(tool.execute, void 0, {
         F: __dxlog_file3,
-        L: 82,
+        L: 84,
         S: void 0,
         A: [
           "tool.execute",
@@ -634,12 +641,14 @@ var runLLM = async (params) => {
             message: toolResult.message
           }, {
             F: __dxlog_file3,
-            L: 86,
+            L: 88,
             S: void 0,
             C: (f, a) => f(...a)
           });
           const resultMessage = {
-            id: ObjectId.random(),
+            id: ObjectId2.random(),
+            spaceId: params.spaceId,
+            threadId: params.threadId,
             role: "user",
             content: [
               {
@@ -662,12 +671,14 @@ var runLLM = async (params) => {
             result: toolResult.result
           }, {
             F: __dxlog_file3,
-            L: 106,
+            L: 110,
             S: void 0,
             C: (f, a) => f(...a)
           });
           const resultMessage = {
-            id: ObjectId.random(),
+            id: ObjectId2.random(),
+            spaceId: params.spaceId,
+            threadId: params.threadId,
             role: "user",
             content: [
               {
@@ -689,7 +700,7 @@ var runLLM = async (params) => {
             result: toolResult.result
           }, {
             F: __dxlog_file3,
-            L: 125,
+            L: 131,
             S: void 0,
             C: (f, a) => f(...a)
           });
@@ -711,7 +722,7 @@ var isToolUse = (message, { onlyToolNames } = {}) => {
   const block = message.content.at(-1);
   invariant2(block, void 0, {
     F: __dxlog_file3,
-    L: 146,
+    L: 152,
     S: void 0,
     A: [
       "block",
@@ -724,7 +735,7 @@ var runTools = async ({ message, tools, extensions }) => {
   const toolCalls = message.content.filter((block) => block.type === "tool_use");
   invariant2(toolCalls.length === 1, void 0, {
     F: __dxlog_file3,
-    L: 168,
+    L: 174,
     S: void 0,
     A: [
       "toolCalls.length === 1",
@@ -734,7 +745,7 @@ var runTools = async ({ message, tools, extensions }) => {
   const toolCall = toolCalls[0];
   const tool = tools.find((tool2) => tool2.name === toolCall.name);
   if (!tool) {
-    const resultMessage = create(Message2, {
+    const resultMessage = createStatic(Message3, {
       role: "user",
       content: [
         {
@@ -754,7 +765,7 @@ var runTools = async ({ message, tools, extensions }) => {
   try {
     invariant2(tool.execute, void 0, {
       F: __dxlog_file3,
-      L: 193,
+      L: 199,
       S: void 0,
       A: [
         "tool.execute",
@@ -769,7 +780,7 @@ var runTools = async ({ message, tools, extensions }) => {
       error
     }, {
       F: __dxlog_file3,
-      L: 196,
+      L: 202,
       S: void 0,
       C: (f, a) => f(...a)
     });
@@ -781,11 +792,11 @@ var runTools = async ({ message, tools, extensions }) => {
         message: toolResult.message
       }, {
         F: __dxlog_file3,
-        L: 202,
+        L: 208,
         S: void 0,
         C: (f, a) => f(...a)
       });
-      const resultMessage = create(Message2, {
+      const resultMessage = createStatic(Message3, {
         role: "user",
         content: [
           {
@@ -806,11 +817,11 @@ var runTools = async ({ message, tools, extensions }) => {
         result: toolResult.result
       }, {
         F: __dxlog_file3,
-        L: 223,
+        L: 229,
         S: void 0,
         C: (f, a) => f(...a)
       });
-      const resultMessage = create(Message2, {
+      const resultMessage = createStatic(Message3, {
         role: "user",
         content: [
           {
@@ -831,7 +842,7 @@ var runTools = async ({ message, tools, extensions }) => {
         result: toolResult.result
       }, {
         F: __dxlog_file3,
-        L: 244,
+        L: 250,
         S: void 0,
         C: (f, a) => f(...a)
       });
@@ -844,9 +855,9 @@ var runTools = async ({ message, tools, extensions }) => {
 };
 
 // packages/experimental/assistant/src/ai-service/ollama-client.ts
-import { Schema as S2 } from "effect";
+import { Schema as S3 } from "effect";
 import { defineTool, ToolResult as ToolResult2 } from "@dxos/artifact";
-import { ObjectId as ObjectId2 } from "@dxos/echo-schema";
+import { ObjectId as ObjectId3 } from "@dxos/echo-schema";
 import { invariant as invariant3 } from "@dxos/invariant";
 import { log as log4 } from "@dxos/log";
 var __dxlog_file4 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/ai-service/ollama-client.ts";
@@ -991,14 +1002,14 @@ var OllamaClient = class {
       signal
     });
     if (!response.ok) {
-      throw await parseErrorResponse2(response);
+      throw await parseErrorResponse(response);
     }
     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 = ObjectId2.random();
+      const messageId = ObjectId3.random();
       yield {
         type: "message_start",
         message: {
@@ -1058,7 +1069,7 @@ var OllamaClient = class {
             index: currentBlockIndex,
             content: {
               type: "tool_use",
-              id: ObjectId2.random(),
+              id: ObjectId3.random(),
               name: data.message.tool_calls[0].function.name,
               input: sanitizeToolArguments(data.message.tool_calls[0].function.arguments)
             }
@@ -1082,13 +1093,7 @@ var OllamaClient = class {
       reader.releaseLock();
     }
   }
-  /**
-  * Generate non-streaming response.
-  */
   async exec(request) {
-    throw new Error("Not implemented");
-  }
-  async execStream(request) {
     const controller = new AbortController();
     try {
       return new GenerationStreamImpl(controller, async function* () {
@@ -1150,15 +1155,15 @@ var WELL_KNOWN_TOOLS = {
   [ToolTypes.TextToImage]: defineTool("system", {
     name: "text-to-image",
     description: "Generate an image from a text prompt",
-    schema: S2.Struct({
-      prompt: S2.String.annotations({
+    schema: S3.Struct({
+      prompt: S3.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 = ObjectId2.random();
+      const id = ObjectId3.random();
       return ToolResult2.Success(`The generation was successful. Image ID: ${id} Prompt: ${prompt}`, [
         {
           type: "image",
@@ -1194,12 +1199,12 @@ var OllamaDecoderStream = class extends TransformStream {
     this._buffer = "";
   }
 };
-var ModelDoesNotSupportToolsError = class extends S2.TaggedError()("ModelDoesNotSupportToolsError", {
-  model: S2.String
+var ModelDoesNotSupportToolsError = class extends S3.TaggedError()("ModelDoesNotSupportToolsError", {
+  model: S3.String
 }) {
 };
-var ModelNotFoundError = class extends S2.TaggedError()("ModelNotFoundError", {
-  model: S2.String
+var ModelNotFoundError = class extends S3.TaggedError()("ModelNotFoundError", {
+  model: S3.String
 }) {
 };
 var parseOllamaError = (error) => {
@@ -1221,7 +1226,7 @@ var parseOllamaError = (error) => {
   }
   throw new Error(`Ollama API error: ${error}`);
 };
-var parseErrorResponse2 = async (response) => {
+var parseErrorResponse = async (response) => {
   if (response.headers.get("content-type")?.includes("application/json")) {
     const body = await response.json();
     return parseOllamaError(body.error);
@@ -1319,9 +1324,9 @@ var StreamTransform = class {
 };
 
 // packages/experimental/assistant/src/ai-service/parser.ts
-import { Message as Message3 } from "@dxos/artifact";
+import { Message as Message4 } from "@dxos/artifact";
 import { Event } from "@dxos/async";
-import { create as create2 } from "@dxos/echo-schema";
+import { createStatic as createStatic2 } from "@dxos/echo-schema";
 import { invariant as invariant4 } from "@dxos/invariant";
 import { log as log5 } from "@dxos/log";
 import { isNotFalsy, safeParseJson } from "@dxos/util";
@@ -1400,7 +1405,7 @@ var MixedStreamParser = class {
               C: (f, a) => f(...a)
             });
           }
-          this._message = create2(Message3, {
+          this._message = createStatic2(Message4, {
             role: event.message.role,
             content: [
               ...event.message.content
@@ -1681,18 +1686,6 @@ var mergeMessageBlock = (contentBlock, streamBlock) => {
           }
           break;
         }
-        case "proposal": {
-          if (streamBlock.content.length === 1 && streamBlock.content[0].type === "text") {
-            return {
-              type: "json",
-              disposition: "proposal",
-              json: JSON.stringify({
-                text: streamBlock.content[0].content
-              })
-            };
-          }
-          break;
-        }
         case "select": {
           return {
             type: "json",
@@ -2190,7 +2183,8 @@ var isReference = (schema) => schema.$id === JSON_SCHEMA_ECHO_REF_ID || schema.t
 // packages/experimental/assistant/src/cypher/echo-data-source.ts
 import { raise as raise2 } from "@dxos/debug";
 import { Filter } from "@dxos/echo-db";
-import { getSchema, getSchemaTypename, StoredSchema, toJsonSchema } from "@dxos/echo-schema";
+import { getSchemaTypename, StoredSchema, toJsonSchema } from "@dxos/echo-schema";
+import { getSchema } from "@dxos/live-object";
 import { log as log8 } from "@dxos/log";
 function _ts_decorate(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
@@ -2274,7 +2268,7 @@ var EchoDataSource = class {
 _ts_decorate([
   log8.method(void 0, void 0, {
     F: __dxlog_file8,
-    L: 16,
+    L: 17,
     S: void 0,
     C: (f, a) => f(...a)
   })
@@ -2282,7 +2276,7 @@ _ts_decorate([
 _ts_decorate([
   log8.method(void 0, void 0, {
     F: __dxlog_file8,
-    L: 33,
+    L: 34,
     S: void 0,
     C: (f, a) => f(...a)
   })
@@ -2303,7 +2297,6 @@ export {
   LLMModel,
   ToolTypes,
   GenerateRequest,
-  GenerateResponse,
   runLLM,
   isToolUse,
   runTools,
@@ -2321,4 +2314,4 @@ export {
   isReference,
   EchoDataSource
 };
-//# sourceMappingURL=chunk-7JI4MDGL.mjs.map
+//# sourceMappingURL=chunk-DCQJOLSR.mjs.map