@dxos/assistant 0.8.4-main.b97322e → 0.8.4-main.f5c0578

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 (129) hide show
  1. package/dist/lib/browser/chunk-6WRGDEMH.mjs +89 -0
  2. package/dist/lib/browser/chunk-6WRGDEMH.mjs.map +7 -0
  3. package/dist/lib/browser/chunk-AVKQ32JL.mjs +352 -0
  4. package/dist/lib/browser/chunk-AVKQ32JL.mjs.map +7 -0
  5. package/dist/lib/browser/extraction/index.mjs +269 -0
  6. package/dist/lib/browser/extraction/index.mjs.map +7 -0
  7. package/dist/lib/browser/index.mjs +170 -1646
  8. package/dist/lib/browser/index.mjs.map +4 -4
  9. package/dist/lib/browser/meta.json +1 -1
  10. package/dist/lib/browser/named-entity-recognition-O46UTCLR.mjs +13 -0
  11. package/dist/lib/browser/named-entity-recognition-O46UTCLR.mjs.map +7 -0
  12. package/dist/lib/node-esm/chunk-3UU5V4FS.mjs +91 -0
  13. package/dist/lib/node-esm/chunk-3UU5V4FS.mjs.map +7 -0
  14. package/dist/lib/node-esm/chunk-MNAVETJA.mjs +354 -0
  15. package/dist/lib/node-esm/chunk-MNAVETJA.mjs.map +7 -0
  16. package/dist/lib/node-esm/extraction/index.mjs +270 -0
  17. package/dist/lib/node-esm/extraction/index.mjs.map +7 -0
  18. package/dist/lib/node-esm/index.mjs +169 -1646
  19. package/dist/lib/node-esm/index.mjs.map +4 -4
  20. package/dist/lib/node-esm/meta.json +1 -1
  21. package/dist/lib/node-esm/named-entity-recognition-W4VMRCH6.mjs +14 -0
  22. package/dist/lib/node-esm/named-entity-recognition-W4VMRCH6.mjs.map +7 -0
  23. package/dist/types/src/conversation/context.d.ts +90 -0
  24. package/dist/types/src/conversation/context.d.ts.map +1 -0
  25. package/dist/types/src/conversation/conversation.d.ts +27 -16
  26. package/dist/types/src/conversation/conversation.d.ts.map +1 -1
  27. package/dist/types/src/conversation/index.d.ts +1 -0
  28. package/dist/types/src/conversation/index.d.ts.map +1 -1
  29. package/dist/types/src/errors.d.ts +5 -4
  30. package/dist/types/src/errors.d.ts.map +1 -1
  31. package/dist/types/src/extraction/extraction-llm-function/extraction-llm-function.d.ts.map +1 -1
  32. package/dist/types/src/extraction/extraction-ner-function/extraction-ner-function.d.ts +27 -23
  33. package/dist/types/src/extraction/extraction-ner-function/extraction-ner-function.d.ts.map +1 -1
  34. package/dist/types/src/extraction/extraction.d.ts +14 -10
  35. package/dist/types/src/extraction/extraction.d.ts.map +1 -1
  36. package/dist/types/src/functions/services.d.ts +4 -3
  37. package/dist/types/src/functions/services.d.ts.map +1 -1
  38. package/dist/types/src/index.d.ts +1 -3
  39. package/dist/types/src/index.d.ts.map +1 -1
  40. package/dist/types/src/session/artifact-diff.d.ts +27 -0
  41. package/dist/types/src/session/artifact-diff.d.ts.map +1 -0
  42. package/dist/types/src/session/format.d.ts +168 -0
  43. package/dist/types/src/session/format.d.ts.map +1 -0
  44. package/dist/types/src/session/format.test.d.ts +2 -0
  45. package/dist/types/src/session/format.test.d.ts.map +1 -0
  46. package/dist/types/src/session/index.d.ts +1 -0
  47. package/dist/types/src/session/index.d.ts.map +1 -1
  48. package/dist/types/src/session/session.d.ts +71 -161
  49. package/dist/types/src/session/session.d.ts.map +1 -1
  50. package/dist/types/src/session/version-pin.d.ts +2 -2
  51. package/dist/types/src/session/version-pin.d.ts.map +1 -1
  52. package/dist/types/src/templates/index.d.ts +1 -1
  53. package/dist/types/src/templates/system.d.ts +22 -0
  54. package/dist/types/src/templates/system.d.ts.map +1 -0
  55. package/dist/types/src/types.d.ts +3 -0
  56. package/dist/types/src/types.d.ts.map +1 -1
  57. package/dist/types/tsconfig.tsbuildinfo +1 -1
  58. package/package.json +43 -27
  59. package/src/{context/binder.ts → conversation/context.ts} +48 -18
  60. package/src/conversation/conversation.ts +70 -39
  61. package/src/conversation/index.ts +1 -0
  62. package/src/extraction/extraction-llm-function/extraction-llm-function.test.ts +1 -0
  63. package/src/extraction/extraction-llm-function/extraction-llm-function.ts +7 -6
  64. package/src/extraction/extraction-ner-function/extraction-ner-function.test.ts +4 -3
  65. package/src/extraction/extraction-ner-function/extraction-ner-function.ts +2 -2
  66. package/src/extraction/quotes.test.ts +1 -1
  67. package/src/extraction/quotes.ts +1 -1
  68. package/src/functions/services.ts +43 -68
  69. package/src/index.ts +1 -3
  70. package/src/session/artifact-diff.ts +37 -0
  71. package/src/session/format.test.ts +60 -0
  72. package/src/session/format.ts +139 -0
  73. package/src/session/index.ts +1 -0
  74. package/src/session/session.test.ts +9 -11
  75. package/src/session/session.ts +196 -229
  76. package/src/session/version-pin.ts +4 -4
  77. package/src/summarization/summarization.test.ts +11 -22
  78. package/src/templates/format.tpl +55 -0
  79. package/src/templates/index.ts +1 -1
  80. package/src/templates/system.tpl +47 -0
  81. package/src/templates/system.ts +33 -0
  82. package/src/types.ts +10 -5
  83. package/src/typings.d.ts +5 -0
  84. package/dist/types/bin/chat.d.ts +0 -2
  85. package/dist/types/bin/chat.d.ts.map +0 -1
  86. package/dist/types/src/context/binder.d.ts +0 -34
  87. package/dist/types/src/context/binder.d.ts.map +0 -1
  88. package/dist/types/src/context/binding.d.ts +0 -56
  89. package/dist/types/src/context/binding.d.ts.map +0 -1
  90. package/dist/types/src/context/index.d.ts +0 -3
  91. package/dist/types/src/context/index.d.ts.map +0 -1
  92. package/dist/types/src/research/exa.d.ts +0 -12
  93. package/dist/types/src/research/exa.d.ts.map +0 -1
  94. package/dist/types/src/research/graph.d.ts +0 -51
  95. package/dist/types/src/research/graph.d.ts.map +0 -1
  96. package/dist/types/src/research/graph.test.d.ts +0 -2
  97. package/dist/types/src/research/graph.test.d.ts.map +0 -1
  98. package/dist/types/src/research/index.d.ts +0 -4
  99. package/dist/types/src/research/index.d.ts.map +0 -1
  100. package/dist/types/src/research/research.d.ts +0 -10
  101. package/dist/types/src/research/research.d.ts.map +0 -1
  102. package/dist/types/src/research/research.test.d.ts +0 -2
  103. package/dist/types/src/research/research.test.d.ts.map +0 -1
  104. package/dist/types/src/templates/prompt.d.ts +0 -30
  105. package/dist/types/src/templates/prompt.d.ts.map +0 -1
  106. package/dist/types/src/testing/exa-search-1748337321991.d.ts +0 -38
  107. package/dist/types/src/testing/exa-search-1748337321991.d.ts.map +0 -1
  108. package/dist/types/src/testing/exa-search-1748337331526.d.ts +0 -37
  109. package/dist/types/src/testing/exa-search-1748337331526.d.ts.map +0 -1
  110. package/dist/types/src/testing/exa-search-1748337344119.d.ts +0 -58
  111. package/dist/types/src/testing/exa-search-1748337344119.d.ts.map +0 -1
  112. package/dist/types/src/testing/index.d.ts +0 -3
  113. package/dist/types/src/testing/index.d.ts.map +0 -1
  114. package/src/context/binding.ts +0 -32
  115. package/src/context/index.ts +0 -6
  116. package/src/research/exa.ts +0 -97
  117. package/src/research/graph.test.ts +0 -64
  118. package/src/research/graph.ts +0 -369
  119. package/src/research/index.ts +0 -7
  120. package/src/research/instructions-research.tpl +0 -98
  121. package/src/research/research.test.ts +0 -190
  122. package/src/research/research.ts +0 -87
  123. package/src/templates/assistant.tpl +0 -111
  124. package/src/templates/prompt.ts +0 -45
  125. package/src/testing/exa-search-1748337321991.ts +0 -131
  126. package/src/testing/exa-search-1748337331526.ts +0 -144
  127. package/src/testing/exa-search-1748337344119.ts +0 -133
  128. package/src/testing/index.ts +0 -11
  129. package/src/typedefs.d.ts +0 -8
@@ -0,0 +1,89 @@
1
+ // src/extraction/extraction-ner-function/named-entity-recognition.ts
2
+ import { pipeline } from "@xenova/transformers";
3
+ import { log } from "@dxos/log";
4
+ var __dxlog_file = "/__w/dxos/dxos/packages/core/assistant/src/extraction/extraction-ner-function/named-entity-recognition.ts";
5
+ var _ner;
6
+ var getNer = () => {
7
+ if (!_ner) {
8
+ _ner = pipeline("ner", "Xenova/bert-base-NER").then((ner) => {
9
+ log.info("NER model is ready", void 0, {
10
+ F: __dxlog_file,
11
+ L: 23,
12
+ S: void 0,
13
+ C: (f, a) => f(...a)
14
+ });
15
+ return ner;
16
+ });
17
+ }
18
+ return _ner;
19
+ };
20
+ var extractFullEntities = async (text) => {
21
+ const ner = await getNer();
22
+ const entities = await ner(text);
23
+ const tokenGroups = createTokenGroups(entities);
24
+ const combinedEntities = tokenGroups.map(combineNerTokens);
25
+ return combinedEntities;
26
+ };
27
+ var createTokenGroups = (tokens) => {
28
+ const tokenGroups = [];
29
+ const canBeCombined = (prevToken, currentToken) => {
30
+ const [currentState, currentEntity] = currentToken.entity.split("-");
31
+ const [prevState, prevEntity] = prevToken.entity.split("-");
32
+ if (prevEntity !== currentEntity) {
33
+ return false;
34
+ }
35
+ if (prevState === "O") {
36
+ return false;
37
+ }
38
+ if (currentState === "I") {
39
+ return true;
40
+ }
41
+ if ((prevState === "B" || prevState === "I") && currentState === "B") {
42
+ return currentToken.word.startsWith("#");
43
+ }
44
+ return false;
45
+ };
46
+ for (const token of tokens) {
47
+ if (tokenGroups.length === 0) {
48
+ tokenGroups.push([
49
+ token
50
+ ]);
51
+ continue;
52
+ }
53
+ const lastTokenGroup = tokenGroups.at(-1);
54
+ if (canBeCombined(lastTokenGroup.at(-1), token)) {
55
+ lastTokenGroup.push(token);
56
+ continue;
57
+ }
58
+ tokenGroups.push([
59
+ token
60
+ ]);
61
+ }
62
+ return tokenGroups;
63
+ };
64
+ var combineNerTokens = (group) => {
65
+ const combinedToken = {
66
+ entity: group.at(0).entity.split("-")[1],
67
+ score: group.reduce((acc, token) => acc + token.score, 0) / group.length,
68
+ index: group.at(0).index,
69
+ word: group.at(0).word,
70
+ start: group.at(0).start,
71
+ end: group.at(-1).end
72
+ };
73
+ for (const token of group.slice(1)) {
74
+ if (token.word.startsWith("#")) {
75
+ combinedToken.word += token.word.replace(/^#+/, "");
76
+ } else {
77
+ combinedToken.word += " " + token.word;
78
+ }
79
+ }
80
+ return combinedToken;
81
+ };
82
+
83
+ export {
84
+ getNer,
85
+ extractFullEntities,
86
+ createTokenGroups,
87
+ combineNerTokens
88
+ };
89
+ //# sourceMappingURL=chunk-6WRGDEMH.mjs.map
@@ -0,0 +1,7 @@
1
+ {
2
+ "version": 3,
3
+ "sources": ["../../../src/extraction/extraction-ner-function/named-entity-recognition.ts"],
4
+ "sourcesContent": ["//\n// Copyright 2025 DXOS.org\n//\n\nimport {\n type TokenClassificationOutput,\n type TokenClassificationPipelineType,\n type TokenClassificationSingle,\n pipeline,\n} from '@xenova/transformers';\n\nimport { log } from '@dxos/log';\n\nlet _ner: Promise<TokenClassificationPipelineType>;\n/**\n * Named Entity Recognition pipeline.\n * Initializes the pipeline on first call.\n * @returns The singleton promise that resolves to a token classification pipeline.\n */\nexport const getNer = () => {\n if (!_ner) {\n _ner = pipeline('ner', 'Xenova/bert-base-NER').then((ner) => {\n log.info('NER model is ready');\n return ner;\n });\n }\n return _ner;\n};\n\n/**\n * Extracts entities from text using the named entity recognition pipeline and does simple grouping of tokens.\n */\nexport const extractFullEntities = async (text: string): Promise<TokenClassificationSingle[]> => {\n const ner = await getNer();\n const entities = (await ner(text)) as TokenClassificationOutput;\n const tokenGroups = createTokenGroups(entities);\n const combinedEntities = tokenGroups.map(combineNerTokens);\n return combinedEntities;\n};\n\nexport const createTokenGroups = (tokens: TokenClassificationSingle[]): TokenClassificationSingle[][] => {\n const tokenGroups: TokenClassificationSingle[][] = [];\n\n const canBeCombined = (prevToken: TokenClassificationSingle, currentToken: TokenClassificationSingle) => {\n const [currentState, currentEntity] = currentToken.entity.split('-'); // B, I, O\n const [prevState, prevEntity] = prevToken.entity.split('-'); // B, I, O\n if (prevEntity !== currentEntity) {\n return false;\n }\n if (prevState === 'O') {\n return false;\n }\n if (currentState === 'I') {\n return true;\n }\n if ((prevState === 'B' || prevState === 'I') && currentState === 'B') {\n return currentToken.word.startsWith('#');\n }\n return false;\n };\n\n for (const token of tokens) {\n if (tokenGroups.length === 0) {\n tokenGroups.push([token]);\n continue;\n }\n const lastTokenGroup = tokenGroups.at(-1)!;\n if (canBeCombined(lastTokenGroup.at(-1)!, token)) {\n lastTokenGroup.push(token);\n continue;\n }\n tokenGroups.push([token]);\n }\n\n return tokenGroups;\n};\n\nexport const combineNerTokens = (group: TokenClassificationSingle[]): TokenClassificationSingle => {\n const combinedToken: TokenClassificationSingle = {\n entity: group.at(0)!.entity.split('-')[1],\n score: group.reduce((acc, token) => acc + token.score, 0) / group.length,\n index: group.at(0)!.index,\n word: group.at(0)!.word,\n start: group.at(0)!.start,\n end: group.at(-1)!.end,\n };\n\n for (const token of group.slice(1)) {\n if (token.word.startsWith('#')) {\n combinedToken.word += token.word.replace(/^#+/, '');\n } else {\n combinedToken.word += ' ' + token.word;\n }\n }\n\n return combinedToken;\n};\n"],
5
+ "mappings": ";AAIA,SAIEA,gBACK;AAEP,SAASC,WAAW;;AAEpB,IAAIC;AAMG,IAAMC,SAAS,MAAA;AACpB,MAAI,CAACD,MAAM;AACTA,WAAOF,SAAS,OAAO,sBAAA,EAAwBI,KAAK,CAACC,QAAAA;AACnDJ,UAAIK,KAAK,sBAAA,QAAA;;;;;;AACT,aAAOD;IACT,CAAA;EACF;AACA,SAAOH;AACT;AAKO,IAAMK,sBAAsB,OAAOC,SAAAA;AACxC,QAAMH,MAAM,MAAMF,OAAAA;AAClB,QAAMM,WAAY,MAAMJ,IAAIG,IAAAA;AAC5B,QAAME,cAAcC,kBAAkBF,QAAAA;AACtC,QAAMG,mBAAmBF,YAAYG,IAAIC,gBAAAA;AACzC,SAAOF;AACT;AAEO,IAAMD,oBAAoB,CAACI,WAAAA;AAChC,QAAML,cAA6C,CAAA;AAEnD,QAAMM,gBAAgB,CAACC,WAAsCC,iBAAAA;AAC3D,UAAM,CAACC,cAAcC,aAAAA,IAAiBF,aAAaG,OAAOC,MAAM,GAAA;AAChE,UAAM,CAACC,WAAWC,UAAAA,IAAcP,UAAUI,OAAOC,MAAM,GAAA;AACvD,QAAIE,eAAeJ,eAAe;AAChC,aAAO;IACT;AACA,QAAIG,cAAc,KAAK;AACrB,aAAO;IACT;AACA,QAAIJ,iBAAiB,KAAK;AACxB,aAAO;IACT;AACA,SAAKI,cAAc,OAAOA,cAAc,QAAQJ,iBAAiB,KAAK;AACpE,aAAOD,aAAaO,KAAKC,WAAW,GAAA;IACtC;AACA,WAAO;EACT;AAEA,aAAWC,SAASZ,QAAQ;AAC1B,QAAIL,YAAYkB,WAAW,GAAG;AAC5BlB,kBAAYmB,KAAK;QAACF;OAAM;AACxB;IACF;AACA,UAAMG,iBAAiBpB,YAAYqB,GAAG,EAAC;AACvC,QAAIf,cAAcc,eAAeC,GAAG,EAAC,GAAKJ,KAAAA,GAAQ;AAChDG,qBAAeD,KAAKF,KAAAA;AACpB;IACF;AACAjB,gBAAYmB,KAAK;MAACF;KAAM;EAC1B;AAEA,SAAOjB;AACT;AAEO,IAAMI,mBAAmB,CAACkB,UAAAA;AAC/B,QAAMC,gBAA2C;IAC/CZ,QAAQW,MAAMD,GAAG,CAAA,EAAIV,OAAOC,MAAM,GAAA,EAAK,CAAA;IACvCY,OAAOF,MAAMG,OAAO,CAACC,KAAKT,UAAUS,MAAMT,MAAMO,OAAO,CAAA,IAAKF,MAAMJ;IAClES,OAAOL,MAAMD,GAAG,CAAA,EAAIM;IACpBZ,MAAMO,MAAMD,GAAG,CAAA,EAAIN;IACnBa,OAAON,MAAMD,GAAG,CAAA,EAAIO;IACpBC,KAAKP,MAAMD,GAAG,EAAC,EAAIQ;EACrB;AAEA,aAAWZ,SAASK,MAAMQ,MAAM,CAAA,GAAI;AAClC,QAAIb,MAAMF,KAAKC,WAAW,GAAA,GAAM;AAC9BO,oBAAcR,QAAQE,MAAMF,KAAKgB,QAAQ,OAAO,EAAA;IAClD,OAAO;AACLR,oBAAcR,QAAQ,MAAME,MAAMF;IACpC;EACF;AAEA,SAAOQ;AACT;",
6
+ "names": ["pipeline", "log", "_ner", "getNer", "then", "ner", "info", "extractFullEntities", "text", "entities", "tokenGroups", "createTokenGroups", "combinedEntities", "map", "combineNerTokens", "tokens", "canBeCombined", "prevToken", "currentToken", "currentState", "currentEntity", "entity", "split", "prevState", "prevEntity", "word", "startsWith", "token", "length", "push", "lastTokenGroup", "at", "group", "combinedToken", "score", "reduce", "acc", "index", "start", "end", "slice", "replace"]
7
+ }
@@ -0,0 +1,352 @@
1
+ // src/session/artifact-diff.ts
2
+ import { Context } from "effect";
3
+ var ArtifactDiffResolver = class extends Context.Tag("@dxos/assistant/ArtifactDiffResolver")() {
4
+ };
5
+
6
+ // src/errors.ts
7
+ import { BaseError } from "@dxos/errors";
8
+ var AiAssistantError = class extends BaseError.extend("AI_ASSISTANT_ERROR") {
9
+ };
10
+
11
+ // src/session/session.ts
12
+ import { AiLanguageModel, AiToolkit } from "@effect/ai";
13
+ import { Chunk, Effect as Effect2, Function, Option as Option2, Queue, Stream } from "effect";
14
+ import { AiParser, AiPreprocessor, ToolExecutionService, ToolResolverService, callTool, getToolCalls } from "@dxos/ai";
15
+ import { todo } from "@dxos/debug";
16
+ import { Obj as Obj2 } from "@dxos/echo";
17
+ import { TracingService } from "@dxos/functions";
18
+ import { log as log2 } from "@dxos/log";
19
+ import { DataType as DataType2 } from "@dxos/schema";
20
+ import { isNotFalsy } from "@dxos/util";
21
+
22
+ // src/session/format.ts
23
+ import { Array, Effect, Option, pipe } from "effect";
24
+ import { Template } from "@dxos/blueprints";
25
+ import { Obj } from "@dxos/echo";
26
+ import { ObjectVersion } from "@dxos/echo-db";
27
+ import { DatabaseService } from "@dxos/functions";
28
+ import { log } from "@dxos/log";
29
+ import { DataType } from "@dxos/schema";
30
+ import { trim } from "@dxos/util";
31
+ var __dxlog_file = "/__w/dxos/dxos/packages/core/assistant/src/session/format.ts";
32
+ var formatSystemPrompt = ({ system, blueprints = [], objects = [] }) => Effect.gen(function* () {
33
+ const blueprintDefs = yield* pipe(blueprints, Effect.forEach((blueprint) => Effect.succeed(blueprint.instructions)), Effect.flatMap(Effect.forEach((template) => DatabaseService.load(template.source))), Effect.map(Array.map((template) => trim`
34
+ <blueprint>
35
+ ${Template.process(template.content)}
36
+ </blueprint>
37
+ `)), Effect.map((blueprints2) => blueprints2.length > 0 ? [
38
+ "## Blueprints Definitions",
39
+ ...blueprints2
40
+ ].join("\n\n") : void 0));
41
+ const objectDefs = yield* pipe(objects, Effect.forEach((object) => Effect.succeed(trim`
42
+ <object>
43
+ <dxn>${Obj.getDXN(object)}</dxn>
44
+ <typename>${Obj.getTypename(object)}</typename>
45
+ </object>
46
+ `)), Effect.map((objects2) => objects2.length > 0 ? [
47
+ "## Context Objects",
48
+ ...objects2
49
+ ].join("\n\n") : void 0));
50
+ return yield* pipe(Effect.succeed([
51
+ system,
52
+ blueprintDefs,
53
+ objectDefs
54
+ ].filter((def) => def !== void 0)), Effect.map((parts) => parts.join("\n\n")));
55
+ });
56
+ var formatUserPrompt = ({ prompt, history = [] }) => Effect.gen(function* () {
57
+ const prelude = [];
58
+ const artifactDiffResolver = yield* Effect.serviceOption(ArtifactDiffResolver);
59
+ if (Option.isSome(artifactDiffResolver)) {
60
+ const versions = gatherObjectVersions(history);
61
+ const artifactDiff = yield* Effect.tryPromise({
62
+ try: () => artifactDiffResolver.value.resolve([
63
+ ...versions.entries()
64
+ ].map(([id, version]) => ({
65
+ id,
66
+ lastVersion: version
67
+ }))),
68
+ catch: AiAssistantError.wrap("Artifact diff resolution error")
69
+ });
70
+ log.info("version", {
71
+ artifactDiff,
72
+ versions
73
+ }, {
74
+ F: __dxlog_file,
75
+ L: 90,
76
+ S: this,
77
+ C: (f, a) => f(...a)
78
+ });
79
+ for (const [id, { version }] of [
80
+ ...artifactDiff.entries()
81
+ ]) {
82
+ if (ObjectVersion.equals(version, versions.get(id))) {
83
+ artifactDiff.delete(id);
84
+ continue;
85
+ }
86
+ prelude.push({
87
+ _tag: "anchor",
88
+ objectId: id,
89
+ version
90
+ });
91
+ }
92
+ if (artifactDiff.size > 0) {
93
+ prelude.push(createArtifactUpdateBlock(artifactDiff));
94
+ }
95
+ }
96
+ return Obj.make(DataType.Message, {
97
+ created: (/* @__PURE__ */ new Date()).toISOString(),
98
+ sender: {
99
+ role: "user"
100
+ },
101
+ blocks: [
102
+ ...prelude,
103
+ {
104
+ _tag: "text",
105
+ text: prompt
106
+ }
107
+ ]
108
+ });
109
+ });
110
+ var gatherObjectVersions = (messages) => {
111
+ const artifactIds = /* @__PURE__ */ new Map();
112
+ for (const message of messages) {
113
+ for (const block of message.blocks) {
114
+ if (block._tag === "anchor") {
115
+ artifactIds.set(block.objectId, block.version);
116
+ }
117
+ }
118
+ }
119
+ return artifactIds;
120
+ };
121
+ var createArtifactUpdateBlock = (artifactDiff) => {
122
+ return {
123
+ _tag: "text",
124
+ // TODO(dmaretskyi): Does this need to be a special content-block?
125
+ disposition: "artifact-update",
126
+ text: trim`
127
+ The following artifacts have been updated since the last message:
128
+ ${[
129
+ ...artifactDiff.entries()
130
+ ].map(([id, { diff }]) => `<changed-artifact id="${id}">${diff ? `
131
+ ${diff}` : ""}</changed-artifact>`).join("\n")}
132
+ `
133
+ };
134
+ };
135
+
136
+ // src/session/session.ts
137
+ var __dxlog_file2 = "/__w/dxos/dxos/packages/core/assistant/src/session/session.ts";
138
+ var GenerationObserver = Object.freeze({
139
+ make: ({ onPart = Function.constant(Effect2.void), onBlock = Function.constant(Effect2.void), onMessage = Function.constant(Effect2.void) } = {}) => ({
140
+ onPart,
141
+ onBlock,
142
+ onMessage
143
+ }),
144
+ noop: () => GenerationObserver.make(),
145
+ /**
146
+ * Debug printer to be used in unit-tests and browser devtools.
147
+ */
148
+ fromPrinter: (printer) => GenerationObserver.make({
149
+ onBlock: (block) => Effect2.sync(() => {
150
+ if (block.pending) {
151
+ return;
152
+ }
153
+ printer.printContentBlock(block);
154
+ }),
155
+ onMessage: (message) => Effect2.sync(() => {
156
+ if (message.sender.role === "assistant") {
157
+ return;
158
+ }
159
+ printer.printMessage(message);
160
+ })
161
+ })
162
+ });
163
+ var AiSession = class {
164
+ _options;
165
+ /** Complete messages fired during the session, both from the model and from the user. */
166
+ messageQueue;
167
+ // TODO(dmaretskyi): Remove the queues and convert everything to observer.
168
+ /**
169
+ * Blocks streaming from the model during the session.
170
+ * @deprecated Use `observer.onBlock` instead.
171
+ */
172
+ blockQueue;
173
+ /** Unparsed events from the underlying generation stream. */
174
+ eventQueue;
175
+ /** Prevents concurrent execution of session. */
176
+ _semaphore;
177
+ /** Pending messages (incl. the current user request). */
178
+ _pending;
179
+ /** Prior history from queue. */
180
+ _history;
181
+ constructor(_options = {}) {
182
+ this._options = _options;
183
+ this.messageQueue = Effect2.runSync(Queue.unbounded());
184
+ this.blockQueue = Effect2.runSync(Queue.unbounded());
185
+ this.eventQueue = Effect2.runSync(Queue.unbounded());
186
+ this._semaphore = Effect2.runSync(Effect2.makeSemaphore(1));
187
+ this._pending = [];
188
+ this._history = [];
189
+ this.run = (params) => Effect2.gen(this, function* () {
190
+ const observer = params.observer ?? GenerationObserver.noop();
191
+ const toolkit = yield* createToolkit(params);
192
+ const system = yield* formatSystemPrompt(params);
193
+ console.log(system);
194
+ const promptMessages = yield* formatUserPrompt(params);
195
+ yield* this.messageQueue.offer(promptMessages);
196
+ yield* observer.onMessage(promptMessages);
197
+ yield* TracingService.emitConverationMessage(promptMessages);
198
+ this._history = [
199
+ ...params.history ?? []
200
+ ];
201
+ this._pending = [
202
+ promptMessages
203
+ ];
204
+ do {
205
+ log2.info("request", {
206
+ prompt: promptMessages,
207
+ system: {
208
+ snippet: [
209
+ system.slice(0, 32),
210
+ "...",
211
+ system.slice(-32)
212
+ ].join(""),
213
+ length: system.length
214
+ },
215
+ pending: this._pending.length,
216
+ history: this._history.length,
217
+ objects: params.objects?.length ?? 0,
218
+ blueprints: params.blueprints?.length ?? 0,
219
+ toolkit: Object.values(toolkit.tools).map((tool) => tool.name)
220
+ }, {
221
+ F: __dxlog_file2,
222
+ L: 182,
223
+ S: this,
224
+ C: (f, a) => f(...a)
225
+ });
226
+ const prompt = yield* AiPreprocessor.preprocessAiInput([
227
+ ...this._history,
228
+ ...this._pending
229
+ ]);
230
+ const blocks = yield* AiLanguageModel.streamText({
231
+ prompt,
232
+ system,
233
+ toolkit,
234
+ // TODO(burdon): Despite this flag, the model still calls tools.
235
+ // Flag is only used in generateText (not streamText); patch and submit bug.
236
+ // https://github.com/Effect-TS/effect/blob/main/packages/ai/ai/src/AiLanguageModel.ts#L401
237
+ disableToolCallResolution: true
238
+ }).pipe(AiParser.parseResponse({
239
+ onBlock: (block) => Effect2.all([
240
+ this.blockQueue.offer(Option2.some(block)),
241
+ observer.onBlock(block)
242
+ ], {
243
+ discard: true
244
+ }),
245
+ onPart: (part) => Effect2.all([
246
+ this.eventQueue.offer(part),
247
+ observer.onPart(part)
248
+ ], {
249
+ discard: true
250
+ })
251
+ }), Stream.runCollect, Effect2.map(Chunk.toArray));
252
+ yield* this.blockQueue.offer(Option2.none());
253
+ const response = Obj2.make(DataType2.Message, {
254
+ created: (/* @__PURE__ */ new Date()).toISOString(),
255
+ sender: {
256
+ role: "assistant"
257
+ },
258
+ blocks
259
+ });
260
+ this._pending.push(response);
261
+ yield* this.messageQueue.offer(response);
262
+ yield* observer.onMessage(response);
263
+ yield* TracingService.emitConverationMessage(response);
264
+ const toolCalls = getToolCalls(response);
265
+ if (toolCalls.length === 0) {
266
+ break;
267
+ }
268
+ const toolResults = yield* Effect2.forEach(toolCalls, (toolCall) => callTool(toolkit, toolCall).pipe(Effect2.provide(TracingService.layerSubframe((context) => ({
269
+ ...context,
270
+ parentMessage: response.id,
271
+ toolCallId: toolCall.toolCallId
272
+ })))));
273
+ const toolResultsMessage = Obj2.make(DataType2.Message, {
274
+ created: (/* @__PURE__ */ new Date()).toISOString(),
275
+ sender: {
276
+ role: "user"
277
+ },
278
+ blocks: toolResults
279
+ });
280
+ this._pending.push(toolResultsMessage);
281
+ yield* this.messageQueue.offer(toolResultsMessage);
282
+ yield* observer.onMessage(toolResultsMessage);
283
+ yield* TracingService.emitConverationMessage(toolResultsMessage);
284
+ } while (true);
285
+ yield* Queue.shutdown(this.messageQueue);
286
+ yield* Queue.shutdown(this.blockQueue);
287
+ yield* Queue.shutdown(this.eventQueue);
288
+ log2("done", {
289
+ pending: this._pending.length
290
+ }, {
291
+ F: __dxlog_file2,
292
+ L: 264,
293
+ S: this,
294
+ C: (f, a) => f(...a)
295
+ });
296
+ return this._pending;
297
+ }).pipe(this._semaphore.withPermits(1), Effect2.withSpan("AiSession.run"));
298
+ }
299
+ run;
300
+ // TODO(burdon): Implement.
301
+ async runStructured(_schema, _options) {
302
+ return todo();
303
+ }
304
+ };
305
+ var createToolkit = ({ toolkit, blueprints = [], toolIds = [] }) => Effect2.gen(function* () {
306
+ const blueprintToolkit = yield* ToolResolverService.resolveToolkit([
307
+ ...blueprints.flatMap(({ tools }) => tools),
308
+ ...toolIds
309
+ ]);
310
+ const blueprintToolkitHandler = yield* blueprintToolkit.toContext(ToolExecutionService.handlersFor(blueprintToolkit));
311
+ return yield* AiToolkit.merge(...[
312
+ toolkit,
313
+ blueprintToolkit
314
+ ].filter(isNotFalsy)).pipe(Effect2.provide(blueprintToolkitHandler));
315
+ });
316
+
317
+ // src/session/version-pin.ts
318
+ import { Schema } from "effect";
319
+ import { getVersion } from "@dxos/echo-db";
320
+ import { ObjectId } from "@dxos/echo-schema";
321
+ var ObjectVersionSchema = Schema.Unknown;
322
+ var VersionPinSchema = Schema.Struct({
323
+ // TODO(dmaretskyi): Use Ref when those support encoding.
324
+ objectId: ObjectId,
325
+ // TODO(dmaretskyi): Could be opaque.
326
+ version: ObjectVersionSchema
327
+ });
328
+ var VersionPin = class extends VersionPinSchema {
329
+ static DISPOSITION = "version-pin";
330
+ static fromObject(object) {
331
+ return VersionPin.make({
332
+ objectId: object.id,
333
+ version: getVersion(object)
334
+ });
335
+ }
336
+ static createBlock(pin) {
337
+ return {
338
+ _tag: "json",
339
+ disposition: VersionPin.DISPOSITION,
340
+ data: JSON.stringify(pin)
341
+ };
342
+ }
343
+ };
344
+
345
+ export {
346
+ ArtifactDiffResolver,
347
+ AiAssistantError,
348
+ GenerationObserver,
349
+ AiSession,
350
+ VersionPin
351
+ };
352
+ //# sourceMappingURL=chunk-AVKQ32JL.mjs.map
@@ -0,0 +1,7 @@
1
+ {
2
+ "version": 3,
3
+ "sources": ["../../../src/session/artifact-diff.ts", "../../../src/errors.ts", "../../../src/session/session.ts", "../../../src/session/format.ts", "../../../src/session/version-pin.ts"],
4
+ "sourcesContent": ["//\n// Copyright 2025 DXOS.org\n//\n\nimport { Context } from 'effect';\n\nimport { type ObjectVersion } from '@dxos/echo-db';\nimport { type ObjectId } from '@dxos/echo-schema';\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(burdon): Move.\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: (artifacts: { id: ObjectId; lastVersion: ObjectVersion }[]) => 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 { BaseError } from '@dxos/errors';\n\n/**\n * Generic error for AI agent execution.\n */\nexport class AiAssistantError extends BaseError.extend('AI_ASSISTANT_ERROR') {}\n", "//\n// Copyright 2025 DXOS.org\n//\n\nimport { type AiError, AiLanguageModel, type AiResponse, type AiTool, AiToolkit } from '@effect/ai';\nimport { Chunk, type Context, Effect, Function, Option, Queue, type Schema, Stream } from 'effect';\n\nimport { type ToolId } from '@dxos/ai';\nimport {\n type AiInputPreprocessingError,\n AiParser,\n AiPreprocessor,\n type AiToolNotFoundError,\n type ConsolePrinter,\n ToolExecutionService,\n 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 { type ContentBlock, DataType } from '@dxos/schema';\nimport { isNotFalsy } from '@dxos/util';\n\nimport { type AiAssistantError } from '../errors';\n\nimport { formatSystemPrompt, formatUserPrompt } from './format';\n\nexport type AiSessionOptions = {};\n\n/**\n * Live observer of the generation process.\n */\nexport interface GenerationObserver {\n /**\n * Unparsed content block parts from the model.\n */\n onPart: (part: AiResponse.Part) => Effect.Effect<void>;\n\n /**\n * Parsed content blocks from the model.\n * NOTE: Use block.pending to determine if the block was completed.\n * For each block this will be called 0..n times with a pending block and then once with the final state of the block.\n *\n * Example:\n * 1. { pending: true, text: \"Hello\"}\n * 2. { pending: true, text: \"Hello, I am a\"}\n * 3. { pending: false, text: \"Hello, I am a helpful assistant!\"}\n */\n onBlock: (block: ContentBlock.Any) => Effect.Effect<void>;\n\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 onPart = Function.constant(Effect.void),\n onBlock = Function.constant(Effect.void),\n onMessage = Function.constant(Effect.void),\n }: Partial<GenerationObserver> = {}): GenerationObserver => ({\n onPart,\n onBlock,\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\nexport type SessionRunParams<Tools extends AiTool.Any> = {\n prompt: string;\n system?: string;\n history?: DataType.Message[];\n objects?: Obj.Any[]; // TODO(burdon): Meta only (typename and id -- write to binder).\n blueprints?: Blueprint.Blueprint[];\n toolIds?: ToolId[];\n toolkit?: AiToolkit.AiToolkit<Tools>;\n observer?: GenerationObserver;\n};\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 /** Complete messages fired during the session, both from the model and from the user. */\n public readonly messageQueue = Effect.runSync(Queue.unbounded<DataType.Message>());\n\n // TODO(dmaretskyi): Remove the queues and convert everything to observer.\n\n /**\n * Blocks streaming from the model during the session.\n * @deprecated Use `observer.onBlock` instead.\n */\n public readonly blockQueue = Effect.runSync(Queue.unbounded<Option.Option<ContentBlock.Any>>());\n\n /** Unparsed events from the underlying generation stream. */\n public readonly eventQueue = Effect.runSync(Queue.unbounded<AiResponse.Part>());\n\n /** Prevents concurrent execution of session. */\n private readonly _semaphore = Effect.runSync(Effect.makeSemaphore(1));\n\n /** Pending messages (incl. the current user request). */\n private _pending: DataType.Message[] = [];\n\n /** Prior history from queue. */\n private _history: DataType.Message[] = [];\n\n constructor(private readonly _options: AiSessionOptions = {}) {}\n\n /**\n * Runs the AI model loop interacting with tools and artifacts.\n * @returns The messages generated by the session, including the user's prompt.\n */\n // TODO(dmaretskyi): Toolkit context doesn't get added to the effect type.\n run = <Tools extends AiTool.Any>(\n params: SessionRunParams<Tools>,\n ): Effect.Effect<\n DataType.Message[],\n AiAssistantError | AiInputPreprocessingError | AiToolNotFoundError | AiError.AiError,\n | AiLanguageModel.AiLanguageModel\n | ToolResolverService\n | ToolExecutionService\n | TracingService\n | AiTool.ToHandler<Tools>\n > =>\n Effect.gen(this, function* () {\n const observer = params.observer ?? GenerationObserver.noop();\n\n // Create toolkit.\n const toolkit: AiToolkit.ToHandler<Tools> = yield* createToolkit(params);\n\n // Generate system prompt.\n // TODO(budon): Dynamically resolve template variables.\n const system = yield* formatSystemPrompt(params);\n console.log(system);\n\n // Generate user prompt.\n const promptMessages = yield* formatUserPrompt(params);\n yield* this.messageQueue.offer(promptMessages);\n yield* observer.onMessage(promptMessages);\n yield* TracingService.emitConverationMessage(promptMessages);\n\n this._history = [...(params.history ?? [])];\n this._pending = [promptMessages];\n\n // Potential tool-use loop.\n do {\n log.info('request', {\n prompt: promptMessages,\n system: { snippet: [system.slice(0, 32), '...', system.slice(-32)].join(''), length: system.length },\n pending: this._pending.length,\n history: this._history.length,\n objects: params.objects?.length ?? 0,\n blueprints: params.blueprints?.length ?? 0,\n toolkit: Object.values(toolkit.tools).map((tool: AiTool.Any) => tool.name),\n });\n\n // Generate the prompt and make request.\n const prompt = yield* AiPreprocessor.preprocessAiInput([...this._history, ...this._pending]);\n const blocks = yield* AiLanguageModel.streamText({\n prompt,\n system,\n toolkit,\n // TODO(burdon): Despite this flag, the model still calls tools.\n // Flag is only used in generateText (not streamText); patch and submit bug.\n // https://github.com/Effect-TS/effect/blob/main/packages/ai/ai/src/AiLanguageModel.ts#L401\n disableToolCallResolution: true,\n }).pipe(\n AiParser.parseResponse({\n onBlock: (block) =>\n Effect.all([this.blockQueue.offer(Option.some(block)), observer.onBlock(block)], { discard: true }),\n onPart: (part) => Effect.all([this.eventQueue.offer(part), observer.onPart(part)], { discard: true }),\n }),\n Stream.runCollect,\n Effect.map(Chunk.toArray),\n );\n\n // Signal to stream consumers that message blocks are complete.\n // Allows for coordination between the block and message queues\n // to prevent the streaming blocks from being rendered twice when the message is produced.\n // TODO(wittjosiah): The block queue should probably be drained at this point in the case that there is no consumer.\n yield* this.blockQueue.offer(Option.none());\n\n // Create response message.\n const response = Obj.make(DataType.Message, {\n created: new Date().toISOString(),\n sender: { role: 'assistant' },\n blocks,\n });\n this._pending.push(response);\n yield* this.messageQueue.offer(response);\n yield* observer.onMessage(response);\n yield* TracingService.emitConverationMessage(response);\n\n // Parse response for tool calls.\n const toolCalls = getToolCalls(response);\n if (toolCalls.length === 0) {\n break;\n }\n\n // TODO(burdon): Report errors to user; with proposed actions.\n const toolResults = yield* Effect.forEach(toolCalls, (toolCall) =>\n 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 const toolResultsMessage = Obj.make(DataType.Message, {\n created: new Date().toISOString(),\n sender: { role: 'user' },\n blocks: toolResults,\n });\n\n this._pending.push(toolResultsMessage);\n yield* this.messageQueue.offer(toolResultsMessage);\n yield* observer.onMessage(toolResultsMessage);\n yield* TracingService.emitConverationMessage(toolResultsMessage);\n } while (true);\n\n // Signals to stream consumers that the session has completed and no more messages are coming.\n yield* Queue.shutdown(this.messageQueue);\n yield* Queue.shutdown(this.blockQueue);\n yield* Queue.shutdown(this.eventQueue);\n\n log('done', { pending: this._pending.length });\n return this._pending;\n }).pipe(this._semaphore.withPermits(1), Effect.withSpan('AiSession.run'));\n\n // TODO(burdon): Implement.\n async runStructured<S extends Schema.Schema.AnyNoContext>(\n _schema: S,\n _options: SessionRunParams<AiTool.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\n/**\n * Build a combined toolkit from the blueprint tools and the provided toolkit.\n */\nconst createToolkit = <Tools extends AiTool.Any>({\n toolkit,\n blueprints = [],\n toolIds = [],\n}: Pick<SessionRunParams<Tools>, 'toolkit' | 'blueprints' | 'toolIds'>) =>\n Effect.gen(function* () {\n const blueprintToolkit = yield* ToolResolverService.resolveToolkit([\n ...blueprints.flatMap(({ tools }) => tools),\n ...toolIds,\n ]);\n const blueprintToolkitHandler: Context.Context<AiTool.ToHandler<AiTool.Any>> = yield* blueprintToolkit.toContext(\n ToolExecutionService.handlersFor(blueprintToolkit),\n );\n\n return yield* AiToolkit.merge(...[toolkit, blueprintToolkit].filter(isNotFalsy)).pipe(\n Effect.provide(blueprintToolkitHandler),\n ) as Effect.Effect<AiToolkit.ToHandler<any>, never, AiTool.ToHandler<Tools>>;\n });\n", "//\n// Copyright 2025 DXOS.org\n//\n\nimport { Array, Effect, Option, pipe } from 'effect';\n\nimport { Template } from '@dxos/blueprints';\nimport { Obj } from '@dxos/echo';\nimport { ObjectVersion } from '@dxos/echo-db';\nimport { type ObjectId } from '@dxos/echo-schema';\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 SessionRunParams } 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<SessionRunParams<any>, 'system' | 'blueprints' | 'objects'>) =>\n Effect.gen(function* () {\n // TOOD(burdon): Should process templates.\n const blueprintDefs = yield* pipe(\n blueprints,\n Effect.forEach((blueprint) => Effect.succeed(blueprint.instructions)),\n Effect.flatMap(Effect.forEach((template) => DatabaseService.load(template.source))),\n Effect.map(\n Array.map(\n (template) => trim`\n <blueprint>\n ${Template.process(template.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* 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* 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<SessionRunParams<any>, 'prompt' | 'history'>) =>\n Effect.gen(function* () {\n const prelude: 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('Artifact diff resolution error'),\n });\n\n log.info('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 prelude.push({ _tag: 'anchor', objectId: id, version });\n }\n\n if (artifactDiff.size > 0) {\n prelude.push(createArtifactUpdateBlock(artifactDiff));\n }\n }\n\n return Obj.make(DataType.Message, {\n created: new Date().toISOString(),\n sender: { role: 'user' },\n blocks: [...prelude, { _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 { Schema } from 'effect';\n\nimport { type ObjectVersion } from '@dxos/echo-db';\nimport { getVersion } from '@dxos/echo-db';\nimport { type AnyEchoObject, ObjectId } from '@dxos/echo-schema';\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,SAASA,eAAe;AAejB,IAAMC,uBAAN,cAAmCC,QAAQC,IAAI,sCAAA,EAAA,EAAA;AAGjD;;;AClBL,SAASC,iBAAiB;AAKnB,IAAMC,mBAAN,cAA+BC,UAAUC,OAAO,oBAAA,EAAA;AAAuB;;;ACL9E,SAAuBC,iBAA+CC,iBAAiB;AACvF,SAASC,OAAqBC,UAAAA,SAAQC,UAAUC,UAAAA,SAAQC,OAAoBC,cAAc;AAG1F,SAEEC,UACAC,gBAGAC,sBACAC,qBACAC,UACAC,oBACK;AAEP,SAASC,YAAY;AACrB,SAASC,OAAAA,YAAW;AACpB,SAASC,sBAAsB;AAC/B,SAASC,OAAAA,YAAW;AACpB,SAA4BC,YAAAA,iBAAgB;AAC5C,SAASC,kBAAkB;;;ACrB3B,SAASC,OAAOC,QAAQC,QAAQC,YAAY;AAE5C,SAASC,gBAAgB;AACzB,SAASC,WAAW;AACpB,SAASC,qBAAqB;AAE9B,SAASC,uBAAuB;AAChC,SAASC,WAAW;AACpB,SAA4BC,gBAAgB;AAC5C,SAASC,YAAY;;AAWd,IAAMC,qBAAqB,CAAC,EACjCC,QACAC,aAAa,CAAA,GACbC,UAAU,CAAA,EAAE,MAEZC,OAAOC,IAAI,aAAA;AAET,QAAMC,gBAAgB,OAAOC,KAC3BL,YACAE,OAAOI,QAAQ,CAACC,cAAcL,OAAOM,QAAQD,UAAUE,YAAY,CAAA,GACnEP,OAAOQ,QAAQR,OAAOI,QAAQ,CAACK,aAAaC,gBAAgBC,KAAKF,SAASG,MAAM,CAAA,CAAA,GAChFZ,OAAOa,IACLC,MAAMD,IACJ,CAACJ,aAAaM;;gBAERC,SAASC,QAAQR,SAASS,OAAO,CAAA;;WAEtC,CAAA,GAGLlB,OAAOa,IAAI,CAACf,gBACVA,YAAWqB,SAAS,IAAI;IAAC;OAAgCrB;IAAYsB,KAAK,MAAA,IAAUC,MAAAA,CAAAA;AAIxF,QAAMC,aAAa,OAAOnB,KACxBJ,SACAC,OAAOI,QAAQ,CAACmB,WACdvB,OAAOM,QAAQS;;mBAEJS,IAAIC,OAAOF,MAAAA,CAAAA;wBACNC,IAAIE,YAAYH,MAAAA,CAAAA;;SAE/B,CAAA,GAEHvB,OAAOa,IAAI,CAACd,aAAaA,SAAQoB,SAAS,IAAI;IAAC;OAAyBpB;IAASqB,KAAK,MAAA,IAAUC,MAAAA,CAAAA;AAGlG,SAAO,OAAOlB,KACZH,OAAOM,QAAQ;IAACT;IAAQK;IAAeoB;IAAYK,OAAO,CAACC,QAAuBA,QAAQP,MAAAA,CAAAA,GAC1FrB,OAAOa,IAAI,CAACgB,UAAUA,MAAMT,KAAK,MAAA,CAAA,CAAA;AAErC,CAAA;AAOK,IAAMU,mBAAmB,CAAC,EAAEC,QAAQC,UAAU,CAAA,EAAE,MACrDhC,OAAOC,IAAI,aAAA;AACT,QAAMgC,UAA8B,CAAA;AAGpC,QAAMC,uBAAuB,OAAOlC,OAAOmC,cAAcC,oBAAAA;AACzD,MAAIC,OAAOC,OAAOJ,oBAAAA,GAAuB;AACvC,UAAMK,WAAWC,qBAAqBR,OAAAA;AACtC,UAAMS,eAAe,OAAOzC,OAAO0C,WAAW;MAC5CC,KAAK,MACHT,qBAAqBU,MAAMC,QACzB;WAAIN,SAASO,QAAO;QAAIjC,IAAI,CAAC,CAACkC,IAAIC,OAAAA,OAAc;QAAED;QAAIE,aAAaD;MAAQ,EAAA,CAAA;MAE/EE,OAAOC,iBAAiBC,KAAK,gCAAA;IAC/B,CAAA;AAEAC,QAAIC,KAAK,WAAW;MAAEb;MAAcF;IAAS,GAAA;;;;;;AAC7C,eAAW,CAACQ,IAAI,EAAEC,QAAO,CAAE,KAAK;SAAIP,aAAaK,QAAO;OAAK;AAC3D,UAAIS,cAAcC,OAAOR,SAAST,SAASkB,IAAIV,EAAAA,CAAAA,GAAO;AACpDN,qBAAaiB,OAAOX,EAAAA;AACpB;MACF;AAEAd,cAAQ0B,KAAK;QAAEC,MAAM;QAAUC,UAAUd;QAAIC;MAAQ,CAAA;IACvD;AAEA,QAAIP,aAAaqB,OAAO,GAAG;AACzB7B,cAAQ0B,KAAKI,0BAA0BtB,YAAAA,CAAAA;IACzC;EACF;AAEA,SAAOjB,IAAIwC,KAAKC,SAASC,SAAS;IAChCC,UAAS,oBAAIC,KAAAA,GAAOC,YAAW;IAC/BC,QAAQ;MAAEC,MAAM;IAAO;IACvBC,QAAQ;SAAIvC;MAAS;QAAE2B,MAAM;QAAQa,MAAM1C;MAAO;;EACpD,CAAA;AACF,CAAA;AAEF,IAAMS,uBAAuB,CAACkC,aAAAA;AAC5B,QAAMC,cAAc,oBAAIC,IAAAA;AACxB,aAAWC,WAAWH,UAAU;AAC9B,eAAWI,SAASD,QAAQL,QAAQ;AAClC,UAAIM,MAAMlB,SAAS,UAAU;AAC3Be,oBAAYI,IAAID,MAAMjB,UAAUiB,MAAM9B,OAAO;MAC/C;IACF;EACF;AAEA,SAAO2B;AACT;AAEA,IAAMZ,4BAA4B,CAChCtB,iBAAAA;AAEA,SAAO;IACLmB,MAAM;;IAENoB,aAAa;IACbP,MAAM1D;;QAEF;SAAI0B,aAAaK,QAAO;MACvBjC,IAAI,CAAC,CAACkC,IAAI,EAAEkC,KAAI,CAAE,MAAM,yBAAyBlC,EAAAA,KAAOkC,OAAO;EAAKA,IAAAA,KAAS,EAAA,qBAAuB,EACpG7D,KAAK,IAAA,CAAA;;EAEZ;AACF;;;;AD7EO,IAAM8D,qBAAqBC,OAAOC,OAAO;EAC9CC,MAAM,CAAC,EACLC,SAASC,SAASC,SAASC,QAAOC,IAAI,GACtCC,UAAUJ,SAASC,SAASC,QAAOC,IAAI,GACvCE,YAAYL,SAASC,SAASC,QAAOC,IAAI,EAAC,IACX,CAAC,OAA2B;IAC3DJ;IACAK;IACAC;EACF;EAEAC,MAAM,MAAMX,mBAAmBG,KAAI;;;;EAKnCS,aAAa,CAACC,YACZb,mBAAmBG,KAAK;IACtBM,SAAS,CAACK,UACRP,QAAOQ,KAAK,MAAA;AACV,UAAID,MAAME,SAAS;AACjB;MACF;AACAH,cAAQI,kBAAkBH,KAAAA;IAC5B,CAAA;IACFJ,WAAW,CAACQ,YACVX,QAAOQ,KAAK,MAAA;AACV,UAAIG,QAAQC,OAAOC,SAAS,aAAa;AACvC;MACF;AACAP,cAAQQ,aAAaH,OAAAA;IACvB,CAAA;EACJ,CAAA;AACJ,CAAA;AAuBO,IAAMI,YAAN,MAAMA;;;EAEKC;;;;;;EAQAC;;EAGAC;;EAGCC;;EAGTC;;EAGAC;EAER,YAA6BC,WAA6B,CAAC,GAAG;SAAjCA,WAAAA;SAtBbN,eAAehB,QAAOuB,QAAQC,MAAMC,UAAS,CAAA;SAQ7CR,aAAajB,QAAOuB,QAAQC,MAAMC,UAAS,CAAA;SAG3CP,aAAalB,QAAOuB,QAAQC,MAAMC,UAAS,CAAA;SAG1CN,aAAanB,QAAOuB,QAAQvB,QAAO0B,cAAc,CAAA,CAAA;SAG1DN,WAA+B,CAAA;SAG/BC,WAA+B,CAAA;SASvCM,MAAM,CACJC,WAUA5B,QAAO6B,IAAI,MAAM,aAAA;AACf,YAAMC,WAAWF,OAAOE,YAAYrC,mBAAmBW,KAAI;AAG3D,YAAM2B,UAAsC,OAAOC,cAAcJ,MAAAA;AAIjE,YAAMK,SAAS,OAAOC,mBAAmBN,MAAAA;AACzCO,cAAQC,IAAIH,MAAAA;AAGZ,YAAMI,iBAAiB,OAAOC,iBAAiBV,MAAAA;AAC/C,aAAO,KAAKZ,aAAauB,MAAMF,cAAAA;AAC/B,aAAOP,SAAS3B,UAAUkC,cAAAA;AAC1B,aAAOG,eAAeC,uBAAuBJ,cAAAA;AAE7C,WAAKhB,WAAW;WAAKO,OAAOc,WAAW,CAAA;;AACvC,WAAKtB,WAAW;QAACiB;;AAGjB,SAAG;AACDD,QAAAA,KAAIO,KAAK,WAAW;UAClBC,QAAQP;UACRJ,QAAQ;YAAEY,SAAS;cAACZ,OAAOa,MAAM,GAAG,EAAA;cAAK;cAAOb,OAAOa,MAAM,GAAC;cAAKC,KAAK,EAAA;YAAKC,QAAQf,OAAOe;UAAO;UACnGvC,SAAS,KAAKW,SAAS4B;UACvBN,SAAS,KAAKrB,SAAS2B;UACvBC,SAASrB,OAAOqB,SAASD,UAAU;UACnCE,YAAYtB,OAAOsB,YAAYF,UAAU;UACzCjB,SAASrC,OAAOyD,OAAOpB,QAAQqB,KAAK,EAAEC,IAAI,CAACC,SAAqBA,KAAKC,IAAI;QAC3E,GAAA;;;;;;AAGA,cAAMX,SAAS,OAAOY,eAAeC,kBAAkB;aAAI,KAAKpC;aAAa,KAAKD;SAAS;AAC3F,cAAMsC,SAAS,OAAOC,gBAAgBC,WAAW;UAC/ChB;UACAX;UACAF;;;;UAIA8B,2BAA2B;QAC7B,CAAA,EAAGC,KACDC,SAASC,cAAc;UACrB9D,SAAS,CAACK,UACRP,QAAOiE,IAAI;YAAC,KAAKhD,WAAWsB,MAAM2B,QAAOC,KAAK5D,KAAAA,CAAAA;YAASuB,SAAS5B,QAAQK,KAAAA;aAAS;YAAE6D,SAAS;UAAK,CAAA;UACnGvE,QAAQ,CAACwE,SAASrE,QAAOiE,IAAI;YAAC,KAAK/C,WAAWqB,MAAM8B,IAAAA;YAAOvC,SAASjC,OAAOwE,IAAAA;aAAQ;YAAED,SAAS;UAAK,CAAA;QACrG,CAAA,GACAE,OAAOC,YACPvE,QAAOqD,IAAImB,MAAMC,OAAO,CAAA;AAO1B,eAAO,KAAKxD,WAAWsB,MAAM2B,QAAOQ,KAAI,CAAA;AAGxC,cAAMC,WAAWC,KAAIhF,KAAKiF,UAASC,SAAS;UAC1CC,UAAS,oBAAIC,KAAAA,GAAOC,YAAW;UAC/BrE,QAAQ;YAAEC,MAAM;UAAY;UAC5B6C;QACF,CAAA;AACA,aAAKtC,SAAS8D,KAAKP,QAAAA;AACnB,eAAO,KAAK3D,aAAauB,MAAMoC,QAAAA;AAC/B,eAAO7C,SAAS3B,UAAUwE,QAAAA;AAC1B,eAAOnC,eAAeC,uBAAuBkC,QAAAA;AAG7C,cAAMQ,YAAYC,aAAaT,QAAAA;AAC/B,YAAIQ,UAAUnC,WAAW,GAAG;AAC1B;QACF;AAGA,cAAMqC,cAAc,OAAOrF,QAAOsF,QAAQH,WAAW,CAACI,aACpDC,SAASzD,SAASwD,QAAAA,EAAUzB,KAC1B9D,QAAOyF,QACLjD,eAAekD,cAAc,CAACC,aAAa;UACzC,GAAGA;UACHC,eAAejB,SAASkB;UACxBC,YAAYP,SAASO;QACvB,EAAA,CAAA,CAAA,CAAA;AAIN,cAAMC,qBAAqBnB,KAAIhF,KAAKiF,UAASC,SAAS;UACpDC,UAAS,oBAAIC,KAAAA,GAAOC,YAAW;UAC/BrE,QAAQ;YAAEC,MAAM;UAAO;UACvB6C,QAAQ2B;QACV,CAAA;AAEA,aAAKjE,SAAS8D,KAAKa,kBAAAA;AACnB,eAAO,KAAK/E,aAAauB,MAAMwD,kBAAAA;AAC/B,eAAOjE,SAAS3B,UAAU4F,kBAAAA;AAC1B,eAAOvD,eAAeC,uBAAuBsD,kBAAAA;MAC/C,SAAS;AAGT,aAAOvE,MAAMwE,SAAS,KAAKhF,YAAY;AACvC,aAAOQ,MAAMwE,SAAS,KAAK/E,UAAU;AACrC,aAAOO,MAAMwE,SAAS,KAAK9E,UAAU;AAErCkB,MAAAA,KAAI,QAAQ;QAAE3B,SAAS,KAAKW,SAAS4B;MAAO,GAAA;;;;;;AAC5C,aAAO,KAAK5B;IACd,CAAA,EAAG0C,KAAK,KAAK3C,WAAW8E,YAAY,CAAA,GAAIjG,QAAOkG,SAAS,eAAA,CAAA;EA5HK;EAO/DvE;;EAwHA,MAAMwE,cACJC,SACA9E,UACgC;AAChC,WAAO+E,KAAAA;EAOT;AACF;AAKA,IAAMrE,gBAAgB,CAA2B,EAC/CD,SACAmB,aAAa,CAAA,GACboD,UAAU,CAAA,EAAE,MAEZtG,QAAO6B,IAAI,aAAA;AACT,QAAM0E,mBAAmB,OAAOC,oBAAoBC,eAAe;OAC9DvD,WAAWwD,QAAQ,CAAC,EAAEtD,MAAK,MAAOA,KAAAA;OAClCkD;GACJ;AACD,QAAMK,0BAAyE,OAAOJ,iBAAiBK,UACrGC,qBAAqBC,YAAYP,gBAAAA,CAAAA;AAGnC,SAAO,OAAOQ,UAAUC,MAAK,GAAI;IAACjF;IAASwE;IAAkBU,OAAOC,UAAAA,CAAAA,EAAapD,KAC/E9D,QAAOyF,QAAQkB,uBAAAA,CAAAA;AAEnB,CAAA;;;AE1SF,SAASQ,cAAc;AAGvB,SAASC,kBAAkB;AAC3B,SAA6BC,gBAAgB;AAI7C,IAAMC,sBAAsBC,OAAOC;AAEnC,IAAMC,mBAAmBF,OAAOG,OAAO;;EAErCC,UAAUC;;EAEVC,SAASP;AACX,CAAA;AAQO,IAAMQ,aAIT,cAAcL,iBAAAA;EAChB,OAAgBM,cAAc;EAC9B,OAAOC,WAAWC,QAAmC;AACnD,WAAOH,WAAWI,KAAK;MACrBP,UAAUM,OAAOE;MACjBN,SAASO,WAAWH,MAAAA;IACtB,CAAA;EACF;EAEA,OAAOI,YAAYC,KAAmC;AACpD,WAAO;MACLC,MAAM;MACNC,aAAaV,WAAWC;MACxBU,MAAMC,KAAKC,UAAUL,GAAAA;IACvB;EACF;AACF;",
6
+ "names": ["Context", "ArtifactDiffResolver", "Context", "Tag", "BaseError", "AiAssistantError", "BaseError", "extend", "AiLanguageModel", "AiToolkit", "Chunk", "Effect", "Function", "Option", "Queue", "Stream", "AiParser", "AiPreprocessor", "ToolExecutionService", "ToolResolverService", "callTool", "getToolCalls", "todo", "Obj", "TracingService", "log", "DataType", "isNotFalsy", "Array", "Effect", "Option", "pipe", "Template", "Obj", "ObjectVersion", "DatabaseService", "log", "DataType", "trim", "formatSystemPrompt", "system", "blueprints", "objects", "Effect", "gen", "blueprintDefs", "pipe", "forEach", "blueprint", "succeed", "instructions", "flatMap", "template", "DatabaseService", "load", "source", "map", "Array", "trim", "Template", "process", "content", "length", "join", "undefined", "objectDefs", "object", "Obj", "getDXN", "getTypename", "filter", "def", "parts", "formatUserPrompt", "prompt", "history", "prelude", "artifactDiffResolver", "serviceOption", "ArtifactDiffResolver", "Option", "isSome", "versions", "gatherObjectVersions", "artifactDiff", "tryPromise", "try", "value", "resolve", "entries", "id", "version", "lastVersion", "catch", "AiAssistantError", "wrap", "log", "info", "ObjectVersion", "equals", "get", "delete", "push", "_tag", "objectId", "size", "createArtifactUpdateBlock", "make", "DataType", "Message", "created", "Date", "toISOString", "sender", "role", "blocks", "text", "messages", "artifactIds", "Map", "message", "block", "set", "disposition", "diff", "GenerationObserver", "Object", "freeze", "make", "onPart", "Function", "constant", "Effect", "void", "onBlock", "onMessage", "noop", "fromPrinter", "printer", "block", "sync", "pending", "printContentBlock", "message", "sender", "role", "printMessage", "AiSession", "messageQueue", "blockQueue", "eventQueue", "_semaphore", "_pending", "_history", "_options", "runSync", "Queue", "unbounded", "makeSemaphore", "run", "params", "gen", "observer", "toolkit", "createToolkit", "system", "formatSystemPrompt", "console", "log", "promptMessages", "formatUserPrompt", "offer", "TracingService", "emitConverationMessage", "history", "info", "prompt", "snippet", "slice", "join", "length", "objects", "blueprints", "values", "tools", "map", "tool", "name", "AiPreprocessor", "preprocessAiInput", "blocks", "AiLanguageModel", "streamText", "disableToolCallResolution", "pipe", "AiParser", "parseResponse", "all", "Option", "some", "discard", "part", "Stream", "runCollect", "Chunk", "toArray", "none", "response", "Obj", "DataType", "Message", "created", "Date", "toISOString", "push", "toolCalls", "getToolCalls", "toolResults", "forEach", "toolCall", "callTool", "provide", "layerSubframe", "context", "parentMessage", "id", "toolCallId", "toolResultsMessage", "shutdown", "withPermits", "withSpan", "runStructured", "_schema", "todo", "toolIds", "blueprintToolkit", "ToolResolverService", "resolveToolkit", "flatMap", "blueprintToolkitHandler", "toContext", "ToolExecutionService", "handlersFor", "AiToolkit", "merge", "filter", "isNotFalsy", "Schema", "getVersion", "ObjectId", "ObjectVersionSchema", "Schema", "Unknown", "VersionPinSchema", "Struct", "objectId", "ObjectId", "version", "VersionPin", "DISPOSITION", "fromObject", "object", "make", "id", "getVersion", "createBlock", "pin", "_tag", "disposition", "data", "JSON", "stringify"]
7
+ }