beeai-framework 0.1.22 → 0.1.24

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 (273) hide show
  1. package/dist/adapters/a2a/agents/agent.d.cts +2 -2
  2. package/dist/adapters/a2a/agents/agent.d.ts +2 -2
  3. package/dist/adapters/a2a/agents/events.d.cts +1 -1
  4. package/dist/adapters/a2a/agents/events.d.ts +1 -1
  5. package/dist/adapters/a2a/agents/types.d.cts +2 -2
  6. package/dist/adapters/a2a/agents/types.d.ts +2 -2
  7. package/dist/adapters/a2a/serve/agent_executor.d.cts +3 -3
  8. package/dist/adapters/a2a/serve/agent_executor.d.ts +3 -3
  9. package/dist/adapters/a2a/serve/server.d.cts +2 -2
  10. package/dist/adapters/a2a/serve/server.d.ts +2 -2
  11. package/dist/adapters/acp/agents/agent.d.cts +2 -2
  12. package/dist/adapters/acp/agents/agent.d.ts +2 -2
  13. package/dist/adapters/acp/agents/events.d.cts +1 -1
  14. package/dist/adapters/acp/agents/events.d.ts +1 -1
  15. package/dist/adapters/agentstack/agents/agent.d.cts +2 -2
  16. package/dist/adapters/agentstack/agents/agent.d.ts +2 -2
  17. package/dist/adapters/agentstack/agents/events.d.cts +1 -1
  18. package/dist/adapters/agentstack/agents/events.d.ts +1 -1
  19. package/dist/adapters/amazon-bedrock/backend/chat.d.cts +4 -4
  20. package/dist/adapters/amazon-bedrock/backend/chat.d.ts +4 -4
  21. package/dist/adapters/amazon-bedrock/backend/embedding.d.cts +2 -2
  22. package/dist/adapters/amazon-bedrock/backend/embedding.d.ts +2 -2
  23. package/dist/adapters/anthropic/backend/chat.d.cts +5 -5
  24. package/dist/adapters/anthropic/backend/chat.d.ts +5 -5
  25. package/dist/adapters/anthropic/backend/embedding.d.cts +2 -2
  26. package/dist/adapters/anthropic/backend/embedding.d.ts +2 -2
  27. package/dist/adapters/azure-openai/backend/chat.d.cts +4 -4
  28. package/dist/adapters/azure-openai/backend/chat.d.ts +4 -4
  29. package/dist/adapters/azure-openai/backend/embedding.d.cts +2 -2
  30. package/dist/adapters/azure-openai/backend/embedding.d.ts +2 -2
  31. package/dist/adapters/dummy/backend/chat.d.cts +5 -4
  32. package/dist/adapters/dummy/backend/chat.d.ts +5 -4
  33. package/dist/adapters/dummy/backend/embedding.d.cts +2 -2
  34. package/dist/adapters/dummy/backend/embedding.d.ts +2 -2
  35. package/dist/adapters/google-vertex/backend/chat.d.cts +4 -4
  36. package/dist/adapters/google-vertex/backend/chat.d.ts +4 -4
  37. package/dist/adapters/google-vertex/backend/embedding.d.cts +2 -2
  38. package/dist/adapters/google-vertex/backend/embedding.d.ts +2 -2
  39. package/dist/adapters/groq/backend/chat.d.cts +4 -4
  40. package/dist/adapters/groq/backend/chat.d.ts +4 -4
  41. package/dist/adapters/groq/backend/embedding.d.cts +2 -2
  42. package/dist/adapters/groq/backend/embedding.d.ts +2 -2
  43. package/dist/adapters/langchain/backend/chat.d.cts +5 -4
  44. package/dist/adapters/langchain/backend/chat.d.ts +5 -4
  45. package/dist/adapters/langchain/backend/embedding.d.cts +2 -2
  46. package/dist/adapters/langchain/backend/embedding.d.ts +2 -2
  47. package/dist/adapters/langchain/tools.d.cts +1 -1
  48. package/dist/adapters/langchain/tools.d.ts +1 -1
  49. package/dist/adapters/ollama/backend/chat.cjs +1 -2
  50. package/dist/adapters/ollama/backend/chat.cjs.map +1 -1
  51. package/dist/adapters/ollama/backend/chat.d.cts +4 -5
  52. package/dist/adapters/ollama/backend/chat.d.ts +4 -5
  53. package/dist/adapters/ollama/backend/chat.js +1 -2
  54. package/dist/adapters/ollama/backend/chat.js.map +1 -1
  55. package/dist/adapters/ollama/backend/embedding.d.cts +2 -2
  56. package/dist/adapters/ollama/backend/embedding.d.ts +2 -2
  57. package/dist/adapters/openai/backend/chat.d.cts +4 -4
  58. package/dist/adapters/openai/backend/chat.d.ts +4 -4
  59. package/dist/adapters/openai/backend/embedding.d.cts +2 -2
  60. package/dist/adapters/openai/backend/embedding.d.ts +2 -2
  61. package/dist/adapters/vercel/backend/chat.cjs +77 -17
  62. package/dist/adapters/vercel/backend/chat.cjs.map +1 -1
  63. package/dist/adapters/vercel/backend/chat.d.cts +5 -4
  64. package/dist/adapters/vercel/backend/chat.d.ts +5 -4
  65. package/dist/adapters/vercel/backend/chat.js +77 -17
  66. package/dist/adapters/vercel/backend/chat.js.map +1 -1
  67. package/dist/adapters/vercel/backend/embedding.d.cts +2 -2
  68. package/dist/adapters/vercel/backend/embedding.d.ts +2 -2
  69. package/dist/adapters/vercel/backend/utils.cjs +17 -0
  70. package/dist/adapters/vercel/backend/utils.cjs.map +1 -1
  71. package/dist/adapters/vercel/backend/utils.d.cts +5 -4
  72. package/dist/adapters/vercel/backend/utils.d.ts +5 -4
  73. package/dist/adapters/vercel/backend/utils.js +17 -1
  74. package/dist/adapters/vercel/backend/utils.js.map +1 -1
  75. package/dist/adapters/watsonx/backend/chat.cjs +40 -13
  76. package/dist/adapters/watsonx/backend/chat.cjs.map +1 -1
  77. package/dist/adapters/watsonx/backend/chat.d.cts +7 -6
  78. package/dist/adapters/watsonx/backend/chat.d.ts +7 -6
  79. package/dist/adapters/watsonx/backend/chat.js +40 -13
  80. package/dist/adapters/watsonx/backend/chat.js.map +1 -1
  81. package/dist/adapters/watsonx/backend/embedding.d.cts +2 -2
  82. package/dist/adapters/watsonx/backend/embedding.d.ts +2 -2
  83. package/dist/adapters/xai/backend/chat.d.cts +4 -4
  84. package/dist/adapters/xai/backend/chat.d.ts +4 -4
  85. package/dist/{agent-BMfyig7X.d.cts → agent-Cy4QtsN8.d.cts} +4 -4
  86. package/dist/{agent-Dxw8AGWb.d.ts → agent-DHLEPXEA.d.ts} +4 -4
  87. package/dist/agents/base.d.cts +2 -2
  88. package/dist/agents/base.d.ts +2 -2
  89. package/dist/agents/experimental/replan/agent.d.cts +3 -3
  90. package/dist/agents/experimental/replan/agent.d.ts +3 -3
  91. package/dist/agents/experimental/replan/prompts.d.cts +1 -1
  92. package/dist/agents/experimental/replan/prompts.d.ts +1 -1
  93. package/dist/agents/experimental/streamlit/agent.d.cts +3 -3
  94. package/dist/agents/experimental/streamlit/agent.d.ts +3 -3
  95. package/dist/agents/react/agent.d.cts +4 -4
  96. package/dist/agents/react/agent.d.ts +4 -4
  97. package/dist/agents/react/prompts.d.cts +1 -1
  98. package/dist/agents/react/prompts.d.ts +1 -1
  99. package/dist/agents/react/runners/base.d.cts +4 -4
  100. package/dist/agents/react/runners/base.d.ts +4 -4
  101. package/dist/agents/react/runners/deep-think/prompts.d.cts +1 -1
  102. package/dist/agents/react/runners/deep-think/prompts.d.ts +1 -1
  103. package/dist/agents/react/runners/deep-think/runner.d.cts +5 -5
  104. package/dist/agents/react/runners/deep-think/runner.d.ts +5 -5
  105. package/dist/agents/react/runners/default/runner.d.cts +6 -6
  106. package/dist/agents/react/runners/default/runner.d.ts +6 -6
  107. package/dist/agents/react/runners/granite/prompts.d.cts +1 -1
  108. package/dist/agents/react/runners/granite/prompts.d.ts +1 -1
  109. package/dist/agents/react/runners/granite/runner.d.cts +5 -5
  110. package/dist/agents/react/runners/granite/runner.d.ts +5 -5
  111. package/dist/agents/react/types.d.cts +2 -2
  112. package/dist/agents/react/types.d.ts +2 -2
  113. package/dist/agents/requirement/agent.cjs +1 -1
  114. package/dist/agents/requirement/agent.cjs.map +1 -1
  115. package/dist/agents/requirement/agent.d.cts +4 -4
  116. package/dist/agents/requirement/agent.d.ts +4 -4
  117. package/dist/agents/requirement/agent.js +1 -1
  118. package/dist/agents/requirement/agent.js.map +1 -1
  119. package/dist/agents/requirement/prompts.d.cts +2 -2
  120. package/dist/agents/requirement/prompts.d.ts +2 -2
  121. package/dist/agents/requirement/requirements/conditional.d.cts +4 -4
  122. package/dist/agents/requirement/requirements/conditional.d.ts +4 -4
  123. package/dist/agents/requirement/requirements/requirement.d.cts +4 -4
  124. package/dist/agents/requirement/requirements/requirement.d.ts +4 -4
  125. package/dist/agents/requirement/requirements/utils.d.cts +1 -1
  126. package/dist/agents/requirement/requirements/utils.d.ts +1 -1
  127. package/dist/agents/requirement/runner.cjs +31 -15
  128. package/dist/agents/requirement/runner.cjs.map +1 -1
  129. package/dist/agents/requirement/runner.d.cts +13 -15
  130. package/dist/agents/requirement/runner.d.ts +13 -15
  131. package/dist/agents/requirement/runner.js +31 -15
  132. package/dist/agents/requirement/runner.js.map +1 -1
  133. package/dist/agents/requirement/types.d.cts +4 -4
  134. package/dist/agents/requirement/types.d.ts +4 -4
  135. package/dist/agents/requirement/utils/llm.d.cts +4 -4
  136. package/dist/agents/requirement/utils/llm.d.ts +4 -4
  137. package/dist/agents/requirement/utils/tool.cjs +2 -1
  138. package/dist/agents/requirement/utils/tool.cjs.map +1 -1
  139. package/dist/agents/requirement/utils/tool.d.cts +4 -4
  140. package/dist/agents/requirement/utils/tool.d.ts +4 -4
  141. package/dist/agents/requirement/utils/tool.js +2 -1
  142. package/dist/agents/requirement/utils/tool.js.map +1 -1
  143. package/dist/agents/toolCalling/agent.cjs +4 -1
  144. package/dist/agents/toolCalling/agent.cjs.map +1 -1
  145. package/dist/agents/toolCalling/agent.d.cts +6 -3
  146. package/dist/agents/toolCalling/agent.d.ts +6 -3
  147. package/dist/agents/toolCalling/agent.js +5 -2
  148. package/dist/agents/toolCalling/agent.js.map +1 -1
  149. package/dist/agents/toolCalling/types.d.cts +1 -1
  150. package/dist/agents/toolCalling/types.d.ts +1 -1
  151. package/dist/agents/types.d.cts +2 -2
  152. package/dist/agents/types.d.ts +2 -2
  153. package/dist/backend/backend.d.cts +2 -2
  154. package/dist/backend/backend.d.ts +2 -2
  155. package/dist/backend/chat.cjs +125 -5
  156. package/dist/backend/chat.cjs.map +1 -1
  157. package/dist/backend/chat.d.cts +3 -3
  158. package/dist/backend/chat.d.ts +3 -3
  159. package/dist/backend/chat.js +127 -7
  160. package/dist/backend/chat.js.map +1 -1
  161. package/dist/backend/core.d.cts +3 -3
  162. package/dist/backend/core.d.ts +3 -3
  163. package/dist/backend/embedding.d.cts +2 -2
  164. package/dist/backend/embedding.d.ts +2 -2
  165. package/dist/backend/message.cjs +20 -13
  166. package/dist/backend/message.cjs.map +1 -1
  167. package/dist/backend/message.d.cts +7 -3
  168. package/dist/backend/message.d.ts +7 -3
  169. package/dist/backend/message.js +20 -13
  170. package/dist/backend/message.js.map +1 -1
  171. package/dist/backend/utils.d.cts +2 -2
  172. package/dist/backend/utils.d.ts +2 -2
  173. package/dist/{base-BurRTBA7.d.ts → base-CyFx7BRL.d.ts} +2 -2
  174. package/dist/{base-v-PYIq5G.d.cts → base-XfWeXQ-S.d.cts} +2 -2
  175. package/dist/{chat-DB_cf15S.d.cts → chat-CC0s8z23.d.cts} +10 -9
  176. package/dist/{chat-BbCYMAdE.d.ts → chat-D8SVtw5Q.d.ts} +10 -9
  177. package/dist/context.d.cts +1 -1
  178. package/dist/context.d.ts +1 -1
  179. package/dist/emitter/emitter.cjs +2 -1
  180. package/dist/emitter/emitter.cjs.map +1 -1
  181. package/dist/emitter/emitter.d.cts +1 -1
  182. package/dist/emitter/emitter.d.ts +1 -1
  183. package/dist/emitter/emitter.js +2 -1
  184. package/dist/emitter/emitter.js.map +1 -1
  185. package/dist/emitter/types.d.cts +1 -1
  186. package/dist/emitter/types.d.ts +1 -1
  187. package/dist/{emitter-CN7I4uSV.d.ts → emitter-36-9MnvA.d.ts} +2 -2
  188. package/dist/{emitter-CGKd_-kK.d.cts → emitter-jN55XZZq.d.cts} +2 -2
  189. package/dist/index.d.cts +3 -3
  190. package/dist/index.d.ts +3 -3
  191. package/dist/internals/fetcher.d.cts +1 -1
  192. package/dist/internals/fetcher.d.ts +1 -1
  193. package/dist/internals/helpers/object.cjs +18 -0
  194. package/dist/internals/helpers/object.cjs.map +1 -1
  195. package/dist/internals/helpers/object.d.cts +2 -1
  196. package/dist/internals/helpers/object.d.ts +2 -1
  197. package/dist/internals/helpers/object.js +18 -1
  198. package/dist/internals/helpers/object.js.map +1 -1
  199. package/dist/memory/summarizeMemory.d.cts +2 -2
  200. package/dist/memory/summarizeMemory.d.ts +2 -2
  201. package/dist/middleware/streamToolCall.cjs +138 -0
  202. package/dist/middleware/streamToolCall.cjs.map +1 -0
  203. package/dist/middleware/streamToolCall.d.cts +96 -0
  204. package/dist/middleware/streamToolCall.d.ts +96 -0
  205. package/dist/middleware/streamToolCall.js +136 -0
  206. package/dist/middleware/streamToolCall.js.map +1 -0
  207. package/dist/middleware/trajectory.cjs +14 -5
  208. package/dist/middleware/trajectory.cjs.map +1 -1
  209. package/dist/middleware/trajectory.d.cts +1 -1
  210. package/dist/middleware/trajectory.d.ts +1 -1
  211. package/dist/middleware/trajectory.js +13 -4
  212. package/dist/middleware/trajectory.js.map +1 -1
  213. package/dist/parsers/linePrefix.d.cts +1 -1
  214. package/dist/parsers/linePrefix.d.ts +1 -1
  215. package/dist/tools/arxiv.d.cts +1 -1
  216. package/dist/tools/arxiv.d.ts +1 -1
  217. package/dist/tools/base.d.cts +1 -1
  218. package/dist/tools/base.d.ts +1 -1
  219. package/dist/tools/calculator.d.cts +1 -1
  220. package/dist/tools/calculator.d.ts +1 -1
  221. package/dist/tools/custom.d.cts +2 -2
  222. package/dist/tools/custom.d.ts +2 -2
  223. package/dist/tools/database/elasticsearch.d.cts +1 -1
  224. package/dist/tools/database/elasticsearch.d.ts +1 -1
  225. package/dist/tools/database/milvus.d.cts +1 -1
  226. package/dist/tools/database/milvus.d.ts +1 -1
  227. package/dist/tools/database/qdrant.d.cts +1 -1
  228. package/dist/tools/database/qdrant.d.ts +1 -1
  229. package/dist/tools/database/sql.d.cts +1 -1
  230. package/dist/tools/database/sql.d.ts +1 -1
  231. package/dist/tools/handoff.cjs +68 -0
  232. package/dist/tools/handoff.cjs.map +1 -0
  233. package/dist/tools/handoff.d.cts +47 -0
  234. package/dist/tools/handoff.d.ts +47 -0
  235. package/dist/tools/handoff.js +66 -0
  236. package/dist/tools/handoff.js.map +1 -0
  237. package/dist/tools/llm.d.cts +2 -2
  238. package/dist/tools/llm.d.ts +2 -2
  239. package/dist/tools/mcp.d.cts +1 -1
  240. package/dist/tools/mcp.d.ts +1 -1
  241. package/dist/tools/openapi.d.cts +1 -1
  242. package/dist/tools/openapi.d.ts +1 -1
  243. package/dist/tools/python/output.d.cts +1 -1
  244. package/dist/tools/python/output.d.ts +1 -1
  245. package/dist/tools/python/python.d.cts +2 -2
  246. package/dist/tools/python/python.d.ts +2 -2
  247. package/dist/tools/search/base.d.cts +1 -1
  248. package/dist/tools/search/base.d.ts +1 -1
  249. package/dist/tools/search/duckDuckGoSearch.d.cts +1 -1
  250. package/dist/tools/search/duckDuckGoSearch.d.ts +1 -1
  251. package/dist/tools/search/googleSearch.d.cts +1 -1
  252. package/dist/tools/search/googleSearch.d.ts +1 -1
  253. package/dist/tools/search/searXNGSearch.d.cts +1 -1
  254. package/dist/tools/search/searXNGSearch.d.ts +1 -1
  255. package/dist/tools/search/wikipedia.d.cts +1 -1
  256. package/dist/tools/search/wikipedia.d.ts +1 -1
  257. package/dist/tools/similarity.d.cts +1 -1
  258. package/dist/tools/similarity.d.ts +1 -1
  259. package/dist/tools/think.d.cts +1 -1
  260. package/dist/tools/think.d.ts +1 -1
  261. package/dist/tools/weather/openMeteo.d.cts +1 -1
  262. package/dist/tools/weather/openMeteo.d.ts +1 -1
  263. package/dist/tools/web/webCrawler.d.cts +1 -1
  264. package/dist/tools/web/webCrawler.d.ts +1 -1
  265. package/dist/{types-Cx6f43du.d.cts → types-D9pn62yF.d.cts} +3 -3
  266. package/dist/{types-BXkkbABd.d.ts → types-Dyd4e1G6.d.ts} +3 -3
  267. package/dist/version.cjs +1 -1
  268. package/dist/version.js +1 -1
  269. package/dist/workflows/agent.d.cts +40 -40
  270. package/dist/workflows/agent.d.ts +40 -40
  271. package/dist/workflows/workflow.d.cts +1 -1
  272. package/dist/workflows/workflow.d.ts +1 -1
  273. package/package.json +8 -8
@@ -0,0 +1,136 @@
1
+ import { Middleware } from '../context.js';
2
+ import { Emitter } from '../emitter/emitter.js';
3
+ import { ChatModelOutput, ChatModel } from '../backend/chat.js';
4
+ import { parseBrokenJson } from '../internals/helpers/schema.js';
5
+ import { isString, isPlainObject } from 'remeda';
6
+ import { hasProp } from '../internals/helpers/object.js';
7
+
8
+ var __defProp = Object.defineProperty;
9
+ var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
10
+ class StreamToolCallMiddleware extends Middleware {
11
+ static {
12
+ __name(this, "StreamToolCallMiddleware");
13
+ }
14
+ target;
15
+ key;
16
+ matchNested;
17
+ forceStreaming;
18
+ cleanups = [];
19
+ output = new ChatModelOutput([]);
20
+ buffer = "";
21
+ delta = "";
22
+ emitter;
23
+ constructor(options) {
24
+ super();
25
+ this.target = options.target;
26
+ this.key = options.key;
27
+ this.matchNested = options.matchNested ?? false;
28
+ this.forceStreaming = options.forceStreaming ?? false;
29
+ this.emitter = Emitter.root.child({
30
+ namespace: [
31
+ "middleware",
32
+ "streamToolCall"
33
+ ]
34
+ });
35
+ }
36
+ reset() {
37
+ this.output = new ChatModelOutput([]);
38
+ this.buffer = "";
39
+ this.delta = "";
40
+ }
41
+ isEmpty() {
42
+ return this.buffer.length === 0;
43
+ }
44
+ bind(ctx) {
45
+ this.reset();
46
+ this.cleanups.push(ctx.instance.emitter.match((meta) => meta.creator instanceof ChatModel && meta.name === "start", this.handleStart.bind(this), this.createEmitterOptions()));
47
+ this.cleanups.push(ctx.instance.emitter.match((meta) => meta.creator instanceof ChatModel && meta.name === "newToken", this.handleNewToken.bind(this), this.createEmitterOptions()));
48
+ this.cleanups.push(ctx.instance.emitter.match((meta) => meta.creator instanceof ChatModel && meta.name === "success", this.handleSuccess.bind(this), this.createEmitterOptions()));
49
+ }
50
+ unbind() {
51
+ while (this.cleanups.length > 0) {
52
+ const fn = this.cleanups.shift();
53
+ fn();
54
+ }
55
+ }
56
+ createEmitterOptions() {
57
+ return {
58
+ matchNested: this.matchNested,
59
+ isBlocking: true
60
+ };
61
+ }
62
+ async process(toolName, args) {
63
+ if (toolName !== this.target.name) {
64
+ return;
65
+ }
66
+ const parsedArgs = isString(args) ? parseBrokenJson(args, {
67
+ pair: [
68
+ "{",
69
+ "}"
70
+ ]
71
+ }) : args;
72
+ const outputStructured = await this.target.parse(parsedArgs).catch(() => null);
73
+ if (!outputStructured) {
74
+ return;
75
+ }
76
+ let output = "";
77
+ if (hasProp(outputStructured, this.key)) {
78
+ output = outputStructured[this.key] || "";
79
+ if (!isString(output)) {
80
+ output = JSON.stringify(output);
81
+ }
82
+ this.delta = output.slice(this.buffer.length);
83
+ this.buffer = output;
84
+ if (!this.delta) {
85
+ return;
86
+ }
87
+ }
88
+ await this.emitter.emit("update", {
89
+ outputStructured,
90
+ delta: this.delta,
91
+ output
92
+ });
93
+ }
94
+ async handleStart(data, _meta) {
95
+ if (this.forceStreaming) {
96
+ data.input.stream = true;
97
+ data.input.streamPartialToolCalls = true;
98
+ }
99
+ }
100
+ async handleSuccess(data, _meta) {
101
+ if (this.output.messages.length === 0) {
102
+ await this.handleNewToken({
103
+ value: data.value,
104
+ callbacks: {
105
+ abort: /* @__PURE__ */ __name(() => {
106
+ }, "abort")
107
+ }
108
+ }, _meta);
109
+ }
110
+ }
111
+ async handleNewToken(data, _meta) {
112
+ await this.output.merge(data.value);
113
+ const toolCalls = this.output.getToolCalls();
114
+ if (toolCalls.length > 0) {
115
+ for (const toolCall2 of toolCalls) {
116
+ await this.process(toolCall2.toolName, toolCall2.input);
117
+ }
118
+ return;
119
+ }
120
+ const textContent = this.output.getTextContent();
121
+ const toolCall = parseBrokenJson(textContent, {
122
+ pair: [
123
+ "{",
124
+ "}"
125
+ ]
126
+ });
127
+ if (!toolCall || !isPlainObject(toolCall)) {
128
+ return;
129
+ }
130
+ await this.process(toolCall.name, toolCall.parameters);
131
+ }
132
+ }
133
+
134
+ export { StreamToolCallMiddleware };
135
+ //# sourceMappingURL=streamToolCall.js.map
136
+ //# sourceMappingURL=streamToolCall.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["../../src/middleware/streamToolCall.ts"],"names":["StreamToolCallMiddleware","Middleware","target","key","matchNested","forceStreaming","cleanups","output","ChatModelOutput","buffer","delta","emitter","options","Emitter","root","child","namespace","reset","isEmpty","length","bind","ctx","push","instance","match","meta","creator","ChatModel","name","handleStart","createEmitterOptions","handleNewToken","handleSuccess","unbind","fn","shift","isBlocking","process","toolName","args","parsedArgs","isString","parseBrokenJson","pair","outputStructured","parse","catch","hasProp","JSON","stringify","slice","emit","data","_meta","input","stream","streamPartialToolCalls","messages","value","callbacks","abort","merge","toolCalls","getToolCalls","toolCall","textContent","getTextContent","isPlainObject","parameters"],"mappings":";;;;;;;;;AAuEO,MAAMA,iCAA0CC,UAAAA,CAAAA;EAvEvD;;;AAwEmBC,EAAAA,MAAAA;AACAC,EAAAA,GAAAA;AACAC,EAAAA,WAAAA;AACAC,EAAAA,cAAAA;AACAC,EAAAA,QAAAA,GAA2B,EAAA;EAEpCC,MAAAA,GAAS,IAAIC,eAAAA,CAAgB,EAAE,CAAA;EAC/BC,MAAAA,GAAS,EAAA;EACTC,KAAAA,GAAQ,EAAA;AAEAC,EAAAA,OAAAA;AAEhB,EAAA,WAAA,CAAYC,OAAAA,EAA0C;AACpD,IAAA,KAAA,EAAK;AAEL,IAAA,IAAA,CAAKV,SAASU,OAAAA,CAAQV,MAAAA;AACtB,IAAA,IAAA,CAAKC,MAAMS,OAAAA,CAAQT,GAAAA;AACnB,IAAA,IAAA,CAAKC,WAAAA,GAAcQ,QAAQR,WAAAA,IAAe,KAAA;AAC1C,IAAA,IAAA,CAAKC,cAAAA,GAAiBO,QAAQP,cAAAA,IAAkB,KAAA;AAEhD,IAAA,IAAA,CAAKM,OAAAA,GAAUE,OAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAA4C;MACtEC,SAAAA,EAAW;AAAC,QAAA,YAAA;AAAc,QAAA;;KAC5B,CAAA;AACF;EAEAC,KAAAA,GAAQ;AACN,IAAA,IAAA,CAAKV,MAAAA,GAAS,IAAIC,eAAAA,CAAgB,EAAE,CAAA;AACpC,IAAA,IAAA,CAAKC,MAAAA,GAAS,EAAA;AACd,IAAA,IAAA,CAAKC,KAAAA,GAAQ,EAAA;AACf;EAEAQ,OAAAA,GAAU;AACR,IAAA,OAAO,IAAA,CAAKT,OAAOU,MAAAA,KAAW,CAAA;AAChC;AAEAC,EAAAA,IAAAA,CAAKC,GAAAA,EAAoC;AACvC,IAAA,IAAA,CAAKJ,KAAAA,EAAK;AAGV,IAAA,IAAA,CAAKX,QAAAA,CAASgB,KACZD,GAAAA,CAAIE,QAAAA,CAASZ,QAAQa,KAAAA,CACnB,CAACC,IAAAA,KAASA,IAAAA,CAAKC,OAAAA,YAAmBC,SAAAA,IAAaF,KAAKG,IAAAA,KAAS,OAAA,EAC7D,KAAKC,WAAAA,CAAYT,IAAAA,CAAK,IAAI,CAAA,EAC1B,IAAA,CAAKU,oBAAAA,EAAoB,CAAA,CAAA;AAK7B,IAAA,IAAA,CAAKxB,QAAAA,CAASgB,KACZD,GAAAA,CAAIE,QAAAA,CAASZ,QAAQa,KAAAA,CACnB,CAACC,IAAAA,KAASA,IAAAA,CAAKC,OAAAA,YAAmBC,SAAAA,IAAaF,KAAKG,IAAAA,KAAS,UAAA,EAC7D,KAAKG,cAAAA,CAAeX,IAAAA,CAAK,IAAI,CAAA,EAC7B,IAAA,CAAKU,oBAAAA,EAAoB,CAAA,CAAA;AAK7B,IAAA,IAAA,CAAKxB,QAAAA,CAASgB,KACZD,GAAAA,CAAIE,QAAAA,CAASZ,QAAQa,KAAAA,CACnB,CAACC,IAAAA,KAASA,IAAAA,CAAKC,OAAAA,YAAmBC,SAAAA,IAAaF,KAAKG,IAAAA,KAAS,SAAA,EAC7D,KAAKI,aAAAA,CAAcZ,IAAAA,CAAK,IAAI,CAAA,EAC5B,IAAA,CAAKU,oBAAAA,EAAoB,CAAA,CAAA;AAG/B;EAEAG,MAAAA,GAAe;AAEb,IAAA,OAAO,IAAA,CAAK3B,QAAAA,CAASa,MAAAA,GAAS,CAAA,EAAG;AAC/B,MAAA,MAAMe,EAAAA,GAAK,IAAA,CAAK5B,QAAAA,CAAS6B,KAAAA,EAAK;AAC9BD,MAAAA,EAAAA,EAAAA;AACF;AACF;EAEUJ,oBAAAA,GAAuC;AAC/C,IAAA,OAAO;AACL1B,MAAAA,WAAAA,EAAa,IAAA,CAAKA,WAAAA;MAClBgC,UAAAA,EAAY;AACd,KAAA;AACF;EAEA,MAAcC,OAAAA,CAAQC,UAAkBC,IAAAA,EAA0B;AAChE,IAAA,IAAID,QAAAA,KAAa,IAAA,CAAKpC,MAAAA,CAAO0B,IAAAA,EAAM;AACjC,MAAA;AACF;AAEA,IAAA,MAAMY,UAAAA,GAAaC,QAAAA,CAASF,IAAAA,CAAAA,GAAQG,gBAAgBH,IAAAA,EAAM;MAAEI,IAAAA,EAAM;AAAC,QAAA,GAAA;AAAK,QAAA;;AAAK,KAAA,CAAA,GAAKJ,IAAAA;AAClF,IAAA,MAAMK,gBAAAA,GAAoB,MAAM,IAAA,CAAK1C,MAAAA,CAAO2C,MAAML,UAAAA,CAAAA,CAAYM,KAAAA,CAAM,MAAM,IAAA,CAAA;AAC1E,IAAA,IAAI,CAACF,gBAAAA,EAAkB;AACrB,MAAA;AACF;AAEA,IAAA,IAAIrC,MAAAA,GAAS,EAAA;AACb,IAAA,IAAIwC,OAAAA,CAAQH,gBAAAA,EAAkB,IAAA,CAAKzC,GAAG,CAAA,EAAc;AAClDI,MAAAA,MAAAA,GAAUqC,gBAAAA,CAAyB,IAAA,CAAKzC,GAAG,CAAA,IAAK,EAAA;AAChD,MAAA,IAAI,CAACsC,QAAAA,CAASlC,MAAAA,CAAAA,EAAS;AACrBA,QAAAA,MAAAA,GAASyC,IAAAA,CAAKC,UAAU1C,MAAAA,CAAAA;AAC1B;AACA,MAAA,IAAA,CAAKG,KAAAA,GAAQH,MAAAA,CAAO2C,KAAAA,CAAM,IAAA,CAAKzC,OAAOU,MAAM,CAAA;AAC5C,MAAA,IAAA,CAAKV,MAAAA,GAASF,MAAAA;AAEd,MAAA,IAAI,CAAC,KAAKG,KAAAA,EAAO;AACf,QAAA;AACF;AACF;AAEA,IAAA,MAAM,IAAA,CAAKC,OAAAA,CAAQwC,IAAAA,CAAK,QAAA,EAAU;AAChCP,MAAAA,gBAAAA;AACAlC,MAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AACZH,MAAAA;KACF,CAAA;AACF;EAEA,MAAcsB,WAAAA,CACZuB,MACAC,KAAAA,EACe;AACf,IAAA,IAAI,KAAKhD,cAAAA,EAAgB;AACvB+C,MAAAA,IAAAA,CAAKE,MAAMC,MAAAA,GAAS,IAAA;AACpBH,MAAAA,IAAAA,CAAKE,MAAME,sBAAAA,GAAyB,IAAA;AACtC;AACF;EAEA,MAAcxB,aAAAA,CACZoB,MACAC,KAAAA,EACe;AAEf,IAAA,IAAI,IAAA,CAAK9C,MAAAA,CAAOkD,QAAAA,CAAStC,MAAAA,KAAW,CAAA,EAAG;AACrC,MAAA,MAAM,KAAKY,cAAAA,CAAe;AAAE2B,QAAAA,KAAAA,EAAON,IAAAA,CAAKM,KAAAA;QAAOC,SAAAA,EAAW;AAAEC,UAAAA,KAAAA,kBAAO,MAAA,CAAA,MAAA;WAAO,EAAP,OAAA;AAAS;AAAE,OAAA,EAAGP,KAAAA,CAAAA;AACnF;AACF;EAEA,MAActB,cAAAA,CACZqB,MACAC,KAAAA,EACe;AACf,IAAA,MAAM,IAAA,CAAK9C,MAAAA,CAAOsD,KAAAA,CAAMT,IAAAA,CAAKM,KAAK,CAAA;AAElC,IAAA,MAAMI,SAAAA,GAAY,IAAA,CAAKvD,MAAAA,CAAOwD,YAAAA,EAAY;AAC1C,IAAA,IAAID,SAAAA,CAAU3C,SAAS,CAAA,EAAG;AACxB,MAAA,KAAA,MAAW6C,aAAYF,SAAAA,EAAW;AAChC,QAAA,MAAM,IAAA,CAAKzB,OAAAA,CAAQ2B,SAAAA,CAAS1B,QAAAA,EAAU0B,UAASV,KAAK,CAAA;AACtD;AACA,MAAA;AACF;AAGA,IAAA,MAAMW,WAAAA,GAAc,IAAA,CAAK1D,MAAAA,CAAO2D,cAAAA,EAAc;AAC9C,IAAA,MAAMF,QAAAA,GAAWtB,gBAAgBuB,WAAAA,EAAa;MAAEtB,IAAAA,EAAM;AAAC,QAAA,GAAA;AAAK,QAAA;;KAAK,CAAA;AAEjE,IAAA,IAAI,CAACqB,QAAAA,IAAY,CAACG,aAAAA,CAAcH,QAAAA,CAAAA,EAAW;AACzC,MAAA;AACF;AAEA,IAAA,MAAM,IAAA,CAAK3B,OAAAA,CAAQ2B,QAAAA,CAASpC,IAAAA,EAAgBoC,SAASI,UAAU,CAAA;AACjE;AACF","file":"streamToolCall.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { Middleware, RunContext, RunInstance } from \"@/context.js\";\nimport { Callback, Emitter, EventMeta } from \"@/emitter/emitter.js\";\nimport { ChatModel, ChatModelEvents } from \"@/backend/chat.js\";\nimport { ChatModelOutput } from \"@/backend/chat.js\";\nimport { Tool } from \"@/tools/base.js\";\nimport { parseBrokenJson } from \"@/internals/helpers/schema.js\";\nimport { isPlainObject, isString } from \"remeda\";\nimport { hasProp } from \"@/internals/helpers/object.js\";\nimport { EmitterOptions, InferCallbackValue } from \"@/emitter/types.js\";\n\n/**\n * Event emitted when the middleware detects an update to the target tool's arguments\n */\nexport interface StreamToolCallMiddlewareUpdateEvent<T = any> {\n /** The validated and structured tool input */\n outputStructured: T | null;\n /** The current value of the target field */\n output: string;\n /** The incremental change since the last update */\n delta: string;\n}\n\n/**\n * Callbacks for StreamToolCallMiddleware events\n */\nexport interface StreamToolCallMiddlewareCallbacks<T = any> {\n update: Callback<StreamToolCallMiddlewareUpdateEvent<T>>;\n}\n\n/**\n * Options for configuring StreamToolCallMiddleware\n */\nexport interface StreamToolCallMiddlewareOptions {\n /** The tool to monitor for streaming updates */\n target: Tool<any>;\n /** The field name in the tool's input schema to stream */\n key: string;\n /** Whether to apply middleware to nested run contexts */\n matchNested?: boolean;\n /** Whether to force streaming on the ChatModel */\n forceStreaming?: boolean;\n}\n\n/**\n * Middleware for handling streaming tool calls in a ChatModel.\n *\n * This middleware observes and listens to ChatModel stream updates and parses\n * the tool calls on demand so that they can be consumed as soon as possible.\n *\n * @example\n * ```typescript\n * const middleware = new StreamToolCallMiddleware({\n * target: thinkTool,\n * key: \"thoughts\",\n * matchNested: false,\n * forceStreaming: true,\n * });\n *\n * middleware.emitter.on(\"update\", (event) => {\n * console.log(\"Delta:\", event.delta);\n * console.log(\"Structured:\", event.outputStructured);\n * });\n *\n * await llm.run(messages, { tools: [thinkTool] }).middleware(middleware);\n * ```\n */\nexport class StreamToolCallMiddleware<T = any> extends Middleware<RunInstance> {\n private readonly target: Tool<any>;\n private readonly key: string;\n private readonly matchNested: boolean;\n private readonly forceStreaming: boolean;\n private readonly cleanups: (() => void)[] = [];\n\n private output = new ChatModelOutput([]);\n private buffer = \"\";\n private delta = \"\";\n\n public readonly emitter: Emitter<StreamToolCallMiddlewareCallbacks<T>>;\n\n constructor(options: StreamToolCallMiddlewareOptions) {\n super();\n\n this.target = options.target;\n this.key = options.key;\n this.matchNested = options.matchNested ?? false;\n this.forceStreaming = options.forceStreaming ?? false;\n\n this.emitter = Emitter.root.child<StreamToolCallMiddlewareCallbacks<T>>({\n namespace: [\"middleware\", \"streamToolCall\"],\n });\n }\n\n reset() {\n this.output = new ChatModelOutput([]);\n this.buffer = \"\";\n this.delta = \"\";\n }\n\n isEmpty() {\n return this.buffer.length === 0;\n }\n\n bind(ctx: RunContext<RunInstance>): void {\n this.reset();\n\n // Listen to ChatModel start event\n this.cleanups.push(\n ctx.instance.emitter.match(\n (meta) => meta.creator instanceof ChatModel && meta.name === \"start\",\n this.handleStart.bind(this),\n this.createEmitterOptions(),\n ),\n );\n\n // Listen to ChatModel newToken event\n this.cleanups.push(\n ctx.instance.emitter.match(\n (meta) => meta.creator instanceof ChatModel && meta.name === \"newToken\",\n this.handleNewToken.bind(this),\n this.createEmitterOptions(),\n ),\n );\n\n // Listen to ChatModel success event\n this.cleanups.push(\n ctx.instance.emitter.match(\n (meta) => meta.creator instanceof ChatModel && meta.name === \"success\",\n this.handleSuccess.bind(this),\n this.createEmitterOptions(),\n ),\n );\n }\n\n unbind(): void {\n // Clean up previous bindings\n while (this.cleanups.length > 0) {\n const fn = this.cleanups.shift()!;\n fn();\n }\n }\n\n protected createEmitterOptions(): EmitterOptions {\n return {\n matchNested: this.matchNested,\n isBlocking: true,\n };\n }\n\n private async process(toolName: string, args: any): Promise<void> {\n if (toolName !== this.target.name) {\n return;\n }\n\n const parsedArgs = isString(args) ? parseBrokenJson(args, { pair: [\"{\", \"}\"] }) : args;\n const outputStructured = (await this.target.parse(parsedArgs).catch(() => null)) as T;\n if (!outputStructured) {\n return;\n }\n\n let output = \"\";\n if (hasProp(outputStructured, this.key as keyof T)) {\n output = (outputStructured as any)[this.key] || \"\";\n if (!isString(output)) {\n output = JSON.stringify(output);\n }\n this.delta = output.slice(this.buffer.length);\n this.buffer = output;\n\n if (!this.delta) {\n return;\n }\n }\n\n await this.emitter.emit(\"update\", {\n outputStructured,\n delta: this.delta,\n output,\n });\n }\n\n private async handleStart(\n data: InferCallbackValue<ChatModelEvents[\"start\"]>,\n _meta: EventMeta,\n ): Promise<void> {\n if (this.forceStreaming) {\n data.input.stream = true;\n data.input.streamPartialToolCalls = true;\n }\n }\n\n private async handleSuccess(\n data: InferCallbackValue<ChatModelEvents[\"success\"]>,\n _meta: EventMeta,\n ): Promise<void> {\n // If we haven't received any tokens yet, process the final output\n if (this.output.messages.length === 0) {\n await this.handleNewToken({ value: data.value, callbacks: { abort: () => {} } }, _meta);\n }\n }\n\n private async handleNewToken(\n data: InferCallbackValue<ChatModelEvents[\"newToken\"]>,\n _meta: EventMeta,\n ): Promise<void> {\n await this.output.merge(data.value);\n\n const toolCalls = this.output.getToolCalls();\n if (toolCalls.length > 0) {\n for (const toolCall of toolCalls) {\n await this.process(toolCall.toolName, toolCall.input);\n }\n return;\n }\n\n // Try to parse text content as a tool call\n const textContent = this.output.getTextContent();\n const toolCall = parseBrokenJson(textContent, { pair: [\"{\", \"}\"] });\n\n if (!toolCall || !isPlainObject(toolCall)) {\n return;\n }\n\n await this.process(toolCall.name as string, toolCall.parameters as any);\n }\n}\n"]}
@@ -9,8 +9,9 @@ var remeda = require('remeda');
9
9
  var chat_cjs = require('../backend/chat.cjs');
10
10
  var base_cjs$1 = require('../tools/base.cjs');
11
11
  var requirement_cjs = require('../agents/requirement/requirements/requirement.cjs');
12
- var utils = require('sequelize/lib/utils');
13
12
  var serializer_cjs = require('../serializer/serializer.cjs');
13
+ var guards_cjs = require('../internals/helpers/guards.cjs');
14
+ var object_cjs = require('../internals/helpers/object.cjs');
14
15
 
15
16
  var __defProp = Object.defineProperty;
16
17
  var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
@@ -112,12 +113,12 @@ class GlobalTrajectoryMiddleware extends context_cjs.Middleware {
112
113
  const eventName = remeda.capitalize(meta.name);
113
114
  await this[`onInternal${eventName}`].call(this, data, meta);
114
115
  }, "handleTopLevelEvent");
115
- this.cleanups.push(emitter.match((event) => [
116
+ this.cleanups.push(emitter.match(matchInternalEvent([
116
117
  "start",
117
118
  "success",
118
119
  "error",
119
120
  "finish"
120
- ].includes(event.name), handleTopLevelEvent, {
121
+ ]), handleTopLevelEvent, {
121
122
  matchNested: false,
122
123
  isBlocking: true,
123
124
  priority: this.emitterPriority
@@ -132,7 +133,9 @@ class GlobalTrajectoryMiddleware extends context_cjs.Middleware {
132
133
  this.bindEmitter(meta.creator.emitter);
133
134
  }
134
135
  }, "handleNestedEvent");
135
- this.cleanups.push(emitter.match((event) => event.name === "start", handleNestedEvent, {
136
+ this.cleanups.push(emitter.match(matchInternalEvent([
137
+ "start"
138
+ ]), handleNestedEvent, {
136
139
  matchNested: true,
137
140
  isBlocking: true,
138
141
  priority: this.emitterPriority
@@ -239,7 +242,7 @@ class GlobalTrajectoryMiddleware extends context_cjs.Middleware {
239
242
  };
240
243
  }
241
244
  async formatPayload(value) {
242
- if (utils.isPrimitive(value)) {
245
+ if (guards_cjs.isPrimitive(value)) {
243
246
  return String(value);
244
247
  }
245
248
  if (value instanceof errors_cjs.FrameworkError) {
@@ -316,6 +319,12 @@ class GlobalTrajectoryMiddleware extends context_cjs.Middleware {
316
319
  });
317
320
  }
318
321
  }
322
+ function matchInternalEvent(names) {
323
+ return (meta) => names.includes(meta.name) && object_cjs.getProp(meta.context, [
324
+ "internal"
325
+ ], false);
326
+ }
327
+ __name(matchInternalEvent, "matchInternalEvent");
319
328
 
320
329
  exports.GlobalTrajectoryMiddleware = GlobalTrajectoryMiddleware;
321
330
  //# sourceMappingURL=trajectory.cjs.map
@@ -1 +1 @@
1
- {"version":3,"sources":["../../src/middleware/trajectory.ts"],"names":["GlobalTrajectoryMiddleware","Middleware","enabled","included","excluded","cleanups","target","ctx","pretty","traceLevel","Map","prefixByType","matchNested","emitterPriority","formatter","emitter","options","createTarget","BaseAgent","ChatModel","Tool","Requirement","Array","from","entries","x","prefix","className","instanceName","eventName","Emitter","root","child","namespace","bind","length","pop","clear","set","runId","relative","absolute","bindEmitter","input","undefined","msg","console","log","Logger","debug","push","match","_","event","logTraceId","handleTopLevelEvent","data","meta","creator","RunContext","trace","isAllowed","capitalize","name","call","includes","isBlocking","priority","handleNestedEvent","has","parentRunId","allowed","parentTrace","get","instance","some","extractName","targetName","type","typePrefix","formatPrefix","indent","getTraceLevel","indentParent","indentDiff","repeat","formatPayload","value","isPrimitive","String","FrameworkError","explain","serialized","Serializer","serialize","JSON","stringify","deserialize","excludedKeys","Set","key","onInternalStart","payload","message","emit","level","origin","onInternalSuccess","onInternalError","onInternalFinish","error","output"],"mappings":";;;;;;;;;;;;;;;;AA0FO,MAAMA,mCAAgEC,sBAAAA,CAAAA;EA1F7E;;;AA2FYC,EAAAA,OAAAA;AACAC,EAAAA,QAAAA;AACAC,EAAAA,QAAAA;AACAC,EAAAA,QAAAA,GAA2B,EAAA;AAC3BC,EAAAA,MAAAA;EACAC,GAAAA,GAA4B,IAAA;AAC5BC,EAAAA,MAAAA;AACAC,EAAAA,UAAAA,uBAAiBC,GAAAA,EAAAA;AACjBC,EAAAA,YAAAA;AACAC,EAAAA,WAAAA;AACAC,EAAAA,eAAAA;AACAC,EAAAA,SAAAA;AACMC,EAAAA,OAAAA;EAEhB,WAAA,CAAYC,OAAAA,GAA6C,EAAC,EAAG;AAC3D,IAAA,KAAA,EAAK;AAEL,IAAA,IAAA,CAAKd,OAAAA,GAAUc,QAAQd,OAAAA,IAAW,IAAA;AAClC,IAAA,IAAA,CAAKC,QAAAA,GAAWa,OAAAA,CAAQb,QAAAA,IAAY,EAAA;AACpC,IAAA,IAAA,CAAKC,QAAAA,GAAWY,OAAAA,CAAQZ,QAAAA,IAAY,EAAA;AACpC,IAAA,IAAA,CAAKE,MAAAA,GAAS,IAAA,CAAKW,YAAAA,CAAaD,OAAAA,CAAQV,MAAM,CAAA;AAC9C,IAAA,IAAA,CAAKE,MAAAA,GAASQ,QAAQR,MAAAA,IAAU,KAAA;AAChC,IAAA,IAAA,CAAKI,WAAAA,GAAcI,QAAQJ,WAAAA,IAAe,IAAA;AAC1C,IAAA,IAAA,CAAKC,eAAAA,GAAkBG,QAAQH,eAAAA,IAAmB,EAAA;AAGlD,IAAA,IAAA,CAAKF,YAAAA,GAAe,IAAID,GAAAA,CAAI;AAC1B,MAAA;AAACQ,QAAAA,kBAAAA;AAAW,QAAA;;AACZ,MAAA;AAACC,QAAAA,kBAAAA;AAAW,QAAA;;AACZ,MAAA;AAACC,QAAAA,eAAAA;AAAM,QAAA;;AACP,MAAA;AAACC,QAAAA,2BAAAA;AAAa,QAAA;;SACTL,OAAAA,CAAQL,YAAAA,GAAeW,MAAMC,IAAAA,CAAKP,OAAAA,CAAQL,aAAaa,OAAAA,EAAO,IAAM;AAC1E,KAAA,CAAA;AAED,IAAA,IAAA,CAAKV,YACHE,OAAAA,CAAQF,SAAAA,KACP,CAACW,CAAAA,KAAM,CAAA,EAAGA,EAAEC,MAAM,CAAA,EAAGD,CAAAA,CAAEE,SAAS,IAAIF,CAAAA,CAAEG,YAAAA,IAAgBH,EAAEE,SAAS,CAAA,EAAA,EAAKF,EAAEI,SAAS,CAAA,CAAA,CAAA,CAAA;AAEpF,IAAA,IAAA,CAAKd,OAAAA,GAAUe,mBAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAA2C;MACrEC,SAAAA,EAAW;AAAC,QAAA,YAAA;AAAc,QAAA;;KAC5B,CAAA;AACF;;;;AAKAC,EAAAA,IAAAA,CAAK3B,GAAAA,EAA0B;AAE7B,IAAA,OAAO,IAAA,CAAKF,QAAAA,CAAS8B,MAAAA,GAAS,CAAA,EAAG;AAC/B,MAAA,IAAA,CAAK9B,QAAAA,CAAS+B,KAAG,EAAA;AACnB;AAEA,IAAA,IAAA,CAAK3B,WAAW4B,KAAAA,EAAK;AACrB,IAAA,IAAA,CAAK5B,UAAAA,CAAW6B,GAAAA,CAAI/B,GAAAA,CAAIgC,KAAAA,EAAO;MAAEC,QAAAA,EAAU,CAAA;MAAGC,QAAAA,EAAU;KAAE,CAAA;AAC1D,IAAA,IAAA,CAAKlC,GAAAA,GAAMA,GAAAA;AAEX,IAAA,IAAA,CAAKmC,WAAAA,CAAYnC,IAAIQ,OAAO,CAAA;AAC9B;AAEQE,EAAAA,YAAAA,CAAa0B,KAAAA,EAAsC;AACzD,IAAA,IAAIA,KAAAA,KAAU,IAAA,IAAQA,KAAAA,KAAUC,MAAAA,EAAW;AAEzC,MAAA,OAAO,CAACC,GAAAA,KAAQC,OAAAA,CAAQC,GAAAA,CAAIF,GAAAA,CAAAA;AAC9B,KAAA,MAAA,IAAWF,iBAAiBK,iBAAAA,EAAQ;AAClC,MAAA,OAAO,CAACH,GAAAA,KAAQF,KAAAA,CAAMM,KAAAA,CAAMJ,GAAAA,CAAAA;KAC9B,MAAO;AACL,MAAA,OAAOF,KAAAA;AACT;AACF;AAEQD,EAAAA,WAAAA,CAAY3B,OAAAA,EAA6B;AAE/C,IAAA,IAAA,CAAKV,QAAAA,CAAS6C,IAAAA,CACZnC,OAAAA,CAAQoC,KAAAA,CAAM,KAAA,EAAO,CAACC,CAAAA,EAAGC,KAAAA,KAAU,IAAA,CAAKC,UAAAA,CAAWD,KAAAA,CAAAA,EAAQ;MACzDzC,WAAAA,EAAa;AACf,KAAA,CAAA,CAAA;AAIF,IAAA,MAAM2C,mBAAAA,mBAAsB,MAAA,CAAA,OAAOC,IAAAA,EAAWC,IAAAA,KAAAA;AAC5C,MAAA,IAAI,EAAEA,IAAAA,CAAKC,OAAAA,YAAmBC,sBAAAA,CAAAA,EAAa;AACzC,QAAA;AACF;AACA,MAAA,IAAI,CAACF,KAAKG,KAAAA,EAAO;AACf,QAAA;AACF;AAEA,MAAA,IAAA,CAAKN,WAAWG,IAAAA,CAAAA;AAChB,MAAA,IAAI,CAAC,IAAA,CAAKI,SAAAA,CAAUJ,IAAAA,CAAAA,EAAO;AACzB,QAAA;AACF;AACA,MAAA,IAAI,CAAC,KAAKvD,OAAAA,EAAS;AACjB,QAAA;AACF;AAEA,MAAA,MAAM2B,SAAAA,GAAYiC,iBAAAA,CAAWL,IAAAA,CAAKM,IAAI,CAAA;AACtC,MAAA,MAAM,IAAA,CAAK,aAAalC,SAAAA,CAAAA,CAAW,EAAWmC,IAAAA,CAAK,IAAA,EAAMR,MAAMC,IAAAA,CAAAA;KACjE,EAlB4B,qBAAA,CAAA;AAoB5B,IAAA,IAAA,CAAKpD,QAAAA,CAAS6C,IAAAA,CACZnC,OAAAA,CAAQoC,KAAAA,CACN,CAACE,KAAAA,KAAU;AAAC,MAAA,OAAA;AAAS,MAAA,SAAA;AAAW,MAAA,OAAA;AAAS,MAAA;MAAUY,QAAAA,CAASZ,KAAAA,CAAMU,IAAI,CAAA,EACtER,mBAAAA,EACA;MACE3C,WAAAA,EAAa,KAAA;MACbsD,UAAAA,EAAY,IAAA;AACZC,MAAAA,QAAAA,EAAU,IAAA,CAAKtD;AACjB,KAAA,CAAA,CAAA;AAKJ,IAAA,IAAI,KAAKD,WAAAA,EAAa;AACpB,MAAA,MAAMwD,iBAAAA,mBAAoB,MAAA,CAAA,OAAOZ,IAAAA,EAAWC,IAAAA,KAAAA;AAC1C,QAAA,IAAI,EAAEA,IAAAA,CAAKC,OAAAA,YAAmBC,sBAAAA,CAAAA,EAAa;AACzC,UAAA;AACF;AACA,QAAA,IAAIF,IAAAA,CAAKC,OAAAA,CAAQ3C,OAAAA,KAAYA,OAAAA,EAAS;AACpC,UAAA,MAAMwC,mBAAAA,CAAoBC,MAAMC,IAAAA,CAAAA;AAChC,UAAA,IAAA,CAAKf,WAAAA,CAAYe,IAAAA,CAAKC,OAAAA,CAAQ3C,OAAO,CAAA;AACvC;OACF,EAR0B,mBAAA,CAAA;AAU1B,MAAA,IAAA,CAAKV,QAAAA,CAAS6C,KACZnC,OAAAA,CAAQoC,KAAAA,CAAM,CAACE,KAAAA,KAAUA,KAAAA,CAAMU,IAAAA,KAAS,OAAA,EAASK,iBAAAA,EAAmB;QAClExD,WAAAA,EAAa,IAAA;QACbsD,UAAAA,EAAY,IAAA;AACZC,QAAAA,QAAAA,EAAU,IAAA,CAAKtD;AACjB,OAAA,CAAA,CAAA;AAEJ;AACF;AAEQyC,EAAAA,UAAAA,CAAWG,IAAAA,EAAuB;AACxC,IAAA,IAAI,CAACA,IAAAA,CAAKG,KAAAA,EAAOrB,KAAAA,EAAO;AACtB,MAAA;AACF;AACA,IAAA,IAAI,KAAK9B,UAAAA,CAAW4D,GAAAA,CAAIZ,IAAAA,CAAKG,KAAAA,CAAMrB,KAAK,CAAA,EAAG;AACzC,MAAA;AACF;AACA,IAAA,IAAIkB,IAAAA,CAAKG,KAAAA,CAAMU,WAAAA,KAAgBb,IAAAA,CAAKG,MAAMrB,KAAAA,EAAO;AAC/C,MAAA;AACF;AAEA,IAAA,IAAIkB,IAAAA,CAAKG,MAAMU,WAAAA,EAAa;AAC1B,MAAA,MAAMC,OAAAA,GAAU,IAAA,CAAKV,SAAAA,CAAUJ,IAAAA,CAAAA;AAC/B,MAAA,MAAMe,cAAc,IAAA,CAAK/D,UAAAA,CAAWgE,IAAIhB,IAAAA,CAAKG,KAAAA,CAAMU,WAAW,CAAA,IAAK;QACjE9B,QAAAA,EAAU,CAAA;QACVC,QAAAA,EAAU;AACZ,OAAA;AACA,MAAA,IAAA,CAAKhC,UAAAA,CAAW6B,GAAAA,CAAImB,IAAAA,CAAKG,KAAAA,CAAMrB,KAAAA,EAAO;QACpCC,QAAAA,EAAUgC,WAAAA,CAAYhC,QAAAA,IAAY+B,OAAAA,GAAU,CAAA,GAAI,CAAA,CAAA;AAChD9B,QAAAA,QAAAA,EAAU+B,YAAY/B,QAAAA,GAAW;OACnC,CAAA;AACF;AACF;AAEQoB,EAAAA,SAAAA,CAAUJ,IAAAA,EAA0B;AAC1C,IAAA,IAAInD,SAAcmD,IAAAA,CAAKC,OAAAA;AACvB,IAAA,IAAIpD,kBAAkBqD,sBAAAA,EAAY;AAChCrD,MAAAA,MAAAA,GAASA,MAAAA,CAAOoE,QAAAA;AAClB;AAEA,IAAA,KAAA,MAAWtE,QAAAA,IAAY,KAAKA,QAAAA,EAAU;AACpC,MAAA,IAAIE,kBAAkBF,QAAAA,EAAU;AAC9B,QAAA,OAAO,KAAA;AACT;AACF;AAEA,IAAA,IAAI,IAAA,CAAKD,QAAAA,CAASgC,MAAAA,KAAW,CAAA,EAAG;AAC9B,MAAA,OAAO,IAAA;AACT;AAEA,IAAA,OAAO,KAAKhC,QAAAA,CAASwE,IAAAA,CAAK,CAACxE,QAAAA,KAAaG,kBAAkBH,QAAAA,CAAAA;AAC5D;AAEQyE,EAAAA,WAAAA,CAAYnB,IAAAA,EAAyB;AAC3C,IAAA,IAAInD,SAAcmD,IAAAA,CAAKC,OAAAA;AACvB,IAAA,IAAIpD,kBAAkBqD,sBAAAA,EAAY;AAChCrD,MAAAA,MAAAA,GAASA,MAAAA,CAAOoE,QAAAA;AAClB;AAEA,IAAA,MAAM/C,SAAAA,GAAYrB,OAAO,WAAA,CAAYyD,IAAAA;AACrC,IAAA,IAAIc,UAAAA,GAA4B,IAAA;AAEhC,IAAA,IAAIvE,MAAAA,YAAkBY,kBAAAA,IAAaZ,MAAAA,CAAOmD,IAAAA,EAAMM,IAAAA,EAAM;AACpDc,MAAAA,UAAAA,GAAavE,OAAOmD,IAAAA,CAAKM,IAAAA;AAC3B,KAAA,MAAA,IAAW,MAAA,IAAUzD,MAAAA,IAAU,OAAOA,MAAAA,CAAOyD,SAAS,QAAA,EAAU;AAC9Dc,MAAAA,UAAAA,GAAavE,MAAAA,CAAOyD,IAAAA;AACtB;AAEA,IAAA,IAAIrC,MAAAA,GAAS,EAAA;AACb,IAAA,KAAA,MAAW,CAACoD,IAAAA,EAAMC,UAAAA,KAAe,IAAA,CAAKpE,YAAAA,CAAaa,SAAO,EAAI;AAC5D,MAAA,IAAIlB,kBAAkBwE,IAAAA,EAAM;AAC1BpD,QAAAA,MAAAA,GAASqD,UAAAA;AACT,QAAA;AACF;AACF;AAEA,IAAA,MAAMpC,KAAAA,GAAkD;AACtDjB,MAAAA,MAAAA;AACAC,MAAAA,SAAAA;MACAC,YAAAA,EAAciD,UAAAA;AACdhD,MAAAA,SAAAA,EAAW4B,IAAAA,CAAKM;AAClB,KAAA;AAEA,IAAA,OAAO,IAAA,CAAKjD,UAAU6B,KAAAA,CAAAA;AACxB;AAEQqC,EAAAA,YAAAA,CAAavB,IAAAA,EAAyB;AAC5C,IAAA,IAAI,CAACA,KAAKG,KAAAA,EAAO;AACf,MAAA,OAAO,EAAA;AACT;AAEA,IAAA,MAAMqB,MAAAA,GAAS,IAAA,CAAKC,aAAAA,CAAczB,IAAAA,EAAM,MAAA,CAAA,CAAQjB,QAAAA;AAChD,IAAA,MAAM2C,YAAAA,GAAe,IAAA,CAAKD,aAAAA,CAAczB,IAAAA,EAAM,QAAA,CAAA,CAAUjB,QAAAA;AACxD,IAAA,MAAM4C,aAAaH,MAAAA,GAASE,YAAAA;AAE5B,IAAA,IAAIzD,MAAAA,GAAS,EAAA;AACbA,IAAAA,MAAAA,IAAU,IAAA,CAAK2D,MAAAA,CAAOF,YAAAA,GAAe,CAAA,CAAA;AAErC,IAAA,IAAI1B,IAAAA,CAAKM,IAAAA,KAAS,OAAA,IAAWkB,MAAAA,EAAQ;AACnCvD,MAAAA,MAAAA,IAAU,GAAA;AACZ;AAEAA,IAAAA,MAAAA,IAAU,IAAA,CAAK2D,OAAOD,UAAAA,CAAAA;AAEtB,IAAA,IAAI3B,IAAAA,CAAKM,IAAAA,KAAS,OAAA,IAAWrC,MAAAA,IAAUuD,MAAAA,EAAQ;AAC7CvD,MAAAA,MAAAA,IAAU,GAAA;AACZ;AAEA,IAAA,IAAIA,MAAAA,EAAQ;AACVA,MAAAA,MAAAA,GAAS,GAAGA,MAAAA,CAAAA,CAAAA,CAAAA;AACd;AAEA,IAAA,MAAMqC,IAAAA,GAAO,IAAA,CAAKa,WAAAA,CAAYnB,IAAAA,CAAAA;AAC9B,IAAA,OAAO,CAAA,EAAG/B,MAAAA,CAAAA,EAASqC,IAAAA,CAAAA,EAAAA,CAAAA;AACrB;EAEQmB,aAAAA,CAAczB,IAAAA,EAAiBqB,OAA0B,MAAA,EAAoB;AACnF,IAAA,IAAI,CAACrB,KAAKG,KAAAA,EAAO;AACf,MAAA,OAAO;QAAEpB,QAAAA,EAAU,CAAA;QAAGC,QAAAA,EAAU;AAAE,OAAA;AACpC;AAEA,IAAA,MAAMF,KAAAA,GAAQuC,SAAS,QAAA,GAAWrB,IAAAA,CAAKG,MAAMU,WAAAA,IAAe,EAAA,GAAKb,KAAKG,KAAAA,CAAMrB,KAAAA;AAC5E,IAAA,OAAO,IAAA,CAAK9B,UAAAA,CAAWgE,GAAAA,CAAIlC,KAAAA,CAAAA,IAAU;MAAEC,QAAAA,EAAU,CAAA;MAAGC,QAAAA,EAAU;AAAE,KAAA;AAClE;AAEA,EAAA,MAAc6C,cAAcC,KAAAA,EAA6B;AACvD,IAAA,IAAIC,iBAAAA,CAAYD,KAAAA,CAAAA,EAAQ;AACtB,MAAA,OAAOE,OAAOF,KAAAA,CAAAA;AAChB;AAEA,IAAA,IAAIA,iBAAiBG,yBAAAA,EAAgB;AACnC,MAAA,OAAOH,MAAMI,OAAAA,EAAO;AACtB;AAEA,IAAA,MAAMC,UAAAA,GAAa,MAAMC,yBAAAA,CAAWC,SAAAA,CAAUP,KAAAA,CAAAA;AAC9C,IAAA,OAAOQ,IAAAA,CAAKC,SAAAA,CACV,MAAMH,yBAAAA,CAAWI,WAAAA,CAAYL,YAAY,EAAA,EAAI,IAAA,CAAA,kBAC5C,CAAA,MAAA;AACC,MAAA,MAAMM,YAAAA,uBAAmBC,GAAAA,CAAI;AAAC,QAAA,SAAA;AAAW,QAAA,UAAA;AAAY,QAAA,SAAA;AAAW,QAAA;AAAY,OAAA,CAAA;AAC5E,MAAA,OAAO,CAACC,KAAKb,MAAAA,KAAAA;AACX,QAAA,IAAIW,YAAAA,CAAa7B,GAAAA,CAAI+B,GAAAA,CAAAA,EAAM;AACzB,UAAA,OAAOxD,MAAAA;AACT;AACA,QAAA,IAAI2C,MAAAA,IAASA,kBAAiBnE,eAAAA,EAAM;AAClC,UAAA,OAAOmE,MAAAA,CAAMxB,IAAAA;AACf;AACA,QAAA,OAAOwB,MAAAA;AACT,OAAA;AACF,KAAA,GAAA,EACA,IAAA,CAAK/E,MAAAA,GAAS,CAAA,GAAIoC,MAAAA,CAAAA;AAEtB;EAEA,MAAcyD,eAAAA,CACZC,SACA7C,IAAAA,EACe;AACf,IAAA,MAAM/B,MAAAA,GAAS,IAAA,CAAKsD,YAAAA,CAAavB,IAAAA,CAAAA;AACjC,IAAA,MAAM8C,OAAAA,GAAU,GAAG7E,MAAAA,CAAAA,EAAS,MAAM,IAAA,CAAK4D,aAAAA,CAAcgB,OAAAA,CAAQ3D,KAAK,CAAA,CAAA,CAAA;AAElE,IAAA,MAAM,IAAA,CAAK5B,OAAAA,CAAQyF,IAAAA,CAAK,OAAA,EAAS;AAC/BD,MAAAA,OAAAA;MACAE,KAAAA,EAAO,IAAA,CAAKvB,cAAczB,IAAAA,CAAAA;MAC1BiD,MAAAA,EAAQ;AAACJ,QAAAA,OAAAA;AAAS7C,QAAAA;;KACpB,CAAA;AAEA,IAAA,IAAA,CAAKnD,OAAOiG,OAAAA,CAAAA;AACd;EAEA,MAAcI,iBAAAA,CACZL,SACA7C,IAAAA,EACe;AACf,IAAA,MAAM/B,MAAAA,GAAS,IAAA,CAAKsD,YAAAA,CAAavB,IAAAA,CAAAA;AAEjC,IAAA,MAAM8C,OAAAA,GAAU,GAAG7E,MAAAA,CAAAA,EAAS,MAAM,IAAA,CAAK4D,aAAAA,CAAcgB,OAAAA,CAAAA,CAAAA,CAAAA;AAErD,IAAA,MAAM,IAAA,CAAKvF,OAAAA,CAAQyF,IAAAA,CAAK,SAAA,EAAW;AACjCD,MAAAA,OAAAA;MACAE,KAAAA,EAAO,IAAA,CAAKvB,cAAczB,IAAAA,CAAAA;MAC1BiD,MAAAA,EAAQ;AAACJ,QAAAA,OAAAA;AAAS7C,QAAAA;;KACpB,CAAA;AAEA,IAAA,IAAA,CAAKnD,OAAOiG,OAAAA,CAAAA;AACd;EAEA,MAAcK,eAAAA,CACZN,SACA7C,IAAAA,EACe;AACf,IAAA,MAAM/B,MAAAA,GAAS,IAAA,CAAKsD,YAAAA,CAAavB,IAAAA,CAAAA;AACjC,IAAA,MAAM8C,OAAAA,GAAU,GAAG7E,MAAAA,CAAAA,EAAS,MAAM,IAAA,CAAK4D,aAAAA,CAAcgB,OAAAA,CAAAA,CAAAA,CAAAA;AAErD,IAAA,MAAM,IAAA,CAAKvF,OAAAA,CAAQyF,IAAAA,CAAK,OAAA,EAAS;AAC/BD,MAAAA,OAAAA;MACAE,KAAAA,EAAO,IAAA,CAAKvB,cAAczB,IAAAA,CAAAA;MAC1BiD,MAAAA,EAAQ;AAACJ,QAAAA,OAAAA;AAAS7C,QAAAA;;KACpB,CAAA;AAEA,IAAA,IAAA,CAAKnD,OAAOiG,OAAAA,CAAAA;AACd;EAEA,MAAcM,gBAAAA,CACZP,SACA7C,IAAAA,EACe;AACf,IAAA,MAAM/B,MAAAA,GAAS,IAAA,CAAKsD,YAAAA,CAAavB,IAAAA,CAAAA;AACjC,IAAA,MAAM8C,OAAAA,GAAU,CAAA,EAAG7E,MAAAA,CAAAA,EAAS,MAAM,IAAA,CAAK4D,aAAAA,CAAcgB,OAAAA,CAAQQ,KAAAA,IAASR,OAAAA,CAAQS,MAAM,CAAA,CAAA,CAAA;AACpF,IAAA,MAAM,IAAA,CAAKhG,OAAAA,CAAQyF,IAAAA,CAAK,QAAA,EAAU;AAChCD,MAAAA,OAAAA;MACAE,KAAAA,EAAO,IAAA,CAAKvB,cAAczB,IAAAA,CAAAA;MAC1BiD,MAAAA,EAAQ;AAACJ,QAAAA,OAAAA;AAAS7C,QAAAA;;KACpB,CAAA;AACF;AACF","file":"trajectory.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { Middleware, RunContext, RunContextCallbacks, RunInstance } from \"@/context.js\";\nimport { Callback, Emitter, EventMeta } from \"@/emitter/emitter.js\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { Logger } from \"@/logger/logger.js\";\nimport { BaseAgent } from \"@/agents/base.js\";\nimport type { AnyConstructable } from \"@/internals/types.js\";\nimport { capitalize } from \"remeda\";\nimport { ChatModel } from \"@/backend/chat.js\";\nimport { Tool } from \"@/tools/base.js\";\nimport { Requirement } from \"@/agents/requirement/requirements/requirement.js\";\nimport { isPrimitive } from \"sequelize/lib/utils\";\nimport type { InferCallbackValue } from \"@/emitter/types.js\";\nimport { Serializer } from \"@/serializer/serializer.js\";\n\n/**\n * Information about how deep the given entity is in the execution tree.\n */\nexport interface TraceLevel {\n /** Relative depth to the included (observed) elements */\n relative: number;\n /** Absolute depth from the root */\n absolute: number;\n}\n\n/**\n * Input for custom formatter function\n */\nexport interface GlobalTrajectoryMiddlewareFormatterInput {\n prefix: string;\n className: string;\n eventName: string;\n instanceName: string | null;\n}\n\nexport interface GlobalTrajectoryMiddlewareCallbacks {\n start: Callback<{\n message: string;\n level: TraceLevel;\n origin: [InferCallbackValue<RunContextCallbacks[\"start\"]>, EventMeta];\n }>;\n success: Callback<{\n message: string;\n level: TraceLevel;\n origin: [InferCallbackValue<RunContextCallbacks[\"success\"]>, EventMeta];\n }>;\n error: Callback<{\n message: string;\n level: TraceLevel;\n origin: [InferCallbackValue<RunContextCallbacks[\"error\"]>, EventMeta];\n }>;\n finish: Callback<{\n message: string;\n level: TraceLevel;\n origin: [InferCallbackValue<RunContextCallbacks[\"finish\"]>, EventMeta];\n }>;\n}\n\ntype OutputTargetFn = (message: string) => void;\ntype OutputTarget = Logger | OutputTargetFn;\n\nexport interface GlobalTrajectoryMiddlewareOptions {\n /** Specify output target: 'console', Logger instance, custom function, or null to disable */\n target?: OutputTarget;\n /** List of classes to include in the trajectory */\n included?: AnyConstructable[];\n /** List of classes to exclude from the trajectory */\n excluded?: AnyConstructable[];\n /** Use pretty formatting for the trajectory */\n pretty?: boolean;\n /** Customize how instances of individual classes should be printed */\n prefixByType?: Map<AnyConstructable, string>;\n /** Enable/Disable the logging */\n enabled?: boolean;\n /** Whether to observe trajectories of nested run contexts */\n matchNested?: boolean;\n /** Defines a priority for registered events */\n emitterPriority?: number;\n /** Custom formatter function */\n formatter?: (input: GlobalTrajectoryMiddlewareFormatterInput) => string;\n}\n\n/**\n * Middleware for capturing and logging execution flow of agents, tools, and models.\n * Provides hierarchical visualization with indentation to show the call stack.\n */\nexport class GlobalTrajectoryMiddleware<T extends RunInstance = any> extends Middleware<T> {\n protected enabled: boolean;\n protected included: AnyConstructable[];\n protected excluded: AnyConstructable[];\n protected cleanups: (() => void)[] = [];\n protected target: (message: string) => void;\n protected ctx: RunContext<T> | null = null;\n protected pretty: boolean;\n protected traceLevel = new Map<string, TraceLevel>();\n protected prefixByType: Map<any, string>;\n protected matchNested: boolean;\n protected emitterPriority: number;\n protected formatter: (input: GlobalTrajectoryMiddlewareFormatterInput) => string;\n public readonly emitter: Emitter<GlobalTrajectoryMiddlewareCallbacks>;\n\n constructor(options: GlobalTrajectoryMiddlewareOptions = {}) {\n super();\n\n this.enabled = options.enabled ?? true;\n this.included = options.included ?? [];\n this.excluded = options.excluded ?? [];\n this.target = this.createTarget(options.target);\n this.pretty = options.pretty ?? false;\n this.matchNested = options.matchNested ?? true;\n this.emitterPriority = options.emitterPriority ?? -1; // run later\n\n // Default prefixes\n this.prefixByType = new Map([\n [BaseAgent, \"🤖 \"],\n [ChatModel, \"💬 \"],\n [Tool, \"🛠️ \"],\n [Requirement, \"🔎 \"],\n ...((options.prefixByType ? Array.from(options.prefixByType.entries()) : []) as any),\n ]);\n\n this.formatter =\n options.formatter ??\n ((x) => `${x.prefix}${x.className}[${x.instanceName || x.className}][${x.eventName}]`);\n\n this.emitter = Emitter.root.child<GlobalTrajectoryMiddlewareCallbacks>({\n namespace: [\"middleware\", \"globalTrajectory\"],\n });\n }\n\n /**\n * Bind the middleware to a run context\n */\n bind(ctx: RunContext<T>): void {\n // Cleanup previous bindings\n while (this.cleanups.length > 0) {\n this.cleanups.pop()!();\n }\n\n this.traceLevel.clear();\n this.traceLevel.set(ctx.runId, { relative: 0, absolute: 0 });\n this.ctx = ctx;\n\n this.bindEmitter(ctx.emitter);\n }\n\n private createTarget(input?: OutputTarget): OutputTargetFn {\n if (input === null || input === undefined) {\n // eslint-disable-next-line no-console\n return (msg) => console.log(msg);\n } else if (input instanceof Logger) {\n return (msg) => input.debug(msg);\n } else {\n return input;\n }\n }\n\n private bindEmitter(emitter: Emitter<any>): void {\n // Track all events for trace ID logging\n this.cleanups.push(\n emitter.match(\"*.*\", (_, event) => this.logTraceId(event), {\n matchNested: true,\n }),\n );\n\n // Handle top-level events\n const handleTopLevelEvent = async (data: any, meta: EventMeta) => {\n if (!(meta.creator instanceof RunContext)) {\n return;\n }\n if (!meta.trace) {\n return;\n }\n\n this.logTraceId(meta);\n if (!this.isAllowed(meta)) {\n return;\n }\n if (!this.enabled) {\n return;\n }\n\n const eventName = capitalize(meta.name) as \"Start\" | \"Success\" | \"Error\" | \"Finish\";\n await this[`onInternal${eventName}` as const].call(this, data, meta);\n };\n\n this.cleanups.push(\n emitter.match(\n (event) => [\"start\", \"success\", \"error\", \"finish\"].includes(event.name),\n handleTopLevelEvent,\n {\n matchNested: false,\n isBlocking: true,\n priority: this.emitterPriority,\n },\n ),\n );\n\n // Handle nested events if enabled\n if (this.matchNested) {\n const handleNestedEvent = async (data: any, meta: EventMeta) => {\n if (!(meta.creator instanceof RunContext)) {\n return;\n }\n if (meta.creator.emitter !== emitter) {\n await handleTopLevelEvent(data, meta);\n this.bindEmitter(meta.creator.emitter);\n }\n };\n\n this.cleanups.push(\n emitter.match((event) => event.name === \"start\", handleNestedEvent, {\n matchNested: true,\n isBlocking: true,\n priority: this.emitterPriority,\n }),\n );\n }\n }\n\n private logTraceId(meta: EventMeta): void {\n if (!meta.trace?.runId) {\n return;\n }\n if (this.traceLevel.has(meta.trace.runId)) {\n return;\n }\n if (meta.trace.parentRunId === meta.trace.runId) {\n return;\n }\n\n if (meta.trace.parentRunId) {\n const allowed = this.isAllowed(meta);\n const parentTrace = this.traceLevel.get(meta.trace.parentRunId) ?? {\n relative: 0,\n absolute: 0,\n };\n this.traceLevel.set(meta.trace.runId, {\n relative: parentTrace.relative + (allowed ? 1 : 0),\n absolute: parentTrace.absolute + 1,\n });\n }\n }\n\n private isAllowed(meta: EventMeta): boolean {\n let target: any = meta.creator;\n if (target instanceof RunContext) {\n target = target.instance;\n }\n\n for (const excluded of this.excluded) {\n if (target instanceof excluded) {\n return false;\n }\n }\n\n if (this.included.length === 0) {\n return true;\n }\n\n return this.included.some((included) => target instanceof included);\n }\n\n private extractName(meta: EventMeta): string {\n let target: any = meta.creator;\n if (target instanceof RunContext) {\n target = target.instance;\n }\n\n const className = target.constructor.name;\n let targetName: string | null = null;\n\n if (target instanceof BaseAgent && target.meta?.name) {\n targetName = target.meta.name;\n } else if (\"name\" in target && typeof target.name === \"string\") {\n targetName = target.name;\n }\n\n let prefix = \"\";\n for (const [type, typePrefix] of this.prefixByType.entries()) {\n if (target instanceof type) {\n prefix = typePrefix;\n break;\n }\n }\n\n const input: GlobalTrajectoryMiddlewareFormatterInput = {\n prefix,\n className,\n instanceName: targetName,\n eventName: meta.name,\n };\n\n return this.formatter(input);\n }\n\n private formatPrefix(meta: EventMeta): string {\n if (!meta.trace) {\n return \"\";\n }\n\n const indent = this.getTraceLevel(meta, \"self\").relative;\n const indentParent = this.getTraceLevel(meta, \"parent\").relative;\n const indentDiff = indent - indentParent;\n\n let prefix = \"\";\n prefix += \" \".repeat(indentParent * 2);\n\n if (meta.name !== \"start\" && indent) {\n prefix += \"<\";\n }\n\n prefix += \"--\".repeat(indentDiff);\n\n if (meta.name === \"start\" && prefix && indent) {\n prefix += \">\";\n }\n\n if (prefix) {\n prefix = `${prefix} `;\n }\n\n const name = this.extractName(meta);\n return `${prefix}${name}: `;\n }\n\n private getTraceLevel(meta: EventMeta, type: \"self\" | \"parent\" = \"self\"): TraceLevel {\n if (!meta.trace) {\n return { relative: 0, absolute: 0 };\n }\n\n const runId = type === \"parent\" ? meta.trace.parentRunId || \"\" : meta.trace.runId;\n return this.traceLevel.get(runId) ?? { relative: 0, absolute: 0 };\n }\n\n private async formatPayload(value: any): Promise<string> {\n if (isPrimitive(value)) {\n return String(value);\n }\n\n if (value instanceof FrameworkError) {\n return value.explain();\n }\n\n const serialized = await Serializer.serialize(value);\n return JSON.stringify(\n await Serializer.deserialize(serialized, [], true),\n (() => {\n const excludedKeys = new Set([\"emitter\", \"cleanups\", \"creator\", \"listeners\"]);\n return (key, value) => {\n if (excludedKeys.has(key)) {\n return undefined;\n }\n if (value && value instanceof Tool) {\n return value.name;\n }\n return value;\n };\n })(),\n this.pretty ? 2 : undefined,\n );\n }\n\n private async onInternalStart(\n payload: InferCallbackValue<RunContextCallbacks[\"start\"]>,\n meta: EventMeta,\n ): Promise<void> {\n const prefix = this.formatPrefix(meta);\n const message = `${prefix}${await this.formatPayload(payload.input)}`;\n\n await this.emitter.emit(\"start\", {\n message,\n level: this.getTraceLevel(meta),\n origin: [payload, meta],\n });\n\n this.target(message);\n }\n\n private async onInternalSuccess(\n payload: InferCallbackValue<RunContextCallbacks[\"success\"]>,\n meta: EventMeta,\n ): Promise<void> {\n const prefix = this.formatPrefix(meta);\n // TODO: change to payload.output once available\n const message = `${prefix}${await this.formatPayload(payload)}`;\n\n await this.emitter.emit(\"success\", {\n message,\n level: this.getTraceLevel(meta),\n origin: [payload, meta],\n });\n\n this.target(message);\n }\n\n private async onInternalError(\n payload: InferCallbackValue<RunContextCallbacks[\"error\"]>,\n meta: EventMeta,\n ): Promise<void> {\n const prefix = this.formatPrefix(meta);\n const message = `${prefix}${await this.formatPayload(payload)}`;\n\n await this.emitter.emit(\"error\", {\n message,\n level: this.getTraceLevel(meta),\n origin: [payload, meta],\n });\n\n this.target(message);\n }\n\n private async onInternalFinish(\n payload: InferCallbackValue<RunContextCallbacks[\"finish\"]>,\n meta: EventMeta,\n ): Promise<void> {\n const prefix = this.formatPrefix(meta);\n const message = `${prefix}${await this.formatPayload(payload.error || payload.output)}`;\n await this.emitter.emit(\"finish\", {\n message,\n level: this.getTraceLevel(meta),\n origin: [payload, meta],\n });\n }\n}\n"]}
1
+ {"version":3,"sources":["../../src/middleware/trajectory.ts"],"names":["GlobalTrajectoryMiddleware","Middleware","enabled","included","excluded","cleanups","target","ctx","pretty","traceLevel","Map","prefixByType","matchNested","emitterPriority","formatter","emitter","options","createTarget","BaseAgent","ChatModel","Tool","Requirement","Array","from","entries","x","prefix","className","instanceName","eventName","Emitter","root","child","namespace","bind","length","pop","clear","set","runId","relative","absolute","bindEmitter","input","undefined","msg","console","log","Logger","debug","push","match","_","event","logTraceId","handleTopLevelEvent","data","meta","creator","RunContext","trace","isAllowed","capitalize","name","call","matchInternalEvent","isBlocking","priority","handleNestedEvent","has","parentRunId","allowed","parentTrace","get","instance","some","extractName","targetName","type","typePrefix","formatPrefix","indent","getTraceLevel","indentParent","indentDiff","repeat","formatPayload","value","isPrimitive","String","FrameworkError","explain","serialized","Serializer","serialize","JSON","stringify","deserialize","excludedKeys","Set","key","onInternalStart","payload","message","emit","level","origin","onInternalSuccess","onInternalError","onInternalFinish","error","output","names","includes","getProp","context"],"mappings":";;;;;;;;;;;;;;;;;AA2FO,MAAMA,mCAAgEC,sBAAAA,CAAAA;EA3F7E;;;AA4FYC,EAAAA,OAAAA;AACAC,EAAAA,QAAAA;AACAC,EAAAA,QAAAA;AACAC,EAAAA,QAAAA,GAA2B,EAAA;AAC3BC,EAAAA,MAAAA;EACAC,GAAAA,GAA4B,IAAA;AAC5BC,EAAAA,MAAAA;AACAC,EAAAA,UAAAA,uBAAiBC,GAAAA,EAAAA;AACjBC,EAAAA,YAAAA;AACAC,EAAAA,WAAAA;AACAC,EAAAA,eAAAA;AACAC,EAAAA,SAAAA;AACMC,EAAAA,OAAAA;EAEhB,WAAA,CAAYC,OAAAA,GAA6C,EAAC,EAAG;AAC3D,IAAA,KAAA,EAAK;AAEL,IAAA,IAAA,CAAKd,OAAAA,GAAUc,QAAQd,OAAAA,IAAW,IAAA;AAClC,IAAA,IAAA,CAAKC,QAAAA,GAAWa,OAAAA,CAAQb,QAAAA,IAAY,EAAA;AACpC,IAAA,IAAA,CAAKC,QAAAA,GAAWY,OAAAA,CAAQZ,QAAAA,IAAY,EAAA;AACpC,IAAA,IAAA,CAAKE,MAAAA,GAAS,IAAA,CAAKW,YAAAA,CAAaD,OAAAA,CAAQV,MAAM,CAAA;AAC9C,IAAA,IAAA,CAAKE,MAAAA,GAASQ,QAAQR,MAAAA,IAAU,KAAA;AAChC,IAAA,IAAA,CAAKI,WAAAA,GAAcI,QAAQJ,WAAAA,IAAe,IAAA;AAC1C,IAAA,IAAA,CAAKC,eAAAA,GAAkBG,QAAQH,eAAAA,IAAmB,EAAA;AAGlD,IAAA,IAAA,CAAKF,YAAAA,GAAe,IAAID,GAAAA,CAAI;AAC1B,MAAA;AAACQ,QAAAA,kBAAAA;AAAW,QAAA;;AACZ,MAAA;AAACC,QAAAA,kBAAAA;AAAW,QAAA;;AACZ,MAAA;AAACC,QAAAA,eAAAA;AAAM,QAAA;;AACP,MAAA;AAACC,QAAAA,2BAAAA;AAAa,QAAA;;SACTL,OAAAA,CAAQL,YAAAA,GAAeW,MAAMC,IAAAA,CAAKP,OAAAA,CAAQL,aAAaa,OAAAA,EAAO,IAAM;AAC1E,KAAA,CAAA;AAED,IAAA,IAAA,CAAKV,YACHE,OAAAA,CAAQF,SAAAA,KACP,CAACW,CAAAA,KAAM,CAAA,EAAGA,EAAEC,MAAM,CAAA,EAAGD,CAAAA,CAAEE,SAAS,IAAIF,CAAAA,CAAEG,YAAAA,IAAgBH,EAAEE,SAAS,CAAA,EAAA,EAAKF,EAAEI,SAAS,CAAA,CAAA,CAAA,CAAA;AAEpF,IAAA,IAAA,CAAKd,OAAAA,GAAUe,mBAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAA2C;MACrEC,SAAAA,EAAW;AAAC,QAAA,YAAA;AAAc,QAAA;;KAC5B,CAAA;AACF;;;;AAKAC,EAAAA,IAAAA,CAAK3B,GAAAA,EAA0B;AAE7B,IAAA,OAAO,IAAA,CAAKF,QAAAA,CAAS8B,MAAAA,GAAS,CAAA,EAAG;AAC/B,MAAA,IAAA,CAAK9B,QAAAA,CAAS+B,KAAG,EAAA;AACnB;AAEA,IAAA,IAAA,CAAK3B,WAAW4B,KAAAA,EAAK;AACrB,IAAA,IAAA,CAAK5B,UAAAA,CAAW6B,GAAAA,CAAI/B,GAAAA,CAAIgC,KAAAA,EAAO;MAAEC,QAAAA,EAAU,CAAA;MAAGC,QAAAA,EAAU;KAAE,CAAA;AAC1D,IAAA,IAAA,CAAKlC,GAAAA,GAAMA,GAAAA;AAEX,IAAA,IAAA,CAAKmC,WAAAA,CAAYnC,IAAIQ,OAAO,CAAA;AAC9B;AAEQE,EAAAA,YAAAA,CAAa0B,KAAAA,EAAsC;AACzD,IAAA,IAAIA,KAAAA,KAAU,IAAA,IAAQA,KAAAA,KAAUC,MAAAA,EAAW;AAEzC,MAAA,OAAO,CAACC,GAAAA,KAAQC,OAAAA,CAAQC,GAAAA,CAAIF,GAAAA,CAAAA;AAC9B,KAAA,MAAA,IAAWF,iBAAiBK,iBAAAA,EAAQ;AAClC,MAAA,OAAO,CAACH,GAAAA,KAAQF,KAAAA,CAAMM,KAAAA,CAAMJ,GAAAA,CAAAA;KAC9B,MAAO;AACL,MAAA,OAAOF,KAAAA;AACT;AACF;AAEQD,EAAAA,WAAAA,CAAY3B,OAAAA,EAA6B;AAE/C,IAAA,IAAA,CAAKV,QAAAA,CAAS6C,IAAAA,CACZnC,OAAAA,CAAQoC,KAAAA,CAAM,KAAA,EAAO,CAACC,CAAAA,EAAGC,KAAAA,KAAU,IAAA,CAAKC,UAAAA,CAAWD,KAAAA,CAAAA,EAAQ;MACzDzC,WAAAA,EAAa;AACf,KAAA,CAAA,CAAA;AAIF,IAAA,MAAM2C,mBAAAA,mBAAsB,MAAA,CAAA,OAAOC,IAAAA,EAAWC,IAAAA,KAAAA;AAC5C,MAAA,IAAI,EAAEA,IAAAA,CAAKC,OAAAA,YAAmBC,sBAAAA,CAAAA,EAAa;AACzC,QAAA;AACF;AACA,MAAA,IAAI,CAACF,KAAKG,KAAAA,EAAO;AACf,QAAA;AACF;AAEA,MAAA,IAAA,CAAKN,WAAWG,IAAAA,CAAAA;AAChB,MAAA,IAAI,CAAC,IAAA,CAAKI,SAAAA,CAAUJ,IAAAA,CAAAA,EAAO;AACzB,QAAA;AACF;AACA,MAAA,IAAI,CAAC,KAAKvD,OAAAA,EAAS;AACjB,QAAA;AACF;AAEA,MAAA,MAAM2B,SAAAA,GAAYiC,iBAAAA,CAAWL,IAAAA,CAAKM,IAAI,CAAA;AACtC,MAAA,MAAM,IAAA,CAAK,aAAalC,SAAAA,CAAAA,CAAW,EAAWmC,IAAAA,CAAK,IAAA,EAAMR,MAAMC,IAAAA,CAAAA;KACjE,EAlB4B,qBAAA,CAAA;AAoB5B,IAAA,IAAA,CAAKpD,QAAAA,CAAS6C,IAAAA,CACZnC,OAAAA,CAAQoC,KAAAA,CACNc,kBAAAA,CAAmB;AAAC,MAAA,OAAA;AAAS,MAAA,SAAA;AAAW,MAAA,OAAA;AAAS,MAAA;AAAS,KAAA,CAAA,EAC1DV,mBAAAA,EACA;MACE3C,WAAAA,EAAa,KAAA;MACbsD,UAAAA,EAAY,IAAA;AACZC,MAAAA,QAAAA,EAAU,IAAA,CAAKtD;AACjB,KAAA,CAAA,CAAA;AAKJ,IAAA,IAAI,KAAKD,WAAAA,EAAa;AACpB,MAAA,MAAMwD,iBAAAA,mBAAoB,MAAA,CAAA,OAAOZ,IAAAA,EAAWC,IAAAA,KAAAA;AAC1C,QAAA,IAAI,EAAEA,IAAAA,CAAKC,OAAAA,YAAmBC,sBAAAA,CAAAA,EAAa;AACzC,UAAA;AACF;AACA,QAAA,IAAIF,IAAAA,CAAKC,OAAAA,CAAQ3C,OAAAA,KAAYA,OAAAA,EAAS;AACpC,UAAA,MAAMwC,mBAAAA,CAAoBC,MAAMC,IAAAA,CAAAA;AAChC,UAAA,IAAA,CAAKf,WAAAA,CAAYe,IAAAA,CAAKC,OAAAA,CAAQ3C,OAAO,CAAA;AACvC;OACF,EAR0B,mBAAA,CAAA;AAU1B,MAAA,IAAA,CAAKV,QAAAA,CAAS6C,IAAAA,CACZnC,OAAAA,CAAQoC,KAAAA,CAAMc,kBAAAA,CAAmB;AAAC,QAAA;AAAQ,OAAA,CAAA,EAAGG,iBAAAA,EAAmB;QAC9DxD,WAAAA,EAAa,IAAA;QACbsD,UAAAA,EAAY,IAAA;AACZC,QAAAA,QAAAA,EAAU,IAAA,CAAKtD;AACjB,OAAA,CAAA,CAAA;AAEJ;AACF;AAEQyC,EAAAA,UAAAA,CAAWG,IAAAA,EAAuB;AACxC,IAAA,IAAI,CAACA,IAAAA,CAAKG,KAAAA,EAAOrB,KAAAA,EAAO;AACtB,MAAA;AACF;AACA,IAAA,IAAI,KAAK9B,UAAAA,CAAW4D,GAAAA,CAAIZ,IAAAA,CAAKG,KAAAA,CAAMrB,KAAK,CAAA,EAAG;AACzC,MAAA;AACF;AACA,IAAA,IAAIkB,IAAAA,CAAKG,KAAAA,CAAMU,WAAAA,KAAgBb,IAAAA,CAAKG,MAAMrB,KAAAA,EAAO;AAC/C,MAAA;AACF;AAEA,IAAA,IAAIkB,IAAAA,CAAKG,MAAMU,WAAAA,EAAa;AAC1B,MAAA,MAAMC,OAAAA,GAAU,IAAA,CAAKV,SAAAA,CAAUJ,IAAAA,CAAAA;AAC/B,MAAA,MAAMe,cAAc,IAAA,CAAK/D,UAAAA,CAAWgE,IAAIhB,IAAAA,CAAKG,KAAAA,CAAMU,WAAW,CAAA,IAAK;QACjE9B,QAAAA,EAAU,CAAA;QACVC,QAAAA,EAAU;AACZ,OAAA;AACA,MAAA,IAAA,CAAKhC,UAAAA,CAAW6B,GAAAA,CAAImB,IAAAA,CAAKG,KAAAA,CAAMrB,KAAAA,EAAO;QACpCC,QAAAA,EAAUgC,WAAAA,CAAYhC,QAAAA,IAAY+B,OAAAA,GAAU,CAAA,GAAI,CAAA,CAAA;AAChD9B,QAAAA,QAAAA,EAAU+B,YAAY/B,QAAAA,GAAW;OACnC,CAAA;AACF;AACF;AAEQoB,EAAAA,SAAAA,CAAUJ,IAAAA,EAA0B;AAC1C,IAAA,IAAInD,SAAcmD,IAAAA,CAAKC,OAAAA;AACvB,IAAA,IAAIpD,kBAAkBqD,sBAAAA,EAAY;AAChCrD,MAAAA,MAAAA,GAASA,MAAAA,CAAOoE,QAAAA;AAClB;AAEA,IAAA,KAAA,MAAWtE,QAAAA,IAAY,KAAKA,QAAAA,EAAU;AACpC,MAAA,IAAIE,kBAAkBF,QAAAA,EAAU;AAC9B,QAAA,OAAO,KAAA;AACT;AACF;AAEA,IAAA,IAAI,IAAA,CAAKD,QAAAA,CAASgC,MAAAA,KAAW,CAAA,EAAG;AAC9B,MAAA,OAAO,IAAA;AACT;AAEA,IAAA,OAAO,KAAKhC,QAAAA,CAASwE,IAAAA,CAAK,CAACxE,QAAAA,KAAaG,kBAAkBH,QAAAA,CAAAA;AAC5D;AAEQyE,EAAAA,WAAAA,CAAYnB,IAAAA,EAAyB;AAC3C,IAAA,IAAInD,SAAcmD,IAAAA,CAAKC,OAAAA;AACvB,IAAA,IAAIpD,kBAAkBqD,sBAAAA,EAAY;AAChCrD,MAAAA,MAAAA,GAASA,MAAAA,CAAOoE,QAAAA;AAClB;AAEA,IAAA,MAAM/C,SAAAA,GAAYrB,OAAO,WAAA,CAAYyD,IAAAA;AACrC,IAAA,IAAIc,UAAAA,GAA4B,IAAA;AAEhC,IAAA,IAAIvE,MAAAA,YAAkBY,kBAAAA,IAAaZ,MAAAA,CAAOmD,IAAAA,EAAMM,IAAAA,EAAM;AACpDc,MAAAA,UAAAA,GAAavE,OAAOmD,IAAAA,CAAKM,IAAAA;AAC3B,KAAA,MAAA,IAAW,MAAA,IAAUzD,MAAAA,IAAU,OAAOA,MAAAA,CAAOyD,SAAS,QAAA,EAAU;AAC9Dc,MAAAA,UAAAA,GAAavE,MAAAA,CAAOyD,IAAAA;AACtB;AAEA,IAAA,IAAIrC,MAAAA,GAAS,EAAA;AACb,IAAA,KAAA,MAAW,CAACoD,IAAAA,EAAMC,UAAAA,KAAe,IAAA,CAAKpE,YAAAA,CAAaa,SAAO,EAAI;AAC5D,MAAA,IAAIlB,kBAAkBwE,IAAAA,EAAM;AAC1BpD,QAAAA,MAAAA,GAASqD,UAAAA;AACT,QAAA;AACF;AACF;AAEA,IAAA,MAAMpC,KAAAA,GAAkD;AACtDjB,MAAAA,MAAAA;AACAC,MAAAA,SAAAA;MACAC,YAAAA,EAAciD,UAAAA;AACdhD,MAAAA,SAAAA,EAAW4B,IAAAA,CAAKM;AAClB,KAAA;AAEA,IAAA,OAAO,IAAA,CAAKjD,UAAU6B,KAAAA,CAAAA;AACxB;AAEQqC,EAAAA,YAAAA,CAAavB,IAAAA,EAAyB;AAC5C,IAAA,IAAI,CAACA,KAAKG,KAAAA,EAAO;AACf,MAAA,OAAO,EAAA;AACT;AAEA,IAAA,MAAMqB,MAAAA,GAAS,IAAA,CAAKC,aAAAA,CAAczB,IAAAA,EAAM,MAAA,CAAA,CAAQjB,QAAAA;AAChD,IAAA,MAAM2C,YAAAA,GAAe,IAAA,CAAKD,aAAAA,CAAczB,IAAAA,EAAM,QAAA,CAAA,CAAUjB,QAAAA;AACxD,IAAA,MAAM4C,aAAaH,MAAAA,GAASE,YAAAA;AAE5B,IAAA,IAAIzD,MAAAA,GAAS,EAAA;AACbA,IAAAA,MAAAA,IAAU,IAAA,CAAK2D,MAAAA,CAAOF,YAAAA,GAAe,CAAA,CAAA;AAErC,IAAA,IAAI1B,IAAAA,CAAKM,IAAAA,KAAS,OAAA,IAAWkB,MAAAA,EAAQ;AACnCvD,MAAAA,MAAAA,IAAU,GAAA;AACZ;AAEAA,IAAAA,MAAAA,IAAU,IAAA,CAAK2D,OAAOD,UAAAA,CAAAA;AAEtB,IAAA,IAAI3B,IAAAA,CAAKM,IAAAA,KAAS,OAAA,IAAWrC,MAAAA,IAAUuD,MAAAA,EAAQ;AAC7CvD,MAAAA,MAAAA,IAAU,GAAA;AACZ;AAEA,IAAA,IAAIA,MAAAA,EAAQ;AACVA,MAAAA,MAAAA,GAAS,GAAGA,MAAAA,CAAAA,CAAAA,CAAAA;AACd;AAEA,IAAA,MAAMqC,IAAAA,GAAO,IAAA,CAAKa,WAAAA,CAAYnB,IAAAA,CAAAA;AAC9B,IAAA,OAAO,CAAA,EAAG/B,MAAAA,CAAAA,EAASqC,IAAAA,CAAAA,EAAAA,CAAAA;AACrB;EAEQmB,aAAAA,CAAczB,IAAAA,EAAiBqB,OAA0B,MAAA,EAAoB;AACnF,IAAA,IAAI,CAACrB,KAAKG,KAAAA,EAAO;AACf,MAAA,OAAO;QAAEpB,QAAAA,EAAU,CAAA;QAAGC,QAAAA,EAAU;AAAE,OAAA;AACpC;AAEA,IAAA,MAAMF,KAAAA,GAAQuC,SAAS,QAAA,GAAWrB,IAAAA,CAAKG,MAAMU,WAAAA,IAAe,EAAA,GAAKb,KAAKG,KAAAA,CAAMrB,KAAAA;AAC5E,IAAA,OAAO,IAAA,CAAK9B,UAAAA,CAAWgE,GAAAA,CAAIlC,KAAAA,CAAAA,IAAU;MAAEC,QAAAA,EAAU,CAAA;MAAGC,QAAAA,EAAU;AAAE,KAAA;AAClE;AAEA,EAAA,MAAc6C,cAAcC,KAAAA,EAA6B;AACvD,IAAA,IAAIC,sBAAAA,CAAYD,KAAAA,CAAAA,EAAQ;AACtB,MAAA,OAAOE,OAAOF,KAAAA,CAAAA;AAChB;AAEA,IAAA,IAAIA,iBAAiBG,yBAAAA,EAAgB;AACnC,MAAA,OAAOH,MAAMI,OAAAA,EAAO;AACtB;AAEA,IAAA,MAAMC,UAAAA,GAAa,MAAMC,yBAAAA,CAAWC,SAAAA,CAAUP,KAAAA,CAAAA;AAC9C,IAAA,OAAOQ,IAAAA,CAAKC,SAAAA,CACV,MAAMH,yBAAAA,CAAWI,WAAAA,CAAYL,YAAY,EAAA,EAAI,IAAA,CAAA,kBAC5C,CAAA,MAAA;AACC,MAAA,MAAMM,YAAAA,uBAAmBC,GAAAA,CAAI;AAAC,QAAA,SAAA;AAAW,QAAA,UAAA;AAAY,QAAA,SAAA;AAAW,QAAA;AAAY,OAAA,CAAA;AAC5E,MAAA,OAAO,CAACC,KAAKb,MAAAA,KAAAA;AACX,QAAA,IAAIW,YAAAA,CAAa7B,GAAAA,CAAI+B,GAAAA,CAAAA,EAAM;AACzB,UAAA,OAAOxD,MAAAA;AACT;AACA,QAAA,IAAI2C,MAAAA,IAASA,kBAAiBnE,eAAAA,EAAM;AAClC,UAAA,OAAOmE,MAAAA,CAAMxB,IAAAA;AACf;AACA,QAAA,OAAOwB,MAAAA;AACT,OAAA;AACF,KAAA,GAAA,EACA,IAAA,CAAK/E,MAAAA,GAAS,CAAA,GAAIoC,MAAAA,CAAAA;AAEtB;EAEA,MAAcyD,eAAAA,CACZC,SACA7C,IAAAA,EACe;AACf,IAAA,MAAM/B,MAAAA,GAAS,IAAA,CAAKsD,YAAAA,CAAavB,IAAAA,CAAAA;AACjC,IAAA,MAAM8C,OAAAA,GAAU,GAAG7E,MAAAA,CAAAA,EAAS,MAAM,IAAA,CAAK4D,aAAAA,CAAcgB,OAAAA,CAAQ3D,KAAK,CAAA,CAAA,CAAA;AAElE,IAAA,MAAM,IAAA,CAAK5B,OAAAA,CAAQyF,IAAAA,CAAK,OAAA,EAAS;AAC/BD,MAAAA,OAAAA;MACAE,KAAAA,EAAO,IAAA,CAAKvB,cAAczB,IAAAA,CAAAA;MAC1BiD,MAAAA,EAAQ;AAACJ,QAAAA,OAAAA;AAAS7C,QAAAA;;KACpB,CAAA;AAEA,IAAA,IAAA,CAAKnD,OAAOiG,OAAAA,CAAAA;AACd;EAEA,MAAcI,iBAAAA,CACZL,SACA7C,IAAAA,EACe;AACf,IAAA,MAAM/B,MAAAA,GAAS,IAAA,CAAKsD,YAAAA,CAAavB,IAAAA,CAAAA;AAEjC,IAAA,MAAM8C,OAAAA,GAAU,GAAG7E,MAAAA,CAAAA,EAAS,MAAM,IAAA,CAAK4D,aAAAA,CAAcgB,OAAAA,CAAAA,CAAAA,CAAAA;AAErD,IAAA,MAAM,IAAA,CAAKvF,OAAAA,CAAQyF,IAAAA,CAAK,SAAA,EAAW;AACjCD,MAAAA,OAAAA;MACAE,KAAAA,EAAO,IAAA,CAAKvB,cAAczB,IAAAA,CAAAA;MAC1BiD,MAAAA,EAAQ;AAACJ,QAAAA,OAAAA;AAAS7C,QAAAA;;KACpB,CAAA;AAEA,IAAA,IAAA,CAAKnD,OAAOiG,OAAAA,CAAAA;AACd;EAEA,MAAcK,eAAAA,CACZN,SACA7C,IAAAA,EACe;AACf,IAAA,MAAM/B,MAAAA,GAAS,IAAA,CAAKsD,YAAAA,CAAavB,IAAAA,CAAAA;AACjC,IAAA,MAAM8C,OAAAA,GAAU,GAAG7E,MAAAA,CAAAA,EAAS,MAAM,IAAA,CAAK4D,aAAAA,CAAcgB,OAAAA,CAAAA,CAAAA,CAAAA;AAErD,IAAA,MAAM,IAAA,CAAKvF,OAAAA,CAAQyF,IAAAA,CAAK,OAAA,EAAS;AAC/BD,MAAAA,OAAAA;MACAE,KAAAA,EAAO,IAAA,CAAKvB,cAAczB,IAAAA,CAAAA;MAC1BiD,MAAAA,EAAQ;AAACJ,QAAAA,OAAAA;AAAS7C,QAAAA;;KACpB,CAAA;AAEA,IAAA,IAAA,CAAKnD,OAAOiG,OAAAA,CAAAA;AACd;EAEA,MAAcM,gBAAAA,CACZP,SACA7C,IAAAA,EACe;AACf,IAAA,MAAM/B,MAAAA,GAAS,IAAA,CAAKsD,YAAAA,CAAavB,IAAAA,CAAAA;AACjC,IAAA,MAAM8C,OAAAA,GAAU,CAAA,EAAG7E,MAAAA,CAAAA,EAAS,MAAM,IAAA,CAAK4D,aAAAA,CAAcgB,OAAAA,CAAQQ,KAAAA,IAASR,OAAAA,CAAQS,MAAM,CAAA,CAAA,CAAA;AACpF,IAAA,MAAM,IAAA,CAAKhG,OAAAA,CAAQyF,IAAAA,CAAK,QAAA,EAAU;AAChCD,MAAAA,OAAAA;MACAE,KAAAA,EAAO,IAAA,CAAKvB,cAAczB,IAAAA,CAAAA;MAC1BiD,MAAAA,EAAQ;AAACJ,QAAAA,OAAAA;AAAS7C,QAAAA;;KACpB,CAAA;AACF;AACF;AAEA,SAASQ,mBAAmB+C,KAAAA,EAAoC;AAC9D,EAAA,OAAO,CAACvD,SACNuD,KAAAA,CAAMC,QAAAA,CAASxD,KAAKM,IAAI,CAAA,IACxBmD,kBAAAA,CAAQzD,IAAAA,CAAK0D,OAAAA,EAAS;AAAC,IAAA;KAAa,KAAA,CAAA;AACxC;AAJSlD,MAAAA,CAAAA,kBAAAA,EAAAA,oBAAAA,CAAAA","file":"trajectory.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { Middleware, RunContext, RunContextCallbacks, RunInstance } from \"@/context.js\";\nimport { Callback, Emitter, EventMeta } from \"@/emitter/emitter.js\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { Logger } from \"@/logger/logger.js\";\nimport { BaseAgent } from \"@/agents/base.js\";\nimport type { AnyConstructable } from \"@/internals/types.js\";\nimport { capitalize } from \"remeda\";\nimport { ChatModel } from \"@/backend/chat.js\";\nimport { Tool } from \"@/tools/base.js\";\nimport { Requirement } from \"@/agents/requirement/requirements/requirement.js\";\nimport type { InferCallbackValue } from \"@/emitter/types.js\";\nimport { Serializer } from \"@/serializer/serializer.js\";\nimport { isPrimitive } from \"@/internals/helpers/guards.js\";\nimport { getProp } from \"@/internals/helpers/object.js\";\n\n/**\n * Information about how deep the given entity is in the execution tree.\n */\nexport interface TraceLevel {\n /** Relative depth to the included (observed) elements */\n relative: number;\n /** Absolute depth from the root */\n absolute: number;\n}\n\n/**\n * Input for custom formatter function\n */\nexport interface GlobalTrajectoryMiddlewareFormatterInput {\n prefix: string;\n className: string;\n eventName: string;\n instanceName: string | null;\n}\n\nexport interface GlobalTrajectoryMiddlewareCallbacks {\n start: Callback<{\n message: string;\n level: TraceLevel;\n origin: [InferCallbackValue<RunContextCallbacks[\"start\"]>, EventMeta];\n }>;\n success: Callback<{\n message: string;\n level: TraceLevel;\n origin: [InferCallbackValue<RunContextCallbacks[\"success\"]>, EventMeta];\n }>;\n error: Callback<{\n message: string;\n level: TraceLevel;\n origin: [InferCallbackValue<RunContextCallbacks[\"error\"]>, EventMeta];\n }>;\n finish: Callback<{\n message: string;\n level: TraceLevel;\n origin: [InferCallbackValue<RunContextCallbacks[\"finish\"]>, EventMeta];\n }>;\n}\n\ntype OutputTargetFn = (message: string) => void;\ntype OutputTarget = Logger | OutputTargetFn;\n\nexport interface GlobalTrajectoryMiddlewareOptions {\n /** Specify output target: 'console', Logger instance, custom function, or null to disable */\n target?: OutputTarget;\n /** List of classes to include in the trajectory */\n included?: AnyConstructable[];\n /** List of classes to exclude from the trajectory */\n excluded?: AnyConstructable[];\n /** Use pretty formatting for the trajectory */\n pretty?: boolean;\n /** Customize how instances of individual classes should be printed */\n prefixByType?: Map<AnyConstructable, string>;\n /** Enable/Disable the logging */\n enabled?: boolean;\n /** Whether to observe trajectories of nested run contexts */\n matchNested?: boolean;\n /** Defines a priority for registered events */\n emitterPriority?: number;\n /** Custom formatter function */\n formatter?: (input: GlobalTrajectoryMiddlewareFormatterInput) => string;\n}\n\n/**\n * Middleware for capturing and logging execution flow of agents, tools, and models.\n * Provides hierarchical visualization with indentation to show the call stack.\n */\nexport class GlobalTrajectoryMiddleware<T extends RunInstance = any> extends Middleware<T> {\n protected enabled: boolean;\n protected included: AnyConstructable[];\n protected excluded: AnyConstructable[];\n protected cleanups: (() => void)[] = [];\n protected target: (message: string) => void;\n protected ctx: RunContext<T> | null = null;\n protected pretty: boolean;\n protected traceLevel = new Map<string, TraceLevel>();\n protected prefixByType: Map<any, string>;\n protected matchNested: boolean;\n protected emitterPriority: number;\n protected formatter: (input: GlobalTrajectoryMiddlewareFormatterInput) => string;\n public readonly emitter: Emitter<GlobalTrajectoryMiddlewareCallbacks>;\n\n constructor(options: GlobalTrajectoryMiddlewareOptions = {}) {\n super();\n\n this.enabled = options.enabled ?? true;\n this.included = options.included ?? [];\n this.excluded = options.excluded ?? [];\n this.target = this.createTarget(options.target);\n this.pretty = options.pretty ?? false;\n this.matchNested = options.matchNested ?? true;\n this.emitterPriority = options.emitterPriority ?? -1; // run later\n\n // Default prefixes\n this.prefixByType = new Map([\n [BaseAgent, \"🤖 \"],\n [ChatModel, \"💬 \"],\n [Tool, \"🛠️ \"],\n [Requirement, \"🔎 \"],\n ...((options.prefixByType ? Array.from(options.prefixByType.entries()) : []) as any),\n ]);\n\n this.formatter =\n options.formatter ??\n ((x) => `${x.prefix}${x.className}[${x.instanceName || x.className}][${x.eventName}]`);\n\n this.emitter = Emitter.root.child<GlobalTrajectoryMiddlewareCallbacks>({\n namespace: [\"middleware\", \"globalTrajectory\"],\n });\n }\n\n /**\n * Bind the middleware to a run context\n */\n bind(ctx: RunContext<T>): void {\n // Cleanup previous bindings\n while (this.cleanups.length > 0) {\n this.cleanups.pop()!();\n }\n\n this.traceLevel.clear();\n this.traceLevel.set(ctx.runId, { relative: 0, absolute: 0 });\n this.ctx = ctx;\n\n this.bindEmitter(ctx.emitter);\n }\n\n private createTarget(input?: OutputTarget): OutputTargetFn {\n if (input === null || input === undefined) {\n // eslint-disable-next-line no-console\n return (msg) => console.log(msg);\n } else if (input instanceof Logger) {\n return (msg) => input.debug(msg);\n } else {\n return input;\n }\n }\n\n private bindEmitter(emitter: Emitter<any>): void {\n // Track all events for trace ID logging\n this.cleanups.push(\n emitter.match(\"*.*\", (_, event) => this.logTraceId(event), {\n matchNested: true,\n }),\n );\n\n // Handle top-level events\n const handleTopLevelEvent = async (data: any, meta: EventMeta) => {\n if (!(meta.creator instanceof RunContext)) {\n return;\n }\n if (!meta.trace) {\n return;\n }\n\n this.logTraceId(meta);\n if (!this.isAllowed(meta)) {\n return;\n }\n if (!this.enabled) {\n return;\n }\n\n const eventName = capitalize(meta.name) as \"Start\" | \"Success\" | \"Error\" | \"Finish\";\n await this[`onInternal${eventName}` as const].call(this, data, meta);\n };\n\n this.cleanups.push(\n emitter.match(\n matchInternalEvent([\"start\", \"success\", \"error\", \"finish\"]),\n handleTopLevelEvent,\n {\n matchNested: false,\n isBlocking: true,\n priority: this.emitterPriority,\n },\n ),\n );\n\n // Handle nested events if enabled\n if (this.matchNested) {\n const handleNestedEvent = async (data: any, meta: EventMeta) => {\n if (!(meta.creator instanceof RunContext)) {\n return;\n }\n if (meta.creator.emitter !== emitter) {\n await handleTopLevelEvent(data, meta);\n this.bindEmitter(meta.creator.emitter);\n }\n };\n\n this.cleanups.push(\n emitter.match(matchInternalEvent([\"start\"]), handleNestedEvent, {\n matchNested: true,\n isBlocking: true,\n priority: this.emitterPriority,\n }),\n );\n }\n }\n\n private logTraceId(meta: EventMeta): void {\n if (!meta.trace?.runId) {\n return;\n }\n if (this.traceLevel.has(meta.trace.runId)) {\n return;\n }\n if (meta.trace.parentRunId === meta.trace.runId) {\n return;\n }\n\n if (meta.trace.parentRunId) {\n const allowed = this.isAllowed(meta);\n const parentTrace = this.traceLevel.get(meta.trace.parentRunId) ?? {\n relative: 0,\n absolute: 0,\n };\n this.traceLevel.set(meta.trace.runId, {\n relative: parentTrace.relative + (allowed ? 1 : 0),\n absolute: parentTrace.absolute + 1,\n });\n }\n }\n\n private isAllowed(meta: EventMeta): boolean {\n let target: any = meta.creator;\n if (target instanceof RunContext) {\n target = target.instance;\n }\n\n for (const excluded of this.excluded) {\n if (target instanceof excluded) {\n return false;\n }\n }\n\n if (this.included.length === 0) {\n return true;\n }\n\n return this.included.some((included) => target instanceof included);\n }\n\n private extractName(meta: EventMeta): string {\n let target: any = meta.creator;\n if (target instanceof RunContext) {\n target = target.instance;\n }\n\n const className = target.constructor.name;\n let targetName: string | null = null;\n\n if (target instanceof BaseAgent && target.meta?.name) {\n targetName = target.meta.name;\n } else if (\"name\" in target && typeof target.name === \"string\") {\n targetName = target.name;\n }\n\n let prefix = \"\";\n for (const [type, typePrefix] of this.prefixByType.entries()) {\n if (target instanceof type) {\n prefix = typePrefix;\n break;\n }\n }\n\n const input: GlobalTrajectoryMiddlewareFormatterInput = {\n prefix,\n className,\n instanceName: targetName,\n eventName: meta.name,\n };\n\n return this.formatter(input);\n }\n\n private formatPrefix(meta: EventMeta): string {\n if (!meta.trace) {\n return \"\";\n }\n\n const indent = this.getTraceLevel(meta, \"self\").relative;\n const indentParent = this.getTraceLevel(meta, \"parent\").relative;\n const indentDiff = indent - indentParent;\n\n let prefix = \"\";\n prefix += \" \".repeat(indentParent * 2);\n\n if (meta.name !== \"start\" && indent) {\n prefix += \"<\";\n }\n\n prefix += \"--\".repeat(indentDiff);\n\n if (meta.name === \"start\" && prefix && indent) {\n prefix += \">\";\n }\n\n if (prefix) {\n prefix = `${prefix} `;\n }\n\n const name = this.extractName(meta);\n return `${prefix}${name}: `;\n }\n\n private getTraceLevel(meta: EventMeta, type: \"self\" | \"parent\" = \"self\"): TraceLevel {\n if (!meta.trace) {\n return { relative: 0, absolute: 0 };\n }\n\n const runId = type === \"parent\" ? meta.trace.parentRunId || \"\" : meta.trace.runId;\n return this.traceLevel.get(runId) ?? { relative: 0, absolute: 0 };\n }\n\n private async formatPayload(value: any): Promise<string> {\n if (isPrimitive(value)) {\n return String(value);\n }\n\n if (value instanceof FrameworkError) {\n return value.explain();\n }\n\n const serialized = await Serializer.serialize(value);\n return JSON.stringify(\n await Serializer.deserialize(serialized, [], true),\n (() => {\n const excludedKeys = new Set([\"emitter\", \"cleanups\", \"creator\", \"listeners\"]);\n return (key, value) => {\n if (excludedKeys.has(key)) {\n return undefined;\n }\n if (value && value instanceof Tool) {\n return value.name;\n }\n return value;\n };\n })(),\n this.pretty ? 2 : undefined,\n );\n }\n\n private async onInternalStart(\n payload: InferCallbackValue<RunContextCallbacks[\"start\"]>,\n meta: EventMeta,\n ): Promise<void> {\n const prefix = this.formatPrefix(meta);\n const message = `${prefix}${await this.formatPayload(payload.input)}`;\n\n await this.emitter.emit(\"start\", {\n message,\n level: this.getTraceLevel(meta),\n origin: [payload, meta],\n });\n\n this.target(message);\n }\n\n private async onInternalSuccess(\n payload: InferCallbackValue<RunContextCallbacks[\"success\"]>,\n meta: EventMeta,\n ): Promise<void> {\n const prefix = this.formatPrefix(meta);\n // TODO: change to payload.output once available\n const message = `${prefix}${await this.formatPayload(payload)}`;\n\n await this.emitter.emit(\"success\", {\n message,\n level: this.getTraceLevel(meta),\n origin: [payload, meta],\n });\n\n this.target(message);\n }\n\n private async onInternalError(\n payload: InferCallbackValue<RunContextCallbacks[\"error\"]>,\n meta: EventMeta,\n ): Promise<void> {\n const prefix = this.formatPrefix(meta);\n const message = `${prefix}${await this.formatPayload(payload)}`;\n\n await this.emitter.emit(\"error\", {\n message,\n level: this.getTraceLevel(meta),\n origin: [payload, meta],\n });\n\n this.target(message);\n }\n\n private async onInternalFinish(\n payload: InferCallbackValue<RunContextCallbacks[\"finish\"]>,\n meta: EventMeta,\n ): Promise<void> {\n const prefix = this.formatPrefix(meta);\n const message = `${prefix}${await this.formatPayload(payload.error || payload.output)}`;\n await this.emitter.emit(\"finish\", {\n message,\n level: this.getTraceLevel(meta),\n origin: [payload, meta],\n });\n }\n}\n\nfunction matchInternalEvent(names: (keyof RunContextCallbacks)[]) {\n return (meta: EventMeta): boolean =>\n names.includes(meta.name as keyof RunContextCallbacks) &&\n getProp(meta.context, [\"internal\"], false);\n}\n"]}
@@ -1,5 +1,5 @@
1
1
  import { RunContextCallbacks, RunInstance, Middleware, RunContext } from '../context.cjs';
2
- import { C as Callback, I as InferCallbackValue, a as EventMeta, E as Emitter } from '../emitter-CGKd_-kK.cjs';
2
+ import { C as Callback, I as InferCallbackValue, b as EventMeta, E as Emitter } from '../emitter-jN55XZZq.cjs';
3
3
  import { Logger } from '../logger/logger.cjs';
4
4
  import { AnyConstructable } from '../internals/types.cjs';
5
5
  import '../internals/serializable.cjs';
@@ -1,5 +1,5 @@
1
1
  import { RunContextCallbacks, RunInstance, Middleware, RunContext } from '../context.js';
2
- import { C as Callback, I as InferCallbackValue, a as EventMeta, E as Emitter } from '../emitter-CN7I4uSV.js';
2
+ import { C as Callback, I as InferCallbackValue, b as EventMeta, E as Emitter } from '../emitter-36-9MnvA.js';
3
3
  import { Logger } from '../logger/logger.js';
4
4
  import { AnyConstructable } from '../internals/types.js';
5
5
  import '../internals/serializable.js';
@@ -7,8 +7,9 @@ import { capitalize } from 'remeda';
7
7
  import { ChatModel } from '../backend/chat.js';
8
8
  import { Tool } from '../tools/base.js';
9
9
  import { Requirement } from '../agents/requirement/requirements/requirement.js';
10
- import { isPrimitive } from 'sequelize/lib/utils';
11
10
  import { Serializer } from '../serializer/serializer.js';
11
+ import { isPrimitive } from '../internals/helpers/guards.js';
12
+ import { getProp } from '../internals/helpers/object.js';
12
13
 
13
14
  var __defProp = Object.defineProperty;
14
15
  var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
@@ -110,12 +111,12 @@ class GlobalTrajectoryMiddleware extends Middleware {
110
111
  const eventName = capitalize(meta.name);
111
112
  await this[`onInternal${eventName}`].call(this, data, meta);
112
113
  }, "handleTopLevelEvent");
113
- this.cleanups.push(emitter.match((event) => [
114
+ this.cleanups.push(emitter.match(matchInternalEvent([
114
115
  "start",
115
116
  "success",
116
117
  "error",
117
118
  "finish"
118
- ].includes(event.name), handleTopLevelEvent, {
119
+ ]), handleTopLevelEvent, {
119
120
  matchNested: false,
120
121
  isBlocking: true,
121
122
  priority: this.emitterPriority
@@ -130,7 +131,9 @@ class GlobalTrajectoryMiddleware extends Middleware {
130
131
  this.bindEmitter(meta.creator.emitter);
131
132
  }
132
133
  }, "handleNestedEvent");
133
- this.cleanups.push(emitter.match((event) => event.name === "start", handleNestedEvent, {
134
+ this.cleanups.push(emitter.match(matchInternalEvent([
135
+ "start"
136
+ ]), handleNestedEvent, {
134
137
  matchNested: true,
135
138
  isBlocking: true,
136
139
  priority: this.emitterPriority
@@ -314,6 +317,12 @@ class GlobalTrajectoryMiddleware extends Middleware {
314
317
  });
315
318
  }
316
319
  }
320
+ function matchInternalEvent(names) {
321
+ return (meta) => names.includes(meta.name) && getProp(meta.context, [
322
+ "internal"
323
+ ], false);
324
+ }
325
+ __name(matchInternalEvent, "matchInternalEvent");
317
326
 
318
327
  export { GlobalTrajectoryMiddleware };
319
328
  //# sourceMappingURL=trajectory.js.map
@@ -1 +1 @@
1
- {"version":3,"sources":["../../src/middleware/trajectory.ts"],"names":["GlobalTrajectoryMiddleware","Middleware","enabled","included","excluded","cleanups","target","ctx","pretty","traceLevel","Map","prefixByType","matchNested","emitterPriority","formatter","emitter","options","createTarget","BaseAgent","ChatModel","Tool","Requirement","Array","from","entries","x","prefix","className","instanceName","eventName","Emitter","root","child","namespace","bind","length","pop","clear","set","runId","relative","absolute","bindEmitter","input","undefined","msg","console","log","Logger","debug","push","match","_","event","logTraceId","handleTopLevelEvent","data","meta","creator","RunContext","trace","isAllowed","capitalize","name","call","includes","isBlocking","priority","handleNestedEvent","has","parentRunId","allowed","parentTrace","get","instance","some","extractName","targetName","type","typePrefix","formatPrefix","indent","getTraceLevel","indentParent","indentDiff","repeat","formatPayload","value","isPrimitive","String","FrameworkError","explain","serialized","Serializer","serialize","JSON","stringify","deserialize","excludedKeys","Set","key","onInternalStart","payload","message","emit","level","origin","onInternalSuccess","onInternalError","onInternalFinish","error","output"],"mappings":";;;;;;;;;;;;;;AA0FO,MAAMA,mCAAgEC,UAAAA,CAAAA;EA1F7E;;;AA2FYC,EAAAA,OAAAA;AACAC,EAAAA,QAAAA;AACAC,EAAAA,QAAAA;AACAC,EAAAA,QAAAA,GAA2B,EAAA;AAC3BC,EAAAA,MAAAA;EACAC,GAAAA,GAA4B,IAAA;AAC5BC,EAAAA,MAAAA;AACAC,EAAAA,UAAAA,uBAAiBC,GAAAA,EAAAA;AACjBC,EAAAA,YAAAA;AACAC,EAAAA,WAAAA;AACAC,EAAAA,eAAAA;AACAC,EAAAA,SAAAA;AACMC,EAAAA,OAAAA;EAEhB,WAAA,CAAYC,OAAAA,GAA6C,EAAC,EAAG;AAC3D,IAAA,KAAA,EAAK;AAEL,IAAA,IAAA,CAAKd,OAAAA,GAAUc,QAAQd,OAAAA,IAAW,IAAA;AAClC,IAAA,IAAA,CAAKC,QAAAA,GAAWa,OAAAA,CAAQb,QAAAA,IAAY,EAAA;AACpC,IAAA,IAAA,CAAKC,QAAAA,GAAWY,OAAAA,CAAQZ,QAAAA,IAAY,EAAA;AACpC,IAAA,IAAA,CAAKE,MAAAA,GAAS,IAAA,CAAKW,YAAAA,CAAaD,OAAAA,CAAQV,MAAM,CAAA;AAC9C,IAAA,IAAA,CAAKE,MAAAA,GAASQ,QAAQR,MAAAA,IAAU,KAAA;AAChC,IAAA,IAAA,CAAKI,WAAAA,GAAcI,QAAQJ,WAAAA,IAAe,IAAA;AAC1C,IAAA,IAAA,CAAKC,eAAAA,GAAkBG,QAAQH,eAAAA,IAAmB,EAAA;AAGlD,IAAA,IAAA,CAAKF,YAAAA,GAAe,IAAID,GAAAA,CAAI;AAC1B,MAAA;AAACQ,QAAAA,SAAAA;AAAW,QAAA;;AACZ,MAAA;AAACC,QAAAA,SAAAA;AAAW,QAAA;;AACZ,MAAA;AAACC,QAAAA,IAAAA;AAAM,QAAA;;AACP,MAAA;AAACC,QAAAA,WAAAA;AAAa,QAAA;;SACTL,OAAAA,CAAQL,YAAAA,GAAeW,MAAMC,IAAAA,CAAKP,OAAAA,CAAQL,aAAaa,OAAAA,EAAO,IAAM;AAC1E,KAAA,CAAA;AAED,IAAA,IAAA,CAAKV,YACHE,OAAAA,CAAQF,SAAAA,KACP,CAACW,CAAAA,KAAM,CAAA,EAAGA,EAAEC,MAAM,CAAA,EAAGD,CAAAA,CAAEE,SAAS,IAAIF,CAAAA,CAAEG,YAAAA,IAAgBH,EAAEE,SAAS,CAAA,EAAA,EAAKF,EAAEI,SAAS,CAAA,CAAA,CAAA,CAAA;AAEpF,IAAA,IAAA,CAAKd,OAAAA,GAAUe,OAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAA2C;MACrEC,SAAAA,EAAW;AAAC,QAAA,YAAA;AAAc,QAAA;;KAC5B,CAAA;AACF;;;;AAKAC,EAAAA,IAAAA,CAAK3B,GAAAA,EAA0B;AAE7B,IAAA,OAAO,IAAA,CAAKF,QAAAA,CAAS8B,MAAAA,GAAS,CAAA,EAAG;AAC/B,MAAA,IAAA,CAAK9B,QAAAA,CAAS+B,KAAG,EAAA;AACnB;AAEA,IAAA,IAAA,CAAK3B,WAAW4B,KAAAA,EAAK;AACrB,IAAA,IAAA,CAAK5B,UAAAA,CAAW6B,GAAAA,CAAI/B,GAAAA,CAAIgC,KAAAA,EAAO;MAAEC,QAAAA,EAAU,CAAA;MAAGC,QAAAA,EAAU;KAAE,CAAA;AAC1D,IAAA,IAAA,CAAKlC,GAAAA,GAAMA,GAAAA;AAEX,IAAA,IAAA,CAAKmC,WAAAA,CAAYnC,IAAIQ,OAAO,CAAA;AAC9B;AAEQE,EAAAA,YAAAA,CAAa0B,KAAAA,EAAsC;AACzD,IAAA,IAAIA,KAAAA,KAAU,IAAA,IAAQA,KAAAA,KAAUC,MAAAA,EAAW;AAEzC,MAAA,OAAO,CAACC,GAAAA,KAAQC,OAAAA,CAAQC,GAAAA,CAAIF,GAAAA,CAAAA;AAC9B,KAAA,MAAA,IAAWF,iBAAiBK,MAAAA,EAAQ;AAClC,MAAA,OAAO,CAACH,GAAAA,KAAQF,KAAAA,CAAMM,KAAAA,CAAMJ,GAAAA,CAAAA;KAC9B,MAAO;AACL,MAAA,OAAOF,KAAAA;AACT;AACF;AAEQD,EAAAA,WAAAA,CAAY3B,OAAAA,EAA6B;AAE/C,IAAA,IAAA,CAAKV,QAAAA,CAAS6C,IAAAA,CACZnC,OAAAA,CAAQoC,KAAAA,CAAM,KAAA,EAAO,CAACC,CAAAA,EAAGC,KAAAA,KAAU,IAAA,CAAKC,UAAAA,CAAWD,KAAAA,CAAAA,EAAQ;MACzDzC,WAAAA,EAAa;AACf,KAAA,CAAA,CAAA;AAIF,IAAA,MAAM2C,mBAAAA,mBAAsB,MAAA,CAAA,OAAOC,IAAAA,EAAWC,IAAAA,KAAAA;AAC5C,MAAA,IAAI,EAAEA,IAAAA,CAAKC,OAAAA,YAAmBC,UAAAA,CAAAA,EAAa;AACzC,QAAA;AACF;AACA,MAAA,IAAI,CAACF,KAAKG,KAAAA,EAAO;AACf,QAAA;AACF;AAEA,MAAA,IAAA,CAAKN,WAAWG,IAAAA,CAAAA;AAChB,MAAA,IAAI,CAAC,IAAA,CAAKI,SAAAA,CAAUJ,IAAAA,CAAAA,EAAO;AACzB,QAAA;AACF;AACA,MAAA,IAAI,CAAC,KAAKvD,OAAAA,EAAS;AACjB,QAAA;AACF;AAEA,MAAA,MAAM2B,SAAAA,GAAYiC,UAAAA,CAAWL,IAAAA,CAAKM,IAAI,CAAA;AACtC,MAAA,MAAM,IAAA,CAAK,aAAalC,SAAAA,CAAAA,CAAW,EAAWmC,IAAAA,CAAK,IAAA,EAAMR,MAAMC,IAAAA,CAAAA;KACjE,EAlB4B,qBAAA,CAAA;AAoB5B,IAAA,IAAA,CAAKpD,QAAAA,CAAS6C,IAAAA,CACZnC,OAAAA,CAAQoC,KAAAA,CACN,CAACE,KAAAA,KAAU;AAAC,MAAA,OAAA;AAAS,MAAA,SAAA;AAAW,MAAA,OAAA;AAAS,MAAA;MAAUY,QAAAA,CAASZ,KAAAA,CAAMU,IAAI,CAAA,EACtER,mBAAAA,EACA;MACE3C,WAAAA,EAAa,KAAA;MACbsD,UAAAA,EAAY,IAAA;AACZC,MAAAA,QAAAA,EAAU,IAAA,CAAKtD;AACjB,KAAA,CAAA,CAAA;AAKJ,IAAA,IAAI,KAAKD,WAAAA,EAAa;AACpB,MAAA,MAAMwD,iBAAAA,mBAAoB,MAAA,CAAA,OAAOZ,IAAAA,EAAWC,IAAAA,KAAAA;AAC1C,QAAA,IAAI,EAAEA,IAAAA,CAAKC,OAAAA,YAAmBC,UAAAA,CAAAA,EAAa;AACzC,UAAA;AACF;AACA,QAAA,IAAIF,IAAAA,CAAKC,OAAAA,CAAQ3C,OAAAA,KAAYA,OAAAA,EAAS;AACpC,UAAA,MAAMwC,mBAAAA,CAAoBC,MAAMC,IAAAA,CAAAA;AAChC,UAAA,IAAA,CAAKf,WAAAA,CAAYe,IAAAA,CAAKC,OAAAA,CAAQ3C,OAAO,CAAA;AACvC;OACF,EAR0B,mBAAA,CAAA;AAU1B,MAAA,IAAA,CAAKV,QAAAA,CAAS6C,KACZnC,OAAAA,CAAQoC,KAAAA,CAAM,CAACE,KAAAA,KAAUA,KAAAA,CAAMU,IAAAA,KAAS,OAAA,EAASK,iBAAAA,EAAmB;QAClExD,WAAAA,EAAa,IAAA;QACbsD,UAAAA,EAAY,IAAA;AACZC,QAAAA,QAAAA,EAAU,IAAA,CAAKtD;AACjB,OAAA,CAAA,CAAA;AAEJ;AACF;AAEQyC,EAAAA,UAAAA,CAAWG,IAAAA,EAAuB;AACxC,IAAA,IAAI,CAACA,IAAAA,CAAKG,KAAAA,EAAOrB,KAAAA,EAAO;AACtB,MAAA;AACF;AACA,IAAA,IAAI,KAAK9B,UAAAA,CAAW4D,GAAAA,CAAIZ,IAAAA,CAAKG,KAAAA,CAAMrB,KAAK,CAAA,EAAG;AACzC,MAAA;AACF;AACA,IAAA,IAAIkB,IAAAA,CAAKG,KAAAA,CAAMU,WAAAA,KAAgBb,IAAAA,CAAKG,MAAMrB,KAAAA,EAAO;AAC/C,MAAA;AACF;AAEA,IAAA,IAAIkB,IAAAA,CAAKG,MAAMU,WAAAA,EAAa;AAC1B,MAAA,MAAMC,OAAAA,GAAU,IAAA,CAAKV,SAAAA,CAAUJ,IAAAA,CAAAA;AAC/B,MAAA,MAAMe,cAAc,IAAA,CAAK/D,UAAAA,CAAWgE,IAAIhB,IAAAA,CAAKG,KAAAA,CAAMU,WAAW,CAAA,IAAK;QACjE9B,QAAAA,EAAU,CAAA;QACVC,QAAAA,EAAU;AACZ,OAAA;AACA,MAAA,IAAA,CAAKhC,UAAAA,CAAW6B,GAAAA,CAAImB,IAAAA,CAAKG,KAAAA,CAAMrB,KAAAA,EAAO;QACpCC,QAAAA,EAAUgC,WAAAA,CAAYhC,QAAAA,IAAY+B,OAAAA,GAAU,CAAA,GAAI,CAAA,CAAA;AAChD9B,QAAAA,QAAAA,EAAU+B,YAAY/B,QAAAA,GAAW;OACnC,CAAA;AACF;AACF;AAEQoB,EAAAA,SAAAA,CAAUJ,IAAAA,EAA0B;AAC1C,IAAA,IAAInD,SAAcmD,IAAAA,CAAKC,OAAAA;AACvB,IAAA,IAAIpD,kBAAkBqD,UAAAA,EAAY;AAChCrD,MAAAA,MAAAA,GAASA,MAAAA,CAAOoE,QAAAA;AAClB;AAEA,IAAA,KAAA,MAAWtE,QAAAA,IAAY,KAAKA,QAAAA,EAAU;AACpC,MAAA,IAAIE,kBAAkBF,QAAAA,EAAU;AAC9B,QAAA,OAAO,KAAA;AACT;AACF;AAEA,IAAA,IAAI,IAAA,CAAKD,QAAAA,CAASgC,MAAAA,KAAW,CAAA,EAAG;AAC9B,MAAA,OAAO,IAAA;AACT;AAEA,IAAA,OAAO,KAAKhC,QAAAA,CAASwE,IAAAA,CAAK,CAACxE,QAAAA,KAAaG,kBAAkBH,QAAAA,CAAAA;AAC5D;AAEQyE,EAAAA,WAAAA,CAAYnB,IAAAA,EAAyB;AAC3C,IAAA,IAAInD,SAAcmD,IAAAA,CAAKC,OAAAA;AACvB,IAAA,IAAIpD,kBAAkBqD,UAAAA,EAAY;AAChCrD,MAAAA,MAAAA,GAASA,MAAAA,CAAOoE,QAAAA;AAClB;AAEA,IAAA,MAAM/C,SAAAA,GAAYrB,OAAO,WAAA,CAAYyD,IAAAA;AACrC,IAAA,IAAIc,UAAAA,GAA4B,IAAA;AAEhC,IAAA,IAAIvE,MAAAA,YAAkBY,SAAAA,IAAaZ,MAAAA,CAAOmD,IAAAA,EAAMM,IAAAA,EAAM;AACpDc,MAAAA,UAAAA,GAAavE,OAAOmD,IAAAA,CAAKM,IAAAA;AAC3B,KAAA,MAAA,IAAW,MAAA,IAAUzD,MAAAA,IAAU,OAAOA,MAAAA,CAAOyD,SAAS,QAAA,EAAU;AAC9Dc,MAAAA,UAAAA,GAAavE,MAAAA,CAAOyD,IAAAA;AACtB;AAEA,IAAA,IAAIrC,MAAAA,GAAS,EAAA;AACb,IAAA,KAAA,MAAW,CAACoD,IAAAA,EAAMC,UAAAA,KAAe,IAAA,CAAKpE,YAAAA,CAAaa,SAAO,EAAI;AAC5D,MAAA,IAAIlB,kBAAkBwE,IAAAA,EAAM;AAC1BpD,QAAAA,MAAAA,GAASqD,UAAAA;AACT,QAAA;AACF;AACF;AAEA,IAAA,MAAMpC,KAAAA,GAAkD;AACtDjB,MAAAA,MAAAA;AACAC,MAAAA,SAAAA;MACAC,YAAAA,EAAciD,UAAAA;AACdhD,MAAAA,SAAAA,EAAW4B,IAAAA,CAAKM;AAClB,KAAA;AAEA,IAAA,OAAO,IAAA,CAAKjD,UAAU6B,KAAAA,CAAAA;AACxB;AAEQqC,EAAAA,YAAAA,CAAavB,IAAAA,EAAyB;AAC5C,IAAA,IAAI,CAACA,KAAKG,KAAAA,EAAO;AACf,MAAA,OAAO,EAAA;AACT;AAEA,IAAA,MAAMqB,MAAAA,GAAS,IAAA,CAAKC,aAAAA,CAAczB,IAAAA,EAAM,MAAA,CAAA,CAAQjB,QAAAA;AAChD,IAAA,MAAM2C,YAAAA,GAAe,IAAA,CAAKD,aAAAA,CAAczB,IAAAA,EAAM,QAAA,CAAA,CAAUjB,QAAAA;AACxD,IAAA,MAAM4C,aAAaH,MAAAA,GAASE,YAAAA;AAE5B,IAAA,IAAIzD,MAAAA,GAAS,EAAA;AACbA,IAAAA,MAAAA,IAAU,IAAA,CAAK2D,MAAAA,CAAOF,YAAAA,GAAe,CAAA,CAAA;AAErC,IAAA,IAAI1B,IAAAA,CAAKM,IAAAA,KAAS,OAAA,IAAWkB,MAAAA,EAAQ;AACnCvD,MAAAA,MAAAA,IAAU,GAAA;AACZ;AAEAA,IAAAA,MAAAA,IAAU,IAAA,CAAK2D,OAAOD,UAAAA,CAAAA;AAEtB,IAAA,IAAI3B,IAAAA,CAAKM,IAAAA,KAAS,OAAA,IAAWrC,MAAAA,IAAUuD,MAAAA,EAAQ;AAC7CvD,MAAAA,MAAAA,IAAU,GAAA;AACZ;AAEA,IAAA,IAAIA,MAAAA,EAAQ;AACVA,MAAAA,MAAAA,GAAS,GAAGA,MAAAA,CAAAA,CAAAA,CAAAA;AACd;AAEA,IAAA,MAAMqC,IAAAA,GAAO,IAAA,CAAKa,WAAAA,CAAYnB,IAAAA,CAAAA;AAC9B,IAAA,OAAO,CAAA,EAAG/B,MAAAA,CAAAA,EAASqC,IAAAA,CAAAA,EAAAA,CAAAA;AACrB;EAEQmB,aAAAA,CAAczB,IAAAA,EAAiBqB,OAA0B,MAAA,EAAoB;AACnF,IAAA,IAAI,CAACrB,KAAKG,KAAAA,EAAO;AACf,MAAA,OAAO;QAAEpB,QAAAA,EAAU,CAAA;QAAGC,QAAAA,EAAU;AAAE,OAAA;AACpC;AAEA,IAAA,MAAMF,KAAAA,GAAQuC,SAAS,QAAA,GAAWrB,IAAAA,CAAKG,MAAMU,WAAAA,IAAe,EAAA,GAAKb,KAAKG,KAAAA,CAAMrB,KAAAA;AAC5E,IAAA,OAAO,IAAA,CAAK9B,UAAAA,CAAWgE,GAAAA,CAAIlC,KAAAA,CAAAA,IAAU;MAAEC,QAAAA,EAAU,CAAA;MAAGC,QAAAA,EAAU;AAAE,KAAA;AAClE;AAEA,EAAA,MAAc6C,cAAcC,KAAAA,EAA6B;AACvD,IAAA,IAAIC,WAAAA,CAAYD,KAAAA,CAAAA,EAAQ;AACtB,MAAA,OAAOE,OAAOF,KAAAA,CAAAA;AAChB;AAEA,IAAA,IAAIA,iBAAiBG,cAAAA,EAAgB;AACnC,MAAA,OAAOH,MAAMI,OAAAA,EAAO;AACtB;AAEA,IAAA,MAAMC,UAAAA,GAAa,MAAMC,UAAAA,CAAWC,SAAAA,CAAUP,KAAAA,CAAAA;AAC9C,IAAA,OAAOQ,IAAAA,CAAKC,SAAAA,CACV,MAAMH,UAAAA,CAAWI,WAAAA,CAAYL,YAAY,EAAA,EAAI,IAAA,CAAA,kBAC5C,CAAA,MAAA;AACC,MAAA,MAAMM,YAAAA,uBAAmBC,GAAAA,CAAI;AAAC,QAAA,SAAA;AAAW,QAAA,UAAA;AAAY,QAAA,SAAA;AAAW,QAAA;AAAY,OAAA,CAAA;AAC5E,MAAA,OAAO,CAACC,KAAKb,MAAAA,KAAAA;AACX,QAAA,IAAIW,YAAAA,CAAa7B,GAAAA,CAAI+B,GAAAA,CAAAA,EAAM;AACzB,UAAA,OAAOxD,MAAAA;AACT;AACA,QAAA,IAAI2C,MAAAA,IAASA,kBAAiBnE,IAAAA,EAAM;AAClC,UAAA,OAAOmE,MAAAA,CAAMxB,IAAAA;AACf;AACA,QAAA,OAAOwB,MAAAA;AACT,OAAA;AACF,KAAA,GAAA,EACA,IAAA,CAAK/E,MAAAA,GAAS,CAAA,GAAIoC,MAAAA,CAAAA;AAEtB;EAEA,MAAcyD,eAAAA,CACZC,SACA7C,IAAAA,EACe;AACf,IAAA,MAAM/B,MAAAA,GAAS,IAAA,CAAKsD,YAAAA,CAAavB,IAAAA,CAAAA;AACjC,IAAA,MAAM8C,OAAAA,GAAU,GAAG7E,MAAAA,CAAAA,EAAS,MAAM,IAAA,CAAK4D,aAAAA,CAAcgB,OAAAA,CAAQ3D,KAAK,CAAA,CAAA,CAAA;AAElE,IAAA,MAAM,IAAA,CAAK5B,OAAAA,CAAQyF,IAAAA,CAAK,OAAA,EAAS;AAC/BD,MAAAA,OAAAA;MACAE,KAAAA,EAAO,IAAA,CAAKvB,cAAczB,IAAAA,CAAAA;MAC1BiD,MAAAA,EAAQ;AAACJ,QAAAA,OAAAA;AAAS7C,QAAAA;;KACpB,CAAA;AAEA,IAAA,IAAA,CAAKnD,OAAOiG,OAAAA,CAAAA;AACd;EAEA,MAAcI,iBAAAA,CACZL,SACA7C,IAAAA,EACe;AACf,IAAA,MAAM/B,MAAAA,GAAS,IAAA,CAAKsD,YAAAA,CAAavB,IAAAA,CAAAA;AAEjC,IAAA,MAAM8C,OAAAA,GAAU,GAAG7E,MAAAA,CAAAA,EAAS,MAAM,IAAA,CAAK4D,aAAAA,CAAcgB,OAAAA,CAAAA,CAAAA,CAAAA;AAErD,IAAA,MAAM,IAAA,CAAKvF,OAAAA,CAAQyF,IAAAA,CAAK,SAAA,EAAW;AACjCD,MAAAA,OAAAA;MACAE,KAAAA,EAAO,IAAA,CAAKvB,cAAczB,IAAAA,CAAAA;MAC1BiD,MAAAA,EAAQ;AAACJ,QAAAA,OAAAA;AAAS7C,QAAAA;;KACpB,CAAA;AAEA,IAAA,IAAA,CAAKnD,OAAOiG,OAAAA,CAAAA;AACd;EAEA,MAAcK,eAAAA,CACZN,SACA7C,IAAAA,EACe;AACf,IAAA,MAAM/B,MAAAA,GAAS,IAAA,CAAKsD,YAAAA,CAAavB,IAAAA,CAAAA;AACjC,IAAA,MAAM8C,OAAAA,GAAU,GAAG7E,MAAAA,CAAAA,EAAS,MAAM,IAAA,CAAK4D,aAAAA,CAAcgB,OAAAA,CAAAA,CAAAA,CAAAA;AAErD,IAAA,MAAM,IAAA,CAAKvF,OAAAA,CAAQyF,IAAAA,CAAK,OAAA,EAAS;AAC/BD,MAAAA,OAAAA;MACAE,KAAAA,EAAO,IAAA,CAAKvB,cAAczB,IAAAA,CAAAA;MAC1BiD,MAAAA,EAAQ;AAACJ,QAAAA,OAAAA;AAAS7C,QAAAA;;KACpB,CAAA;AAEA,IAAA,IAAA,CAAKnD,OAAOiG,OAAAA,CAAAA;AACd;EAEA,MAAcM,gBAAAA,CACZP,SACA7C,IAAAA,EACe;AACf,IAAA,MAAM/B,MAAAA,GAAS,IAAA,CAAKsD,YAAAA,CAAavB,IAAAA,CAAAA;AACjC,IAAA,MAAM8C,OAAAA,GAAU,CAAA,EAAG7E,MAAAA,CAAAA,EAAS,MAAM,IAAA,CAAK4D,aAAAA,CAAcgB,OAAAA,CAAQQ,KAAAA,IAASR,OAAAA,CAAQS,MAAM,CAAA,CAAA,CAAA;AACpF,IAAA,MAAM,IAAA,CAAKhG,OAAAA,CAAQyF,IAAAA,CAAK,QAAA,EAAU;AAChCD,MAAAA,OAAAA;MACAE,KAAAA,EAAO,IAAA,CAAKvB,cAAczB,IAAAA,CAAAA;MAC1BiD,MAAAA,EAAQ;AAACJ,QAAAA,OAAAA;AAAS7C,QAAAA;;KACpB,CAAA;AACF;AACF","file":"trajectory.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { Middleware, RunContext, RunContextCallbacks, RunInstance } from \"@/context.js\";\nimport { Callback, Emitter, EventMeta } from \"@/emitter/emitter.js\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { Logger } from \"@/logger/logger.js\";\nimport { BaseAgent } from \"@/agents/base.js\";\nimport type { AnyConstructable } from \"@/internals/types.js\";\nimport { capitalize } from \"remeda\";\nimport { ChatModel } from \"@/backend/chat.js\";\nimport { Tool } from \"@/tools/base.js\";\nimport { Requirement } from \"@/agents/requirement/requirements/requirement.js\";\nimport { isPrimitive } from \"sequelize/lib/utils\";\nimport type { InferCallbackValue } from \"@/emitter/types.js\";\nimport { Serializer } from \"@/serializer/serializer.js\";\n\n/**\n * Information about how deep the given entity is in the execution tree.\n */\nexport interface TraceLevel {\n /** Relative depth to the included (observed) elements */\n relative: number;\n /** Absolute depth from the root */\n absolute: number;\n}\n\n/**\n * Input for custom formatter function\n */\nexport interface GlobalTrajectoryMiddlewareFormatterInput {\n prefix: string;\n className: string;\n eventName: string;\n instanceName: string | null;\n}\n\nexport interface GlobalTrajectoryMiddlewareCallbacks {\n start: Callback<{\n message: string;\n level: TraceLevel;\n origin: [InferCallbackValue<RunContextCallbacks[\"start\"]>, EventMeta];\n }>;\n success: Callback<{\n message: string;\n level: TraceLevel;\n origin: [InferCallbackValue<RunContextCallbacks[\"success\"]>, EventMeta];\n }>;\n error: Callback<{\n message: string;\n level: TraceLevel;\n origin: [InferCallbackValue<RunContextCallbacks[\"error\"]>, EventMeta];\n }>;\n finish: Callback<{\n message: string;\n level: TraceLevel;\n origin: [InferCallbackValue<RunContextCallbacks[\"finish\"]>, EventMeta];\n }>;\n}\n\ntype OutputTargetFn = (message: string) => void;\ntype OutputTarget = Logger | OutputTargetFn;\n\nexport interface GlobalTrajectoryMiddlewareOptions {\n /** Specify output target: 'console', Logger instance, custom function, or null to disable */\n target?: OutputTarget;\n /** List of classes to include in the trajectory */\n included?: AnyConstructable[];\n /** List of classes to exclude from the trajectory */\n excluded?: AnyConstructable[];\n /** Use pretty formatting for the trajectory */\n pretty?: boolean;\n /** Customize how instances of individual classes should be printed */\n prefixByType?: Map<AnyConstructable, string>;\n /** Enable/Disable the logging */\n enabled?: boolean;\n /** Whether to observe trajectories of nested run contexts */\n matchNested?: boolean;\n /** Defines a priority for registered events */\n emitterPriority?: number;\n /** Custom formatter function */\n formatter?: (input: GlobalTrajectoryMiddlewareFormatterInput) => string;\n}\n\n/**\n * Middleware for capturing and logging execution flow of agents, tools, and models.\n * Provides hierarchical visualization with indentation to show the call stack.\n */\nexport class GlobalTrajectoryMiddleware<T extends RunInstance = any> extends Middleware<T> {\n protected enabled: boolean;\n protected included: AnyConstructable[];\n protected excluded: AnyConstructable[];\n protected cleanups: (() => void)[] = [];\n protected target: (message: string) => void;\n protected ctx: RunContext<T> | null = null;\n protected pretty: boolean;\n protected traceLevel = new Map<string, TraceLevel>();\n protected prefixByType: Map<any, string>;\n protected matchNested: boolean;\n protected emitterPriority: number;\n protected formatter: (input: GlobalTrajectoryMiddlewareFormatterInput) => string;\n public readonly emitter: Emitter<GlobalTrajectoryMiddlewareCallbacks>;\n\n constructor(options: GlobalTrajectoryMiddlewareOptions = {}) {\n super();\n\n this.enabled = options.enabled ?? true;\n this.included = options.included ?? [];\n this.excluded = options.excluded ?? [];\n this.target = this.createTarget(options.target);\n this.pretty = options.pretty ?? false;\n this.matchNested = options.matchNested ?? true;\n this.emitterPriority = options.emitterPriority ?? -1; // run later\n\n // Default prefixes\n this.prefixByType = new Map([\n [BaseAgent, \"🤖 \"],\n [ChatModel, \"💬 \"],\n [Tool, \"🛠️ \"],\n [Requirement, \"🔎 \"],\n ...((options.prefixByType ? Array.from(options.prefixByType.entries()) : []) as any),\n ]);\n\n this.formatter =\n options.formatter ??\n ((x) => `${x.prefix}${x.className}[${x.instanceName || x.className}][${x.eventName}]`);\n\n this.emitter = Emitter.root.child<GlobalTrajectoryMiddlewareCallbacks>({\n namespace: [\"middleware\", \"globalTrajectory\"],\n });\n }\n\n /**\n * Bind the middleware to a run context\n */\n bind(ctx: RunContext<T>): void {\n // Cleanup previous bindings\n while (this.cleanups.length > 0) {\n this.cleanups.pop()!();\n }\n\n this.traceLevel.clear();\n this.traceLevel.set(ctx.runId, { relative: 0, absolute: 0 });\n this.ctx = ctx;\n\n this.bindEmitter(ctx.emitter);\n }\n\n private createTarget(input?: OutputTarget): OutputTargetFn {\n if (input === null || input === undefined) {\n // eslint-disable-next-line no-console\n return (msg) => console.log(msg);\n } else if (input instanceof Logger) {\n return (msg) => input.debug(msg);\n } else {\n return input;\n }\n }\n\n private bindEmitter(emitter: Emitter<any>): void {\n // Track all events for trace ID logging\n this.cleanups.push(\n emitter.match(\"*.*\", (_, event) => this.logTraceId(event), {\n matchNested: true,\n }),\n );\n\n // Handle top-level events\n const handleTopLevelEvent = async (data: any, meta: EventMeta) => {\n if (!(meta.creator instanceof RunContext)) {\n return;\n }\n if (!meta.trace) {\n return;\n }\n\n this.logTraceId(meta);\n if (!this.isAllowed(meta)) {\n return;\n }\n if (!this.enabled) {\n return;\n }\n\n const eventName = capitalize(meta.name) as \"Start\" | \"Success\" | \"Error\" | \"Finish\";\n await this[`onInternal${eventName}` as const].call(this, data, meta);\n };\n\n this.cleanups.push(\n emitter.match(\n (event) => [\"start\", \"success\", \"error\", \"finish\"].includes(event.name),\n handleTopLevelEvent,\n {\n matchNested: false,\n isBlocking: true,\n priority: this.emitterPriority,\n },\n ),\n );\n\n // Handle nested events if enabled\n if (this.matchNested) {\n const handleNestedEvent = async (data: any, meta: EventMeta) => {\n if (!(meta.creator instanceof RunContext)) {\n return;\n }\n if (meta.creator.emitter !== emitter) {\n await handleTopLevelEvent(data, meta);\n this.bindEmitter(meta.creator.emitter);\n }\n };\n\n this.cleanups.push(\n emitter.match((event) => event.name === \"start\", handleNestedEvent, {\n matchNested: true,\n isBlocking: true,\n priority: this.emitterPriority,\n }),\n );\n }\n }\n\n private logTraceId(meta: EventMeta): void {\n if (!meta.trace?.runId) {\n return;\n }\n if (this.traceLevel.has(meta.trace.runId)) {\n return;\n }\n if (meta.trace.parentRunId === meta.trace.runId) {\n return;\n }\n\n if (meta.trace.parentRunId) {\n const allowed = this.isAllowed(meta);\n const parentTrace = this.traceLevel.get(meta.trace.parentRunId) ?? {\n relative: 0,\n absolute: 0,\n };\n this.traceLevel.set(meta.trace.runId, {\n relative: parentTrace.relative + (allowed ? 1 : 0),\n absolute: parentTrace.absolute + 1,\n });\n }\n }\n\n private isAllowed(meta: EventMeta): boolean {\n let target: any = meta.creator;\n if (target instanceof RunContext) {\n target = target.instance;\n }\n\n for (const excluded of this.excluded) {\n if (target instanceof excluded) {\n return false;\n }\n }\n\n if (this.included.length === 0) {\n return true;\n }\n\n return this.included.some((included) => target instanceof included);\n }\n\n private extractName(meta: EventMeta): string {\n let target: any = meta.creator;\n if (target instanceof RunContext) {\n target = target.instance;\n }\n\n const className = target.constructor.name;\n let targetName: string | null = null;\n\n if (target instanceof BaseAgent && target.meta?.name) {\n targetName = target.meta.name;\n } else if (\"name\" in target && typeof target.name === \"string\") {\n targetName = target.name;\n }\n\n let prefix = \"\";\n for (const [type, typePrefix] of this.prefixByType.entries()) {\n if (target instanceof type) {\n prefix = typePrefix;\n break;\n }\n }\n\n const input: GlobalTrajectoryMiddlewareFormatterInput = {\n prefix,\n className,\n instanceName: targetName,\n eventName: meta.name,\n };\n\n return this.formatter(input);\n }\n\n private formatPrefix(meta: EventMeta): string {\n if (!meta.trace) {\n return \"\";\n }\n\n const indent = this.getTraceLevel(meta, \"self\").relative;\n const indentParent = this.getTraceLevel(meta, \"parent\").relative;\n const indentDiff = indent - indentParent;\n\n let prefix = \"\";\n prefix += \" \".repeat(indentParent * 2);\n\n if (meta.name !== \"start\" && indent) {\n prefix += \"<\";\n }\n\n prefix += \"--\".repeat(indentDiff);\n\n if (meta.name === \"start\" && prefix && indent) {\n prefix += \">\";\n }\n\n if (prefix) {\n prefix = `${prefix} `;\n }\n\n const name = this.extractName(meta);\n return `${prefix}${name}: `;\n }\n\n private getTraceLevel(meta: EventMeta, type: \"self\" | \"parent\" = \"self\"): TraceLevel {\n if (!meta.trace) {\n return { relative: 0, absolute: 0 };\n }\n\n const runId = type === \"parent\" ? meta.trace.parentRunId || \"\" : meta.trace.runId;\n return this.traceLevel.get(runId) ?? { relative: 0, absolute: 0 };\n }\n\n private async formatPayload(value: any): Promise<string> {\n if (isPrimitive(value)) {\n return String(value);\n }\n\n if (value instanceof FrameworkError) {\n return value.explain();\n }\n\n const serialized = await Serializer.serialize(value);\n return JSON.stringify(\n await Serializer.deserialize(serialized, [], true),\n (() => {\n const excludedKeys = new Set([\"emitter\", \"cleanups\", \"creator\", \"listeners\"]);\n return (key, value) => {\n if (excludedKeys.has(key)) {\n return undefined;\n }\n if (value && value instanceof Tool) {\n return value.name;\n }\n return value;\n };\n })(),\n this.pretty ? 2 : undefined,\n );\n }\n\n private async onInternalStart(\n payload: InferCallbackValue<RunContextCallbacks[\"start\"]>,\n meta: EventMeta,\n ): Promise<void> {\n const prefix = this.formatPrefix(meta);\n const message = `${prefix}${await this.formatPayload(payload.input)}`;\n\n await this.emitter.emit(\"start\", {\n message,\n level: this.getTraceLevel(meta),\n origin: [payload, meta],\n });\n\n this.target(message);\n }\n\n private async onInternalSuccess(\n payload: InferCallbackValue<RunContextCallbacks[\"success\"]>,\n meta: EventMeta,\n ): Promise<void> {\n const prefix = this.formatPrefix(meta);\n // TODO: change to payload.output once available\n const message = `${prefix}${await this.formatPayload(payload)}`;\n\n await this.emitter.emit(\"success\", {\n message,\n level: this.getTraceLevel(meta),\n origin: [payload, meta],\n });\n\n this.target(message);\n }\n\n private async onInternalError(\n payload: InferCallbackValue<RunContextCallbacks[\"error\"]>,\n meta: EventMeta,\n ): Promise<void> {\n const prefix = this.formatPrefix(meta);\n const message = `${prefix}${await this.formatPayload(payload)}`;\n\n await this.emitter.emit(\"error\", {\n message,\n level: this.getTraceLevel(meta),\n origin: [payload, meta],\n });\n\n this.target(message);\n }\n\n private async onInternalFinish(\n payload: InferCallbackValue<RunContextCallbacks[\"finish\"]>,\n meta: EventMeta,\n ): Promise<void> {\n const prefix = this.formatPrefix(meta);\n const message = `${prefix}${await this.formatPayload(payload.error || payload.output)}`;\n await this.emitter.emit(\"finish\", {\n message,\n level: this.getTraceLevel(meta),\n origin: [payload, meta],\n });\n }\n}\n"]}
1
+ {"version":3,"sources":["../../src/middleware/trajectory.ts"],"names":["GlobalTrajectoryMiddleware","Middleware","enabled","included","excluded","cleanups","target","ctx","pretty","traceLevel","Map","prefixByType","matchNested","emitterPriority","formatter","emitter","options","createTarget","BaseAgent","ChatModel","Tool","Requirement","Array","from","entries","x","prefix","className","instanceName","eventName","Emitter","root","child","namespace","bind","length","pop","clear","set","runId","relative","absolute","bindEmitter","input","undefined","msg","console","log","Logger","debug","push","match","_","event","logTraceId","handleTopLevelEvent","data","meta","creator","RunContext","trace","isAllowed","capitalize","name","call","matchInternalEvent","isBlocking","priority","handleNestedEvent","has","parentRunId","allowed","parentTrace","get","instance","some","extractName","targetName","type","typePrefix","formatPrefix","indent","getTraceLevel","indentParent","indentDiff","repeat","formatPayload","value","isPrimitive","String","FrameworkError","explain","serialized","Serializer","serialize","JSON","stringify","deserialize","excludedKeys","Set","key","onInternalStart","payload","message","emit","level","origin","onInternalSuccess","onInternalError","onInternalFinish","error","output","names","includes","getProp","context"],"mappings":";;;;;;;;;;;;;;;AA2FO,MAAMA,mCAAgEC,UAAAA,CAAAA;EA3F7E;;;AA4FYC,EAAAA,OAAAA;AACAC,EAAAA,QAAAA;AACAC,EAAAA,QAAAA;AACAC,EAAAA,QAAAA,GAA2B,EAAA;AAC3BC,EAAAA,MAAAA;EACAC,GAAAA,GAA4B,IAAA;AAC5BC,EAAAA,MAAAA;AACAC,EAAAA,UAAAA,uBAAiBC,GAAAA,EAAAA;AACjBC,EAAAA,YAAAA;AACAC,EAAAA,WAAAA;AACAC,EAAAA,eAAAA;AACAC,EAAAA,SAAAA;AACMC,EAAAA,OAAAA;EAEhB,WAAA,CAAYC,OAAAA,GAA6C,EAAC,EAAG;AAC3D,IAAA,KAAA,EAAK;AAEL,IAAA,IAAA,CAAKd,OAAAA,GAAUc,QAAQd,OAAAA,IAAW,IAAA;AAClC,IAAA,IAAA,CAAKC,QAAAA,GAAWa,OAAAA,CAAQb,QAAAA,IAAY,EAAA;AACpC,IAAA,IAAA,CAAKC,QAAAA,GAAWY,OAAAA,CAAQZ,QAAAA,IAAY,EAAA;AACpC,IAAA,IAAA,CAAKE,MAAAA,GAAS,IAAA,CAAKW,YAAAA,CAAaD,OAAAA,CAAQV,MAAM,CAAA;AAC9C,IAAA,IAAA,CAAKE,MAAAA,GAASQ,QAAQR,MAAAA,IAAU,KAAA;AAChC,IAAA,IAAA,CAAKI,WAAAA,GAAcI,QAAQJ,WAAAA,IAAe,IAAA;AAC1C,IAAA,IAAA,CAAKC,eAAAA,GAAkBG,QAAQH,eAAAA,IAAmB,EAAA;AAGlD,IAAA,IAAA,CAAKF,YAAAA,GAAe,IAAID,GAAAA,CAAI;AAC1B,MAAA;AAACQ,QAAAA,SAAAA;AAAW,QAAA;;AACZ,MAAA;AAACC,QAAAA,SAAAA;AAAW,QAAA;;AACZ,MAAA;AAACC,QAAAA,IAAAA;AAAM,QAAA;;AACP,MAAA;AAACC,QAAAA,WAAAA;AAAa,QAAA;;SACTL,OAAAA,CAAQL,YAAAA,GAAeW,MAAMC,IAAAA,CAAKP,OAAAA,CAAQL,aAAaa,OAAAA,EAAO,IAAM;AAC1E,KAAA,CAAA;AAED,IAAA,IAAA,CAAKV,YACHE,OAAAA,CAAQF,SAAAA,KACP,CAACW,CAAAA,KAAM,CAAA,EAAGA,EAAEC,MAAM,CAAA,EAAGD,CAAAA,CAAEE,SAAS,IAAIF,CAAAA,CAAEG,YAAAA,IAAgBH,EAAEE,SAAS,CAAA,EAAA,EAAKF,EAAEI,SAAS,CAAA,CAAA,CAAA,CAAA;AAEpF,IAAA,IAAA,CAAKd,OAAAA,GAAUe,OAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAA2C;MACrEC,SAAAA,EAAW;AAAC,QAAA,YAAA;AAAc,QAAA;;KAC5B,CAAA;AACF;;;;AAKAC,EAAAA,IAAAA,CAAK3B,GAAAA,EAA0B;AAE7B,IAAA,OAAO,IAAA,CAAKF,QAAAA,CAAS8B,MAAAA,GAAS,CAAA,EAAG;AAC/B,MAAA,IAAA,CAAK9B,QAAAA,CAAS+B,KAAG,EAAA;AACnB;AAEA,IAAA,IAAA,CAAK3B,WAAW4B,KAAAA,EAAK;AACrB,IAAA,IAAA,CAAK5B,UAAAA,CAAW6B,GAAAA,CAAI/B,GAAAA,CAAIgC,KAAAA,EAAO;MAAEC,QAAAA,EAAU,CAAA;MAAGC,QAAAA,EAAU;KAAE,CAAA;AAC1D,IAAA,IAAA,CAAKlC,GAAAA,GAAMA,GAAAA;AAEX,IAAA,IAAA,CAAKmC,WAAAA,CAAYnC,IAAIQ,OAAO,CAAA;AAC9B;AAEQE,EAAAA,YAAAA,CAAa0B,KAAAA,EAAsC;AACzD,IAAA,IAAIA,KAAAA,KAAU,IAAA,IAAQA,KAAAA,KAAUC,MAAAA,EAAW;AAEzC,MAAA,OAAO,CAACC,GAAAA,KAAQC,OAAAA,CAAQC,GAAAA,CAAIF,GAAAA,CAAAA;AAC9B,KAAA,MAAA,IAAWF,iBAAiBK,MAAAA,EAAQ;AAClC,MAAA,OAAO,CAACH,GAAAA,KAAQF,KAAAA,CAAMM,KAAAA,CAAMJ,GAAAA,CAAAA;KAC9B,MAAO;AACL,MAAA,OAAOF,KAAAA;AACT;AACF;AAEQD,EAAAA,WAAAA,CAAY3B,OAAAA,EAA6B;AAE/C,IAAA,IAAA,CAAKV,QAAAA,CAAS6C,IAAAA,CACZnC,OAAAA,CAAQoC,KAAAA,CAAM,KAAA,EAAO,CAACC,CAAAA,EAAGC,KAAAA,KAAU,IAAA,CAAKC,UAAAA,CAAWD,KAAAA,CAAAA,EAAQ;MACzDzC,WAAAA,EAAa;AACf,KAAA,CAAA,CAAA;AAIF,IAAA,MAAM2C,mBAAAA,mBAAsB,MAAA,CAAA,OAAOC,IAAAA,EAAWC,IAAAA,KAAAA;AAC5C,MAAA,IAAI,EAAEA,IAAAA,CAAKC,OAAAA,YAAmBC,UAAAA,CAAAA,EAAa;AACzC,QAAA;AACF;AACA,MAAA,IAAI,CAACF,KAAKG,KAAAA,EAAO;AACf,QAAA;AACF;AAEA,MAAA,IAAA,CAAKN,WAAWG,IAAAA,CAAAA;AAChB,MAAA,IAAI,CAAC,IAAA,CAAKI,SAAAA,CAAUJ,IAAAA,CAAAA,EAAO;AACzB,QAAA;AACF;AACA,MAAA,IAAI,CAAC,KAAKvD,OAAAA,EAAS;AACjB,QAAA;AACF;AAEA,MAAA,MAAM2B,SAAAA,GAAYiC,UAAAA,CAAWL,IAAAA,CAAKM,IAAI,CAAA;AACtC,MAAA,MAAM,IAAA,CAAK,aAAalC,SAAAA,CAAAA,CAAW,EAAWmC,IAAAA,CAAK,IAAA,EAAMR,MAAMC,IAAAA,CAAAA;KACjE,EAlB4B,qBAAA,CAAA;AAoB5B,IAAA,IAAA,CAAKpD,QAAAA,CAAS6C,IAAAA,CACZnC,OAAAA,CAAQoC,KAAAA,CACNc,kBAAAA,CAAmB;AAAC,MAAA,OAAA;AAAS,MAAA,SAAA;AAAW,MAAA,OAAA;AAAS,MAAA;AAAS,KAAA,CAAA,EAC1DV,mBAAAA,EACA;MACE3C,WAAAA,EAAa,KAAA;MACbsD,UAAAA,EAAY,IAAA;AACZC,MAAAA,QAAAA,EAAU,IAAA,CAAKtD;AACjB,KAAA,CAAA,CAAA;AAKJ,IAAA,IAAI,KAAKD,WAAAA,EAAa;AACpB,MAAA,MAAMwD,iBAAAA,mBAAoB,MAAA,CAAA,OAAOZ,IAAAA,EAAWC,IAAAA,KAAAA;AAC1C,QAAA,IAAI,EAAEA,IAAAA,CAAKC,OAAAA,YAAmBC,UAAAA,CAAAA,EAAa;AACzC,UAAA;AACF;AACA,QAAA,IAAIF,IAAAA,CAAKC,OAAAA,CAAQ3C,OAAAA,KAAYA,OAAAA,EAAS;AACpC,UAAA,MAAMwC,mBAAAA,CAAoBC,MAAMC,IAAAA,CAAAA;AAChC,UAAA,IAAA,CAAKf,WAAAA,CAAYe,IAAAA,CAAKC,OAAAA,CAAQ3C,OAAO,CAAA;AACvC;OACF,EAR0B,mBAAA,CAAA;AAU1B,MAAA,IAAA,CAAKV,QAAAA,CAAS6C,IAAAA,CACZnC,OAAAA,CAAQoC,KAAAA,CAAMc,kBAAAA,CAAmB;AAAC,QAAA;AAAQ,OAAA,CAAA,EAAGG,iBAAAA,EAAmB;QAC9DxD,WAAAA,EAAa,IAAA;QACbsD,UAAAA,EAAY,IAAA;AACZC,QAAAA,QAAAA,EAAU,IAAA,CAAKtD;AACjB,OAAA,CAAA,CAAA;AAEJ;AACF;AAEQyC,EAAAA,UAAAA,CAAWG,IAAAA,EAAuB;AACxC,IAAA,IAAI,CAACA,IAAAA,CAAKG,KAAAA,EAAOrB,KAAAA,EAAO;AACtB,MAAA;AACF;AACA,IAAA,IAAI,KAAK9B,UAAAA,CAAW4D,GAAAA,CAAIZ,IAAAA,CAAKG,KAAAA,CAAMrB,KAAK,CAAA,EAAG;AACzC,MAAA;AACF;AACA,IAAA,IAAIkB,IAAAA,CAAKG,KAAAA,CAAMU,WAAAA,KAAgBb,IAAAA,CAAKG,MAAMrB,KAAAA,EAAO;AAC/C,MAAA;AACF;AAEA,IAAA,IAAIkB,IAAAA,CAAKG,MAAMU,WAAAA,EAAa;AAC1B,MAAA,MAAMC,OAAAA,GAAU,IAAA,CAAKV,SAAAA,CAAUJ,IAAAA,CAAAA;AAC/B,MAAA,MAAMe,cAAc,IAAA,CAAK/D,UAAAA,CAAWgE,IAAIhB,IAAAA,CAAKG,KAAAA,CAAMU,WAAW,CAAA,IAAK;QACjE9B,QAAAA,EAAU,CAAA;QACVC,QAAAA,EAAU;AACZ,OAAA;AACA,MAAA,IAAA,CAAKhC,UAAAA,CAAW6B,GAAAA,CAAImB,IAAAA,CAAKG,KAAAA,CAAMrB,KAAAA,EAAO;QACpCC,QAAAA,EAAUgC,WAAAA,CAAYhC,QAAAA,IAAY+B,OAAAA,GAAU,CAAA,GAAI,CAAA,CAAA;AAChD9B,QAAAA,QAAAA,EAAU+B,YAAY/B,QAAAA,GAAW;OACnC,CAAA;AACF;AACF;AAEQoB,EAAAA,SAAAA,CAAUJ,IAAAA,EAA0B;AAC1C,IAAA,IAAInD,SAAcmD,IAAAA,CAAKC,OAAAA;AACvB,IAAA,IAAIpD,kBAAkBqD,UAAAA,EAAY;AAChCrD,MAAAA,MAAAA,GAASA,MAAAA,CAAOoE,QAAAA;AAClB;AAEA,IAAA,KAAA,MAAWtE,QAAAA,IAAY,KAAKA,QAAAA,EAAU;AACpC,MAAA,IAAIE,kBAAkBF,QAAAA,EAAU;AAC9B,QAAA,OAAO,KAAA;AACT;AACF;AAEA,IAAA,IAAI,IAAA,CAAKD,QAAAA,CAASgC,MAAAA,KAAW,CAAA,EAAG;AAC9B,MAAA,OAAO,IAAA;AACT;AAEA,IAAA,OAAO,KAAKhC,QAAAA,CAASwE,IAAAA,CAAK,CAACxE,QAAAA,KAAaG,kBAAkBH,QAAAA,CAAAA;AAC5D;AAEQyE,EAAAA,WAAAA,CAAYnB,IAAAA,EAAyB;AAC3C,IAAA,IAAInD,SAAcmD,IAAAA,CAAKC,OAAAA;AACvB,IAAA,IAAIpD,kBAAkBqD,UAAAA,EAAY;AAChCrD,MAAAA,MAAAA,GAASA,MAAAA,CAAOoE,QAAAA;AAClB;AAEA,IAAA,MAAM/C,SAAAA,GAAYrB,OAAO,WAAA,CAAYyD,IAAAA;AACrC,IAAA,IAAIc,UAAAA,GAA4B,IAAA;AAEhC,IAAA,IAAIvE,MAAAA,YAAkBY,SAAAA,IAAaZ,MAAAA,CAAOmD,IAAAA,EAAMM,IAAAA,EAAM;AACpDc,MAAAA,UAAAA,GAAavE,OAAOmD,IAAAA,CAAKM,IAAAA;AAC3B,KAAA,MAAA,IAAW,MAAA,IAAUzD,MAAAA,IAAU,OAAOA,MAAAA,CAAOyD,SAAS,QAAA,EAAU;AAC9Dc,MAAAA,UAAAA,GAAavE,MAAAA,CAAOyD,IAAAA;AACtB;AAEA,IAAA,IAAIrC,MAAAA,GAAS,EAAA;AACb,IAAA,KAAA,MAAW,CAACoD,IAAAA,EAAMC,UAAAA,KAAe,IAAA,CAAKpE,YAAAA,CAAaa,SAAO,EAAI;AAC5D,MAAA,IAAIlB,kBAAkBwE,IAAAA,EAAM;AAC1BpD,QAAAA,MAAAA,GAASqD,UAAAA;AACT,QAAA;AACF;AACF;AAEA,IAAA,MAAMpC,KAAAA,GAAkD;AACtDjB,MAAAA,MAAAA;AACAC,MAAAA,SAAAA;MACAC,YAAAA,EAAciD,UAAAA;AACdhD,MAAAA,SAAAA,EAAW4B,IAAAA,CAAKM;AAClB,KAAA;AAEA,IAAA,OAAO,IAAA,CAAKjD,UAAU6B,KAAAA,CAAAA;AACxB;AAEQqC,EAAAA,YAAAA,CAAavB,IAAAA,EAAyB;AAC5C,IAAA,IAAI,CAACA,KAAKG,KAAAA,EAAO;AACf,MAAA,OAAO,EAAA;AACT;AAEA,IAAA,MAAMqB,MAAAA,GAAS,IAAA,CAAKC,aAAAA,CAAczB,IAAAA,EAAM,MAAA,CAAA,CAAQjB,QAAAA;AAChD,IAAA,MAAM2C,YAAAA,GAAe,IAAA,CAAKD,aAAAA,CAAczB,IAAAA,EAAM,QAAA,CAAA,CAAUjB,QAAAA;AACxD,IAAA,MAAM4C,aAAaH,MAAAA,GAASE,YAAAA;AAE5B,IAAA,IAAIzD,MAAAA,GAAS,EAAA;AACbA,IAAAA,MAAAA,IAAU,IAAA,CAAK2D,MAAAA,CAAOF,YAAAA,GAAe,CAAA,CAAA;AAErC,IAAA,IAAI1B,IAAAA,CAAKM,IAAAA,KAAS,OAAA,IAAWkB,MAAAA,EAAQ;AACnCvD,MAAAA,MAAAA,IAAU,GAAA;AACZ;AAEAA,IAAAA,MAAAA,IAAU,IAAA,CAAK2D,OAAOD,UAAAA,CAAAA;AAEtB,IAAA,IAAI3B,IAAAA,CAAKM,IAAAA,KAAS,OAAA,IAAWrC,MAAAA,IAAUuD,MAAAA,EAAQ;AAC7CvD,MAAAA,MAAAA,IAAU,GAAA;AACZ;AAEA,IAAA,IAAIA,MAAAA,EAAQ;AACVA,MAAAA,MAAAA,GAAS,GAAGA,MAAAA,CAAAA,CAAAA,CAAAA;AACd;AAEA,IAAA,MAAMqC,IAAAA,GAAO,IAAA,CAAKa,WAAAA,CAAYnB,IAAAA,CAAAA;AAC9B,IAAA,OAAO,CAAA,EAAG/B,MAAAA,CAAAA,EAASqC,IAAAA,CAAAA,EAAAA,CAAAA;AACrB;EAEQmB,aAAAA,CAAczB,IAAAA,EAAiBqB,OAA0B,MAAA,EAAoB;AACnF,IAAA,IAAI,CAACrB,KAAKG,KAAAA,EAAO;AACf,MAAA,OAAO;QAAEpB,QAAAA,EAAU,CAAA;QAAGC,QAAAA,EAAU;AAAE,OAAA;AACpC;AAEA,IAAA,MAAMF,KAAAA,GAAQuC,SAAS,QAAA,GAAWrB,IAAAA,CAAKG,MAAMU,WAAAA,IAAe,EAAA,GAAKb,KAAKG,KAAAA,CAAMrB,KAAAA;AAC5E,IAAA,OAAO,IAAA,CAAK9B,UAAAA,CAAWgE,GAAAA,CAAIlC,KAAAA,CAAAA,IAAU;MAAEC,QAAAA,EAAU,CAAA;MAAGC,QAAAA,EAAU;AAAE,KAAA;AAClE;AAEA,EAAA,MAAc6C,cAAcC,KAAAA,EAA6B;AACvD,IAAA,IAAIC,WAAAA,CAAYD,KAAAA,CAAAA,EAAQ;AACtB,MAAA,OAAOE,OAAOF,KAAAA,CAAAA;AAChB;AAEA,IAAA,IAAIA,iBAAiBG,cAAAA,EAAgB;AACnC,MAAA,OAAOH,MAAMI,OAAAA,EAAO;AACtB;AAEA,IAAA,MAAMC,UAAAA,GAAa,MAAMC,UAAAA,CAAWC,SAAAA,CAAUP,KAAAA,CAAAA;AAC9C,IAAA,OAAOQ,IAAAA,CAAKC,SAAAA,CACV,MAAMH,UAAAA,CAAWI,WAAAA,CAAYL,YAAY,EAAA,EAAI,IAAA,CAAA,kBAC5C,CAAA,MAAA;AACC,MAAA,MAAMM,YAAAA,uBAAmBC,GAAAA,CAAI;AAAC,QAAA,SAAA;AAAW,QAAA,UAAA;AAAY,QAAA,SAAA;AAAW,QAAA;AAAY,OAAA,CAAA;AAC5E,MAAA,OAAO,CAACC,KAAKb,MAAAA,KAAAA;AACX,QAAA,IAAIW,YAAAA,CAAa7B,GAAAA,CAAI+B,GAAAA,CAAAA,EAAM;AACzB,UAAA,OAAOxD,MAAAA;AACT;AACA,QAAA,IAAI2C,MAAAA,IAASA,kBAAiBnE,IAAAA,EAAM;AAClC,UAAA,OAAOmE,MAAAA,CAAMxB,IAAAA;AACf;AACA,QAAA,OAAOwB,MAAAA;AACT,OAAA;AACF,KAAA,GAAA,EACA,IAAA,CAAK/E,MAAAA,GAAS,CAAA,GAAIoC,MAAAA,CAAAA;AAEtB;EAEA,MAAcyD,eAAAA,CACZC,SACA7C,IAAAA,EACe;AACf,IAAA,MAAM/B,MAAAA,GAAS,IAAA,CAAKsD,YAAAA,CAAavB,IAAAA,CAAAA;AACjC,IAAA,MAAM8C,OAAAA,GAAU,GAAG7E,MAAAA,CAAAA,EAAS,MAAM,IAAA,CAAK4D,aAAAA,CAAcgB,OAAAA,CAAQ3D,KAAK,CAAA,CAAA,CAAA;AAElE,IAAA,MAAM,IAAA,CAAK5B,OAAAA,CAAQyF,IAAAA,CAAK,OAAA,EAAS;AAC/BD,MAAAA,OAAAA;MACAE,KAAAA,EAAO,IAAA,CAAKvB,cAAczB,IAAAA,CAAAA;MAC1BiD,MAAAA,EAAQ;AAACJ,QAAAA,OAAAA;AAAS7C,QAAAA;;KACpB,CAAA;AAEA,IAAA,IAAA,CAAKnD,OAAOiG,OAAAA,CAAAA;AACd;EAEA,MAAcI,iBAAAA,CACZL,SACA7C,IAAAA,EACe;AACf,IAAA,MAAM/B,MAAAA,GAAS,IAAA,CAAKsD,YAAAA,CAAavB,IAAAA,CAAAA;AAEjC,IAAA,MAAM8C,OAAAA,GAAU,GAAG7E,MAAAA,CAAAA,EAAS,MAAM,IAAA,CAAK4D,aAAAA,CAAcgB,OAAAA,CAAAA,CAAAA,CAAAA;AAErD,IAAA,MAAM,IAAA,CAAKvF,OAAAA,CAAQyF,IAAAA,CAAK,SAAA,EAAW;AACjCD,MAAAA,OAAAA;MACAE,KAAAA,EAAO,IAAA,CAAKvB,cAAczB,IAAAA,CAAAA;MAC1BiD,MAAAA,EAAQ;AAACJ,QAAAA,OAAAA;AAAS7C,QAAAA;;KACpB,CAAA;AAEA,IAAA,IAAA,CAAKnD,OAAOiG,OAAAA,CAAAA;AACd;EAEA,MAAcK,eAAAA,CACZN,SACA7C,IAAAA,EACe;AACf,IAAA,MAAM/B,MAAAA,GAAS,IAAA,CAAKsD,YAAAA,CAAavB,IAAAA,CAAAA;AACjC,IAAA,MAAM8C,OAAAA,GAAU,GAAG7E,MAAAA,CAAAA,EAAS,MAAM,IAAA,CAAK4D,aAAAA,CAAcgB,OAAAA,CAAAA,CAAAA,CAAAA;AAErD,IAAA,MAAM,IAAA,CAAKvF,OAAAA,CAAQyF,IAAAA,CAAK,OAAA,EAAS;AAC/BD,MAAAA,OAAAA;MACAE,KAAAA,EAAO,IAAA,CAAKvB,cAAczB,IAAAA,CAAAA;MAC1BiD,MAAAA,EAAQ;AAACJ,QAAAA,OAAAA;AAAS7C,QAAAA;;KACpB,CAAA;AAEA,IAAA,IAAA,CAAKnD,OAAOiG,OAAAA,CAAAA;AACd;EAEA,MAAcM,gBAAAA,CACZP,SACA7C,IAAAA,EACe;AACf,IAAA,MAAM/B,MAAAA,GAAS,IAAA,CAAKsD,YAAAA,CAAavB,IAAAA,CAAAA;AACjC,IAAA,MAAM8C,OAAAA,GAAU,CAAA,EAAG7E,MAAAA,CAAAA,EAAS,MAAM,IAAA,CAAK4D,aAAAA,CAAcgB,OAAAA,CAAQQ,KAAAA,IAASR,OAAAA,CAAQS,MAAM,CAAA,CAAA,CAAA;AACpF,IAAA,MAAM,IAAA,CAAKhG,OAAAA,CAAQyF,IAAAA,CAAK,QAAA,EAAU;AAChCD,MAAAA,OAAAA;MACAE,KAAAA,EAAO,IAAA,CAAKvB,cAAczB,IAAAA,CAAAA;MAC1BiD,MAAAA,EAAQ;AAACJ,QAAAA,OAAAA;AAAS7C,QAAAA;;KACpB,CAAA;AACF;AACF;AAEA,SAASQ,mBAAmB+C,KAAAA,EAAoC;AAC9D,EAAA,OAAO,CAACvD,SACNuD,KAAAA,CAAMC,QAAAA,CAASxD,KAAKM,IAAI,CAAA,IACxBmD,OAAAA,CAAQzD,IAAAA,CAAK0D,OAAAA,EAAS;AAAC,IAAA;KAAa,KAAA,CAAA;AACxC;AAJSlD,MAAAA,CAAAA,kBAAAA,EAAAA,oBAAAA,CAAAA","file":"trajectory.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { Middleware, RunContext, RunContextCallbacks, RunInstance } from \"@/context.js\";\nimport { Callback, Emitter, EventMeta } from \"@/emitter/emitter.js\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { Logger } from \"@/logger/logger.js\";\nimport { BaseAgent } from \"@/agents/base.js\";\nimport type { AnyConstructable } from \"@/internals/types.js\";\nimport { capitalize } from \"remeda\";\nimport { ChatModel } from \"@/backend/chat.js\";\nimport { Tool } from \"@/tools/base.js\";\nimport { Requirement } from \"@/agents/requirement/requirements/requirement.js\";\nimport type { InferCallbackValue } from \"@/emitter/types.js\";\nimport { Serializer } from \"@/serializer/serializer.js\";\nimport { isPrimitive } from \"@/internals/helpers/guards.js\";\nimport { getProp } from \"@/internals/helpers/object.js\";\n\n/**\n * Information about how deep the given entity is in the execution tree.\n */\nexport interface TraceLevel {\n /** Relative depth to the included (observed) elements */\n relative: number;\n /** Absolute depth from the root */\n absolute: number;\n}\n\n/**\n * Input for custom formatter function\n */\nexport interface GlobalTrajectoryMiddlewareFormatterInput {\n prefix: string;\n className: string;\n eventName: string;\n instanceName: string | null;\n}\n\nexport interface GlobalTrajectoryMiddlewareCallbacks {\n start: Callback<{\n message: string;\n level: TraceLevel;\n origin: [InferCallbackValue<RunContextCallbacks[\"start\"]>, EventMeta];\n }>;\n success: Callback<{\n message: string;\n level: TraceLevel;\n origin: [InferCallbackValue<RunContextCallbacks[\"success\"]>, EventMeta];\n }>;\n error: Callback<{\n message: string;\n level: TraceLevel;\n origin: [InferCallbackValue<RunContextCallbacks[\"error\"]>, EventMeta];\n }>;\n finish: Callback<{\n message: string;\n level: TraceLevel;\n origin: [InferCallbackValue<RunContextCallbacks[\"finish\"]>, EventMeta];\n }>;\n}\n\ntype OutputTargetFn = (message: string) => void;\ntype OutputTarget = Logger | OutputTargetFn;\n\nexport interface GlobalTrajectoryMiddlewareOptions {\n /** Specify output target: 'console', Logger instance, custom function, or null to disable */\n target?: OutputTarget;\n /** List of classes to include in the trajectory */\n included?: AnyConstructable[];\n /** List of classes to exclude from the trajectory */\n excluded?: AnyConstructable[];\n /** Use pretty formatting for the trajectory */\n pretty?: boolean;\n /** Customize how instances of individual classes should be printed */\n prefixByType?: Map<AnyConstructable, string>;\n /** Enable/Disable the logging */\n enabled?: boolean;\n /** Whether to observe trajectories of nested run contexts */\n matchNested?: boolean;\n /** Defines a priority for registered events */\n emitterPriority?: number;\n /** Custom formatter function */\n formatter?: (input: GlobalTrajectoryMiddlewareFormatterInput) => string;\n}\n\n/**\n * Middleware for capturing and logging execution flow of agents, tools, and models.\n * Provides hierarchical visualization with indentation to show the call stack.\n */\nexport class GlobalTrajectoryMiddleware<T extends RunInstance = any> extends Middleware<T> {\n protected enabled: boolean;\n protected included: AnyConstructable[];\n protected excluded: AnyConstructable[];\n protected cleanups: (() => void)[] = [];\n protected target: (message: string) => void;\n protected ctx: RunContext<T> | null = null;\n protected pretty: boolean;\n protected traceLevel = new Map<string, TraceLevel>();\n protected prefixByType: Map<any, string>;\n protected matchNested: boolean;\n protected emitterPriority: number;\n protected formatter: (input: GlobalTrajectoryMiddlewareFormatterInput) => string;\n public readonly emitter: Emitter<GlobalTrajectoryMiddlewareCallbacks>;\n\n constructor(options: GlobalTrajectoryMiddlewareOptions = {}) {\n super();\n\n this.enabled = options.enabled ?? true;\n this.included = options.included ?? [];\n this.excluded = options.excluded ?? [];\n this.target = this.createTarget(options.target);\n this.pretty = options.pretty ?? false;\n this.matchNested = options.matchNested ?? true;\n this.emitterPriority = options.emitterPriority ?? -1; // run later\n\n // Default prefixes\n this.prefixByType = new Map([\n [BaseAgent, \"🤖 \"],\n [ChatModel, \"💬 \"],\n [Tool, \"🛠️ \"],\n [Requirement, \"🔎 \"],\n ...((options.prefixByType ? Array.from(options.prefixByType.entries()) : []) as any),\n ]);\n\n this.formatter =\n options.formatter ??\n ((x) => `${x.prefix}${x.className}[${x.instanceName || x.className}][${x.eventName}]`);\n\n this.emitter = Emitter.root.child<GlobalTrajectoryMiddlewareCallbacks>({\n namespace: [\"middleware\", \"globalTrajectory\"],\n });\n }\n\n /**\n * Bind the middleware to a run context\n */\n bind(ctx: RunContext<T>): void {\n // Cleanup previous bindings\n while (this.cleanups.length > 0) {\n this.cleanups.pop()!();\n }\n\n this.traceLevel.clear();\n this.traceLevel.set(ctx.runId, { relative: 0, absolute: 0 });\n this.ctx = ctx;\n\n this.bindEmitter(ctx.emitter);\n }\n\n private createTarget(input?: OutputTarget): OutputTargetFn {\n if (input === null || input === undefined) {\n // eslint-disable-next-line no-console\n return (msg) => console.log(msg);\n } else if (input instanceof Logger) {\n return (msg) => input.debug(msg);\n } else {\n return input;\n }\n }\n\n private bindEmitter(emitter: Emitter<any>): void {\n // Track all events for trace ID logging\n this.cleanups.push(\n emitter.match(\"*.*\", (_, event) => this.logTraceId(event), {\n matchNested: true,\n }),\n );\n\n // Handle top-level events\n const handleTopLevelEvent = async (data: any, meta: EventMeta) => {\n if (!(meta.creator instanceof RunContext)) {\n return;\n }\n if (!meta.trace) {\n return;\n }\n\n this.logTraceId(meta);\n if (!this.isAllowed(meta)) {\n return;\n }\n if (!this.enabled) {\n return;\n }\n\n const eventName = capitalize(meta.name) as \"Start\" | \"Success\" | \"Error\" | \"Finish\";\n await this[`onInternal${eventName}` as const].call(this, data, meta);\n };\n\n this.cleanups.push(\n emitter.match(\n matchInternalEvent([\"start\", \"success\", \"error\", \"finish\"]),\n handleTopLevelEvent,\n {\n matchNested: false,\n isBlocking: true,\n priority: this.emitterPriority,\n },\n ),\n );\n\n // Handle nested events if enabled\n if (this.matchNested) {\n const handleNestedEvent = async (data: any, meta: EventMeta) => {\n if (!(meta.creator instanceof RunContext)) {\n return;\n }\n if (meta.creator.emitter !== emitter) {\n await handleTopLevelEvent(data, meta);\n this.bindEmitter(meta.creator.emitter);\n }\n };\n\n this.cleanups.push(\n emitter.match(matchInternalEvent([\"start\"]), handleNestedEvent, {\n matchNested: true,\n isBlocking: true,\n priority: this.emitterPriority,\n }),\n );\n }\n }\n\n private logTraceId(meta: EventMeta): void {\n if (!meta.trace?.runId) {\n return;\n }\n if (this.traceLevel.has(meta.trace.runId)) {\n return;\n }\n if (meta.trace.parentRunId === meta.trace.runId) {\n return;\n }\n\n if (meta.trace.parentRunId) {\n const allowed = this.isAllowed(meta);\n const parentTrace = this.traceLevel.get(meta.trace.parentRunId) ?? {\n relative: 0,\n absolute: 0,\n };\n this.traceLevel.set(meta.trace.runId, {\n relative: parentTrace.relative + (allowed ? 1 : 0),\n absolute: parentTrace.absolute + 1,\n });\n }\n }\n\n private isAllowed(meta: EventMeta): boolean {\n let target: any = meta.creator;\n if (target instanceof RunContext) {\n target = target.instance;\n }\n\n for (const excluded of this.excluded) {\n if (target instanceof excluded) {\n return false;\n }\n }\n\n if (this.included.length === 0) {\n return true;\n }\n\n return this.included.some((included) => target instanceof included);\n }\n\n private extractName(meta: EventMeta): string {\n let target: any = meta.creator;\n if (target instanceof RunContext) {\n target = target.instance;\n }\n\n const className = target.constructor.name;\n let targetName: string | null = null;\n\n if (target instanceof BaseAgent && target.meta?.name) {\n targetName = target.meta.name;\n } else if (\"name\" in target && typeof target.name === \"string\") {\n targetName = target.name;\n }\n\n let prefix = \"\";\n for (const [type, typePrefix] of this.prefixByType.entries()) {\n if (target instanceof type) {\n prefix = typePrefix;\n break;\n }\n }\n\n const input: GlobalTrajectoryMiddlewareFormatterInput = {\n prefix,\n className,\n instanceName: targetName,\n eventName: meta.name,\n };\n\n return this.formatter(input);\n }\n\n private formatPrefix(meta: EventMeta): string {\n if (!meta.trace) {\n return \"\";\n }\n\n const indent = this.getTraceLevel(meta, \"self\").relative;\n const indentParent = this.getTraceLevel(meta, \"parent\").relative;\n const indentDiff = indent - indentParent;\n\n let prefix = \"\";\n prefix += \" \".repeat(indentParent * 2);\n\n if (meta.name !== \"start\" && indent) {\n prefix += \"<\";\n }\n\n prefix += \"--\".repeat(indentDiff);\n\n if (meta.name === \"start\" && prefix && indent) {\n prefix += \">\";\n }\n\n if (prefix) {\n prefix = `${prefix} `;\n }\n\n const name = this.extractName(meta);\n return `${prefix}${name}: `;\n }\n\n private getTraceLevel(meta: EventMeta, type: \"self\" | \"parent\" = \"self\"): TraceLevel {\n if (!meta.trace) {\n return { relative: 0, absolute: 0 };\n }\n\n const runId = type === \"parent\" ? meta.trace.parentRunId || \"\" : meta.trace.runId;\n return this.traceLevel.get(runId) ?? { relative: 0, absolute: 0 };\n }\n\n private async formatPayload(value: any): Promise<string> {\n if (isPrimitive(value)) {\n return String(value);\n }\n\n if (value instanceof FrameworkError) {\n return value.explain();\n }\n\n const serialized = await Serializer.serialize(value);\n return JSON.stringify(\n await Serializer.deserialize(serialized, [], true),\n (() => {\n const excludedKeys = new Set([\"emitter\", \"cleanups\", \"creator\", \"listeners\"]);\n return (key, value) => {\n if (excludedKeys.has(key)) {\n return undefined;\n }\n if (value && value instanceof Tool) {\n return value.name;\n }\n return value;\n };\n })(),\n this.pretty ? 2 : undefined,\n );\n }\n\n private async onInternalStart(\n payload: InferCallbackValue<RunContextCallbacks[\"start\"]>,\n meta: EventMeta,\n ): Promise<void> {\n const prefix = this.formatPrefix(meta);\n const message = `${prefix}${await this.formatPayload(payload.input)}`;\n\n await this.emitter.emit(\"start\", {\n message,\n level: this.getTraceLevel(meta),\n origin: [payload, meta],\n });\n\n this.target(message);\n }\n\n private async onInternalSuccess(\n payload: InferCallbackValue<RunContextCallbacks[\"success\"]>,\n meta: EventMeta,\n ): Promise<void> {\n const prefix = this.formatPrefix(meta);\n // TODO: change to payload.output once available\n const message = `${prefix}${await this.formatPayload(payload)}`;\n\n await this.emitter.emit(\"success\", {\n message,\n level: this.getTraceLevel(meta),\n origin: [payload, meta],\n });\n\n this.target(message);\n }\n\n private async onInternalError(\n payload: InferCallbackValue<RunContextCallbacks[\"error\"]>,\n meta: EventMeta,\n ): Promise<void> {\n const prefix = this.formatPrefix(meta);\n const message = `${prefix}${await this.formatPayload(payload)}`;\n\n await this.emitter.emit(\"error\", {\n message,\n level: this.getTraceLevel(meta),\n origin: [payload, meta],\n });\n\n this.target(message);\n }\n\n private async onInternalFinish(\n payload: InferCallbackValue<RunContextCallbacks[\"finish\"]>,\n meta: EventMeta,\n ): Promise<void> {\n const prefix = this.formatPrefix(meta);\n const message = `${prefix}${await this.formatPayload(payload.error || payload.output)}`;\n await this.emitter.emit(\"finish\", {\n message,\n level: this.getTraceLevel(meta),\n origin: [payload, meta],\n });\n }\n}\n\nfunction matchInternalEvent(names: (keyof RunContextCallbacks)[]) {\n return (meta: EventMeta): boolean =>\n names.includes(meta.name as keyof RunContextCallbacks) &&\n getProp(meta.context, [\"internal\"], false);\n}\n"]}
@@ -1,5 +1,5 @@
1
1
  import * as node_modules_remeda_dist_ToString_DKW_jYvM_js from 'node_modules/remeda/dist/ToString-DKW-jYvM.js';
2
- import { E as Emitter, C as Callback, I as InferCallbackValue } from '../emitter-CGKd_-kK.cjs';
2
+ import { E as Emitter, C as Callback, I as InferCallbackValue } from '../emitter-jN55XZZq.cjs';
3
3
  import { Serializable } from '../internals/serializable.cjs';
4
4
  import { ParserField } from './field.cjs';
5
5
  import { ValueOf } from '../internals/types.cjs';
@@ -1,5 +1,5 @@
1
1
  import * as node_modules_remeda_dist_ToString_DKW_jYvM_js from 'node_modules/remeda/dist/ToString-DKW-jYvM.js';
2
- import { E as Emitter, C as Callback, I as InferCallbackValue } from '../emitter-CN7I4uSV.js';
2
+ import { E as Emitter, C as Callback, I as InferCallbackValue } from '../emitter-36-9MnvA.js';
3
3
  import { Serializable } from '../internals/serializable.js';
4
4
  import { ParserField } from './field.js';
5
5
  import { ValueOf } from '../internals/types.js';
@@ -9,7 +9,7 @@ import '../internals/helpers/guards.cjs';
9
9
  import '../internals/serializable.cjs';
10
10
  import 'promise-based-task';
11
11
  import '../cache/base.cjs';
12
- import '../emitter-CGKd_-kK.cjs';
12
+ import '../emitter-jN55XZZq.cjs';
13
13
  import '../internals/helpers/promise.cjs';
14
14
  import 'zod-to-json-schema';
15
15
 
@@ -9,7 +9,7 @@ import '../internals/helpers/guards.js';
9
9
  import '../internals/serializable.js';
10
10
  import 'promise-based-task';
11
11
  import '../cache/base.js';
12
- import '../emitter-CN7I4uSV.js';
12
+ import '../emitter-36-9MnvA.js';
13
13
  import '../internals/helpers/promise.js';
14
14
  import 'zod-to-json-schema';
15
15
 
@@ -7,7 +7,7 @@ import { Task } from 'promise-based-task';
7
7
  import { BaseCache } from '../cache/base.cjs';
8
8
  import { AnyToolSchemaLike, FromSchemaLikeRaw, FromSchemaLike } from '../internals/helpers/schema.cjs';
9
9
  import { ZodSchema, z } from 'zod';
10
- import { E as Emitter, C as Callback } from '../emitter-CGKd_-kK.cjs';
10
+ import { E as Emitter, C as Callback } from '../emitter-jN55XZZq.cjs';
11
11
  import '../internals/helpers/promise.cjs';
12
12
  import '../internals/types.cjs';
13
13
  import '../internals/helpers/guards.cjs';
@@ -7,7 +7,7 @@ import { Task } from 'promise-based-task';
7
7
  import { BaseCache } from '../cache/base.js';
8
8
  import { AnyToolSchemaLike, FromSchemaLikeRaw, FromSchemaLike } from '../internals/helpers/schema.js';
9
9
  import { ZodSchema, z } from 'zod';
10
- import { E as Emitter, C as Callback } from '../emitter-CN7I4uSV.js';
10
+ import { E as Emitter, C as Callback } from '../emitter-36-9MnvA.js';
11
11
  import '../internals/helpers/promise.js';
12
12
  import '../internals/types.js';
13
13
  import '../internals/helpers/guards.js';
@@ -3,7 +3,7 @@ import { z } from 'zod';
3
3
  import { ConfigOptions, ImportObject, ImportOptions, evaluate } from 'mathjs';
4
4
  import 'ajv';
5
5
  import '../context.cjs';
6
- import '../emitter-CGKd_-kK.cjs';
6
+ import '../emitter-jN55XZZq.cjs';
7
7
  import '../internals/types.cjs';
8
8
  import '../internals/helpers/guards.cjs';
9
9
  import '../internals/serializable.cjs';
@@ -3,7 +3,7 @@ import { z } from 'zod';
3
3
  import { ConfigOptions, ImportObject, ImportOptions, evaluate } from 'mathjs';
4
4
  import 'ajv';
5
5
  import '../context.js';
6
- import '../emitter-CN7I4uSV.js';
6
+ import '../emitter-36-9MnvA.js';
7
7
  import '../internals/types.js';
8
8
  import '../internals/helpers/guards.js';
9
9
  import '../internals/serializable.js';