@copilotkit/runtime 1.57.3 → 1.58.0-canary.thread-id-propagation

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 (346) hide show
  1. package/dist/lib/observability.d.cts +1 -1
  2. package/dist/lib/observability.d.cts.map +1 -1
  3. package/dist/lib/observability.d.mts +1 -1
  4. package/dist/lib/observability.d.mts.map +1 -1
  5. package/dist/lib/runtime/copilot-runtime.cjs +2 -0
  6. package/dist/lib/runtime/copilot-runtime.cjs.map +1 -1
  7. package/dist/lib/runtime/copilot-runtime.d.cts.map +1 -1
  8. package/dist/lib/runtime/copilot-runtime.d.mts.map +1 -1
  9. package/dist/lib/runtime/copilot-runtime.mjs +2 -0
  10. package/dist/lib/runtime/copilot-runtime.mjs.map +1 -1
  11. package/dist/package.cjs +5 -8
  12. package/dist/package.mjs +5 -8
  13. package/dist/v2/runtime/core/runtime.cjs +4 -1
  14. package/dist/v2/runtime/core/runtime.cjs.map +1 -1
  15. package/dist/v2/runtime/core/runtime.d.cts.map +1 -1
  16. package/dist/v2/runtime/core/runtime.d.mts.map +1 -1
  17. package/dist/v2/runtime/core/runtime.mjs +4 -1
  18. package/dist/v2/runtime/core/runtime.mjs.map +1 -1
  19. package/dist/v2/runtime/handlers/get-runtime-info.cjs +1 -1
  20. package/dist/v2/runtime/handlers/get-runtime-info.mjs +1 -1
  21. package/dist/v2/runtime/handlers/handle-connect.cjs +1 -1
  22. package/dist/v2/runtime/handlers/handle-connect.mjs +1 -1
  23. package/dist/v2/runtime/handlers/handle-run.cjs +1 -1
  24. package/dist/v2/runtime/handlers/handle-run.mjs +1 -1
  25. package/dist/v2/runtime/handlers/shared/agent-utils.cjs +1 -1
  26. package/dist/v2/runtime/handlers/shared/agent-utils.cjs.map +1 -1
  27. package/dist/v2/runtime/handlers/shared/agent-utils.mjs +1 -1
  28. package/dist/v2/runtime/handlers/shared/agent-utils.mjs.map +1 -1
  29. package/dist/v2/runtime/telemetry/telemetry-client.cjs +22 -6
  30. package/dist/v2/runtime/telemetry/telemetry-client.cjs.map +1 -1
  31. package/dist/v2/runtime/telemetry/telemetry-client.mjs +27 -11
  32. package/dist/v2/runtime/telemetry/telemetry-client.mjs.map +1 -1
  33. package/package.json +9 -19
  34. package/skills/runtime/SKILL.md +98 -0
  35. package/skills/runtime/references/agent-runners-custom.md +161 -0
  36. package/skills/runtime/references/agent-runners-in-memory.md +64 -0
  37. package/skills/runtime/references/agent-runners-sqlite.md +90 -0
  38. package/skills/runtime/references/agent-runners.md +304 -0
  39. package/skills/runtime/references/built-in-agent-factory-modes.md +232 -0
  40. package/skills/runtime/references/built-in-agent-helper-utilities.md +123 -0
  41. package/skills/runtime/references/built-in-agent-model-identifiers.md +59 -0
  42. package/skills/runtime/references/built-in-agent.md +523 -0
  43. package/skills/runtime/references/intelligence-mode.md +336 -0
  44. package/skills/runtime/references/middleware.md +376 -0
  45. package/skills/runtime/references/server-side-tools.md +414 -0
  46. package/skills/runtime/references/setup-endpoint.md +503 -0
  47. package/skills/runtime/references/transcription.md +287 -0
  48. package/skills/runtime/references/wiring-a2a.md +40 -0
  49. package/skills/runtime/references/wiring-adk.md +45 -0
  50. package/skills/runtime/references/wiring-ag2.md +41 -0
  51. package/skills/runtime/references/wiring-agno.md +39 -0
  52. package/skills/runtime/references/wiring-aws-strands.md +59 -0
  53. package/skills/runtime/references/wiring-crewai-crews.md +51 -0
  54. package/skills/runtime/references/wiring-crewai-flows.md +45 -0
  55. package/skills/runtime/references/wiring-external-agents.md +348 -0
  56. package/skills/runtime/references/wiring-langgraph.md +50 -0
  57. package/skills/runtime/references/wiring-llamaindex.md +39 -0
  58. package/skills/runtime/references/wiring-mastra.md +70 -0
  59. package/skills/runtime/references/wiring-mcp-apps-middleware.md +68 -0
  60. package/skills/runtime/references/wiring-ms-agent-framework.md +41 -0
  61. package/skills/runtime/references/wiring-pydantic-ai.md +45 -0
  62. package/CHANGELOG.md +0 -3624
  63. package/__snapshots__/schema/schema.graphql +0 -371
  64. package/dist/v2/runtime/telemetry/scarf-client.cjs +0 -32
  65. package/dist/v2/runtime/telemetry/scarf-client.cjs.map +0 -1
  66. package/dist/v2/runtime/telemetry/scarf-client.mjs +0 -32
  67. package/dist/v2/runtime/telemetry/scarf-client.mjs.map +0 -1
  68. package/scripts/generate-gql-schema.ts +0 -16
  69. package/src/agent/__tests__/agent-test-helpers.ts +0 -476
  70. package/src/agent/__tests__/agent.test.ts +0 -593
  71. package/src/agent/__tests__/ai-sdk-v6-compat.test.ts +0 -116
  72. package/src/agent/__tests__/basic-agent.test.ts +0 -1698
  73. package/src/agent/__tests__/capabilities.test.ts +0 -81
  74. package/src/agent/__tests__/config-tools-execution.test.ts +0 -516
  75. package/src/agent/__tests__/converter-aisdk.test.ts +0 -692
  76. package/src/agent/__tests__/converter-custom.test.ts +0 -319
  77. package/src/agent/__tests__/converter-tanstack-input.test.ts +0 -211
  78. package/src/agent/__tests__/converter-tanstack.test.ts +0 -594
  79. package/src/agent/__tests__/mcp-clients.test.ts +0 -246
  80. package/src/agent/__tests__/mcp-servers-integration.test.ts +0 -373
  81. package/src/agent/__tests__/multimodal-tanstack.test.ts +0 -284
  82. package/src/agent/__tests__/multimodal.test.ts +0 -176
  83. package/src/agent/__tests__/property-overrides.test.ts +0 -598
  84. package/src/agent/__tests__/provider-id-collision.test.ts +0 -195
  85. package/src/agent/__tests__/standard-schema-tools.test.ts +0 -313
  86. package/src/agent/__tests__/standard-schema-types.test.ts +0 -158
  87. package/src/agent/__tests__/state-tools.test.ts +0 -436
  88. package/src/agent/__tests__/test-helpers.ts +0 -197
  89. package/src/agent/__tests__/utils.test.ts +0 -536
  90. package/src/agent/__tests__/zod-regression.test.ts +0 -350
  91. package/src/agent/converters/aisdk.ts +0 -326
  92. package/src/agent/converters/index.ts +0 -7
  93. package/src/agent/converters/tanstack.ts +0 -451
  94. package/src/agent/index.ts +0 -1743
  95. package/src/agents/langgraph/__tests__/event-source.test.ts +0 -256
  96. package/src/agents/langgraph/event-source.ts +0 -365
  97. package/src/agents/langgraph/events.ts +0 -394
  98. package/src/graphql/inputs/action.input.ts +0 -16
  99. package/src/graphql/inputs/agent-session.input.ts +0 -13
  100. package/src/graphql/inputs/agent-state.input.ts +0 -13
  101. package/src/graphql/inputs/cloud-guardrails.input.ts +0 -16
  102. package/src/graphql/inputs/cloud.input.ts +0 -8
  103. package/src/graphql/inputs/context-property.input.ts +0 -10
  104. package/src/graphql/inputs/copilot-context.input.ts +0 -10
  105. package/src/graphql/inputs/custom-property.input.ts +0 -15
  106. package/src/graphql/inputs/extensions.input.ts +0 -21
  107. package/src/graphql/inputs/forwarded-parameters.input.ts +0 -22
  108. package/src/graphql/inputs/frontend.input.ts +0 -14
  109. package/src/graphql/inputs/generate-copilot-response.input.ts +0 -59
  110. package/src/graphql/inputs/load-agent-state.input.ts +0 -10
  111. package/src/graphql/inputs/message.input.ts +0 -110
  112. package/src/graphql/inputs/meta-event.input.ts +0 -18
  113. package/src/graphql/message-conversion/agui-to-gql.test.ts +0 -1384
  114. package/src/graphql/message-conversion/agui-to-gql.ts +0 -384
  115. package/src/graphql/message-conversion/gql-to-agui.test.ts +0 -1653
  116. package/src/graphql/message-conversion/gql-to-agui.ts +0 -297
  117. package/src/graphql/message-conversion/index.ts +0 -2
  118. package/src/graphql/message-conversion/roundtrip-conversion.test.ts +0 -561
  119. package/src/graphql/resolvers/__tests__/resolve-message-id.test.ts +0 -25
  120. package/src/graphql/resolvers/copilot.resolver.ts +0 -785
  121. package/src/graphql/resolvers/resolve-message-id.ts +0 -14
  122. package/src/graphql/resolvers/state.resolver.ts +0 -30
  123. package/src/graphql/types/agents-response.type.ts +0 -19
  124. package/src/graphql/types/base/index.ts +0 -10
  125. package/src/graphql/types/converted/index.ts +0 -183
  126. package/src/graphql/types/copilot-response.type.ts +0 -141
  127. package/src/graphql/types/enums.ts +0 -38
  128. package/src/graphql/types/extensions-response.type.ts +0 -23
  129. package/src/graphql/types/guardrails-result.type.ts +0 -20
  130. package/src/graphql/types/load-agent-state-response.type.ts +0 -17
  131. package/src/graphql/types/message-status.type.ts +0 -48
  132. package/src/graphql/types/meta-events.type.ts +0 -78
  133. package/src/graphql/types/response-status.type.ts +0 -77
  134. package/src/index.ts +0 -3
  135. package/src/langgraph.ts +0 -1
  136. package/src/lib/__tests__/telemetry-disclosure.test.ts +0 -55
  137. package/src/lib/cloud/index.ts +0 -4
  138. package/src/lib/error-messages.ts +0 -211
  139. package/src/lib/index.ts +0 -52
  140. package/src/lib/integrations/index.ts +0 -6
  141. package/src/lib/integrations/nest/index.ts +0 -21
  142. package/src/lib/integrations/nextjs/app-router.ts +0 -47
  143. package/src/lib/integrations/nextjs/pages-router.ts +0 -45
  144. package/src/lib/integrations/node-express/index.ts +0 -21
  145. package/src/lib/integrations/node-http/__tests__/request-duck-type.test.ts +0 -66
  146. package/src/lib/integrations/node-http/index.ts +0 -187
  147. package/src/lib/integrations/node-http/request-handler.ts +0 -128
  148. package/src/lib/integrations/shared.ts +0 -112
  149. package/src/lib/logger.ts +0 -31
  150. package/src/lib/observability.ts +0 -167
  151. package/src/lib/runtime/__tests__/copilot-runtime-error.test.ts +0 -183
  152. package/src/lib/runtime/__tests__/handle-service-adapter.test.ts +0 -108
  153. package/src/lib/runtime/__tests__/mcp-tools-utils.test.ts +0 -499
  154. package/src/lib/runtime/__tests__/on-after-request.test.ts +0 -122
  155. package/src/lib/runtime/__tests__/retry-utils.test.ts +0 -137
  156. package/src/lib/runtime/__tests__/v1-agent-factory.test.ts +0 -109
  157. package/src/lib/runtime/agent-integrations/langgraph/__tests__/dispatch-event-filtering.test.ts +0 -345
  158. package/src/lib/runtime/agent-integrations/langgraph/__tests__/run-message-filtering.test.ts +0 -156
  159. package/src/lib/runtime/agent-integrations/langgraph/agent.ts +0 -263
  160. package/src/lib/runtime/agent-integrations/langgraph/consts.ts +0 -37
  161. package/src/lib/runtime/agent-integrations/langgraph/index.ts +0 -2
  162. package/src/lib/runtime/copilot-runtime.ts +0 -863
  163. package/src/lib/runtime/mcp-tools-utils.ts +0 -313
  164. package/src/lib/runtime/retry-utils.ts +0 -141
  165. package/src/lib/runtime/telemetry-agent-runner.ts +0 -151
  166. package/src/lib/runtime/types.ts +0 -48
  167. package/src/lib/runtime/utils.ts +0 -93
  168. package/src/lib/streaming.ts +0 -220
  169. package/src/lib/telemetry-client.ts +0 -66
  170. package/src/lib/telemetry-disclosure.ts +0 -53
  171. package/src/service-adapters/anthropic/anthropic-adapter.ts +0 -532
  172. package/src/service-adapters/anthropic/utils.ts +0 -219
  173. package/src/service-adapters/bedrock/bedrock-adapter.ts +0 -73
  174. package/src/service-adapters/conversion.test.ts +0 -56
  175. package/src/service-adapters/conversion.ts +0 -69
  176. package/src/service-adapters/empty/empty-adapter.ts +0 -38
  177. package/src/service-adapters/events.ts +0 -337
  178. package/src/service-adapters/experimental/ollama/ollama-adapter.ts +0 -84
  179. package/src/service-adapters/google/google-genai-adapter.test.ts +0 -151
  180. package/src/service-adapters/google/google-genai-adapter.ts +0 -95
  181. package/src/service-adapters/groq/groq-adapter.ts +0 -229
  182. package/src/service-adapters/index.ts +0 -18
  183. package/src/service-adapters/langchain/langchain-adapter.ts +0 -113
  184. package/src/service-adapters/langchain/langserve.ts +0 -88
  185. package/src/service-adapters/langchain/types.ts +0 -20
  186. package/src/service-adapters/langchain/utils.ts +0 -330
  187. package/src/service-adapters/openai/__tests__/openai-v5-compat.test.ts +0 -177
  188. package/src/service-adapters/openai/openai-adapter.ts +0 -324
  189. package/src/service-adapters/openai/openai-assistant-adapter.ts +0 -385
  190. package/src/service-adapters/openai/utils.ts +0 -305
  191. package/src/service-adapters/service-adapter.ts +0 -50
  192. package/src/service-adapters/shared/error-utils.ts +0 -64
  193. package/src/service-adapters/shared/index.ts +0 -2
  194. package/src/service-adapters/shared/sdk-client-utils.ts +0 -19
  195. package/src/service-adapters/unify/unify-adapter.ts +0 -165
  196. package/src/utils/failed-response-status-reasons.ts +0 -70
  197. package/src/utils/index.ts +0 -1
  198. package/src/v2/express.ts +0 -1
  199. package/src/v2/hono.ts +0 -1
  200. package/src/v2/index.ts +0 -5
  201. package/src/v2/node.ts +0 -1
  202. package/src/v2/runtime/__tests__/agents-factory.test.ts +0 -136
  203. package/src/v2/runtime/__tests__/backward-compat.test.ts +0 -261
  204. package/src/v2/runtime/__tests__/code-review-fixes.test.ts +0 -500
  205. package/src/v2/runtime/__tests__/cors-credentials.test.ts +0 -320
  206. package/src/v2/runtime/__tests__/debug-sse-response.test.ts +0 -302
  207. package/src/v2/runtime/__tests__/express-adapter.test.ts +0 -188
  208. package/src/v2/runtime/__tests__/express-body-order.test.ts +0 -76
  209. package/src/v2/runtime/__tests__/express-fetch-bridge.test.ts +0 -344
  210. package/src/v2/runtime/__tests__/express-single-sse.test.ts +0 -122
  211. package/src/v2/runtime/__tests__/express-single-telemetry.integration.test.ts +0 -65
  212. package/src/v2/runtime/__tests__/express-telemetry.integration.test.ts +0 -101
  213. package/src/v2/runtime/__tests__/fetch-cors.test.ts +0 -205
  214. package/src/v2/runtime/__tests__/fetch-handler-validation.test.ts +0 -440
  215. package/src/v2/runtime/__tests__/fetch-handler.test.ts +0 -456
  216. package/src/v2/runtime/__tests__/fetch-router.test.ts +0 -276
  217. package/src/v2/runtime/__tests__/get-runtime-info.test.ts +0 -335
  218. package/src/v2/runtime/__tests__/handle-connect.test.ts +0 -585
  219. package/src/v2/runtime/__tests__/handle-run.test.ts +0 -1388
  220. package/src/v2/runtime/__tests__/handle-threads.test.ts +0 -930
  221. package/src/v2/runtime/__tests__/handle-transcribe.test.ts +0 -301
  222. package/src/v2/runtime/__tests__/header-utils.test.ts +0 -88
  223. package/src/v2/runtime/__tests__/hono-adapter.test.ts +0 -150
  224. package/src/v2/runtime/__tests__/hono-single-telemetry.integration.test.ts +0 -46
  225. package/src/v2/runtime/__tests__/hono-telemetry.integration.test.ts +0 -99
  226. package/src/v2/runtime/__tests__/hooks-edge-cases.test.ts +0 -457
  227. package/src/v2/runtime/__tests__/hooks.test.ts +0 -557
  228. package/src/v2/runtime/__tests__/in-process-agent-runner-messages.test.ts +0 -230
  229. package/src/v2/runtime/__tests__/in-process-agent-runner.test.ts +0 -1030
  230. package/src/v2/runtime/__tests__/integration/bun/bun-servers.integration.test.ts +0 -27
  231. package/src/v2/runtime/__tests__/integration/bun/elysia-multi.ts +0 -32
  232. package/src/v2/runtime/__tests__/integration/bun/elysia-single.ts +0 -33
  233. package/src/v2/runtime/__tests__/integration/bun/hono-bun-multi.ts +0 -25
  234. package/src/v2/runtime/__tests__/integration/bun/hono-bun-single.ts +0 -32
  235. package/src/v2/runtime/__tests__/integration/helpers/create-test-runtime.ts +0 -15
  236. package/src/v2/runtime/__tests__/integration/helpers/sse-reader.ts +0 -45
  237. package/src/v2/runtime/__tests__/integration/helpers/test-agent.ts +0 -58
  238. package/src/v2/runtime/__tests__/integration/node-servers.integration.test.ts +0 -58
  239. package/src/v2/runtime/__tests__/integration/servers/express-multi.ts +0 -35
  240. package/src/v2/runtime/__tests__/integration/servers/express-single.ts +0 -36
  241. package/src/v2/runtime/__tests__/integration/servers/fetch-direct.ts +0 -39
  242. package/src/v2/runtime/__tests__/integration/servers/hono-multi.ts +0 -30
  243. package/src/v2/runtime/__tests__/integration/servers/hono-single.ts +0 -37
  244. package/src/v2/runtime/__tests__/integration/servers/node-multi.ts +0 -45
  245. package/src/v2/runtime/__tests__/integration/servers/node-single.ts +0 -46
  246. package/src/v2/runtime/__tests__/integration/servers/types.ts +0 -18
  247. package/src/v2/runtime/__tests__/integration/suites/debug-events.suite.ts +0 -253
  248. package/src/v2/runtime/__tests__/integration/suites/multi-endpoint.suite.ts +0 -358
  249. package/src/v2/runtime/__tests__/integration/suites/single-endpoint.suite.ts +0 -363
  250. package/src/v2/runtime/__tests__/intelligence-run-telemetry.test.ts +0 -194
  251. package/src/v2/runtime/__tests__/mcp-apps-middleware-integration.test.ts +0 -275
  252. package/src/v2/runtime/__tests__/middleware-express.test.ts +0 -208
  253. package/src/v2/runtime/__tests__/middleware-single-express.test.ts +0 -213
  254. package/src/v2/runtime/__tests__/middleware-single.test.ts +0 -225
  255. package/src/v2/runtime/__tests__/middleware-sse-parser.test.ts +0 -237
  256. package/src/v2/runtime/__tests__/middleware.test.ts +0 -250
  257. package/src/v2/runtime/__tests__/node-fetch-handler.test.ts +0 -157
  258. package/src/v2/runtime/__tests__/open-generative-ui-middleware.e2e.test.ts +0 -728
  259. package/src/v2/runtime/__tests__/router-edge-cases.test.ts +0 -217
  260. package/src/v2/runtime/__tests__/routing-express.test.ts +0 -174
  261. package/src/v2/runtime/__tests__/routing-single-express.test.ts +0 -168
  262. package/src/v2/runtime/__tests__/routing-single.test.ts +0 -193
  263. package/src/v2/runtime/__tests__/routing.test.ts +0 -257
  264. package/src/v2/runtime/__tests__/runtime.test.ts +0 -234
  265. package/src/v2/runtime/__tests__/sse-response-telemetry.test.ts +0 -108
  266. package/src/v2/runtime/__tests__/telemetry.test.ts +0 -167
  267. package/src/v2/runtime/__tests__/thread-names.test.ts +0 -188
  268. package/src/v2/runtime/core/__tests__/debug-event-bus.test.ts +0 -156
  269. package/src/v2/runtime/core/debug-event-bus.ts +0 -45
  270. package/src/v2/runtime/core/fetch-cors.ts +0 -136
  271. package/src/v2/runtime/core/fetch-handler.ts +0 -492
  272. package/src/v2/runtime/core/fetch-router.ts +0 -203
  273. package/src/v2/runtime/core/hooks.ts +0 -160
  274. package/src/v2/runtime/core/middleware-sse-parser.ts +0 -210
  275. package/src/v2/runtime/core/middleware.ts +0 -115
  276. package/src/v2/runtime/core/runtime.ts +0 -432
  277. package/src/v2/runtime/endpoints/express-fetch-bridge.ts +0 -137
  278. package/src/v2/runtime/endpoints/express-single.ts +0 -54
  279. package/src/v2/runtime/endpoints/express.ts +0 -179
  280. package/src/v2/runtime/endpoints/hono-single.ts +0 -60
  281. package/src/v2/runtime/endpoints/hono.ts +0 -89
  282. package/src/v2/runtime/endpoints/index.ts +0 -4
  283. package/src/v2/runtime/endpoints/node-fetch-handler.ts +0 -48
  284. package/src/v2/runtime/endpoints/node.ts +0 -28
  285. package/src/v2/runtime/endpoints/single-route-helpers.ts +0 -125
  286. package/src/v2/runtime/express.ts +0 -2
  287. package/src/v2/runtime/handlers/__tests__/handle-debug-events.test.ts +0 -176
  288. package/src/v2/runtime/handlers/get-runtime-info.ts +0 -101
  289. package/src/v2/runtime/handlers/handle-connect.ts +0 -80
  290. package/src/v2/runtime/handlers/handle-debug-events.ts +0 -52
  291. package/src/v2/runtime/handlers/handle-run.ts +0 -111
  292. package/src/v2/runtime/handlers/handle-stop.ts +0 -77
  293. package/src/v2/runtime/handlers/handle-threads.ts +0 -11
  294. package/src/v2/runtime/handlers/handle-transcribe.ts +0 -269
  295. package/src/v2/runtime/handlers/header-utils.ts +0 -24
  296. package/src/v2/runtime/handlers/intelligence/connect.ts +0 -102
  297. package/src/v2/runtime/handlers/intelligence/run.ts +0 -351
  298. package/src/v2/runtime/handlers/intelligence/thread-names.ts +0 -246
  299. package/src/v2/runtime/handlers/intelligence/threads.ts +0 -420
  300. package/src/v2/runtime/handlers/shared/agent-utils.ts +0 -154
  301. package/src/v2/runtime/handlers/shared/intelligence-utils.ts +0 -41
  302. package/src/v2/runtime/handlers/shared/json-response.ts +0 -9
  303. package/src/v2/runtime/handlers/shared/resolve-intelligence-user.ts +0 -28
  304. package/src/v2/runtime/handlers/shared/sse-response.ts +0 -215
  305. package/src/v2/runtime/handlers/sse/__tests__/sse-connect-agent-id.test.ts +0 -71
  306. package/src/v2/runtime/handlers/sse/connect.ts +0 -30
  307. package/src/v2/runtime/handlers/sse/run.ts +0 -40
  308. package/src/v2/runtime/hono.ts +0 -2
  309. package/src/v2/runtime/index.ts +0 -51
  310. package/src/v2/runtime/intelligence-platform/__tests__/client.test.ts +0 -601
  311. package/src/v2/runtime/intelligence-platform/__tests__/intelligence-mcp-helper.test.ts +0 -246
  312. package/src/v2/runtime/intelligence-platform/client.ts +0 -818
  313. package/src/v2/runtime/intelligence-platform/index.ts +0 -10
  314. package/src/v2/runtime/node.ts +0 -6
  315. package/src/v2/runtime/open-generative-ui-middleware.ts +0 -373
  316. package/src/v2/runtime/runner/__tests__/finalize-events.test.ts +0 -109
  317. package/src/v2/runtime/runner/__tests__/in-memory-runner.e2e.test.ts +0 -775
  318. package/src/v2/runtime/runner/__tests__/in-memory-runner.test.ts +0 -777
  319. package/src/v2/runtime/runner/__tests__/intelligence-runner.test.ts +0 -1039
  320. package/src/v2/runtime/runner/agent-runner.ts +0 -35
  321. package/src/v2/runtime/runner/in-memory.ts +0 -467
  322. package/src/v2/runtime/runner/index.ts +0 -4
  323. package/src/v2/runtime/runner/intelligence.ts +0 -498
  324. package/src/v2/runtime/telemetry/__tests__/instance-created.test.ts +0 -96
  325. package/src/v2/runtime/telemetry/events.ts +0 -35
  326. package/src/v2/runtime/telemetry/index.ts +0 -7
  327. package/src/v2/runtime/telemetry/instance-created.ts +0 -44
  328. package/src/v2/runtime/telemetry/scarf-client.ts +0 -39
  329. package/src/v2/runtime/telemetry/telemetry-client.ts +0 -70
  330. package/src/v2/runtime/transcription-service/transcription-service.ts +0 -11
  331. package/tests/global.d.ts +0 -1
  332. package/tests/service-adapters/anthropic/allowlist-approach.test.ts +0 -259
  333. package/tests/service-adapters/anthropic/anthropic-adapter-language-model.test.ts +0 -101
  334. package/tests/service-adapters/anthropic/anthropic-adapter.test.ts +0 -645
  335. package/tests/service-adapters/anthropic/utils-token-trimming.test.ts +0 -301
  336. package/tests/service-adapters/groq/groq-adapter-language-model.test.ts +0 -102
  337. package/tests/service-adapters/openai/allowlist-approach.test.ts +0 -294
  338. package/tests/service-adapters/openai/openai-adapter-language-model.test.ts +0 -122
  339. package/tests/service-adapters/openai/openai-adapter.test.ts +0 -291
  340. package/tests/service-adapters/shared/sdk-client-utils.test.ts +0 -36
  341. package/tests/setup.vitest.ts +0 -8
  342. package/tests/tsconfig.json +0 -10
  343. package/tsconfig.json +0 -20
  344. package/tsdown.config.ts +0 -45
  345. package/typedoc.json +0 -4
  346. package/vitest.config.mjs +0 -13
@@ -1,728 +0,0 @@
1
- import { describe, it, expect } from "vitest";
2
- import {
3
- AbstractAgent,
4
- BaseEvent,
5
- EventType,
6
- RunAgentInput,
7
- ActivitySnapshotEvent,
8
- ActivityDeltaEvent,
9
- } from "@ag-ui/client";
10
- import { Observable, firstValueFrom } from "rxjs";
11
- import { toArray } from "rxjs/operators";
12
- import {
13
- OpenGenerativeUIMiddleware,
14
- ArgsParser,
15
- } from "../open-generative-ui-middleware";
16
-
17
- /**
18
- * A minimal agent that records the input it receives and emits scripted events.
19
- */
20
- class MockAgent extends AbstractAgent {
21
- public receivedInput: RunAgentInput | null = null;
22
-
23
- constructor(private readonly events: BaseEvent[]) {
24
- super();
25
- }
26
-
27
- run(input: RunAgentInput): Observable<BaseEvent> {
28
- this.receivedInput = input;
29
- return new Observable<BaseEvent>((subscriber) => {
30
- for (const event of this.events) {
31
- subscriber.next(event);
32
- }
33
- subscriber.complete();
34
- });
35
- }
36
-
37
- clone(): AbstractAgent {
38
- return new MockAgent([...this.events]);
39
- }
40
-
41
- protected connect(): ReturnType<AbstractAgent["connect"]> {
42
- throw new Error("not used");
43
- }
44
- }
45
-
46
- function createRunInput(overrides: Partial<RunAgentInput> = {}): RunAgentInput {
47
- return {
48
- threadId: "thread-1",
49
- runId: "run-1",
50
- state: {},
51
- messages: [],
52
- tools: [],
53
- context: [],
54
- forwardedProps: undefined,
55
- ...overrides,
56
- };
57
- }
58
-
59
- async function collectEvents(
60
- observable: Observable<BaseEvent>,
61
- ): Promise<BaseEvent[]> {
62
- return firstValueFrom(observable.pipe(toArray()));
63
- }
64
-
65
- describe("OpenGenerativeUIMiddleware e2e", () => {
66
- describe("Tool passthrough", () => {
67
- it("does not modify the tools list", async () => {
68
- const middleware = new OpenGenerativeUIMiddleware();
69
- const agent = new MockAgent([
70
- {
71
- type: EventType.RUN_STARTED,
72
- threadId: "thread-1",
73
- runId: "run-1",
74
- } as BaseEvent,
75
- {
76
- type: EventType.RUN_FINISHED,
77
- threadId: "thread-1",
78
- runId: "run-1",
79
- } as BaseEvent,
80
- ]);
81
-
82
- const input = createRunInput();
83
- await collectEvents(middleware.run(input, agent));
84
-
85
- expect(agent.receivedInput!.tools).toEqual(input.tools);
86
- });
87
- });
88
-
89
- describe("ArgsParser (streaming JSON via clarinet)", () => {
90
- const noop = () => {};
91
-
92
- it("parses a complete JSON object in one chunk", () => {
93
- const parser = new ArgsParser("tc-1", noop);
94
- parser.write(
95
- '{"initialHeight":400,"html":"<div>hi</div>","jsFunctions":"function foo(){}","jsExpressions":["expr1","expr2"]}',
96
- );
97
-
98
- expect(parser.params).toEqual({
99
- initialHeight: 400,
100
- html: "<div>hi</div>",
101
- jsFunctions: "function foo(){}",
102
- jsExpressions: ["expr1", "expr2"],
103
- });
104
- });
105
-
106
- it("streams html as incremental chunks via textNode", () => {
107
- const emitted: BaseEvent[] = [];
108
- const parser = new ArgsParser("tc-1", (e) => emitted.push(e));
109
-
110
- parser.write('{"initialHeight":200,');
111
- emitted.length = 0; // clear snapshot
112
-
113
- // Start streaming html — first chunk
114
- parser.write('"html":"<div');
115
- const htmlDeltas1 = emitted.filter(
116
- (e) => e.type === EventType.ACTIVITY_DELTA,
117
- ) as ActivityDeltaEvent[];
118
- // Should have array creation + first chunk
119
- expect(htmlDeltas1.length).toBeGreaterThanOrEqual(1);
120
- expect(htmlDeltas1[0].patch).toEqual([
121
- { op: "add", path: "/html", value: [] },
122
- ]);
123
-
124
- emitted.length = 0;
125
- // More html content — note: clarinet needs a delimiter after the
126
- // closing quote to emit onvalue, so include the trailing }
127
- parser.write('>hello</div>"}');
128
- // Should have more chunk(s) + htmlComplete
129
- const completeDelta = emitted.find(
130
- (e) =>
131
- e.type === EventType.ACTIVITY_DELTA &&
132
- (e as ActivityDeltaEvent).patch.some(
133
- (p) => p.path === "/htmlComplete",
134
- ),
135
- );
136
- expect(completeDelta).toBeDefined();
137
-
138
- // Final params should have the complete html
139
- expect(parser.params.html).toBe("<div>hello</div>");
140
- });
141
-
142
- it("parses JSON streamed in small chunks", () => {
143
- const parser = new ArgsParser("tc-1", noop);
144
- const json = '{"initialHeight":300,"html":"<p>hello</p>"}';
145
-
146
- for (const ch of json) {
147
- parser.write(ch);
148
- }
149
-
150
- expect(parser.params.initialHeight).toBe(300);
151
- expect(parser.params.html).toBe("<p>hello</p>");
152
- });
153
-
154
- it("parses jsExpressions array streamed incrementally", () => {
155
- const parser = new ArgsParser("tc-1", noop);
156
-
157
- parser.write('{"jsExpressions":');
158
- expect(parser.params.jsExpressions).toBeUndefined();
159
-
160
- parser.write('["alert(1)",');
161
- expect(parser.params.jsExpressions).toEqual(["alert(1)"]);
162
-
163
- parser.write('"console.log(2)",');
164
- expect(parser.params.jsExpressions).toEqual([
165
- "alert(1)",
166
- "console.log(2)",
167
- ]);
168
-
169
- parser.write('"document.title"]}');
170
- expect(parser.params.jsExpressions).toEqual([
171
- "alert(1)",
172
- "console.log(2)",
173
- "document.title",
174
- ]);
175
- });
176
-
177
- it("handles partial chunks that split across keys and values", () => {
178
- const parser = new ArgsParser("tc-1", noop);
179
-
180
- parser.write('{"ini');
181
- parser.write('tialHeight":');
182
- parser.write("25");
183
- parser.write('0,"ht');
184
- parser.write('ml":"<div');
185
- parser.write('>test</div>"}');
186
-
187
- expect(parser.params.initialHeight).toBe(250);
188
- expect(parser.params.html).toBe("<div>test</div>");
189
- });
190
-
191
- it("ignores unknown keys", () => {
192
- const parser = new ArgsParser("tc-1", noop);
193
- parser.write(
194
- '{"initialHeight":100,"unknown_field":"ignored","html":"ok"}',
195
- );
196
-
197
- expect(parser.params.initialHeight).toBe(100);
198
- expect(parser.params.html).toBe("ok");
199
- expect(parser.params).not.toHaveProperty("unknown_field");
200
- });
201
- });
202
-
203
- describe("Activity event emission", () => {
204
- it("emits ACTIVITY_SNAPSHOT when initialHeight finishes", () => {
205
- const emitted: BaseEvent[] = [];
206
- const parser = new ArgsParser("tc-1", (e) => emitted.push(e));
207
-
208
- parser.write('{"initialHeight":400}');
209
-
210
- expect(emitted).toHaveLength(1);
211
- const snapshot = emitted[0] as ActivitySnapshotEvent;
212
- expect(snapshot.type).toBe(EventType.ACTIVITY_SNAPSHOT);
213
- expect(snapshot.messageId).toBe("tc-1-activity");
214
- expect(snapshot.activityType).toBe("open-generative-ui");
215
- expect(snapshot.content).toEqual({
216
- initialHeight: 400,
217
- generating: true,
218
- });
219
- });
220
-
221
- it("emits ACTIVITY_DELTA array for html and single delta for jsFunctions", () => {
222
- const emitted: BaseEvent[] = [];
223
- const parser = new ArgsParser("tc-1", (e) => emitted.push(e));
224
-
225
- parser.write('{"initialHeight":200,');
226
- emitted.length = 0; // clear snapshot
227
-
228
- parser.write('"html":"<div/>",');
229
- // Should have: array creation, chunk(s), htmlComplete
230
- const htmlDeltas = emitted.filter(
231
- (e) => e.type === EventType.ACTIVITY_DELTA,
232
- ) as ActivityDeltaEvent[];
233
- expect(htmlDeltas[0].patch).toEqual([
234
- { op: "add", path: "/html", value: [] },
235
- ]);
236
- // Last html delta should be htmlComplete
237
- const completeIdx = htmlDeltas.findIndex((d) =>
238
- d.patch.some((p) => p.path === "/htmlComplete"),
239
- );
240
- expect(completeIdx).toBeGreaterThan(0);
241
-
242
- emitted.length = 0;
243
- parser.write('"jsFunctions":"fn(){}"}');
244
- const fnDeltas = emitted.filter(
245
- (e) => e.type === EventType.ACTIVITY_DELTA,
246
- ) as ActivityDeltaEvent[];
247
- expect(fnDeltas).toHaveLength(2);
248
- expect(fnDeltas[0].patch).toEqual([
249
- { op: "add", path: "/jsFunctions", value: "fn(){}" },
250
- ]);
251
- expect(fnDeltas[1].patch).toEqual([
252
- { op: "add", path: "/jsFunctionsComplete", value: true },
253
- ]);
254
- });
255
-
256
- it("emits ACTIVITY_DELTA with add op for each jsExpressions item", () => {
257
- const emitted: BaseEvent[] = [];
258
- const parser = new ArgsParser("tc-1", (e) => emitted.push(e));
259
-
260
- parser.write('{"initialHeight":100,');
261
- emitted.length = 0; // clear snapshot
262
-
263
- parser.write('"jsExpressions":["expr1",');
264
- // First: array-creation delta, then the first item append
265
- expect(emitted).toHaveLength(2);
266
- const arrayCreate = emitted[0] as ActivityDeltaEvent;
267
- expect(arrayCreate.type).toBe(EventType.ACTIVITY_DELTA);
268
- expect(arrayCreate.patch).toEqual([
269
- { op: "add", path: "/jsExpressions", value: [] },
270
- ]);
271
- const delta1 = emitted[1] as ActivityDeltaEvent;
272
- expect(delta1.type).toBe(EventType.ACTIVITY_DELTA);
273
- expect(delta1.patch).toEqual([
274
- { op: "add", path: "/jsExpressions/-", value: "expr1" },
275
- ]);
276
-
277
- emitted.length = 0;
278
- parser.write('"expr2",');
279
- expect(emitted).toHaveLength(1);
280
- const delta2 = emitted[0] as ActivityDeltaEvent;
281
- expect(delta2.patch).toEqual([
282
- { op: "add", path: "/jsExpressions/-", value: "expr2" },
283
- ]);
284
-
285
- emitted.length = 0;
286
- parser.write('"expr3"]}');
287
- expect(emitted).toHaveLength(2);
288
- const delta3 = emitted[0] as ActivityDeltaEvent;
289
- expect(delta3.patch).toEqual([
290
- { op: "add", path: "/jsExpressions/-", value: "expr3" },
291
- ]);
292
- expect((emitted[1] as ActivityDeltaEvent).patch).toEqual([
293
- { op: "add", path: "/jsExpressionsComplete", value: true },
294
- ]);
295
- });
296
-
297
- it("emits html chunks immediately without throttling", () => {
298
- const emitted: BaseEvent[] = [];
299
- const parser = new ArgsParser("tc-1", (e) => emitted.push(e));
300
-
301
- parser.write('{"initialHeight":200,');
302
- emitted.length = 0;
303
-
304
- // Start html streaming — first write should emit
305
- parser.write('"html":"chunk1');
306
- const firstDeltas = emitted.filter(
307
- (e) => e.type === EventType.ACTIVITY_DELTA,
308
- ) as ActivityDeltaEvent[];
309
- // Should have array creation + first chunk
310
- expect(firstDeltas.length).toBeGreaterThanOrEqual(1);
311
-
312
- // Immediate second write should also emit (no throttle)
313
- emitted.length = 0;
314
- parser.write("chunk2");
315
- const secondDeltas = emitted.filter(
316
- (e) => e.type === EventType.ACTIVITY_DELTA,
317
- ) as ActivityDeltaEvent[];
318
- expect(secondDeltas).toHaveLength(1);
319
- expect(secondDeltas[0].patch[0].value).toContain("chunk2");
320
-
321
- // Completing the html string should flush remaining + htmlComplete
322
- emitted.length = 0;
323
- parser.write('",');
324
- const completeDeltas = emitted.filter(
325
- (e) => e.type === EventType.ACTIVITY_DELTA,
326
- ) as ActivityDeltaEvent[];
327
- const completeDelta = completeDeltas.find((d) =>
328
- d.patch.some((p) => p.path === "/htmlComplete"),
329
- );
330
- expect(completeDelta).toBeDefined();
331
- });
332
-
333
- it("emits snapshot only once even with multiple params", () => {
334
- const emitted: BaseEvent[] = [];
335
- const parser = new ArgsParser("tc-1", (e) => emitted.push(e));
336
-
337
- parser.write('{"initialHeight":100,"html":"a","jsFunctions":"b"}');
338
-
339
- const snapshots = emitted.filter(
340
- (e) => e.type === EventType.ACTIVITY_SNAPSHOT,
341
- );
342
- expect(snapshots).toHaveLength(1);
343
- });
344
-
345
- it("produces patches that build complete content when applied sequentially", () => {
346
- const emitted: BaseEvent[] = [];
347
- const parser = new ArgsParser("tc-1", (e) => emitted.push(e));
348
-
349
- // Simulate a full tool call with all parameter types
350
- parser.write(
351
- '{"initialHeight":400,"html":"<body>game</body>","jsFunctions":"function init(){}","jsExpressions":["init()","update()"]}',
352
- );
353
-
354
- // Reconstruct content by applying snapshot + deltas in order
355
- let content: Record<string, unknown> = {};
356
- for (const event of emitted) {
357
- if (event.type === EventType.ACTIVITY_SNAPSHOT) {
358
- content = { ...(event as ActivitySnapshotEvent).content } as Record<
359
- string,
360
- unknown
361
- >;
362
- } else if (event.type === EventType.ACTIVITY_DELTA) {
363
- const delta = event as ActivityDeltaEvent;
364
- for (const op of delta.patch) {
365
- if (op.op === "add") {
366
- if (op.path.endsWith("/-")) {
367
- // Array append: path like "/jsExpressions/-" or "/html/-"
368
- const arrayKey = op.path.slice(1, -2);
369
- (content[arrayKey] as unknown[]).push(op.value);
370
- } else {
371
- // Direct property: path like "/htmlComplete"
372
- content[op.path.slice(1)] = op.value;
373
- }
374
- }
375
- }
376
- }
377
- }
378
-
379
- // html is now an array of chunks; join to verify full content
380
- expect(Array.isArray(content.html)).toBe(true);
381
- expect((content.html as string[]).join("")).toBe("<body>game</body>");
382
- expect(content.htmlComplete).toBe(true);
383
- expect(content.jsFunctions).toBe("function init(){}");
384
- expect(content.jsExpressions).toEqual(["init()", "update()"]);
385
- });
386
-
387
- it("produces patches that build content correctly when streamed in chunks", () => {
388
- const emitted: BaseEvent[] = [];
389
- const parser = new ArgsParser("tc-1", (e) => emitted.push(e));
390
-
391
- // Stream in small chunks like a real LLM would
392
- parser.write('{"initialHeight":300,');
393
- parser.write('"html":"<div>hi</div>",');
394
- parser.write('"jsFunctions":"function go(){}",');
395
- parser.write('"jsExpressions":["go()",');
396
- parser.write('"render()","done()"]}');
397
-
398
- // Reconstruct content
399
- let content: Record<string, unknown> = {};
400
- for (const event of emitted) {
401
- if (event.type === EventType.ACTIVITY_SNAPSHOT) {
402
- content = { ...(event as ActivitySnapshotEvent).content } as Record<
403
- string,
404
- unknown
405
- >;
406
- } else if (event.type === EventType.ACTIVITY_DELTA) {
407
- const delta = event as ActivityDeltaEvent;
408
- for (const op of delta.patch) {
409
- if (op.op === "add") {
410
- if (op.path.endsWith("/-")) {
411
- const arrayKey = op.path.slice(1, -2);
412
- (content[arrayKey] as unknown[]).push(op.value);
413
- } else {
414
- content[op.path.slice(1)] = op.value;
415
- }
416
- }
417
- }
418
- }
419
- }
420
-
421
- // html is now an array of chunks
422
- expect(Array.isArray(content.html)).toBe(true);
423
- expect((content.html as string[]).join("")).toBe("<div>hi</div>");
424
- expect(content.htmlComplete).toBe(true);
425
- expect(content.jsFunctions).toBe("function go(){}");
426
- expect(content.jsExpressions).toEqual(["go()", "render()", "done()"]);
427
- });
428
-
429
- it("emits array-creation delta before first jsExpressions item", () => {
430
- const emitted: BaseEvent[] = [];
431
- const parser = new ArgsParser("tc-1", (e) => emitted.push(e));
432
-
433
- parser.write('{"initialHeight":100,');
434
- emitted.length = 0; // clear snapshot
435
-
436
- // Trailing comma needed — clarinet fires onvalue after a delimiter
437
- parser.write('"jsExpressions":["first",');
438
-
439
- // Should get array creation delta followed by item delta
440
- expect(emitted).toHaveLength(2);
441
- expect((emitted[0] as ActivityDeltaEvent).patch).toEqual([
442
- { op: "add", path: "/jsExpressions", value: [] },
443
- ]);
444
- expect((emitted[1] as ActivityDeltaEvent).patch).toEqual([
445
- { op: "add", path: "/jsExpressions/-", value: "first" },
446
- ]);
447
- });
448
-
449
- it("holds genui tool call events and flushes after first activity event", async () => {
450
- const middleware = new OpenGenerativeUIMiddleware();
451
- const toolCallId = "tc-stream";
452
- const parentMessageId = "msg-1";
453
-
454
- const agent = new MockAgent([
455
- {
456
- type: EventType.RUN_STARTED,
457
- threadId: "thread-1",
458
- runId: "run-1",
459
- } as BaseEvent,
460
- {
461
- type: EventType.TEXT_MESSAGE_START,
462
- messageId: parentMessageId,
463
- role: "assistant",
464
- } as BaseEvent,
465
- {
466
- type: EventType.TEXT_MESSAGE_END,
467
- messageId: parentMessageId,
468
- } as BaseEvent,
469
- {
470
- type: EventType.TOOL_CALL_START,
471
- toolCallId,
472
- toolCallName: "generateSandboxedUi",
473
- parentMessageId,
474
- } as BaseEvent,
475
- {
476
- type: EventType.TOOL_CALL_ARGS,
477
- toolCallId,
478
- delta: '{"initialHeight":300,',
479
- } as BaseEvent,
480
- {
481
- type: EventType.TOOL_CALL_ARGS,
482
- toolCallId,
483
- delta: '"html":"<p>hi</p>"}',
484
- } as BaseEvent,
485
- {
486
- type: EventType.TOOL_CALL_END,
487
- toolCallId,
488
- } as BaseEvent,
489
- {
490
- type: EventType.RUN_FINISHED,
491
- threadId: "thread-1",
492
- runId: "run-1",
493
- } as BaseEvent,
494
- ]);
495
-
496
- const events = await collectEvents(
497
- middleware.run(createRunInput(), agent),
498
- );
499
-
500
- // ACTIVITY_SNAPSHOT should appear before any tool call events
501
- const snapshotIdx = events.findIndex(
502
- (e) => e.type === EventType.ACTIVITY_SNAPSHOT,
503
- );
504
- const toolCallStartIdx = events.findIndex(
505
- (e) => e.type === EventType.TOOL_CALL_START,
506
- );
507
- expect(snapshotIdx).toBeGreaterThan(-1);
508
- expect(toolCallStartIdx).toBeGreaterThan(-1);
509
- expect(snapshotIdx).toBeLessThan(toolCallStartIdx);
510
-
511
- // Activity content is correct
512
- const snapshots = events.filter(
513
- (e) => e.type === EventType.ACTIVITY_SNAPSHOT,
514
- ) as ActivitySnapshotEvent[];
515
- expect(snapshots).toHaveLength(1);
516
- expect(snapshots[0].content).toEqual({
517
- initialHeight: 300,
518
- generating: true,
519
- });
520
-
521
- const deltas = events.filter(
522
- (e) => e.type === EventType.ACTIVITY_DELTA,
523
- ) as ActivityDeltaEvent[];
524
- // html deltas: array creation, chunk(s), htmlComplete, then generating: false
525
- expect(deltas.length).toBeGreaterThanOrEqual(3);
526
- expect(deltas[0].patch).toEqual([
527
- { op: "add", path: "/html", value: [] },
528
- ]);
529
- // Last delta should be generating: false
530
- expect(deltas[deltas.length - 1].patch).toEqual([
531
- { op: "add", path: "/generating", value: false },
532
- ]);
533
- // htmlComplete should be emitted
534
- const htmlCompleteDelta = deltas.find((d) =>
535
- d.patch.some((p) => p.path === "/htmlComplete" && p.value === true),
536
- );
537
- expect(htmlCompleteDelta).toBeDefined();
538
- });
539
-
540
- it("passes through tool call events for non-genui tools", async () => {
541
- const middleware = new OpenGenerativeUIMiddleware();
542
- const toolCallId = "tc-other";
543
- const parentMessageId = "msg-1";
544
-
545
- const agent = new MockAgent([
546
- {
547
- type: EventType.RUN_STARTED,
548
- threadId: "thread-1",
549
- runId: "run-1",
550
- } as BaseEvent,
551
- {
552
- type: EventType.TEXT_MESSAGE_START,
553
- messageId: parentMessageId,
554
- role: "assistant",
555
- } as BaseEvent,
556
- {
557
- type: EventType.TEXT_MESSAGE_END,
558
- messageId: parentMessageId,
559
- } as BaseEvent,
560
- {
561
- type: EventType.TOOL_CALL_START,
562
- toolCallId,
563
- toolCallName: "some_other_tool",
564
- parentMessageId,
565
- } as BaseEvent,
566
- {
567
- type: EventType.TOOL_CALL_ARGS,
568
- toolCallId,
569
- delta: "{}",
570
- } as BaseEvent,
571
- {
572
- type: EventType.TOOL_CALL_END,
573
- toolCallId,
574
- } as BaseEvent,
575
- {
576
- type: EventType.RUN_FINISHED,
577
- threadId: "thread-1",
578
- runId: "run-1",
579
- } as BaseEvent,
580
- ]);
581
-
582
- const events = await collectEvents(
583
- middleware.run(createRunInput(), agent),
584
- );
585
-
586
- const types = events.map((e) => e.type);
587
- expect(types).toEqual([
588
- EventType.RUN_STARTED,
589
- EventType.TEXT_MESSAGE_START,
590
- EventType.TEXT_MESSAGE_END,
591
- EventType.TOOL_CALL_START,
592
- EventType.TOOL_CALL_ARGS,
593
- EventType.TOOL_CALL_END,
594
- EventType.RUN_FINISHED,
595
- ]);
596
- });
597
-
598
- it("emits full activity events for jsFunctions and jsExpressions through middleware", async () => {
599
- const middleware = new OpenGenerativeUIMiddleware();
600
- const toolCallId = "tc-js";
601
- const parentMessageId = "msg-1";
602
-
603
- const agent = new MockAgent([
604
- {
605
- type: EventType.RUN_STARTED,
606
- threadId: "thread-1",
607
- runId: "run-1",
608
- } as BaseEvent,
609
- {
610
- type: EventType.TEXT_MESSAGE_START,
611
- messageId: parentMessageId,
612
- role: "assistant",
613
- } as BaseEvent,
614
- {
615
- type: EventType.TEXT_MESSAGE_END,
616
- messageId: parentMessageId,
617
- } as BaseEvent,
618
- {
619
- type: EventType.TOOL_CALL_START,
620
- toolCallId,
621
- toolCallName: "generateSandboxedUi",
622
- parentMessageId,
623
- } as BaseEvent,
624
- {
625
- type: EventType.TOOL_CALL_ARGS,
626
- toolCallId,
627
- delta: '{"initialHeight":400,',
628
- } as BaseEvent,
629
- {
630
- type: EventType.TOOL_CALL_ARGS,
631
- toolCallId,
632
- delta: '"html":"<body>game</body>",',
633
- } as BaseEvent,
634
- {
635
- type: EventType.TOOL_CALL_ARGS,
636
- toolCallId,
637
- delta: '"jsFunctions":"function init(){}",',
638
- } as BaseEvent,
639
- {
640
- type: EventType.TOOL_CALL_ARGS,
641
- toolCallId,
642
- delta: '"jsExpressions":["init()",',
643
- } as BaseEvent,
644
- {
645
- type: EventType.TOOL_CALL_ARGS,
646
- toolCallId,
647
- delta: '"render()"]}',
648
- } as BaseEvent,
649
- {
650
- type: EventType.TOOL_CALL_END,
651
- toolCallId,
652
- } as BaseEvent,
653
- {
654
- type: EventType.RUN_FINISHED,
655
- threadId: "thread-1",
656
- runId: "run-1",
657
- } as BaseEvent,
658
- ]);
659
-
660
- const events = await collectEvents(
661
- middleware.run(createRunInput(), agent),
662
- );
663
-
664
- // Verify snapshot
665
- const snapshots = events.filter(
666
- (e) => e.type === EventType.ACTIVITY_SNAPSHOT,
667
- ) as ActivitySnapshotEvent[];
668
- expect(snapshots).toHaveLength(1);
669
- expect(snapshots[0].content).toEqual({
670
- initialHeight: 400,
671
- generating: true,
672
- });
673
-
674
- // Verify deltas
675
- const deltas = events.filter(
676
- (e) => e.type === EventType.ACTIVITY_DELTA,
677
- ) as ActivityDeltaEvent[];
678
-
679
- // html is now streamed as array: creation, chunk(s), htmlComplete
680
- // Then: jsFunctions, jsExpressions array creation, items, generating: false
681
- // Verify key structural deltas exist
682
- expect(deltas[0].patch).toEqual([
683
- { op: "add", path: "/html", value: [] },
684
- ]);
685
- const htmlCompleteDelta = deltas.find((d) =>
686
- d.patch.some((p) => p.path === "/htmlComplete" && p.value === true),
687
- );
688
- expect(htmlCompleteDelta).toBeDefined();
689
- const jsFuncDelta = deltas.find((d) =>
690
- d.patch.some((p) => p.path === "/jsFunctions"),
691
- );
692
- expect(jsFuncDelta).toBeDefined();
693
- expect(deltas[deltas.length - 1].patch).toEqual([
694
- { op: "add", path: "/generating", value: false },
695
- ]);
696
-
697
- // Reconstruct content to prove patches work end-to-end
698
- let content: Record<string, unknown> = {};
699
- for (const event of events) {
700
- if (event.type === EventType.ACTIVITY_SNAPSHOT) {
701
- content = { ...(event as ActivitySnapshotEvent).content } as Record<
702
- string,
703
- unknown
704
- >;
705
- } else if (event.type === EventType.ACTIVITY_DELTA) {
706
- for (const op of (event as ActivityDeltaEvent).patch) {
707
- if (op.op === "add") {
708
- if (op.path.endsWith("/-")) {
709
- const arrayKey = op.path.slice(1, -2);
710
- (content[arrayKey] as unknown[]).push(op.value);
711
- } else {
712
- content[op.path.slice(1)] = op.value;
713
- }
714
- }
715
- }
716
- }
717
- }
718
-
719
- // html is now an array of chunks
720
- expect(Array.isArray(content.html)).toBe(true);
721
- expect((content.html as string[]).join("")).toBe("<body>game</body>");
722
- expect(content.htmlComplete).toBe(true);
723
- expect(content.generating).toBe(false);
724
- expect(content.jsFunctions).toBe("function init(){}");
725
- expect(content.jsExpressions).toEqual(["init()", "render()"]);
726
- });
727
- });
728
- });