@dxos/assistant 0.8.2-main.fbd8ed0 → 0.8.2-staging.4d6ad0f

package/dist/lib/node-esm/testing/index.mjs

@@ -1,659 +0,0 @@
-import { createRequire } from 'node:module';const require = createRequire(import.meta.url);
-import {
-  DEFAULT_OLLAMA_MODEL,
-  OllamaClient,
-  executeQuery,
-  formatInferredRelationshipLabel,
-  formatJsonSchemaForLLM,
-  formatNodeLabel,
-  trim
-} from "../chunk-SAWYU3V5.mjs";
-
-// packages/experimental/assistant/src/testing/defs.ts
-var AI_SERVICE_ENDPOINT = {
-  LOCAL: "http://localhost:8788",
-  REMOTE: "https://ai-service.dxos.workers.dev"
-};
-
-// packages/experimental/assistant/src/testing/logger.ts
-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;
-      }
-    }
-  };
-};
-
-// packages/experimental/assistant/src/testing/ollama.ts
-var createTestOllamaClient = (options) => {
-  return new OllamaClient({
-    tools: options?.tools,
-    overrides: {
-      model: DEFAULT_OLLAMA_MODEL,
-      temperature: 0
-    }
-  });
-};
-
-// packages/experimental/assistant/src/testing/printer.ts
-import { inspect } from "node:util";
-var ConsolePrinter = class {
-  constructor() {
-    this.printMessage = (message) => {
-      console.log(`${message.role.toUpperCase()}
-`);
-      for (const content of message.content) {
-        this.printContentBlock(content);
-      }
-    };
-    this.printContentBlock = (content) => {
-      switch (content.type) {
-        case "text":
-          console.log(content.text);
-          break;
-        case "tool_use":
-          console.log(`\u2699\uFE0F [Tool Use] ${content.name} ${inspect(content.input, {
-            depth: null,
-            colors: true
-          })}`);
-          break;
-        case "tool_result": {
-          let data;
-          try {
-            data = JSON.parse(content.content);
-          } catch {
-            data = content.content;
-          }
-          console.log(`\u2699\uFE0F [Tool Result] ${content.toolUseId} ${inspect(data, {
-            depth: null,
-            colors: true
-          })}`);
-          break;
-        }
-      }
-    };
-  }
-};
-
-// packages/experimental/assistant/src/testing/query-promts.ts
-import { Schema } from "effect";
-import { defineTool, ToolResult } from "@dxos/artifact";
-import { toJsonSchema } from "@dxos/echo-schema";
-import { log } from "@dxos/log";
-var __dxlog_file = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/testing/query-promts.ts";
-var createCypherTool = (dataSource, schemaTypes = []) => 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: Schema.Struct({
-    query: 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 {
-      log("cypher query", {
-        query
-      }, {
-        F: __dxlog_file,
-        L: 47,
-        S: void 0,
-        C: (f, a) => f(...a)
-      });
-      const results = await executeQuery(dataSource, query);
-      log("query complete", {
-        results
-      }, {
-        F: __dxlog_file,
-        L: 49,
-        S: void 0,
-        C: (f, a) => f(...a)
-      });
-      return ToolResult.Success(results);
-    } catch (e) {
-      log.catch(e, void 0, {
-        F: __dxlog_file,
-        L: 52,
-        S: void 0,
-        C: (f, a) => f(...a)
-      });
-      return ToolResult.Error(e.message);
-    }
-  }
-});
-var createSystemPrompt = (schemaTypes) => 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>
-    ${formatJsonSchemaForLLM(schemaTypes.map((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.
-`;
-
-// packages/experimental/assistant/src/testing/test-data.ts
-import { raise } from "@dxos/debug";
-import { JSON_SCHEMA_ECHO_REF_ID, ObjectId, toJsonSchema as toJsonSchema2, Ref as RefImpl } from "@dxos/echo-schema";
-import { invariant } from "@dxos/invariant";
-import { live } from "@dxos/live-object";
-var __dxlog_file2 = "/home/runner/work/dxos/dxos/packages/experimental/assistant/src/testing/test-data.ts";
-var seedTestData = (db) => {
-  const contactRich = db.add(live(Contact, {
-    name: "Rich"
-  }));
-  const contactJosiah = db.add(live(Contact, {
-    name: "Josiah"
-  }));
-  const contactDima = db.add(live(Contact, {
-    name: "Dima"
-  }));
-  const contactFred = db.add(live(Contact, {
-    name: "Fred"
-  }));
-  const projectComposer = db.add(live(Project, {
-    name: "Composer"
-  }));
-  const projectEcho = db.add(live(Project, {
-    name: "ECHO"
-  }));
-  const projectDoodles = db.add(live(Project, {
-    name: "Doodles"
-  }));
-  const _taskComposer1 = db.add(live(Task, {
-    name: "Optimize startup performance",
-    project: RefImpl.make(projectComposer),
-    assignee: RefImpl.make(contactJosiah)
-  }));
-  const _taskComposer2 = db.add(live(Task, {
-    name: "Create form builder",
-    project: RefImpl.make(projectComposer),
-    assignee: RefImpl.make(contactRich)
-  }));
-  const _taskComposer3 = db.add(live(Task, {
-    name: "Add support for custom themes",
-    project: RefImpl.make(projectComposer),
-    assignee: RefImpl.make(contactJosiah)
-  }));
-  const _taskComposer5 = db.add(live(Task, {
-    name: "Implement community plugin",
-    project: RefImpl.make(projectComposer),
-    assignee: RefImpl.make(contactFred)
-  }));
-  const _taskComposer4 = db.add(live(Task, {
-    name: "Implement dark mode",
-    project: RefImpl.make(projectComposer),
-    assignee: RefImpl.make(contactRich)
-  }));
-  const _taskEcho1 = db.add(live(Task, {
-    name: "Implement cypher query engine",
-    project: RefImpl.make(projectEcho),
-    assignee: RefImpl.make(contactDima)
-  }));
-  const _taskEcho2 = db.add(live(Task, {
-    name: "Add schema editor",
-    project: RefImpl.make(projectEcho),
-    assignee: RefImpl.make(contactRich)
-  }));
-  const _taskDoodles1 = db.add(live(Task, {
-    name: "Add support for custom themes",
-    project: RefImpl.make(projectDoodles),
-    assignee: RefImpl.make(contactFred)
-  }));
-  const _taskDoodles2 = db.add(live(Task, {
-    name: "Implement dark mode",
-    project: RefImpl.make(projectDoodles),
-    assignee: RefImpl.make(contactJosiah)
-  }));
-  const _orgDxos = db.add(live(Organization, {
-    name: "DXOS",
-    employees: [
-      RefImpl.make(contactRich),
-      RefImpl.make(contactJosiah),
-      RefImpl.make(contactDima)
-    ],
-    projects: [
-      RefImpl.make(projectEcho)
-    ]
-  }));
-  const _orgBraneframe = db.add(live(Organization, {
-    name: "Braneframe",
-    employees: [
-      RefImpl.make(contactJosiah),
-      RefImpl.make(contactRich)
-    ],
-    projects: [
-      RefImpl.make(projectComposer)
-    ]
-  }));
-};
-var createTestData = () => {
-  const dataSource = new MockDataSource();
-  const contactRich = dataSource.add(Contact, {
-    id: ObjectId.random(),
-    name: "Rich"
-  });
-  const contactJosiah = dataSource.add(Contact, {
-    id: ObjectId.random(),
-    name: "Josiah"
-  });
-  const contactDima = dataSource.add(Contact, {
-    id: ObjectId.random(),
-    name: "Dima"
-  });
-  const contactFred = dataSource.add(Contact, {
-    id: ObjectId.random(),
-    name: "Fred"
-  });
-  const projectComposer = dataSource.add(Project, {
-    id: ObjectId.random(),
-    name: "Composer"
-  });
-  const projectEcho = dataSource.add(Project, {
-    id: ObjectId.random(),
-    name: "ECHO"
-  });
-  const projectDoodles = dataSource.add(Project, {
-    id: ObjectId.random(),
-    name: "Doodles"
-  });
-  const _taskComposer1 = dataSource.add(Task, {
-    id: ObjectId.random(),
-    name: "Optimize startup performance",
-    project: RefImpl.make(projectComposer),
-    assignee: RefImpl.make(contactJosiah)
-  });
-  const _taskComposer2 = dataSource.add(Task, {
-    id: ObjectId.random(),
-    name: "Create form builder",
-    project: RefImpl.make(projectComposer),
-    assignee: RefImpl.make(contactRich)
-  });
-  const _taskComposer3 = dataSource.add(Task, {
-    id: ObjectId.random(),
-    name: "Add support for custom themes",
-    project: RefImpl.make(projectComposer),
-    assignee: RefImpl.make(contactJosiah)
-  });
-  const _taskComposer5 = dataSource.add(Task, {
-    id: ObjectId.random(),
-    name: "Implement community plugin",
-    project: RefImpl.make(projectComposer),
-    assignee: RefImpl.make(contactFred)
-  });
-  const _taskComposer4 = dataSource.add(Task, {
-    id: ObjectId.random(),
-    name: "Implement dark mode",
-    project: RefImpl.make(projectComposer),
-    assignee: RefImpl.make(contactRich)
-  });
-  const _taskEcho1 = dataSource.add(Task, {
-    id: ObjectId.random(),
-    name: "Implement cypher query engine",
-    project: RefImpl.make(projectEcho),
-    assignee: RefImpl.make(contactDima)
-  });
-  const _taskEcho2 = dataSource.add(Task, {
-    id: ObjectId.random(),
-    name: "Add schema editor",
-    project: RefImpl.make(projectEcho),
-    assignee: RefImpl.make(contactRich)
-  });
-  const _taskDoodles1 = dataSource.add(Task, {
-    id: ObjectId.random(),
-    name: "Add support for custom themes",
-    project: RefImpl.make(projectDoodles),
-    assignee: RefImpl.make(contactFred)
-  });
-  const _taskDoodles2 = dataSource.add(Task, {
-    id: ObjectId.random(),
-    name: "Implement dark mode",
-    project: RefImpl.make(projectDoodles),
-    assignee: RefImpl.make(contactJosiah)
-  });
-  const _orgDxos = dataSource.add(Organization, {
-    id: ObjectId.random(),
-    name: "DXOS",
-    employees: [
-      RefImpl.make(contactRich),
-      RefImpl.make(contactJosiah),
-      RefImpl.make(contactDima)
-    ],
-    projects: [
-      RefImpl.make(projectEcho)
-    ]
-  });
-  const _orgBraneframe = dataSource.add(Organization, {
-    id: ObjectId.random(),
-    name: "Braneframe",
-    employees: [
-      RefImpl.make(contactJosiah),
-      RefImpl.make(contactRich)
-    ],
-    projects: [
-      RefImpl.make(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) {
-    invariant(typeof data.id === "string", "Data must have an id", {
-      F: __dxlog_file2,
-      L: 284,
-      S: this,
-      A: [
-        "typeof data.id === 'string'",
-        "'Data must have an id'"
-      ]
-    });
-    const jsonSchema = toJsonSchema2(schema);
-    if (!this.schema[jsonSchema.$id]) {
-      this.schema[jsonSchema.$id] = {
-        json: jsonSchema,
-        instance: schema
-      };
-    }
-    const typenameDxn = jsonSchema.$id ?? 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: 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 ?? raise(new Error("Schema must have $id"));
-      for (const [name, propSchema] of Object.entries(schema.properties)) {
-        if (propSchema.$id === JSON_SCHEMA_ECHO_REF_ID) {
-          const label = 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 === JSON_SCHEMA_ECHO_REF_ID) {
-          const label = 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
-            });
-          }
-        }
-      }
-    }
-  }
-};
-
-// packages/experimental/assistant/src/testing/test-schema.ts
-import { Schema as Schema2 } from "effect";
-import { EntityKind, TypeAnnotationId, Ref, ObjectId as ObjectId2 } from "@dxos/echo-schema";
-var Contact = Schema2.Struct({
-  id: ObjectId2,
-  name: Schema2.String.annotations({
-    description: "The name of the person."
-  }),
-  email: Schema2.optional(Schema2.String).annotations({
-    description: "Email address."
-  })
-}).pipe(Schema2.mutable).annotations({
-  [TypeAnnotationId]: {
-    kind: EntityKind.Object,
-    typename: "example.com/type/Contact",
-    version: "0.1.0"
-  },
-  description: "Contact information."
-});
-var Project = Schema2.Struct({
-  id: ObjectId2,
-  name: Schema2.String.annotations({
-    description: "The name of the project."
-  }),
-  description: Schema2.optional(Schema2.String).annotations({
-    description: "The description of the project."
-  })
-}).pipe(Schema2.mutable).annotations({
-  [TypeAnnotationId]: {
-    kind: EntityKind.Object,
-    typename: "example.com/type/Project",
-    version: "0.1.0"
-  },
-  description: "Contact information."
-});
-var Task = Schema2.Struct({
-  id: ObjectId2,
-  name: Schema2.String.annotations({
-    description: "The name of the task."
-  }),
-  description: Schema2.optional(Schema2.String).annotations({
-    description: "The description of the task."
-  }),
-  project: Ref(Project),
-  assignee: Ref(Contact)
-}).pipe(Schema2.mutable).annotations({
-  [TypeAnnotationId]: {
-    kind: EntityKind.Object,
-    typename: "example.com/type/Task",
-    version: "0.1.0"
-  },
-  description: "Contact information."
-});
-var Organization = Schema2.Struct({
-  id: ObjectId2,
-  name: Schema2.String.annotations({
-    description: "The name of the organization."
-  }),
-  projects: Schema2.Array(Ref(Project)),
-  employees: Schema2.Array(Ref(Contact))
-}).pipe(Schema2.mutable).annotations({
-  [TypeAnnotationId]: {
-    kind: EntityKind.Object,
-    typename: "example.com/type/Organization",
-    version: "0.1.0"
-  },
-  description: "Contact information."
-});
-export {
-  AI_SERVICE_ENDPOINT,
-  ConsolePrinter,
-  Contact,
-  MockDataSource,
-  Organization,
-  Project,
-  Task,
-  createCypherTool,
-  createLogger,
-  createSystemPrompt,
-  createTestData,
-  createTestOllamaClient,
-  seedTestData
-};
-//# sourceMappingURL=index.mjs.map