@google/adk 2.4.0 → 2.5.0

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 (326) hide show
  1. package/dist/cjs/a2a/a2a_remote_agent.js +17 -2
  2. package/dist/cjs/a2a/a2a_remote_agent_utils.js +8 -4
  3. package/dist/cjs/a2a/agent_card.js +5 -81
  4. package/dist/cjs/a2a/agent_to_a2a.js +7 -0
  5. package/dist/cjs/a2a/event_converter_utils.js +10 -3
  6. package/dist/cjs/agents/agent_engine_client.js +282 -55
  7. package/dist/cjs/agents/base_agent.js +111 -68
  8. package/dist/cjs/agents/functions.js +62 -85
  9. package/dist/cjs/agents/instructions.js +7 -6
  10. package/dist/cjs/agents/invocation_context.js +2 -2
  11. package/dist/cjs/agents/llm_agent.js +105 -95
  12. package/dist/cjs/agents/processors/content_processor_utils.js +42 -15
  13. package/dist/cjs/agents/processors/content_request_processor.js +28 -11
  14. package/dist/cjs/agents/processors/context_compactor_request_processor.js +5 -1
  15. package/dist/cjs/agents/processors/instructions_llm_request_processor.js +1 -1
  16. package/dist/cjs/agents/processors/nl_planning_processor.js +2 -1
  17. package/dist/cjs/artifacts/file_artifact_service.js +1 -1
  18. package/dist/cjs/cli/cli_eval.js +884 -17
  19. package/dist/cjs/cli/utils/cleanup.js +84 -0
  20. package/dist/cjs/cli/utils/envs.js +33 -2
  21. package/dist/cjs/cli/utils/evals.js +75 -8
  22. package/dist/cjs/cli/utils/logs.js +128 -9
  23. package/dist/cjs/cli/utils/path_validation.js +83 -0
  24. package/dist/cjs/code_executors/container_code_executor.js +158 -36
  25. package/dist/cjs/common.js +50 -12
  26. package/dist/cjs/context/compaction_utils.js +19 -0
  27. package/dist/cjs/events/event.js +7 -3
  28. package/dist/cjs/features/feature_registry.js +5 -0
  29. package/dist/cjs/index.js +29 -14
  30. package/dist/cjs/models/google_llm.js +33 -2
  31. package/dist/cjs/models/interactions_utils.js +3 -0
  32. package/dist/cjs/plugins/base_plugin.js +45 -21
  33. package/dist/cjs/plugins/plugin_manager.js +27 -0
  34. package/dist/cjs/runner/runner.js +306 -189
  35. package/dist/cjs/sessions/base_session_service.js +1 -4
  36. package/dist/cjs/sessions/database_session_service.js +191 -19
  37. package/dist/cjs/sessions/db/operations.js +70 -1
  38. package/dist/cjs/sessions/db/schema.js +4 -0
  39. package/dist/cjs/sessions/in_memory_session_service.js +2 -2
  40. package/dist/cjs/sessions/vertex_ai_session_service.js +11 -3
  41. package/dist/cjs/skills/loader.js +29 -28
  42. package/dist/cjs/telemetry/api/instrumentation.js +632 -0
  43. package/dist/cjs/telemetry/api/internal/common/accumulators.js +169 -0
  44. package/dist/cjs/telemetry/api/internal/common/attributes.js +131 -0
  45. package/dist/cjs/telemetry/api/internal/common/config.js +159 -0
  46. package/dist/cjs/telemetry/api/internal/common/error.js +98 -0
  47. package/dist/cjs/telemetry/api/internal/common/finish_reason.js +63 -0
  48. package/dist/cjs/telemetry/api/internal/common/genai_messages.js +217 -0
  49. package/dist/cjs/telemetry/api/internal/common/http_exchange.js +169 -0
  50. package/dist/cjs/telemetry/api/internal/common/provider_name.js +87 -0
  51. package/dist/cjs/telemetry/api/internal/common/redaction.js +148 -0
  52. package/dist/cjs/telemetry/api/internal/common/serialization.js +58 -0
  53. package/dist/cjs/telemetry/api/internal/common/token_usage.js +118 -0
  54. package/dist/cjs/telemetry/api/internal/logs/inference.js +50 -0
  55. package/dist/cjs/telemetry/api/internal/logs/logger.js +46 -0
  56. package/dist/cjs/telemetry/api/internal/logs/mcp.js +141 -0
  57. package/dist/cjs/telemetry/api/internal/metrics/agent.js +128 -0
  58. package/dist/cjs/telemetry/api/internal/metrics/inference_start_hook.js +56 -0
  59. package/dist/cjs/telemetry/api/internal/metrics/meters.js +189 -0
  60. package/dist/cjs/telemetry/api/internal/metrics/tool.js +147 -0
  61. package/dist/cjs/telemetry/api/internal/metrics/workflow.js +132 -0
  62. package/dist/cjs/telemetry/api/internal/trace/agent.js +52 -0
  63. package/dist/cjs/telemetry/api/internal/trace/compaction.js +78 -0
  64. package/dist/cjs/telemetry/api/internal/trace/inference.js +132 -0
  65. package/dist/cjs/telemetry/api/internal/trace/live.js +52 -0
  66. package/dist/cjs/telemetry/api/internal/trace/node.js +73 -0
  67. package/dist/cjs/telemetry/api/internal/trace/propagation.js +67 -0
  68. package/dist/cjs/telemetry/api/internal/trace/tool.js +122 -0
  69. package/dist/cjs/telemetry/api/internal/trace/tracer.js +77 -0
  70. package/dist/cjs/telemetry/api/internal/trace/workflow.js +77 -0
  71. package/dist/cjs/telemetry/export/agent_engine.js +90 -0
  72. package/dist/cjs/telemetry/export/agent_runtime.js +73 -0
  73. package/dist/cjs/telemetry/export/google_cloud.js +763 -0
  74. package/dist/cjs/telemetry/export/request_driven_metric_reader.js +303 -0
  75. package/dist/cjs/telemetry/{setup.js → export/setup.js} +72 -5
  76. package/dist/cjs/tools/agent_tool.js +9 -1
  77. package/dist/cjs/tools/application_integration_tool/application_integration_toolset.js +22 -34
  78. package/dist/cjs/tools/application_integration_tool/clients/connections_client.js +60 -20
  79. package/dist/cjs/tools/application_integration_tool/clients/integration_client.js +26 -10
  80. package/dist/cjs/tools/application_integration_tool/integration_connector_tool.js +14 -23
  81. package/dist/cjs/tools/base_toolset.js +1 -1
  82. package/dist/cjs/tools/computer_use/base_computer.js +125 -0
  83. package/dist/cjs/tools/computer_use/computer_use_actions.js +502 -0
  84. package/dist/cjs/tools/computer_use/computer_use_tool.js +244 -0
  85. package/dist/cjs/tools/computer_use/computer_use_toolset.js +360 -0
  86. package/dist/cjs/tools/function_tool.js +15 -5
  87. package/dist/cjs/tools/google_api_tool/google_api_toolset.js +3 -2
  88. package/dist/cjs/tools/google_api_tool/googleapi_to_openapi_converter.js +3 -1
  89. package/dist/cjs/tools/load_artifacts_tool.js +15 -5
  90. package/dist/cjs/tools/load_memory_tool.js +2 -5
  91. package/dist/cjs/tools/mcp/mcp_session_manager.js +23 -3
  92. package/dist/cjs/tools/mcp/mcp_tool.js +7 -3
  93. package/dist/cjs/tools/mcp/mcp_toolset.js +18 -6
  94. package/dist/cjs/tools/openapi_tool/auth/auth_helpers.js +30 -29
  95. package/dist/cjs/tools/openapi_tool/auth/credential_exchangers/oauth2_exchanger.js +2 -2
  96. package/dist/cjs/tools/openapi_tool/rest_api_tool.js +45 -14
  97. package/dist/cjs/tools/skill/load_skill_resource_tool.js +7 -0
  98. package/dist/cjs/tools/skill/load_skill_tool.js +3 -0
  99. package/dist/cjs/tools/skill/run_skill_script_tool.js +12 -2
  100. package/dist/cjs/tools/toolbox_toolset.js +47 -28
  101. package/dist/cjs/utils/async_queue.js +37 -3
  102. package/dist/cjs/utils/client_labels.js +9 -11
  103. package/dist/cjs/utils/env_aware_utils.js +13 -7
  104. package/dist/cjs/utils/error_utils.js +40 -2
  105. package/dist/cjs/utils/failover_utils.js +10 -9
  106. package/dist/cjs/utils/gemini_schema_util.js +125 -12
  107. package/dist/cjs/utils/id_provider.js +63 -0
  108. package/dist/cjs/utils/model_name.js +15 -0
  109. package/dist/cjs/utils/simple_zod_to_json.js +13 -31
  110. package/dist/cjs/utils/ssrf_guard.js +46 -16
  111. package/dist/cjs/version.js +1 -1
  112. package/dist/cjs/workflow/index.js +2 -0
  113. package/dist/cjs/workflow/node_runner.js +24 -80
  114. package/dist/cjs/workflow/nodes/parallel_worker.js +18 -7
  115. package/dist/cjs/workflow/run_llm_agent_as_node.js +19 -14
  116. package/dist/cjs/workflow/utils/retry_utils.js +2 -2
  117. package/dist/cjs/workflow/workflow.js +12 -12
  118. package/dist/esm/a2a/a2a_remote_agent.js +23 -3
  119. package/dist/esm/a2a/a2a_remote_agent_utils.js +8 -4
  120. package/dist/esm/a2a/agent_card.js +5 -83
  121. package/dist/esm/a2a/agent_to_a2a.js +7 -0
  122. package/dist/esm/a2a/event_converter_utils.js +10 -3
  123. package/dist/esm/agents/agent_engine_client.js +293 -55
  124. package/dist/esm/agents/base_agent.js +111 -72
  125. package/dist/esm/agents/functions.js +67 -89
  126. package/dist/esm/agents/instructions.js +7 -6
  127. package/dist/esm/agents/invocation_context.js +2 -2
  128. package/dist/esm/agents/llm_agent.js +111 -100
  129. package/dist/esm/agents/processors/content_processor_utils.js +42 -15
  130. package/dist/esm/agents/processors/content_request_processor.js +28 -11
  131. package/dist/esm/agents/processors/context_compactor_request_processor.js +5 -1
  132. package/dist/esm/agents/processors/instructions_llm_request_processor.js +1 -1
  133. package/dist/esm/agents/processors/nl_planning_processor.js +2 -1
  134. package/dist/esm/artifacts/file_artifact_service.js +1 -1
  135. package/dist/esm/cli/cli_eval.js +880 -17
  136. package/dist/esm/cli/utils/cleanup.js +56 -0
  137. package/dist/esm/cli/utils/envs.js +33 -2
  138. package/dist/esm/cli/utils/evals.js +74 -8
  139. package/dist/esm/cli/utils/logs.js +128 -9
  140. package/dist/esm/cli/utils/path_validation.js +45 -0
  141. package/dist/esm/code_executors/container_code_executor.js +158 -36
  142. package/dist/esm/common.js +49 -6
  143. package/dist/esm/context/compaction_utils.js +19 -0
  144. package/dist/esm/events/event.js +7 -3
  145. package/dist/esm/features/feature_registry.js +5 -0
  146. package/dist/esm/index.js +27 -15
  147. package/dist/esm/models/google_llm.js +33 -2
  148. package/dist/esm/models/interactions_utils.js +3 -0
  149. package/dist/esm/plugins/base_plugin.js +45 -21
  150. package/dist/esm/plugins/plugin_manager.js +27 -0
  151. package/dist/esm/runner/runner.js +310 -192
  152. package/dist/esm/sessions/base_session_service.js +1 -4
  153. package/dist/esm/sessions/database_session_service.js +191 -19
  154. package/dist/esm/sessions/db/operations.js +70 -1
  155. package/dist/esm/sessions/db/schema.js +4 -0
  156. package/dist/esm/sessions/in_memory_session_service.js +2 -2
  157. package/dist/esm/sessions/vertex_ai_session_service.js +11 -3
  158. package/dist/esm/skills/loader.js +29 -28
  159. package/dist/esm/telemetry/api/instrumentation.js +655 -0
  160. package/dist/esm/telemetry/api/internal/common/accumulators.js +139 -0
  161. package/dist/esm/telemetry/api/internal/common/attributes.js +94 -0
  162. package/dist/esm/telemetry/api/internal/common/config.js +128 -0
  163. package/dist/esm/telemetry/api/internal/common/error.js +69 -0
  164. package/dist/esm/telemetry/api/internal/common/finish_reason.js +35 -0
  165. package/dist/esm/telemetry/api/internal/common/genai_messages.js +188 -0
  166. package/dist/esm/telemetry/api/internal/common/http_exchange.js +142 -0
  167. package/dist/esm/telemetry/api/internal/common/provider_name.js +59 -0
  168. package/dist/esm/telemetry/api/internal/common/redaction.js +121 -0
  169. package/dist/esm/telemetry/api/internal/common/serialization.js +30 -0
  170. package/dist/esm/telemetry/api/internal/common/token_usage.js +85 -0
  171. package/dist/esm/telemetry/api/internal/logs/inference.js +23 -0
  172. package/dist/esm/telemetry/api/internal/logs/logger.js +19 -0
  173. package/dist/esm/telemetry/api/internal/logs/mcp.js +113 -0
  174. package/dist/esm/telemetry/api/internal/metrics/agent.js +107 -0
  175. package/dist/esm/telemetry/api/internal/metrics/inference_start_hook.js +29 -0
  176. package/dist/esm/telemetry/api/internal/metrics/meters.js +165 -0
  177. package/dist/esm/telemetry/api/internal/metrics/tool.js +119 -0
  178. package/dist/esm/telemetry/api/internal/metrics/workflow.js +112 -0
  179. package/dist/esm/telemetry/api/internal/trace/agent.js +25 -0
  180. package/dist/esm/telemetry/api/internal/trace/compaction.js +50 -0
  181. package/dist/esm/telemetry/api/internal/trace/inference.js +107 -0
  182. package/dist/esm/telemetry/api/internal/trace/live.js +28 -0
  183. package/dist/esm/telemetry/api/internal/trace/node.js +44 -0
  184. package/dist/esm/telemetry/api/internal/trace/propagation.js +39 -0
  185. package/dist/esm/telemetry/api/internal/trace/tool.js +96 -0
  186. package/dist/esm/telemetry/api/internal/trace/tracer.js +52 -0
  187. package/dist/esm/telemetry/api/internal/trace/workflow.js +54 -0
  188. package/dist/esm/telemetry/export/agent_engine.js +67 -0
  189. package/dist/esm/telemetry/export/agent_runtime.js +46 -0
  190. package/dist/esm/telemetry/export/google_cloud.js +733 -0
  191. package/dist/esm/telemetry/export/request_driven_metric_reader.js +276 -0
  192. package/dist/esm/telemetry/export/setup.js +160 -0
  193. package/dist/esm/tools/agent_tool.js +9 -1
  194. package/dist/esm/tools/application_integration_tool/application_integration_toolset.js +23 -35
  195. package/dist/esm/tools/application_integration_tool/clients/connections_client.js +64 -21
  196. package/dist/esm/tools/application_integration_tool/clients/integration_client.js +26 -10
  197. package/dist/esm/tools/application_integration_tool/integration_connector_tool.js +14 -23
  198. package/dist/esm/tools/base_toolset.js +1 -1
  199. package/dist/esm/tools/computer_use/base_computer.js +99 -0
  200. package/dist/esm/tools/computer_use/computer_use_actions.js +477 -0
  201. package/dist/esm/tools/computer_use/computer_use_tool.js +217 -0
  202. package/dist/esm/tools/computer_use/computer_use_toolset.js +344 -0
  203. package/dist/esm/tools/function_tool.js +15 -5
  204. package/dist/esm/tools/google_api_tool/google_api_toolset.js +3 -2
  205. package/dist/esm/tools/google_api_tool/googleapi_to_openapi_converter.js +3 -1
  206. package/dist/esm/tools/load_artifacts_tool.js +15 -5
  207. package/dist/esm/tools/load_memory_tool.js +2 -5
  208. package/dist/esm/tools/mcp/mcp_session_manager.js +23 -3
  209. package/dist/esm/tools/mcp/mcp_tool.js +7 -3
  210. package/dist/esm/tools/mcp/mcp_toolset.js +18 -6
  211. package/dist/esm/tools/openapi_tool/auth/auth_helpers.js +27 -26
  212. package/dist/esm/tools/openapi_tool/auth/credential_exchangers/oauth2_exchanger.js +2 -2
  213. package/dist/esm/tools/openapi_tool/rest_api_tool.js +45 -14
  214. package/dist/esm/tools/skill/load_skill_resource_tool.js +7 -0
  215. package/dist/esm/tools/skill/load_skill_tool.js +3 -0
  216. package/dist/esm/tools/skill/run_skill_script_tool.js +12 -2
  217. package/dist/esm/tools/toolbox_toolset.js +47 -28
  218. package/dist/esm/utils/async_queue.js +37 -3
  219. package/dist/esm/utils/client_labels.js +9 -11
  220. package/dist/esm/utils/env_aware_utils.js +12 -7
  221. package/dist/esm/utils/error_utils.js +37 -1
  222. package/dist/esm/utils/failover_utils.js +10 -9
  223. package/dist/esm/utils/gemini_schema_util.js +124 -11
  224. package/dist/esm/utils/id_provider.js +33 -0
  225. package/dist/esm/utils/model_name.js +14 -0
  226. package/dist/esm/utils/simple_zod_to_json.js +13 -31
  227. package/dist/esm/utils/ssrf_guard.js +55 -15
  228. package/dist/esm/version.js +1 -1
  229. package/dist/esm/workflow/index.js +2 -1
  230. package/dist/esm/workflow/node_runner.js +26 -80
  231. package/dist/esm/workflow/nodes/parallel_worker.js +16 -6
  232. package/dist/esm/workflow/run_llm_agent_as_node.js +19 -14
  233. package/dist/esm/workflow/utils/retry_utils.js +2 -2
  234. package/dist/esm/workflow/workflow.js +12 -12
  235. package/dist/types/a2a/a2a_remote_agent.d.ts +6 -0
  236. package/dist/types/a2a/a2a_remote_agent_utils.d.ts +3 -2
  237. package/dist/types/agents/agent_engine_client.d.ts +121 -3
  238. package/dist/types/agents/base_agent.d.ts +2 -0
  239. package/dist/types/agents/processors/content_processor_utils.d.ts +9 -9
  240. package/dist/types/agents/run_config.d.ts +14 -0
  241. package/dist/types/cli/cli_eval.d.ts +172 -9
  242. package/dist/types/cli/utils/cleanup.d.ts +18 -0
  243. package/dist/types/cli/utils/envs.d.ts +13 -3
  244. package/dist/types/cli/utils/evals.d.ts +64 -1
  245. package/dist/types/cli/utils/logs.d.ts +19 -2
  246. package/dist/types/cli/utils/path_validation.d.ts +30 -0
  247. package/dist/types/code_executors/container_code_executor.d.ts +49 -8
  248. package/dist/types/common.d.ts +16 -9
  249. package/dist/types/context/compaction_utils.d.ts +13 -1
  250. package/dist/types/events/event.d.ts +3 -0
  251. package/dist/types/features/feature_registry.d.ts +1 -0
  252. package/dist/types/index.d.ts +9 -6
  253. package/dist/types/plugins/base_plugin.d.ts +50 -21
  254. package/dist/types/plugins/plugin_manager.d.ts +13 -0
  255. package/dist/types/runner/runner.d.ts +11 -0
  256. package/dist/types/sessions/base_session_service.d.ts +0 -2
  257. package/dist/types/sessions/db/operations.d.ts +2 -1
  258. package/dist/types/sessions/db/schema.d.ts +5 -3
  259. package/dist/types/telemetry/api/instrumentation.d.ts +259 -0
  260. package/dist/types/telemetry/api/internal/common/accumulators.d.ts +91 -0
  261. package/dist/types/telemetry/api/internal/common/attributes.d.ts +90 -0
  262. package/dist/types/telemetry/api/internal/common/config.d.ts +61 -0
  263. package/dist/types/telemetry/api/internal/common/error.d.ts +29 -0
  264. package/dist/types/telemetry/api/internal/common/finish_reason.d.ts +18 -0
  265. package/dist/types/telemetry/api/internal/common/genai_messages.d.ts +26 -0
  266. package/dist/types/telemetry/api/internal/common/http_exchange.d.ts +50 -0
  267. package/dist/types/telemetry/api/internal/common/provider_name.d.ts +16 -0
  268. package/dist/types/telemetry/api/internal/common/redaction.d.ts +14 -0
  269. package/dist/types/telemetry/api/internal/common/serialization.d.ts +18 -0
  270. package/dist/types/telemetry/api/internal/common/token_usage.d.ts +47 -0
  271. package/dist/types/telemetry/api/internal/logs/inference.d.ts +15 -0
  272. package/dist/types/telemetry/api/internal/logs/logger.d.ts +13 -0
  273. package/dist/types/telemetry/api/internal/logs/mcp.d.ts +31 -0
  274. package/dist/types/telemetry/api/internal/metrics/agent.d.ts +28 -0
  275. package/dist/types/telemetry/api/internal/metrics/inference_start_hook.d.ts +12 -0
  276. package/dist/types/telemetry/api/internal/metrics/meters.d.ts +57 -0
  277. package/dist/types/telemetry/api/internal/metrics/tool.d.ts +44 -0
  278. package/dist/types/telemetry/api/internal/metrics/workflow.d.ts +32 -0
  279. package/dist/types/telemetry/api/internal/trace/agent.d.ts +14 -0
  280. package/dist/types/telemetry/api/internal/trace/compaction.d.ts +20 -0
  281. package/dist/types/telemetry/api/internal/trace/inference.d.ts +31 -0
  282. package/dist/types/telemetry/api/internal/trace/live.d.ts +21 -0
  283. package/dist/types/telemetry/api/internal/trace/node.d.ts +22 -0
  284. package/dist/types/telemetry/api/internal/trace/propagation.d.ts +22 -0
  285. package/dist/types/telemetry/api/internal/trace/tool.d.ts +57 -0
  286. package/dist/types/telemetry/api/internal/trace/tracer.d.ts +31 -0
  287. package/dist/types/telemetry/api/internal/trace/workflow.d.ts +21 -0
  288. package/dist/types/telemetry/export/agent_engine.d.ts +23 -0
  289. package/dist/types/telemetry/export/agent_runtime.d.ts +40 -0
  290. package/dist/types/telemetry/export/google_cloud.d.ts +189 -0
  291. package/dist/types/telemetry/export/request_driven_metric_reader.d.ts +144 -0
  292. package/dist/types/telemetry/{setup.d.ts → export/setup.d.ts} +3 -0
  293. package/dist/types/tools/agent_tool.d.ts +1 -1
  294. package/dist/types/tools/application_integration_tool/clients/connections_client.d.ts +13 -4
  295. package/dist/types/tools/base_toolset.d.ts +1 -1
  296. package/dist/types/tools/computer_use/base_computer.d.ts +281 -0
  297. package/dist/types/tools/computer_use/computer_use_actions.d.ts +53 -0
  298. package/dist/types/tools/computer_use/computer_use_tool.d.ts +108 -0
  299. package/dist/types/tools/computer_use/computer_use_toolset.d.ts +132 -0
  300. package/dist/types/tools/google_api_tool/google_api_toolset.d.ts +2 -0
  301. package/dist/types/tools/load_artifacts_tool.d.ts +5 -1
  302. package/dist/types/tools/load_memory_tool.d.ts +13 -7
  303. package/dist/types/tools/mcp/mcp_session_manager.d.ts +16 -0
  304. package/dist/types/tools/openapi_tool/auth/auth_helpers.d.ts +8 -8
  305. package/dist/types/tools/openapi_tool/openapi_spec_parser/operation_parser.d.ts +7 -3
  306. package/dist/types/tools/toolbox_toolset.d.ts +15 -20
  307. package/dist/types/utils/async_queue.d.ts +14 -0
  308. package/dist/types/utils/env_aware_utils.d.ts +11 -0
  309. package/dist/types/utils/error_utils.d.ts +4 -0
  310. package/dist/types/utils/gemini_schema_util.d.ts +27 -6
  311. package/dist/types/utils/id_provider.d.ts +43 -0
  312. package/dist/types/utils/model_name.d.ts +8 -0
  313. package/dist/types/utils/ssrf_guard.d.ts +2 -0
  314. package/dist/types/version.d.ts +1 -1
  315. package/dist/types/workflow/index.d.ts +1 -1
  316. package/dist/types/workflow/nodes/parallel_worker.d.ts +8 -1
  317. package/dist/web/index_web.js +86 -88
  318. package/dist/web/index_web.js.map +4 -4
  319. package/package.json +13 -15
  320. package/dist/cjs/telemetry/google_cloud.js +0 -116
  321. package/dist/cjs/telemetry/tracing.js +0 -393
  322. package/dist/esm/telemetry/google_cloud.js +0 -99
  323. package/dist/esm/telemetry/setup.js +0 -88
  324. package/dist/esm/telemetry/tracing.js +0 -361
  325. package/dist/types/telemetry/google_cloud.d.ts +0 -9
  326. package/dist/types/telemetry/tracing.d.ts +0 -143
@@ -0,0 +1,107 @@
1
+ import {createRequire as topLevelCreateRequire} from 'module';
2
+ const require = topLevelCreateRequire(import.meta.url);
3
+ var __defProp = Object.defineProperty;
4
+ var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
5
+ /**
6
+ * @license
7
+ * Copyright 2026 Google LLC
8
+ * SPDX-License-Identifier: Apache-2.0
9
+ */
10
+ import { AdkAttr, GenAiAttr } from "../common/attributes.js";
11
+ import {
12
+ isExperimentalFeatureEnabled,
13
+ shouldAddContentToSpans
14
+ } from "../common/config.js";
15
+ import { isReportedFinishReason } from "../common/finish_reason.js";
16
+ import {
17
+ withoutContent
18
+ } from "../common/genai_messages.js";
19
+ import { safeJsonSerialize } from "../common/serialization.js";
20
+ import { TokenUsage } from "../common/token_usage.js";
21
+ import { getTracer } from "./tracer.js";
22
+ function startInferenceSpan({
23
+ llmRequest,
24
+ commonAttributes
25
+ }) {
26
+ var _a, _b;
27
+ const config = llmRequest.config;
28
+ return getTracer().startSpan(
29
+ `generate_content ${(_a = llmRequest.model) != null ? _a : ""}`.trim(),
30
+ {
31
+ attributes: {
32
+ [GenAiAttr.OPERATION_NAME]: "generate_content",
33
+ [GenAiAttr.REQUEST_MODEL]: (_b = llmRequest.model) != null ? _b : "",
34
+ ...(config == null ? void 0 : config.topP) === void 0 ? {} : { [GenAiAttr.REQUEST_TOP_P]: config.topP },
35
+ ...(config == null ? void 0 : config.maxOutputTokens) === void 0 ? {} : { [GenAiAttr.REQUEST_MAX_TOKENS]: config.maxOutputTokens },
36
+ ...commonAttributes
37
+ }
38
+ }
39
+ );
40
+ }
41
+ __name(startInferenceSpan, "startInferenceSpan");
42
+ function buildResponseAttributes(llmResponse) {
43
+ return {
44
+ ...isReportedFinishReason(llmResponse.finishReason) ? {
45
+ [GenAiAttr.RESPONSE_FINISH_REASONS]: [
46
+ String(llmResponse.finishReason).toLowerCase()
47
+ ]
48
+ } : {},
49
+ ...llmResponse.usageMetadata ? new TokenUsage(llmResponse.usageMetadata).asAttributes() : {}
50
+ };
51
+ }
52
+ __name(buildResponseAttributes, "buildResponseAttributes");
53
+ function groundingSourceUris(chunks) {
54
+ var _a, _b, _c;
55
+ const uris = /* @__PURE__ */ new Set();
56
+ for (const chunk of chunks) {
57
+ for (const uri of [
58
+ (_a = chunk.web) == null ? void 0 : _a.uri,
59
+ (_b = chunk.retrievedContext) == null ? void 0 : _b.uri,
60
+ (_c = chunk.maps) == null ? void 0 : _c.uri
61
+ ]) {
62
+ if (uri) {
63
+ uris.add(uri);
64
+ }
65
+ }
66
+ }
67
+ return [...uris];
68
+ }
69
+ __name(groundingSourceUris, "groundingSourceUris");
70
+ function buildGroundingAttributes(groundingMetadata, captureContent) {
71
+ var _a, _b;
72
+ const queries = ((_a = groundingMetadata.webSearchQueries) != null ? _a : []).filter(Boolean);
73
+ const chunks = (_b = groundingMetadata.groundingChunks) != null ? _b : [];
74
+ const sourceUris = groundingSourceUris(chunks);
75
+ return {
76
+ [AdkAttr.EXPERIMENTAL_GROUNDING_QUERY_COUNT]: new Set(queries).size,
77
+ [AdkAttr.EXPERIMENTAL_GROUNDING_CHUNK_COUNT]: chunks.length,
78
+ ...captureContent && queries.length > 0 ? { [AdkAttr.EXPERIMENTAL_GROUNDING_WEB_SEARCH_QUERIES]: queries } : {},
79
+ ...captureContent && sourceUris.length > 0 ? { [AdkAttr.EXPERIMENTAL_GROUNDING_SOURCE_URIS]: sourceUris } : {}
80
+ };
81
+ }
82
+ __name(buildGroundingAttributes, "buildGroundingAttributes");
83
+ function setInferenceResponseAttributes(span, llmResponse, config) {
84
+ span.setAttributes(buildResponseAttributes(llmResponse));
85
+ if (llmResponse.groundingMetadata && isExperimentalFeatureEnabled(config, "grounding")) {
86
+ span.setAttributes(
87
+ buildGroundingAttributes(
88
+ llmResponse.groundingMetadata,
89
+ shouldAddContentToSpans(config)
90
+ )
91
+ );
92
+ }
93
+ }
94
+ __name(setInferenceResponseAttributes, "setInferenceResponseAttributes");
95
+ function setInferenceOperationDetails(span, details, config) {
96
+ const recorded = shouldAddContentToSpans(config) ? details : withoutContent(details);
97
+ for (const [key, value] of Object.entries(recorded)) {
98
+ span.setAttribute(key, safeJsonSerialize(value));
99
+ }
100
+ }
101
+ __name(setInferenceOperationDetails, "setInferenceOperationDetails");
102
+ export {
103
+ buildResponseAttributes,
104
+ setInferenceOperationDetails,
105
+ setInferenceResponseAttributes,
106
+ startInferenceSpan
107
+ };
@@ -0,0 +1,28 @@
1
+ import {createRequire as topLevelCreateRequire} from 'module';
2
+ const require = topLevelCreateRequire(import.meta.url);
3
+ var __defProp = Object.defineProperty;
4
+ var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
5
+ /**
6
+ * @license
7
+ * Copyright 2026 Google LLC
8
+ * SPDX-License-Identifier: Apache-2.0
9
+ */
10
+ import { GcpAttr } from "../common/attributes.js";
11
+ import {
12
+ safeJsonSerialize,
13
+ shouldAddRequestResponseToSpans
14
+ } from "../common/serialization.js";
15
+ function setSendDataAttributes(span, { invocationContext, eventId, data }) {
16
+ span.setAttributes({
17
+ [GcpAttr.INVOCATION_ID]: invocationContext.invocationId,
18
+ [GcpAttr.EVENT_ID]: eventId
19
+ });
20
+ span.setAttribute(
21
+ GcpAttr.DATA,
22
+ shouldAddRequestResponseToSpans() ? safeJsonSerialize(data) : "{}"
23
+ );
24
+ }
25
+ __name(setSendDataAttributes, "setSendDataAttributes");
26
+ export {
27
+ setSendDataAttributes
28
+ };
@@ -0,0 +1,44 @@
1
+ import {createRequire as topLevelCreateRequire} from 'module';
2
+ const require = topLevelCreateRequire(import.meta.url);
3
+ var __defProp = Object.defineProperty;
4
+ var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
5
+ /**
6
+ * @license
7
+ * Copyright 2026 Google LLC
8
+ * SPDX-License-Identifier: Apache-2.0
9
+ */
10
+ import { GcpAttr, GenAiAttr } from "../common/attributes.js";
11
+ import { getTracer } from "./tracer.js";
12
+ const SELF_TRACED_NODE_BRANDS = [
13
+ /* @__PURE__ */ Symbol.for("google.adk.baseAgent"),
14
+ /* @__PURE__ */ Symbol.for("google.adk.workflow.workflow")
15
+ ];
16
+ function hasOwnInvocationSpan(node) {
17
+ return SELF_TRACED_NODE_BRANDS.some(
18
+ (brand) => brand in node && Reflect.get(node, brand) === true
19
+ );
20
+ }
21
+ __name(hasOwnInvocationSpan, "hasOwnInvocationSpan");
22
+ function startNodeSpan({
23
+ nodeName,
24
+ sessionId
25
+ }) {
26
+ return getTracer().startSpan(`invoke_node ${nodeName}`, {
27
+ attributes: {
28
+ [GenAiAttr.OPERATION_NAME]: "invoke_node",
29
+ [GenAiAttr.CONVERSATION_ID]: sessionId
30
+ }
31
+ });
32
+ }
33
+ __name(startNodeSpan, "startNodeSpan");
34
+ function setAssociatedEventIds(span, eventIds) {
35
+ if (eventIds.length > 0) {
36
+ span.setAttribute(GcpAttr.ASSOCIATED_EVENT_IDS, [...eventIds]);
37
+ }
38
+ }
39
+ __name(setAssociatedEventIds, "setAssociatedEventIds");
40
+ export {
41
+ hasOwnInvocationSpan,
42
+ setAssociatedEventIds,
43
+ startNodeSpan
44
+ };
@@ -0,0 +1,39 @@
1
+ import {createRequire as topLevelCreateRequire} from 'module';
2
+ const require = topLevelCreateRequire(import.meta.url);
3
+ var __defProp = Object.defineProperty;
4
+ var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
5
+ /**
6
+ * @license
7
+ * Copyright 2026 Google LLC
8
+ * SPDX-License-Identifier: Apache-2.0
9
+ */
10
+ const INCOMING_HEADERS_GETTER = Object.freeze({
11
+ keys: /* @__PURE__ */ __name((carrier) => Object.keys(carrier), "keys"),
12
+ get(carrier, key) {
13
+ const name = key.toLowerCase();
14
+ for (const [headerName, value] of Object.entries(carrier)) {
15
+ if (headerName.toLowerCase() === name) {
16
+ return value;
17
+ }
18
+ }
19
+ return void 0;
20
+ }
21
+ });
22
+ function addMissingHeaders(input, init, injected) {
23
+ var _a;
24
+ const isRequest = typeof Request !== "undefined" && input instanceof Request;
25
+ const headers = new Headers(
26
+ (_a = init == null ? void 0 : init.headers) != null ? _a : isRequest ? input.headers : void 0
27
+ );
28
+ for (const [name, value] of Object.entries(injected)) {
29
+ if (!headers.has(name)) {
30
+ headers.set(name, value);
31
+ }
32
+ }
33
+ return headers;
34
+ }
35
+ __name(addMissingHeaders, "addMissingHeaders");
36
+ export {
37
+ INCOMING_HEADERS_GETTER,
38
+ addMissingHeaders
39
+ };
@@ -0,0 +1,96 @@
1
+ import {createRequire as topLevelCreateRequire} from 'module';
2
+ const require = topLevelCreateRequire(import.meta.url);
3
+ var __defProp = Object.defineProperty;
4
+ var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
5
+ /**
6
+ * @license
7
+ * Copyright 2026 Google LLC
8
+ * SPDX-License-Identifier: Apache-2.0
9
+ */
10
+ import {
11
+ AdkAttr,
12
+ ERROR_TYPE_ATTR,
13
+ GcpAttr,
14
+ GenAiAttr
15
+ } from "../common/attributes.js";
16
+ import { shouldAddContentToSpans } from "../common/config.js";
17
+ import { redactCredentialSecrets } from "../common/redaction.js";
18
+ import { safeJsonSerialize } from "../common/serialization.js";
19
+ const UNSPECIFIED = "<not specified>";
20
+ function serializeToolResult(result) {
21
+ try {
22
+ return safeJsonSerialize(redactCredentialSecrets(result));
23
+ } catch {
24
+ return "<not serializable>";
25
+ }
26
+ }
27
+ __name(serializeToolResult, "serializeToolResult");
28
+ function buildToolCallAttributes({
29
+ tool,
30
+ args,
31
+ agentName,
32
+ outcome,
33
+ config
34
+ }) {
35
+ var _a, _b, _c, _d;
36
+ const functionResponseEvent = outcome.kind === "response" ? outcome.functionResponseEvent : void 0;
37
+ const functionResponse = (_c = (_b = (_a = functionResponseEvent == null ? void 0 : functionResponseEvent.content) == null ? void 0 : _a.parts) == null ? void 0 : _b[0]) == null ? void 0 : _c.functionResponse;
38
+ const errorType = outcome.errorType;
39
+ const captureContent = shouldAddContentToSpans(config);
40
+ const result = functionResponse == null ? void 0 : functionResponse.response;
41
+ return {
42
+ [GenAiAttr.OPERATION_NAME]: "execute_tool",
43
+ [GenAiAttr.TOOL_DESCRIPTION]: tool.description,
44
+ [GenAiAttr.TOOL_NAME]: tool.name,
45
+ [GenAiAttr.TOOL_TYPE]: tool.constructor.name,
46
+ ...agentName ? { [GenAiAttr.AGENT_NAME]: agentName } : {},
47
+ ...errorType === void 0 ? {} : { [ERROR_TYPE_ATTR]: errorType },
48
+ [GenAiAttr.TOOL_CALL_ID]: (_d = functionResponse == null ? void 0 : functionResponse.id) != null ? _d : UNSPECIFIED,
49
+ ...functionResponseEvent ? { [GcpAttr.EVENT_ID]: functionResponseEvent.id } : {},
50
+ ...captureContent ? { [GenAiAttr.TOOL_CALL_ARGUMENTS]: safeJsonSerialize(args) } : {},
51
+ ...captureContent && result !== void 0 && result !== null ? { [GenAiAttr.TOOL_CALL_RESULT]: serializeToolResult(result) } : {}
52
+ };
53
+ }
54
+ __name(buildToolCallAttributes, "buildToolCallAttributes");
55
+ function buildMergedToolCallsAttributes(responseEventId) {
56
+ return {
57
+ [GenAiAttr.OPERATION_NAME]: "execute_tool",
58
+ [GenAiAttr.TOOL_NAME]: "(merged tools)",
59
+ [GenAiAttr.TOOL_DESCRIPTION]: "(merged tools)",
60
+ [GenAiAttr.TOOL_CALL_ID]: responseEventId,
61
+ [GcpAttr.EVENT_ID]: responseEventId
62
+ };
63
+ }
64
+ __name(buildMergedToolCallsAttributes, "buildMergedToolCallsAttributes");
65
+ function buildSkillSpanName(toolName, report) {
66
+ return report.resourceName === void 0 ? `execute_tool ${toolName} ${report.skillName}` : `execute_tool ${toolName} ${report.skillName} ${report.resourceName}`;
67
+ }
68
+ __name(buildSkillSpanName, "buildSkillSpanName");
69
+ function buildSkillAttributes(report) {
70
+ var _a;
71
+ const { skill } = report;
72
+ const additionalTools = (_a = skill == null ? void 0 : skill.frontmatter.metadata) == null ? void 0 : _a["adk_additional_tools"];
73
+ return {
74
+ [GenAiAttr.SKILL_NAME]: report.skillName,
75
+ ...skill ? { [GenAiAttr.SKILL_DESCRIPTION]: skill.frontmatter.description } : {},
76
+ ...report.resourceName === void 0 ? {} : { [GenAiAttr.SKILL_RESOURCE_NAME]: report.resourceName },
77
+ ...report.isSkillLoad && skill && Array.isArray(additionalTools) ? {
78
+ [AdkAttr.SKILL_ADDITIONAL_TOOLS]: additionalTools.filter(
79
+ (name) => typeof name === "string"
80
+ )
81
+ } : {}
82
+ };
83
+ }
84
+ __name(buildSkillAttributes, "buildSkillAttributes");
85
+ function setSkillSpanAttributes(span, toolName, report) {
86
+ span.updateName(buildSkillSpanName(toolName, report));
87
+ span.setAttributes(buildSkillAttributes(report));
88
+ }
89
+ __name(setSkillSpanAttributes, "setSkillSpanAttributes");
90
+ export {
91
+ buildMergedToolCallsAttributes,
92
+ buildSkillAttributes,
93
+ buildSkillSpanName,
94
+ buildToolCallAttributes,
95
+ setSkillSpanAttributes
96
+ };
@@ -0,0 +1,52 @@
1
+ import {createRequire as topLevelCreateRequire} from 'module';
2
+ const require = topLevelCreateRequire(import.meta.url);
3
+ var __defProp = Object.defineProperty;
4
+ var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
5
+ /**
6
+ * @license
7
+ * Copyright 2026 Google LLC
8
+ * SPDX-License-Identifier: Apache-2.0
9
+ */
10
+ import {
11
+ context,
12
+ SpanStatusCode,
13
+ trace
14
+ } from "@opentelemetry/api";
15
+ import { version } from "../../../../version.js";
16
+ import { GEN_AI_SYSTEM_VALUE, SCHEMA_URL } from "../common/attributes.js";
17
+ import { resolveErrorType } from "../common/error.js";
18
+ function getTracer() {
19
+ return trace.getTracerProvider().getTracer(GEN_AI_SYSTEM_VALUE, version, { schemaUrl: SCHEMA_URL });
20
+ }
21
+ __name(getTracer, "getTracer");
22
+ function bindOtelContextToAsyncGenerator(ctx, generator) {
23
+ return {
24
+ next: context.bind(ctx, generator.next.bind(generator)),
25
+ return: context.bind(ctx, generator.return.bind(generator)),
26
+ throw: context.bind(ctx, generator.throw.bind(generator)),
27
+ [Symbol.asyncIterator]() {
28
+ return bindOtelContextToAsyncGenerator(
29
+ ctx,
30
+ generator[Symbol.asyncIterator]()
31
+ );
32
+ }
33
+ };
34
+ }
35
+ __name(bindOtelContextToAsyncGenerator, "bindOtelContextToAsyncGenerator");
36
+ function setSpanError(span, error) {
37
+ const errorType = resolveErrorType(error);
38
+ let exception;
39
+ try {
40
+ exception = error instanceof Error ? error : errorType;
41
+ } catch {
42
+ exception = errorType;
43
+ }
44
+ span.recordException(exception);
45
+ span.setStatus({ code: SpanStatusCode.ERROR, message: errorType });
46
+ }
47
+ __name(setSpanError, "setSpanError");
48
+ export {
49
+ bindOtelContextToAsyncGenerator,
50
+ getTracer,
51
+ setSpanError
52
+ };
@@ -0,0 +1,54 @@
1
+ import {createRequire as topLevelCreateRequire} from 'module';
2
+ const require = topLevelCreateRequire(import.meta.url);
3
+ var __defProp = Object.defineProperty;
4
+ var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
5
+ /**
6
+ * @license
7
+ * Copyright 2026 Google LLC
8
+ * SPDX-License-Identifier: Apache-2.0
9
+ */
10
+ import {
11
+ context,
12
+ createContextKey,
13
+ trace
14
+ } from "@opentelemetry/api";
15
+ import { GenAiAttr, SUPPORT_ID_ATTR } from "../common/attributes.js";
16
+ import { getTracer } from "./tracer.js";
17
+ const SUPPORT_ID_KEY = createContextKey("google.adk.agent_engine.support_id");
18
+ function getSupportId(parentContext, nested) {
19
+ if (nested) {
20
+ return void 0;
21
+ }
22
+ const parentSpanContext = trace.getSpanContext(parentContext);
23
+ if (parentSpanContext && !parentSpanContext.isRemote) {
24
+ return void 0;
25
+ }
26
+ const supportId = parentContext.getValue(SUPPORT_ID_KEY);
27
+ return typeof supportId === "string" ? supportId : void 0;
28
+ }
29
+ __name(getSupportId, "getSupportId");
30
+ function startWorkflowSpan({
31
+ workflowName,
32
+ sessionId,
33
+ nested,
34
+ parentContext = context.active()
35
+ }) {
36
+ const supportId = getSupportId(parentContext, nested);
37
+ return getTracer().startSpan(
38
+ workflowName ? `invoke_workflow ${workflowName}` : "invoke_workflow",
39
+ {
40
+ attributes: {
41
+ [GenAiAttr.OPERATION_NAME]: "invoke_workflow",
42
+ [GenAiAttr.CONVERSATION_ID]: sessionId,
43
+ ...nested ? { [GenAiAttr.WORKFLOW_NESTED]: true } : {},
44
+ ...workflowName ? { [GenAiAttr.WORKFLOW_NAME]: workflowName } : {},
45
+ ...supportId ? { [SUPPORT_ID_ATTR]: supportId } : {}
46
+ }
47
+ },
48
+ parentContext
49
+ );
50
+ }
51
+ __name(startWorkflowSpan, "startWorkflowSpan");
52
+ export {
53
+ startWorkflowSpan
54
+ };
@@ -0,0 +1,67 @@
1
+ import {createRequire as topLevelCreateRequire} from 'module';
2
+ const require = topLevelCreateRequire(import.meta.url);
3
+ var __defProp = Object.defineProperty;
4
+ var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
5
+ /**
6
+ * @license
7
+ * Copyright 2026 Google LLC
8
+ * SPDX-License-Identifier: Apache-2.0
9
+ */
10
+ import {
11
+ context,
12
+ createContextKey,
13
+ trace
14
+ } from "@opentelemetry/api";
15
+ const AGENT_ENGINE_TRACEPARENT_HEADER = "google-agent-engine-traceparent";
16
+ const TRACEPARENT_HEADER = "traceparent";
17
+ const MAX_SUPPORT_ID_LENGTH = 256;
18
+ const SUPPORT_ID_KEY = createContextKey("google.adk.agent_engine.support_id");
19
+ const TRACEPARENT_PATTERN = /^\s?([0-9a-f]{2})-([0-9a-f]{32})-([0-9a-f]{16})-([0-9a-f]{2})(-.*)?\s?$/;
20
+ function parseTraceparent(value) {
21
+ const match = TRACEPARENT_PATTERN.exec(value);
22
+ if (!match) {
23
+ return void 0;
24
+ }
25
+ const [, version, traceId, spanId, flags, rest] = match;
26
+ if (version === "ff" || version === "00" && rest) {
27
+ return void 0;
28
+ }
29
+ const spanContext = {
30
+ traceId,
31
+ spanId,
32
+ traceFlags: parseInt(flags, 16),
33
+ isRemote: true
34
+ };
35
+ return trace.isSpanContextValid(spanContext) ? spanContext : void 0;
36
+ }
37
+ __name(parseTraceparent, "parseTraceparent");
38
+ function getHeader(headers, name) {
39
+ for (const [key, value] of Object.entries(headers)) {
40
+ if (key.toLowerCase() === name) {
41
+ return Array.isArray(value) ? value[0] : value;
42
+ }
43
+ }
44
+ return void 0;
45
+ }
46
+ __name(getHeader, "getHeader");
47
+ function runWithAgentEngineContext(headers, fn) {
48
+ var _a;
49
+ let ctx = context.active();
50
+ const agentEngineTraceparent = getHeader(
51
+ headers,
52
+ AGENT_ENGINE_TRACEPARENT_HEADER
53
+ );
54
+ const remoteParent = agentEngineTraceparent === void 0 ? void 0 : parseTraceparent(agentEngineTraceparent);
55
+ if (remoteParent && !trace.getSpan(ctx)) {
56
+ ctx = trace.setSpanContext(ctx, remoteParent);
57
+ }
58
+ const supportId = (_a = getHeader(headers, TRACEPARENT_HEADER)) == null ? void 0 : _a.trim();
59
+ if (supportId && supportId.length <= MAX_SUPPORT_ID_LENGTH) {
60
+ ctx = ctx.setValue(SUPPORT_ID_KEY, supportId);
61
+ }
62
+ return context.with(ctx, fn);
63
+ }
64
+ __name(runWithAgentEngineContext, "runWithAgentEngineContext");
65
+ export {
66
+ runWithAgentEngineContext
67
+ };
@@ -0,0 +1,46 @@
1
+ import {createRequire as topLevelCreateRequire} from 'module';
2
+ const require = topLevelCreateRequire(import.meta.url);
3
+ var __defProp = Object.defineProperty;
4
+ var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
5
+ /**
6
+ * @license
7
+ * Copyright 2026 Google LLC
8
+ * SPDX-License-Identifier: Apache-2.0
9
+ */
10
+ import { formatError } from "../../utils/error_utils.js";
11
+ import { logger } from "../../utils/logger.js";
12
+ import { isRequestDrivenMetricReader } from "./request_driven_metric_reader.js";
13
+ function agentRuntimeRequestHooks(otelHooks) {
14
+ var _a, _b, _c;
15
+ const readers = ((_a = otelHooks.metricReaders) != null ? _a : []).filter(
16
+ isRequestDrivenMetricReader
17
+ );
18
+ const flushables = [
19
+ ...(_b = otelHooks.spanProcessors) != null ? _b : [],
20
+ ...(_c = otelHooks.logRecordProcessors) != null ? _c : []
21
+ ];
22
+ return {
23
+ onRequestStart() {
24
+ for (const reader of readers) {
25
+ reader.onRequestStart();
26
+ }
27
+ },
28
+ async onRequestEnd() {
29
+ const results = await Promise.allSettled([
30
+ ...readers.map(async (reader) => reader.onRequestEnd()),
31
+ ...flushables.map(async (processor) => processor.forceFlush())
32
+ ]);
33
+ for (const result of results) {
34
+ if (result.status === "rejected") {
35
+ logger.warn(
36
+ `Telemetry flush on request end failed: ${formatError(result.reason)}`
37
+ );
38
+ }
39
+ }
40
+ }
41
+ };
42
+ }
43
+ __name(agentRuntimeRequestHooks, "agentRuntimeRequestHooks");
44
+ export {
45
+ agentRuntimeRequestHooks
46
+ };