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

package/dist/lib/node/index.cjs

@@ -18,362 +18,42 @@ var __copyProps = (to, from, except, desc) => {
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var node_exports = {};
 __export(node_exports, {
-  AIServiceEdgeClient: () => import_chunk_7VH7NMIM.AIServiceEdgeClient,
-  AIServiecOverloadedError: () => AIServiecOverloadedError,
-  AISession: () => AISession,
-  DEFAULT_EDGE_MODEL: () => import_chunk_7VH7NMIM.DEFAULT_EDGE_MODEL,
-  DEFAULT_EDGE_MODELS: () => import_chunk_7VH7NMIM.DEFAULT_EDGE_MODELS,
-  DEFAULT_OLLAMA_ENDPOINT: () => import_chunk_7VH7NMIM.DEFAULT_OLLAMA_ENDPOINT,
-  DEFAULT_OLLAMA_MODEL: () => import_chunk_7VH7NMIM.DEFAULT_OLLAMA_MODEL,
-  DEFAULT_OLLAMA_MODELS: () => import_chunk_7VH7NMIM.DEFAULT_OLLAMA_MODELS,
-  EchoDataSource: () => import_chunk_7VH7NMIM.EchoDataSource,
-  GenerateRequest: () => import_chunk_7VH7NMIM.GenerateRequest,
-  GenerateResponse: () => import_chunk_7VH7NMIM.GenerateResponse,
-  GenerationStreamImpl: () => import_chunk_7VH7NMIM.GenerationStreamImpl,
-  LLMModel: () => import_chunk_7VH7NMIM.LLMModel,
-  MessageCollector: () => import_chunk_7VH7NMIM.MessageCollector,
-  MixedStreamParser: () => import_chunk_7VH7NMIM.MixedStreamParser,
-  OllamaClient: () => import_chunk_7VH7NMIM.OllamaClient,
-  StreamTransform: () => import_chunk_7VH7NMIM.StreamTransform,
-  ToolTypes: () => import_chunk_7VH7NMIM.ToolTypes,
-  createArtifactElement: () => import_chunk_7VH7NMIM.createArtifactElement,
-  createGenerationStream: () => import_chunk_7VH7NMIM.createGenerationStream,
-  emitMessageAsEvents: () => import_chunk_7VH7NMIM.emitMessageAsEvents,
-  executeQuery: () => import_chunk_7VH7NMIM.executeQuery,
-  executeQueryPlan: () => import_chunk_7VH7NMIM.executeQueryPlan,
-  formatInferredRelationshipLabel: () => import_chunk_7VH7NMIM.formatInferredRelationshipLabel,
-  formatJsonSchemaForLLM: () => import_chunk_7VH7NMIM.formatJsonSchemaForLLM,
-  formatNodeLabel: () => import_chunk_7VH7NMIM.formatNodeLabel,
-  isReference: () => import_chunk_7VH7NMIM.isReference,
-  isToolUse: () => import_chunk_7VH7NMIM.isToolUse,
-  mergeMessageBlock: () => import_chunk_7VH7NMIM.mergeMessageBlock,
-  parseCypherQuery: () => import_chunk_7VH7NMIM.parseCypherQuery,
-  runLLM: () => import_chunk_7VH7NMIM.runLLM,
-  runTools: () => import_chunk_7VH7NMIM.runTools
+  AIServiceEdgeClient: () => import_chunk_NOB7CIIC.AIServiceEdgeClient,
+  AIServiecOverloadedError: () => import_chunk_NOB7CIIC.AIServiecOverloadedError,
+  AISession: () => import_chunk_NOB7CIIC.AISession,
+  DEFAULT_EDGE_MODEL: () => import_chunk_NOB7CIIC.DEFAULT_EDGE_MODEL,
+  DEFAULT_EDGE_MODELS: () => import_chunk_NOB7CIIC.DEFAULT_EDGE_MODELS,
+  DEFAULT_OLLAMA_ENDPOINT: () => import_chunk_NOB7CIIC.DEFAULT_OLLAMA_ENDPOINT,
+  DEFAULT_OLLAMA_MODEL: () => import_chunk_NOB7CIIC.DEFAULT_OLLAMA_MODEL,
+  DEFAULT_OLLAMA_MODELS: () => import_chunk_NOB7CIIC.DEFAULT_OLLAMA_MODELS,
+  EchoDataSource: () => import_chunk_NOB7CIIC.EchoDataSource,
+  GenerateRequest: () => import_chunk_NOB7CIIC.GenerateRequest,
+  GenerateResponse: () => import_chunk_NOB7CIIC.GenerateResponse,
+  GenerationStreamImpl: () => import_chunk_NOB7CIIC.GenerationStreamImpl,
+  LLMModel: () => import_chunk_NOB7CIIC.LLMModel,
+  MessageCollector: () => import_chunk_NOB7CIIC.MessageCollector,
+  MixedStreamParser: () => import_chunk_NOB7CIIC.MixedStreamParser,
+  OllamaClient: () => import_chunk_NOB7CIIC.OllamaClient,
+  StreamTransform: () => import_chunk_NOB7CIIC.StreamTransform,
+  ToolTypes: () => import_chunk_NOB7CIIC.ToolTypes,
+  VersionPin: () => import_chunk_NOB7CIIC.VersionPin,
+  createArtifactElement: () => import_chunk_NOB7CIIC.createArtifactElement,
+  createGenerationStream: () => import_chunk_NOB7CIIC.createGenerationStream,
+  emitMessageAsEvents: () => import_chunk_NOB7CIIC.emitMessageAsEvents,
+  executeQuery: () => import_chunk_NOB7CIIC.executeQuery,
+  executeQueryPlan: () => import_chunk_NOB7CIIC.executeQueryPlan,
+  formatInferredRelationshipLabel: () => import_chunk_NOB7CIIC.formatInferredRelationshipLabel,
+  formatJsonSchemaForLLM: () => import_chunk_NOB7CIIC.formatJsonSchemaForLLM,
+  formatNodeLabel: () => import_chunk_NOB7CIIC.formatNodeLabel,
+  isReference: () => import_chunk_NOB7CIIC.isReference,
+  isToolUse: () => import_chunk_NOB7CIIC.isToolUse,
+  mergeMessageBlock: () => import_chunk_NOB7CIIC.mergeMessageBlock,
+  parseCypherQuery: () => import_chunk_NOB7CIIC.parseCypherQuery,
+  runLLM: () => import_chunk_NOB7CIIC.runLLM,
+  runTools: () => import_chunk_NOB7CIIC.runTools
 });
 module.exports = __toCommonJS(node_exports);
-var import_chunk_7VH7NMIM = require("./chunk-7VH7NMIM.cjs");
-var import_effect = require("effect");
-var import_artifact = require("@dxos/artifact");
-var import_async = require("@dxos/async");
-var import_echo_schema = require("@dxos/echo-schema");
-var import_invariant = require("@dxos/invariant");
-var import_log = require("@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;
-  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_file = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/session/session.ts";
-var AISession = class _AISession {
-  constructor(_options) {
-    this._options = _options;
-    this._parser = new import_chunk_7VH7NMIM.MixedStreamParser();
-    this._pending = [];
-    this._history = [];
-    this.message = this._parser.message;
-    this.block = this._parser.block;
-    this.update = this._parser.update;
-    this.streamEvent = this._parser.streamEvent;
-    this.userMessage = new import_async.Event();
-    this._parser.message.on((message) => {
-      this._pending.push(message);
-    });
-  }
-  async run(options) {
-    const systemTools = [
-      (0, import_artifact.defineTool)("system", {
-        name: "query_artifact_definitions",
-        description: "Query the available artifact definitions",
-        schema: import_effect.Schema.Struct({}),
-        execute: async () => {
-          return import_artifact.ToolResult.Success({
-            artifactDefinitions: options.artifacts.map((artifact) => ({
-              id: artifact.id,
-              name: artifact.name,
-              description: artifact.description
-            }))
-          });
-        }
-      })
-    ];
-    switch (this._options.operationModel) {
-      case "planning":
-        systemTools.push((0, import_artifact.defineTool)("system", {
-          name: "create_plan",
-          description: "Create a plan. Make sure that each step is independent and can be executed solely on the data returned by the previous step. The steps only share the data that is specified in the plan.",
-          schema: import_effect.Schema.Struct({
-            goal: import_effect.Schema.String.annotations({
-              description: "The goal that the plan will achieve."
-            }),
-            steps: import_effect.Schema.Array(import_effect.Schema.Struct({
-              action: import_effect.Schema.String.annotations({
-                description: "Complete, detailed action to perform."
-              }),
-              input: import_effect.Schema.String.annotations({
-                description: "The required input data for this step. First step must not require any data."
-              }),
-              output: import_effect.Schema.String.annotations({
-                description: "The output data from this step. This is passed to the next step as input."
-              }),
-              requiredArtifactIds: import_effect.Schema.Array(import_effect.Schema.String).annotations({
-                description: "The ids of the artifacts required to perform this step.",
-                examples: [
-                  [
-                    "artifact:dxos.org/example/Test"
-                  ]
-                ]
-              })
-            })).annotations({
-              description: "Steps"
-            })
-          }),
-          execute: async ({ goal, steps }) => {
-            const missingArtifactIds = steps.flatMap((step) => step.requiredArtifactIds).filter((artifactId) => !options.artifacts.some((artifact) => artifact.id === artifactId));
-            if (missingArtifactIds.length > 0) {
-              return import_artifact.ToolResult.Error(`One or more artifact ids are invalid: ${missingArtifactIds.join(", ")}`);
-            }
-            const stepResults = [];
-            for (const step of steps) {
-              (0, import_log.log)("executing step", {
-                action: step.action
-              }, {
-                F: __dxlog_file,
-                L: 160,
-                S: this,
-                C: (f, a) => f(...a)
-              });
-              const session = new _AISession({
-                operationModel: "immediate"
-              });
-              session.message.on((ev) => this.message.emit(ev));
-              session.block.on((ev) => this.block.emit(ev));
-              session.update.on((ev) => this.update.emit(ev));
-              session.streamEvent.on((ev) => this.streamEvent.emit(ev));
-              session.userMessage.on((ev) => this.userMessage.emit(ev));
-              const messages = await session.run({
-                client: options.client,
-                artifacts: options.artifacts,
-                tools: options.tools,
-                history: options.history,
-                extensions: options.extensions,
-                generationOptions: options.generationOptions,
-                requiredArtifactIds: step.requiredArtifactIds.slice(),
-                prompt: `
-                    You are an agent that executes the subtask in a plan.
-                    
-                    While you know the plan's goal only focus on your subtask
-                    Use the data from the previous step to complete your task.
-                    Return all of the data from "output" in your last message.
-                    
-                    Current step (that you need to complete):
-                    ${JSON.stringify({
-                  goal,
-                  input: step.input,
-                  output: step.output,
-                  yourTask: step.action
-                }, null, 2)}
-                    
-                    Previous step results:
-                    ${JSON.stringify(stepResults.at(-1))}.
-                  `
-              });
-              const result = messages.at(-1);
-              stepResults.push(result);
-            }
-            return import_artifact.ToolResult.Success({
-              stepResults
-            });
-          }
-        }));
-        break;
-      case "immediate":
-        systemTools.push((0, import_artifact.defineTool)("system", {
-          name: "require_artifact_definitions",
-          description: "Require the use of specific artifact definitions. This will allow the model to interact with artifact definitions and use their tools.",
-          schema: import_effect.Schema.Struct({
-            artifactDefinitionIds: import_effect.Schema.Array(import_effect.Schema.String).annotations({
-              description: "The ids of the artifact definitions to require",
-              examples: [
-                [
-                  "artifact:dxos.org/example/Test"
-                ]
-              ]
-            })
-          }),
-          execute: async ({ artifactDefinitionIds }) => {
-            const missingArtifactDefinitionIds = artifactDefinitionIds.filter((artifactId) => !options.artifacts.some((artifact) => artifact.id === artifactId));
-            if (missingArtifactDefinitionIds.length > 0) {
-              return import_artifact.ToolResult.Error(`One or more artifact definition ids are invalid: ${missingArtifactDefinitionIds.join(", ")}`);
-            }
-            for (const artifactDefinitionId of artifactDefinitionIds) {
-              requiredArtifactIds.add(artifactDefinitionId);
-            }
-            return import_artifact.ToolResult.Success({});
-          }
-        }));
-        break;
-    }
-    this._history = [
-      ...options.history
-    ];
-    this._pending = [
-      (0, import_echo_schema.create)(import_artifact.Message, {
-        role: "user",
-        content: [
-          {
-            type: "text",
-            text: options.prompt
-          }
-        ]
-      })
-    ];
-    this.userMessage.emit(this._pending.at(-1));
-    this._stream = void 0;
-    let error;
-    const requiredArtifactIds = new Set(options.requiredArtifactIds ?? []);
-    try {
-      let more = false;
-      do {
-        const tools = [
-          ...systemTools,
-          ...options.tools,
-          ...options.artifacts.filter((artifact) => requiredArtifactIds.has(artifact.id)).flatMap((artifact) => artifact.tools)
-        ];
-        (0, import_log.log)("request", {
-          pending: this._pending.length,
-          history: this._history.length,
-          tools: tools.map((tool) => tool.name)
-        }, {
-          F: __dxlog_file,
-          L: 264,
-          S: this,
-          C: (f, a) => f(...a)
-        });
-        this._stream = await options.client.execStream({
-          ...options.generationOptions ?? {},
-          // TODO(burdon): Rename messages or separate history/message.
-          history: [
-            ...this._history,
-            ...this._pending
-          ],
-          tools,
-          systemPrompt: options.systemPrompt ?? createBaseInstructions({
-            availableArtifacts: Array.from(requiredArtifactIds),
-            operationModel: this._options.operationModel
-          })
-        });
-        await this._parser.parse(this._stream);
-        await this._stream.complete();
-        (0, import_log.log)("response", {
-          pending: this._pending
-        }, {
-          F: __dxlog_file,
-          L: 289,
-          S: this,
-          C: (f, a) => f(...a)
-        });
-        more = false;
-        const message = this._pending.at(-1);
-        (0, import_invariant.invariant)(message, void 0, {
-          F: __dxlog_file,
-          L: 294,
-          S: this,
-          A: [
-            "message",
-            ""
-          ]
-        });
-        if ((0, import_chunk_7VH7NMIM.isToolUse)(message)) {
-          (0, import_log.log)("tool request...", void 0, {
-            F: __dxlog_file,
-            L: 296,
-            S: this,
-            C: (f, a) => f(...a)
-          });
-          const response = await (0, import_chunk_7VH7NMIM.runTools)({
-            message: this._pending.at(-1),
-            tools,
-            extensions: options.extensions ?? {}
-          });
-          (0, import_log.log)("tool response", {
-            response
-          }, {
-            F: __dxlog_file,
-            L: 303,
-            S: this,
-            C: (f, a) => f(...a)
-          });
-          switch (response.type) {
-            case "continue": {
-              this._pending = [
-                ...this._pending,
-                response.message
-              ];
-              this.userMessage.emit(response.message);
-              more = true;
-              break;
-            }
-          }
-        }
-      } while (more);
-    } catch (err) {
-      import_log.log.catch(err, void 0, {
-        F: __dxlog_file,
-        L: 315,
-        S: this,
-        C: (f, a) => f(...a)
-      });
-      if (err instanceof Error && err.message.includes("Overloaded")) {
-        error = new AIServiecOverloadedError("AI service overloaded", {
-          cause: err
-        });
-      } else {
-        error = new Error("AI service error", {
-          cause: err
-        });
-      }
-    } finally {
-      this._stream = void 0;
-    }
-    if (error) {
-      throw error;
-    }
-    return this._pending;
-  }
-  abort() {
-    this._stream?.abort();
-  }
-};
-_ts_decorate([
-  import_async.synchronized
-], AISession.prototype, "run", null);
-var createBaseInstructions = ({ availableArtifacts, operationModel }) => `
-  You are a friendly, advanced AI assistant capable of creating and managing artifacts from available data and tools. 
-  Your task is to process user commands and questions and decide how best to respond.
-  In some cases, you will need to create or reference data objects called artifacts.
-
-  Follow these guidelines carefully:
-
-  Decision-making:
-
-  - Analyze the structure and type of the content in the user's message.
-  - Can you complete the task using the available artifacts?
-  - If you can't complete the task using the available artifacts, query the list of available artifacts using the appropriate tool.
-  - Identify which artifacts are relevant to the user's request.
-  ${operationModel === "planning" ? `
-        - Break down the user's request into a step-by-step plan that you will use to execute the user's request, the plan items should include artifacts which will be used to perform the actions.
-      ` : `
-      - Are the required artifacts already available?
-      - If not, select which artifact(s) will be the most relevant and require them using the require_artifacts tool.
-      - The require'd artifact tools will be available for use after require.
-    `}
-
-  ${availableArtifacts.length > 0 ? `Artifacts already in context: ${availableArtifacts.join("\n")}` : ""}
-`;
-var AIServiecOverloadedError = class extends import_effect.Schema.TaggedError()("AIServiecOverloadedError", {}) {
-};
+var import_chunk_NOB7CIIC = require("./chunk-NOB7CIIC.cjs");
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   AIServiceEdgeClient,
@@ -394,6 +74,7 @@ var AIServiecOverloadedError = class extends import_effect.Schema.TaggedError()(
   OllamaClient,
   StreamTransform,
   ToolTypes,
+  VersionPin,
   createArtifactElement,
   createGenerationStream,
   emitMessageAsEvents,

package/dist/lib/node/index.cjs.map

@@ -1,7 +1,7 @@
 {
   "version": 3,
-  "sources": ["../../../src/session/session.ts"],
-  "sourcesContent": ["//\n// Copyright 2025 DXOS.org\n//\n\nimport { Schema as S } from 'effect';\n\nimport { defineTool, Message, ToolResult, type ArtifactDefinition, type Tool } from '@dxos/artifact';\nimport { Event, synchronized } from '@dxos/async';\nimport { create } from '@dxos/echo-schema';\nimport { invariant } from '@dxos/invariant';\nimport { log } from '@dxos/log';\n\nimport { MixedStreamParser, type AIServiceClient, type GenerateRequest, type GenerationStream } from '../ai-service';\nimport { isToolUse, runTools } from '../conversation';\n/**\n * Contains message history, tools, current context.\n * Current context means the state of the app, time of day, and other contextual information.\n * It makes requests to the model, its a state machine.\n * It keeps track of the current goal.\n * It manages the context window.\n * Tracks the success criteria of reaching the goal, exposing metrics (stretch)\n * Could be run locally in the app or remotely.\n * Could be personal or shared.\n */\n\nexport type SessionRunOptions = {\n  client: AIServiceClient;\n\n  artifacts: ArtifactDefinition[];\n\n  /**\n   * Non-artifact specific tools.\n   */\n  tools: Tool[];\n\n  history: Message[];\n\n  prompt: string;\n\n  generationOptions?: Pick<GenerateRequest, 'model'>;\n\n  extensions?: ToolContextExtensions;\n\n  systemPrompt?: string;\n\n  /**\n   * Pre-require artifacts.\n   */\n  requiredArtifactIds?: string[];\n};\n\ntype OperationModel = 'planning' | 'immediate';\n\nexport type AiSessionOptions = {\n  /**\n   * In planning mode, the model will create a plan before executing the user's request.\n   * The plan will determine the artifacts that will be used to perform the actions.\n   * In require mode, the model will select the artifacts that will be used to perform the actions.\n   */\n  operationModel: OperationModel;\n};\n\nexport class AISession {\n  /** SSE stream parser. */\n  private readonly _parser = new MixedStreamParser();\n\n  /** Pending messages (incl. the current user request). */\n  private _pending: Message[] = [];\n\n  /** Current streaming response. */\n  private _stream: GenerationStream | undefined;\n\n  /** Prior history from queue. */\n  private _history: Message[] = [];\n\n  /**\n   * New message.\n   */\n  public readonly message = this._parser.message;\n\n  /**\n   * Complete block added to Message.\n   */\n  public readonly block = this._parser.block;\n\n  /**\n   * Update partial block (while streaming).\n   */\n  public readonly update = this._parser.update;\n\n  public readonly streamEvent = this._parser.streamEvent;\n\n  /**\n   * User prompt or tool result.\n   */\n  public readonly userMessage = new Event<Message>();\n\n  constructor(private readonly _options: AiSessionOptions) {\n    // Message complete.\n    this._parser.message.on((message) => {\n      this._pending.push(message);\n    });\n  }\n\n  @synchronized\n  async run(options: SessionRunOptions): Promise<Message[]> {\n    const systemTools: Tool[] = [\n      defineTool('system', {\n        name: 'query_artifact_definitions',\n        description: 'Query the available artifact definitions',\n        schema: S.Struct({}),\n        execute: async () => {\n          return ToolResult.Success({\n            artifactDefinitions: options.artifacts.map((artifact) => ({\n              id: artifact.id,\n              name: artifact.name,\n              description: artifact.description,\n            })),\n          });\n        },\n      }),\n    ];\n    switch (this._options.operationModel) {\n      case 'planning':\n        systemTools.push(\n          defineTool('system', {\n            name: 'create_plan',\n            description:\n              'Create a plan. Make sure that each step is independent and can be executed solely on the data returned by the previous step. The steps only share the data that is specified in the plan.',\n            schema: S.Struct({\n              goal: S.String.annotations({ description: 'The goal that the plan will achieve.' }),\n              steps: S.Array(\n                S.Struct({\n                  action: S.String.annotations({\n                    description: 'Complete, detailed action to perform.',\n                  }),\n                  input: S.String.annotations({\n                    description: 'The required input data for this step. First step must not require any data.',\n                  }),\n                  output: S.String.annotations({\n                    description: 'The output data from this step. This is passed to the next step as input.',\n                  }),\n                  requiredArtifactIds: S.Array(S.String).annotations({\n                    description: 'The ids of the artifacts required to perform this step.',\n                    examples: [['artifact:dxos.org/example/Test']],\n                  }),\n                }),\n              ).annotations({ description: 'Steps' }),\n            }),\n            execute: async ({ goal, steps }) => {\n              const missingArtifactIds = steps\n                .flatMap((step) => step.requiredArtifactIds)\n                .filter((artifactId) => !options.artifacts.some((artifact) => artifact.id === artifactId));\n              if (missingArtifactIds.length > 0) {\n                return ToolResult.Error(`One or more artifact ids are invalid: ${missingArtifactIds.join(', ')}`);\n              }\n\n              const stepResults: any[] = [];\n              for (const step of steps) {\n                log('executing step', { action: step.action });\n                const session = new AISession({ operationModel: 'immediate' });\n                session.message.on((ev) => this.message.emit(ev));\n                session.block.on((ev) => this.block.emit(ev));\n                session.update.on((ev) => this.update.emit(ev));\n                session.streamEvent.on((ev) => this.streamEvent.emit(ev));\n                session.userMessage.on((ev) => this.userMessage.emit(ev));\n\n                const messages = await session.run({\n                  client: options.client,\n                  artifacts: options.artifacts,\n                  tools: options.tools,\n                  history: options.history,\n                  extensions: options.extensions,\n                  generationOptions: options.generationOptions,\n                  requiredArtifactIds: step.requiredArtifactIds.slice(),\n                  prompt: `\n                    You are an agent that executes the subtask in a plan.\n                    \n                    While you know the plan's goal only focus on your subtask\n                    Use the data from the previous step to complete your task.\n                    Return all of the data from \"output\" in your last message.\n                    \n                    Current step (that you need to complete):\n                    ${JSON.stringify(\n                      {\n                        goal,\n                        input: step.input,\n                        output: step.output,\n                        yourTask: step.action,\n                      },\n                      null,\n                      2,\n                    )}\n                    \n                    Previous step results:\n                    ${JSON.stringify(stepResults.at(-1))}.\n                  `,\n                });\n                const result = messages.at(-1);\n                stepResults.push(result);\n              }\n\n              return ToolResult.Success({ stepResults });\n            },\n          }),\n        );\n        break;\n      case 'immediate':\n        systemTools.push(\n          defineTool('system', {\n            name: 'require_artifact_definitions',\n            description:\n              'Require the use of specific artifact definitions. This will allow the model to interact with artifact definitions and use their tools.',\n            schema: S.Struct({\n              artifactDefinitionIds: S.Array(S.String).annotations({\n                description: 'The ids of the artifact definitions to require',\n                examples: [['artifact:dxos.org/example/Test']],\n              }),\n            }),\n            execute: async ({ artifactDefinitionIds }) => {\n              const missingArtifactDefinitionIds = artifactDefinitionIds.filter(\n                (artifactId) => !options.artifacts.some((artifact) => artifact.id === artifactId),\n              );\n              if (missingArtifactDefinitionIds.length > 0) {\n                return ToolResult.Error(\n                  `One or more artifact definition ids are invalid: ${missingArtifactDefinitionIds.join(', ')}`,\n                );\n              }\n\n              for (const artifactDefinitionId of artifactDefinitionIds) {\n                requiredArtifactIds.add(artifactDefinitionId);\n              }\n\n              return ToolResult.Success({});\n            },\n          }),\n        );\n        break;\n    }\n\n    this._history = [...options.history];\n    this._pending = [\n      create(Message, {\n        role: 'user',\n        content: [{ type: 'text', text: options.prompt }],\n      }),\n    ];\n    this.userMessage.emit(this._pending.at(-1)!);\n    this._stream = undefined;\n    let error: Error | undefined;\n\n    const requiredArtifactIds = new Set<string>(options.requiredArtifactIds ?? []);\n    try {\n      let more = false;\n      do {\n        const tools = [\n          ...systemTools,\n          ...options.tools,\n          ...options.artifacts\n            .filter((artifact) => requiredArtifactIds.has(artifact.id))\n            .flatMap((artifact) => artifact.tools),\n        ];\n\n        log('request', {\n          pending: this._pending.length,\n          history: this._history.length,\n          tools: tools.map((tool) => tool.name),\n        });\n\n        // Open request stream.\n        this._stream = await options.client.execStream({\n          ...(options.generationOptions ?? {}),\n          // TODO(burdon): Rename messages or separate history/message.\n          history: [...this._history, ...this._pending],\n          tools,\n          systemPrompt:\n            options.systemPrompt ??\n            createBaseInstructions({\n              availableArtifacts: Array.from(requiredArtifactIds),\n              operationModel: this._options.operationModel,\n            }),\n        });\n\n        // Wait until complete.\n        await this._parser.parse(this._stream);\n        await this._stream.complete();\n\n        // Add messages.\n        log('response', { pending: this._pending });\n\n        // Resolve tool use locally.\n        more = false;\n        const message = this._pending.at(-1);\n        invariant(message);\n        if (isToolUse(message)) {\n          log('tool request...');\n          const response = await runTools({\n            message: this._pending.at(-1)!,\n            tools,\n            extensions: options.extensions ?? {},\n          });\n\n          log('tool response', { response });\n          switch (response.type) {\n            case 'continue': {\n              this._pending = [...this._pending, response.message];\n              this.userMessage.emit(response.message);\n              more = true;\n              break;\n            }\n          }\n        }\n      } while (more);\n    } catch (err) {\n      log.catch(err);\n      if (err instanceof Error && err.message.includes('Overloaded')) {\n        error = new AIServiecOverloadedError('AI service overloaded', { cause: err });\n      } else {\n        error = new Error('AI service error', { cause: err });\n      }\n    } finally {\n      this._stream = undefined;\n    }\n\n    if (error) {\n      throw error;\n    }\n\n    return this._pending;\n  }\n\n  abort() {\n    this._stream?.abort();\n  }\n}\n\n// TODO(burdon): Use handlebars template with effect-schema input.\nconst createBaseInstructions = ({\n  availableArtifacts,\n  operationModel,\n}: {\n  availableArtifacts: string[];\n  operationModel: OperationModel;\n}) => `\n  You are a friendly, advanced AI assistant capable of creating and managing artifacts from available data and tools. \n  Your task is to process user commands and questions and decide how best to respond.\n  In some cases, you will need to create or reference data objects called artifacts.\n\n  Follow these guidelines carefully:\n\n  Decision-making:\n\n  - Analyze the structure and type of the content in the user's message.\n  - Can you complete the task using the available artifacts?\n  - If you can't complete the task using the available artifacts, query the list of available artifacts using the appropriate tool.\n  - Identify which artifacts are relevant to the user's request.\n  ${\n    operationModel === 'planning'\n      ? `\n        - Break down the user's request into a step-by-step plan that you will use to execute the user's request, the plan items should include artifacts which will be used to perform the actions.\n      `\n      : `\n      - Are the required artifacts already available?\n      - If not, select which artifact(s) will be the most relevant and require them using the require_artifacts tool.\n      - The require'd artifact tools will be available for use after require.\n    `\n  }\n\n  ${availableArtifacts.length > 0 ? `Artifacts already in context: ${availableArtifacts.join('\\n')}` : ''}\n`;\n\nexport class AIServiecOverloadedError extends S.TaggedError<AIServiecOverloadedError>()(\n  'AIServiecOverloadedError',\n  {},\n) {}\n"],
-  "mappings": ";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAIA,oBAA4B;AAE5B,sBAAoF;AACpF,mBAAoC;AACpC,yBAAuB;AACvB,uBAA0B;AAC1B,iBAAoB;;;;;;;;AAoDb,IAAMA,YAAN,MAAMA,WAAAA;EAmCXC,YAA6BC,UAA4B;SAA5BA,WAAAA;SAjCZC,UAAU,IAAIC,wCAAAA;SAGvBC,WAAsB,CAAA;SAMtBC,WAAsB,CAAA;SAKdC,UAAU,KAAKJ,QAAQI;SAKvBC,QAAQ,KAAKL,QAAQK;SAKrBC,SAAS,KAAKN,QAAQM;SAEtBC,cAAc,KAAKP,QAAQO;SAK3BC,cAAc,IAAIC,mBAAAA;AAIhC,SAAKT,QAAQI,QAAQM,GAAG,CAACN,YAAAA;AACvB,WAAKF,SAASS,KAAKP,OAAAA;IACrB,CAAA;EACF;EAEA,MACMQ,IAAIC,SAAgD;AACxD,UAAMC,cAAsB;UAC1BC,4BAAW,UAAU;QACnBC,MAAM;QACNC,aAAa;QACbC,QAAQC,cAAAA,OAAEC,OAAO,CAAC,CAAA;QAClBC,SAAS,YAAA;AACP,iBAAOC,2BAAWC,QAAQ;YACxBC,qBAAqBX,QAAQY,UAAUC,IAAI,CAACC,cAAc;cACxDC,IAAID,SAASC;cACbZ,MAAMW,SAASX;cACfC,aAAaU,SAASV;YACxB,EAAA;UACF,CAAA;QACF;MACF,CAAA;;AAEF,YAAQ,KAAKlB,SAAS8B,gBAAc;MAClC,KAAK;AACHf,oBAAYH,SACVI,4BAAW,UAAU;UACnBC,MAAM;UACNC,aACE;UACFC,QAAQC,cAAAA,OAAEC,OAAO;YACfU,MAAMX,cAAAA,OAAEY,OAAOC,YAAY;cAAEf,aAAa;YAAuC,CAAA;YACjFgB,OAAOd,cAAAA,OAAEe,MACPf,cAAAA,OAAEC,OAAO;cACPe,QAAQhB,cAAAA,OAAEY,OAAOC,YAAY;gBAC3Bf,aAAa;cACf,CAAA;cACAmB,OAAOjB,cAAAA,OAAEY,OAAOC,YAAY;gBAC1Bf,aAAa;cACf,CAAA;cACAoB,QAAQlB,cAAAA,OAAEY,OAAOC,YAAY;gBAC3Bf,aAAa;cACf,CAAA;cACAqB,qBAAqBnB,cAAAA,OAAEe,MAAMf,cAAAA,OAAEY,MAAM,EAAEC,YAAY;gBACjDf,aAAa;gBACbsB,UAAU;kBAAC;oBAAC;;;cACd,CAAA;YACF,CAAA,CAAA,EACAP,YAAY;cAAEf,aAAa;YAAQ,CAAA;UACvC,CAAA;UACAI,SAAS,OAAO,EAAES,MAAMG,MAAK,MAAE;AAC7B,kBAAMO,qBAAqBP,MACxBQ,QAAQ,CAACC,SAASA,KAAKJ,mBAAmB,EAC1CK,OAAO,CAACC,eAAe,CAAC/B,QAAQY,UAAUoB,KAAK,CAAClB,aAAaA,SAASC,OAAOgB,UAAAA,CAAAA;AAChF,gBAAIJ,mBAAmBM,SAAS,GAAG;AACjC,qBAAOxB,2BAAWyB,MAAM,yCAAyCP,mBAAmBQ,KAAK,IAAA,CAAA,EAAO;YAClG;AAEA,kBAAMC,cAAqB,CAAA;AAC3B,uBAAWP,QAAQT,OAAO;AACxBiB,kCAAI,kBAAkB;gBAAEf,QAAQO,KAAKP;cAAO,GAAA;;;;;;AAC5C,oBAAMgB,UAAU,IAAItD,WAAU;gBAAEgC,gBAAgB;cAAY,CAAA;AAC5DsB,sBAAQ/C,QAAQM,GAAG,CAAC0C,OAAO,KAAKhD,QAAQiD,KAAKD,EAAAA,CAAAA;AAC7CD,sBAAQ9C,MAAMK,GAAG,CAAC0C,OAAO,KAAK/C,MAAMgD,KAAKD,EAAAA,CAAAA;AACzCD,sBAAQ7C,OAAOI,GAAG,CAAC0C,OAAO,KAAK9C,OAAO+C,KAAKD,EAAAA,CAAAA;AAC3CD,sBAAQ5C,YAAYG,GAAG,CAAC0C,OAAO,KAAK7C,YAAY8C,KAAKD,EAAAA,CAAAA;AACrDD,sBAAQ3C,YAAYE,GAAG,CAAC0C,OAAO,KAAK5C,YAAY6C,KAAKD,EAAAA,CAAAA;AAErD,oBAAME,WAAW,MAAMH,QAAQvC,IAAI;gBACjC2C,QAAQ1C,QAAQ0C;gBAChB9B,WAAWZ,QAAQY;gBACnB+B,OAAO3C,QAAQ2C;gBACfC,SAAS5C,QAAQ4C;gBACjBC,YAAY7C,QAAQ6C;gBACpBC,mBAAmB9C,QAAQ8C;gBAC3BrB,qBAAqBI,KAAKJ,oBAAoBsB,MAAK;gBACnDC,QAAQ;;;;;;;;sBAQJC,KAAKC,UACL;kBACEjC;kBACAM,OAAOM,KAAKN;kBACZC,QAAQK,KAAKL;kBACb2B,UAAUtB,KAAKP;gBACjB,GACA,MACA,CAAA,CAAA;;;sBAIA2B,KAAKC,UAAUd,YAAYgB,GAAG,EAAC,CAAA,CAAA;;cAErC,CAAA;AACA,oBAAMC,SAASZ,SAASW,GAAG,EAAC;AAC5BhB,0BAAYtC,KAAKuD,MAAAA;YACnB;AAEA,mBAAO5C,2BAAWC,QAAQ;cAAE0B;YAAY,CAAA;UAC1C;QACF,CAAA,CAAA;AAEF;MACF,KAAK;AACHnC,oBAAYH,SACVI,4BAAW,UAAU;UACnBC,MAAM;UACNC,aACE;UACFC,QAAQC,cAAAA,OAAEC,OAAO;YACf+C,uBAAuBhD,cAAAA,OAAEe,MAAMf,cAAAA,OAAEY,MAAM,EAAEC,YAAY;cACnDf,aAAa;cACbsB,UAAU;gBAAC;kBAAC;;;YACd,CAAA;UACF,CAAA;UACAlB,SAAS,OAAO,EAAE8C,sBAAqB,MAAE;AACvC,kBAAMC,+BAA+BD,sBAAsBxB,OACzD,CAACC,eAAe,CAAC/B,QAAQY,UAAUoB,KAAK,CAAClB,aAAaA,SAASC,OAAOgB,UAAAA,CAAAA;AAExE,gBAAIwB,6BAA6BtB,SAAS,GAAG;AAC3C,qBAAOxB,2BAAWyB,MAChB,oDAAoDqB,6BAA6BpB,KAAK,IAAA,CAAA,EAAO;YAEjG;AAEA,uBAAWqB,wBAAwBF,uBAAuB;AACxD7B,kCAAoBgC,IAAID,oBAAAA;YAC1B;AAEA,mBAAO/C,2BAAWC,QAAQ,CAAC,CAAA;UAC7B;QACF,CAAA,CAAA;AAEF;IACJ;AAEA,SAAKpB,WAAW;SAAIU,QAAQ4C;;AAC5B,SAAKvD,WAAW;UACdqE,2BAAOC,yBAAS;QACdC,MAAM;QACNC,SAAS;UAAC;YAAEC,MAAM;YAAQC,MAAM/D,QAAQgD;UAAO;;MACjD,CAAA;;AAEF,SAAKrD,YAAY6C,KAAK,KAAKnD,SAAS+D,GAAG,EAAC,CAAA;AACxC,SAAKY,UAAUC;AACf,QAAIC;AAEJ,UAAMzC,sBAAsB,IAAI0C,IAAYnE,QAAQyB,uBAAuB,CAAA,CAAE;AAC7E,QAAI;AACF,UAAI2C,OAAO;AACX,SAAG;AACD,cAAMzB,QAAQ;aACT1C;aACAD,QAAQ2C;aACR3C,QAAQY,UACRkB,OAAO,CAAChB,aAAaW,oBAAoB4C,IAAIvD,SAASC,EAAE,CAAA,EACxDa,QAAQ,CAACd,aAAaA,SAAS6B,KAAK;;AAGzCN,4BAAI,WAAW;UACbiC,SAAS,KAAKjF,SAAS4C;UACvBW,SAAS,KAAKtD,SAAS2C;UACvBU,OAAOA,MAAM9B,IAAI,CAAC0D,SAASA,KAAKpE,IAAI;QACtC,GAAA;;;;;;AAGA,aAAK6D,UAAU,MAAMhE,QAAQ0C,OAAO8B,WAAW;UAC7C,GAAIxE,QAAQ8C,qBAAqB,CAAC;;UAElCF,SAAS;eAAI,KAAKtD;eAAa,KAAKD;;UACpCsD;UACA8B,cACEzE,QAAQyE,gBACRC,uBAAuB;YACrBC,oBAAoBtD,MAAMuD,KAAKnD,mBAAAA;YAC/BT,gBAAgB,KAAK9B,SAAS8B;UAChC,CAAA;QACJ,CAAA;AAGA,cAAM,KAAK7B,QAAQ0F,MAAM,KAAKb,OAAO;AACrC,cAAM,KAAKA,QAAQc,SAAQ;AAG3BzC,4BAAI,YAAY;UAAEiC,SAAS,KAAKjF;QAAS,GAAA;;;;;;AAGzC+E,eAAO;AACP,cAAM7E,UAAU,KAAKF,SAAS+D,GAAG,EAAC;AAClC2B,wCAAUxF,SAAAA,QAAAA;;;;;;;;;AACV,gBAAIyF,iCAAUzF,OAAAA,GAAU;AACtB8C,8BAAI,mBAAA,QAAA;;;;;;AACJ,gBAAM4C,WAAW,UAAMC,gCAAS;YAC9B3F,SAAS,KAAKF,SAAS+D,GAAG,EAAC;YAC3BT;YACAE,YAAY7C,QAAQ6C,cAAc,CAAC;UACrC,CAAA;AAEAR,8BAAI,iBAAiB;YAAE4C;UAAS,GAAA;;;;;;AAChC,kBAAQA,SAASnB,MAAI;YACnB,KAAK,YAAY;AACf,mBAAKzE,WAAW;mBAAI,KAAKA;gBAAU4F,SAAS1F;;AAC5C,mBAAKI,YAAY6C,KAAKyC,SAAS1F,OAAO;AACtC6E,qBAAO;AACP;YACF;UACF;QACF;MACF,SAASA;IACX,SAASe,KAAK;AACZ9C,qBAAI+C,MAAMD,KAAAA,QAAAA;;;;;;AACV,UAAIA,eAAejD,SAASiD,IAAI5F,QAAQ8F,SAAS,YAAA,GAAe;AAC9DnB,gBAAQ,IAAIoB,yBAAyB,yBAAyB;UAAEC,OAAOJ;QAAI,CAAA;MAC7E,OAAO;AACLjB,gBAAQ,IAAIhC,MAAM,oBAAoB;UAAEqD,OAAOJ;QAAI,CAAA;MACrD;IACF,UAAA;AACE,WAAKnB,UAAUC;IACjB;AAEA,QAAIC,OAAO;AACT,YAAMA;IACR;AAEA,WAAO,KAAK7E;EACd;EAEAmG,QAAQ;AACN,SAAKxB,SAASwB,MAAAA;EAChB;AACF;;EAtOGC;GA1CUzG,UAAAA,WAAAA,OAAAA,IAAAA;AAmRb,IAAM0F,yBAAyB,CAAC,EAC9BC,oBACA3D,eAAc,MAIV;;;;;;;;;;;;;IAcFA,mBAAmB,aACf;;UAGA;;;;KAIH;;IAGD2D,mBAAmB1C,SAAS,IAAI,iCAAiC0C,mBAAmBxC,KAAK,IAAA,CAAA,KAAU,EAAA;;AAGhG,IAAMmD,2BAAN,cAAuChF,cAAAA,OAAEoF,YAAW,EACzD,4BACA,CAAC,CAAA,EAAA;AACA;",
-  "names": ["AISession", "constructor", "_options", "_parser", "MixedStreamParser", "_pending", "_history", "message", "block", "update", "streamEvent", "userMessage", "Event", "on", "push", "run", "options", "systemTools", "defineTool", "name", "description", "schema", "S", "Struct", "execute", "ToolResult", "Success", "artifactDefinitions", "artifacts", "map", "artifact", "id", "operationModel", "goal", "String", "annotations", "steps", "Array", "action", "input", "output", "requiredArtifactIds", "examples", "missingArtifactIds", "flatMap", "step", "filter", "artifactId", "some", "length", "Error", "join", "stepResults", "log", "session", "ev", "emit", "messages", "client", "tools", "history", "extensions", "generationOptions", "slice", "prompt", "JSON", "stringify", "yourTask", "at", "result", "artifactDefinitionIds", "missingArtifactDefinitionIds", "artifactDefinitionId", "add", "create", "Message", "role", "content", "type", "text", "_stream", "undefined", "error", "Set", "more", "has", "pending", "tool", "execStream", "systemPrompt", "createBaseInstructions", "availableArtifacts", "from", "parse", "complete", "invariant", "isToolUse", "response", "runTools", "err", "catch", "includes", "AIServiecOverloadedError", "cause", "abort", "synchronized", "TaggedError"]
+  "sources": ["index.cjs"],
+  "sourcesContent": ["import {\n  AIServiceEdgeClient,\n  AIServiecOverloadedError,\n  AISession,\n  DEFAULT_EDGE_MODEL,\n  DEFAULT_EDGE_MODELS,\n  DEFAULT_OLLAMA_ENDPOINT,\n  DEFAULT_OLLAMA_MODEL,\n  DEFAULT_OLLAMA_MODELS,\n  EchoDataSource,\n  GenerateRequest,\n  GenerateResponse,\n  GenerationStreamImpl,\n  LLMModel,\n  MessageCollector,\n  MixedStreamParser,\n  OllamaClient,\n  StreamTransform,\n  ToolTypes,\n  VersionPin,\n  createArtifactElement,\n  createGenerationStream,\n  emitMessageAsEvents,\n  executeQuery,\n  executeQueryPlan,\n  formatInferredRelationshipLabel,\n  formatJsonSchemaForLLM,\n  formatNodeLabel,\n  isReference,\n  isToolUse,\n  mergeMessageBlock,\n  parseCypherQuery,\n  runLLM,\n  runTools\n} from \"./chunk-NOB7CIIC.cjs\";\nexport {\n  AIServiceEdgeClient,\n  AIServiecOverloadedError,\n  AISession,\n  DEFAULT_EDGE_MODEL,\n  DEFAULT_EDGE_MODELS,\n  DEFAULT_OLLAMA_ENDPOINT,\n  DEFAULT_OLLAMA_MODEL,\n  DEFAULT_OLLAMA_MODELS,\n  EchoDataSource,\n  GenerateRequest,\n  GenerateResponse,\n  GenerationStreamImpl,\n  LLMModel,\n  MessageCollector,\n  MixedStreamParser,\n  OllamaClient,\n  StreamTransform,\n  ToolTypes,\n  VersionPin,\n  createArtifactElement,\n  createGenerationStream,\n  emitMessageAsEvents,\n  executeQuery,\n  executeQueryPlan,\n  formatInferredRelationshipLabel,\n  formatJsonSchemaForLLM,\n  formatNodeLabel,\n  isReference,\n  isToolUse,\n  mergeMessageBlock,\n  parseCypherQuery,\n  runLLM,\n  runTools\n};\n//# sourceMappingURL=index.cjs.map\n"],
+  "mappings": ";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,4BAkCO;",
+  "names": []
 }