@dxos/assistant 0.8.1 → 0.8.2-main.10c050d

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

@@ -1,637 +0,0 @@
-"use strict";
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
-};
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
-      if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
-  }
-  return to;
-};
-var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-var testing_exports = {};
-__export(testing_exports, {
-  AI_SERVICE_ENDPOINT: () => AI_SERVICE_ENDPOINT,
-  Contact: () => Contact,
-  MockDataSource: () => MockDataSource,
-  ObjectId: () => ObjectId2,
-  Org: () => Org,
-  Project: () => Project,
-  Task: () => Task,
-  createCypherTool: () => createCypherTool,
-  createLogger: () => createLogger,
-  createSystemPrompt: () => createSystemPrompt,
-  createTestData: () => createTestData,
-  createTestOllamaClient: () => createTestOllamaClient,
-  seedTestData: () => seedTestData
-});
-module.exports = __toCommonJS(testing_exports);
-var import_chunk_IBOBRBDF = require("../chunk-IBOBRBDF.cjs");
-var import_effect = require("effect");
-var import_artifact = require("@dxos/artifact");
-var import_echo_schema = require("@dxos/echo-schema");
-var import_log = require("@dxos/log");
-var import_debug = require("@dxos/debug");
-var import_echo_schema2 = require("@dxos/echo-schema");
-var import_invariant = require("@dxos/invariant");
-var import_live_object = require("@dxos/live-object");
-var import_effect2 = require("effect");
-var import_echo_schema3 = require("@dxos/echo-schema");
-var AI_SERVICE_ENDPOINT = {
-  LOCAL: "http://localhost:8788",
-  REMOTE: "https://ai-service.dxos.workers.dev"
-};
-var createLogger = ({ stream, filter, onImage } = {}) => {
-  const images = {};
-  let currentText = "";
-  return (event) => {
-    if (typeof filter === "function" && !filter(event)) {
-      return;
-    }
-    const printContentBlock = (content) => {
-      switch (content.type) {
-        case "text":
-          process.stdout.write(content.text);
-          break;
-        case "tool_use":
-          process.stdout.write(`\u2699\uFE0F [Tool Use] ${content.name}
-`);
-          break;
-        case "image": {
-          if (content.id && content.source) {
-            images[content.id] = content.source;
-          }
-          process.stdout.write(`[Image id=${content.id} mediaType=${content.source?.mediaType}]`);
-          break;
-        }
-      }
-    };
-    const onTextBlockPrinted = (text) => {
-      const imageMatch = text.match(/<image id="([^"]+)"(?:\s+prompt="([^"]+)")?\s*\/>/);
-      if (imageMatch && onImage && images[imageMatch[1]]) {
-        onImage({
-          id: imageMatch[1],
-          description: imageMatch[2],
-          source: images[imageMatch[1]]
-        });
-      }
-    };
-    if (stream) {
-      switch (event.type) {
-        case "message_start": {
-          process.stdout.write(`${event.message.role.toUpperCase()}
-
-`);
-          for (const content of event.message.content) {
-            printContentBlock(content);
-          }
-          break;
-        }
-        case "content_block_start": {
-          currentText = "";
-          printContentBlock(event.content);
-          break;
-        }
-        case "content_block_delta": {
-          switch (event.delta.type) {
-            case "text_delta": {
-              currentText += event.delta.text;
-              process.stdout.write(event.delta.text);
-              break;
-            }
-            case "input_json_delta": {
-              process.stdout.write(event.delta.partial_json);
-              break;
-            }
-          }
-          break;
-        }
-        case "content_block_stop": {
-          process.stdout.write("\n");
-          onTextBlockPrinted(currentText);
-          currentText = "";
-          break;
-        }
-        case "message_delta": {
-          break;
-        }
-        case "message_stop": {
-          process.stdout.write("\n\n");
-          break;
-        }
-      }
-    }
-    switch (event.type) {
-      case "message": {
-        if (!stream || event.message.role !== "assistant") {
-          process.stdout.write(`${event.message.role.toUpperCase()}
-
-`);
-          for (const block of event.message.content) {
-            switch (block.type) {
-              case "text":
-                process.stdout.write(block.text + "\n");
-                onTextBlockPrinted(block.text);
-                break;
-              case "tool_use":
-                process.stdout.write(`\u2699\uFE0F [Tool Use] ${block.name}
-`);
-                process.stdout.write(`  ${JSON.stringify(block.input)}
-`);
-                break;
-              case "tool_result":
-                if (block.isError) {
-                  process.stdout.write("\u274C [Tool Error]\n");
-                  process.stdout.write(block.content + "\n");
-                } else {
-                  process.stdout.write("\u2705 [Tool Success]\n");
-                  process.stdout.write(block.content + "\n");
-                }
-            }
-            process.stdout.write("\n\n");
-          }
-        }
-        break;
-      }
-    }
-  };
-};
-var createTestOllamaClient = (options) => {
-  return new import_chunk_IBOBRBDF.OllamaClient({
-    tools: options?.tools,
-    overrides: {
-      model: import_chunk_IBOBRBDF.DEFAULT_OLLAMA_MODEL,
-      temperature: 0
-    }
-  });
-};
-var __dxlog_file = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/testing/query-promts.ts";
-var createCypherTool = (dataSource, schemaTypes = []) => (0, import_artifact.defineTool)("dxos.org/echo", {
-  name: "query",
-  description: "Query the ECHO graph database in cypher query language. Returns data from the database." + (schemaTypes.length > 0 ? `
-
-${createSystemPrompt(schemaTypes)}` : ""),
-  schema: import_effect.Schema.Struct({
-    query: import_effect.Schema.String.annotations({
-      description: `
-          A valid cypher query string to execute.
-          Query must have one MATCH clause.
-          Match clause can have multiple patterns but they must all be connected via a relationship chain.
-          Query might have zero or one WHERE clauses.
-          Query must have one RETURN clause.
-          RETURN clause can have multiple expressions separated by commas.
-          DISTINCT keyword is not allowed.
-
-          <example>
-          MATCH (n:Person) RETURN n.name
-          </example>
-
-          <example>
-          MATCH (s:Studio)-[:PRODUCED]->(m:Movie)-[:PRODUCED_IN]->(c:City {name: "Las Vegas"}) RETURN s
-          </example>
-          
-          <example>
-          MATCH (a:Actor)-[:STARRED_IN]->(m:Movie)<-[:PRODUCED]-(s:Studio {name: "Cinema Pictures"}) WHERE m.name = "Once upon a time" RETURN m.name, a.name
-          </example>
-        `
-    })
-  }),
-  execute: async ({ query }) => {
-    try {
-      (0, import_log.log)("cypher query", {
-        query
-      }, {
-        F: __dxlog_file,
-        L: 47,
-        S: void 0,
-        C: (f, a) => f(...a)
-      });
-      const results = await (0, import_chunk_IBOBRBDF.executeQuery)(dataSource, query);
-      (0, import_log.log)("query complete", {
-        results
-      }, {
-        F: __dxlog_file,
-        L: 49,
-        S: void 0,
-        C: (f, a) => f(...a)
-      });
-      return import_artifact.ToolResult.Success(results);
-    } catch (e) {
-      import_log.log.catch(e, void 0, {
-        F: __dxlog_file,
-        L: 52,
-        S: void 0,
-        C: (f, a) => f(...a)
-      });
-      return import_artifact.ToolResult.Error(e.message);
-    }
-  }
-});
-var createSystemPrompt = (schemaTypes) => import_chunk_IBOBRBDF.trim`
-  You have access to ECHO graph database. You can query the database to get data to satisfy user prompts.
-
-  Database schema is defined in pseud-cypher syntax. The schema is:
-
-  <schema>
-    ${(0, import_chunk_IBOBRBDF.formatJsonSchemaForLLM)(schemaTypes.map((schema) => (0, import_echo_schema.toJsonSchema)(schema)))}
-  <schema>
-
-  Before replying always think step-by-step on how to proceed.
-  Print your thoughts inside <cot> tags.
-  
-  <example>
-    <cot>To answer the question I need to ...</cot>
-  </example>
-
-  Before answering the user's question, decide what tools you need to use to answer the question.
-`;
-var __dxlog_file2 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/testing/test-data.ts";
-var seedTestData = (db) => {
-  const contactRich = db.add((0, import_live_object.create)(Contact, {
-    name: "Rich"
-  }));
-  const contactJosiah = db.add((0, import_live_object.create)(Contact, {
-    name: "Josiah"
-  }));
-  const contactDima = db.add((0, import_live_object.create)(Contact, {
-    name: "Dima"
-  }));
-  const contactFred = db.add((0, import_live_object.create)(Contact, {
-    name: "Fred"
-  }));
-  const projectComposer = db.add((0, import_live_object.create)(Project, {
-    name: "Composer"
-  }));
-  const projectEcho = db.add((0, import_live_object.create)(Project, {
-    name: "ECHO"
-  }));
-  const projectDoodles = db.add((0, import_live_object.create)(Project, {
-    name: "Doodles"
-  }));
-  const _taskComposer1 = db.add((0, import_live_object.create)(Task, {
-    name: "Optimize startup performance",
-    project: (0, import_live_object.makeRef)(projectComposer),
-    assignee: (0, import_live_object.makeRef)(contactJosiah)
-  }));
-  const _taskComposer2 = db.add((0, import_live_object.create)(Task, {
-    name: "Create form builder",
-    project: (0, import_live_object.makeRef)(projectComposer),
-    assignee: (0, import_live_object.makeRef)(contactRich)
-  }));
-  const _taskComposer3 = db.add((0, import_live_object.create)(Task, {
-    name: "Add support for custom themes",
-    project: (0, import_live_object.makeRef)(projectComposer),
-    assignee: (0, import_live_object.makeRef)(contactJosiah)
-  }));
-  const _taskComposer5 = db.add((0, import_live_object.create)(Task, {
-    name: "Implement community plugin",
-    project: (0, import_live_object.makeRef)(projectComposer),
-    assignee: (0, import_live_object.makeRef)(contactFred)
-  }));
-  const _taskComposer4 = db.add((0, import_live_object.create)(Task, {
-    name: "Implement dark mode",
-    project: (0, import_live_object.makeRef)(projectComposer),
-    assignee: (0, import_live_object.makeRef)(contactRich)
-  }));
-  const _taskEcho1 = db.add((0, import_live_object.create)(Task, {
-    name: "Implement cypher query engine",
-    project: (0, import_live_object.makeRef)(projectEcho),
-    assignee: (0, import_live_object.makeRef)(contactDima)
-  }));
-  const _taskEcho2 = db.add((0, import_live_object.create)(Task, {
-    name: "Add schema editor",
-    project: (0, import_live_object.makeRef)(projectEcho),
-    assignee: (0, import_live_object.makeRef)(contactRich)
-  }));
-  const _taskDoodles1 = db.add((0, import_live_object.create)(Task, {
-    name: "Add support for custom themes",
-    project: (0, import_live_object.makeRef)(projectDoodles),
-    assignee: (0, import_live_object.makeRef)(contactFred)
-  }));
-  const _taskDoodles2 = db.add((0, import_live_object.create)(Task, {
-    name: "Implement dark mode",
-    project: (0, import_live_object.makeRef)(projectDoodles),
-    assignee: (0, import_live_object.makeRef)(contactJosiah)
-  }));
-  const _orgDxos = db.add((0, import_live_object.create)(Org, {
-    name: "DXOS",
-    employees: [
-      (0, import_live_object.makeRef)(contactRich),
-      (0, import_live_object.makeRef)(contactJosiah),
-      (0, import_live_object.makeRef)(contactDima)
-    ],
-    projects: [
-      (0, import_live_object.makeRef)(projectEcho)
-    ]
-  }));
-  const _orgBraneframe = db.add((0, import_live_object.create)(Org, {
-    name: "Braneframe",
-    employees: [
-      (0, import_live_object.makeRef)(contactJosiah),
-      (0, import_live_object.makeRef)(contactRich)
-    ],
-    projects: [
-      (0, import_live_object.makeRef)(projectComposer)
-    ]
-  }));
-};
-var createTestData = () => {
-  const dataSource = new MockDataSource();
-  const contactRich = dataSource.add(Contact, {
-    id: import_echo_schema2.ObjectId.random(),
-    name: "Rich"
-  });
-  const contactJosiah = dataSource.add(Contact, {
-    id: import_echo_schema2.ObjectId.random(),
-    name: "Josiah"
-  });
-  const contactDima = dataSource.add(Contact, {
-    id: import_echo_schema2.ObjectId.random(),
-    name: "Dima"
-  });
-  const contactFred = dataSource.add(Contact, {
-    id: import_echo_schema2.ObjectId.random(),
-    name: "Fred"
-  });
-  const projectComposer = dataSource.add(Project, {
-    id: import_echo_schema2.ObjectId.random(),
-    name: "Composer"
-  });
-  const projectEcho = dataSource.add(Project, {
-    id: import_echo_schema2.ObjectId.random(),
-    name: "ECHO"
-  });
-  const projectDoodles = dataSource.add(Project, {
-    id: import_echo_schema2.ObjectId.random(),
-    name: "Doodles"
-  });
-  const _taskComposer1 = dataSource.add(Task, {
-    id: import_echo_schema2.ObjectId.random(),
-    name: "Optimize startup performance",
-    project: (0, import_live_object.makeRef)(projectComposer),
-    assignee: (0, import_live_object.makeRef)(contactJosiah)
-  });
-  const _taskComposer2 = dataSource.add(Task, {
-    id: import_echo_schema2.ObjectId.random(),
-    name: "Create form builder",
-    project: (0, import_live_object.makeRef)(projectComposer),
-    assignee: (0, import_live_object.makeRef)(contactRich)
-  });
-  const _taskComposer3 = dataSource.add(Task, {
-    id: import_echo_schema2.ObjectId.random(),
-    name: "Add support for custom themes",
-    project: (0, import_live_object.makeRef)(projectComposer),
-    assignee: (0, import_live_object.makeRef)(contactJosiah)
-  });
-  const _taskComposer5 = dataSource.add(Task, {
-    id: import_echo_schema2.ObjectId.random(),
-    name: "Implement community plugin",
-    project: (0, import_live_object.makeRef)(projectComposer),
-    assignee: (0, import_live_object.makeRef)(contactFred)
-  });
-  const _taskComposer4 = dataSource.add(Task, {
-    id: import_echo_schema2.ObjectId.random(),
-    name: "Implement dark mode",
-    project: (0, import_live_object.makeRef)(projectComposer),
-    assignee: (0, import_live_object.makeRef)(contactRich)
-  });
-  const _taskEcho1 = dataSource.add(Task, {
-    id: import_echo_schema2.ObjectId.random(),
-    name: "Implement cypher query engine",
-    project: (0, import_live_object.makeRef)(projectEcho),
-    assignee: (0, import_live_object.makeRef)(contactDima)
-  });
-  const _taskEcho2 = dataSource.add(Task, {
-    id: import_echo_schema2.ObjectId.random(),
-    name: "Add schema editor",
-    project: (0, import_live_object.makeRef)(projectEcho),
-    assignee: (0, import_live_object.makeRef)(contactRich)
-  });
-  const _taskDoodles1 = dataSource.add(Task, {
-    id: import_echo_schema2.ObjectId.random(),
-    name: "Add support for custom themes",
-    project: (0, import_live_object.makeRef)(projectDoodles),
-    assignee: (0, import_live_object.makeRef)(contactFred)
-  });
-  const _taskDoodles2 = dataSource.add(Task, {
-    id: import_echo_schema2.ObjectId.random(),
-    name: "Implement dark mode",
-    project: (0, import_live_object.makeRef)(projectDoodles),
-    assignee: (0, import_live_object.makeRef)(contactJosiah)
-  });
-  const _orgDxos = dataSource.add(Org, {
-    id: import_echo_schema2.ObjectId.random(),
-    name: "DXOS",
-    employees: [
-      (0, import_live_object.makeRef)(contactRich),
-      (0, import_live_object.makeRef)(contactJosiah),
-      (0, import_live_object.makeRef)(contactDima)
-    ],
-    projects: [
-      (0, import_live_object.makeRef)(projectEcho)
-    ]
-  });
-  const _orgBraneframe = dataSource.add(Org, {
-    id: import_echo_schema2.ObjectId.random(),
-    name: "Braneframe",
-    employees: [
-      (0, import_live_object.makeRef)(contactJosiah),
-      (0, import_live_object.makeRef)(contactRich)
-    ],
-    projects: [
-      (0, import_live_object.makeRef)(projectComposer)
-    ]
-  });
-  dataSource.computeGraph();
-  return dataSource;
-};
-var MockDataSource = class {
-  constructor() {
-    this.objects = {};
-    this.schema = {};
-    this.nodes = [];
-    this.relationships = [];
-  }
-  async getNodes({ label }) {
-    return this.nodes.filter((node) => !label || node.label === label);
-  }
-  async getRelationships({ label }) {
-    return this.relationships.filter((relationship) => !label || relationship.label === label);
-  }
-  add(schema, data) {
-    (0, import_invariant.invariant)(typeof data.id === "string", "Data must have an id", {
-      F: __dxlog_file2,
-      L: 277,
-      S: this,
-      A: [
-        "typeof data.id === 'string'",
-        "'Data must have an id'"
-      ]
-    });
-    const jsonSchema = (0, import_echo_schema2.toJsonSchema)(schema);
-    if (!this.schema[jsonSchema.$id]) {
-      this.schema[jsonSchema.$id] = {
-        json: jsonSchema,
-        instance: schema
-      };
-    }
-    const typenameDxn = jsonSchema.$id ?? (0, import_debug.raise)(new Error("Schema must have $id"));
-    this.objects[data.id] = {
-      schema,
-      id: data.id,
-      typeDxn: typenameDxn,
-      data
-    };
-    return data;
-  }
-  computeGraph() {
-    this.nodes = [];
-    this.relationships = [];
-    for (const object of Object.values(this.objects)) {
-      this.nodes.push({
-        kind: "node",
-        id: object.id,
-        label: (0, import_chunk_IBOBRBDF.formatNodeLabel)(object.typeDxn),
-        properties: object.data
-      });
-    }
-    for (const object of Object.values(this.objects)) {
-      const schema = this.schema[object.typeDxn].json;
-      const typenameDxn = schema.$id ?? (0, import_debug.raise)(new Error("Schema must have $id"));
-      for (const [name, propSchema] of Object.entries(schema.properties)) {
-        if (propSchema.$id === import_echo_schema2.JSON_SCHEMA_ECHO_REF_ID) {
-          const label = (0, import_chunk_IBOBRBDF.formatInferredRelationshipLabel)(typenameDxn, name);
-          const source = this.nodes.find((node) => node.id === object.id);
-          if (!source) {
-            continue;
-          }
-          const { id: targetId } = object.data[name].target;
-          const target = this.nodes.find((node) => node.id === targetId);
-          if (!target) {
-            continue;
-          }
-          this.relationships.push({
-            kind: "relationship",
-            id: `${object.id}-${name}-${target.id}`,
-            label,
-            properties: {},
-            source,
-            target
-          });
-        } else if (propSchema.type === "array" && propSchema.items && typeof propSchema.items === "object" && "$id" in propSchema.items && propSchema.items.$id === import_echo_schema2.JSON_SCHEMA_ECHO_REF_ID) {
-          const label = (0, import_chunk_IBOBRBDF.formatInferredRelationshipLabel)(typenameDxn, name);
-          const source = this.nodes.find((node) => node.id === object.id);
-          if (!source) {
-            continue;
-          }
-          for (const { id: targetId } of object.data[name].map((x) => x.target)) {
-            const target = this.nodes.find((node) => node.id === targetId);
-            if (!target) {
-              continue;
-            }
-            this.relationships.push({
-              kind: "relationship",
-              id: `${object.id}-${name}-${targetId}`,
-              label,
-              properties: {},
-              source,
-              target
-            });
-          }
-        }
-      }
-    }
-  }
-};
-var ObjectId2 = import_effect2.Schema.String.annotations({
-  description: "The unique object identifier."
-});
-var Contact = import_effect2.Schema.Struct({
-  id: ObjectId2,
-  name: import_effect2.Schema.String.annotations({
-    description: "The name of the person."
-  }),
-  email: import_effect2.Schema.optional(import_effect2.Schema.String).annotations({
-    description: "Email address."
-  })
-}).pipe(import_effect2.Schema.mutable).annotations({
-  [import_echo_schema3.TypeAnnotationId]: {
-    kind: import_echo_schema3.EntityKind.Object,
-    typename: "example.com/type/Contact",
-    version: "0.1.0"
-  },
-  description: "Contact information."
-});
-var Project = import_effect2.Schema.Struct({
-  id: ObjectId2,
-  name: import_effect2.Schema.String.annotations({
-    description: "The name of the project."
-  }),
-  description: import_effect2.Schema.optional(import_effect2.Schema.String).annotations({
-    description: "The description of the project."
-  })
-}).pipe(import_effect2.Schema.mutable).annotations({
-  [import_echo_schema3.TypeAnnotationId]: {
-    kind: import_echo_schema3.EntityKind.Object,
-    typename: "example.com/type/Project",
-    version: "0.1.0"
-  },
-  description: "Contact information."
-});
-var Task = import_effect2.Schema.Struct({
-  id: ObjectId2,
-  name: import_effect2.Schema.String.annotations({
-    description: "The name of the task."
-  }),
-  description: import_effect2.Schema.optional(import_effect2.Schema.String).annotations({
-    description: "The description of the task."
-  }),
-  project: (0, import_echo_schema3.Ref)(Project),
-  assignee: (0, import_echo_schema3.Ref)(Contact)
-}).pipe(import_effect2.Schema.mutable).annotations({
-  [import_echo_schema3.TypeAnnotationId]: {
-    kind: import_echo_schema3.EntityKind.Object,
-    typename: "example.com/type/Task",
-    version: "0.1.0"
-  },
-  description: "Contact information."
-});
-var Org = import_effect2.Schema.Struct({
-  id: ObjectId2,
-  name: import_effect2.Schema.String.annotations({
-    description: "The name of the organization."
-  }),
-  projects: import_effect2.Schema.Array((0, import_echo_schema3.Ref)(Project)),
-  employees: import_effect2.Schema.Array((0, import_echo_schema3.Ref)(Contact))
-}).pipe(import_effect2.Schema.mutable).annotations({
-  [import_echo_schema3.TypeAnnotationId]: {
-    kind: import_echo_schema3.EntityKind.Object,
-    typename: "example.com/type/Org",
-    version: "0.1.0"
-  },
-  description: "Contact information."
-});
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
-  AI_SERVICE_ENDPOINT,
-  Contact,
-  MockDataSource,
-  ObjectId,
-  Org,
-  Project,
-  Task,
-  createCypherTool,
-  createLogger,
-  createSystemPrompt,
-  createTestData,
-  createTestOllamaClient,
-  seedTestData
-});
-//# sourceMappingURL=index.cjs.map