@dxos/assistant 0.8.4-main.e098934 → 0.8.4-main.ead640a

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (86) hide show
  1. package/dist/lib/browser/{chunk-3K5SRE5A.mjs → chunk-CWGVG235.mjs} +59 -83
  2. package/dist/lib/browser/chunk-CWGVG235.mjs.map +7 -0
  3. package/dist/lib/browser/extraction/index.mjs +17 -13
  4. package/dist/lib/browser/extraction/index.mjs.map +4 -4
  5. package/dist/lib/browser/index.mjs +550 -86
  6. package/dist/lib/browser/index.mjs.map +4 -4
  7. package/dist/lib/browser/meta.json +1 -1
  8. package/dist/lib/node-esm/{chunk-W7AIB6DP.mjs → chunk-OV5PMQPB.mjs} +59 -83
  9. package/dist/lib/node-esm/chunk-OV5PMQPB.mjs.map +7 -0
  10. package/dist/lib/node-esm/extraction/index.mjs +17 -13
  11. package/dist/lib/node-esm/extraction/index.mjs.map +4 -4
  12. package/dist/lib/node-esm/index.mjs +550 -86
  13. package/dist/lib/node-esm/index.mjs.map +4 -4
  14. package/dist/lib/node-esm/meta.json +1 -1
  15. package/dist/types/src/conversation/context.d.ts +14 -7
  16. package/dist/types/src/conversation/context.d.ts.map +1 -1
  17. package/dist/types/src/conversation/conversation.d.ts +9 -10
  18. package/dist/types/src/conversation/conversation.d.ts.map +1 -1
  19. package/dist/types/src/errors.d.ts +8 -8
  20. package/dist/types/src/execution-graph.d.ts +82 -0
  21. package/dist/types/src/execution-graph.d.ts.map +1 -0
  22. package/dist/types/src/execution-graph.test.d.ts +2 -0
  23. package/dist/types/src/execution-graph.test.d.ts.map +1 -0
  24. package/dist/types/src/extraction/extraction-llm-function/extraction-llm-function.d.ts.map +1 -1
  25. package/dist/types/src/extraction/extraction-ner-function/extraction-ner-function.d.ts +61 -57
  26. package/dist/types/src/extraction/extraction-ner-function/extraction-ner-function.d.ts.map +1 -1
  27. package/dist/types/src/extraction/extraction.d.ts +10 -6
  28. package/dist/types/src/extraction/extraction.d.ts.map +1 -1
  29. package/dist/types/src/extraction/quotes.d.ts +1 -1
  30. package/dist/types/src/extraction/quotes.d.ts.map +1 -1
  31. package/dist/types/src/functions/services.d.ts +17 -5
  32. package/dist/types/src/functions/services.d.ts.map +1 -1
  33. package/dist/types/src/index.d.ts +2 -1
  34. package/dist/types/src/index.d.ts.map +1 -1
  35. package/dist/types/src/session/artifact-diff.d.ts +2 -2
  36. package/dist/types/src/session/artifact-diff.d.ts.map +1 -1
  37. package/dist/types/src/session/error-handling.d.ts +2 -2
  38. package/dist/types/src/session/error-handling.d.ts.map +1 -1
  39. package/dist/types/src/session/format.d.ts +33 -31
  40. package/dist/types/src/session/format.d.ts.map +1 -1
  41. package/dist/types/src/session/observer.d.ts +1 -1
  42. package/dist/types/src/session/observer.d.ts.map +1 -1
  43. package/dist/types/src/session/session.d.ts +13 -9
  44. package/dist/types/src/session/session.d.ts.map +1 -1
  45. package/dist/types/src/session/toolkit.d.ts +7 -6
  46. package/dist/types/src/session/toolkit.d.ts.map +1 -1
  47. package/dist/types/src/session/version-pin.d.ts +2 -2
  48. package/dist/types/src/session/version-pin.d.ts.map +1 -1
  49. package/dist/types/src/{types.d.ts → types/artifact.d.ts} +3 -3
  50. package/dist/types/src/types/artifact.d.ts.map +1 -0
  51. package/dist/types/src/types/index.d.ts +2 -0
  52. package/dist/types/src/types/index.d.ts.map +1 -0
  53. package/dist/types/tsconfig.tsbuildinfo +1 -1
  54. package/package.json +26 -31
  55. package/src/conversation/context.ts +49 -17
  56. package/src/conversation/conversation.ts +23 -18
  57. package/src/execution-graph.test.ts +738 -0
  58. package/src/execution-graph.ts +500 -0
  59. package/src/extraction/extraction-llm-function/extraction-llm-function.ts +3 -1
  60. package/src/extraction/extraction-ner-function/extraction-ner-function.ts +3 -1
  61. package/src/extraction/extraction-ner-function/named-entity-recognition.test.ts +2 -1
  62. package/src/extraction/extraction.ts +2 -2
  63. package/src/extraction/quotes.ts +2 -2
  64. package/src/functions/services.ts +85 -46
  65. package/src/index.ts +2 -1
  66. package/src/session/artifact-diff.ts +2 -2
  67. package/src/session/error-handling.ts +6 -4
  68. package/src/session/format.test.ts +1 -1
  69. package/src/session/format.ts +8 -5
  70. package/src/session/observer.ts +2 -1
  71. package/src/session/session.conversations.json +1 -0
  72. package/src/session/session.test.ts +50 -209
  73. package/src/session/session.ts +31 -20
  74. package/src/session/toolkit.ts +19 -13
  75. package/src/session/version-pin.ts +2 -2
  76. package/src/summarization/summarization.test.ts +8 -3
  77. package/src/templates/system.tpl +10 -12
  78. package/src/{types.ts → types/artifact.ts} +2 -2
  79. package/src/types/index.ts +5 -0
  80. package/dist/lib/browser/chunk-3K5SRE5A.mjs.map +0 -7
  81. package/dist/lib/node-esm/chunk-W7AIB6DP.mjs.map +0 -7
  82. package/dist/types/src/session/version-pin.test.d.ts +0 -2
  83. package/dist/types/src/session/version-pin.test.d.ts.map +0 -1
  84. package/dist/types/src/types.d.ts.map +0 -1
  85. package/src/session/__snapshots__/session.test.ts.snap +0 -32
  86. package/src/session/version-pin.test.ts +0 -7
@@ -1,12 +1,13 @@
1
1
  import { createRequire } from 'node:module';const require = createRequire(import.meta.url);
2
2
 
3
3
  // src/session/artifact-diff.ts
4
- import { Context } from "effect";
4
+ import * as Context from "effect/Context";
5
5
  var ArtifactDiffResolver = class extends Context.Tag("@dxos/assistant/ArtifactDiffResolver")() {
6
6
  };
7
7
 
8
8
  // src/session/observer.ts
9
- import { Effect, Function } from "effect";
9
+ import * as Effect from "effect/Effect";
10
+ import * as Function from "effect/Function";
10
11
  var GenerationObserver = Object.freeze({
11
12
  make: ({ onBegin = Function.constant(Effect.void), onPart = Function.constant(Effect.void), onBlock = Function.constant(Effect.void), onEnd = Function.constant(Effect.void), onMessage = Function.constant(Effect.void) } = {}) => ({
12
13
  onBegin,
@@ -43,8 +44,10 @@ var AiAssistantError = class extends BaseError.extend("AI_ASSISTANT_ERROR", "AI
43
44
  };
44
45
 
45
46
  // src/session/session.ts
46
- import { AiLanguageModel } from "@effect/ai";
47
- import { Chunk, Effect as Effect4, Stream } from "effect";
47
+ import * as LanguageModel from "@effect/ai/LanguageModel";
48
+ import * as Chunk from "effect/Chunk";
49
+ import * as Effect3 from "effect/Effect";
50
+ import * as Stream from "effect/Stream";
48
51
  import { AiParser, AiPreprocessor, callTool, getToolCalls } from "@dxos/ai";
49
52
  import { todo } from "@dxos/debug";
50
53
  import { Obj as Obj2 } from "@dxos/echo";
@@ -52,42 +55,11 @@ import { TracingService } from "@dxos/functions";
52
55
  import { log as log2 } from "@dxos/log";
53
56
  import { DataType as DataType2 } from "@dxos/schema";
54
57
 
55
- // src/session/error-handling.ts
56
- import { AiError } from "@effect/ai";
57
- import { HttpClientError } from "@effect/platform";
58
- import { Effect as Effect2, Schema } from "effect";
59
- var AnthropicErrorResponse = Schema.Struct({
60
- type: Schema.Literal("error"),
61
- error: Schema.Struct({
62
- type: Schema.String,
63
- message: Schema.String
64
- })
65
- });
66
- var mapAiError = (err) => Effect2.gen(function* () {
67
- const cause = err.cause;
68
- if (HttpClientError.isHttpClientError(cause) && cause.reason === "StatusCode") {
69
- const body = yield* cause.response.json.pipe(Effect2.flatMap(Schema.decodeUnknown(AnthropicErrorResponse, {
70
- exact: false
71
- })));
72
- const parsedCause = new AiModelError({
73
- message: body.error.message,
74
- context: {
75
- model: "anthropic",
76
- type: body.error.type
77
- }
78
- });
79
- return new AiError.AiError({
80
- description: body.error.message,
81
- module: err.module,
82
- method: err.method,
83
- cause: parsedCause
84
- });
85
- }
86
- return err;
87
- }).pipe(Effect2.catchAll(() => Effect2.succeed(err)));
88
-
89
58
  // src/session/format.ts
90
- import { Array, Effect as Effect3, Option, pipe } from "effect";
59
+ import * as Array from "effect/Array";
60
+ import * as Effect2 from "effect/Effect";
61
+ import * as Function2 from "effect/Function";
62
+ import * as Option from "effect/Option";
91
63
  import { Template } from "@dxos/blueprints";
92
64
  import { Obj } from "@dxos/echo";
93
65
  import { ObjectVersion } from "@dxos/echo-db";
@@ -96,36 +68,36 @@ import { log } from "@dxos/log";
96
68
  import { DataType } from "@dxos/schema";
97
69
  import { trim } from "@dxos/util";
98
70
  var __dxlog_file = "/__w/dxos/dxos/packages/core/assistant/src/session/format.ts";
99
- var formatSystemPrompt = ({ system, blueprints = [], objects = [] }) => Effect3.gen(function* () {
100
- const blueprintDefs = yield* pipe(blueprints, Effect3.forEach((blueprint) => Effect3.succeed(blueprint.instructions)), Effect3.flatMap(Effect3.forEach((template) => DatabaseService.loadOption(template.source))), Effect3.map(Array.filter(Option.isSome)), Effect3.map(Array.map((template) => trim`
71
+ var formatSystemPrompt = ({ system, blueprints = [], objects = [] }) => Effect2.gen(function* () {
72
+ const blueprintDefs = yield* Function2.pipe(blueprints, Effect2.forEach((blueprint) => Effect2.succeed(blueprint.instructions)), Effect2.flatMap(Effect2.forEach((template) => DatabaseService.loadOption(template.source))), Effect2.map(Array.filter(Option.isSome)), Effect2.map(Array.map((template) => trim`
101
73
  <blueprint>
102
74
  ${Template.process(template.value.content)}
103
75
  </blueprint>
104
- `)), Effect3.map((blueprints2) => blueprints2.length > 0 ? [
76
+ `)), Effect2.map((blueprints2) => blueprints2.length > 0 ? [
105
77
  "## Blueprints Definitions",
106
78
  ...blueprints2
107
79
  ].join("\n\n") : void 0));
108
- const objectDefs = yield* pipe(objects, Effect3.forEach((object) => Effect3.succeed(trim`
80
+ const objectDefs = yield* Function2.pipe(objects, Effect2.forEach((object) => Effect2.succeed(trim`
109
81
  <object>
110
82
  <dxn>${Obj.getDXN(object)}</dxn>
111
83
  <typename>${Obj.getTypename(object)}</typename>
112
84
  </object>
113
- `)), Effect3.map((objects2) => objects2.length > 0 ? [
85
+ `)), Effect2.map((objects2) => objects2.length > 0 ? [
114
86
  "## Context Objects",
115
87
  ...objects2
116
88
  ].join("\n\n") : void 0));
117
- return yield* pipe(Effect3.succeed([
89
+ return yield* Function2.pipe(Effect2.succeed([
118
90
  system,
119
91
  blueprintDefs,
120
92
  objectDefs
121
- ].filter((def) => def !== void 0)), Effect3.map((parts) => parts.join("\n\n")));
93
+ ].filter((def) => def !== void 0)), Effect2.map((parts) => parts.join("\n\n")));
122
94
  });
123
- var formatUserPrompt = ({ prompt, history = [] }) => Effect3.gen(function* () {
95
+ var formatUserPrompt = ({ prompt, history = [] }) => Effect2.gen(function* () {
124
96
  const blocks = [];
125
- const artifactDiffResolver = yield* Effect3.serviceOption(ArtifactDiffResolver);
97
+ const artifactDiffResolver = yield* Effect2.serviceOption(ArtifactDiffResolver);
126
98
  if (Option.isSome(artifactDiffResolver)) {
127
99
  const versions = gatherObjectVersions(history);
128
- const artifactDiff = yield* Effect3.tryPromise({
100
+ const artifactDiff = yield* Effect2.tryPromise({
129
101
  try: () => artifactDiffResolver.value.resolve([
130
102
  ...versions.entries()
131
103
  ].map(([id, version]) => ({
@@ -141,7 +113,7 @@ var formatUserPrompt = ({ prompt, history = [] }) => Effect3.gen(function* () {
141
113
  versions
142
114
  }, {
143
115
  F: __dxlog_file,
144
- L: 90,
116
+ L: 93,
145
117
  S: this,
146
118
  C: (f, a) => f(...a)
147
119
  });
@@ -232,16 +204,16 @@ var AiSession = class {
232
204
  _define_property(this, "_pending", void 0);
233
205
  _define_property(this, "run", void 0);
234
206
  this._options = _options;
235
- this._semaphore = Effect4.runSync(Effect4.makeSemaphore(1));
207
+ this._semaphore = Effect3.runSync(Effect3.makeSemaphore(1));
236
208
  this._history = [];
237
209
  this._pending = [];
238
- this.run = ({ prompt, system: systemTemplate, history = [], objects = [], blueprints = [], toolkit, observer = GenerationObserver.noop() }) => Effect4.gen(this, function* () {
210
+ this.run = ({ prompt, system: systemTemplate, history = [], objects = [], blueprints = [], toolkit, observer = GenerationObserver.noop() }) => Effect3.gen(this, function* () {
239
211
  this._history = [
240
212
  ...history
241
213
  ];
242
214
  this._pending = [];
243
215
  const pending = this._pending;
244
- const submitMessage = Effect4.fnUntraced(function* (message) {
216
+ const submitMessage = Effect3.fnUntraced(function* (message) {
245
217
  pending.push(message);
246
218
  yield* observer.onMessage(message);
247
219
  yield* TracingService.emitConverationMessage(message);
@@ -268,27 +240,31 @@ var AiSession = class {
268
240
  objects: objects?.length ?? 0
269
241
  }, {
270
242
  F: __dxlog_file2,
271
- L: 104,
243
+ L: 109,
272
244
  S: this,
273
245
  C: (f, a) => f(...a)
274
246
  });
275
- const prompt2 = yield* AiPreprocessor.preprocessAiInput([
247
+ const prompt2 = yield* AiPreprocessor.preprocessPrompt([
276
248
  ...this._history,
277
249
  ...this._pending
278
- ]);
279
- const blocks = yield* AiLanguageModel.streamText({
250
+ ], {
251
+ system
252
+ });
253
+ const blocks = yield* LanguageModel.streamText({
280
254
  prompt: prompt2,
281
- system,
282
255
  toolkit,
283
256
  disableToolCallResolution: true
284
- }).pipe(Stream.catchTag("AiError", Effect4.fnUntraced(function* (err) {
285
- return yield* Effect4.fail(yield* mapAiError(err));
286
- })), AiParser.parseResponse({
287
- onBegin: () => observer.onBegin(),
288
- onBlock: (block) => observer.onBlock(block),
289
- onPart: (part) => observer.onPart(part),
290
- onEnd: (summary) => observer.onEnd(summary)
291
- }), Stream.runCollect, Effect4.map(Chunk.toArray));
257
+ }).pipe(
258
+ // TOOD(dmaretskyi): Error mapping.
259
+ AiParser.parseResponse({
260
+ onBegin: () => observer.onBegin(),
261
+ onBlock: (block) => observer.onBlock(block),
262
+ onPart: (part) => observer.onPart(part),
263
+ onEnd: (summary) => observer.onEnd(summary)
264
+ }),
265
+ Stream.runCollect,
266
+ Effect3.map(Chunk.toArray)
267
+ );
292
268
  const response = yield* submitMessage(Obj2.make(DataType2.Message, {
293
269
  created: (/* @__PURE__ */ new Date()).toISOString(),
294
270
  sender: {
@@ -302,8 +278,8 @@ var AiSession = class {
302
278
  } else if (!toolkit) {
303
279
  throw new Error("No toolkit provided");
304
280
  }
305
- const toolResults = yield* Effect4.forEach(toolCalls, (toolCall) => {
306
- return callTool(toolkit, toolCall).pipe(Effect4.provide(TracingService.layerSubframe((context) => ({
281
+ const toolResults = yield* Effect3.forEach(toolCalls, (toolCall) => {
282
+ return callTool(toolkit, toolCall).pipe(Effect3.provide(TracingService.layerSubframe((context) => ({
307
283
  ...context,
308
284
  parentMessage: response.id,
309
285
  toolCallId: toolCall.toolCallId
@@ -312,7 +288,7 @@ var AiSession = class {
312
288
  yield* submitMessage(Obj2.make(DataType2.Message, {
313
289
  created: (/* @__PURE__ */ new Date()).toISOString(),
314
290
  sender: {
315
- role: "user"
291
+ role: "tool"
316
292
  },
317
293
  blocks: toolResults
318
294
  }));
@@ -321,12 +297,12 @@ var AiSession = class {
321
297
  pending: this._pending.length
322
298
  }, {
323
299
  F: __dxlog_file2,
324
- L: 178,
300
+ L: 180,
325
301
  S: this,
326
302
  C: (f, a) => f(...a)
327
303
  });
328
304
  return this._pending;
329
- }).pipe(this._semaphore.withPermits(1), Effect4.withSpan("AiSession.run"));
305
+ }).pipe(this._semaphore.withPermits(1), Effect3.withSpan("AiSession.run"));
330
306
  }
331
307
  };
332
308
  var createSnippet = (text, len = 32) => text.length <= len * 2 ? text : [
@@ -336,27 +312,27 @@ var createSnippet = (text, len = 32) => text.length <= len * 2 ? text : [
336
312
  ].join("");
337
313
 
338
314
  // src/session/toolkit.ts
339
- import { AiToolkit } from "@effect/ai";
340
- import { Effect as Effect5 } from "effect";
315
+ import * as Toolkit from "@effect/ai/Toolkit";
316
+ import * as Effect4 from "effect/Effect";
341
317
  import { ToolExecutionService, ToolResolverService } from "@dxos/ai";
342
- import { isNotFalsy } from "@dxos/util";
343
- var createToolkit = ({ toolkit: toolkitParam, toolIds = [], blueprints = [] }) => Effect5.gen(function* () {
318
+ import { isTruthy } from "@dxos/util";
319
+ var createToolkit = ({ toolkit: toolkitParam, toolIds = [], blueprints = [] }) => Effect4.gen(function* () {
344
320
  const blueprintToolkit = yield* ToolResolverService.resolveToolkit([
345
321
  ...blueprints.flatMap(({ tools }) => tools),
346
322
  ...toolIds
347
323
  ]);
348
324
  const blueprintToolkitHandler = yield* blueprintToolkit.toContext(ToolExecutionService.handlersFor(blueprintToolkit));
349
- const toolkit = AiToolkit.merge(...[
325
+ const toolkit = Toolkit.merge(...[
350
326
  toolkitParam,
351
327
  blueprintToolkit
352
- ].filter(isNotFalsy));
353
- return yield* toolkit.pipe(Effect5.provide(blueprintToolkitHandler));
328
+ ].filter(isTruthy));
329
+ return yield* toolkit.pipe(Effect4.provide(blueprintToolkitHandler));
354
330
  });
355
331
 
356
332
  // src/session/version-pin.ts
357
- import { Schema as Schema2 } from "effect";
333
+ import * as Schema from "effect/Schema";
334
+ import { ObjectId } from "@dxos/echo/internal";
358
335
  import { getVersion } from "@dxos/echo-db";
359
- import { ObjectId } from "@dxos/echo-schema";
360
336
  function _define_property2(obj, key, value) {
361
337
  if (key in obj) {
362
338
  Object.defineProperty(obj, key, {
@@ -371,8 +347,8 @@ function _define_property2(obj, key, value) {
371
347
  return obj;
372
348
  }
373
349
  var _class;
374
- var ObjectVersionSchema = Schema2.Unknown;
375
- var VersionPinSchema = Schema2.Struct({
350
+ var ObjectVersionSchema = Schema.Unknown;
351
+ var VersionPinSchema = Schema.Struct({
376
352
  // TODO(dmaretskyi): Use Ref when those support encoding.
377
353
  objectId: ObjectId,
378
354
  // TODO(dmaretskyi): Could be opaque.
@@ -403,4 +379,4 @@ export {
403
379
  createToolkit,
404
380
  VersionPin
405
381
  };
406
- //# sourceMappingURL=chunk-W7AIB6DP.mjs.map
382
+ //# sourceMappingURL=chunk-OV5PMQPB.mjs.map
@@ -0,0 +1,7 @@
1
+ {
2
+ "version": 3,
3
+ "sources": ["../../../src/session/artifact-diff.ts", "../../../src/session/observer.ts", "../../../src/errors.ts", "../../../src/session/session.ts", "../../../src/session/format.ts", "../../../src/session/toolkit.ts", "../../../src/session/version-pin.ts"],
4
+ "sourcesContent": ["//\n// Copyright 2025 DXOS.org\n//\n\nimport * as Context from 'effect/Context';\n\nimport { type ObjectId } from '@dxos/echo/internal';\nimport { type ObjectVersion } from '@dxos/echo-db';\n\n/**\n * Resolves artifact ids to their versions.\n * Used to give the model a sense of the changes to the artifacts made by users during the conversation.\n * The artifacts versions are pinned in the history, and whenever the artifact changes in-between assistant's steps,\n * a diff is inserted into the conversation.\n *\n * Can be optionally provided to the session run call.\n */\n// TODO(dmaretskyi): Convert to Context.Reference\nexport class ArtifactDiffResolver extends Context.Tag('@dxos/assistant/ArtifactDiffResolver')<\n ArtifactDiffResolver,\n ArtifactDiffResolver.Service\n>() {}\n\nexport namespace ArtifactDiffResolver {\n export type Service = {\n resolve: (\n artifacts: {\n id: ObjectId;\n lastVersion: ObjectVersion;\n }[],\n ) => Promise<\n Map<\n ObjectId,\n {\n version: ObjectVersion;\n diff?: string;\n }\n >\n >;\n };\n}\n", "//\n// Copyright 2025 DXOS.org\n//\n\nimport * as Effect from 'effect/Effect';\nimport * as Function from 'effect/Function';\n\nimport { type AiParser, type ConsolePrinter } from '@dxos/ai';\nimport { type DataType } from '@dxos/schema';\n\n/**\n * Live observer of the generation process.\n */\nexport interface GenerationObserver extends AiParser.ParseResponseCallbacks {\n /**\n * Complete messages fired during the session, both from the model and from the user.\n * This message will contain all message blocks emitted through the `onBlock` callback.\n */\n onMessage: (message: DataType.Message) => Effect.Effect<void>;\n}\n\nexport const GenerationObserver = Object.freeze({\n make: ({\n onBegin = Function.constant(Effect.void),\n onPart = Function.constant(Effect.void),\n onBlock = Function.constant(Effect.void),\n onEnd = Function.constant(Effect.void),\n onMessage = Function.constant(Effect.void),\n }: Partial<GenerationObserver> = {}): GenerationObserver => ({\n onBegin,\n onPart,\n onBlock,\n onEnd,\n onMessage,\n }),\n\n noop: () => GenerationObserver.make(),\n\n /**\n * Debug printer to be used in unit-tests and browser devtools.\n */\n fromPrinter: (printer: ConsolePrinter) =>\n GenerationObserver.make({\n onBlock: (block) =>\n Effect.sync(() => {\n if (block.pending) {\n return; // Only prints full blocks (better for unit-tests and browser devtools).\n }\n printer.printContentBlock(block);\n }),\n onMessage: (message) =>\n Effect.sync(() => {\n if (message.sender.role === 'assistant') {\n return; // Skip assistant messages since they are printed in the `onBlock` callback.\n }\n printer.printMessage(message);\n }),\n }),\n});\n", "//\n// Copyright 2025 DXOS.org\n//\n\nimport { BaseError } from '@dxos/errors';\n\n/**\n * Generic error from AI model.\n */\nexport class AiModelError extends BaseError.extend('AI_MODEL_ERROR', 'AI model error') {}\n\n/**\n * Generic error for AI agent execution.\n */\nexport class AiAssistantError extends BaseError.extend('AI_ASSISTANT_ERROR', 'AI assistant error') {}\n", "//\n// Copyright 2025 DXOS.org\n//\n\nimport type * as AiError from '@effect/ai/AiError';\nimport * as LanguageModel from '@effect/ai/LanguageModel';\nimport type * as Tool from '@effect/ai/Tool';\nimport type * as Toolkit from '@effect/ai/Toolkit';\nimport * as Chunk from 'effect/Chunk';\nimport * as Effect from 'effect/Effect';\nimport type * as Schema from 'effect/Schema';\nimport * as Stream from 'effect/Stream';\n\nimport {\n AiParser,\n AiPreprocessor,\n type AiToolNotFoundError,\n type PromptPreprocessingError,\n type ToolExecutionService,\n type ToolResolverService,\n callTool,\n getToolCalls,\n} from '@dxos/ai';\nimport { type Blueprint } from '@dxos/blueprints';\nimport { todo } from '@dxos/debug';\nimport { Obj } from '@dxos/echo';\nimport { TracingService } from '@dxos/functions';\nimport { log } from '@dxos/log';\nimport { DataType } from '@dxos/schema';\n\nimport { type AiAssistantError } from '../errors';\n\nimport { formatSystemPrompt, formatUserPrompt } from './format';\nimport { GenerationObserver } from './observer';\n\nexport type AiSessionRunError = AiError.AiError | PromptPreprocessingError | AiToolNotFoundError | AiAssistantError;\n\nexport type AiSessionRunRequirements =\n | LanguageModel.LanguageModel\n | ToolExecutionService\n | ToolResolverService\n | TracingService;\n\nexport type AiSessionRunParams<Tools extends Record<string, Tool.Any>> = {\n prompt: string;\n system?: string;\n history?: DataType.Message[];\n objects?: Obj.Any[];\n blueprints?: Blueprint.Blueprint[];\n toolkit?: Toolkit.WithHandler<Tools>;\n observer?: GenerationObserver;\n};\n\nexport type AiSessionOptions = {};\n\n/**\n * Contains message history, tools, current context.\n * Current context means the state of the app, time of day, and other contextual information.\n * It makes requests to the model, its a state machine.\n * It keeps track of the current goal.\n * It manages the context window.\n * Tracks the success criteria of reaching the goal, exposing metrics (stretch).\n * Could be run locally in the app or remotely.\n * Could be personal or shared.\n */\nexport class AiSession {\n /** Prevents concurrent execution of session. */\n private readonly _semaphore = Effect.runSync(Effect.makeSemaphore(1));\n\n /** Message history from queue. */\n // TODO(burdon): Evolve into supporting a git-like graph of messages.\n private _history: DataType.Message[] = [];\n\n /** Pending messages for this session (incl. the current prompt). */\n private _pending: DataType.Message[] = [];\n\n constructor(private readonly _options: AiSessionOptions = {}) {}\n\n run = <Tools extends Record<string, Tool.Any>>({\n prompt,\n system: systemTemplate,\n history = [],\n objects = [],\n blueprints = [],\n toolkit,\n observer = GenerationObserver.noop(),\n }: AiSessionRunParams<Tools>): Effect.Effect<DataType.Message[], AiSessionRunError, AiSessionRunRequirements> =>\n Effect.gen(this, function* () {\n this._history = [...history];\n this._pending = [];\n const pending = this._pending;\n\n const submitMessage = Effect.fnUntraced(function* (message: DataType.Message) {\n pending.push(message);\n yield* observer.onMessage(message);\n yield* TracingService.emitConverationMessage(message);\n return message;\n });\n\n // Submit the prompt.\n // TODO(burdon): Remove if cancelled?\n const promptMessage = yield* submitMessage(yield* formatUserPrompt({ prompt, history }));\n\n // Generate system and prompt messages.\n const system = yield* formatSystemPrompt({ system: systemTemplate, blueprints, objects });\n\n // Tool call loop.\n do {\n log('request', {\n prompt: promptMessage,\n system: { snippet: createSnippet(system), length: system.length },\n pending: this._pending.length,\n history: this._history.length,\n objects: objects?.length ?? 0,\n });\n\n const prompt = yield* AiPreprocessor.preprocessPrompt([...this._history, ...this._pending], { system });\n\n // Execute the stream request.\n // logDump('prompt', Prompt.Prompt.pipe(Schema.encodeSync)(prompt));\n const blocks = yield* LanguageModel.streamText({\n prompt,\n toolkit,\n disableToolCallResolution: true,\n }).pipe(\n // TOOD(dmaretskyi): Error mapping.\n AiParser.parseResponse({\n onBegin: () => observer.onBegin(),\n onBlock: (block) => observer.onBlock(block),\n onPart: (part) => observer.onPart(part),\n onEnd: (summary) => observer.onEnd(summary),\n }),\n Stream.runCollect,\n Effect.map(Chunk.toArray),\n );\n\n // Create the response message.\n const response = yield* submitMessage(\n Obj.make(DataType.Message, {\n created: new Date().toISOString(),\n sender: { role: 'assistant' },\n blocks,\n }),\n );\n\n // Parse the response for tool calls.\n const toolCalls = getToolCalls(response);\n if (toolCalls.length === 0) {\n break;\n } else if (!toolkit) {\n throw new Error('No toolkit provided'); // TODO(burdon): Throw user error?\n }\n\n // Execute the tool calls.\n // TODO(burdon): Retry errors? Write result when each completes individually?\n const toolResults = yield* Effect.forEach(toolCalls, (toolCall) => {\n return callTool(toolkit, toolCall).pipe(\n Effect.provide(\n TracingService.layerSubframe((context) => ({\n ...context,\n parentMessage: response.id,\n toolCallId: toolCall.toolCallId,\n })),\n ),\n );\n });\n\n // Add to queue and continue loop.\n // TODO(wittjosiah): Sometimes tool error results are added to the queue before the tool agent statuses.\n // This results in a broken execution graph.\n yield* submitMessage(\n Obj.make(DataType.Message, {\n created: new Date().toISOString(),\n sender: { role: 'tool' },\n blocks: toolResults,\n }),\n );\n } while (true);\n\n log('done', { pending: this._pending.length });\n return this._pending;\n }).pipe(this._semaphore.withPermits(1), Effect.withSpan('AiSession.run'));\n\n /**\n * @deprecated\n */\n // TODO(burdon): Implement or remove.\n async runStructured<S extends Schema.Schema.AnyNoContext>(\n _schema: S,\n _options: AiSessionRunParams<any>,\n ): Promise<Schema.Schema.Type<S>> {\n return todo();\n // const parser = structuredOutputParser(schema);\n // const result = await this.run({\n // ...options,\n // executableTools: [...(options.executableTools ?? []), parser.tool],\n // });\n // return parser.getResult(result);\n }\n}\n\nconst createSnippet = (text: string, len = 32) =>\n text.length <= len * 2 ? text : [text.slice(0, len), '...', text.slice(-len)].join('');\n\n// TODO(dmaretskyi): Extract as a general util.\nconst logDump = (message: string, data: unknown) => {\n // eslint-disable-next-line @typescript-eslint/no-require-imports\n const { writeFileSync } = require('node:fs');\n const path = `/tmp/log-data-${Date.now()}.json`;\n writeFileSync(path, JSON.stringify(data, null, 2));\n log.info(message, { path });\n};\n", "//\n// Copyright 2025 DXOS.org\n//\n\nimport * as Array from 'effect/Array';\nimport * as Effect from 'effect/Effect';\nimport * as Function from 'effect/Function';\nimport * as Option from 'effect/Option';\n\nimport { Template } from '@dxos/blueprints';\nimport { Obj } from '@dxos/echo';\nimport { type ObjectId } from '@dxos/echo/internal';\nimport { ObjectVersion } from '@dxos/echo-db';\nimport { DatabaseService } from '@dxos/functions';\nimport { log } from '@dxos/log';\nimport { type ContentBlock, DataType } from '@dxos/schema';\nimport { trim } from '@dxos/util';\n\nimport { AiAssistantError } from '../errors';\n\nimport { ArtifactDiffResolver } from './artifact-diff';\nimport { type AiSessionRunParams } from './session';\n\n/**\n * Formats the system prompt.\n */\n// TODO(burdon): Move to AiPreprocessor.\nexport const formatSystemPrompt = ({\n system,\n blueprints = [],\n objects = [],\n}: Pick<AiSessionRunParams<any>, 'system' | 'blueprints' | 'objects'>) =>\n Effect.gen(function* () {\n const blueprintDefs = yield* Function.pipe(\n blueprints,\n Effect.forEach((blueprint) => Effect.succeed(blueprint.instructions)),\n Effect.flatMap(Effect.forEach((template) => DatabaseService.loadOption(template.source))),\n Effect.map(Array.filter(Option.isSome)),\n Effect.map(\n Array.map(\n (template) => trim`\n <blueprint>\n ${Template.process(template.value.content)}\n </blueprint>\n `,\n ),\n ),\n Effect.map((blueprints) =>\n blueprints.length > 0 ? ['## Blueprints Definitions', ...blueprints].join('\\n\\n') : undefined,\n ),\n );\n\n const objectDefs = yield* Function.pipe(\n objects,\n Effect.forEach((object) =>\n Effect.succeed(trim`\n <object>\n <dxn>${Obj.getDXN(object)}</dxn>\n <typename>${Obj.getTypename(object)}</typename>\n </object>\n `),\n ),\n Effect.map((objects) => (objects.length > 0 ? ['## Context Objects', ...objects].join('\\n\\n') : undefined)),\n );\n\n return yield* Function.pipe(\n Effect.succeed([system, blueprintDefs, objectDefs].filter((def): def is string => def !== undefined)),\n Effect.map((parts) => parts.join('\\n\\n')),\n );\n });\n\n/**\n * Formats the user prompt.\n */\n// TODO(burdon): Move to AiPreprocessor.\n// TODO(burdon): Convert util below to `Effect.fn` (to preserve stack info)\nexport const formatUserPrompt = ({ prompt, history = [] }: Pick<AiSessionRunParams<any>, 'prompt' | 'history'>) =>\n Effect.gen(function* () {\n const blocks: ContentBlock.Any[] = [];\n\n // TODO(dmaretskyi): Evaluate other approaches as `serviceOption` isn't represented in the type system.\n const artifactDiffResolver = yield* Effect.serviceOption(ArtifactDiffResolver);\n if (Option.isSome(artifactDiffResolver)) {\n const versions = gatherObjectVersions(history);\n const artifactDiff = yield* Effect.tryPromise({\n try: () =>\n artifactDiffResolver.value.resolve(\n [...versions.entries()].map(([id, version]) => ({ id, lastVersion: version })),\n ),\n catch: AiAssistantError.wrap({ message: 'Artifact diff resolution error' }),\n });\n\n log('version', { artifactDiff, versions });\n for (const [id, { version }] of [...artifactDiff.entries()]) {\n if (ObjectVersion.equals(version, versions.get(id)!)) {\n artifactDiff.delete(id);\n continue;\n }\n\n blocks.push({ _tag: 'anchor', objectId: id, version });\n }\n\n if (artifactDiff.size > 0) {\n blocks.push(createArtifactUpdateBlock(artifactDiff));\n }\n }\n\n return Obj.make(DataType.Message, {\n created: new Date().toISOString(),\n sender: { role: 'user' },\n blocks: [...blocks, { _tag: 'text', text: prompt }],\n });\n });\n\nconst gatherObjectVersions = (messages: DataType.Message[]): Map<ObjectId, ObjectVersion> => {\n const artifactIds = new Map<ObjectId, ObjectVersion>();\n for (const message of messages) {\n for (const block of message.blocks) {\n if (block._tag === 'anchor') {\n artifactIds.set(block.objectId, block.version as ObjectVersion);\n }\n }\n }\n\n return artifactIds;\n};\n\nconst createArtifactUpdateBlock = (\n artifactDiff: Map<ObjectId, { version: ObjectVersion; diff?: string }>,\n): ContentBlock.Any => {\n return {\n _tag: 'text',\n // TODO(dmaretskyi): Does this need to be a special content-block?\n disposition: 'artifact-update',\n text: trim`\n The following artifacts have been updated since the last message:\n ${[...artifactDiff.entries()]\n .map(([id, { diff }]) => `<changed-artifact id=\"${id}\">${diff ? `\\n${diff}` : ''}</changed-artifact>`)\n .join('\\n')}\n `,\n };\n};\n", "//\n// Copyright 2025 DXOS.org\n//\n\nimport type * as Tool from '@effect/ai/Tool';\nimport * as Toolkit from '@effect/ai/Toolkit';\nimport type * as Context from 'effect/Context';\nimport * as Effect from 'effect/Effect';\n\nimport { type AiToolNotFoundError, ToolExecutionService, type ToolId, ToolResolverService } from '@dxos/ai';\nimport { type Blueprint } from '@dxos/blueprints';\nimport { isTruthy } from '@dxos/util';\n\nexport type ToolkitParams<Tools extends Record<string, Tool.Any>> = {\n toolkit?: Toolkit.Toolkit<Tools>;\n toolIds?: ToolId[];\n blueprints?: Blueprint.Blueprint[];\n};\n\n/**\n * Build a combined toolkit from the blueprint tools and the provided toolkit.\n */\nexport const createToolkit = <Tools extends Record<string, Tool.Any> = {}>({\n toolkit: toolkitParam,\n toolIds = [],\n blueprints = [],\n}: ToolkitParams<Tools>): Effect.Effect<\n Toolkit.WithHandler<any>,\n AiToolNotFoundError,\n ToolResolverService | ToolExecutionService | Tool.HandlersFor<Tools>\n> =>\n Effect.gen(function* () {\n const blueprintToolkit = yield* ToolResolverService.resolveToolkit([\n ...blueprints.flatMap(({ tools }) => tools),\n ...toolIds,\n ]);\n\n const blueprintToolkitHandler: Context.Context<Tool.HandlersFor<any>> = yield* blueprintToolkit.toContext(\n ToolExecutionService.handlersFor(blueprintToolkit),\n );\n\n const toolkit = Toolkit.merge(...[toolkitParam, blueprintToolkit].filter(isTruthy));\n return yield* toolkit.pipe(Effect.provide(blueprintToolkitHandler)) as any as Effect.Effect<\n Toolkit.WithHandler<any>,\n never,\n Tool.HandlersFor<Tools>\n >;\n });\n", "//\n// Copyright 2025 DXOS.org\n//\n\nimport * as Schema from 'effect/Schema';\n\nimport { type AnyEchoObject, ObjectId } from '@dxos/echo/internal';\nimport { type ObjectVersion } from '@dxos/echo-db';\nimport { getVersion } from '@dxos/echo-db';\nimport { type ContentBlock } from '@dxos/schema';\n\n// TODO(dmaretskyi): Extract.\nconst ObjectVersionSchema = Schema.Unknown as Schema.Schema<ObjectVersion>;\n\nconst VersionPinSchema = Schema.Struct({\n // TODO(dmaretskyi): Use Ref when those support encoding.\n objectId: ObjectId,\n // TODO(dmaretskyi): Could be opaque.\n version: ObjectVersionSchema,\n});\n\n/**\n * Used to pin a specific version of an object during the course of a conversation.\n * This allows us to detect when the object changes during the course of the conversation.\n */\nexport interface VersionPin extends Schema.Schema.Type<typeof VersionPinSchema> {}\n\nexport const VersionPin: typeof VersionPinSchema & {\n DISPOSITION: 'version-pin';\n fromObject: (object: AnyEchoObject) => VersionPin;\n createBlock: (pin: VersionPin) => ContentBlock.Any;\n} = class extends VersionPinSchema {\n static readonly DISPOSITION = 'version-pin';\n static fromObject(object: AnyEchoObject): VersionPin {\n return VersionPin.make({\n objectId: object.id,\n version: getVersion(object),\n });\n }\n\n static createBlock(pin: VersionPin): ContentBlock.Any {\n return {\n _tag: 'json',\n disposition: VersionPin.DISPOSITION,\n data: JSON.stringify(pin),\n };\n }\n};\n"],
5
+ "mappings": ";;;AAIA,YAAYA,aAAa;AAclB,IAAMC,uBAAN,cAA2CC,YAAI,sCAAA,EAAA,EAAA;AAGjD;;;ACjBL,YAAYC,YAAY;AACxB,YAAYC,cAAc;AAgBnB,IAAMC,qBAAqBC,OAAOC,OAAO;EAC9CC,MAAM,CAAC,EACLC,UAAmBC,kBAAgBC,WAAI,GACvCC,SAAkBF,kBAAgBC,WAAI,GACtCE,UAAmBH,kBAAgBC,WAAI,GACvCG,QAAiBJ,kBAAgBC,WAAI,GACrCI,YAAqBL,kBAAgBC,WAAI,EAAC,IACX,CAAC,OAA2B;IAC3DF;IACAG;IACAC;IACAC;IACAC;EACF;EAEAC,MAAM,MAAMX,mBAAmBG,KAAI;;;;EAKnCS,aAAa,CAACC,YACZb,mBAAmBG,KAAK;IACtBK,SAAS,CAACM,UACDC,YAAK,MAAA;AACV,UAAID,MAAME,SAAS;AACjB;MACF;AACAH,cAAQI,kBAAkBH,KAAAA;IAC5B,CAAA;IACFJ,WAAW,CAACQ,YACHH,YAAK,MAAA;AACV,UAAIG,QAAQC,OAAOC,SAAS,aAAa;AACvC;MACF;AACAP,cAAQQ,aAAaH,OAAAA;IACvB,CAAA;EACJ,CAAA;AACJ,CAAA;;;ACtDA,SAASI,iBAAiB;AAKnB,IAAMC,eAAN,cAA2BC,UAAUC,OAAO,kBAAkB,gBAAA,EAAA;AAAmB;AAKjF,IAAMC,mBAAN,cAA+BF,UAAUC,OAAO,sBAAsB,oBAAA,EAAA;AAAuB;;;ACTpG,YAAYE,mBAAmB;AAG/B,YAAYC,WAAW;AACvB,YAAYC,aAAY;AAExB,YAAYC,YAAY;AAExB,SACEC,UACAC,gBAKAC,UACAC,oBACK;AAEP,SAASC,YAAY;AACrB,SAASC,OAAAA,YAAW;AACpB,SAASC,sBAAsB;AAC/B,SAASC,OAAAA,YAAW;AACpB,SAASC,YAAAA,iBAAgB;;;ACxBzB,YAAYC,WAAW;AACvB,YAAYC,aAAY;AACxB,YAAYC,eAAc;AAC1B,YAAYC,YAAY;AAExB,SAASC,gBAAgB;AACzB,SAASC,WAAW;AAEpB,SAASC,qBAAqB;AAC9B,SAASC,uBAAuB;AAChC,SAASC,WAAW;AACpB,SAA4BC,gBAAgB;AAC5C,SAASC,YAAY;;AAWd,IAAMC,qBAAqB,CAAC,EACjCC,QACAC,aAAa,CAAA,GACbC,UAAU,CAAA,EAAE,MAELC,YAAI,aAAA;AACT,QAAMC,gBAAgB,OAAgBC,eACpCJ,YACOK,gBAAQ,CAACC,cAAqBC,gBAAQD,UAAUE,YAAY,CAAA,GAC5DC,gBAAeJ,gBAAQ,CAACK,aAAaC,gBAAgBC,WAAWF,SAASG,MAAM,CAAA,CAAA,GAC/EC,YAAUC,aAAcC,aAAM,CAAA,GAC9BF,YACCA,UACJ,CAACJ,aAAaO;;gBAERC,SAASC,QAAQT,SAASU,MAAMC,OAAO,CAAA;;WAE5C,CAAA,GAGEP,YAAI,CAACd,gBACVA,YAAWsB,SAAS,IAAI;IAAC;OAAgCtB;IAAYuB,KAAK,MAAA,IAAUC,MAAAA,CAAAA;AAIxF,QAAMC,aAAa,OAAgBrB,eACjCH,SACOI,gBAAQ,CAACqB,WACPnB,gBAAQU;;mBAEJU,IAAIC,OAAOF,MAAAA,CAAAA;wBACNC,IAAIE,YAAYH,MAAAA,CAAAA;;SAE/B,CAAA,GAEIZ,YAAI,CAACb,aAAaA,SAAQqB,SAAS,IAAI;IAAC;OAAyBrB;IAASsB,KAAK,MAAA,IAAUC,MAAAA,CAAAA;AAGlG,SAAO,OAAgBpB,eACdG,gBAAQ;IAACR;IAAQI;IAAesB;IAAYV,OAAO,CAACe,QAAuBA,QAAQN,MAAAA,CAAAA,GACnFV,YAAI,CAACiB,UAAUA,MAAMR,KAAK,MAAA,CAAA,CAAA;AAErC,CAAA;AAOK,IAAMS,mBAAmB,CAAC,EAAEC,QAAQC,UAAU,CAAA,EAAE,MAC9ChC,YAAI,aAAA;AACT,QAAMiC,SAA6B,CAAA;AAGnC,QAAMC,uBAAuB,OAAcC,sBAAcC,oBAAAA;AACzD,MAAWtB,cAAOoB,oBAAAA,GAAuB;AACvC,UAAMG,WAAWC,qBAAqBN,OAAAA;AACtC,UAAMO,eAAe,OAAcC,mBAAW;MAC5CC,KAAK,MACHP,qBAAqBhB,MAAMwB,QACzB;WAAIL,SAASM,QAAO;QAAI/B,IAAI,CAAC,CAACgC,IAAIC,OAAAA,OAAc;QAAED;QAAIE,aAAaD;MAAQ,EAAA,CAAA;MAE/EE,OAAOC,iBAAiBC,KAAK;QAAEC,SAAS;MAAiC,CAAA;IAC3E,CAAA;AAEAC,QAAI,WAAW;MAAEZ;MAAcF;IAAS,GAAA;;;;;;AACxC,eAAW,CAACO,IAAI,EAAEC,QAAO,CAAE,KAAK;SAAIN,aAAaI,QAAO;OAAK;AAC3D,UAAIS,cAAcC,OAAOR,SAASR,SAASiB,IAAIV,EAAAA,CAAAA,GAAO;AACpDL,qBAAagB,OAAOX,EAAAA;AACpB;MACF;AAEAX,aAAOuB,KAAK;QAAEC,MAAM;QAAUC,UAAUd;QAAIC;MAAQ,CAAA;IACtD;AAEA,QAAIN,aAAaoB,OAAO,GAAG;AACzB1B,aAAOuB,KAAKI,0BAA0BrB,YAAAA,CAAAA;IACxC;EACF;AAEA,SAAOd,IAAIoC,KAAKC,SAASC,SAAS;IAChCC,UAAS,oBAAIC,KAAAA,GAAOC,YAAW;IAC/BC,QAAQ;MAAEC,MAAM;IAAO;IACvBnC,QAAQ;SAAIA;MAAQ;QAAEwB,MAAM;QAAQY,MAAMtC;MAAO;;EACnD,CAAA;AACF,CAAA;AAEF,IAAMO,uBAAuB,CAACgC,aAAAA;AAC5B,QAAMC,cAAc,oBAAIC,IAAAA;AACxB,aAAWtB,WAAWoB,UAAU;AAC9B,eAAWG,SAASvB,QAAQjB,QAAQ;AAClC,UAAIwC,MAAMhB,SAAS,UAAU;AAC3Bc,oBAAYG,IAAID,MAAMf,UAAUe,MAAM5B,OAAO;MAC/C;IACF;EACF;AAEA,SAAO0B;AACT;AAEA,IAAMX,4BAA4B,CAChCrB,iBAAAA;AAEA,SAAO;IACLkB,MAAM;;IAENkB,aAAa;IACbN,MAAMtD;;QAEF;SAAIwB,aAAaI,QAAO;MACvB/B,IAAI,CAAC,CAACgC,IAAI,EAAEgC,KAAI,CAAE,MAAM,yBAAyBhC,EAAAA,KAAOgC,OAAO;EAAKA,IAAAA,KAAS,EAAA,qBAAuB,EACpGvD,KAAK,IAAA,CAAA;;EAEZ;AACF;;;;;;;;;;;;;;;;;AD5EO,IAAMwD,YAAN,MAAMA;;;;;EA0HX,MAAMC,cACJC,SACAC,UACgC;AAChC,WAAOC,KAAAA;EAOT;EA1HA,YAA6BD,WAA6B,CAAC,GAAG;;AAT9D,qBAAA,MAAiBE,cAAjB,MAAA;AAIA,qBAAA,MAAQC,YAAR,MAAA;AAGA,qBAAA,MAAQC,YAAR,MAAA;AAIAC,qBAAAA,MAAAA,OAAAA,MAAAA;SAF6BL,WAAAA;SATZE,aAAoBI,gBAAeC,sBAAc,CAAA,CAAA;SAI1DJ,WAA+B,CAAA;SAG/BC,WAA+B,CAAA;SAIvCC,MAAM,CAAyC,EAC7CG,QACAC,QAAQC,gBACRC,UAAU,CAAA,GACVC,UAAU,CAAA,GACVC,aAAa,CAAA,GACbC,SACAC,WAAWC,mBAAmBC,KAAI,EAAE,MAE7BC,YAAI,MAAM,aAAA;AACf,WAAKf,WAAW;WAAIQ;;AACpB,WAAKP,WAAW,CAAA;AAChB,YAAMe,UAAU,KAAKf;AAErB,YAAMgB,gBAAuBC,mBAAW,WAAWC,SAAyB;AAC1EH,gBAAQI,KAAKD,OAAAA;AACb,eAAOP,SAASS,UAAUF,OAAAA;AAC1B,eAAOG,eAAeC,uBAAuBJ,OAAAA;AAC7C,eAAOA;MACT,CAAA;AAIA,YAAMK,gBAAgB,OAAOP,cAAc,OAAOQ,iBAAiB;QAAEpB;QAAQG;MAAQ,CAAA,CAAC;AAGtF,YAAMF,SAAS,OAAOoB,mBAAmB;QAAEpB,QAAQC;QAAgBG;QAAYD;MAAQ,CAAA;AAGvF,SAAG;AACDkB,QAAAA,KAAI,WAAW;UACbtB,QAAQmB;UACRlB,QAAQ;YAAEsB,SAASC,cAAcvB,MAAAA;YAASwB,QAAQxB,OAAOwB;UAAO;UAChEd,SAAS,KAAKf,SAAS6B;UACvBtB,SAAS,KAAKR,SAAS8B;UACvBrB,SAASA,SAASqB,UAAU;QAC9B,GAAA;;;;;;AAEA,cAAMzB,UAAS,OAAO0B,eAAeC,iBAAiB;aAAI,KAAKhC;aAAa,KAAKC;WAAW;UAAEK;QAAO,CAAA;AAIrG,cAAM2B,SAAS,OAAqBC,yBAAW;UAC7C7B,QAAAA;UACAM;UACAwB,2BAA2B;QAC7B,CAAA,EAAGC;;UAEDC,SAASC,cAAc;YACrBC,SAAS,MAAM3B,SAAS2B,QAAO;YAC/BC,SAAS,CAACC,UAAU7B,SAAS4B,QAAQC,KAAAA;YACrCC,QAAQ,CAACC,SAAS/B,SAAS8B,OAAOC,IAAAA;YAClCC,OAAO,CAACC,YAAYjC,SAASgC,MAAMC,OAAAA;UACrC,CAAA;UACOC;UACAC,YAAUC,aAAO;QAAA;AAI1B,cAAMC,WAAW,OAAOhC,cACtBiC,KAAIC,KAAKC,UAASC,SAAS;UACzBC,UAAS,oBAAIC,KAAAA,GAAOC,YAAW;UAC/BC,QAAQ;YAAEC,MAAM;UAAY;UAC5BzB;QACF,CAAA,CAAA;AAIF,cAAM0B,YAAYC,aAAaX,QAAAA;AAC/B,YAAIU,UAAU7B,WAAW,GAAG;AAC1B;QACF,WAAW,CAACnB,SAAS;AACnB,gBAAM,IAAIkD,MAAM,qBAAA;QAClB;AAIA,cAAMC,cAAc,OAAcC,gBAAQJ,WAAW,CAACK,aAAAA;AACpD,iBAAOC,SAAStD,SAASqD,QAAAA,EAAU5B,KAC1B8B,gBACL5C,eAAe6C,cAAc,CAACC,aAAa;YACzC,GAAGA;YACHC,eAAepB,SAASqB;YACxBC,YAAYP,SAASO;UACvB,EAAA,CAAA,CAAA;QAGN,CAAA;AAKA,eAAOtD,cACLiC,KAAIC,KAAKC,UAASC,SAAS;UACzBC,UAAS,oBAAIC,KAAAA,GAAOC,YAAW;UAC/BC,QAAQ;YAAEC,MAAM;UAAO;UACvBzB,QAAQ6B;QACV,CAAA,CAAA;MAEJ,SAAS;AAETnC,MAAAA,KAAI,QAAQ;QAAEX,SAAS,KAAKf,SAAS6B;MAAO,GAAA;;;;;;AAC5C,aAAO,KAAK7B;IACd,CAAA,EAAGmC,KAAK,KAAKrC,WAAWyE,YAAY,CAAA,GAAWC,iBAAS,eAAA,CAAA;EAzGK;AA2HjE;AAEA,IAAM5C,gBAAgB,CAAC6C,MAAcC,MAAM,OACzCD,KAAK5C,UAAU6C,MAAM,IAAID,OAAO;EAACA,KAAKE,MAAM,GAAGD,GAAAA;EAAM;EAAOD,KAAKE,MAAM,CAACD,GAAAA;EAAME,KAAK,EAAA;;;AErMrF,YAAYC,aAAa;AAEzB,YAAYC,aAAY;AAExB,SAAmCC,sBAAmCC,2BAA2B;AAEjG,SAASC,gBAAgB;AAWlB,IAAMC,gBAAgB,CAA8C,EACzEC,SAASC,cACTC,UAAU,CAAA,GACVC,aAAa,CAAA,EAAE,MAMRC,YAAI,aAAA;AACT,QAAMC,mBAAmB,OAAOC,oBAAoBC,eAAe;OAC9DJ,WAAWK,QAAQ,CAAC,EAAEC,MAAK,MAAOA,KAAAA;OAClCP;GACJ;AAED,QAAMQ,0BAAkE,OAAOL,iBAAiBM,UAC9FC,qBAAqBC,YAAYR,gBAAAA,CAAAA;AAGnC,QAAML,UAAkBc,cAAK,GAAI;IAACb;IAAcI;IAAkBU,OAAOC,QAAAA,CAAAA;AACzE,SAAO,OAAOhB,QAAQiB,KAAYC,gBAAQR,uBAAAA,CAAAA;AAK5C,CAAA;;;AC3CF,YAAYS,YAAY;AAExB,SAA6BC,gBAAgB;AAE7C,SAASC,kBAAkB;;;;;;;;;;;;;;;AAI3B,IAAMC,sBAA6BC;AAEnC,IAAMC,mBAA0BC,cAAO;;EAErCC,UAAUC;;EAEVC,SAASN;AACX,CAAA;AAQO,IAAMO,cAAAA,SAIT,cAAcL,iBAAAA;EAEhB,OAAOM,WAAWC,QAAmC;AACnD,WAAOF,WAAWG,KAAK;MACrBN,UAAUK,OAAOE;MACjBL,SAASM,WAAWH,MAAAA;IACtB,CAAA;EACF;EAEA,OAAOI,YAAYC,KAAmC;AACpD,WAAO;MACLC,MAAM;MACNC,aAAaT,WAAWU;MACxBC,MAAMC,KAAKC,UAAUN,GAAAA;IACvB;EACF;AACF,GAfEO,kBAAA,QAAgBJ,eAAc,aAAA,GAAA;",
6
+ "names": ["Context", "ArtifactDiffResolver", "Tag", "Effect", "Function", "GenerationObserver", "Object", "freeze", "make", "onBegin", "constant", "void", "onPart", "onBlock", "onEnd", "onMessage", "noop", "fromPrinter", "printer", "block", "sync", "pending", "printContentBlock", "message", "sender", "role", "printMessage", "BaseError", "AiModelError", "BaseError", "extend", "AiAssistantError", "LanguageModel", "Chunk", "Effect", "Stream", "AiParser", "AiPreprocessor", "callTool", "getToolCalls", "todo", "Obj", "TracingService", "log", "DataType", "Array", "Effect", "Function", "Option", "Template", "Obj", "ObjectVersion", "DatabaseService", "log", "DataType", "trim", "formatSystemPrompt", "system", "blueprints", "objects", "gen", "blueprintDefs", "pipe", "forEach", "blueprint", "succeed", "instructions", "flatMap", "template", "DatabaseService", "loadOption", "source", "map", "filter", "isSome", "trim", "Template", "process", "value", "content", "length", "join", "undefined", "objectDefs", "object", "Obj", "getDXN", "getTypename", "def", "parts", "formatUserPrompt", "prompt", "history", "blocks", "artifactDiffResolver", "serviceOption", "ArtifactDiffResolver", "versions", "gatherObjectVersions", "artifactDiff", "tryPromise", "try", "resolve", "entries", "id", "version", "lastVersion", "catch", "AiAssistantError", "wrap", "message", "log", "ObjectVersion", "equals", "get", "delete", "push", "_tag", "objectId", "size", "createArtifactUpdateBlock", "make", "DataType", "Message", "created", "Date", "toISOString", "sender", "role", "text", "messages", "artifactIds", "Map", "block", "set", "disposition", "diff", "AiSession", "runStructured", "_schema", "_options", "todo", "_semaphore", "_history", "_pending", "run", "runSync", "makeSemaphore", "prompt", "system", "systemTemplate", "history", "objects", "blueprints", "toolkit", "observer", "GenerationObserver", "noop", "gen", "pending", "submitMessage", "fnUntraced", "message", "push", "onMessage", "TracingService", "emitConverationMessage", "promptMessage", "formatUserPrompt", "formatSystemPrompt", "log", "snippet", "createSnippet", "length", "AiPreprocessor", "preprocessPrompt", "blocks", "streamText", "disableToolCallResolution", "pipe", "AiParser", "parseResponse", "onBegin", "onBlock", "block", "onPart", "part", "onEnd", "summary", "runCollect", "map", "toArray", "response", "Obj", "make", "DataType", "Message", "created", "Date", "toISOString", "sender", "role", "toolCalls", "getToolCalls", "Error", "toolResults", "forEach", "toolCall", "callTool", "provide", "layerSubframe", "context", "parentMessage", "id", "toolCallId", "withPermits", "withSpan", "text", "len", "slice", "join", "Toolkit", "Effect", "ToolExecutionService", "ToolResolverService", "isTruthy", "createToolkit", "toolkit", "toolkitParam", "toolIds", "blueprints", "gen", "blueprintToolkit", "ToolResolverService", "resolveToolkit", "flatMap", "tools", "blueprintToolkitHandler", "toContext", "ToolExecutionService", "handlersFor", "merge", "filter", "isTruthy", "pipe", "provide", "Schema", "ObjectId", "getVersion", "ObjectVersionSchema", "Unknown", "VersionPinSchema", "Struct", "objectId", "ObjectId", "version", "VersionPin", "fromObject", "object", "make", "id", "getVersion", "createBlock", "pin", "_tag", "disposition", "DISPOSITION", "data", "JSON", "stringify", "_define_property"]
7
+ }
@@ -1,7 +1,7 @@
1
1
  import { createRequire } from 'node:module';const require = createRequire(import.meta.url);
2
2
  import {
3
3
  AiSession
4
- } from "../chunk-W7AIB6DP.mjs";
4
+ } from "../chunk-OV5PMQPB.mjs";
5
5
  import {
6
6
  combineNerTokens,
7
7
  createTokenGroups,
@@ -10,9 +10,9 @@ import {
10
10
  } from "../chunk-BCFXTNSK.mjs";
11
11
 
12
12
  // src/extraction/extraction.ts
13
- import { Schema } from "effect";
13
+ import * as Schema from "effect/Schema";
14
14
  import { asyncTimeout } from "@dxos/async";
15
- import { Expando } from "@dxos/echo-schema";
15
+ import { Expando } from "@dxos/echo/internal";
16
16
  import { log } from "@dxos/log";
17
17
  import { DataType } from "@dxos/schema";
18
18
  var __dxlog_file = "/__w/dxos/dxos/packages/core/assistant/src/extraction/extraction.ts";
@@ -58,14 +58,14 @@ var processTranscriptMessage = async (params) => {
58
58
  // src/extraction/extraction-llm-function/extraction-llm-function.ts
59
59
  import { AiService } from "@dxos/ai";
60
60
  import { Obj } from "@dxos/echo";
61
- import { create } from "@dxos/echo-schema";
61
+ import { create } from "@dxos/echo/internal";
62
62
  import { defineFunction } from "@dxos/functions";
63
63
  import { DataType as DataType2 } from "@dxos/schema";
64
64
 
65
65
  // src/extraction/quotes.ts
66
- import { Schema as Schema2 } from "effect";
66
+ import * as Schema2 from "effect/Schema";
67
67
  import { Filter, Query } from "@dxos/echo";
68
- import { ObjectId } from "@dxos/echo-schema";
68
+ import { ObjectId } from "@dxos/echo/internal";
69
69
  import { DXN } from "@dxos/keys";
70
70
  import { log as log2 } from "@dxos/log";
71
71
  var __dxlog_file2 = "/__w/dxos/dxos/packages/core/assistant/src/extraction/quotes.ts";
@@ -117,11 +117,13 @@ var insertReferences = (text, quotes) => {
117
117
  return text;
118
118
  };
119
119
 
120
- // raw-loader:./instructions.tpl?raw
120
+ // raw-loader:/__w/dxos/dxos/packages/core/assistant/src/extraction/extraction-llm-function/instructions.tpl?raw
121
121
  var instructions_default = 'You are an AI assistant specialized in processing and enhancing transcripts with contextual information.\nYour task is to analyze a given transcript and extract references to the context objects.\n\nThe context and transcript are provided in the user message in `context` and `transcript` XML tags.\n\nHere are your instructions:\n\n1. Read the transcript and context information provided above.\n\n2. Process each segment of the transcript separately.\n\n3. For each segment:\n a. Analyze the content and identify relevant information from the context.\n b. Extract a number of references to the context objects that are mentioned in the segment.\n c. Output the references in the format `{ "quote": "earnings report", "id": "01JT0JP9AX0XKGZX4MV4B69VT6" }`.\n d. If there are no references, just output an empty array.\n\n4. Maintain the exact order and structure of the original transcript segments.\n\nOutput Format:\nYou have been provided with a tool that specifies output format\n\nRemember:\n- Only quote the text that is in the transcript.\n- Analyze each segment carefully before processing it.\n- Only reference objects that appear in the context and exactly by their id, If the object is not in the context, do not insert a reference. \n\nNow, process the transcript and context provided below according to these instructions.';
122
122
 
123
123
  // src/extraction/extraction-llm-function/extraction-llm-function.ts
124
124
  var extractionAnthropicFunction = defineFunction({
125
+ key: "dxos.org/function/extraction/extract-entities",
126
+ name: "Extract Entities",
125
127
  description: "Extract entities from the transcript message and add them to the message.",
126
128
  inputSchema: ExtractionInput,
127
129
  outputSchema: ExtractionOutput,
@@ -174,13 +176,15 @@ var extractionAnthropicFunction = defineFunction({
174
176
 
175
177
  // src/extraction/extraction-ner-function/extraction-ner-function.ts
176
178
  import { asyncTimeout as asyncTimeout2 } from "@dxos/async";
177
- import { create as create2 } from "@dxos/echo-schema";
179
+ import { create as create2 } from "@dxos/echo/internal";
178
180
  import { DatabaseService, defineFunction as defineFunction2 } from "@dxos/functions";
179
181
  import { invariant } from "@dxos/invariant";
180
182
  import { log as log3 } from "@dxos/log";
181
183
  import { DataType as DataType3 } from "@dxos/schema";
182
184
  var __dxlog_file3 = "/__w/dxos/dxos/packages/core/assistant/src/extraction/extraction-ner-function/extraction-ner-function.ts";
183
185
  var extractionNerFunction = defineFunction2({
186
+ key: "dxos.org/function/extraction/extract-entities",
187
+ name: "Extract Entities",
184
188
  description: "Extracts named entities from a text",
185
189
  inputSchema: ExtractionInput,
186
190
  outputSchema: ExtractionOutput,
@@ -190,7 +194,7 @@ var extractionNerFunction = defineFunction2({
190
194
  options
191
195
  }, {
192
196
  F: __dxlog_file3,
193
- L: 27,
197
+ L: 29,
194
198
  S: void 0,
195
199
  C: (f, a) => f(...a)
196
200
  });
@@ -200,7 +204,7 @@ var extractionNerFunction = defineFunction2({
200
204
  const entitiesPromise = Promise.all(message.blocks.map(async (block) => {
201
205
  invariant(block._tag === "transcript" || block._tag === "text", void 0, {
202
206
  F: __dxlog_file3,
203
- L: 34,
207
+ L: 36,
204
208
  S: void 0,
205
209
  A: [
206
210
  "block._tag === 'transcript' || block._tag === 'text'",
@@ -214,7 +218,7 @@ var extractionNerFunction = defineFunction2({
214
218
  entities
215
219
  }, {
216
220
  F: __dxlog_file3,
217
- L: 40,
221
+ L: 42,
218
222
  S: void 0,
219
223
  C: (f, a) => f(...a)
220
224
  });
@@ -222,7 +226,7 @@ var extractionNerFunction = defineFunction2({
222
226
  const blocksWithReferences = message.blocks.map((block) => {
223
227
  invariant(block._tag === "transcript" || block._tag === "text", void 0, {
224
228
  F: __dxlog_file3,
225
- L: 47,
229
+ L: 49,
226
230
  S: void 0,
227
231
  A: [
228
232
  "block._tag === 'transcript' || block._tag === 'text'",
@@ -242,7 +246,7 @@ var extractionNerFunction = defineFunction2({
242
246
  messageWithReferences
243
247
  }, {
244
248
  F: __dxlog_file3,
245
- L: 56,
249
+ L: 58,
246
250
  S: void 0,
247
251
  C: (f, a) => f(...a)
248
252
  });
@@ -1,7 +1,7 @@
1
1
  {
2
2
  "version": 3,
3
- "sources": ["../../../../src/extraction/extraction.ts", "../../../../src/extraction/extraction-llm-function/extraction-llm-function.ts", "../../../../src/extraction/quotes.ts", "raw-loader:./instructions.tpl?raw", "../../../../src/extraction/extraction-ner-function/extraction-ner-function.ts"],
4
- "sourcesContent": ["//\n// Copyright 2025 DXOS.org\n//\n\nimport { Schema } from 'effect';\n\nimport { asyncTimeout } from '@dxos/async';\nimport { Expando } from '@dxos/echo-schema';\nimport { type FunctionDefinition, type FunctionExecutor } from '@dxos/functions';\nimport { log } from '@dxos/log';\nimport { DataType } from '@dxos/schema';\n\nexport const ExtractionInput = Schema.Struct({\n message: DataType.Message,\n objects: Schema.optional(Schema.Array(Expando)),\n options: Schema.optional(\n Schema.Struct({\n timeout: Schema.optional(Schema.Number),\n fallbackToRaw: Schema.optional(Schema.Boolean),\n }),\n ),\n});\nexport interface ExtractionInput extends Schema.Schema.Type<typeof ExtractionInput> {}\n\nexport const ExtractionOutput = Schema.Struct({\n message: DataType.Message,\n timeElapsed: Schema.Number,\n});\nexport interface ExtractionOutput extends Schema.Schema.Type<typeof ExtractionOutput> {}\n\nexport type ProcessTranscriptMessageParams = {\n input: ExtractionInput;\n function: FunctionDefinition<ExtractionInput, ExtractionOutput>;\n executor: FunctionExecutor;\n\n options?: {\n /**\n * Timeout for the entity extraction.\n */\n timeout?: number;\n\n /**\n * Fallback to raw text if the entity extraction fails.\n * Otherwise the function will throw an error.\n * @default false\n */\n fallbackToRaw?: boolean;\n };\n};\n\nexport type ExtractionFunction = FunctionDefinition<ExtractionInput, ExtractionOutput>;\n\n/**\n * Extract entities from the transcript message and add them to the message.\n */\n// TODO(dmaretskyi): Move context to a vector search index.\nexport const processTranscriptMessage = async (params: ProcessTranscriptMessageParams): Promise<ExtractionOutput> => {\n try {\n if (params.options?.timeout && params.options.timeout > 0) {\n return await asyncTimeout(params.executor.invoke(params.function, params.input), params.options.timeout);\n } else {\n return await params.executor.invoke(params.function, params.input);\n }\n } catch (error) {\n if (params.options?.fallbackToRaw) {\n log.warn('failed to process message, falling back to raw text', { error });\n return {\n message: params.input.message,\n timeElapsed: 0,\n };\n } else {\n throw error;\n }\n }\n};\n", "//\n// Copyright 2025 DXOS.org\n//\n\n// ISSUE(burdon): defineFunction\n// @ts-nocheck\n\nimport { AiService } from '@dxos/ai';\nimport { Obj } from '@dxos/echo';\nimport { create } from '@dxos/echo-schema';\nimport { type FunctionDefinition, defineFunction } from '@dxos/functions';\nimport { type ContentBlock, DataType } from '@dxos/schema';\n\nimport { AiSession } from '../../session';\nimport { ExtractionInput, ExtractionOutput } from '../extraction';\nimport { ReferencedQuotes, insertReferences } from '../quotes';\n\nimport PROMPT from './instructions.tpl?raw';\n\nexport const extractionAnthropicFunction: FunctionDefinition<ExtractionInput, ExtractionOutput> = defineFunction({\n description: 'Extract entities from the transcript message and add them to the message.',\n inputSchema: ExtractionInput,\n outputSchema: ExtractionOutput,\n handler: async ({ data: { message, objects }, context }) => {\n const startTime = performance.now();\n const ai = context.getService(AiService.AiService);\n const session = new AiSession({ operationModel: 'configured' });\n const result = await session.runStructured(ReferencedQuotes, {\n generationOptions: {\n model: '@anthropic/claude-3-5-haiku-20241022',\n },\n client: ai.client,\n systemPrompt: PROMPT,\n history: [\n Obj.make(DataType.Message, {\n created: new Date().toISOString(),\n sender: { role: 'user' },\n blocks: [\n {\n _tag: 'text',\n text: `<context>${JSON.stringify(objects)}</context>`,\n } satisfies ContentBlock.Text,\n {\n _tag: 'text',\n text: `<transcript>${JSON.stringify(message.blocks)}</transcript>`,\n } satisfies ContentBlock.Text,\n ],\n }),\n ],\n artifacts: [],\n prompt: '',\n tools: [],\n } as any); // TODO(burdon): Rewrite test.\n\n return {\n timeElapsed: performance.now() - startTime,\n message: create(DataType.Message, {\n ...message,\n blocks: message.blocks.map((block) =>\n block._tag !== 'transcript'\n ? block\n : {\n ...block,\n text: insertReferences(block.text, result),\n },\n ),\n }),\n };\n },\n});\n", "//\n// Copyright 2025 DXOS.org\n//\n\nimport { Schema } from 'effect';\n\nimport { Filter, Query } from '@dxos/echo';\nimport { type EchoDatabase } from '@dxos/echo-db';\nimport { ObjectId } from '@dxos/echo-schema';\nimport { DXN } from '@dxos/keys';\nimport { log } from '@dxos/log';\n\nexport const ReferencedQuotes = Schema.Struct({\n references: Schema.Array(\n Schema.Struct({\n id: Schema.String,\n quote: Schema.String, // TODO(burdon): Quote?\n }),\n )\n .pipe(Schema.mutable)\n .annotations({\n // TODO(burdon): Does this description make sense?\n description: `\n The references to the context objects that are mentioned in the transcript. \n Quote should match the original transcript text exactly, while id is the id of the context object.\n `,\n }),\n});\nexport type ReferencedQuotes = Schema.Schema.Type<typeof ReferencedQuotes>;\n\nexport const findQuotes = async (quotes: string[], db: EchoDatabase): Promise<ReferencedQuotes> => {\n const references = await Promise.all(\n quotes.map(async (quote) => {\n const { objects } = await db.query(Query.select(Filter.text(quote, { type: 'vector' }))).run();\n log.info('found quotes in vector index', { objects, quote });\n return objects.length > 0 ? [{ id: objects[0].id.toString(), quote }] : [];\n }),\n );\n return { references: references.flat() };\n};\n\n/**\n * Finds and replaces all quotes with DXNs references.\n */\n// TODO(dmaretskyi): Lookup and verifiy ids from provided context.\nexport const insertReferences = (text: string, quotes: ReferencedQuotes) => {\n for (const quote of quotes.references) {\n if (!ObjectId.isValid(quote.id)) {\n continue;\n }\n\n // Use a case-insensitive regular expression to replace the quote.\n const regex = new RegExp(quote.quote.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\$&'), 'gi');\n text = text.replace(regex, `[${quote.quote}](${DXN.fromLocalObjectId(quote.id).toString()})`);\n }\n\n return text;\n};\n", "You are an AI assistant specialized in processing and enhancing transcripts with contextual information.\nYour task is to analyze a given transcript and extract references to the context objects.\n\nThe context and transcript are provided in the user message in `context` and `transcript` XML tags.\n\nHere are your instructions:\n\n1. Read the transcript and context information provided above.\n\n2. Process each segment of the transcript separately.\n\n3. For each segment:\n a. Analyze the content and identify relevant information from the context.\n b. Extract a number of references to the context objects that are mentioned in the segment.\n c. Output the references in the format `{ \"quote\": \"earnings report\", \"id\": \"01JT0JP9AX0XKGZX4MV4B69VT6\" }`.\n d. If there are no references, just output an empty array.\n\n4. Maintain the exact order and structure of the original transcript segments.\n\nOutput Format:\nYou have been provided with a tool that specifies output format\n\nRemember:\n- Only quote the text that is in the transcript.\n- Analyze each segment carefully before processing it.\n- Only reference objects that appear in the context and exactly by their id, If the object is not in the context, do not insert a reference. \n\nNow, process the transcript and context provided below according to these instructions.", "//\n// Copyright 2025 DXOS.org\n//\n\n// ISSUE(burdon): defineFunction\n// @ts-nocheck\n\nimport { asyncTimeout } from '@dxos/async';\nimport { create } from '@dxos/echo-schema';\nimport { DatabaseService, defineFunction } from '@dxos/functions';\nimport { invariant } from '@dxos/invariant';\nimport { log } from '@dxos/log';\nimport { DataType } from '@dxos/schema';\n\nimport { ExtractionInput, ExtractionOutput } from '../extraction';\nimport { findQuotes, insertReferences } from '../quotes';\n\n/**\n * Named Entity Recognition model\n */\n// TODO(mykola): Make it use vector index to query objects and not pass objects as input.\nexport const extractionNerFunction = defineFunction({\n description: 'Extracts named entities from a text',\n inputSchema: ExtractionInput,\n outputSchema: ExtractionOutput,\n handler: async ({ data: { message, options }, context }) => {\n log.info('input', { message, options });\n const startTime = performance.now();\n const { db } = context.getService(DatabaseService);\n const { extractFullEntities } = await import('./named-entity-recognition');\n\n const entitiesPromise = Promise.all(\n message.blocks.map(async (block) => {\n invariant(block._tag === 'transcript' || block._tag === 'text');\n return extractFullEntities(block.text);\n }),\n ).then((entities) => entities.flat());\n const entities = options?.timeout ? await asyncTimeout(entitiesPromise, options.timeout) : await entitiesPromise;\n\n log.info('entities', { entities });\n const quoteReferences = await findQuotes(\n entities.map((entity) => entity.word),\n db,\n );\n\n const blocksWithReferences = message.blocks.map((block) => {\n invariant(block._tag === 'transcript' || block._tag === 'text');\n return { ...block, text: insertReferences(block._tag, quoteReferences) };\n });\n\n const messageWithReferences = create(DataType.Message, {\n ...message,\n blocks: blocksWithReferences,\n });\n\n log.info('output', { messageWithReferences });\n return {\n message: messageWithReferences,\n timeElapsed: performance.now() - startTime,\n };\n },\n});\n"],
5
- "mappings": ";;;;;;;;;;;;AAIA,SAASA,cAAc;AAEvB,SAASC,oBAAoB;AAC7B,SAASC,eAAe;AAExB,SAASC,WAAW;AACpB,SAASC,gBAAgB;;AAElB,IAAMC,kBAAkBL,OAAOM,OAAO;EAC3CC,SAASH,SAASI;EAClBC,SAAST,OAAOU,SAASV,OAAOW,MAAMT,OAAAA,CAAAA;EACtCU,SAASZ,OAAOU,SACdV,OAAOM,OAAO;IACZO,SAASb,OAAOU,SAASV,OAAOc,MAAM;IACtCC,eAAef,OAAOU,SAASV,OAAOgB,OAAO;EAC/C,CAAA,CAAA;AAEJ,CAAA;AAGO,IAAMC,mBAAmBjB,OAAOM,OAAO;EAC5CC,SAASH,SAASI;EAClBU,aAAalB,OAAOc;AACtB,CAAA;AA6BO,IAAMK,2BAA2B,OAAOC,WAAAA;AAC7C,MAAI;AACF,QAAIA,OAAOR,SAASC,WAAWO,OAAOR,QAAQC,UAAU,GAAG;AACzD,aAAO,MAAMZ,aAAamB,OAAOC,SAASC,OAAOF,OAAOG,UAAUH,OAAOI,KAAK,GAAGJ,OAAOR,QAAQC,OAAO;IACzG,OAAO;AACL,aAAO,MAAMO,OAAOC,SAASC,OAAOF,OAAOG,UAAUH,OAAOI,KAAK;IACnE;EACF,SAASC,OAAO;AACd,QAAIL,OAAOR,SAASG,eAAe;AACjCZ,UAAIuB,KAAK,uDAAuD;QAAED;MAAM,GAAA;;;;;;AACxE,aAAO;QACLlB,SAASa,OAAOI,MAAMjB;QACtBW,aAAa;MACf;IACF,OAAO;AACL,YAAMO;IACR;EACF;AACF;;;ACnEA,SAASE,iBAAiB;AAC1B,SAASC,WAAW;AACpB,SAASC,cAAc;AACvB,SAAkCC,sBAAsB;AACxD,SAA4BC,YAAAA,iBAAgB;;;ACP5C,SAASC,UAAAA,eAAc;AAEvB,SAASC,QAAQC,aAAa;AAE9B,SAASC,gBAAgB;AACzB,SAASC,WAAW;AACpB,SAASC,OAAAA,YAAW;;AAEb,IAAMC,mBAAmBN,QAAOO,OAAO;EAC5CC,YAAYR,QAAOS,MACjBT,QAAOO,OAAO;IACZG,IAAIV,QAAOW;IACXC,OAAOZ,QAAOW;EAChB,CAAA,CAAA,EAECE,KAAKb,QAAOc,OAAO,EACnBC,YAAY;;IAEXC,aAAa;;;;EAIf,CAAA;AACJ,CAAA;AAGO,IAAMC,aAAa,OAAOC,QAAkBC,OAAAA;AACjD,QAAMX,aAAa,MAAMY,QAAQC,IAC/BH,OAAOI,IAAI,OAAOV,UAAAA;AAChB,UAAM,EAAEW,QAAO,IAAK,MAAMJ,GAAGK,MAAMtB,MAAMuB,OAAOxB,OAAOyB,KAAKd,OAAO;MAAEe,MAAM;IAAS,CAAA,CAAA,CAAA,EAAKC,IAAG;AAC5FvB,IAAAA,KAAIwB,KAAK,gCAAgC;MAAEN;MAASX;IAAM,GAAA;;;;;;AAC1D,WAAOW,QAAQO,SAAS,IAAI;MAAC;QAAEpB,IAAIa,QAAQ,CAAA,EAAGb,GAAGqB,SAAQ;QAAInB;MAAM;QAAK,CAAA;EAC1E,CAAA,CAAA;AAEF,SAAO;IAAEJ,YAAYA,WAAWwB,KAAI;EAAG;AACzC;AAMO,IAAMC,mBAAmB,CAACP,MAAcR,WAAAA;AAC7C,aAAWN,SAASM,OAAOV,YAAY;AACrC,QAAI,CAACL,SAAS+B,QAAQtB,MAAMF,EAAE,GAAG;AAC/B;IACF;AAGA,UAAMyB,QAAQ,IAAIC,OAAOxB,MAAMA,MAAMyB,QAAQ,uBAAuB,MAAA,GAAS,IAAA;AAC7EX,WAAOA,KAAKW,QAAQF,OAAO,IAAIvB,MAAMA,KAAK,KAAKR,IAAIkC,kBAAkB1B,MAAMF,EAAE,EAAEqB,SAAQ,CAAA,GAAK;EAC9F;AAEA,SAAOL;AACT;;;ACzDA;;;AFmBO,IAAMa,8BAAqFC,eAAe;EAC/GC,aAAa;EACbC,aAAaC;EACbC,cAAcC;EACdC,SAAS,OAAO,EAAEC,MAAM,EAAEC,SAASC,QAAO,GAAIC,QAAO,MAAE;AACrD,UAAMC,YAAYC,YAAYC,IAAG;AACjC,UAAMC,KAAKJ,QAAQK,WAAWC,UAAUA,SAAS;AACjD,UAAMC,UAAU,IAAIC,UAAU;MAAEC,gBAAgB;IAAa,CAAA;AAC7D,UAAMC,SAAS,MAAMH,QAAQI,cAAcC,kBAAkB;MAC3DC,mBAAmB;QACjBC,OAAO;MACT;MACAC,QAAQX,GAAGW;MACXC,cAAcC;MACdC,SAAS;QACPC,IAAIC,KAAKC,UAASC,SAAS;UACzBC,UAAS,oBAAIC,KAAAA,GAAOC,YAAW;UAC/BC,QAAQ;YAAEC,MAAM;UAAO;UACvBC,QAAQ;YACN;cACEC,MAAM;cACNC,MAAM,YAAYC,KAAKC,UAAUjC,OAAAA,CAAAA;YACnC;YACA;cACE8B,MAAM;cACNC,MAAM,eAAeC,KAAKC,UAAUlC,QAAQ8B,MAAM,CAAA;YACpD;;QAEJ,CAAA;;MAEFK,WAAW,CAAA;MACXC,QAAQ;MACRC,OAAO,CAAA;IACT,CAAA;AAEA,WAAO;MACLC,aAAalC,YAAYC,IAAG,IAAKF;MACjCH,SAASuC,OAAOhB,UAASC,SAAS;QAChC,GAAGxB;QACH8B,QAAQ9B,QAAQ8B,OAAOU,IAAI,CAACC,UAC1BA,MAAMV,SAAS,eACXU,QACA;UACE,GAAGA;UACHT,MAAMU,iBAAiBD,MAAMT,MAAMpB,MAAAA;QACrC,CAAA;MAER,CAAA;IACF;EACF;AACF,CAAA;;;AG9DA,SAAS+B,gBAAAA,qBAAoB;AAC7B,SAASC,UAAAA,eAAc;AACvB,SAASC,iBAAiBC,kBAAAA,uBAAsB;AAChD,SAASC,iBAAiB;AAC1B,SAASC,OAAAA,YAAW;AACpB,SAASC,YAAAA,iBAAgB;;AASlB,IAAMC,wBAAwBC,gBAAe;EAClDC,aAAa;EACbC,aAAaC;EACbC,cAAcC;EACdC,SAAS,OAAO,EAAEC,MAAM,EAAEC,SAASC,QAAO,GAAIC,QAAO,MAAE;AACrDC,IAAAA,KAAIC,KAAK,SAAS;MAAEJ;MAASC;IAAQ,GAAA;;;;;;AACrC,UAAMI,YAAYC,YAAYC,IAAG;AACjC,UAAM,EAAEC,GAAE,IAAKN,QAAQO,WAAWC,eAAAA;AAClC,UAAM,EAAEC,qBAAAA,qBAAmB,IAAK,MAAM,OAAO,0CAAA;AAE7C,UAAMC,kBAAkBC,QAAQC,IAC9Bd,QAAQe,OAAOC,IAAI,OAAOC,UAAAA;AACxBC,gBAAUD,MAAME,SAAS,gBAAgBF,MAAME,SAAS,QAAA,QAAA;;;;;;;;;AACxD,aAAOR,qBAAoBM,MAAMG,IAAI;IACvC,CAAA,CAAA,EACAC,KAAK,CAACC,cAAaA,UAASC,KAAI,CAAA;AAClC,UAAMD,WAAWrB,SAASuB,UAAU,MAAMC,cAAab,iBAAiBX,QAAQuB,OAAO,IAAI,MAAMZ;AAEjGT,IAAAA,KAAIC,KAAK,YAAY;MAAEkB;IAAS,GAAA;;;;;;AAChC,UAAMI,kBAAkB,MAAMC,WAC5BL,SAASN,IAAI,CAACY,WAAWA,OAAOC,IAAI,GACpCrB,EAAAA;AAGF,UAAMsB,uBAAuB9B,QAAQe,OAAOC,IAAI,CAACC,UAAAA;AAC/CC,gBAAUD,MAAME,SAAS,gBAAgBF,MAAME,SAAS,QAAA,QAAA;;;;;;;;;AACxD,aAAO;QAAE,GAAGF;QAAOG,MAAMW,iBAAiBd,MAAME,MAAMO,eAAAA;MAAiB;IACzE,CAAA;AAEA,UAAMM,wBAAwBC,QAAOC,UAASC,SAAS;MACrD,GAAGnC;MACHe,QAAQe;IACV,CAAA;AAEA3B,IAAAA,KAAIC,KAAK,UAAU;MAAE4B;IAAsB,GAAA;;;;;;AAC3C,WAAO;MACLhC,SAASgC;MACTI,aAAa9B,YAAYC,IAAG,IAAKF;IACnC;EACF;AACF,CAAA;",
6
- "names": ["Schema", "asyncTimeout", "Expando", "log", "DataType", "ExtractionInput", "Struct", "message", "Message", "objects", "optional", "Array", "options", "timeout", "Number", "fallbackToRaw", "Boolean", "ExtractionOutput", "timeElapsed", "processTranscriptMessage", "params", "executor", "invoke", "function", "input", "error", "warn", "AiService", "Obj", "create", "defineFunction", "DataType", "Schema", "Filter", "Query", "ObjectId", "DXN", "log", "ReferencedQuotes", "Struct", "references", "Array", "id", "String", "quote", "pipe", "mutable", "annotations", "description", "findQuotes", "quotes", "db", "Promise", "all", "map", "objects", "query", "select", "text", "type", "run", "info", "length", "toString", "flat", "insertReferences", "isValid", "regex", "RegExp", "replace", "fromLocalObjectId", "extractionAnthropicFunction", "defineFunction", "description", "inputSchema", "ExtractionInput", "outputSchema", "ExtractionOutput", "handler", "data", "message", "objects", "context", "startTime", "performance", "now", "ai", "getService", "AiService", "session", "AiSession", "operationModel", "result", "runStructured", "ReferencedQuotes", "generationOptions", "model", "client", "systemPrompt", "PROMPT", "history", "Obj", "make", "DataType", "Message", "created", "Date", "toISOString", "sender", "role", "blocks", "_tag", "text", "JSON", "stringify", "artifacts", "prompt", "tools", "timeElapsed", "create", "map", "block", "insertReferences", "asyncTimeout", "create", "DatabaseService", "defineFunction", "invariant", "log", "DataType", "extractionNerFunction", "defineFunction", "description", "inputSchema", "ExtractionInput", "outputSchema", "ExtractionOutput", "handler", "data", "message", "options", "context", "log", "info", "startTime", "performance", "now", "db", "getService", "DatabaseService", "extractFullEntities", "entitiesPromise", "Promise", "all", "blocks", "map", "block", "invariant", "_tag", "text", "then", "entities", "flat", "timeout", "asyncTimeout", "quoteReferences", "findQuotes", "entity", "word", "blocksWithReferences", "insertReferences", "messageWithReferences", "create", "DataType", "Message", "timeElapsed"]
3
+ "sources": ["../../../../src/extraction/extraction.ts", "../../../../src/extraction/extraction-llm-function/extraction-llm-function.ts", "../../../../src/extraction/quotes.ts", "raw-loader:/__w/dxos/dxos/packages/core/assistant/src/extraction/extraction-llm-function/instructions.tpl?raw", "../../../../src/extraction/extraction-ner-function/extraction-ner-function.ts"],
4
+ "sourcesContent": ["//\n// Copyright 2025 DXOS.org\n//\n\nimport * as Schema from 'effect/Schema';\n\nimport { asyncTimeout } from '@dxos/async';\nimport { Expando } from '@dxos/echo/internal';\nimport { type FunctionDefinition, type FunctionExecutor } from '@dxos/functions';\nimport { log } from '@dxos/log';\nimport { DataType } from '@dxos/schema';\n\nexport const ExtractionInput = Schema.Struct({\n message: DataType.Message,\n objects: Schema.optional(Schema.Array(Expando)),\n options: Schema.optional(\n Schema.Struct({\n timeout: Schema.optional(Schema.Number),\n fallbackToRaw: Schema.optional(Schema.Boolean),\n }),\n ),\n});\nexport interface ExtractionInput extends Schema.Schema.Type<typeof ExtractionInput> {}\n\nexport const ExtractionOutput = Schema.Struct({\n message: DataType.Message,\n timeElapsed: Schema.Number,\n});\nexport interface ExtractionOutput extends Schema.Schema.Type<typeof ExtractionOutput> {}\n\nexport type ProcessTranscriptMessageParams = {\n input: ExtractionInput;\n function: FunctionDefinition<ExtractionInput, ExtractionOutput>;\n executor: FunctionExecutor;\n\n options?: {\n /**\n * Timeout for the entity extraction.\n */\n timeout?: number;\n\n /**\n * Fallback to raw text if the entity extraction fails.\n * Otherwise the function will throw an error.\n * @default false\n */\n fallbackToRaw?: boolean;\n };\n};\n\nexport type ExtractionFunction = FunctionDefinition<ExtractionInput, ExtractionOutput>;\n\n/**\n * Extract entities from the transcript message and add them to the message.\n */\n// TODO(dmaretskyi): Move context to a vector search index.\nexport const processTranscriptMessage = async (params: ProcessTranscriptMessageParams): Promise<ExtractionOutput> => {\n try {\n if (params.options?.timeout && params.options.timeout > 0) {\n return await asyncTimeout(params.executor.invoke(params.function, params.input), params.options.timeout);\n } else {\n return await params.executor.invoke(params.function, params.input);\n }\n } catch (error) {\n if (params.options?.fallbackToRaw) {\n log.warn('failed to process message, falling back to raw text', { error });\n return {\n message: params.input.message,\n timeElapsed: 0,\n };\n } else {\n throw error;\n }\n }\n};\n", "//\n// Copyright 2025 DXOS.org\n//\n\n// ISSUE(burdon): defineFunction\n// @ts-nocheck\n\nimport { AiService } from '@dxos/ai';\nimport { Obj } from '@dxos/echo';\nimport { create } from '@dxos/echo/internal';\nimport { type FunctionDefinition, defineFunction } from '@dxos/functions';\nimport { type ContentBlock, DataType } from '@dxos/schema';\n\nimport { AiSession } from '../../session';\nimport { ExtractionInput, ExtractionOutput } from '../extraction';\nimport { ReferencedQuotes, insertReferences } from '../quotes';\n\nimport PROMPT from './instructions.tpl?raw';\n\nexport const extractionAnthropicFunction: FunctionDefinition<ExtractionInput, ExtractionOutput> = defineFunction({\n key: 'dxos.org/function/extraction/extract-entities',\n name: 'Extract Entities',\n description: 'Extract entities from the transcript message and add them to the message.',\n inputSchema: ExtractionInput,\n outputSchema: ExtractionOutput,\n handler: async ({ data: { message, objects }, context }) => {\n const startTime = performance.now();\n const ai = context.getService(AiService.AiService);\n const session = new AiSession({ operationModel: 'configured' });\n const result = await session.runStructured(ReferencedQuotes, {\n generationOptions: {\n model: '@anthropic/claude-3-5-haiku-20241022',\n },\n client: ai.client,\n systemPrompt: PROMPT,\n history: [\n Obj.make(DataType.Message, {\n created: new Date().toISOString(),\n sender: { role: 'user' },\n blocks: [\n {\n _tag: 'text',\n text: `<context>${JSON.stringify(objects)}</context>`,\n } satisfies ContentBlock.Text,\n {\n _tag: 'text',\n text: `<transcript>${JSON.stringify(message.blocks)}</transcript>`,\n } satisfies ContentBlock.Text,\n ],\n }),\n ],\n artifacts: [],\n prompt: '',\n tools: [],\n } as any); // TODO(burdon): Rewrite test.\n\n return {\n timeElapsed: performance.now() - startTime,\n message: create(DataType.Message, {\n ...message,\n blocks: message.blocks.map((block) =>\n block._tag !== 'transcript'\n ? block\n : {\n ...block,\n text: insertReferences(block.text, result),\n },\n ),\n }),\n };\n },\n});\n", "//\n// Copyright 2025 DXOS.org\n//\n\nimport * as Schema from 'effect/Schema';\n\nimport { Filter, Query } from '@dxos/echo';\nimport { ObjectId } from '@dxos/echo/internal';\nimport { type EchoDatabase } from '@dxos/echo-db';\nimport { DXN } from '@dxos/keys';\nimport { log } from '@dxos/log';\n\nexport const ReferencedQuotes = Schema.Struct({\n references: Schema.Array(\n Schema.Struct({\n id: Schema.String,\n quote: Schema.String, // TODO(burdon): Quote?\n }),\n )\n .pipe(Schema.mutable)\n .annotations({\n // TODO(burdon): Does this description make sense?\n description: `\n The references to the context objects that are mentioned in the transcript. \n Quote should match the original transcript text exactly, while id is the id of the context object.\n `,\n }),\n});\nexport type ReferencedQuotes = Schema.Schema.Type<typeof ReferencedQuotes>;\n\nexport const findQuotes = async (quotes: string[], db: EchoDatabase): Promise<ReferencedQuotes> => {\n const references = await Promise.all(\n quotes.map(async (quote) => {\n const { objects } = await db.query(Query.select(Filter.text(quote, { type: 'vector' }))).run();\n log.info('found quotes in vector index', { objects, quote });\n return objects.length > 0 ? [{ id: objects[0].id.toString(), quote }] : [];\n }),\n );\n return { references: references.flat() };\n};\n\n/**\n * Finds and replaces all quotes with DXNs references.\n */\n// TODO(dmaretskyi): Lookup and verifiy ids from provided context.\nexport const insertReferences = (text: string, quotes: ReferencedQuotes) => {\n for (const quote of quotes.references) {\n if (!ObjectId.isValid(quote.id)) {\n continue;\n }\n\n // Use a case-insensitive regular expression to replace the quote.\n const regex = new RegExp(quote.quote.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\$&'), 'gi');\n text = text.replace(regex, `[${quote.quote}](${DXN.fromLocalObjectId(quote.id).toString()})`);\n }\n\n return text;\n};\n", "You are an AI assistant specialized in processing and enhancing transcripts with contextual information.\nYour task is to analyze a given transcript and extract references to the context objects.\n\nThe context and transcript are provided in the user message in `context` and `transcript` XML tags.\n\nHere are your instructions:\n\n1. Read the transcript and context information provided above.\n\n2. Process each segment of the transcript separately.\n\n3. For each segment:\n a. Analyze the content and identify relevant information from the context.\n b. Extract a number of references to the context objects that are mentioned in the segment.\n c. Output the references in the format `{ \"quote\": \"earnings report\", \"id\": \"01JT0JP9AX0XKGZX4MV4B69VT6\" }`.\n d. If there are no references, just output an empty array.\n\n4. Maintain the exact order and structure of the original transcript segments.\n\nOutput Format:\nYou have been provided with a tool that specifies output format\n\nRemember:\n- Only quote the text that is in the transcript.\n- Analyze each segment carefully before processing it.\n- Only reference objects that appear in the context and exactly by their id, If the object is not in the context, do not insert a reference. \n\nNow, process the transcript and context provided below according to these instructions.", "//\n// Copyright 2025 DXOS.org\n//\n\n// ISSUE(burdon): defineFunction\n// @ts-nocheck\n\nimport { asyncTimeout } from '@dxos/async';\nimport { create } from '@dxos/echo/internal';\nimport { DatabaseService, defineFunction } from '@dxos/functions';\nimport { invariant } from '@dxos/invariant';\nimport { log } from '@dxos/log';\nimport { DataType } from '@dxos/schema';\n\nimport { ExtractionInput, ExtractionOutput } from '../extraction';\nimport { findQuotes, insertReferences } from '../quotes';\n\n/**\n * Named Entity Recognition model\n */\n// TODO(mykola): Make it use vector index to query objects and not pass objects as input.\nexport const extractionNerFunction = defineFunction({\n key: 'dxos.org/function/extraction/extract-entities',\n name: 'Extract Entities',\n description: 'Extracts named entities from a text',\n inputSchema: ExtractionInput,\n outputSchema: ExtractionOutput,\n handler: async ({ data: { message, options }, context }) => {\n log.info('input', { message, options });\n const startTime = performance.now();\n const { db } = context.getService(DatabaseService);\n const { extractFullEntities } = await import('./named-entity-recognition');\n\n const entitiesPromise = Promise.all(\n message.blocks.map(async (block) => {\n invariant(block._tag === 'transcript' || block._tag === 'text');\n return extractFullEntities(block.text);\n }),\n ).then((entities) => entities.flat());\n const entities = options?.timeout ? await asyncTimeout(entitiesPromise, options.timeout) : await entitiesPromise;\n\n log.info('entities', { entities });\n const quoteReferences = await findQuotes(\n entities.map((entity) => entity.word),\n db,\n );\n\n const blocksWithReferences = message.blocks.map((block) => {\n invariant(block._tag === 'transcript' || block._tag === 'text');\n return { ...block, text: insertReferences(block._tag, quoteReferences) };\n });\n\n const messageWithReferences = create(DataType.Message, {\n ...message,\n blocks: blocksWithReferences,\n });\n\n log.info('output', { messageWithReferences });\n return {\n message: messageWithReferences,\n timeElapsed: performance.now() - startTime,\n };\n },\n});\n"],
5
+ "mappings": ";;;;;;;;;;;;AAIA,YAAYA,YAAY;AAExB,SAASC,oBAAoB;AAC7B,SAASC,eAAe;AAExB,SAASC,WAAW;AACpB,SAASC,gBAAgB;;AAElB,IAAMC,kBAAyBC,cAAO;EAC3CC,SAASH,SAASI;EAClBC,SAAgBC,gBAAgBC,aAAMT,OAAAA,CAAAA;EACtCU,SAAgBF,gBACPJ,cAAO;IACZO,SAAgBH,gBAAgBI,aAAM;IACtCC,eAAsBL,gBAAgBM,cAAO;EAC/C,CAAA,CAAA;AAEJ,CAAA;AAGO,IAAMC,mBAA0BX,cAAO;EAC5CC,SAASH,SAASI;EAClBU,aAAoBJ;AACtB,CAAA;AA6BO,IAAMK,2BAA2B,OAAOC,WAAAA;AAC7C,MAAI;AACF,QAAIA,OAAOR,SAASC,WAAWO,OAAOR,QAAQC,UAAU,GAAG;AACzD,aAAO,MAAMZ,aAAamB,OAAOC,SAASC,OAAOF,OAAOG,UAAUH,OAAOI,KAAK,GAAGJ,OAAOR,QAAQC,OAAO;IACzG,OAAO;AACL,aAAO,MAAMO,OAAOC,SAASC,OAAOF,OAAOG,UAAUH,OAAOI,KAAK;IACnE;EACF,SAASC,OAAO;AACd,QAAIL,OAAOR,SAASG,eAAe;AACjCZ,UAAIuB,KAAK,uDAAuD;QAAED;MAAM,GAAA;;;;;;AACxE,aAAO;QACLlB,SAASa,OAAOI,MAAMjB;QACtBW,aAAa;MACf;IACF,OAAO;AACL,YAAMO;IACR;EACF;AACF;;;ACnEA,SAASE,iBAAiB;AAC1B,SAASC,WAAW;AACpB,SAASC,cAAc;AACvB,SAAkCC,sBAAsB;AACxD,SAA4BC,YAAAA,iBAAgB;;;ACP5C,YAAYC,aAAY;AAExB,SAASC,QAAQC,aAAa;AAC9B,SAASC,gBAAgB;AAEzB,SAASC,WAAW;AACpB,SAASC,OAAAA,YAAW;;AAEb,IAAMC,mBAA0BC,eAAO;EAC5CC,YAAmBC,cACVF,eAAO;IACZG,IAAWC;IACXC,OAAcD;EAChB,CAAA,CAAA,EAECE,KAAYC,eAAO,EACnBC,YAAY;;IAEXC,aAAa;;;;EAIf,CAAA;AACJ,CAAA;AAGO,IAAMC,aAAa,OAAOC,QAAkBC,OAAAA;AACjD,QAAMX,aAAa,MAAMY,QAAQC,IAC/BH,OAAOI,IAAI,OAAOV,UAAAA;AAChB,UAAM,EAAEW,QAAO,IAAK,MAAMJ,GAAGK,MAAMtB,MAAMuB,OAAOxB,OAAOyB,KAAKd,OAAO;MAAEe,MAAM;IAAS,CAAA,CAAA,CAAA,EAAKC,IAAG;AAC5FvB,IAAAA,KAAIwB,KAAK,gCAAgC;MAAEN;MAASX;IAAM,GAAA;;;;;;AAC1D,WAAOW,QAAQO,SAAS,IAAI;MAAC;QAAEpB,IAAIa,QAAQ,CAAA,EAAGb,GAAGqB,SAAQ;QAAInB;MAAM;QAAK,CAAA;EAC1E,CAAA,CAAA;AAEF,SAAO;IAAEJ,YAAYA,WAAWwB,KAAI;EAAG;AACzC;AAMO,IAAMC,mBAAmB,CAACP,MAAcR,WAAAA;AAC7C,aAAWN,SAASM,OAAOV,YAAY;AACrC,QAAI,CAACL,SAAS+B,QAAQtB,MAAMF,EAAE,GAAG;AAC/B;IACF;AAGA,UAAMyB,QAAQ,IAAIC,OAAOxB,MAAMA,MAAMyB,QAAQ,uBAAuB,MAAA,GAAS,IAAA;AAC7EX,WAAOA,KAAKW,QAAQF,OAAO,IAAIvB,MAAMA,KAAK,KAAKR,IAAIkC,kBAAkB1B,MAAMF,EAAE,EAAEqB,SAAQ,CAAA,GAAK;EAC9F;AAEA,SAAOL;AACT;;;ACzDA;;;AFmBO,IAAMa,8BAAqFC,eAAe;EAC/GC,KAAK;EACLC,MAAM;EACNC,aAAa;EACbC,aAAaC;EACbC,cAAcC;EACdC,SAAS,OAAO,EAAEC,MAAM,EAAEC,SAASC,QAAO,GAAIC,QAAO,MAAE;AACrD,UAAMC,YAAYC,YAAYC,IAAG;AACjC,UAAMC,KAAKJ,QAAQK,WAAWC,UAAUA,SAAS;AACjD,UAAMC,UAAU,IAAIC,UAAU;MAAEC,gBAAgB;IAAa,CAAA;AAC7D,UAAMC,SAAS,MAAMH,QAAQI,cAAcC,kBAAkB;MAC3DC,mBAAmB;QACjBC,OAAO;MACT;MACAC,QAAQX,GAAGW;MACXC,cAAcC;MACdC,SAAS;QACPC,IAAIC,KAAKC,UAASC,SAAS;UACzBC,UAAS,oBAAIC,KAAAA,GAAOC,YAAW;UAC/BC,QAAQ;YAAEC,MAAM;UAAO;UACvBC,QAAQ;YACN;cACEC,MAAM;cACNC,MAAM,YAAYC,KAAKC,UAAUjC,OAAAA,CAAAA;YACnC;YACA;cACE8B,MAAM;cACNC,MAAM,eAAeC,KAAKC,UAAUlC,QAAQ8B,MAAM,CAAA;YACpD;;QAEJ,CAAA;;MAEFK,WAAW,CAAA;MACXC,QAAQ;MACRC,OAAO,CAAA;IACT,CAAA;AAEA,WAAO;MACLC,aAAalC,YAAYC,IAAG,IAAKF;MACjCH,SAASuC,OAAOhB,UAASC,SAAS;QAChC,GAAGxB;QACH8B,QAAQ9B,QAAQ8B,OAAOU,IAAI,CAACC,UAC1BA,MAAMV,SAAS,eACXU,QACA;UACE,GAAGA;UACHT,MAAMU,iBAAiBD,MAAMT,MAAMpB,MAAAA;QACrC,CAAA;MAER,CAAA;IACF;EACF;AACF,CAAA;;;AGhEA,SAAS+B,gBAAAA,qBAAoB;AAC7B,SAASC,UAAAA,eAAc;AACvB,SAASC,iBAAiBC,kBAAAA,uBAAsB;AAChD,SAASC,iBAAiB;AAC1B,SAASC,OAAAA,YAAW;AACpB,SAASC,YAAAA,iBAAgB;;AASlB,IAAMC,wBAAwBC,gBAAe;EAClDC,KAAK;EACLC,MAAM;EACNC,aAAa;EACbC,aAAaC;EACbC,cAAcC;EACdC,SAAS,OAAO,EAAEC,MAAM,EAAEC,SAASC,QAAO,GAAIC,QAAO,MAAE;AACrDC,IAAAA,KAAIC,KAAK,SAAS;MAAEJ;MAASC;IAAQ,GAAA;;;;;;AACrC,UAAMI,YAAYC,YAAYC,IAAG;AACjC,UAAM,EAAEC,GAAE,IAAKN,QAAQO,WAAWC,eAAAA;AAClC,UAAM,EAAEC,qBAAAA,qBAAmB,IAAK,MAAM,OAAO,0CAAA;AAE7C,UAAMC,kBAAkBC,QAAQC,IAC9Bd,QAAQe,OAAOC,IAAI,OAAOC,UAAAA;AACxBC,gBAAUD,MAAME,SAAS,gBAAgBF,MAAME,SAAS,QAAA,QAAA;;;;;;;;;AACxD,aAAOR,qBAAoBM,MAAMG,IAAI;IACvC,CAAA,CAAA,EACAC,KAAK,CAACC,cAAaA,UAASC,KAAI,CAAA;AAClC,UAAMD,WAAWrB,SAASuB,UAAU,MAAMC,cAAab,iBAAiBX,QAAQuB,OAAO,IAAI,MAAMZ;AAEjGT,IAAAA,KAAIC,KAAK,YAAY;MAAEkB;IAAS,GAAA;;;;;;AAChC,UAAMI,kBAAkB,MAAMC,WAC5BL,SAASN,IAAI,CAACY,WAAWA,OAAOC,IAAI,GACpCrB,EAAAA;AAGF,UAAMsB,uBAAuB9B,QAAQe,OAAOC,IAAI,CAACC,UAAAA;AAC/CC,gBAAUD,MAAME,SAAS,gBAAgBF,MAAME,SAAS,QAAA,QAAA;;;;;;;;;AACxD,aAAO;QAAE,GAAGF;QAAOG,MAAMW,iBAAiBd,MAAME,MAAMO,eAAAA;MAAiB;IACzE,CAAA;AAEA,UAAMM,wBAAwBC,QAAOC,UAASC,SAAS;MACrD,GAAGnC;MACHe,QAAQe;IACV,CAAA;AAEA3B,IAAAA,KAAIC,KAAK,UAAU;MAAE4B;IAAsB,GAAA;;;;;;AAC3C,WAAO;MACLhC,SAASgC;MACTI,aAAa9B,YAAYC,IAAG,IAAKF;IACnC;EACF;AACF,CAAA;",
6
+ "names": ["Schema", "asyncTimeout", "Expando", "log", "DataType", "ExtractionInput", "Struct", "message", "Message", "objects", "optional", "Array", "options", "timeout", "Number", "fallbackToRaw", "Boolean", "ExtractionOutput", "timeElapsed", "processTranscriptMessage", "params", "executor", "invoke", "function", "input", "error", "warn", "AiService", "Obj", "create", "defineFunction", "DataType", "Schema", "Filter", "Query", "ObjectId", "DXN", "log", "ReferencedQuotes", "Struct", "references", "Array", "id", "String", "quote", "pipe", "mutable", "annotations", "description", "findQuotes", "quotes", "db", "Promise", "all", "map", "objects", "query", "select", "text", "type", "run", "info", "length", "toString", "flat", "insertReferences", "isValid", "regex", "RegExp", "replace", "fromLocalObjectId", "extractionAnthropicFunction", "defineFunction", "key", "name", "description", "inputSchema", "ExtractionInput", "outputSchema", "ExtractionOutput", "handler", "data", "message", "objects", "context", "startTime", "performance", "now", "ai", "getService", "AiService", "session", "AiSession", "operationModel", "result", "runStructured", "ReferencedQuotes", "generationOptions", "model", "client", "systemPrompt", "PROMPT", "history", "Obj", "make", "DataType", "Message", "created", "Date", "toISOString", "sender", "role", "blocks", "_tag", "text", "JSON", "stringify", "artifacts", "prompt", "tools", "timeElapsed", "create", "map", "block", "insertReferences", "asyncTimeout", "create", "DatabaseService", "defineFunction", "invariant", "log", "DataType", "extractionNerFunction", "defineFunction", "key", "name", "description", "inputSchema", "ExtractionInput", "outputSchema", "ExtractionOutput", "handler", "data", "message", "options", "context", "log", "info", "startTime", "performance", "now", "db", "getService", "DatabaseService", "extractFullEntities", "entitiesPromise", "Promise", "all", "blocks", "map", "block", "invariant", "_tag", "text", "then", "entities", "flat", "timeout", "asyncTimeout", "quoteReferences", "findQuotes", "entity", "word", "blocksWithReferences", "insertReferences", "messageWithReferences", "create", "DataType", "Message", "timeElapsed"]
7
7
  }