@dxos/assistant 0.7.4 → 0.7.5-feature-compute.4d9d99a

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

@@ -1,7 +1,7 @@
 {
   "version": 3,
-  "sources": ["../../../../src/testing/logger.ts", "../../../../src/testing/query-promts.ts", "../../../../src/testing/test-data.ts", "../../../../src/testing/test-schema.ts"],
-  "sourcesContent": ["//\n// Copyright 2024 DXOS.org\n//\n\nimport type { ImageSource, MessageContentBlock } from '../ai-service';\nimport type { ConversationEvent } from '../conversation';\n\nexport const createLogger = ({\n  stream,\n  filter,\n  onImage,\n}: {\n  stream?: boolean;\n  filter?: (event: ConversationEvent) => boolean;\n  onImage?: (img: { id: string; description?: string; source: ImageSource }) => void;\n} = {}) => {\n  const images: Record<string, any> = {};\n  let currentText = '';\n\n  return (event: ConversationEvent) => {\n    if (typeof filter === 'function' && !filter(event)) {\n      return;\n    }\n\n    const printContentBlock = (content: MessageContentBlock) => {\n      switch (content.type) {\n        case 'text':\n          process.stdout.write(content.text);\n          break;\n        case 'tool_use':\n          process.stdout.write(`⚙️ [Tool Use] ${content.name}\\n`);\n          break;\n        case 'image': {\n          if (content.id && content.source) {\n            images[content.id] = content.source;\n          }\n          process.stdout.write(`[Image id=${content.id} media_type=${content.source?.media_type}]`);\n          break;\n        }\n      }\n    };\n\n    const onTextBlockPrinted = (text: string) => {\n      const imageMatch = text.match(/<image id=\"([^\"]+)\"(?:\\s+prompt=\"([^\"]+)\")?\\s*\\/>/);\n      if (imageMatch && onImage && images[imageMatch[1]]) {\n        onImage({\n          id: imageMatch[1],\n          description: imageMatch[2], // Extract prompt as description if present\n          source: images[imageMatch[1]],\n        });\n      }\n    };\n\n    if (stream) {\n      switch (event.type) {\n        case 'message_start': {\n          process.stdout.write(`${event.message.role.toUpperCase()}\\n\\n`);\n          for (const content of event.message.content) {\n            printContentBlock(content);\n          }\n          break;\n        }\n        case 'content_block_start': {\n          currentText = '';\n          printContentBlock(event.content);\n          break;\n        }\n        case 'content_block_delta': {\n          switch (event.delta.type) {\n            case 'text_delta': {\n              currentText += event.delta.text;\n              process.stdout.write(event.delta.text);\n              break;\n            }\n            case 'input_json_delta': {\n              process.stdout.write(event.delta.partial_json);\n              break;\n            }\n          }\n          break;\n        }\n        case 'content_block_stop': {\n          process.stdout.write('\\n');\n          onTextBlockPrinted(currentText);\n          currentText = '';\n          break;\n        }\n        case 'message_delta': {\n          break;\n        }\n        case 'message_stop': {\n          process.stdout.write('\\n\\n');\n          break;\n        }\n      }\n    }\n\n    switch (event.type) {\n      case 'message': {\n        if (!stream || event.message.role !== 'assistant') {\n          process.stdout.write(`${event.message.role.toUpperCase()}\\n\\n`);\n          for (const block of event.message.content) {\n            switch (block.type) {\n              case 'text':\n                process.stdout.write(block.text + '\\n');\n                onTextBlockPrinted(block.text);\n                break;\n              case 'tool_use':\n                process.stdout.write(`⚙️ [Tool Use] ${block.name}\\n`);\n                process.stdout.write(`  ${JSON.stringify(block.input)}\\n`);\n                break;\n              case 'tool_result':\n                if (block.isError) {\n                  process.stdout.write('❌ [Tool Error]\\n');\n                  process.stdout.write(block.content + '\\n');\n                } else {\n                  process.stdout.write('✅ [Tool Success]\\n');\n                  process.stdout.write(block.content + '\\n');\n                }\n            }\n            process.stdout.write('\\n\\n');\n          }\n        }\n        break;\n      }\n    }\n  };\n};\n", "//\n// Copyright 2024 DXOS.org\n//\n\nimport { Schema as S } from '@effect/schema';\n\nimport { toJsonSchema } from '@dxos/echo-schema';\nimport { log } from '@dxos/log';\n\nimport { defineTool, LLMToolResult } from '../conversation';\nimport { executeQuery, formatJsonSchemaForLLM, type DataSource } from '../cypher';\nimport { trim } from '../util';\n\nexport const createCypherTool = (dataSource: DataSource, schemaTypes: S.Schema.Any[] = []) =>\n  defineTool({\n    name: 'graphQuery',\n    description:\n      'Query the ECHO graph database in cypher query language. Returns data from the database.' +\n      (schemaTypes.length > 0 ? `\\n\\n${createSystemPrompt(schemaTypes)}` : ''),\n    schema: S.Struct({\n      query: S.String.annotations({\n        description: `\n      A valid cypher query string to execute.\n      Query must have one MATCH clause.\n      Match clause can have multiple patterns but they must all be connected via a relationship chain.\n      Query might have zero or one WHERE clauses.\n      Query must have one RETURN clause.\n      RETURN clause can have multiple expressions separated by commas.\n      DISTINCT keyword is not allowed.\n\n      <example>\n      MATCH (n:Person) RETURN n.name\n      </example>\n\n      <example>\n      MATCH (s:Studio)-[:PRODUCED]->(m:Movie)-[:PRODUCED_IN]->(c:City {name: \"Las Vegas\"}) RETURN s\n      </example>\n      \n      <example>\n      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\n      </example>\n      `,\n      }),\n    }),\n    execute: async ({ query }) => {\n      try {\n        log('cypher query', { query });\n        const results = await executeQuery(dataSource, query);\n        log('query complete', { results });\n        return LLMToolResult.Success(results);\n      } catch (e: any) {\n        log.catch(e);\n        return LLMToolResult.Error(e.message);\n      }\n    },\n  });\n\nexport const createSystemPrompt = (schemaTypes: S.Schema.Any[]) => trim`\n  You have access to ECHO graph database. You can query the database to get data to satisfy user prompts.\n\n  Database schema is defined in pseud-cypher syntax. The schema is:\n\n  <schema>\n    ${formatJsonSchemaForLLM(schemaTypes.map((schema) => toJsonSchema(schema)))}\n  <schema>\n\n  Before replying always think step-by-step on how to proceed.\n  Print your thoughts inside <cot> tags.\n  \n  <example>\n    <cot>To answer the question I need to ...</cot>\n  </example>\n\n  Before answering the user's question, decide what tools you need to use to answer the question.\n`;\n", "//\n// Copyright 2024 DXOS.org\n//\n\nimport { type Schema as S } from '@effect/schema';\n\nimport { raise } from '@dxos/debug';\nimport { JSON_SCHEMA_ECHO_REF_ID, ObjectId, toJsonSchema, type JsonSchemaType, type Ref } from '@dxos/echo-schema';\nimport { invariant } from '@dxos/invariant';\nimport { create, makeRef } from '@dxos/live-object';\n\nimport {\n  type DataSource,\n  type Node,\n  type Relationship,\n  formatInferredRelationshipLabel,\n  formatNodeLabel,\n} from '../cypher';\nimport { Contact, Org, Project, Task } from '../testing';\nimport type { EchoDatabase } from '@dxos/echo-db';\n\nexport const seedTestData = (db: EchoDatabase) => {\n  const contactRich = db.add(\n    create(Contact, {\n      name: 'Rich',\n    }),\n  );\n  const contactJosiah = db.add(\n    create(Contact, {\n      name: 'Josiah',\n    }),\n  );\n  const contactDima = db.add(\n    create(Contact, {\n      name: 'Dima',\n    }),\n  );\n  const contactFred = db.add(\n    create(Contact, {\n      name: 'Fred',\n    }),\n  );\n\n  const projectComposer = db.add(\n    create(Project, {\n      name: 'Composer',\n    }),\n  );\n  const projectEcho = db.add(\n    create(Project, {\n      name: 'ECHO',\n    }),\n  );\n  const projectDoodles = db.add(\n    create(Project, {\n      name: 'Doodles',\n    }),\n  );\n\n  const _taskComposer1 = db.add(\n    create(Task, {\n      name: 'Optimize startup performance',\n      project: makeRef(projectComposer),\n      assignee: makeRef(contactJosiah),\n    }),\n  );\n  const _taskComposer2 = db.add(\n    create(Task, {\n      name: 'Create form builder',\n      project: makeRef(projectComposer),\n      assignee: makeRef(contactRich),\n    }),\n  );\n  const _taskComposer3 = db.add(\n    create(Task, {\n      name: 'Add support for custom themes',\n      project: makeRef(projectComposer),\n      assignee: makeRef(contactJosiah),\n    }),\n  );\n  const _taskComposer5 = db.add(\n    create(Task, {\n      name: 'Implement community plugin',\n      project: makeRef(projectComposer),\n      assignee: makeRef(contactFred),\n    }),\n  );\n  const _taskComposer4 = db.add(\n    create(Task, {\n      name: 'Implement dark mode',\n      project: makeRef(projectComposer),\n      assignee: makeRef(contactRich),\n    }),\n  );\n  const _taskEcho1 = db.add(\n    create(Task, {\n      name: 'Implement cypher query engine',\n      project: makeRef(projectEcho),\n      assignee: makeRef(contactDima),\n    }),\n  );\n  const _taskEcho2 = db.add(\n    create(Task, {\n      name: 'Add schema editor',\n      project: makeRef(projectEcho),\n      assignee: makeRef(contactRich),\n    }),\n  );\n  const _taskDoodles1 = db.add(\n    create(Task, {\n      name: 'Add support for custom themes',\n      project: makeRef(projectDoodles),\n      assignee: makeRef(contactFred),\n    }),\n  );\n  const _taskDoodles2 = db.add(\n    create(Task, {\n      name: 'Implement dark mode',\n      project: makeRef(projectDoodles),\n      assignee: makeRef(contactJosiah),\n    }),\n  );\n\n  const _orgDxos = db.add(\n    create(Org, {\n      name: 'DXOS',\n      employees: [makeRef(contactRich), makeRef(contactJosiah), makeRef(contactDima)],\n      projects: [makeRef(projectEcho)],\n    }),\n  );\n  const _orgBraneframe = db.add(\n    create(Org, {\n      name: 'Braneframe',\n      employees: [makeRef(contactJosiah), makeRef(contactRich)],\n      projects: [makeRef(projectComposer)],\n    }),\n  );\n};\n\n// TODO(burdon): Use schema/testing.\nexport const createTestData = (): MockDataSource => {\n  const dataSource = new MockDataSource();\n\n  const contactRich = dataSource.add(Contact, {\n    id: ObjectId.random(),\n    name: 'Rich',\n  });\n  const contactJosiah = dataSource.add(Contact, {\n    id: ObjectId.random(),\n    name: 'Josiah',\n  });\n  const contactDima = dataSource.add(Contact, {\n    id: ObjectId.random(),\n    name: 'Dima',\n  });\n  const contactFred = dataSource.add(Contact, {\n    id: ObjectId.random(),\n    name: 'Fred',\n  });\n\n  const projectComposer = dataSource.add(Project, {\n    id: ObjectId.random(),\n    name: 'Composer',\n  });\n  const projectEcho = dataSource.add(Project, {\n    id: ObjectId.random(),\n    name: 'ECHO',\n  });\n  const projectDoodles = dataSource.add(Project, {\n    id: ObjectId.random(),\n    name: 'Doodles',\n  });\n\n  const _taskComposer1 = dataSource.add(Task, {\n    id: ObjectId.random(),\n    name: 'Optimize startup performance',\n    project: makeRef(projectComposer),\n    assignee: makeRef(contactJosiah),\n  });\n  const _taskComposer2 = dataSource.add(Task, {\n    id: ObjectId.random(),\n    name: 'Create form builder',\n    project: makeRef(projectComposer),\n    assignee: makeRef(contactRich),\n  });\n  const _taskComposer3 = dataSource.add(Task, {\n    id: ObjectId.random(),\n    name: 'Add support for custom themes',\n    project: makeRef(projectComposer),\n    assignee: makeRef(contactJosiah),\n  });\n  const _taskComposer5 = dataSource.add(Task, {\n    id: ObjectId.random(),\n    name: 'Implement community plugin',\n    project: makeRef(projectComposer),\n    assignee: makeRef(contactFred),\n  });\n  const _taskComposer4 = dataSource.add(Task, {\n    id: ObjectId.random(),\n    name: 'Implement dark mode',\n    project: makeRef(projectComposer),\n    assignee: makeRef(contactRich),\n  });\n  const _taskEcho1 = dataSource.add(Task, {\n    id: ObjectId.random(),\n    name: 'Implement cypher query engine',\n    project: makeRef(projectEcho),\n    assignee: makeRef(contactDima),\n  });\n  const _taskEcho2 = dataSource.add(Task, {\n    id: ObjectId.random(),\n    name: 'Add schema editor',\n    project: makeRef(projectEcho),\n    assignee: makeRef(contactRich),\n  });\n  const _taskDoodles1 = dataSource.add(Task, {\n    id: ObjectId.random(),\n    name: 'Add support for custom themes',\n    project: makeRef(projectDoodles),\n    assignee: makeRef(contactFred),\n  });\n  const _taskDoodles2 = dataSource.add(Task, {\n    id: ObjectId.random(),\n    name: 'Implement dark mode',\n    project: makeRef(projectDoodles),\n    assignee: makeRef(contactJosiah),\n  });\n\n  const _orgDxos = dataSource.add(Org, {\n    id: ObjectId.random(),\n    name: 'DXOS',\n    employees: [makeRef(contactRich), makeRef(contactJosiah), makeRef(contactDima)],\n    projects: [makeRef(projectEcho)],\n  });\n  const _orgBraneframe = dataSource.add(Org, {\n    id: ObjectId.random(),\n    name: 'Braneframe',\n    employees: [makeRef(contactJosiah), makeRef(contactRich)],\n    projects: [makeRef(projectComposer)],\n  });\n  dataSource.computeGraph();\n\n  return dataSource;\n};\n\nexport class MockDataSource implements DataSource {\n  objects: Record<\n    string,\n    {\n      schema: S.Schema.All;\n      typeDxn: string;\n      id: string;\n      data: unknown;\n    }\n  > = {};\n\n  schema: Record<\n    string,\n    {\n      json: JsonSchemaType;\n      instance: S.Schema.All;\n    }\n  > = {};\n\n  nodes: Node[] = [];\n  relationships: Relationship[] = [];\n\n  async getNodes({ label }: { label?: string }) {\n    return this.nodes.filter((node) => !label || node.label === label);\n  }\n\n  async getRelationships({ label }: { label?: string }) {\n    return this.relationships.filter((relationship) => !label || relationship.label === label);\n  }\n\n  add<S extends S.Schema.All>(schema: S, data: S.Schema.Type<S>) {\n    invariant(typeof data.id === 'string', 'Data must have an id');\n    const jsonSchema = toJsonSchema(schema);\n    if (!this.schema[jsonSchema.$id!]) {\n      this.schema[jsonSchema.$id!] = {\n        json: jsonSchema,\n        instance: schema,\n      };\n    }\n\n    const typenameDxn = jsonSchema.$id ?? raise(new Error('Schema must have $id'));\n\n    this.objects[data.id] = {\n      schema,\n      id: data.id,\n      typeDxn: typenameDxn,\n      data,\n    };\n\n    return data;\n  }\n\n  computeGraph() {\n    this.nodes = [];\n    this.relationships = [];\n\n    for (const object of Object.values(this.objects)) {\n      this.nodes.push({\n        kind: 'node',\n        id: object.id,\n        label: formatNodeLabel(object.typeDxn),\n        properties: object.data as any,\n      });\n    }\n\n    for (const object of Object.values(this.objects)) {\n      const schema = this.schema[object.typeDxn].json;\n      const typenameDxn = schema.$id ?? raise(new Error('Schema must have $id'));\n\n      for (const [name, propSchema] of Object.entries(schema.properties!)) {\n        if (propSchema.$id === JSON_SCHEMA_ECHO_REF_ID) {\n          const label = formatInferredRelationshipLabel(typenameDxn, name);\n\n          const source = this.nodes.find((node) => node.id === object.id);\n          if (!source) {\n            continue;\n          }\n\n          const { id: targetId } = (object.data as any)[name].target;\n\n          const target = this.nodes.find((node) => node.id === targetId);\n          if (!target) {\n            continue;\n          }\n\n          this.relationships.push({\n            kind: 'relationship',\n            id: `${object.id}-${name}-${target.id}`,\n            label,\n            properties: {},\n            source,\n            target,\n          });\n        } else if (propSchema.type === 'array' && propSchema.items?.$id === JSON_SCHEMA_ECHO_REF_ID) {\n          const label = formatInferredRelationshipLabel(typenameDxn, name);\n\n          const source = this.nodes.find((node) => node.id === object.id);\n          if (!source) {\n            continue;\n          }\n\n          for (const { id: targetId } of (object.data as any)[name].map((x: Ref<any>) => x.target)) {\n            const target = this.nodes.find((node) => node.id === targetId);\n            if (!target) {\n              continue;\n            }\n\n            this.relationships.push({\n              kind: 'relationship',\n              id: `${object.id}-${name}-${targetId}`,\n              label,\n              properties: {},\n              source,\n              target,\n            });\n          }\n        }\n      }\n    }\n  }\n}\n", "//\n// Copyright 2024 DXOS.org\n//\n\nimport { Schema as S } from '@effect/schema';\n\nimport { EntityKind, ObjectAnnotationId, Ref } from '@dxos/echo-schema';\n\n// TODO(burdon): Reconcile with sdk/schema/testing.\n// TODO(burdon): Convert interface to type.\n\n// TODO(dmaretskyi): Extract.\nexport const ObjectId = S.String.annotations({ description: 'The unique object identifier.' });\n\nexport const Contact = S.Struct({\n  id: ObjectId,\n  name: S.String.annotations({ description: 'The name of the person.' }),\n  email: S.optional(S.String).annotations({ description: 'Email address.' }),\n})\n  .pipe(S.mutable)\n  .annotations({\n    [ObjectAnnotationId]: {\n      kind: EntityKind.Object,\n      typename: 'example.com/type/Contact',\n      version: '0.1.0',\n    },\n    description: 'Contact information.',\n  });\nexport interface Contact extends S.Schema.Type<typeof Contact> {}\n\nexport const Project = S.Struct({\n  id: ObjectId,\n  name: S.String.annotations({ description: 'The name of the project.' }),\n  description: S.optional(S.String).annotations({ description: 'The description of the project.' }),\n})\n  .pipe(S.mutable)\n  .annotations({\n    [ObjectAnnotationId]: {\n      kind: EntityKind.Object,\n      typename: 'example.com/type/Project',\n      version: '0.1.0',\n    },\n    description: 'Contact information.',\n  });\nexport interface Project extends S.Schema.Type<typeof Project> {}\n\nexport const Task = S.Struct({\n  id: ObjectId,\n  name: S.String.annotations({ description: 'The name of the task.' }),\n  description: S.optional(S.String).annotations({ description: 'The description of the task.' }),\n  project: Ref(Project),\n  assignee: Ref(Contact),\n})\n  .pipe(S.mutable)\n  .annotations({\n    [ObjectAnnotationId]: {\n      kind: EntityKind.Object,\n      typename: 'example.com/type/Task',\n      version: '0.1.0',\n    },\n    description: 'Contact information.',\n  });\nexport interface Task extends S.Schema.Type<typeof Task> {}\n\nexport const Org = S.Struct({\n  id: ObjectId,\n  name: S.String.annotations({ description: 'The name of the organization.' }),\n  projects: S.Array(Ref(Project)),\n  employees: S.Array(Ref(Contact)),\n})\n  .pipe(S.mutable)\n  .annotations({\n    [ObjectAnnotationId]: {\n      kind: EntityKind.Object,\n      typename: 'example.com/type/Org',\n      version: '0.1.0',\n    },\n    description: 'Contact information.',\n  });\nexport interface Org extends S.Schema.Type<typeof Org> {}\n"],
-  "mappings": ";;;;;;;;;;;;AAOO,IAAMA,eAAe,CAAC,EAC3BC,QACAC,QACAC,QAAO,IAKL,CAAC,MAAC;AACJ,QAAMC,SAA8B,CAAC;AACrC,MAAIC,cAAc;AAElB,SAAO,CAACC,UAAAA;AACN,QAAI,OAAOJ,WAAW,cAAc,CAACA,OAAOI,KAAAA,GAAQ;AAClD;IACF;AAEA,UAAMC,oBAAoB,CAACC,YAAAA;AACzB,cAAQA,QAAQC,MAAI;QAClB,KAAK;AACHC,kBAAQC,OAAOC,MAAMJ,QAAQK,IAAI;AACjC;QACF,KAAK;AACHH,kBAAQC,OAAOC,MAAM,2BAAiBJ,QAAQM,IAAI;CAAI;AACtD;QACF,KAAK,SAAS;AACZ,cAAIN,QAAQO,MAAMP,QAAQQ,QAAQ;AAChCZ,mBAAOI,QAAQO,EAAE,IAAIP,QAAQQ;UAC/B;AACAN,kBAAQC,OAAOC,MAAM,aAAaJ,QAAQO,EAAE,eAAeP,QAAQQ,QAAQC,UAAAA,GAAa;AACxF;QACF;MACF;IACF;AAEA,UAAMC,qBAAqB,CAACL,SAAAA;AAC1B,YAAMM,aAAaN,KAAKO,MAAM,mDAAA;AAC9B,UAAID,cAAchB,WAAWC,OAAOe,WAAW,CAAA,CAAE,GAAG;AAClDhB,gBAAQ;UACNY,IAAII,WAAW,CAAA;UACfE,aAAaF,WAAW,CAAA;UACxBH,QAAQZ,OAAOe,WAAW,CAAA,CAAE;QAC9B,CAAA;MACF;IACF;AAEA,QAAIlB,QAAQ;AACV,cAAQK,MAAMG,MAAI;QAChB,KAAK,iBAAiB;AACpBC,kBAAQC,OAAOC,MAAM,GAAGN,MAAMgB,QAAQC,KAAKC,YAAW,CAAA;;CAAQ;AAC9D,qBAAWhB,WAAWF,MAAMgB,QAAQd,SAAS;AAC3CD,8BAAkBC,OAAAA;UACpB;AACA;QACF;QACA,KAAK,uBAAuB;AAC1BH,wBAAc;AACdE,4BAAkBD,MAAME,OAAO;AAC/B;QACF;QACA,KAAK,uBAAuB;AAC1B,kBAAQF,MAAMmB,MAAMhB,MAAI;YACtB,KAAK,cAAc;AACjBJ,6BAAeC,MAAMmB,MAAMZ;AAC3BH,sBAAQC,OAAOC,MAAMN,MAAMmB,MAAMZ,IAAI;AACrC;YACF;YACA,KAAK,oBAAoB;AACvBH,sBAAQC,OAAOC,MAAMN,MAAMmB,MAAMC,YAAY;AAC7C;YACF;UACF;AACA;QACF;QACA,KAAK,sBAAsB;AACzBhB,kBAAQC,OAAOC,MAAM,IAAA;AACrBM,6BAAmBb,WAAAA;AACnBA,wBAAc;AACd;QACF;QACA,KAAK,iBAAiB;AACpB;QACF;QACA,KAAK,gBAAgB;AACnBK,kBAAQC,OAAOC,MAAM,MAAA;AACrB;QACF;MACF;IACF;AAEA,YAAQN,MAAMG,MAAI;MAChB,KAAK,WAAW;AACd,YAAI,CAACR,UAAUK,MAAMgB,QAAQC,SAAS,aAAa;AACjDb,kBAAQC,OAAOC,MAAM,GAAGN,MAAMgB,QAAQC,KAAKC,YAAW,CAAA;;CAAQ;AAC9D,qBAAWG,SAASrB,MAAMgB,QAAQd,SAAS;AACzC,oBAAQmB,MAAMlB,MAAI;cAChB,KAAK;AACHC,wBAAQC,OAAOC,MAAMe,MAAMd,OAAO,IAAA;AAClCK,mCAAmBS,MAAMd,IAAI;AAC7B;cACF,KAAK;AACHH,wBAAQC,OAAOC,MAAM,2BAAiBe,MAAMb,IAAI;CAAI;AACpDJ,wBAAQC,OAAOC,MAAM,KAAKgB,KAAKC,UAAUF,MAAMG,KAAK,CAAA;CAAK;AACzD;cACF,KAAK;AACH,oBAAIH,MAAMI,SAAS;AACjBrB,0BAAQC,OAAOC,MAAM,uBAAA;AACrBF,0BAAQC,OAAOC,MAAMe,MAAMnB,UAAU,IAAA;gBACvC,OAAO;AACLE,0BAAQC,OAAOC,MAAM,yBAAA;AACrBF,0BAAQC,OAAOC,MAAMe,MAAMnB,UAAU,IAAA;gBACvC;YACJ;AACAE,oBAAQC,OAAOC,MAAM,MAAA;UACvB;QACF;AACA;MACF;IACF;EACF;AACF;;;AC3HA,SAASoB,UAAUC,SAAS;AAE5B,SAASC,oBAAoB;AAC7B,SAASC,WAAW;;AAMb,IAAMC,mBAAmB,CAACC,YAAwBC,cAA8B,CAAA,MACrFC,WAAW;EACTC,MAAM;EACNC,aACE,6FACCH,YAAYI,SAAS,IAAI;;EAAOC,mBAAmBL,WAAAA,CAAAA,KAAiB;EACvEM,QAAQC,EAAEC,OAAO;IACfC,OAAOF,EAAEG,OAAOC,YAAY;MAC1BR,aAAa;;;;;;;;;;;;;;;;;;;;;IAqBf,CAAA;EACF,CAAA;EACAS,SAAS,OAAO,EAAEH,MAAK,MAAE;AACvB,QAAI;AACFI,UAAI,gBAAgB;QAAEJ;MAAM,GAAA;;;;;;AAC5B,YAAMK,UAAU,MAAMC,aAAahB,YAAYU,KAAAA;AAC/CI,UAAI,kBAAkB;QAAEC;MAAQ,GAAA;;;;;;AAChC,aAAOE,cAAcC,QAAQH,OAAAA;IAC/B,SAASI,GAAQ;AACfL,UAAIM,MAAMD,GAAAA,QAAAA;;;;;;AACV,aAAOF,cAAcI,MAAMF,EAAEG,OAAO;IACtC;EACF;AACF,CAAA;AAEK,IAAMhB,qBAAqB,CAACL,gBAAgCsB;;;;;;MAM7DC,uBAAuBvB,YAAYwB,IAAI,CAAClB,WAAWmB,aAAanB,MAAAA,CAAAA,CAAAA,CAAAA;;;;;;;;;;;;;;ACzDtE,SAASoB,aAAa;AACtB,SAASC,yBAAyBC,UAAUC,gBAAAA,qBAAmD;AAC/F,SAASC,iBAAiB;AAC1B,SAASC,QAAQC,eAAe;;AAYzB,IAAMC,eAAe,CAACC,OAAAA;AAC3B,QAAMC,cAAcD,GAAGE,IACrBC,OAAOC,SAAS;IACdC,MAAM;EACR,CAAA,CAAA;AAEF,QAAMC,gBAAgBN,GAAGE,IACvBC,OAAOC,SAAS;IACdC,MAAM;EACR,CAAA,CAAA;AAEF,QAAME,cAAcP,GAAGE,IACrBC,OAAOC,SAAS;IACdC,MAAM;EACR,CAAA,CAAA;AAEF,QAAMG,cAAcR,GAAGE,IACrBC,OAAOC,SAAS;IACdC,MAAM;EACR,CAAA,CAAA;AAGF,QAAMI,kBAAkBT,GAAGE,IACzBC,OAAOO,SAAS;IACdL,MAAM;EACR,CAAA,CAAA;AAEF,QAAMM,cAAcX,GAAGE,IACrBC,OAAOO,SAAS;IACdL,MAAM;EACR,CAAA,CAAA;AAEF,QAAMO,iBAAiBZ,GAAGE,IACxBC,OAAOO,SAAS;IACdL,MAAM;EACR,CAAA,CAAA;AAGF,QAAMQ,iBAAiBb,GAAGE,IACxBC,OAAOW,MAAM;IACXT,MAAM;IACNU,SAASC,QAAQP,eAAAA;IACjBQ,UAAUD,QAAQV,aAAAA;EACpB,CAAA,CAAA;AAEF,QAAMY,iBAAiBlB,GAAGE,IACxBC,OAAOW,MAAM;IACXT,MAAM;IACNU,SAASC,QAAQP,eAAAA;IACjBQ,UAAUD,QAAQf,WAAAA;EACpB,CAAA,CAAA;AAEF,QAAMkB,iBAAiBnB,GAAGE,IACxBC,OAAOW,MAAM;IACXT,MAAM;IACNU,SAASC,QAAQP,eAAAA;IACjBQ,UAAUD,QAAQV,aAAAA;EACpB,CAAA,CAAA;AAEF,QAAMc,iBAAiBpB,GAAGE,IACxBC,OAAOW,MAAM;IACXT,MAAM;IACNU,SAASC,QAAQP,eAAAA;IACjBQ,UAAUD,QAAQR,WAAAA;EACpB,CAAA,CAAA;AAEF,QAAMa,iBAAiBrB,GAAGE,IACxBC,OAAOW,MAAM;IACXT,MAAM;IACNU,SAASC,QAAQP,eAAAA;IACjBQ,UAAUD,QAAQf,WAAAA;EACpB,CAAA,CAAA;AAEF,QAAMqB,aAAatB,GAAGE,IACpBC,OAAOW,MAAM;IACXT,MAAM;IACNU,SAASC,QAAQL,WAAAA;IACjBM,UAAUD,QAAQT,WAAAA;EACpB,CAAA,CAAA;AAEF,QAAMgB,aAAavB,GAAGE,IACpBC,OAAOW,MAAM;IACXT,MAAM;IACNU,SAASC,QAAQL,WAAAA;IACjBM,UAAUD,QAAQf,WAAAA;EACpB,CAAA,CAAA;AAEF,QAAMuB,gBAAgBxB,GAAGE,IACvBC,OAAOW,MAAM;IACXT,MAAM;IACNU,SAASC,QAAQJ,cAAAA;IACjBK,UAAUD,QAAQR,WAAAA;EACpB,CAAA,CAAA;AAEF,QAAMiB,gBAAgBzB,GAAGE,IACvBC,OAAOW,MAAM;IACXT,MAAM;IACNU,SAASC,QAAQJ,cAAAA;IACjBK,UAAUD,QAAQV,aAAAA;EACpB,CAAA,CAAA;AAGF,QAAMoB,WAAW1B,GAAGE,IAClBC,OAAOwB,KAAK;IACVtB,MAAM;IACNuB,WAAW;MAACZ,QAAQf,WAAAA;MAAce,QAAQV,aAAAA;MAAgBU,QAAQT,WAAAA;;IAClEsB,UAAU;MAACb,QAAQL,WAAAA;;EACrB,CAAA,CAAA;AAEF,QAAMmB,iBAAiB9B,GAAGE,IACxBC,OAAOwB,KAAK;IACVtB,MAAM;IACNuB,WAAW;MAACZ,QAAQV,aAAAA;MAAgBU,QAAQf,WAAAA;;IAC5C4B,UAAU;MAACb,QAAQP,eAAAA;;EACrB,CAAA,CAAA;AAEJ;AAGO,IAAMsB,iBAAiB,MAAA;AAC5B,QAAMC,aAAa,IAAIC,eAAAA;AAEvB,QAAMhC,cAAc+B,WAAW9B,IAAIE,SAAS;IAC1C8B,IAAIC,SAASC,OAAM;IACnB/B,MAAM;EACR,CAAA;AACA,QAAMC,gBAAgB0B,WAAW9B,IAAIE,SAAS;IAC5C8B,IAAIC,SAASC,OAAM;IACnB/B,MAAM;EACR,CAAA;AACA,QAAME,cAAcyB,WAAW9B,IAAIE,SAAS;IAC1C8B,IAAIC,SAASC,OAAM;IACnB/B,MAAM;EACR,CAAA;AACA,QAAMG,cAAcwB,WAAW9B,IAAIE,SAAS;IAC1C8B,IAAIC,SAASC,OAAM;IACnB/B,MAAM;EACR,CAAA;AAEA,QAAMI,kBAAkBuB,WAAW9B,IAAIQ,SAAS;IAC9CwB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;EACR,CAAA;AACA,QAAMM,cAAcqB,WAAW9B,IAAIQ,SAAS;IAC1CwB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;EACR,CAAA;AACA,QAAMO,iBAAiBoB,WAAW9B,IAAIQ,SAAS;IAC7CwB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;EACR,CAAA;AAEA,QAAMQ,iBAAiBmB,WAAW9B,IAAIY,MAAM;IAC1CoB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNU,SAASC,QAAQP,eAAAA;IACjBQ,UAAUD,QAAQV,aAAAA;EACpB,CAAA;AACA,QAAMY,iBAAiBc,WAAW9B,IAAIY,MAAM;IAC1CoB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNU,SAASC,QAAQP,eAAAA;IACjBQ,UAAUD,QAAQf,WAAAA;EACpB,CAAA;AACA,QAAMkB,iBAAiBa,WAAW9B,IAAIY,MAAM;IAC1CoB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNU,SAASC,QAAQP,eAAAA;IACjBQ,UAAUD,QAAQV,aAAAA;EACpB,CAAA;AACA,QAAMc,iBAAiBY,WAAW9B,IAAIY,MAAM;IAC1CoB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNU,SAASC,QAAQP,eAAAA;IACjBQ,UAAUD,QAAQR,WAAAA;EACpB,CAAA;AACA,QAAMa,iBAAiBW,WAAW9B,IAAIY,MAAM;IAC1CoB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNU,SAASC,QAAQP,eAAAA;IACjBQ,UAAUD,QAAQf,WAAAA;EACpB,CAAA;AACA,QAAMqB,aAAaU,WAAW9B,IAAIY,MAAM;IACtCoB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNU,SAASC,QAAQL,WAAAA;IACjBM,UAAUD,QAAQT,WAAAA;EACpB,CAAA;AACA,QAAMgB,aAAaS,WAAW9B,IAAIY,MAAM;IACtCoB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNU,SAASC,QAAQL,WAAAA;IACjBM,UAAUD,QAAQf,WAAAA;EACpB,CAAA;AACA,QAAMuB,gBAAgBQ,WAAW9B,IAAIY,MAAM;IACzCoB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNU,SAASC,QAAQJ,cAAAA;IACjBK,UAAUD,QAAQR,WAAAA;EACpB,CAAA;AACA,QAAMiB,gBAAgBO,WAAW9B,IAAIY,MAAM;IACzCoB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNU,SAASC,QAAQJ,cAAAA;IACjBK,UAAUD,QAAQV,aAAAA;EACpB,CAAA;AAEA,QAAMoB,WAAWM,WAAW9B,IAAIyB,KAAK;IACnCO,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNuB,WAAW;MAACZ,QAAQf,WAAAA;MAAce,QAAQV,aAAAA;MAAgBU,QAAQT,WAAAA;;IAClEsB,UAAU;MAACb,QAAQL,WAAAA;;EACrB,CAAA;AACA,QAAMmB,iBAAiBE,WAAW9B,IAAIyB,KAAK;IACzCO,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNuB,WAAW;MAACZ,QAAQV,aAAAA;MAAgBU,QAAQf,WAAAA;;IAC5C4B,UAAU;MAACb,QAAQP,eAAAA;;EACrB,CAAA;AACAuB,aAAWK,aAAY;AAEvB,SAAOL;AACT;AAEO,IAAMC,iBAAN,MAAMA;EAAN;AACLK,mBAQI,CAAC;AAELC,kBAMI,CAAC;AAELC,iBAAgB,CAAA;AAChBC,yBAAgC,CAAA;;EAEhC,MAAMC,SAAS,EAAEC,MAAK,GAAwB;AAC5C,WAAO,KAAKH,MAAMI,OAAO,CAACC,SAAS,CAACF,SAASE,KAAKF,UAAUA,KAAAA;EAC9D;EAEA,MAAMG,iBAAiB,EAAEH,MAAK,GAAwB;AACpD,WAAO,KAAKF,cAAcG,OAAO,CAACG,iBAAiB,CAACJ,SAASI,aAAaJ,UAAUA,KAAAA;EACtF;EAEAzC,IAA4BqC,QAAWS,MAAwB;AAC7DC,cAAU,OAAOD,KAAKd,OAAO,UAAU,wBAAA;;;;;;;;;AACvC,UAAMgB,aAAaC,cAAaZ,MAAAA;AAChC,QAAI,CAAC,KAAKA,OAAOW,WAAWE,GAAG,GAAI;AACjC,WAAKb,OAAOW,WAAWE,GAAG,IAAK;QAC7BC,MAAMH;QACNI,UAAUf;MACZ;IACF;AAEA,UAAMgB,cAAcL,WAAWE,OAAOI,MAAM,IAAIC,MAAM,sBAAA,CAAA;AAEtD,SAAKnB,QAAQU,KAAKd,EAAE,IAAI;MACtBK;MACAL,IAAIc,KAAKd;MACTwB,SAASH;MACTP;IACF;AAEA,WAAOA;EACT;EAEAX,eAAe;AACb,SAAKG,QAAQ,CAAA;AACb,SAAKC,gBAAgB,CAAA;AAErB,eAAWkB,UAAUC,OAAOC,OAAO,KAAKvB,OAAO,GAAG;AAChD,WAAKE,MAAMsB,KAAK;QACdC,MAAM;QACN7B,IAAIyB,OAAOzB;QACXS,OAAOqB,gBAAgBL,OAAOD,OAAO;QACrCO,YAAYN,OAAOX;MACrB,CAAA;IACF;AAEA,eAAWW,UAAUC,OAAOC,OAAO,KAAKvB,OAAO,GAAG;AAChD,YAAMC,SAAS,KAAKA,OAAOoB,OAAOD,OAAO,EAAEL;AAC3C,YAAME,cAAchB,OAAOa,OAAOI,MAAM,IAAIC,MAAM,sBAAA,CAAA;AAElD,iBAAW,CAACpD,MAAM6D,UAAAA,KAAeN,OAAOO,QAAQ5B,OAAO0B,UAAU,GAAI;AACnE,YAAIC,WAAWd,QAAQgB,yBAAyB;AAC9C,gBAAMzB,QAAQ0B,gCAAgCd,aAAalD,IAAAA;AAE3D,gBAAMiE,SAAS,KAAK9B,MAAM+B,KAAK,CAAC1B,SAASA,KAAKX,OAAOyB,OAAOzB,EAAE;AAC9D,cAAI,CAACoC,QAAQ;AACX;UACF;AAEA,gBAAM,EAAEpC,IAAIsC,SAAQ,IAAMb,OAAOX,KAAa3C,IAAAA,EAAMoE;AAEpD,gBAAMA,SAAS,KAAKjC,MAAM+B,KAAK,CAAC1B,SAASA,KAAKX,OAAOsC,QAAAA;AACrD,cAAI,CAACC,QAAQ;AACX;UACF;AAEA,eAAKhC,cAAcqB,KAAK;YACtBC,MAAM;YACN7B,IAAI,GAAGyB,OAAOzB,EAAE,IAAI7B,IAAAA,IAAQoE,OAAOvC,EAAE;YACrCS;YACAsB,YAAY,CAAC;YACbK;YACAG;UACF,CAAA;QACF,WAAWP,WAAWQ,SAAS,WAAWR,WAAWS,OAAOvB,QAAQgB,yBAAyB;AAC3F,gBAAMzB,QAAQ0B,gCAAgCd,aAAalD,IAAAA;AAE3D,gBAAMiE,SAAS,KAAK9B,MAAM+B,KAAK,CAAC1B,SAASA,KAAKX,OAAOyB,OAAOzB,EAAE;AAC9D,cAAI,CAACoC,QAAQ;AACX;UACF;AAEA,qBAAW,EAAEpC,IAAIsC,SAAQ,KAAOb,OAAOX,KAAa3C,IAAAA,EAAMuE,IAAI,CAACC,MAAgBA,EAAEJ,MAAM,GAAG;AACxF,kBAAMA,SAAS,KAAKjC,MAAM+B,KAAK,CAAC1B,SAASA,KAAKX,OAAOsC,QAAAA;AACrD,gBAAI,CAACC,QAAQ;AACX;YACF;AAEA,iBAAKhC,cAAcqB,KAAK;cACtBC,MAAM;cACN7B,IAAI,GAAGyB,OAAOzB,EAAE,IAAI7B,IAAAA,IAAQmE,QAAAA;cAC5B7B;cACAsB,YAAY,CAAC;cACbK;cACAG;YACF,CAAA;UACF;QACF;MACF;IACF;EACF;AACF;;;ACzWA,SAASK,UAAUC,UAAS;AAE5B,SAASC,YAAYC,oBAAoBC,WAAW;AAM7C,IAAMC,YAAWC,GAAEC,OAAOC,YAAY;EAAEC,aAAa;AAAgC,CAAA;AAErF,IAAMC,UAAUJ,GAAEK,OAAO;EAC9BC,IAAIP;EACJQ,MAAMP,GAAEC,OAAOC,YAAY;IAAEC,aAAa;EAA0B,CAAA;EACpEK,OAAOR,GAAES,SAAST,GAAEC,MAAM,EAAEC,YAAY;IAAEC,aAAa;EAAiB,CAAA;AAC1E,CAAA,EACGO,KAAKV,GAAEW,OAAO,EACdT,YAAY;EACX,CAACU,kBAAAA,GAAqB;IACpBC,MAAMC,WAAWC;IACjBC,UAAU;IACVC,SAAS;EACX;EACAd,aAAa;AACf,CAAA;AAGK,IAAMe,UAAUlB,GAAEK,OAAO;EAC9BC,IAAIP;EACJQ,MAAMP,GAAEC,OAAOC,YAAY;IAAEC,aAAa;EAA2B,CAAA;EACrEA,aAAaH,GAAES,SAAST,GAAEC,MAAM,EAAEC,YAAY;IAAEC,aAAa;EAAkC,CAAA;AACjG,CAAA,EACGO,KAAKV,GAAEW,OAAO,EACdT,YAAY;EACX,CAACU,kBAAAA,GAAqB;IACpBC,MAAMC,WAAWC;IACjBC,UAAU;IACVC,SAAS;EACX;EACAd,aAAa;AACf,CAAA;AAGK,IAAMgB,OAAOnB,GAAEK,OAAO;EAC3BC,IAAIP;EACJQ,MAAMP,GAAEC,OAAOC,YAAY;IAAEC,aAAa;EAAwB,CAAA;EAClEA,aAAaH,GAAES,SAAST,GAAEC,MAAM,EAAEC,YAAY;IAAEC,aAAa;EAA+B,CAAA;EAC5FiB,SAASC,IAAIH,OAAAA;EACbI,UAAUD,IAAIjB,OAAAA;AAChB,CAAA,EACGM,KAAKV,GAAEW,OAAO,EACdT,YAAY;EACX,CAACU,kBAAAA,GAAqB;IACpBC,MAAMC,WAAWC;IACjBC,UAAU;IACVC,SAAS;EACX;EACAd,aAAa;AACf,CAAA;AAGK,IAAMoB,MAAMvB,GAAEK,OAAO;EAC1BC,IAAIP;EACJQ,MAAMP,GAAEC,OAAOC,YAAY;IAAEC,aAAa;EAAgC,CAAA;EAC1EqB,UAAUxB,GAAEyB,MAAMJ,IAAIH,OAAAA,CAAAA;EACtBQ,WAAW1B,GAAEyB,MAAMJ,IAAIjB,OAAAA,CAAAA;AACzB,CAAA,EACGM,KAAKV,GAAEW,OAAO,EACdT,YAAY;EACX,CAACU,kBAAAA,GAAqB;IACpBC,MAAMC,WAAWC;IACjBC,UAAU;IACVC,SAAS;EACX;EACAd,aAAa;AACf,CAAA;",
-  "names": ["createLogger", "stream", "filter", "onImage", "images", "currentText", "event", "printContentBlock", "content", "type", "process", "stdout", "write", "text", "name", "id", "source", "media_type", "onTextBlockPrinted", "imageMatch", "match", "description", "message", "role", "toUpperCase", "delta", "partial_json", "block", "JSON", "stringify", "input", "isError", "Schema", "S", "toJsonSchema", "log", "createCypherTool", "dataSource", "schemaTypes", "defineTool", "name", "description", "length", "createSystemPrompt", "schema", "S", "Struct", "query", "String", "annotations", "execute", "log", "results", "executeQuery", "LLMToolResult", "Success", "e", "catch", "Error", "message", "trim", "formatJsonSchemaForLLM", "map", "toJsonSchema", "raise", "JSON_SCHEMA_ECHO_REF_ID", "ObjectId", "toJsonSchema", "invariant", "create", "makeRef", "seedTestData", "db", "contactRich", "add", "create", "Contact", "name", "contactJosiah", "contactDima", "contactFred", "projectComposer", "Project", "projectEcho", "projectDoodles", "_taskComposer1", "Task", "project", "makeRef", "assignee", "_taskComposer2", "_taskComposer3", "_taskComposer5", "_taskComposer4", "_taskEcho1", "_taskEcho2", "_taskDoodles1", "_taskDoodles2", "_orgDxos", "Org", "employees", "projects", "_orgBraneframe", "createTestData", "dataSource", "MockDataSource", "id", "ObjectId", "random", "computeGraph", "objects", "schema", "nodes", "relationships", "getNodes", "label", "filter", "node", "getRelationships", "relationship", "data", "invariant", "jsonSchema", "toJsonSchema", "$id", "json", "instance", "typenameDxn", "raise", "Error", "typeDxn", "object", "Object", "values", "push", "kind", "formatNodeLabel", "properties", "propSchema", "entries", "JSON_SCHEMA_ECHO_REF_ID", "formatInferredRelationshipLabel", "source", "find", "targetId", "target", "type", "items", "map", "x", "Schema", "S", "EntityKind", "ObjectAnnotationId", "Ref", "ObjectId", "S", "String", "annotations", "description", "Contact", "Struct", "id", "name", "email", "optional", "pipe", "mutable", "ObjectAnnotationId", "kind", "EntityKind", "Object", "typename", "version", "Project", "Task", "project", "Ref", "assignee", "Org", "projects", "Array", "employees"]
+  "sources": ["../../../../src/testing/defs.ts", "../../../../src/testing/logger.ts", "../../../../src/testing/query-promts.ts", "../../../../src/testing/test-data.ts", "../../../../src/testing/test-schema.ts"],
+  "sourcesContent": ["//\n// Copyright 2025 DXOS.org\n//\n\nexport const AI_SERVICE_ENDPOINT = {\n  LOCAL: 'http://localhost:8788',\n  REMOTE: 'https://ai-service.dxos.workers.dev',\n};\n", "//\n// Copyright 2024 DXOS.org\n//\n\nimport type { ImageSource, MessageContentBlock } from '../ai-service';\nimport type { ConversationEvent } from '../conversation';\n\nexport const createLogger = ({\n  stream,\n  filter,\n  onImage,\n}: {\n  stream?: boolean;\n  filter?: (event: ConversationEvent) => boolean;\n  onImage?: (img: { id: string; description?: string; source: ImageSource }) => void;\n} = {}) => {\n  const images: Record<string, any> = {};\n  let currentText = '';\n\n  return (event: ConversationEvent) => {\n    if (typeof filter === 'function' && !filter(event)) {\n      return;\n    }\n\n    const printContentBlock = (content: MessageContentBlock) => {\n      switch (content.type) {\n        case 'text':\n          process.stdout.write(content.text);\n          break;\n        case 'tool_use':\n          process.stdout.write(`⚙️ [Tool Use] ${content.name}\\n`);\n          break;\n        case 'image': {\n          if (content.id && content.source) {\n            images[content.id] = content.source;\n          }\n          process.stdout.write(`[Image id=${content.id} media_type=${content.source?.media_type}]`);\n          break;\n        }\n      }\n    };\n\n    const onTextBlockPrinted = (text: string) => {\n      const imageMatch = text.match(/<image id=\"([^\"]+)\"(?:\\s+prompt=\"([^\"]+)\")?\\s*\\/>/);\n      if (imageMatch && onImage && images[imageMatch[1]]) {\n        onImage({\n          id: imageMatch[1],\n          description: imageMatch[2], // Extract prompt as description if present\n          source: images[imageMatch[1]],\n        });\n      }\n    };\n\n    if (stream) {\n      switch (event.type) {\n        case 'message_start': {\n          process.stdout.write(`${event.message.role.toUpperCase()}\\n\\n`);\n          for (const content of event.message.content) {\n            printContentBlock(content);\n          }\n          break;\n        }\n        case 'content_block_start': {\n          currentText = '';\n          printContentBlock(event.content);\n          break;\n        }\n        case 'content_block_delta': {\n          switch (event.delta.type) {\n            case 'text_delta': {\n              currentText += event.delta.text;\n              process.stdout.write(event.delta.text);\n              break;\n            }\n            case 'input_json_delta': {\n              process.stdout.write(event.delta.partial_json);\n              break;\n            }\n          }\n          break;\n        }\n        case 'content_block_stop': {\n          process.stdout.write('\\n');\n          onTextBlockPrinted(currentText);\n          currentText = '';\n          break;\n        }\n        case 'message_delta': {\n          break;\n        }\n        case 'message_stop': {\n          process.stdout.write('\\n\\n');\n          break;\n        }\n      }\n    }\n\n    switch (event.type) {\n      case 'message': {\n        if (!stream || event.message.role !== 'assistant') {\n          process.stdout.write(`${event.message.role.toUpperCase()}\\n\\n`);\n          for (const block of event.message.content) {\n            switch (block.type) {\n              case 'text':\n                process.stdout.write(block.text + '\\n');\n                onTextBlockPrinted(block.text);\n                break;\n              case 'tool_use':\n                process.stdout.write(`⚙️ [Tool Use] ${block.name}\\n`);\n                process.stdout.write(`  ${JSON.stringify(block.input)}\\n`);\n                break;\n              case 'tool_result':\n                if (block.isError) {\n                  process.stdout.write('❌ [Tool Error]\\n');\n                  process.stdout.write(block.content + '\\n');\n                } else {\n                  process.stdout.write('✅ [Tool Success]\\n');\n                  process.stdout.write(block.content + '\\n');\n                }\n            }\n            process.stdout.write('\\n\\n');\n          }\n        }\n        break;\n      }\n    }\n  };\n};\n", "//\n// Copyright 2024 DXOS.org\n//\n\nimport { Schema as S } from '@effect/schema';\n\nimport { toJsonSchema } from '@dxos/echo-schema';\nimport { log } from '@dxos/log';\n\nimport { defineTool, LLMToolResult } from '../conversation';\nimport { executeQuery, formatJsonSchemaForLLM, type DataSource } from '../cypher';\nimport { trim } from '../util';\n\nexport const createCypherTool = (dataSource: DataSource, schemaTypes: S.Schema.Any[] = []) =>\n  defineTool({\n    name: 'graphQuery',\n    description:\n      'Query the ECHO graph database in cypher query language. Returns data from the database.' +\n      (schemaTypes.length > 0 ? `\\n\\n${createSystemPrompt(schemaTypes)}` : ''),\n    schema: S.Struct({\n      query: S.String.annotations({\n        description: `\n      A valid cypher query string to execute.\n      Query must have one MATCH clause.\n      Match clause can have multiple patterns but they must all be connected via a relationship chain.\n      Query might have zero or one WHERE clauses.\n      Query must have one RETURN clause.\n      RETURN clause can have multiple expressions separated by commas.\n      DISTINCT keyword is not allowed.\n\n      <example>\n      MATCH (n:Person) RETURN n.name\n      </example>\n\n      <example>\n      MATCH (s:Studio)-[:PRODUCED]->(m:Movie)-[:PRODUCED_IN]->(c:City {name: \"Las Vegas\"}) RETURN s\n      </example>\n      \n      <example>\n      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\n      </example>\n      `,\n      }),\n    }),\n    execute: async ({ query }) => {\n      try {\n        log('cypher query', { query });\n        const results = await executeQuery(dataSource, query);\n        log('query complete', { results });\n        return LLMToolResult.Success(results);\n      } catch (e: any) {\n        log.catch(e);\n        return LLMToolResult.Error(e.message);\n      }\n    },\n  });\n\nexport const createSystemPrompt = (schemaTypes: S.Schema.Any[]) => trim`\n  You have access to ECHO graph database. You can query the database to get data to satisfy user prompts.\n\n  Database schema is defined in pseud-cypher syntax. The schema is:\n\n  <schema>\n    ${formatJsonSchemaForLLM(schemaTypes.map((schema) => toJsonSchema(schema)))}\n  <schema>\n\n  Before replying always think step-by-step on how to proceed.\n  Print your thoughts inside <cot> tags.\n  \n  <example>\n    <cot>To answer the question I need to ...</cot>\n  </example>\n\n  Before answering the user's question, decide what tools you need to use to answer the question.\n`;\n", "//\n// Copyright 2024 DXOS.org\n//\n\nimport { type Schema as S } from '@effect/schema';\n\nimport { raise } from '@dxos/debug';\nimport type { EchoDatabase } from '@dxos/echo-db';\nimport { JSON_SCHEMA_ECHO_REF_ID, ObjectId, toJsonSchema, type JsonSchemaType, type Ref } from '@dxos/echo-schema';\nimport { invariant } from '@dxos/invariant';\nimport { create, makeRef } from '@dxos/live-object';\n\nimport {\n  type DataSource,\n  type Node,\n  type Relationship,\n  formatInferredRelationshipLabel,\n  formatNodeLabel,\n} from '../cypher';\nimport { Contact, Org, Project, Task } from '../testing';\n\nexport const seedTestData = (db: EchoDatabase) => {\n  const contactRich = db.add(\n    create(Contact, {\n      name: 'Rich',\n    }),\n  );\n  const contactJosiah = db.add(\n    create(Contact, {\n      name: 'Josiah',\n    }),\n  );\n  const contactDima = db.add(\n    create(Contact, {\n      name: 'Dima',\n    }),\n  );\n  const contactFred = db.add(\n    create(Contact, {\n      name: 'Fred',\n    }),\n  );\n\n  const projectComposer = db.add(\n    create(Project, {\n      name: 'Composer',\n    }),\n  );\n  const projectEcho = db.add(\n    create(Project, {\n      name: 'ECHO',\n    }),\n  );\n  const projectDoodles = db.add(\n    create(Project, {\n      name: 'Doodles',\n    }),\n  );\n\n  const _taskComposer1 = db.add(\n    create(Task, {\n      name: 'Optimize startup performance',\n      project: makeRef(projectComposer),\n      assignee: makeRef(contactJosiah),\n    }),\n  );\n  const _taskComposer2 = db.add(\n    create(Task, {\n      name: 'Create form builder',\n      project: makeRef(projectComposer),\n      assignee: makeRef(contactRich),\n    }),\n  );\n  const _taskComposer3 = db.add(\n    create(Task, {\n      name: 'Add support for custom themes',\n      project: makeRef(projectComposer),\n      assignee: makeRef(contactJosiah),\n    }),\n  );\n  const _taskComposer5 = db.add(\n    create(Task, {\n      name: 'Implement community plugin',\n      project: makeRef(projectComposer),\n      assignee: makeRef(contactFred),\n    }),\n  );\n  const _taskComposer4 = db.add(\n    create(Task, {\n      name: 'Implement dark mode',\n      project: makeRef(projectComposer),\n      assignee: makeRef(contactRich),\n    }),\n  );\n  const _taskEcho1 = db.add(\n    create(Task, {\n      name: 'Implement cypher query engine',\n      project: makeRef(projectEcho),\n      assignee: makeRef(contactDima),\n    }),\n  );\n  const _taskEcho2 = db.add(\n    create(Task, {\n      name: 'Add schema editor',\n      project: makeRef(projectEcho),\n      assignee: makeRef(contactRich),\n    }),\n  );\n  const _taskDoodles1 = db.add(\n    create(Task, {\n      name: 'Add support for custom themes',\n      project: makeRef(projectDoodles),\n      assignee: makeRef(contactFred),\n    }),\n  );\n  const _taskDoodles2 = db.add(\n    create(Task, {\n      name: 'Implement dark mode',\n      project: makeRef(projectDoodles),\n      assignee: makeRef(contactJosiah),\n    }),\n  );\n\n  const _orgDxos = db.add(\n    create(Org, {\n      name: 'DXOS',\n      employees: [makeRef(contactRich), makeRef(contactJosiah), makeRef(contactDima)],\n      projects: [makeRef(projectEcho)],\n    }),\n  );\n  const _orgBraneframe = db.add(\n    create(Org, {\n      name: 'Braneframe',\n      employees: [makeRef(contactJosiah), makeRef(contactRich)],\n      projects: [makeRef(projectComposer)],\n    }),\n  );\n};\n\n// TODO(burdon): Use schema/testing.\nexport const createTestData = (): MockDataSource => {\n  const dataSource = new MockDataSource();\n\n  const contactRich = dataSource.add(Contact, {\n    id: ObjectId.random(),\n    name: 'Rich',\n  });\n  const contactJosiah = dataSource.add(Contact, {\n    id: ObjectId.random(),\n    name: 'Josiah',\n  });\n  const contactDima = dataSource.add(Contact, {\n    id: ObjectId.random(),\n    name: 'Dima',\n  });\n  const contactFred = dataSource.add(Contact, {\n    id: ObjectId.random(),\n    name: 'Fred',\n  });\n\n  const projectComposer = dataSource.add(Project, {\n    id: ObjectId.random(),\n    name: 'Composer',\n  });\n  const projectEcho = dataSource.add(Project, {\n    id: ObjectId.random(),\n    name: 'ECHO',\n  });\n  const projectDoodles = dataSource.add(Project, {\n    id: ObjectId.random(),\n    name: 'Doodles',\n  });\n\n  const _taskComposer1 = dataSource.add(Task, {\n    id: ObjectId.random(),\n    name: 'Optimize startup performance',\n    project: makeRef(projectComposer),\n    assignee: makeRef(contactJosiah),\n  });\n  const _taskComposer2 = dataSource.add(Task, {\n    id: ObjectId.random(),\n    name: 'Create form builder',\n    project: makeRef(projectComposer),\n    assignee: makeRef(contactRich),\n  });\n  const _taskComposer3 = dataSource.add(Task, {\n    id: ObjectId.random(),\n    name: 'Add support for custom themes',\n    project: makeRef(projectComposer),\n    assignee: makeRef(contactJosiah),\n  });\n  const _taskComposer5 = dataSource.add(Task, {\n    id: ObjectId.random(),\n    name: 'Implement community plugin',\n    project: makeRef(projectComposer),\n    assignee: makeRef(contactFred),\n  });\n  const _taskComposer4 = dataSource.add(Task, {\n    id: ObjectId.random(),\n    name: 'Implement dark mode',\n    project: makeRef(projectComposer),\n    assignee: makeRef(contactRich),\n  });\n  const _taskEcho1 = dataSource.add(Task, {\n    id: ObjectId.random(),\n    name: 'Implement cypher query engine',\n    project: makeRef(projectEcho),\n    assignee: makeRef(contactDima),\n  });\n  const _taskEcho2 = dataSource.add(Task, {\n    id: ObjectId.random(),\n    name: 'Add schema editor',\n    project: makeRef(projectEcho),\n    assignee: makeRef(contactRich),\n  });\n  const _taskDoodles1 = dataSource.add(Task, {\n    id: ObjectId.random(),\n    name: 'Add support for custom themes',\n    project: makeRef(projectDoodles),\n    assignee: makeRef(contactFred),\n  });\n  const _taskDoodles2 = dataSource.add(Task, {\n    id: ObjectId.random(),\n    name: 'Implement dark mode',\n    project: makeRef(projectDoodles),\n    assignee: makeRef(contactJosiah),\n  });\n\n  const _orgDxos = dataSource.add(Org, {\n    id: ObjectId.random(),\n    name: 'DXOS',\n    employees: [makeRef(contactRich), makeRef(contactJosiah), makeRef(contactDima)],\n    projects: [makeRef(projectEcho)],\n  });\n  const _orgBraneframe = dataSource.add(Org, {\n    id: ObjectId.random(),\n    name: 'Braneframe',\n    employees: [makeRef(contactJosiah), makeRef(contactRich)],\n    projects: [makeRef(projectComposer)],\n  });\n  dataSource.computeGraph();\n\n  return dataSource;\n};\n\nexport class MockDataSource implements DataSource {\n  objects: Record<\n    string,\n    {\n      schema: S.Schema.All;\n      typeDxn: string;\n      id: string;\n      data: unknown;\n    }\n  > = {};\n\n  schema: Record<\n    string,\n    {\n      json: JsonSchemaType;\n      instance: S.Schema.All;\n    }\n  > = {};\n\n  nodes: Node[] = [];\n  relationships: Relationship[] = [];\n\n  async getNodes({ label }: { label?: string }) {\n    return this.nodes.filter((node) => !label || node.label === label);\n  }\n\n  async getRelationships({ label }: { label?: string }) {\n    return this.relationships.filter((relationship) => !label || relationship.label === label);\n  }\n\n  add<S extends S.Schema.All>(schema: S, data: S.Schema.Type<S>) {\n    invariant(typeof data.id === 'string', 'Data must have an id');\n    const jsonSchema = toJsonSchema(schema);\n    if (!this.schema[jsonSchema.$id!]) {\n      this.schema[jsonSchema.$id!] = {\n        json: jsonSchema,\n        instance: schema,\n      };\n    }\n\n    const typenameDxn = jsonSchema.$id ?? raise(new Error('Schema must have $id'));\n\n    this.objects[data.id] = {\n      schema,\n      id: data.id,\n      typeDxn: typenameDxn,\n      data,\n    };\n\n    return data;\n  }\n\n  computeGraph() {\n    this.nodes = [];\n    this.relationships = [];\n\n    for (const object of Object.values(this.objects)) {\n      this.nodes.push({\n        kind: 'node',\n        id: object.id,\n        label: formatNodeLabel(object.typeDxn),\n        properties: object.data as any,\n      });\n    }\n\n    for (const object of Object.values(this.objects)) {\n      const schema = this.schema[object.typeDxn].json;\n      const typenameDxn = schema.$id ?? raise(new Error('Schema must have $id'));\n\n      for (const [name, propSchema] of Object.entries(schema.properties!)) {\n        if (propSchema.$id === JSON_SCHEMA_ECHO_REF_ID) {\n          const label = formatInferredRelationshipLabel(typenameDxn, name);\n\n          const source = this.nodes.find((node) => node.id === object.id);\n          if (!source) {\n            continue;\n          }\n\n          const { id: targetId } = (object.data as any)[name].target;\n\n          const target = this.nodes.find((node) => node.id === targetId);\n          if (!target) {\n            continue;\n          }\n\n          this.relationships.push({\n            kind: 'relationship',\n            id: `${object.id}-${name}-${target.id}`,\n            label,\n            properties: {},\n            source,\n            target,\n          });\n        } else if (propSchema.type === 'array' && propSchema.items?.$id === JSON_SCHEMA_ECHO_REF_ID) {\n          const label = formatInferredRelationshipLabel(typenameDxn, name);\n\n          const source = this.nodes.find((node) => node.id === object.id);\n          if (!source) {\n            continue;\n          }\n\n          for (const { id: targetId } of (object.data as any)[name].map((x: Ref<any>) => x.target)) {\n            const target = this.nodes.find((node) => node.id === targetId);\n            if (!target) {\n              continue;\n            }\n\n            this.relationships.push({\n              kind: 'relationship',\n              id: `${object.id}-${name}-${targetId}`,\n              label,\n              properties: {},\n              source,\n              target,\n            });\n          }\n        }\n      }\n    }\n  }\n}\n", "//\n// Copyright 2024 DXOS.org\n//\n\nimport { Schema as S } from '@effect/schema';\n\nimport { EntityKind, ObjectAnnotationId, Ref } from '@dxos/echo-schema';\n\n// TODO(burdon): Reconcile with sdk/schema/testing.\n// TODO(burdon): Convert interface to type.\n\n// TODO(dmaretskyi): Extract.\nexport const ObjectId = S.String.annotations({ description: 'The unique object identifier.' });\n\nexport const Contact = S.Struct({\n  id: ObjectId,\n  name: S.String.annotations({ description: 'The name of the person.' }),\n  email: S.optional(S.String).annotations({ description: 'Email address.' }),\n})\n  .pipe(S.mutable)\n  .annotations({\n    [ObjectAnnotationId]: {\n      kind: EntityKind.Object,\n      typename: 'example.com/type/Contact',\n      version: '0.1.0',\n    },\n    description: 'Contact information.',\n  });\nexport interface Contact extends S.Schema.Type<typeof Contact> {}\n\nexport const Project = S.Struct({\n  id: ObjectId,\n  name: S.String.annotations({ description: 'The name of the project.' }),\n  description: S.optional(S.String).annotations({ description: 'The description of the project.' }),\n})\n  .pipe(S.mutable)\n  .annotations({\n    [ObjectAnnotationId]: {\n      kind: EntityKind.Object,\n      typename: 'example.com/type/Project',\n      version: '0.1.0',\n    },\n    description: 'Contact information.',\n  });\nexport interface Project extends S.Schema.Type<typeof Project> {}\n\nexport const Task = S.Struct({\n  id: ObjectId,\n  name: S.String.annotations({ description: 'The name of the task.' }),\n  description: S.optional(S.String).annotations({ description: 'The description of the task.' }),\n  project: Ref(Project),\n  assignee: Ref(Contact),\n})\n  .pipe(S.mutable)\n  .annotations({\n    [ObjectAnnotationId]: {\n      kind: EntityKind.Object,\n      typename: 'example.com/type/Task',\n      version: '0.1.0',\n    },\n    description: 'Contact information.',\n  });\nexport interface Task extends S.Schema.Type<typeof Task> {}\n\nexport const Org = S.Struct({\n  id: ObjectId,\n  name: S.String.annotations({ description: 'The name of the organization.' }),\n  projects: S.Array(Ref(Project)),\n  employees: S.Array(Ref(Contact)),\n})\n  .pipe(S.mutable)\n  .annotations({\n    [ObjectAnnotationId]: {\n      kind: EntityKind.Object,\n      typename: 'example.com/type/Org',\n      version: '0.1.0',\n    },\n    description: 'Contact information.',\n  });\nexport interface Org extends S.Schema.Type<typeof Org> {}\n"],
+  "mappings": ";;;;;;;;;;;;AAIO,IAAMA,sBAAsB;EACjCC,OAAO;EACPC,QAAQ;AACV;;;ACAO,IAAMC,eAAe,CAAC,EAC3BC,QACAC,QACAC,QAAO,IAKL,CAAC,MAAC;AACJ,QAAMC,SAA8B,CAAC;AACrC,MAAIC,cAAc;AAElB,SAAO,CAACC,UAAAA;AACN,QAAI,OAAOJ,WAAW,cAAc,CAACA,OAAOI,KAAAA,GAAQ;AAClD;IACF;AAEA,UAAMC,oBAAoB,CAACC,YAAAA;AACzB,cAAQA,QAAQC,MAAI;QAClB,KAAK;AACHC,kBAAQC,OAAOC,MAAMJ,QAAQK,IAAI;AACjC;QACF,KAAK;AACHH,kBAAQC,OAAOC,MAAM,2BAAiBJ,QAAQM,IAAI;CAAI;AACtD;QACF,KAAK,SAAS;AACZ,cAAIN,QAAQO,MAAMP,QAAQQ,QAAQ;AAChCZ,mBAAOI,QAAQO,EAAE,IAAIP,QAAQQ;UAC/B;AACAN,kBAAQC,OAAOC,MAAM,aAAaJ,QAAQO,EAAE,eAAeP,QAAQQ,QAAQC,UAAAA,GAAa;AACxF;QACF;MACF;IACF;AAEA,UAAMC,qBAAqB,CAACL,SAAAA;AAC1B,YAAMM,aAAaN,KAAKO,MAAM,mDAAA;AAC9B,UAAID,cAAchB,WAAWC,OAAOe,WAAW,CAAA,CAAE,GAAG;AAClDhB,gBAAQ;UACNY,IAAII,WAAW,CAAA;UACfE,aAAaF,WAAW,CAAA;UACxBH,QAAQZ,OAAOe,WAAW,CAAA,CAAE;QAC9B,CAAA;MACF;IACF;AAEA,QAAIlB,QAAQ;AACV,cAAQK,MAAMG,MAAI;QAChB,KAAK,iBAAiB;AACpBC,kBAAQC,OAAOC,MAAM,GAAGN,MAAMgB,QAAQC,KAAKC,YAAW,CAAA;;CAAQ;AAC9D,qBAAWhB,WAAWF,MAAMgB,QAAQd,SAAS;AAC3CD,8BAAkBC,OAAAA;UACpB;AACA;QACF;QACA,KAAK,uBAAuB;AAC1BH,wBAAc;AACdE,4BAAkBD,MAAME,OAAO;AAC/B;QACF;QACA,KAAK,uBAAuB;AAC1B,kBAAQF,MAAMmB,MAAMhB,MAAI;YACtB,KAAK,cAAc;AACjBJ,6BAAeC,MAAMmB,MAAMZ;AAC3BH,sBAAQC,OAAOC,MAAMN,MAAMmB,MAAMZ,IAAI;AACrC;YACF;YACA,KAAK,oBAAoB;AACvBH,sBAAQC,OAAOC,MAAMN,MAAMmB,MAAMC,YAAY;AAC7C;YACF;UACF;AACA;QACF;QACA,KAAK,sBAAsB;AACzBhB,kBAAQC,OAAOC,MAAM,IAAA;AACrBM,6BAAmBb,WAAAA;AACnBA,wBAAc;AACd;QACF;QACA,KAAK,iBAAiB;AACpB;QACF;QACA,KAAK,gBAAgB;AACnBK,kBAAQC,OAAOC,MAAM,MAAA;AACrB;QACF;MACF;IACF;AAEA,YAAQN,MAAMG,MAAI;MAChB,KAAK,WAAW;AACd,YAAI,CAACR,UAAUK,MAAMgB,QAAQC,SAAS,aAAa;AACjDb,kBAAQC,OAAOC,MAAM,GAAGN,MAAMgB,QAAQC,KAAKC,YAAW,CAAA;;CAAQ;AAC9D,qBAAWG,SAASrB,MAAMgB,QAAQd,SAAS;AACzC,oBAAQmB,MAAMlB,MAAI;cAChB,KAAK;AACHC,wBAAQC,OAAOC,MAAMe,MAAMd,OAAO,IAAA;AAClCK,mCAAmBS,MAAMd,IAAI;AAC7B;cACF,KAAK;AACHH,wBAAQC,OAAOC,MAAM,2BAAiBe,MAAMb,IAAI;CAAI;AACpDJ,wBAAQC,OAAOC,MAAM,KAAKgB,KAAKC,UAAUF,MAAMG,KAAK,CAAA;CAAK;AACzD;cACF,KAAK;AACH,oBAAIH,MAAMI,SAAS;AACjBrB,0BAAQC,OAAOC,MAAM,uBAAA;AACrBF,0BAAQC,OAAOC,MAAMe,MAAMnB,UAAU,IAAA;gBACvC,OAAO;AACLE,0BAAQC,OAAOC,MAAM,yBAAA;AACrBF,0BAAQC,OAAOC,MAAMe,MAAMnB,UAAU,IAAA;gBACvC;YACJ;AACAE,oBAAQC,OAAOC,MAAM,MAAA;UACvB;QACF;AACA;MACF;IACF;EACF;AACF;;;AC3HA,SAASoB,UAAUC,SAAS;AAE5B,SAASC,oBAAoB;AAC7B,SAASC,WAAW;;AAMb,IAAMC,mBAAmB,CAACC,YAAwBC,cAA8B,CAAA,MACrFC,WAAW;EACTC,MAAM;EACNC,aACE,6FACCH,YAAYI,SAAS,IAAI;;EAAOC,mBAAmBL,WAAAA,CAAAA,KAAiB;EACvEM,QAAQC,EAAEC,OAAO;IACfC,OAAOF,EAAEG,OAAOC,YAAY;MAC1BR,aAAa;;;;;;;;;;;;;;;;;;;;;IAqBf,CAAA;EACF,CAAA;EACAS,SAAS,OAAO,EAAEH,MAAK,MAAE;AACvB,QAAI;AACFI,UAAI,gBAAgB;QAAEJ;MAAM,GAAA;;;;;;AAC5B,YAAMK,UAAU,MAAMC,aAAahB,YAAYU,KAAAA;AAC/CI,UAAI,kBAAkB;QAAEC;MAAQ,GAAA;;;;;;AAChC,aAAOE,cAAcC,QAAQH,OAAAA;IAC/B,SAASI,GAAQ;AACfL,UAAIM,MAAMD,GAAAA,QAAAA;;;;;;AACV,aAAOF,cAAcI,MAAMF,EAAEG,OAAO;IACtC;EACF;AACF,CAAA;AAEK,IAAMhB,qBAAqB,CAACL,gBAAgCsB;;;;;;MAM7DC,uBAAuBvB,YAAYwB,IAAI,CAAClB,WAAWmB,aAAanB,MAAAA,CAAAA,CAAAA,CAAAA;;;;;;;;;;;;;;ACzDtE,SAASoB,aAAa;AAEtB,SAASC,yBAAyBC,UAAUC,gBAAAA,qBAAmD;AAC/F,SAASC,iBAAiB;AAC1B,SAASC,QAAQC,eAAe;;AAWzB,IAAMC,eAAe,CAACC,OAAAA;AAC3B,QAAMC,cAAcD,GAAGE,IACrBC,OAAOC,SAAS;IACdC,MAAM;EACR,CAAA,CAAA;AAEF,QAAMC,gBAAgBN,GAAGE,IACvBC,OAAOC,SAAS;IACdC,MAAM;EACR,CAAA,CAAA;AAEF,QAAME,cAAcP,GAAGE,IACrBC,OAAOC,SAAS;IACdC,MAAM;EACR,CAAA,CAAA;AAEF,QAAMG,cAAcR,GAAGE,IACrBC,OAAOC,SAAS;IACdC,MAAM;EACR,CAAA,CAAA;AAGF,QAAMI,kBAAkBT,GAAGE,IACzBC,OAAOO,SAAS;IACdL,MAAM;EACR,CAAA,CAAA;AAEF,QAAMM,cAAcX,GAAGE,IACrBC,OAAOO,SAAS;IACdL,MAAM;EACR,CAAA,CAAA;AAEF,QAAMO,iBAAiBZ,GAAGE,IACxBC,OAAOO,SAAS;IACdL,MAAM;EACR,CAAA,CAAA;AAGF,QAAMQ,iBAAiBb,GAAGE,IACxBC,OAAOW,MAAM;IACXT,MAAM;IACNU,SAASC,QAAQP,eAAAA;IACjBQ,UAAUD,QAAQV,aAAAA;EACpB,CAAA,CAAA;AAEF,QAAMY,iBAAiBlB,GAAGE,IACxBC,OAAOW,MAAM;IACXT,MAAM;IACNU,SAASC,QAAQP,eAAAA;IACjBQ,UAAUD,QAAQf,WAAAA;EACpB,CAAA,CAAA;AAEF,QAAMkB,iBAAiBnB,GAAGE,IACxBC,OAAOW,MAAM;IACXT,MAAM;IACNU,SAASC,QAAQP,eAAAA;IACjBQ,UAAUD,QAAQV,aAAAA;EACpB,CAAA,CAAA;AAEF,QAAMc,iBAAiBpB,GAAGE,IACxBC,OAAOW,MAAM;IACXT,MAAM;IACNU,SAASC,QAAQP,eAAAA;IACjBQ,UAAUD,QAAQR,WAAAA;EACpB,CAAA,CAAA;AAEF,QAAMa,iBAAiBrB,GAAGE,IACxBC,OAAOW,MAAM;IACXT,MAAM;IACNU,SAASC,QAAQP,eAAAA;IACjBQ,UAAUD,QAAQf,WAAAA;EACpB,CAAA,CAAA;AAEF,QAAMqB,aAAatB,GAAGE,IACpBC,OAAOW,MAAM;IACXT,MAAM;IACNU,SAASC,QAAQL,WAAAA;IACjBM,UAAUD,QAAQT,WAAAA;EACpB,CAAA,CAAA;AAEF,QAAMgB,aAAavB,GAAGE,IACpBC,OAAOW,MAAM;IACXT,MAAM;IACNU,SAASC,QAAQL,WAAAA;IACjBM,UAAUD,QAAQf,WAAAA;EACpB,CAAA,CAAA;AAEF,QAAMuB,gBAAgBxB,GAAGE,IACvBC,OAAOW,MAAM;IACXT,MAAM;IACNU,SAASC,QAAQJ,cAAAA;IACjBK,UAAUD,QAAQR,WAAAA;EACpB,CAAA,CAAA;AAEF,QAAMiB,gBAAgBzB,GAAGE,IACvBC,OAAOW,MAAM;IACXT,MAAM;IACNU,SAASC,QAAQJ,cAAAA;IACjBK,UAAUD,QAAQV,aAAAA;EACpB,CAAA,CAAA;AAGF,QAAMoB,WAAW1B,GAAGE,IAClBC,OAAOwB,KAAK;IACVtB,MAAM;IACNuB,WAAW;MAACZ,QAAQf,WAAAA;MAAce,QAAQV,aAAAA;MAAgBU,QAAQT,WAAAA;;IAClEsB,UAAU;MAACb,QAAQL,WAAAA;;EACrB,CAAA,CAAA;AAEF,QAAMmB,iBAAiB9B,GAAGE,IACxBC,OAAOwB,KAAK;IACVtB,MAAM;IACNuB,WAAW;MAACZ,QAAQV,aAAAA;MAAgBU,QAAQf,WAAAA;;IAC5C4B,UAAU;MAACb,QAAQP,eAAAA;;EACrB,CAAA,CAAA;AAEJ;AAGO,IAAMsB,iBAAiB,MAAA;AAC5B,QAAMC,aAAa,IAAIC,eAAAA;AAEvB,QAAMhC,cAAc+B,WAAW9B,IAAIE,SAAS;IAC1C8B,IAAIC,SAASC,OAAM;IACnB/B,MAAM;EACR,CAAA;AACA,QAAMC,gBAAgB0B,WAAW9B,IAAIE,SAAS;IAC5C8B,IAAIC,SAASC,OAAM;IACnB/B,MAAM;EACR,CAAA;AACA,QAAME,cAAcyB,WAAW9B,IAAIE,SAAS;IAC1C8B,IAAIC,SAASC,OAAM;IACnB/B,MAAM;EACR,CAAA;AACA,QAAMG,cAAcwB,WAAW9B,IAAIE,SAAS;IAC1C8B,IAAIC,SAASC,OAAM;IACnB/B,MAAM;EACR,CAAA;AAEA,QAAMI,kBAAkBuB,WAAW9B,IAAIQ,SAAS;IAC9CwB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;EACR,CAAA;AACA,QAAMM,cAAcqB,WAAW9B,IAAIQ,SAAS;IAC1CwB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;EACR,CAAA;AACA,QAAMO,iBAAiBoB,WAAW9B,IAAIQ,SAAS;IAC7CwB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;EACR,CAAA;AAEA,QAAMQ,iBAAiBmB,WAAW9B,IAAIY,MAAM;IAC1CoB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNU,SAASC,QAAQP,eAAAA;IACjBQ,UAAUD,QAAQV,aAAAA;EACpB,CAAA;AACA,QAAMY,iBAAiBc,WAAW9B,IAAIY,MAAM;IAC1CoB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNU,SAASC,QAAQP,eAAAA;IACjBQ,UAAUD,QAAQf,WAAAA;EACpB,CAAA;AACA,QAAMkB,iBAAiBa,WAAW9B,IAAIY,MAAM;IAC1CoB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNU,SAASC,QAAQP,eAAAA;IACjBQ,UAAUD,QAAQV,aAAAA;EACpB,CAAA;AACA,QAAMc,iBAAiBY,WAAW9B,IAAIY,MAAM;IAC1CoB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNU,SAASC,QAAQP,eAAAA;IACjBQ,UAAUD,QAAQR,WAAAA;EACpB,CAAA;AACA,QAAMa,iBAAiBW,WAAW9B,IAAIY,MAAM;IAC1CoB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNU,SAASC,QAAQP,eAAAA;IACjBQ,UAAUD,QAAQf,WAAAA;EACpB,CAAA;AACA,QAAMqB,aAAaU,WAAW9B,IAAIY,MAAM;IACtCoB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNU,SAASC,QAAQL,WAAAA;IACjBM,UAAUD,QAAQT,WAAAA;EACpB,CAAA;AACA,QAAMgB,aAAaS,WAAW9B,IAAIY,MAAM;IACtCoB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNU,SAASC,QAAQL,WAAAA;IACjBM,UAAUD,QAAQf,WAAAA;EACpB,CAAA;AACA,QAAMuB,gBAAgBQ,WAAW9B,IAAIY,MAAM;IACzCoB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNU,SAASC,QAAQJ,cAAAA;IACjBK,UAAUD,QAAQR,WAAAA;EACpB,CAAA;AACA,QAAMiB,gBAAgBO,WAAW9B,IAAIY,MAAM;IACzCoB,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNU,SAASC,QAAQJ,cAAAA;IACjBK,UAAUD,QAAQV,aAAAA;EACpB,CAAA;AAEA,QAAMoB,WAAWM,WAAW9B,IAAIyB,KAAK;IACnCO,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNuB,WAAW;MAACZ,QAAQf,WAAAA;MAAce,QAAQV,aAAAA;MAAgBU,QAAQT,WAAAA;;IAClEsB,UAAU;MAACb,QAAQL,WAAAA;;EACrB,CAAA;AACA,QAAMmB,iBAAiBE,WAAW9B,IAAIyB,KAAK;IACzCO,IAAIC,SAASC,OAAM;IACnB/B,MAAM;IACNuB,WAAW;MAACZ,QAAQV,aAAAA;MAAgBU,QAAQf,WAAAA;;IAC5C4B,UAAU;MAACb,QAAQP,eAAAA;;EACrB,CAAA;AACAuB,aAAWK,aAAY;AAEvB,SAAOL;AACT;AAEO,IAAMC,iBAAN,MAAMA;EAAN;AACLK,mBAQI,CAAC;AAELC,kBAMI,CAAC;AAELC,iBAAgB,CAAA;AAChBC,yBAAgC,CAAA;;EAEhC,MAAMC,SAAS,EAAEC,MAAK,GAAwB;AAC5C,WAAO,KAAKH,MAAMI,OAAO,CAACC,SAAS,CAACF,SAASE,KAAKF,UAAUA,KAAAA;EAC9D;EAEA,MAAMG,iBAAiB,EAAEH,MAAK,GAAwB;AACpD,WAAO,KAAKF,cAAcG,OAAO,CAACG,iBAAiB,CAACJ,SAASI,aAAaJ,UAAUA,KAAAA;EACtF;EAEAzC,IAA4BqC,QAAWS,MAAwB;AAC7DC,cAAU,OAAOD,KAAKd,OAAO,UAAU,wBAAA;;;;;;;;;AACvC,UAAMgB,aAAaC,cAAaZ,MAAAA;AAChC,QAAI,CAAC,KAAKA,OAAOW,WAAWE,GAAG,GAAI;AACjC,WAAKb,OAAOW,WAAWE,GAAG,IAAK;QAC7BC,MAAMH;QACNI,UAAUf;MACZ;IACF;AAEA,UAAMgB,cAAcL,WAAWE,OAAOI,MAAM,IAAIC,MAAM,sBAAA,CAAA;AAEtD,SAAKnB,QAAQU,KAAKd,EAAE,IAAI;MACtBK;MACAL,IAAIc,KAAKd;MACTwB,SAASH;MACTP;IACF;AAEA,WAAOA;EACT;EAEAX,eAAe;AACb,SAAKG,QAAQ,CAAA;AACb,SAAKC,gBAAgB,CAAA;AAErB,eAAWkB,UAAUC,OAAOC,OAAO,KAAKvB,OAAO,GAAG;AAChD,WAAKE,MAAMsB,KAAK;QACdC,MAAM;QACN7B,IAAIyB,OAAOzB;QACXS,OAAOqB,gBAAgBL,OAAOD,OAAO;QACrCO,YAAYN,OAAOX;MACrB,CAAA;IACF;AAEA,eAAWW,UAAUC,OAAOC,OAAO,KAAKvB,OAAO,GAAG;AAChD,YAAMC,SAAS,KAAKA,OAAOoB,OAAOD,OAAO,EAAEL;AAC3C,YAAME,cAAchB,OAAOa,OAAOI,MAAM,IAAIC,MAAM,sBAAA,CAAA;AAElD,iBAAW,CAACpD,MAAM6D,UAAAA,KAAeN,OAAOO,QAAQ5B,OAAO0B,UAAU,GAAI;AACnE,YAAIC,WAAWd,QAAQgB,yBAAyB;AAC9C,gBAAMzB,QAAQ0B,gCAAgCd,aAAalD,IAAAA;AAE3D,gBAAMiE,SAAS,KAAK9B,MAAM+B,KAAK,CAAC1B,SAASA,KAAKX,OAAOyB,OAAOzB,EAAE;AAC9D,cAAI,CAACoC,QAAQ;AACX;UACF;AAEA,gBAAM,EAAEpC,IAAIsC,SAAQ,IAAMb,OAAOX,KAAa3C,IAAAA,EAAMoE;AAEpD,gBAAMA,SAAS,KAAKjC,MAAM+B,KAAK,CAAC1B,SAASA,KAAKX,OAAOsC,QAAAA;AACrD,cAAI,CAACC,QAAQ;AACX;UACF;AAEA,eAAKhC,cAAcqB,KAAK;YACtBC,MAAM;YACN7B,IAAI,GAAGyB,OAAOzB,EAAE,IAAI7B,IAAAA,IAAQoE,OAAOvC,EAAE;YACrCS;YACAsB,YAAY,CAAC;YACbK;YACAG;UACF,CAAA;QACF,WAAWP,WAAWQ,SAAS,WAAWR,WAAWS,OAAOvB,QAAQgB,yBAAyB;AAC3F,gBAAMzB,QAAQ0B,gCAAgCd,aAAalD,IAAAA;AAE3D,gBAAMiE,SAAS,KAAK9B,MAAM+B,KAAK,CAAC1B,SAASA,KAAKX,OAAOyB,OAAOzB,EAAE;AAC9D,cAAI,CAACoC,QAAQ;AACX;UACF;AAEA,qBAAW,EAAEpC,IAAIsC,SAAQ,KAAOb,OAAOX,KAAa3C,IAAAA,EAAMuE,IAAI,CAACC,MAAgBA,EAAEJ,MAAM,GAAG;AACxF,kBAAMA,SAAS,KAAKjC,MAAM+B,KAAK,CAAC1B,SAASA,KAAKX,OAAOsC,QAAAA;AACrD,gBAAI,CAACC,QAAQ;AACX;YACF;AAEA,iBAAKhC,cAAcqB,KAAK;cACtBC,MAAM;cACN7B,IAAI,GAAGyB,OAAOzB,EAAE,IAAI7B,IAAAA,IAAQmE,QAAAA;cAC5B7B;cACAsB,YAAY,CAAC;cACbK;cACAG;YACF,CAAA;UACF;QACF;MACF;IACF;EACF;AACF;;;ACzWA,SAASK,UAAUC,UAAS;AAE5B,SAASC,YAAYC,oBAAoBC,WAAW;AAM7C,IAAMC,YAAWC,GAAEC,OAAOC,YAAY;EAAEC,aAAa;AAAgC,CAAA;AAErF,IAAMC,UAAUJ,GAAEK,OAAO;EAC9BC,IAAIP;EACJQ,MAAMP,GAAEC,OAAOC,YAAY;IAAEC,aAAa;EAA0B,CAAA;EACpEK,OAAOR,GAAES,SAAST,GAAEC,MAAM,EAAEC,YAAY;IAAEC,aAAa;EAAiB,CAAA;AAC1E,CAAA,EACGO,KAAKV,GAAEW,OAAO,EACdT,YAAY;EACX,CAACU,kBAAAA,GAAqB;IACpBC,MAAMC,WAAWC;IACjBC,UAAU;IACVC,SAAS;EACX;EACAd,aAAa;AACf,CAAA;AAGK,IAAMe,UAAUlB,GAAEK,OAAO;EAC9BC,IAAIP;EACJQ,MAAMP,GAAEC,OAAOC,YAAY;IAAEC,aAAa;EAA2B,CAAA;EACrEA,aAAaH,GAAES,SAAST,GAAEC,MAAM,EAAEC,YAAY;IAAEC,aAAa;EAAkC,CAAA;AACjG,CAAA,EACGO,KAAKV,GAAEW,OAAO,EACdT,YAAY;EACX,CAACU,kBAAAA,GAAqB;IACpBC,MAAMC,WAAWC;IACjBC,UAAU;IACVC,SAAS;EACX;EACAd,aAAa;AACf,CAAA;AAGK,IAAMgB,OAAOnB,GAAEK,OAAO;EAC3BC,IAAIP;EACJQ,MAAMP,GAAEC,OAAOC,YAAY;IAAEC,aAAa;EAAwB,CAAA;EAClEA,aAAaH,GAAES,SAAST,GAAEC,MAAM,EAAEC,YAAY;IAAEC,aAAa;EAA+B,CAAA;EAC5FiB,SAASC,IAAIH,OAAAA;EACbI,UAAUD,IAAIjB,OAAAA;AAChB,CAAA,EACGM,KAAKV,GAAEW,OAAO,EACdT,YAAY;EACX,CAACU,kBAAAA,GAAqB;IACpBC,MAAMC,WAAWC;IACjBC,UAAU;IACVC,SAAS;EACX;EACAd,aAAa;AACf,CAAA;AAGK,IAAMoB,MAAMvB,GAAEK,OAAO;EAC1BC,IAAIP;EACJQ,MAAMP,GAAEC,OAAOC,YAAY;IAAEC,aAAa;EAAgC,CAAA;EAC1EqB,UAAUxB,GAAEyB,MAAMJ,IAAIH,OAAAA,CAAAA;EACtBQ,WAAW1B,GAAEyB,MAAMJ,IAAIjB,OAAAA,CAAAA;AACzB,CAAA,EACGM,KAAKV,GAAEW,OAAO,EACdT,YAAY;EACX,CAACU,kBAAAA,GAAqB;IACpBC,MAAMC,WAAWC;IACjBC,UAAU;IACVC,SAAS;EACX;EACAd,aAAa;AACf,CAAA;",
+  "names": ["AI_SERVICE_ENDPOINT", "LOCAL", "REMOTE", "createLogger", "stream", "filter", "onImage", "images", "currentText", "event", "printContentBlock", "content", "type", "process", "stdout", "write", "text", "name", "id", "source", "media_type", "onTextBlockPrinted", "imageMatch", "match", "description", "message", "role", "toUpperCase", "delta", "partial_json", "block", "JSON", "stringify", "input", "isError", "Schema", "S", "toJsonSchema", "log", "createCypherTool", "dataSource", "schemaTypes", "defineTool", "name", "description", "length", "createSystemPrompt", "schema", "S", "Struct", "query", "String", "annotations", "execute", "log", "results", "executeQuery", "LLMToolResult", "Success", "e", "catch", "Error", "message", "trim", "formatJsonSchemaForLLM", "map", "toJsonSchema", "raise", "JSON_SCHEMA_ECHO_REF_ID", "ObjectId", "toJsonSchema", "invariant", "create", "makeRef", "seedTestData", "db", "contactRich", "add", "create", "Contact", "name", "contactJosiah", "contactDima", "contactFred", "projectComposer", "Project", "projectEcho", "projectDoodles", "_taskComposer1", "Task", "project", "makeRef", "assignee", "_taskComposer2", "_taskComposer3", "_taskComposer5", "_taskComposer4", "_taskEcho1", "_taskEcho2", "_taskDoodles1", "_taskDoodles2", "_orgDxos", "Org", "employees", "projects", "_orgBraneframe", "createTestData", "dataSource", "MockDataSource", "id", "ObjectId", "random", "computeGraph", "objects", "schema", "nodes", "relationships", "getNodes", "label", "filter", "node", "getRelationships", "relationship", "data", "invariant", "jsonSchema", "toJsonSchema", "$id", "json", "instance", "typenameDxn", "raise", "Error", "typeDxn", "object", "Object", "values", "push", "kind", "formatNodeLabel", "properties", "propSchema", "entries", "JSON_SCHEMA_ECHO_REF_ID", "formatInferredRelationshipLabel", "source", "find", "targetId", "target", "type", "items", "map", "x", "Schema", "S", "EntityKind", "ObjectAnnotationId", "Ref", "ObjectId", "S", "String", "annotations", "description", "Contact", "Struct", "id", "name", "email", "optional", "pipe", "mutable", "ObjectAnnotationId", "kind", "EntityKind", "Object", "typename", "version", "Project", "Task", "project", "Ref", "assignee", "Org", "projects", "Array", "employees"]
 }

package/dist/types/src/ai-service/client.d.ts

@@ -1,48 +1,31 @@
-import type { SpaceId } from '@dxos/keys';
-import { Message, type GenerateRequest, type ObjectId, type ResultStreamEvent, type Space, type Thread } from './schema';
-export type AIServiceClientParams = {
-    endpoint: string;
-};
+import { type SpaceId } from '@dxos/keys';
+import { GenerationStream } from './stream';
+import { Message, type GenerateRequest, type Space, type Thread } from './types';
 export interface AIServiceClient {
     getSpace(spaceId: SpaceId): Promise<Space>;
     getThread(spaceId: SpaceId, threadId: string): Promise<Thread>;
-    getMessagesInThread(spaceId: SpaceId, threadId: string): Promise<Message[]>;
-    insertMessages(messages: Message[]): Promise<void>;
+    getMessages(spaceId: SpaceId, threadId: string): Promise<Message[]>;
+    appendMessages(messages: Message[]): Promise<void>;
     updateMessage(spaceId: SpaceId, threadId: string, messageId: string, message: Message): Promise<void>;
     generate(request: GenerateRequest): Promise<GenerationStream>;
 }
+export type AIServiceClientParams = {
+    endpoint: string;
+};
+/**
+ * Edge GPT client.
+ */
 export declare class AIServiceClientImpl implements AIServiceClient {
     private readonly _endpoint;
     constructor({ endpoint }: AIServiceClientParams);
     getSpace(spaceId: SpaceId): Promise<Space>;
     getThread(spaceId: SpaceId, threadId: string): Promise<Thread>;
-    getMessagesInThread(spaceId: SpaceId, threadId: string): Promise<Message[]>;
-    insertMessages(messages: Message[]): Promise<void>;
+    getMessages(spaceId: SpaceId, threadId: string): Promise<Message[]>;
+    appendMessages(messages: Message[]): Promise<void>;
     updateMessage(spaceId: SpaceId, threadId: string, messageId: string, message: Message): Promise<void>;
-    generate(request: GenerateRequest): Promise<GenerationStream>;
-}
-type GenerationParams = {
-    spaceId: SpaceId;
-    threadId: ObjectId;
-};
-export declare class GenerationStream implements AsyncIterable<ResultStreamEvent> {
-    private readonly _params;
-    private readonly _getIterator;
-    static fromSSEResponse(params: GenerationParams, response: Response): GenerationStream;
-    private _previousMessages;
-    private _accumulatedMessage?;
-    private _done;
-    private _iterator?;
-    constructor(_params: GenerationParams, _getIterator: () => AsyncIterableIterator<ResultStreamEvent>);
-    [Symbol.asyncIterator](): AsyncIterator<ResultStreamEvent>;
-    cancel(): void;
-    get accumulatedMessages(): Message[];
     /**
-     * Returns the complete message list generated by the AI service.
+     * Open message stream.
      */
-    complete(): Promise<Message[]>;
-    private _createIterator;
-    private _processEvent;
+    generate(request: GenerateRequest): Promise<GenerationStream>;
 }
-export {};
 //# sourceMappingURL=client.d.ts.map

package/dist/types/src/ai-service/client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/client.ts"],"names":[],"mappings":"AAUA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAE1C,OAAO,EACL,OAAO,EACP,KAAK,eAAe,EACpB,KAAK,QAAQ,EACb,KAAK,iBAAiB,EACtB,KAAK,KAAK,EACV,KAAK,MAAM,EACZ,MAAM,UAAU,CAAC;AAIlB,MAAM,MAAM,qBAAqB,GAAG;IAClC,QAAQ,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF,MAAM,WAAW,eAAe;IAC9B,QAAQ,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;IAC3C,SAAS,CAAC,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IAC/D,mBAAmB,CAAC,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IAC5E,cAAc,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACnD,aAAa,CAAC,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACtG,QAAQ,CAAC,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAAC;CAC/D;AAED,qBAAa,mBAAoB,YAAW,eAAe;IACzD,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;gBACvB,EAAE,QAAQ,EAAE,EAAE,qBAAqB;IAIzC,QAAQ,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC;IAI1C,SAAS,CAAC,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAI9D,mBAAmB,CAAC,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IAK3E,cAAc,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAelD,aAAa,CAAC,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IAIrG,QAAQ,CAAC,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAgBpE;AAED,KAAK,gBAAgB,GAAG;IACtB,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,EAAE,QAAQ,CAAC;CACpB,CAAC;AAEF,qBAAa,gBAAiB,YAAW,aAAa,CAAC,iBAAiB,CAAC;IAkDrE,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,YAAY;IAlD/B,MAAM,CAAC,eAAe,CAAC,MAAM,EAAE,gBAAgB,EAAE,QAAQ,EAAE,QAAQ;IA2CnE,OAAO,CAAC,iBAAiB,CAAiB;IAC1C,OAAO,CAAC,mBAAmB,CAAC,CAAsB;IAClD,OAAO,CAAC,KAAK,CAAiB;IAC9B,OAAO,CAAC,SAAS,CAAC,CAA+C;gBAG9C,OAAO,EAAE,gBAAgB,EACzB,YAAY,EAAE,MAAM,qBAAqB,CAAC,iBAAiB,CAAC;IAG/E,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,aAAa,CAAC,iBAAiB,CAAC;IAI1D,MAAM;IAIN,IAAI,mBAAmB,IAAI,OAAO,EAAE,CAGnC;IAED;;OAEG;IACG,QAAQ,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;IAKpC,OAAO,CAAC,eAAe;IAUvB,OAAO,CAAC,aAAa;CA4DtB"}
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/client.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,KAAK,OAAO,EAAE,MAAM,YAAY,CAAC;AAE1C,OAAO,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAC5C,OAAO,EAAE,OAAO,EAAE,KAAK,eAAe,EAAE,KAAK,KAAK,EAAE,KAAK,MAAM,EAAE,MAAM,SAAS,CAAC;AAEjF,MAAM,WAAW,eAAe;IAC9B,QAAQ,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;IAC3C,SAAS,CAAC,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IAC/D,WAAW,CAAC,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IACpE,cAAc,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACnD,aAAa,CAAC,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACtG,QAAQ,CAAC,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAAC;CAC/D;AAID,MAAM,MAAM,qBAAqB,GAAG;IAClC,QAAQ,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF;;GAEG;AACH,qBAAa,mBAAoB,YAAW,eAAe;IACzD,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;gBAEvB,EAAE,QAAQ,EAAE,EAAE,qBAAqB;IAIzC,QAAQ,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC;IAI1C,SAAS,CAAC,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAI9D,WAAW,CAAC,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IAKnE,cAAc,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAgBlD,aAAa,CAAC,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IAI3G;;OAEG;IACG,QAAQ,CAAC,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAqBpE"}

package/dist/types/src/ai-service/index.d.ts

@@ -1,3 +1,4 @@
 export * from './client';
-export * from './schema';
+export * from './stream';
+export * from './types';
 //# sourceMappingURL=index.d.ts.map

package/dist/types/src/ai-service/index.d.ts.map

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

package/dist/types/src/ai-service/stream.d.ts

@@ -0,0 +1,34 @@
+import { type ObjectId } from '@dxos/echo-schema';
+import { type SpaceId } from '@dxos/keys';
+import { type Message, type ResultStreamEvent } from './types';
+export type GenerationParams = {
+    spaceId?: SpaceId;
+    threadId?: ObjectId;
+};
+export declare class GenerationStream implements AsyncIterable<ResultStreamEvent> {
+    private readonly _controller;
+    private readonly _getIterator;
+    private readonly _params;
+    /**
+     * Creates a stream from an SSE response.
+     */
+    static fromSSEResponse(params: GenerationParams, response: Response, controller?: AbortController): GenerationStream;
+    private readonly _done;
+    private _previousMessages;
+    private _accumulatedMessage?;
+    private _iterator?;
+    constructor(_controller: AbortController, _getIterator: () => AsyncIterableIterator<ResultStreamEvent>, _params: GenerationParams);
+    [Symbol.asyncIterator](): AsyncIterator<ResultStreamEvent>;
+    get accumulatedMessages(): Message[];
+    /**
+     * Aborts the stream.
+     */
+    abort(): void;
+    /**
+     * Returns the complete message list generated by the AI service.
+     */
+    complete(): Promise<Message[]>;
+    private _createIterator;
+    private _processEvent;
+}
+//# sourceMappingURL=stream.d.ts.map

package/dist/types/src/ai-service/stream.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/stream.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,KAAK,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAElD,OAAO,EAAE,KAAK,OAAO,EAAE,MAAM,YAAY,CAAC;AAG1C,OAAO,EAAE,KAAK,OAAO,EAAE,KAAK,iBAAiB,EAAE,MAAM,SAAS,CAAC;AAG/D,MAAM,MAAM,gBAAgB,GAAG;IAC7B,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,QAAQ,CAAC,EAAE,QAAQ,CAAC;CACrB,CAAC;AAEF,qBAAa,gBAAiB,YAAW,aAAa,CAAC,iBAAiB,CAAC;IAmDrE,OAAO,CAAC,QAAQ,CAAC,WAAW;IAC5B,OAAO,CAAC,QAAQ,CAAC,YAAY;IAC7B,OAAO,CAAC,QAAQ,CAAC,OAAO;IApD1B;;OAEG;IACH,MAAM,CAAC,eAAe,CAAC,MAAM,EAAE,gBAAgB,EAAE,QAAQ,EAAE,QAAQ,EAAE,UAAU,kBAAwB;IAyCvG,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAiB;IACvC,OAAO,CAAC,iBAAiB,CAAiB;IAC1C,OAAO,CAAC,mBAAmB,CAAC,CAAsB;IAClD,OAAO,CAAC,SAAS,CAAC,CAA+C;gBAG9C,WAAW,EAAE,eAAe,EAC5B,YAAY,EAAE,MAAM,qBAAqB,CAAC,iBAAiB,CAAC,EAC5D,OAAO,EAAE,gBAAgB;IAG5C,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,aAAa,CAAC,iBAAiB,CAAC;IAI1D,IAAI,mBAAmB,IAAI,OAAO,EAAE,CAGnC;IAED;;OAEG;IACH,KAAK;IAIL;;OAEG;IACG,QAAQ,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;IAKpC,OAAO,CAAC,eAAe;IAkBvB,OAAO,CAAC,aAAa;CAgEtB"}

package/dist/types/src/ai-service/stream.test.d.ts

@@ -0,0 +1,2 @@
+export declare const createTestStream: (blocks: string[]) => ReadableStream;
+//# sourceMappingURL=stream.test.d.ts.map

package/dist/types/src/ai-service/stream.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream.test.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/stream.test.ts"],"names":[],"mappings":"AAeA,eAAO,MAAM,gBAAgB,WAAY,MAAM,EAAE,KAAG,cA2CnD,CAAC"}

package/dist/types/src/ai-service/types/index.d.ts

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

package/dist/types/src/ai-service/types/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../src/ai-service/types/index.ts"],"names":[],"mappings":"AAIA,cAAc,WAAW,CAAC;AAC1B,cAAc,UAAU,CAAC"}

package/dist/types/src/ai-service/types/message.d.ts

@@ -0,0 +1,191 @@
+import { Schema as S } from '@effect/schema';
+import { SpaceId } from '@dxos/keys';
+export declare const SpaceIdSchema: S.Schema<SpaceId, string>;
+export declare const Space: S.Struct<{
+    id: S.Schema<SpaceId, string, never>;
+}>;
+export interface Space extends S.Schema.Type<typeof Space> {
+}
+/**
+ * Thread.
+ */
+export declare const Thread: S.Struct<{
+    id: import("@dxos/echo-schema").ObjectIdClass;
+    spaceId: S.Schema<SpaceId, string, never>;
+}>;
+export interface Thread extends S.Schema.Type<typeof Thread> {
+}
+export declare const MessageRole: S.refine<"user" | "assistant", S.Schema<string, string, never>>;
+export type MessageRole = S.Schema.Type<typeof MessageRole>;
+/**
+ * Tagged block.
+ */
+export declare const TagContentBlock: S.mutable<S.Struct<{
+    type: S.Literal<["tag"]>;
+    tag: typeof S.String;
+    text: typeof S.String;
+}>>;
+export type TagContentBlock = S.Schema.Type<typeof TagContentBlock>;
+/**
+ * Text.
+ */
+export declare const TextContentBlock: S.mutable<S.Struct<{
+    type: S.Literal<["text"]>;
+    text: typeof S.String;
+}>>;
+export type TextContentBlock = S.Schema.Type<typeof TextContentBlock>;
+/**
+ * Image.
+ */
+export declare const ImageSource: S.mutable<S.Struct<{
+    type: S.Literal<["base64"]>;
+    media_type: S.Literal<["image/jpeg", "image/png", "image/gif", "image/webp"]>;
+    data: typeof S.String;
+}>>;
+export type ImageSource = S.Schema.Type<typeof ImageSource>;
+export declare const ImageContentBlock: S.mutable<S.Struct<{
+    type: S.Literal<["image"]>;
+    id: S.optional<typeof S.String>;
+    source: S.optional<S.mutable<S.Struct<{
+        type: S.Literal<["base64"]>;
+        media_type: S.Literal<["image/jpeg", "image/png", "image/gif", "image/webp"]>;
+        data: typeof S.String;
+    }>>>;
+}>>;
+export type ImageContentBlock = S.Schema.Type<typeof ImageContentBlock>;
+/**
+ * Tool use.
+ */
+export declare const ToolUseContentBlock: S.mutable<S.Struct<{
+    type: S.Literal<["tool_use"]>;
+    /**
+     * Opaque service-defined ID of the tool invocation.
+     * Used to match the tool result block.
+     */
+    id: typeof S.String;
+    /**
+     * Tool name.
+     */
+    name: typeof S.String;
+    input: typeof S.Unknown;
+    /**
+     * Used to accumulate the partial tool input JSON in streaming mode.
+     */
+    inputJson: S.optional<typeof S.String>;
+}>>;
+export declare const ToolResultContentBlock: S.Struct<{
+    type: S.Literal<["tool_result"]>;
+    toolUseId: typeof S.String;
+    content: typeof S.String;
+    isError: S.optional<typeof S.Boolean>;
+}>;
+/**
+ * Content union.
+ */
+export declare const MessageContentBlock: S.Union<[S.mutable<S.Struct<{
+    type: S.Literal<["tag"]>;
+    tag: typeof S.String;
+    text: typeof S.String;
+}>>, S.mutable<S.Struct<{
+    type: S.Literal<["text"]>;
+    text: typeof S.String;
+}>>, S.mutable<S.Struct<{
+    type: S.Literal<["image"]>;
+    id: S.optional<typeof S.String>;
+    source: S.optional<S.mutable<S.Struct<{
+        type: S.Literal<["base64"]>;
+        media_type: S.Literal<["image/jpeg", "image/png", "image/gif", "image/webp"]>;
+        data: typeof S.String;
+    }>>>;
+}>>, S.mutable<S.Struct<{
+    type: S.Literal<["tool_use"]>;
+    /**
+     * Opaque service-defined ID of the tool invocation.
+     * Used to match the tool result block.
+     */
+    id: typeof S.String;
+    /**
+     * Tool name.
+     */
+    name: typeof S.String;
+    input: typeof S.Unknown;
+    /**
+     * Used to accumulate the partial tool input JSON in streaming mode.
+     */
+    inputJson: S.optional<typeof S.String>;
+}>>, S.Struct<{
+    type: S.Literal<["tool_result"]>;
+    toolUseId: typeof S.String;
+    content: typeof S.String;
+    isError: S.optional<typeof S.Boolean>;
+}>]>;
+export type MessageContentBlock = S.Schema.Type<typeof MessageContentBlock>;
+export declare const Message: S.Schema<{
+    readonly id: string;
+    spaceId?: SpaceId | undefined;
+    content: ({
+        type: "tag";
+        tag: string;
+        text: string;
+    } | {
+        type: "text";
+        text: string;
+    } | {
+        id?: string | undefined;
+        type: "image";
+        source?: {
+            type: "base64";
+            media_type: "image/jpeg" | "image/png" | "image/gif" | "image/webp";
+            data: string;
+        } | undefined;
+    } | {
+        id: string;
+        type: "tool_use";
+        name: string;
+        input: unknown;
+        inputJson?: string | undefined;
+    } | {
+        readonly type: "tool_result";
+        readonly toolUseId: string;
+        readonly content: string;
+        readonly isError?: boolean | undefined;
+    })[];
+    threadId?: string | undefined;
+    foreignId?: string | undefined;
+    role: "user" | "assistant";
+}, {
+    readonly id: string;
+    spaceId?: SpaceId | undefined;
+    content: ({
+        type: "tag";
+        tag: string;
+        text: string;
+    } | {
+        type: "text";
+        text: string;
+    } | {
+        id?: string | undefined;
+        type: "image";
+        source?: {
+            type: "base64";
+            media_type: "image/jpeg" | "image/png" | "image/gif" | "image/webp";
+            data: string;
+        } | undefined;
+    } | {
+        id: string;
+        type: "tool_use";
+        name: string;
+        input: unknown;
+        inputJson?: string | undefined;
+    } | {
+        readonly type: "tool_result";
+        readonly toolUseId: string;
+        readonly content: string;
+        readonly isError?: boolean | undefined;
+    })[];
+    threadId?: string | undefined;
+    foreignId?: string | undefined;
+    role: "user" | "assistant";
+}, never>;
+export type Message = S.Schema.Type<typeof Message>;
+//# sourceMappingURL=message.d.ts.map

package/dist/types/src/ai-service/types/message.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"message.d.ts","sourceRoot":"","sources":["../../../../../src/ai-service/types/message.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,MAAM,IAAI,CAAC,EAAE,MAAM,gBAAgB,CAAC;AAG7C,OAAO,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAKrC,eAAO,MAAM,aAAa,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAA4C,CAAC;AAEjG,eAAO,MAAM,KAAK;;EAEhB,CAAC;AAEH,MAAM,WAAW,KAAM,SAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,KAAK,CAAC;CAAG;AAE7D;;GAEG;AACH,eAAO,MAAM,MAAM;;;EAGjB,CAAC;AAEH,MAAM,WAAW,MAAO,SAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,MAAM,CAAC;CAAG;AAE/D,eAAO,MAAM,WAAW,iEAA6E,CAAC;AACtG,MAAM,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,WAAW,CAAC,CAAC;AAE5D;;GAEG;AACH,eAAO,MAAM,eAAe;;;;GAIV,CAAC;AAEnB,MAAM,MAAM,eAAe,GAAG,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,eAAe,CAAC,CAAC;AAEpE;;GAEG;AACH,eAAO,MAAM,gBAAgB;;;GAGX,CAAC;AAEnB,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,gBAAgB,CAAC,CAAC;AAEtE;;GAEG;AACH,eAAO,MAAM,WAAW;;;;GAIN,CAAC;AAEnB,MAAM,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,WAAW,CAAC,CAAC;AAE5D,eAAO,MAAM,iBAAiB;;;;;;;;GAIZ,CAAC;AAEnB,MAAM,MAAM,iBAAiB,GAAG,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,iBAAiB,CAAC,CAAC;AAExE;;GAEG;AACH,eAAO,MAAM,mBAAmB;;IAG9B;;;OAGG;;IAGH;;OAEG;;;IAKH;;OAEG;;GAGa,CAAC;AAEnB,eAAO,MAAM,sBAAsB;;;;;EAKjC,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,mBAAmB;;;;;;;;;;;;;;;;;IA9B9B;;;OAGG;;IAGH;;OAEG;;;IAKH;;OAEG;;;;;;;IAqBJ,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,mBAAmB,CAAC,CAAC;AA2B5E,eAAO,MAAM,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;SAAmE,CAAC;AAExF,MAAM,MAAM,OAAO,GAAG,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,OAAO,CAAC,CAAC"}