@dxos/assistant 0.7.4-df37807

package/dist/types/src/cypher/execution-plan.d.ts

@@ -0,0 +1,49 @@
+import type { JsonPath } from '@dxos/echo-schema';
+import type { ParsedQuery } from './parser';
+export type VariableName = string & {
+    __VariableName: never;
+};
+export type ExecutionPlanStep = {
+    type: 'NodeScan';
+    label: string;
+    into: VariableName;
+} | {
+    type: 'FilterByLabel';
+    variable: VariableName;
+    label: string;
+} | {
+    type: 'FilterByProperty';
+    variable: VariableName;
+    property: JsonPath;
+    value: unknown;
+} | {
+    /**
+     * Scans (anchor)-[relationshipInto:label]->(nodeInto).
+     */
+    type: 'RelationshipScan';
+    label: string;
+    anchor: VariableName;
+    relationshipInto: VariableName;
+    nodeInto: VariableName;
+    /**
+     * `in`: (anchor)<-[relationshipInto:label]-(nodeInto)
+     * `out`: (anchor)-[relationshipInto:label]->(nodeInto)
+     * `both`: (anchor)-[relationshipInto:label]-(nodeInto)
+     */
+    direction: 'in' | 'out' | 'both';
+} | {
+    type: 'ProduceResults';
+    fields: {
+        name: string;
+        variable: VariableName;
+        path: JsonPath;
+    }[];
+};
+export type ExecutionPlan = {
+    steps: ExecutionPlanStep[];
+    output: {
+        fields: string[];
+    };
+};
+export declare const createExecutionPlan: (query: ParsedQuery) => ExecutionPlan;
+//# sourceMappingURL=execution-plan.d.ts.map

package/dist/types/src/cypher/execution-plan.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"execution-plan.d.ts","sourceRoot":"","sources":["../../../../src/cypher/execution-plan.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAKlD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAE5C,MAAM,MAAM,YAAY,GAAG,MAAM,GAAG;IAAE,cAAc,EAAE,KAAK,CAAA;CAAE,CAAC;AAE9D,MAAM,MAAM,iBAAiB,GACzB;IACE,IAAI,EAAE,UAAU,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,YAAY,CAAC;CACpB,GACD;IACE,IAAI,EAAE,eAAe,CAAC;IACtB,QAAQ,EAAE,YAAY,CAAC;IACvB,KAAK,EAAE,MAAM,CAAC;CACf,GACD;IACE,IAAI,EAAE,kBAAkB,CAAC;IACzB,QAAQ,EAAE,YAAY,CAAC;IACvB,QAAQ,EAAE,QAAQ,CAAC;IACnB,KAAK,EAAE,OAAO,CAAC;CAChB,GACD;IACE;;OAEG;IACH,IAAI,EAAE,kBAAkB,CAAC;IACzB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,YAAY,CAAC;IACrB,gBAAgB,EAAE,YAAY,CAAC;IAC/B,QAAQ,EAAE,YAAY,CAAC;IAEvB;;;;OAIG;IACH,SAAS,EAAE,IAAI,GAAG,KAAK,GAAG,MAAM,CAAC;CAClC,GACD;IACE,IAAI,EAAE,gBAAgB,CAAC;IACvB,MAAM,EAAE;QACN,IAAI,EAAE,MAAM,CAAC;QACb,QAAQ,EAAE,YAAY,CAAC;QACvB,IAAI,EAAE,QAAQ,CAAC;KAChB,EAAE,CAAC;CACL,CAAC;AAEN,MAAM,MAAM,aAAa,GAAG;IAC1B,KAAK,EAAE,iBAAiB,EAAE,CAAC;IAC3B,MAAM,EAAE;QACN,MAAM,EAAE,MAAM,EAAE,CAAC;KAClB,CAAC;CACH,CAAC;AAEF,eAAO,MAAM,mBAAmB,UAAW,WAAW,KAAG,aAExD,CAAC"}

package/dist/types/src/cypher/execution-plan.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=execution-plan.test.d.ts.map

package/dist/types/src/cypher/execution-plan.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"execution-plan.test.d.ts","sourceRoot":"","sources":["../../../../src/cypher/execution-plan.test.ts"],"names":[],"mappings":""}

package/dist/types/src/cypher/index.d.ts

@@ -0,0 +1,5 @@
+export * from './parser';
+export * from './query-executor';
+export * from './schema';
+export * from './echo-data-source';
+//# sourceMappingURL=index.d.ts.map

package/dist/types/src/cypher/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/cypher/index.ts"],"names":[],"mappings":"AAIA,cAAc,UAAU,CAAC;AACzB,cAAc,kBAAkB,CAAC;AACjC,cAAc,UAAU,CAAC;AACzB,cAAc,oBAAoB,CAAC"}

package/dist/types/src/cypher/parjs.d.ts

@@ -0,0 +1,4 @@
+import { parjsCombinators, parjs } from '#parjs';
+import type { Parjser } from 'parjs';
+export { type Parjser, parjsCombinators, parjs };
+//# sourceMappingURL=parjs.d.ts.map

package/dist/types/src/cypher/parjs.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"parjs.d.ts","sourceRoot":"","sources":["../../../../src/cypher/parjs.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,gBAAgB,EAAE,KAAK,EAAE,MAAM,QAAQ,CAAC;AAEjD,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AAErC,OAAO,EAAE,KAAK,OAAO,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC"}

package/dist/types/src/cypher/parser.d.ts

@@ -0,0 +1,6 @@
+import type { CypherQuery } from './ast';
+export type ParsedQuery = {
+    ast: CypherQuery;
+};
+export declare const parseCypherQuery: (query: string) => ParsedQuery;
+//# sourceMappingURL=parser.d.ts.map

package/dist/types/src/cypher/parser.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"parser.d.ts","sourceRoot":"","sources":["../../../../src/cypher/parser.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,OAAO,CAAC;AAGzC,MAAM,MAAM,WAAW,GAAG;IACxB,GAAG,EAAE,WAAW,CAAC;CAClB,CAAC;AAEF,eAAO,MAAM,gBAAgB,UAAW,MAAM,KAAG,WAOhD,CAAC"}

package/dist/types/src/cypher/query-executor.d.ts

@@ -0,0 +1,26 @@
+import { type ExecutionPlan } from './execution-plan';
+export interface DataSource {
+    getNodes: (params: {
+        label?: string;
+    }) => Promise<Node[]>;
+    getRelationships: (params: {
+        label?: string;
+    }) => Promise<Relationship[]>;
+}
+export type Node = {
+    id: string;
+    kind: 'node';
+    label: string;
+    properties: Record<string, unknown>;
+};
+export type Relationship = {
+    id: string;
+    kind: 'relationship';
+    label: string;
+    properties: Record<string, unknown>;
+    source: Node;
+    target: Node;
+};
+export declare const executeQueryPlan: (dataSource: DataSource, plan: ExecutionPlan) => Promise<Record<string, unknown>[]>;
+export declare const executeQuery: (dataSource: DataSource, query: string) => Promise<Record<string, unknown>[]>;
+//# sourceMappingURL=query-executor.d.ts.map

package/dist/types/src/cypher/query-executor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"query-executor.d.ts","sourceRoot":"","sources":["../../../../src/cypher/query-executor.ts"],"names":[],"mappings":"AAQA,OAAO,EAAuB,KAAK,aAAa,EAAqB,MAAM,kBAAkB,CAAC;AAG9F,MAAM,WAAW,UAAU;IACzB,QAAQ,EAAE,CAAC,MAAM,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,KAAK,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IAC1D,gBAAgB,EAAE,CAAC,MAAM,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,KAAK,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;CAC3E;AAED,MAAM,MAAM,IAAI,GAAG;IACjB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACrC,CAAC;AAEF,MAAM,MAAM,YAAY,GAAG;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,cAAc,CAAC;IACrB,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACpC,MAAM,EAAE,IAAI,CAAC;IACb,MAAM,EAAE,IAAI,CAAC;CACd,CAAC;AAEF,eAAO,MAAM,gBAAgB,eACf,UAAU,QAChB,aAAa,KAClB,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAsFnC,CAAC;AAEF,eAAO,MAAM,YAAY,eAAgB,UAAU,SAAS,MAAM,KAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAGrG,CAAC"}

package/dist/types/src/cypher/query-executor.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=query-executor.test.d.ts.map

package/dist/types/src/cypher/query-executor.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"query-executor.test.d.ts","sourceRoot":"","sources":["../../../../src/cypher/query-executor.test.ts"],"names":[],"mappings":""}

package/dist/types/src/cypher/schema.d.ts

@@ -0,0 +1,12 @@
+import { type JsonSchemaType } from '@dxos/echo-schema';
+/**
+ * Formats ECHO schema to be consumed by LLM.
+ * Schema is formatted in cypher format.
+ * Objects are represented as nodes.
+ * References are represented as relationships.
+ */
+export declare const formatJsonSchemaForLLM: (schemaTypes: JsonSchemaType[]) => string[];
+export declare const formatNodeLabel: (typename: string) => string;
+export declare const formatInferredRelationshipLabel: (typename: string, property: string) => string;
+export declare const isReference: (schema: JsonSchemaType) => boolean;
+//# sourceMappingURL=schema.d.ts.map

package/dist/types/src/cypher/schema.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.d.ts","sourceRoot":"","sources":["../../../../src/cypher/schema.ts"],"names":[],"mappings":"AAKA,OAAO,EAA2B,KAAK,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAKjF;;;;;GAKG;AACH,eAAO,MAAM,sBAAsB,gBAAiB,cAAc,EAAE,KAAG,MAAM,EAK5E,CAAC;AAkDF,eAAO,MAAM,eAAe,aAAc,MAAM,WAE/C,CAAC;AAEF,eAAO,MAAM,+BAA+B,aAAc,MAAM,YAAY,MAAM,WAGjF,CAAC;AAEF,eAAO,MAAM,WAAW,WAAY,cAAc,YACoE,CAAC"}

package/dist/types/src/cypher/schema.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=schema.test.d.ts.map

package/dist/types/src/cypher/schema.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.test.d.ts","sourceRoot":"","sources":["../../../../src/cypher/schema.test.ts"],"names":[],"mappings":""}

package/dist/types/src/index.d.ts

@@ -0,0 +1,4 @@
+export * from './ai-service';
+export * from './cypher';
+export * from './conversation';
+//# sourceMappingURL=index.d.ts.map

package/dist/types/src/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/index.ts"],"names":[],"mappings":"AAIA,cAAc,cAAc,CAAC;AAC7B,cAAc,UAAU,CAAC;AACzB,cAAc,gBAAgB,CAAC"}

package/dist/types/src/sanity.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=sanity.test.d.ts.map

package/dist/types/src/sanity.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sanity.test.d.ts","sourceRoot":"","sources":["../../../src/sanity.test.ts"],"names":[],"mappings":""}

package/dist/types/src/testing/index.d.ts

@@ -0,0 +1,5 @@
+export * from './logger';
+export * from './query-promts';
+export * from './test-data';
+export * from './test-schema';
+//# sourceMappingURL=index.d.ts.map

package/dist/types/src/testing/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/testing/index.ts"],"names":[],"mappings":"AAIA,cAAc,UAAU,CAAC;AACzB,cAAc,gBAAgB,CAAC;AAC/B,cAAc,aAAa,CAAC;AAC5B,cAAc,eAAe,CAAC"}

package/dist/types/src/testing/logger.d.ts

@@ -0,0 +1,12 @@
+import type { ImageSource } from '../ai-service';
+import type { ConversationEvent } from '../conversation';
+export declare const createLogger: ({ stream, filter, onImage, }?: {
+    stream?: boolean;
+    filter?: (event: ConversationEvent) => boolean;
+    onImage?: (img: {
+        id: string;
+        description?: string;
+        source: ImageSource;
+    }) => void;
+}) => (event: ConversationEvent) => void;
+//# sourceMappingURL=logger.d.ts.map

package/dist/types/src/testing/logger.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../../../../src/testing/logger.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,WAAW,EAAuB,MAAM,eAAe,CAAC;AACtE,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AAEzD,eAAO,MAAM,YAAY,kCAItB;IACD,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,iBAAiB,KAAK,OAAO,CAAC;IAC/C,OAAO,CAAC,EAAE,CAAC,GAAG,EAAE;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,WAAW,CAAA;KAAE,KAAK,IAAI,CAAC;CACpF,aAIgB,iBAAiB,SA4GjC,CAAC"}

package/dist/types/src/testing/object-extraction.test.d.ts

@@ -0,0 +1 @@
+//# sourceMappingURL=object-extraction.test.d.ts.map

package/dist/types/src/testing/object-extraction.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"object-extraction.test.d.ts","sourceRoot":"","sources":["../../../../src/testing/object-extraction.test.ts"],"names":[],"mappings":""}

package/dist/types/src/testing/query-promts.d.ts

@@ -0,0 +1,5 @@
+import { Schema as S } from '@effect/schema';
+import { type DataSource } from '../cypher';
+export declare const createCypherTool: (dataSource: DataSource, schemaTypes?: S.Schema.Any[]) => import("../conversation").LLMToolDefinition;
+export declare const createSystemPrompt: (schemaTypes: S.Schema.Any[]) => string;
+//# sourceMappingURL=query-promts.d.ts.map

package/dist/types/src/testing/query-promts.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"query-promts.d.ts","sourceRoot":"","sources":["../../../../src/testing/query-promts.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,MAAM,IAAI,CAAC,EAAE,MAAM,gBAAgB,CAAC;AAM7C,OAAO,EAAwC,KAAK,UAAU,EAAE,MAAM,WAAW,CAAC;AAGlF,eAAO,MAAM,gBAAgB,eAAgB,UAAU,gBAAe,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,gDA0ChF,CAAC;AAEL,eAAO,MAAM,kBAAkB,gBAAiB,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,WAiB7D,CAAC"}

package/dist/types/src/testing/query.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=query.test.d.ts.map

package/dist/types/src/testing/query.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"query.test.d.ts","sourceRoot":"","sources":["../../../../src/testing/query.test.ts"],"names":[],"mappings":""}

package/dist/types/src/testing/test-data.d.ts

@@ -0,0 +1,29 @@
+import { type Schema as S } from '@effect/schema';
+import { type JsonSchemaType } from '@dxos/echo-schema';
+import { type DataSource, type Node, type Relationship } from '../cypher';
+import type { EchoDatabase } from '@dxos/echo-db';
+export declare const seedTestData: (db: EchoDatabase) => void;
+export declare const createTestData: () => MockDataSource;
+export declare class MockDataSource implements DataSource {
+    objects: Record<string, {
+        schema: S.Schema.All;
+        typeDxn: string;
+        id: string;
+        data: unknown;
+    }>;
+    schema: Record<string, {
+        json: JsonSchemaType;
+        instance: S.Schema.All;
+    }>;
+    nodes: Node[];
+    relationships: Relationship[];
+    getNodes({ label }: {
+        label?: string;
+    }): Promise<Node[]>;
+    getRelationships({ label }: {
+        label?: string;
+    }): Promise<Relationship[]>;
+    add<S extends S.Schema.All>(schema: S, data: S.Schema.Type<S>): S.Schema.Type<S>;
+    computeGraph(): void;
+}
+//# sourceMappingURL=test-data.d.ts.map

package/dist/types/src/testing/test-data.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-data.d.ts","sourceRoot":"","sources":["../../../../src/testing/test-data.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,KAAK,MAAM,IAAI,CAAC,EAAE,MAAM,gBAAgB,CAAC;AAGlD,OAAO,EAAmD,KAAK,cAAc,EAAY,MAAM,mBAAmB,CAAC;AAInH,OAAO,EACL,KAAK,UAAU,EACf,KAAK,IAAI,EACT,KAAK,YAAY,EAGlB,MAAM,WAAW,CAAC;AAEnB,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAElD,eAAO,MAAM,YAAY,OAAQ,YAAY,SAoH5C,CAAC;AAGF,eAAO,MAAM,cAAc,QAAO,cAuGjC,CAAC;AAEF,qBAAa,cAAe,YAAW,UAAU;IAC/C,OAAO,EAAE,MAAM,CACb,MAAM,EACN;QACE,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;QACrB,OAAO,EAAE,MAAM,CAAC;QAChB,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,OAAO,CAAC;KACf,CACF,CAAM;IAEP,MAAM,EAAE,MAAM,CACZ,MAAM,EACN;QACE,IAAI,EAAE,cAAc,CAAC;QACrB,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;KACxB,CACF,CAAM;IAEP,KAAK,EAAE,IAAI,EAAE,CAAM;IACnB,aAAa,EAAE,YAAY,EAAE,CAAM;IAE7B,QAAQ,CAAC,EAAE,KAAK,EAAE,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE;IAItC,gBAAgB,CAAC,EAAE,KAAK,EAAE,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE;IAIpD,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;IAsB7D,YAAY;CAoEb"}

package/dist/types/src/testing/test-schema.d.ts

@@ -0,0 +1,50 @@
+import { Schema as S } from '@effect/schema';
+export declare const ObjectId: S.SchemaClass<string, string, never>;
+export declare const Contact: S.mutable<S.Struct<{
+    id: S.SchemaClass<string, string, never>;
+    name: S.SchemaClass<string, string, never>;
+    email: S.optional<typeof S.String>;
+}>>;
+export interface Contact extends S.Schema.Type<typeof Contact> {
+}
+export declare const Project: S.mutable<S.Struct<{
+    id: S.SchemaClass<string, string, never>;
+    name: S.SchemaClass<string, string, never>;
+    description: S.optional<typeof S.String>;
+}>>;
+export interface Project extends S.Schema.Type<typeof Project> {
+}
+export declare const Task: S.mutable<S.Struct<{
+    id: S.SchemaClass<string, string, never>;
+    name: S.SchemaClass<string, string, never>;
+    description: S.optional<typeof S.String>;
+    project: import("@dxos/echo-schema").Ref$<{
+        id: string;
+        name: string;
+        description?: string | undefined;
+    }>;
+    assignee: import("@dxos/echo-schema").Ref$<{
+        id: string;
+        name: string;
+        email?: string | undefined;
+    }>;
+}>>;
+export interface Task extends S.Schema.Type<typeof Task> {
+}
+export declare const Org: S.mutable<S.Struct<{
+    id: S.SchemaClass<string, string, never>;
+    name: S.SchemaClass<string, string, never>;
+    projects: S.Array$<import("@dxos/echo-schema").Ref$<{
+        id: string;
+        name: string;
+        description?: string | undefined;
+    }>>;
+    employees: S.Array$<import("@dxos/echo-schema").Ref$<{
+        id: string;
+        name: string;
+        email?: string | undefined;
+    }>>;
+}>>;
+export interface Org extends S.Schema.Type<typeof Org> {
+}
+//# sourceMappingURL=test-schema.d.ts.map

package/dist/types/src/testing/test-schema.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-schema.d.ts","sourceRoot":"","sources":["../../../../src/testing/test-schema.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,MAAM,IAAI,CAAC,EAAE,MAAM,gBAAgB,CAAC;AAQ7C,eAAO,MAAM,QAAQ,sCAAyE,CAAC;AAE/F,eAAO,MAAM,OAAO;;;;GAahB,CAAC;AACL,MAAM,WAAW,OAAQ,SAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,OAAO,CAAC;CAAG;AAEjE,eAAO,MAAM,OAAO;;;;GAahB,CAAC;AACL,MAAM,WAAW,OAAQ,SAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,OAAO,CAAC;CAAG;AAEjE,eAAO,MAAM,IAAI;;;;;;;;;;;;;;GAeb,CAAC;AACL,MAAM,WAAW,IAAK,SAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC;CAAG;AAE3D,eAAO,MAAM,GAAG;;;;;;;;;;;;;GAcZ,CAAC;AACL,MAAM,WAAW,GAAI,SAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,GAAG,CAAC;CAAG"}

package/dist/types/src/util/index.d.ts

@@ -0,0 +1,2 @@
+export * from './trim';
+//# sourceMappingURL=index.d.ts.map

package/dist/types/src/util/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/util/index.ts"],"names":[],"mappings":"AAIA,cAAc,QAAQ,CAAC"}

package/dist/types/src/util/trim.d.ts

@@ -0,0 +1,5 @@
+/**
+ * Trims multiline strings.
+ */
+export declare const trim: (strings: TemplateStringsArray, ...values: any[]) => string;
+//# sourceMappingURL=trim.d.ts.map

package/dist/types/src/util/trim.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"trim.d.ts","sourceRoot":"","sources":["../../../../src/util/trim.ts"],"names":[],"mappings":"AAIA;;GAEG;AAEH,eAAO,MAAM,IAAI,YAAa,oBAAoB,aAAa,GAAG,EAAE,WAMnE,CAAC"}

package/dist/types/tsconfig.tsbuildinfo

@@ -0,0 +1 @@
+{"version":"5.7.2"}

package/package.json

@@ -0,0 +1,64 @@
+{
+  "name": "@dxos/assistant",
+  "version": "0.7.4-df37807",
+  "description": "Natural language interface to DXOS",
+  "homepage": "https://dxos.org",
+  "bugs": "https://github.com/dxos/dxos/issues",
+  "license": "MIT",
+  "author": "info@dxos.org",
+  "sideEffects": true,
+  "type": "module",
+  "imports": {
+    "#parjs": {
+      "browser": "./vendor/parjs-browser.mjs",
+      "default": "./vendor/parjs-default.mjs"
+    },
+    "#parjs/combinators": {
+      "browser": "./vendor/parjs-combinators-browser.mjs",
+      "default": "./vendor/parjs-combinators-default.mjs"
+    }
+  },
+  "exports": {
+    ".": {
+      "types": "./dist/types/src/index.d.ts",
+      "browser": "./dist/lib/browser/index.mjs",
+      "node": "./dist/lib/node-esm/index.mjs"
+    },
+    "./testing": {
+      "types": "./dist/types/src/testing/index.d.ts",
+      "browser": "./dist/lib/browser/testing/index.mjs",
+      "node": "./dist/lib/node-esm/testing/index.mjs"
+    }
+  },
+  "types": "dist/types/src/index.d.ts",
+  "files": [
+    "dist",
+    "src"
+  ],
+  "scripts": {
+    "run:chat": "tsx src/bin/chat.ts"
+  },
+  "dependencies": {
+    "@dxos/async": "workspace:*",
+    "@dxos/context": "workspace:*",
+    "@dxos/debug": "workspace:*",
+    "@dxos/echo-db": "workspace:*",
+    "@dxos/echo-schema": "workspace:*",
+    "@dxos/invariant": "workspace:*",
+    "@dxos/keys": "workspace:*",
+    "@dxos/live-object": "workspace:*",
+    "@dxos/log": "workspace:*",
+    "@dxos/node-std": "workspace:*",
+    "@dxos/util": "workspace:*",
+    "@effect/schema": "^0.75.5",
+    "parjs": "^1.3.9",
+    "ulidx": "^2.3.0"
+  },
+  "devDependencies": {
+    "inquirer": "^12.0.0"
+  },
+  "publishConfig": {
+    "access": "restricted"
+  },
+  "beast": {}
+}

package/src/ai-service/client.test.ts

@@ -0,0 +1,162 @@
+//
+// Copyright 2024 DXOS.org
+//
+
+import { Schema as S } from '@effect/schema';
+import { test, describe } from 'vitest';
+
+import { toJsonSchema } from '@dxos/echo-schema';
+import { invariant } from '@dxos/invariant';
+import { SpaceId } from '@dxos/keys';
+import { log } from '@dxos/log';
+
+import { AIServiceClientImpl } from './client';
+import { ObjectId, ToolTypes, type LLMTool } from './schema';
+
+const ENDPOINT = 'http://localhost:8787';
+
+describe('AI Service Client', () => {
+  test('client generation', async () => {
+    const client = new AIServiceClientImpl({
+      endpoint: ENDPOINT,
+    });
+
+    const spaceId = SpaceId.random();
+    const threadId = ObjectId.random();
+
+    await client.insertMessages([
+      {
+        id: ObjectId.random(),
+        spaceId,
+        threadId,
+        role: 'user',
+        content: [{ type: 'text', text: 'Hello' }],
+      },
+    ]);
+
+    const stream = await client.generate({
+      model: '@anthropic/claude-3-5-haiku-20241022',
+      spaceId,
+      threadId,
+      systemPrompt: 'You are a poet',
+      tools: [],
+    });
+    for await (const event of stream) {
+      log.info('event', event);
+    }
+
+    log.info('full message', {
+      message: await stream.complete(),
+    });
+  });
+
+  test('tool calls', async () => {
+    const client = new AIServiceClientImpl({
+      endpoint: ENDPOINT,
+    });
+
+    const custodian: LLMTool = {
+      name: 'custodian',
+      description: 'Custodian can tell you the password if you say the magic word',
+      parameters: toJsonSchema(
+        S.Struct({
+          magicWord: S.String.annotations({ description: 'The magic word. Should be exactly "pretty please"' }),
+        }),
+      ),
+    };
+
+    const spaceId = SpaceId.random();
+    const threadId = ObjectId.random();
+
+    await client.insertMessages([
+      {
+        id: ObjectId.random(),
+        spaceId,
+        threadId,
+        role: 'user',
+        content: [{ type: 'text', text: 'What is the password? Ask the custodian' }],
+      },
+    ]);
+
+    const stream = await client.generate({
+      model: '@anthropic/claude-3-5-haiku-20241022',
+      spaceId,
+      threadId,
+      systemPrompt: 'You are a helpful assistant.',
+      tools: [custodian],
+    });
+    for await (const event of stream) {
+      log.info('event', event);
+    }
+    const [message] = await stream.complete();
+    log.info('full message', {
+      message,
+    });
+    await client.insertMessages([message]);
+
+    const toolUse = message.content.find(({ type }) => type === 'tool_use')!;
+    invariant(toolUse.type === 'tool_use');
+    await client.insertMessages([
+      {
+        id: ObjectId.random(),
+        spaceId,
+        threadId,
+        role: 'user',
+        content: [{ type: 'tool_result', toolUseId: toolUse.id, content: 'password="The sky is gray"' }],
+      },
+    ]);
+
+    const stream2 = await client.generate({
+      model: '@anthropic/claude-3-5-haiku-20241022',
+      spaceId,
+      threadId,
+      systemPrompt: 'You are a helpful assistant.',
+      tools: [custodian],
+    });
+    for await (const event of stream2) {
+      log.info('event', event);
+    }
+    const [message2] = await stream2.complete();
+    log.info('full message', {
+      message: message2,
+    });
+  });
+
+  test.only('image generation', async () => {
+    const client = new AIServiceClientImpl({
+      endpoint: ENDPOINT,
+    });
+
+    const spaceId = SpaceId.random();
+    const threadId = ObjectId.random();
+
+    await client.insertMessages([
+      {
+        id: ObjectId.random(),
+        spaceId,
+        threadId,
+        role: 'user',
+        content: [{ type: 'text', text: 'Generate an image of a cat' }],
+      },
+    ]);
+
+    const stream = await client.generate({
+      model: '@anthropic/claude-3-5-sonnet-20241022',
+      spaceId,
+      threadId,
+      tools: [
+        {
+          name: 'text-to-image',
+          type: ToolTypes.TextToImage,
+        },
+      ],
+    });
+    for await (const event of stream) {
+      log.info('event', event);
+    }
+
+    log.info('full message', {
+      message: await stream.complete(),
+    });
+  });
+});