@dxos/assistant 0.7.4-df37807

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

@@ -0,0 +1,619 @@
+"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, {
+  Contact: () => Contact,
+  MockDataSource: () => MockDataSource,
+  ObjectId: () => ObjectId2,
+  Org: () => Org,
+  Project: () => Project,
+  Task: () => Task,
+  createCypherTool: () => createCypherTool,
+  createLogger: () => createLogger,
+  createSystemPrompt: () => createSystemPrompt,
+  createTestData: () => createTestData,
+  seedTestData: () => seedTestData
+});
+module.exports = __toCommonJS(testing_exports);
+var import_chunk_G6CGWEMO = require("../chunk-G6CGWEMO.cjs");
+var import_schema = require("@effect/schema");
+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_schema2 = require("@effect/schema");
+var import_echo_schema3 = require("@dxos/echo-schema");
+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} media_type=${content.source?.media_type}]`);
+          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 __dxlog_file = "/Users/yarolegovich/Documents/Workspace/dxos/packages/experimental/assistant/src/testing/query-promts.ts";
+var createCypherTool = (dataSource, schemaTypes = []) => (0, import_chunk_G6CGWEMO.defineTool)({
+  name: "graphQuery",
+  description: "Query the ECHO graph database in cypher query language. Returns data from the database." + (schemaTypes.length > 0 ? `
+
+${createSystemPrompt(schemaTypes)}` : ""),
+  schema: import_schema.Schema.Struct({
+    query: import_schema.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_G6CGWEMO.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_chunk_G6CGWEMO.LLMToolResult.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_chunk_G6CGWEMO.LLMToolResult.Error(e.message);
+    }
+  }
+});
+var createSystemPrompt = (schemaTypes) => import_chunk_G6CGWEMO.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_G6CGWEMO.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 = "/Users/yarolegovich/Documents/Workspace/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_G6CGWEMO.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_G6CGWEMO.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?.$id === import_echo_schema2.JSON_SCHEMA_ECHO_REF_ID) {
+          const label = (0, import_chunk_G6CGWEMO.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_schema2.Schema.String.annotations({
+  description: "The unique object identifier."
+});
+var Contact = import_schema2.Schema.Struct({
+  id: ObjectId2,
+  name: import_schema2.Schema.String.annotations({
+    description: "The name of the person."
+  }),
+  email: import_schema2.Schema.optional(import_schema2.Schema.String).annotations({
+    description: "Email address."
+  })
+}).pipe(import_schema2.Schema.mutable).annotations({
+  [import_echo_schema3.ObjectAnnotationId]: {
+    kind: import_echo_schema3.EntityKind.Object,
+    typename: "example.com/type/Contact",
+    version: "0.1.0"
+  },
+  description: "Contact information."
+});
+var Project = import_schema2.Schema.Struct({
+  id: ObjectId2,
+  name: import_schema2.Schema.String.annotations({
+    description: "The name of the project."
+  }),
+  description: import_schema2.Schema.optional(import_schema2.Schema.String).annotations({
+    description: "The description of the project."
+  })
+}).pipe(import_schema2.Schema.mutable).annotations({
+  [import_echo_schema3.ObjectAnnotationId]: {
+    kind: import_echo_schema3.EntityKind.Object,
+    typename: "example.com/type/Project",
+    version: "0.1.0"
+  },
+  description: "Contact information."
+});
+var Task = import_schema2.Schema.Struct({
+  id: ObjectId2,
+  name: import_schema2.Schema.String.annotations({
+    description: "The name of the task."
+  }),
+  description: import_schema2.Schema.optional(import_schema2.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_schema2.Schema.mutable).annotations({
+  [import_echo_schema3.ObjectAnnotationId]: {
+    kind: import_echo_schema3.EntityKind.Object,
+    typename: "example.com/type/Task",
+    version: "0.1.0"
+  },
+  description: "Contact information."
+});
+var Org = import_schema2.Schema.Struct({
+  id: ObjectId2,
+  name: import_schema2.Schema.String.annotations({
+    description: "The name of the organization."
+  }),
+  projects: import_schema2.Schema.Array((0, import_echo_schema3.Ref)(Project)),
+  employees: import_schema2.Schema.Array((0, import_echo_schema3.Ref)(Contact))
+}).pipe(import_schema2.Schema.mutable).annotations({
+  [import_echo_schema3.ObjectAnnotationId]: {
+    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 = {
+  Contact,
+  MockDataSource,
+  ObjectId,
+  Org,
+  Project,
+  Task,
+  createCypherTool,
+  createLogger,
+  createSystemPrompt,
+  createTestData,
+  seedTestData
+});
+//# sourceMappingURL=index.cjs.map