@datalayer/agent-runtimes 1.0.2 → 1.0.3

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 (615) hide show
  1. package/README.md +1 -1
  2. package/lib/Agent.js +2 -2
  3. package/lib/AgentLexical.d.ts +1 -12
  4. package/lib/AgentLexical.js +3 -14
  5. package/lib/AgentNotebook.js +3 -3
  6. package/lib/App.js +2 -2
  7. package/lib/{components/chat/components → agents}/AgentDetails.d.ts +13 -2
  8. package/lib/{components/chat/components → agents}/AgentDetails.js +6 -12
  9. package/lib/api/agents.d.ts +47 -0
  10. package/lib/api/agents.js +106 -0
  11. package/lib/api/context.d.ts +17 -0
  12. package/lib/api/context.js +45 -0
  13. package/lib/api/evals.d.ts +27 -0
  14. package/lib/api/evals.js +63 -0
  15. package/lib/api/events.d.ts +17 -0
  16. package/lib/api/events.js +93 -0
  17. package/lib/{components/chat → api}/handler.js +1 -1
  18. package/lib/api/index.d.ts +15 -0
  19. package/lib/api/index.js +20 -0
  20. package/lib/api/notifications.d.ts +39 -0
  21. package/lib/api/notifications.js +103 -0
  22. package/lib/api/output.d.ts +28 -0
  23. package/lib/api/output.js +64 -0
  24. package/lib/api/tool-approvals.d.ts +62 -0
  25. package/lib/api/tool-approvals.js +145 -0
  26. package/lib/{components/chat/components → chat}/Chat.d.ts +24 -15
  27. package/lib/{components/chat/components → chat}/Chat.js +16 -16
  28. package/lib/{components/chat/components → chat}/ChatFloating.d.ts +17 -9
  29. package/lib/{components/chat/components → chat}/ChatFloating.js +17 -9
  30. package/lib/{components/chat/components → chat}/ChatInline.d.ts +2 -2
  31. package/lib/{components/chat/components → chat}/ChatInline.js +8 -6
  32. package/lib/{components/chat/components → chat}/ChatPopupStandalone.d.ts +3 -4
  33. package/lib/{components/chat/components → chat}/ChatPopupStandalone.js +6 -6
  34. package/lib/{components/chat/components → chat}/ChatSidebar.d.ts +14 -5
  35. package/lib/{components/chat/components → chat}/ChatSidebar.js +48 -26
  36. package/lib/{components/chat/components → chat}/ChatStandalone.d.ts +3 -4
  37. package/lib/{components/chat/components → chat}/ChatStandalone.js +5 -5
  38. package/lib/chat/base/ChatBase.d.ts +6 -0
  39. package/lib/chat/base/ChatBase.js +1287 -0
  40. package/lib/chat/display/EmptyState.d.ts +27 -0
  41. package/lib/chat/display/EmptyState.js +41 -0
  42. package/lib/{components/chat/components/elements → chat/display}/FloatingBrandButton.d.ts +1 -1
  43. package/lib/{components/chat/components/elements → chat/display}/FloatingBrandButton.js +1 -1
  44. package/lib/{components/chat/components/elements → chat/display}/PoweredByTag.d.ts +1 -1
  45. package/lib/{components/chat/components/elements → chat/header}/ChatHeader.d.ts +1 -4
  46. package/lib/{components/chat/components/elements → chat/header}/ChatHeader.js +1 -1
  47. package/lib/chat/header/ChatHeaderBase.d.ts +38 -0
  48. package/lib/chat/header/ChatHeaderBase.js +83 -0
  49. package/lib/chat/index.d.ts +66 -0
  50. package/lib/chat/index.js +74 -0
  51. package/lib/chat/indicators/McpStatusIndicator.d.ts +9 -0
  52. package/lib/chat/indicators/McpStatusIndicator.js +128 -0
  53. package/lib/chat/indicators/SandboxStatusIndicator.d.ts +10 -0
  54. package/lib/chat/indicators/SandboxStatusIndicator.js +175 -0
  55. package/lib/chat/indicators/index.d.ts +17 -0
  56. package/lib/chat/indicators/index.js +19 -0
  57. package/lib/chat/messages/ChatMessageList.d.ts +49 -0
  58. package/lib/chat/messages/ChatMessageList.js +319 -0
  59. package/lib/{components/chat/components/elements → chat/messages}/ChatMessages.d.ts +2 -2
  60. package/lib/{components/chat/components/elements → chat/messages}/ChatMessages.js +6 -4
  61. package/lib/{components/chat/components → chat}/parts/DynamicToolPart.d.ts +1 -1
  62. package/lib/{components/chat/components/elements → chat/parts}/MessagePart.d.ts +1 -1
  63. package/lib/{components/chat/components/elements → chat/parts}/MessagePart.js +4 -4
  64. package/lib/{components/chat/components → chat}/parts/ReasoningPart.js +1 -1
  65. package/lib/{components/chat/components → chat}/parts/TextPart.d.ts +1 -1
  66. package/lib/{components/chat/components → chat}/parts/ToolPart.js +1 -1
  67. package/lib/{components/chat/components → chat}/parts/index.d.ts +2 -1
  68. package/lib/{components/chat/components → chat}/parts/index.js +2 -1
  69. package/lib/chat/prompt/InputFooter.d.ts +43 -0
  70. package/lib/chat/prompt/InputFooter.js +135 -0
  71. package/lib/chat/prompt/InputPrompt.d.ts +60 -0
  72. package/lib/chat/prompt/InputPrompt.js +83 -0
  73. package/lib/chat/prompt/InputPromptFooter.d.ts +25 -0
  74. package/lib/chat/prompt/InputPromptFooter.js +15 -0
  75. package/lib/chat/prompt/InputPromptHeader.d.ts +15 -0
  76. package/lib/chat/prompt/InputPromptHeader.js +15 -0
  77. package/lib/chat/prompt/InputPromptLexical.d.ts +16 -0
  78. package/lib/chat/prompt/InputPromptLexical.js +122 -0
  79. package/lib/chat/prompt/InputPromptText.d.ts +24 -0
  80. package/lib/chat/prompt/InputPromptText.js +66 -0
  81. package/lib/chat/prompt/index.d.ts +11 -0
  82. package/lib/chat/prompt/index.js +15 -0
  83. package/lib/{components/chat/components → chat}/styles/streamdownStyles.d.ts +1 -1
  84. package/lib/chat/tools/ToolApprovalBanner.d.ts +31 -0
  85. package/lib/chat/tools/ToolApprovalBanner.js +62 -0
  86. package/lib/{components/chat/components/elements → chat/tools}/ToolApprovalDialog.js +1 -1
  87. package/lib/{components/chat/components/display → chat/tools}/ToolCallDisplay.d.ts +13 -4
  88. package/lib/{components/chat/components/display → chat/tools}/ToolCallDisplay.js +18 -8
  89. package/lib/chat/tools/index.d.ts +8 -0
  90. package/lib/{components/chat/components/display → chat/tools}/index.js +3 -1
  91. package/lib/chat/usage/TokenUsageBar.d.ts +8 -0
  92. package/lib/chat/usage/TokenUsageBar.js +213 -0
  93. package/lib/client/AgentsMixin.d.ts +169 -0
  94. package/lib/client/AgentsMixin.js +279 -0
  95. package/lib/client/index.d.ts +6 -0
  96. package/lib/client/index.js +10 -0
  97. package/lib/components/NotificationEventCard.d.ts +8 -0
  98. package/lib/components/NotificationEventCard.js +152 -0
  99. package/lib/components/OutputCard.d.ts +8 -0
  100. package/lib/components/OutputCard.js +80 -0
  101. package/lib/components/ToolApprovalCard.d.ts +33 -0
  102. package/lib/components/ToolApprovalCard.js +60 -0
  103. package/lib/components/index.d.ts +3 -7
  104. package/lib/components/index.js +3 -4
  105. package/lib/{components → config}/AgentConfiguration.d.ts +43 -21
  106. package/lib/{components → config}/AgentConfiguration.js +48 -28
  107. package/lib/config/index.d.ts +2 -0
  108. package/lib/config/index.js +1 -0
  109. package/lib/{components/chat/components → context}/ContextInspector.js +38 -8
  110. package/lib/{components/chat/components → context}/ContextPanel.js +27 -6
  111. package/lib/context/CostTracker.d.ts +37 -0
  112. package/lib/context/CostTracker.js +124 -0
  113. package/lib/context/TokenUsageChart.d.ts +10 -0
  114. package/lib/context/TokenUsageChart.js +288 -0
  115. package/lib/examples/A2UiComponentGalleryExample.d.ts +20 -0
  116. package/lib/examples/A2UiComponentGalleryExample.js +568 -0
  117. package/lib/examples/A2UiContactCardExample.d.ts +21 -0
  118. package/lib/examples/A2UiContactCardExample.js +432 -0
  119. package/lib/examples/A2UiRestaurantExample.d.ts +11 -3
  120. package/lib/examples/A2UiRestaurantExample.js +63 -98
  121. package/lib/examples/A2UiViewerExample.d.ts +21 -0
  122. package/lib/examples/A2UiViewerExample.js +563 -0
  123. package/lib/examples/AgUiAgenticExample.js +3 -3
  124. package/lib/examples/AgUiBackendToolRenderingExample.js +3 -3
  125. package/lib/examples/{AgUiHaikuGenUIExample.d.ts → AgUiHaikuGenUiExample.d.ts} +4 -4
  126. package/lib/examples/{AgUiHaikuGenUIExample.js → AgUiHaikuGenUiExample.js} +7 -7
  127. package/lib/examples/AgUiHumanInTheLoopExample.js +3 -3
  128. package/lib/examples/AgUiSharedStateExample.js +3 -3
  129. package/lib/examples/{AgUiToolsBasedGenUIExample.d.ts → AgUiToolsBasedGenUiExample.d.ts} +4 -4
  130. package/lib/examples/{AgUiToolsBasedGenUIExample.js → AgUiToolsBasedGenUiExample.js} +7 -7
  131. package/lib/examples/AgentCheckpointsExample.d.ts +19 -0
  132. package/lib/examples/AgentCheckpointsExample.js +506 -0
  133. package/lib/examples/AgentCodemodeExample.d.ts +14 -0
  134. package/lib/examples/AgentCodemodeExample.js +262 -0
  135. package/lib/examples/AgentEvalsExample.d.ts +14 -0
  136. package/lib/examples/AgentEvalsExample.js +216 -0
  137. package/lib/examples/AgentGuardrailsExample.d.ts +14 -0
  138. package/lib/examples/AgentGuardrailsExample.js +218 -0
  139. package/lib/examples/AgentMemoryExample.d.ts +14 -0
  140. package/lib/examples/AgentMemoryExample.js +234 -0
  141. package/lib/examples/AgentMonitoringExample.d.ts +13 -0
  142. package/lib/examples/AgentMonitoringExample.js +311 -0
  143. package/lib/examples/AgentNotificationsExample.d.ts +14 -0
  144. package/lib/examples/AgentNotificationsExample.js +273 -0
  145. package/lib/examples/AgentOtelExample.d.ts +25 -0
  146. package/lib/examples/AgentOtelExample.js +281 -0
  147. package/lib/examples/AgentOutputsExample.d.ts +14 -0
  148. package/lib/examples/AgentOutputsExample.js +211 -0
  149. package/lib/examples/AgentSandboxExample.d.ts +17 -0
  150. package/lib/examples/AgentSandboxExample.js +496 -0
  151. package/lib/examples/AgentSkillsExample.d.ts +3 -0
  152. package/lib/examples/AgentSkillsExample.js +290 -0
  153. package/lib/examples/AgentToolApprovalsExample.d.ts +3 -0
  154. package/lib/examples/AgentToolApprovalsExample.js +672 -0
  155. package/lib/examples/AgentTriggersExample.d.ts +14 -0
  156. package/lib/examples/AgentTriggersExample.js +523 -0
  157. package/lib/examples/{AgentRuntimeFormExample.d.ts → AgentspecExample.d.ts} +3 -4
  158. package/lib/examples/{AgentRuntimeFormExample.js → AgentspecExample.js} +92 -34
  159. package/lib/examples/{JupyterCellExample.js → CellSimpleExample.js} +1 -1
  160. package/lib/examples/{AgentRuntimeCustomExample.js → ChatCustomExample.js} +3 -3
  161. package/lib/examples/{AgentRuntimeChatExample.js → ChatExample.js} +4 -4
  162. package/lib/examples/{AgentRuntimeStandaloneExample.js → ChatStandaloneExample.js} +2 -2
  163. package/lib/examples/CopilotKitLexicalExample.d.ts +3 -14
  164. package/lib/examples/CopilotKitLexicalExample.js +4 -15
  165. package/lib/examples/CopilotKitNotebookExample.js +4 -6
  166. package/lib/examples/DatalayerNotebookExample.js +2 -2
  167. package/lib/examples/{AgentRuntimeLexical2Example.d.ts → Lexical2Example.d.ts} +2 -13
  168. package/lib/examples/{AgentRuntimeLexical2Example.js → Lexical2Example.js} +8 -17
  169. package/lib/examples/{AgentRuntimeLexicalExample.d.ts → LexicalExample.d.ts} +1 -12
  170. package/lib/examples/{AgentRuntimeLexicalExample.js → LexicalExample.js} +28 -27
  171. package/lib/examples/{AgentRuntimeLexicalSidebarExample.d.ts → LexicalSidebarExample.d.ts} +2 -13
  172. package/lib/examples/{AgentRuntimeLexicalSidebarExample.js → LexicalSidebarExample.js} +6 -18
  173. package/lib/examples/{AgentRuntimeNotebookExample.js → NotebookExample.js} +14 -10
  174. package/lib/examples/{AgentRuntimeNotebookSidebarExample.js → NotebookSidebarExample.js} +5 -8
  175. package/lib/examples/{JupyterNotebookExample.js → NotebookSimpleExample.js} +2 -2
  176. package/lib/examples/ag-ui/weather/InlineWeatherCard.js +1 -1
  177. package/lib/examples/components/ExampleWrapper.d.ts +12 -0
  178. package/lib/examples/components/ExampleWrapper.js +16 -0
  179. package/lib/examples/components/Header.d.ts +2 -2
  180. package/lib/examples/components/HeaderControls.js +1 -1
  181. package/lib/examples/components/LexicalEditor.d.ts +1 -12
  182. package/lib/examples/components/LexicalEditor.js +1 -12
  183. package/lib/examples/components/MainContent.d.ts +4 -11
  184. package/lib/examples/components/MainContent.js +6 -60
  185. package/lib/examples/components/index.d.ts +1 -0
  186. package/lib/examples/components/index.js +1 -0
  187. package/lib/examples/example-selector.d.ts +1 -1
  188. package/lib/examples/example-selector.js +35 -22
  189. package/lib/examples/index.d.ts +26 -13
  190. package/lib/examples/index.js +26 -12
  191. package/lib/examples/main.d.ts +6 -0
  192. package/lib/examples/main.js +20 -43
  193. package/lib/examples/utils/examplesStore.d.ts +4 -0
  194. package/lib/examples/{stores → utils}/examplesStore.js +1 -1
  195. package/lib/examples/utils/notebooks/Empty.ipynb.json +33 -0
  196. package/lib/examples/utils/notebooks/NotebookExample2.ipynb.json +48 -0
  197. package/lib/examples/utils/themeStore.d.ts +8 -0
  198. package/lib/examples/utils/themeStore.js +14 -0
  199. package/lib/extensions/A2UIExtension.d.ts +65 -0
  200. package/lib/extensions/A2UIExtension.js +202 -0
  201. package/lib/{components/chat/extensions → extensions}/ExtensionRegistry.d.ts +2 -3
  202. package/lib/{components/chat/extensions → extensions}/ExtensionRegistry.js +0 -2
  203. package/lib/{components/chat/extensions → extensions}/MCPUIExtension.d.ts +2 -2
  204. package/lib/{components/chat/extensions → extensions}/MCPUIExtension.js +1 -1
  205. package/lib/extensions/index.d.ts +9 -0
  206. package/lib/{components/chat/extensions → extensions}/index.js +2 -2
  207. package/lib/hooks/index.d.ts +33 -16
  208. package/lib/hooks/index.js +33 -16
  209. package/lib/hooks/useAIAgentsWebSocket.d.ts +29 -0
  210. package/lib/hooks/useAIAgentsWebSocket.js +136 -0
  211. package/lib/hooks/{useAGUI.d.ts → useAgUi.d.ts} +2 -2
  212. package/lib/hooks/{useAGUI.js → useAgUi.js} +2 -2
  213. package/lib/hooks/useAgents.d.ts +150 -11
  214. package/lib/hooks/useAgents.js +623 -61
  215. package/lib/hooks/{useAgentStore.d.ts → useAgentsCatalog.d.ts} +3 -8
  216. package/lib/hooks/{useAgentStore.js → useAgentsCatalog.js} +9 -3
  217. package/lib/hooks/useAgentsRegistry.d.ts +10 -0
  218. package/lib/hooks/useAgentsRegistry.js +20 -0
  219. package/lib/hooks/useAgentsService.d.ts +22 -0
  220. package/lib/hooks/useAgentsService.js +146 -0
  221. package/lib/hooks/useChat.d.ts +2 -2
  222. package/lib/hooks/useChat.js +14 -8
  223. package/lib/hooks/useCheckpoints.d.ts +176 -0
  224. package/lib/hooks/useCheckpoints.js +466 -0
  225. package/lib/hooks/useConfig.d.ts +11 -0
  226. package/lib/hooks/useConfig.js +46 -0
  227. package/lib/hooks/useContextSnapshot.d.ts +11 -0
  228. package/lib/hooks/useContextSnapshot.js +44 -0
  229. package/lib/hooks/useMonitoring.d.ts +24 -0
  230. package/lib/hooks/useMonitoring.js +111 -0
  231. package/lib/hooks/useNotifications.d.ts +67 -0
  232. package/lib/hooks/useNotifications.js +208 -0
  233. package/lib/hooks/useSandbox.d.ts +12 -0
  234. package/lib/hooks/useSandbox.js +49 -0
  235. package/lib/hooks/useSkills.d.ts +13 -0
  236. package/lib/hooks/useSkills.js +46 -0
  237. package/lib/hooks/useToolApprovals.d.ts +45 -0
  238. package/lib/hooks/useToolApprovals.js +126 -0
  239. package/lib/hooks/useTools.d.ts +4 -4
  240. package/lib/hooks/useTools.js +2 -2
  241. package/lib/hooks/{useVercelChat.d.ts → useVercelAI.d.ts} +3 -3
  242. package/lib/hooks/{useVercelChat.js → useVercelAI.js} +2 -2
  243. package/lib/{components/chat/components → identity}/AgentIdentity.d.ts +1 -1
  244. package/lib/{components/chat/components → identity}/AgentIdentity.js +4 -3
  245. package/lib/identity/index.d.ts +1 -0
  246. package/lib/identity/index.js +2 -0
  247. package/lib/index.d.ts +4 -3
  248. package/lib/index.js +3 -2
  249. package/lib/{components/chat/inference → inference}/BaseInferenceProvider.d.ts +3 -3
  250. package/lib/{components/chat/inference → inference}/DatalayerInferenceProvider.d.ts +3 -3
  251. package/lib/{components/chat/inference → inference}/DatalayerInferenceProvider.js +1 -1
  252. package/lib/{components/chat/inference → inference}/SelfHostedInferenceProvider.d.ts +2 -2
  253. package/lib/{components/chat/inference → inference}/SelfHostedInferenceProvider.js +1 -1
  254. package/lib/{components/chat/inference → inference}/index.d.ts +1 -1
  255. package/lib/{components/chat/inference → inference}/index.js +1 -1
  256. package/lib/lexical/ChatInlinePlugin.d.ts +1 -1
  257. package/lib/lexical/ChatInlinePlugin.js +1 -1
  258. package/lib/{components → mcp}/McpServerManager.d.ts +1 -2
  259. package/lib/mcp/index.d.ts +1 -0
  260. package/lib/{specs/agents/codeai → mcp}/index.js +1 -1
  261. package/lib/{components/chat/middleware → middleware}/MiddlewarePipeline.d.ts +3 -3
  262. package/lib/{components/chat/middleware → middleware}/index.d.ts +1 -1
  263. package/lib/{components/chat/middleware → middleware}/index.js +1 -1
  264. package/lib/{components/chat/protocols → protocols}/A2AAdapter.d.ts +6 -6
  265. package/lib/{components/chat/protocols → protocols}/A2AAdapter.js +3 -3
  266. package/lib/{components/chat/protocols → protocols}/ACPAdapter.d.ts +6 -6
  267. package/lib/{components/chat/protocols → protocols}/ACPAdapter.js +4 -4
  268. package/lib/{components/chat/protocols → protocols}/AGUIAdapter.d.ts +14 -6
  269. package/lib/{components/chat/protocols → protocols}/AGUIAdapter.js +72 -10
  270. package/lib/{components/chat/protocols → protocols}/BaseProtocolAdapter.d.ts +6 -6
  271. package/lib/{components/chat/protocols → protocols}/BaseProtocolAdapter.js +1 -1
  272. package/lib/{components/chat/protocols → protocols}/VercelAIAdapter.d.ts +31 -7
  273. package/lib/protocols/VercelAIAdapter.js +682 -0
  274. package/lib/{components/chat/protocols → protocols}/index.d.ts +1 -2
  275. package/lib/{components/chat/protocols → protocols}/index.js +1 -1
  276. package/lib/specs/agents/agents.d.ts +54 -0
  277. package/lib/specs/agents/{mocks/agents.js → agents.js} +1144 -799
  278. package/lib/specs/agents/index.js +14 -9
  279. package/lib/specs/envvars.d.ts +11 -19
  280. package/lib/specs/envvars.js +42 -21
  281. package/lib/specs/evals.d.ts +20 -0
  282. package/lib/specs/evals.js +133 -0
  283. package/lib/specs/events.d.ts +18 -0
  284. package/lib/specs/events.js +182 -0
  285. package/lib/specs/frontendTools.d.ts +14 -0
  286. package/lib/specs/frontendTools.js +53 -0
  287. package/lib/specs/guardrails.d.ts +22 -0
  288. package/lib/specs/guardrails.js +391 -0
  289. package/lib/specs/index.d.ts +15 -2
  290. package/lib/specs/index.js +15 -2
  291. package/lib/specs/mcpServers.d.ts +13 -11
  292. package/lib/specs/mcpServers.js +82 -28
  293. package/lib/specs/memory.d.ts +34 -0
  294. package/lib/specs/memory.js +99 -0
  295. package/lib/specs/models.d.ts +21 -34
  296. package/lib/specs/models.js +61 -41
  297. package/lib/specs/notifications.d.ts +17 -0
  298. package/lib/specs/notifications.js +187 -0
  299. package/lib/specs/outputs.d.ts +19 -0
  300. package/lib/specs/outputs.js +112 -0
  301. package/lib/specs/skills.d.ts +7 -16
  302. package/lib/specs/skills.js +89 -12
  303. package/lib/specs/teams/index.d.ts +17 -0
  304. package/lib/specs/teams/index.js +37 -0
  305. package/lib/specs/teams/teams.d.ts +27 -0
  306. package/lib/specs/teams/teams.js +1120 -0
  307. package/lib/specs/tools.d.ts +15 -0
  308. package/lib/specs/tools.js +83 -0
  309. package/lib/specs/triggers.d.ts +15 -0
  310. package/lib/specs/triggers.js +117 -0
  311. package/lib/stores/agentsStore.d.ts +123 -0
  312. package/lib/stores/agentsStore.js +270 -0
  313. package/lib/{components/chat/store → stores}/chatStore.d.ts +3 -2
  314. package/lib/{components/chat/store → stores}/chatStore.js +2 -2
  315. package/lib/{components/chat/store → stores}/conversationStore.d.ts +1 -1
  316. package/lib/{components/chat/store → stores}/conversationStore.js +1 -1
  317. package/lib/{components/chat/store → stores}/index.d.ts +3 -2
  318. package/lib/{components/chat/store → stores}/index.js +3 -2
  319. package/lib/{components/chat/tools → tools}/ToolExecutor.d.ts +2 -2
  320. package/lib/{components/chat/tools → tools}/ToolExecutor.js +1 -1
  321. package/lib/tools/adapters/agent-runtimes/AgentRuntimesToolAdapter.d.ts +1 -1
  322. package/lib/tools/adapters/agent-runtimes/lexicalHooks.d.ts +14 -10
  323. package/lib/tools/adapters/agent-runtimes/lexicalHooks.js +31 -21
  324. package/lib/tools/adapters/agent-runtimes/notebookHooks.d.ts +1 -1
  325. package/lib/tools/index.d.ts +3 -0
  326. package/lib/tools/index.js +3 -7
  327. package/lib/types/a2a.d.ts +39 -0
  328. package/lib/types/acp.d.ts +21 -0
  329. package/lib/types/ag-ui.d.ts +25 -0
  330. package/lib/types/agents-lifecycle.d.ts +36 -0
  331. package/lib/types/agents.d.ts +80 -0
  332. package/lib/types/agents.js +22 -0
  333. package/lib/types/agentspecs.d.ts +90 -0
  334. package/lib/{components/chat/components/base/ChatBase.d.ts → types/chat.d.ts} +59 -99
  335. package/lib/types/checkpoints.d.ts +32 -0
  336. package/lib/types/checkpoints.js +5 -0
  337. package/lib/types/config.d.ts +67 -0
  338. package/lib/{runtime/types.js → types/config.js} +2 -2
  339. package/lib/types/connection.d.ts +31 -0
  340. package/lib/types/connection.js +5 -0
  341. package/lib/types/context.d.ts +67 -0
  342. package/lib/types/context.js +5 -0
  343. package/lib/types/cost.d.ts +42 -0
  344. package/lib/types/cost.js +5 -0
  345. package/lib/types/envvars.d.ts +21 -0
  346. package/lib/types/envvars.js +5 -0
  347. package/lib/types/evals.d.ts +66 -0
  348. package/lib/types/evals.js +5 -0
  349. package/lib/types/events.d.ts +49 -0
  350. package/lib/types/events.js +5 -0
  351. package/lib/types/eventspecs.d.ts +39 -0
  352. package/lib/types/eventspecs.js +5 -0
  353. package/lib/types/examples.d.ts +31 -0
  354. package/lib/types/examples.js +5 -0
  355. package/lib/{components/chat/types → types}/execution.d.ts +10 -1
  356. package/lib/{components/chat/types/extension.d.ts → types/extensions.d.ts} +3 -3
  357. package/lib/types/guardrails.d.ts +106 -0
  358. package/lib/types/guardrails.js +5 -0
  359. package/lib/types/index.d.ts +36 -2
  360. package/lib/types/index.js +35 -2
  361. package/lib/{components/chat/types → types}/inference.d.ts +3 -3
  362. package/lib/types/inference.js +5 -0
  363. package/lib/types/mcp.d.ts +117 -0
  364. package/lib/types/mcp.js +27 -0
  365. package/lib/types/memory.d.ts +23 -0
  366. package/lib/types/memory.js +5 -0
  367. package/lib/{components/chat/types/message.d.ts → types/messages.d.ts} +20 -1
  368. package/lib/{components/chat/types → types}/middleware.d.ts +3 -3
  369. package/lib/types/models.d.ts +63 -0
  370. package/lib/types/models.js +5 -0
  371. package/lib/types/notifications.d.ts +85 -0
  372. package/lib/types/notifications.js +5 -0
  373. package/lib/types/outputs.d.ts +51 -0
  374. package/lib/types/outputs.js +5 -0
  375. package/lib/{components/chat/types → types}/protocol.d.ts +37 -99
  376. package/lib/types/protocol.js +5 -0
  377. package/lib/types/sandbox.d.ts +27 -0
  378. package/lib/types/sandbox.js +24 -0
  379. package/lib/types/skills.d.ts +74 -0
  380. package/lib/types/skills.js +5 -0
  381. package/lib/types/teams.d.ts +133 -0
  382. package/lib/types/teams.js +5 -0
  383. package/lib/types/tool-approvals.d.ts +39 -0
  384. package/lib/types/tool-approvals.js +5 -0
  385. package/lib/{components/chat/types/tool.d.ts → types/tools.d.ts} +59 -4
  386. package/lib/types/triggers.d.ts +48 -0
  387. package/lib/types/triggers.js +5 -0
  388. package/lib/types/usage.d.ts +36 -0
  389. package/lib/types/usage.js +5 -0
  390. package/lib/utils/index.d.ts +1 -0
  391. package/lib/utils/index.js +5 -0
  392. package/lib/utils/utils.d.ts +60 -0
  393. package/lib/utils/utils.js +205 -0
  394. package/package.json +13 -14
  395. package/scripts/codegen/__pycache__/generate_agents.cpython-313.pyc +0 -0
  396. package/scripts/codegen/__pycache__/generate_envvars.cpython-313.pyc +0 -0
  397. package/scripts/codegen/__pycache__/generate_evals.cpython-313.pyc +0 -0
  398. package/scripts/codegen/__pycache__/generate_guardrails.cpython-313.pyc +0 -0
  399. package/scripts/codegen/__pycache__/generate_mcp_servers.cpython-313.pyc +0 -0
  400. package/scripts/codegen/__pycache__/generate_memory.cpython-313.pyc +0 -0
  401. package/scripts/codegen/__pycache__/generate_models.cpython-313.pyc +0 -0
  402. package/scripts/codegen/__pycache__/generate_notifications.cpython-313.pyc +0 -0
  403. package/scripts/codegen/__pycache__/generate_outputs.cpython-313.pyc +0 -0
  404. package/scripts/codegen/__pycache__/generate_skills.cpython-313.pyc +0 -0
  405. package/scripts/codegen/__pycache__/generate_teams.cpython-313.pyc +0 -0
  406. package/scripts/codegen/__pycache__/generate_tools.cpython-313.pyc +0 -0
  407. package/scripts/codegen/__pycache__/generate_triggers.cpython-313.pyc +0 -0
  408. package/scripts/codegen/__pycache__/versioning.cpython-313.pyc +0 -0
  409. package/scripts/codegen/generate_agents.py +373 -60
  410. package/scripts/codegen/generate_envvars.py +36 -35
  411. package/scripts/codegen/generate_evals.py +279 -0
  412. package/scripts/codegen/generate_events.py +312 -0
  413. package/scripts/codegen/generate_frontend_tools.py +266 -0
  414. package/scripts/codegen/generate_guardrails.py +475 -0
  415. package/scripts/codegen/generate_mcp_servers.py +36 -9
  416. package/scripts/codegen/generate_memory.py +468 -0
  417. package/scripts/codegen/generate_models.py +22 -46
  418. package/scripts/codegen/generate_notifications.py +309 -0
  419. package/scripts/codegen/generate_outputs.py +267 -0
  420. package/scripts/codegen/generate_skills.py +108 -51
  421. package/scripts/codegen/generate_teams.py +922 -0
  422. package/scripts/codegen/generate_tools.py +326 -0
  423. package/scripts/codegen/generate_triggers.py +295 -0
  424. package/scripts/codegen/versioning.py +53 -0
  425. package/lib/components/chat/components/base/ChatBase.js +0 -2242
  426. package/lib/components/chat/components/base/InputPrompt.d.ts +0 -42
  427. package/lib/components/chat/components/base/InputPrompt.js +0 -131
  428. package/lib/components/chat/components/display/index.d.ts +0 -6
  429. package/lib/components/chat/components/index.d.ts +0 -26
  430. package/lib/components/chat/components/index.js +0 -39
  431. package/lib/components/chat/extensions/A2UIExtension.d.ts +0 -87
  432. package/lib/components/chat/extensions/A2UIExtension.js +0 -312
  433. package/lib/components/chat/extensions/index.d.ts +0 -10
  434. package/lib/components/chat/index.d.ts +0 -61
  435. package/lib/components/chat/index.js +0 -76
  436. package/lib/components/chat/protocols/VercelAIAdapter.js +0 -315
  437. package/lib/components/chat/tools/index.d.ts +0 -8
  438. package/lib/components/chat/tools/index.js +0 -11
  439. package/lib/components/chat/types/index.d.ts +0 -12
  440. package/lib/components/chat/types/index.js +0 -17
  441. package/lib/components/sparklines/Sparklines.d.ts +0 -16
  442. package/lib/components/sparklines/Sparklines.js +0 -65
  443. package/lib/components/sparklines/SparklinesLine.d.ts +0 -8
  444. package/lib/components/sparklines/SparklinesLine.js +0 -37
  445. package/lib/components/sparklines/dataProcessing.d.ts +0 -25
  446. package/lib/components/sparklines/dataProcessing.js +0 -35
  447. package/lib/components/sparklines/index.d.ts +0 -4
  448. package/lib/components/sparklines/index.js +0 -7
  449. package/lib/components/sparklines/types.d.ts +0 -36
  450. package/lib/examples/stores/examplesStore.d.ts +0 -5
  451. package/lib/examples/stores/notebooks/Empty.ipynb.json +0 -33
  452. package/lib/examples/stores/notebooks/NotebookExample2.ipynb.json +0 -48
  453. package/lib/examples/stores/themeStore.d.ts +0 -33
  454. package/lib/examples/stores/themeStore.js +0 -38
  455. package/lib/hooks/useAgentRuntimes.d.ts +0 -350
  456. package/lib/hooks/useAgentRuntimes.js +0 -78
  457. package/lib/hooks/useKeyboardShortcuts.d.ts +0 -47
  458. package/lib/hooks/useKeyboardShortcuts.js +0 -153
  459. package/lib/hooks/useMobile.d.ts +0 -1
  460. package/lib/hooks/useMobile.js +0 -19
  461. package/lib/hooks/useNotebookAIAgent.d.ts +0 -8
  462. package/lib/hooks/useNotebookAIAgent.js +0 -73
  463. package/lib/renderers/a2ui/components/A2UIRenderer.d.ts +0 -7
  464. package/lib/renderers/a2ui/components/A2UIRenderer.js +0 -102
  465. package/lib/renderers/a2ui/components/SurfaceRenderer.d.ts +0 -7
  466. package/lib/renderers/a2ui/components/SurfaceRenderer.js +0 -101
  467. package/lib/renderers/a2ui/components/content/AudioPlayer.d.ts +0 -9
  468. package/lib/renderers/a2ui/components/content/AudioPlayer.js +0 -38
  469. package/lib/renderers/a2ui/components/content/Divider.d.ts +0 -9
  470. package/lib/renderers/a2ui/components/content/Divider.js +0 -35
  471. package/lib/renderers/a2ui/components/content/Icon.d.ts +0 -9
  472. package/lib/renderers/a2ui/components/content/Icon.js +0 -110
  473. package/lib/renderers/a2ui/components/content/Image.d.ts +0 -9
  474. package/lib/renderers/a2ui/components/content/Image.js +0 -61
  475. package/lib/renderers/a2ui/components/content/Text.d.ts +0 -9
  476. package/lib/renderers/a2ui/components/content/Text.js +0 -64
  477. package/lib/renderers/a2ui/components/content/Video.d.ts +0 -9
  478. package/lib/renderers/a2ui/components/content/Video.js +0 -37
  479. package/lib/renderers/a2ui/components/content/index.d.ts +0 -6
  480. package/lib/renderers/a2ui/components/content/index.js +0 -25
  481. package/lib/renderers/a2ui/components/index.d.ts +0 -5
  482. package/lib/renderers/a2ui/components/index.js +0 -24
  483. package/lib/renderers/a2ui/components/interactive/Button.d.ts +0 -11
  484. package/lib/renderers/a2ui/components/interactive/Button.js +0 -71
  485. package/lib/renderers/a2ui/components/interactive/CheckBox.d.ts +0 -9
  486. package/lib/renderers/a2ui/components/interactive/CheckBox.js +0 -48
  487. package/lib/renderers/a2ui/components/interactive/DateTimeInput.d.ts +0 -9
  488. package/lib/renderers/a2ui/components/interactive/DateTimeInput.js +0 -62
  489. package/lib/renderers/a2ui/components/interactive/MultipleChoice.d.ts +0 -9
  490. package/lib/renderers/a2ui/components/interactive/MultipleChoice.js +0 -73
  491. package/lib/renderers/a2ui/components/interactive/Slider.d.ts +0 -9
  492. package/lib/renderers/a2ui/components/interactive/Slider.js +0 -53
  493. package/lib/renderers/a2ui/components/interactive/TextField.d.ts +0 -9
  494. package/lib/renderers/a2ui/components/interactive/TextField.js +0 -72
  495. package/lib/renderers/a2ui/components/interactive/index.d.ts +0 -6
  496. package/lib/renderers/a2ui/components/interactive/index.js +0 -25
  497. package/lib/renderers/a2ui/components/layout/Card.d.ts +0 -11
  498. package/lib/renderers/a2ui/components/layout/Card.js +0 -30
  499. package/lib/renderers/a2ui/components/layout/Column.d.ts +0 -11
  500. package/lib/renderers/a2ui/components/layout/Column.js +0 -65
  501. package/lib/renderers/a2ui/components/layout/List.d.ts +0 -11
  502. package/lib/renderers/a2ui/components/layout/List.js +0 -55
  503. package/lib/renderers/a2ui/components/layout/Modal.d.ts +0 -11
  504. package/lib/renderers/a2ui/components/layout/Modal.js +0 -58
  505. package/lib/renderers/a2ui/components/layout/Row.d.ts +0 -11
  506. package/lib/renderers/a2ui/components/layout/Row.js +0 -65
  507. package/lib/renderers/a2ui/components/layout/Tabs.d.ts +0 -11
  508. package/lib/renderers/a2ui/components/layout/Tabs.js +0 -48
  509. package/lib/renderers/a2ui/components/layout/index.d.ts +0 -6
  510. package/lib/renderers/a2ui/components/layout/index.js +0 -25
  511. package/lib/renderers/a2ui/context/A2UIContext.d.ts +0 -17
  512. package/lib/renderers/a2ui/context/A2UIContext.js +0 -54
  513. package/lib/renderers/a2ui/context/ThemeContext.d.ts +0 -20
  514. package/lib/renderers/a2ui/context/ThemeContext.js +0 -333
  515. package/lib/renderers/a2ui/hooks/useA2UI.d.ts +0 -36
  516. package/lib/renderers/a2ui/hooks/useA2UI.js +0 -62
  517. package/lib/renderers/a2ui/hooks/useDataBinding.d.ts +0 -8
  518. package/lib/renderers/a2ui/hooks/useDataBinding.js +0 -83
  519. package/lib/renderers/a2ui/index.d.ts +0 -9
  520. package/lib/renderers/a2ui/index.js +0 -28
  521. package/lib/renderers/a2ui/lib/utils.d.ts +0 -11
  522. package/lib/renderers/a2ui/lib/utils.js +0 -38
  523. package/lib/renderers/a2ui/types/index.d.ts +0 -17
  524. package/lib/runtime/index.d.ts +0 -38
  525. package/lib/runtime/index.js +0 -40
  526. package/lib/runtime/runtimeStore.d.ts +0 -76
  527. package/lib/runtime/runtimeStore.js +0 -184
  528. package/lib/runtime/types.d.ts +0 -84
  529. package/lib/runtime/useAgentConnection.d.ts +0 -45
  530. package/lib/runtime/useAgentConnection.js +0 -112
  531. package/lib/runtime/useAgentRuntime.d.ts +0 -93
  532. package/lib/runtime/useAgentRuntime.js +0 -125
  533. package/lib/specs/agents/codeai/agents.d.ts +0 -28
  534. package/lib/specs/agents/codeai/agents.js +0 -179
  535. package/lib/specs/agents/codeai/index.d.ts +0 -1
  536. package/lib/specs/agents/codemode-paper/agents.d.ts +0 -31
  537. package/lib/specs/agents/codemode-paper/agents.js +0 -378
  538. package/lib/specs/agents/codemode-paper/index.d.ts +0 -1
  539. package/lib/specs/agents/codemode-paper/index.js +0 -5
  540. package/lib/specs/agents/datalayer-ai/agents.d.ts +0 -31
  541. package/lib/specs/agents/datalayer-ai/agents.js +0 -352
  542. package/lib/specs/agents/datalayer-ai/index.d.ts +0 -1
  543. package/lib/specs/agents/datalayer-ai/index.js +0 -5
  544. package/lib/specs/agents/mocks/agents.d.ts +0 -43
  545. package/lib/specs/agents/mocks/index.d.ts +0 -1
  546. package/lib/specs/agents/mocks/index.js +0 -5
  547. package/lib/state/index.d.ts +0 -1
  548. package/lib/state/index.js +0 -5
  549. package/lib/state/substates/AIAgentState.d.ts +0 -80
  550. package/lib/state/substates/AIAgentState.js +0 -108
  551. package/lib/state/substates/index.d.ts +0 -1
  552. package/lib/state/substates/index.js +0 -5
  553. package/lib/types/AIAgent.d.ts +0 -17
  554. package/lib/types/Types.d.ts +0 -217
  555. /package/lib/{components/chat → api}/handler.d.ts +0 -0
  556. /package/lib/{renderers/index.d.ts → api/utils.d.ts} +0 -0
  557. /package/lib/{renderers/index.js → api/utils.js} +0 -0
  558. /package/lib/{components/chat/components/elements → chat/display}/PoweredByTag.js +0 -0
  559. /package/lib/{components/chat/components → chat}/parts/DynamicToolPart.js +0 -0
  560. /package/lib/{components/chat/components → chat}/parts/ReasoningPart.d.ts +0 -0
  561. /package/lib/{components/chat/components → chat}/parts/TextPart.js +0 -0
  562. /package/lib/{components/chat/components → chat}/parts/ToolPart.d.ts +0 -0
  563. /package/lib/{components/chat/components → chat}/styles/streamdownStyles.js +0 -0
  564. /package/lib/{components/chat/components/elements → chat/tools}/ToolApprovalDialog.d.ts +0 -0
  565. /package/lib/{components/chat/components → context}/ContextDistribution.d.ts +0 -0
  566. /package/lib/{components/chat/components → context}/ContextDistribution.js +0 -0
  567. /package/lib/{components/chat/components → context}/ContextInspector.d.ts +0 -0
  568. /package/lib/{components/chat/components → context}/ContextPanel.d.ts +0 -0
  569. /package/lib/{components/chat/components → context}/ContextUsage.d.ts +0 -0
  570. /package/lib/{components/chat/components → context}/ContextUsage.js +0 -0
  571. /package/lib/examples/{JupyterCellExample.d.ts → CellSimpleExample.d.ts} +0 -0
  572. /package/lib/examples/{AgentRuntimeCustomExample.d.ts → ChatCustomExample.d.ts} +0 -0
  573. /package/lib/examples/{AgentRuntimeChatExample.d.ts → ChatExample.d.ts} +0 -0
  574. /package/lib/examples/{AgentRuntimeStandaloneExample.d.ts → ChatStandaloneExample.d.ts} +0 -0
  575. /package/lib/examples/{AgentRuntimeNotebookExample.d.ts → NotebookExample.d.ts} +0 -0
  576. /package/lib/examples/{AgentRuntimeNotebookSidebarExample.d.ts → NotebookSidebarExample.d.ts} +0 -0
  577. /package/lib/examples/{JupyterNotebookExample.d.ts → NotebookSimpleExample.d.ts} +0 -0
  578. /package/lib/examples/{stores → utils}/agents/earthquake-detector.ipynb.json +0 -0
  579. /package/lib/examples/{stores → utils}/agents/earthquake-detector.json +0 -0
  580. /package/lib/examples/{stores → utils}/agents/earthquake-detector.lexical.json +0 -0
  581. /package/lib/examples/{stores → utils}/agents/sales-forecaster.ipynb.json +0 -0
  582. /package/lib/examples/{stores → utils}/agents/sales-forecaster.json +0 -0
  583. /package/lib/examples/{stores → utils}/agents/sales-forecaster.lexical.json +0 -0
  584. /package/lib/examples/{stores → utils}/agents/social-post-generator.ipynb.json +0 -0
  585. /package/lib/examples/{stores → utils}/agents/social-post-generator.json +0 -0
  586. /package/lib/examples/{stores → utils}/agents/social-post-generator.lexical.json +0 -0
  587. /package/lib/examples/{stores → utils}/agents/stock-market.ipynb.json +0 -0
  588. /package/lib/examples/{stores → utils}/agents/stock-market.json +0 -0
  589. /package/lib/examples/{stores → utils}/agents/stock-market.lexical.json +0 -0
  590. /package/lib/examples/{stores → utils}/notebooks/IPyWidgetsExample.ipynb.json +0 -0
  591. /package/lib/examples/{stores → utils}/notebooks/IPyWidgetsExampleWithState.ipynb.json +0 -0
  592. /package/lib/examples/{stores → utils}/notebooks/Lite.ipynb.json +0 -0
  593. /package/lib/examples/{stores → utils}/notebooks/Matplotlib.ipynb.json +0 -0
  594. /package/lib/examples/{stores → utils}/notebooks/NotebookExample1.ipynb.json +0 -0
  595. /package/lib/examples/{stores → utils}/notebooks/NotebookOutputs.ipynb.json +0 -0
  596. /package/lib/examples/{stores → utils}/notebooks/NotebookToCExample.ipynb.json +0 -0
  597. /package/lib/examples/{stores → utils}/notebooks/OutputIPyWidgetsExample.d.ts +0 -0
  598. /package/lib/examples/{stores → utils}/notebooks/OutputIPyWidgetsExample.js +0 -0
  599. /package/lib/examples/{stores → utils}/notebooks/PyGWalker.ipynb.json +0 -0
  600. /package/lib/examples/{stores → utils}/themedProvider.d.ts +0 -0
  601. /package/lib/examples/{stores → utils}/themedProvider.js +0 -0
  602. /package/lib/{components/chat/inference → inference}/BaseInferenceProvider.js +0 -0
  603. /package/lib/{components → mcp}/McpServerManager.js +0 -0
  604. /package/lib/{components/chat/middleware → middleware}/MiddlewarePipeline.js +0 -0
  605. /package/lib/{components/chat/types/inference.js → types/a2a.js} +0 -0
  606. /package/lib/{components/chat/types/protocol.js → types/acp.js} +0 -0
  607. /package/lib/{components/sparklines/types.js → types/ag-ui.js} +0 -0
  608. /package/lib/{renderers/a2ui/types/index.js → types/agents-lifecycle.js} +0 -0
  609. /package/lib/types/{AIAgent.js → agentspecs.js} +0 -0
  610. /package/lib/types/{Types.js → chat.js} +0 -0
  611. /package/lib/{components/chat/types → types}/execution.js +0 -0
  612. /package/lib/{components/chat/types/extension.js → types/extensions.js} +0 -0
  613. /package/lib/{components/chat/types/message.js → types/messages.js} +0 -0
  614. /package/lib/{components/chat/types → types}/middleware.js +0 -0
  615. /package/lib/{components/chat/types/tool.js → types/tools.js} +0 -0
@@ -0,0 +1,111 @@
1
+ /*
2
+ * Copyright (c) 2025-2026 Datalayer, Inc.
3
+ * Distributed under the terms of the Modified BSD License.
4
+ */
5
+ /**
6
+ * Agent monitoring hooks.
7
+ *
8
+ * @module hooks/useMonitoring
9
+ */
10
+ import { useEffect, useState } from 'react';
11
+ import { createOtelClient } from '@datalayer/core/lib/otel';
12
+ export function toMetricValue(row) {
13
+ const candidates = [row.value_double, row.value_int, row.value];
14
+ for (const candidate of candidates) {
15
+ if (typeof candidate === 'number' && Number.isFinite(candidate)) {
16
+ return candidate;
17
+ }
18
+ if (typeof candidate === 'string') {
19
+ const parsed = Number(candidate);
20
+ if (Number.isFinite(parsed)) {
21
+ return parsed;
22
+ }
23
+ }
24
+ }
25
+ return 0;
26
+ }
27
+ export async function fetchOtelMetricRows({ metric, serviceName, runUrl, apiKey, limit = 500, }) {
28
+ const client = createOtelClient({
29
+ baseUrl: runUrl,
30
+ token: apiKey,
31
+ });
32
+ const filtered = await client.fetchMetrics({
33
+ metricName: metric,
34
+ serviceName,
35
+ limit,
36
+ });
37
+ if (filtered.data.length > 0 || !serviceName) {
38
+ return filtered.data;
39
+ }
40
+ const fallback = await client.fetchMetrics({
41
+ metricName: metric,
42
+ limit,
43
+ });
44
+ return fallback.data;
45
+ }
46
+ export async function fetchOtelMetricTotal(options) {
47
+ const rows = await fetchOtelMetricRows(options);
48
+ return rows.reduce((sum, row) => sum + toMetricValue(row), 0);
49
+ }
50
+ export async function fetchOtelTotalTokens({ serviceName, runUrl, apiKey, limit = 500, }) {
51
+ const total = await fetchOtelMetricTotal({
52
+ metric: 'agent_runtimes.prompt.turn.total_tokens',
53
+ serviceName,
54
+ runUrl,
55
+ apiKey,
56
+ limit,
57
+ });
58
+ if (total > 0) {
59
+ return total;
60
+ }
61
+ const prompt = await fetchOtelMetricTotal({
62
+ metric: 'agent_runtimes.prompt.turn.prompt_tokens',
63
+ serviceName,
64
+ runUrl,
65
+ apiKey,
66
+ limit,
67
+ });
68
+ const completion = await fetchOtelMetricTotal({
69
+ metric: 'agent_runtimes.prompt.turn.completion_tokens',
70
+ serviceName,
71
+ runUrl,
72
+ apiKey,
73
+ limit,
74
+ });
75
+ return prompt + completion;
76
+ }
77
+ export function useOtelTotalTokens({ serviceName, runUrl, apiKey, limit = 500, }) {
78
+ const [tokensLabel, setTokensLabel] = useState('-');
79
+ useEffect(() => {
80
+ let cancelled = false;
81
+ const load = async () => {
82
+ try {
83
+ const total = await fetchOtelTotalTokens({
84
+ serviceName,
85
+ runUrl,
86
+ apiKey,
87
+ limit,
88
+ });
89
+ if (cancelled) {
90
+ return;
91
+ }
92
+ if (total > 0) {
93
+ setTokensLabel(Math.round(total).toLocaleString());
94
+ }
95
+ else {
96
+ setTokensLabel('-');
97
+ }
98
+ }
99
+ catch {
100
+ if (!cancelled) {
101
+ setTokensLabel('-');
102
+ }
103
+ }
104
+ };
105
+ load();
106
+ return () => {
107
+ cancelled = true;
108
+ };
109
+ }, [apiKey, limit, runUrl, serviceName]);
110
+ return tokensLabel;
111
+ }
@@ -0,0 +1,67 @@
1
+ import type { NotificationFilters, ListAgentEventsParams, CreateAgentEventRequest, UpdateAgentEventRequest } from '../types';
2
+ export declare function useFilteredNotifications(filters?: NotificationFilters): import("@tanstack/react-query").UseQueryResult<import("../types").AgentNotification[], Error>;
3
+ export declare function useUnreadNotificationCount(): import("@tanstack/react-query").UseQueryResult<{
4
+ count: number;
5
+ }, Error>;
6
+ export declare function useMarkNotificationRead(): import("@tanstack/react-query").UseMutationResult<void, Error, string, unknown>;
7
+ export declare function useMarkAllNotificationsRead(): import("@tanstack/react-query").UseMutationResult<void, Error, void, unknown>;
8
+ /**
9
+ * List events across all agents for the authenticated user.
10
+ */
11
+ export declare function useAllAgentEvents(params?: Omit<ListAgentEventsParams, 'agent_id'>, enabled?: boolean): import("@tanstack/react-query").UseQueryResult<import("../types").ListAgentEventsResponse, Error>;
12
+ export declare function useAgentEvents(agentId: string, params?: Omit<ListAgentEventsParams, 'agent_id'>, enabled?: boolean): import("@tanstack/react-query").UseQueryResult<import("../types").ListAgentEventsResponse, Error>;
13
+ export declare function useAgentEvent(agentId: string, eventId?: string): import("@tanstack/react-query").UseQueryResult<import("../types").GetAgentEventResponse, Error>;
14
+ export declare function useCreateAgentEvent(): import("@tanstack/react-query").UseMutationResult<{
15
+ success: boolean;
16
+ event: import("../types").AgentEvent;
17
+ }, Error, CreateAgentEventRequest, unknown>;
18
+ export declare function useUpdateAgentEvent(agentId: string): import("@tanstack/react-query").UseMutationResult<import("../types").GetAgentEventResponse, Error, {
19
+ eventId: string;
20
+ payload: UpdateAgentEventRequest;
21
+ }, unknown>;
22
+ export declare function useDeleteAgentEvent(agentId: string): import("@tanstack/react-query").UseMutationResult<{
23
+ success: boolean;
24
+ }, Error, string | {
25
+ eventId: string;
26
+ eventAgentId?: string;
27
+ }, unknown>;
28
+ export declare function useMarkEventRead(agentId: string): import("@tanstack/react-query").UseMutationResult<import("../types").GetAgentEventResponse, Error, {
29
+ eventId: string;
30
+ eventAgentId?: string;
31
+ }, unknown>;
32
+ export declare function useMarkEventUnread(agentId: string): import("@tanstack/react-query").UseMutationResult<import("../types").GetAgentEventResponse, Error, {
33
+ eventId: string;
34
+ eventAgentId?: string;
35
+ }, unknown>;
36
+ export declare function useNotifications(agentId: string, filters?: NotificationFilters, eventParams?: Omit<ListAgentEventsParams, 'agent_id'>, eventId?: string): {
37
+ notificationsQuery: import("@tanstack/react-query").UseQueryResult<import("../types").AgentNotification[], Error>;
38
+ unreadCountQuery: import("@tanstack/react-query").UseQueryResult<{
39
+ count: number;
40
+ }, Error>;
41
+ markRead: import("@tanstack/react-query").UseMutationResult<void, Error, string, unknown>;
42
+ markAllRead: import("@tanstack/react-query").UseMutationResult<void, Error, void, unknown>;
43
+ eventsQuery: import("@tanstack/react-query").UseQueryResult<import("../types").ListAgentEventsResponse, Error>;
44
+ eventQuery: import("@tanstack/react-query").UseQueryResult<import("../types").GetAgentEventResponse, Error>;
45
+ createEvent: import("@tanstack/react-query").UseMutationResult<{
46
+ success: boolean;
47
+ event: import("../types").AgentEvent;
48
+ }, Error, CreateAgentEventRequest, unknown>;
49
+ updateEvent: import("@tanstack/react-query").UseMutationResult<import("../types").GetAgentEventResponse, Error, {
50
+ eventId: string;
51
+ payload: UpdateAgentEventRequest;
52
+ }, unknown>;
53
+ deleteEvent: import("@tanstack/react-query").UseMutationResult<{
54
+ success: boolean;
55
+ }, Error, string | {
56
+ eventId: string;
57
+ eventAgentId?: string;
58
+ }, unknown>;
59
+ markEventRead: import("@tanstack/react-query").UseMutationResult<import("../types").GetAgentEventResponse, Error, {
60
+ eventId: string;
61
+ eventAgentId?: string;
62
+ }, unknown>;
63
+ markEventUnread: import("@tanstack/react-query").UseMutationResult<import("../types").GetAgentEventResponse, Error, {
64
+ eventId: string;
65
+ eventAgentId?: string;
66
+ }, unknown>;
67
+ };
@@ -0,0 +1,208 @@
1
+ /*
2
+ * Copyright (c) 2025-2026 Datalayer, Inc.
3
+ * Distributed under the terms of the Modified BSD License.
4
+ */
5
+ /**
6
+ * Agent notification hooks.
7
+ *
8
+ * @module hooks/useNotifications
9
+ */
10
+ import { useMemo } from 'react';
11
+ import { useQuery, useMutation, useQueryClient } from '@tanstack/react-query';
12
+ import { useCoreStore, useIAMStore } from '@datalayer/core/lib/state';
13
+ import { DEFAULT_SERVICE_URLS } from '@datalayer/core/lib/api/constants';
14
+ import { events, notifications } from '../api';
15
+ // ─── Auth helpers ────────────────────────────────────────────────────
16
+ function useAuthToken() {
17
+ const token = useIAMStore((s) => s.token);
18
+ return token ?? '';
19
+ }
20
+ function useBaseUrl() {
21
+ const config = useCoreStore((s) => s.configuration);
22
+ return config?.aiagentsRunUrl ?? DEFAULT_SERVICE_URLS.AI_AGENTS;
23
+ }
24
+ // ─── Base hooks ──────────────────────────────────────────────────────
25
+ export function useFilteredNotifications(filters) {
26
+ const token = useAuthToken();
27
+ const baseUrl = useBaseUrl();
28
+ return useQuery({
29
+ queryKey: ['agent-notifications', filters],
30
+ queryFn: () => notifications.getNotifications(token, filters, baseUrl),
31
+ enabled: !!token,
32
+ staleTime: 10_000,
33
+ });
34
+ }
35
+ export function useUnreadNotificationCount() {
36
+ const token = useAuthToken();
37
+ const baseUrl = useBaseUrl();
38
+ return useQuery({
39
+ queryKey: ['agent-notifications', 'unread-count'],
40
+ queryFn: () => notifications.getUnreadCount(token, baseUrl),
41
+ enabled: !!token,
42
+ staleTime: 5_000,
43
+ });
44
+ }
45
+ export function useMarkNotificationRead() {
46
+ const token = useAuthToken();
47
+ const baseUrl = useBaseUrl();
48
+ const queryClient = useQueryClient();
49
+ return useMutation({
50
+ mutationFn: (id) => notifications.markNotificationRead(token, id, baseUrl),
51
+ onSuccess: () => {
52
+ queryClient.invalidateQueries({ queryKey: ['agent-notifications'] });
53
+ },
54
+ });
55
+ }
56
+ export function useMarkAllNotificationsRead() {
57
+ const token = useAuthToken();
58
+ const baseUrl = useBaseUrl();
59
+ const queryClient = useQueryClient();
60
+ return useMutation({
61
+ mutationFn: () => notifications.markAllRead(token, baseUrl),
62
+ onSuccess: () => {
63
+ queryClient.invalidateQueries({ queryKey: ['agent-notifications'] });
64
+ },
65
+ });
66
+ }
67
+ // ─── Event hooks ─────────────────────────────────────────────────────
68
+ /**
69
+ * List events across all agents for the authenticated user.
70
+ */
71
+ export function useAllAgentEvents(params, enabled = true) {
72
+ const token = useAuthToken();
73
+ const baseUrl = useBaseUrl();
74
+ return useQuery({
75
+ queryKey: ['agent-events', params],
76
+ queryFn: () => events.listAllEvents(token, params ?? {}, baseUrl),
77
+ enabled: !!token && enabled,
78
+ staleTime: 10_000,
79
+ });
80
+ }
81
+ export function useAgentEvents(agentId, params, enabled = true) {
82
+ const token = useAuthToken();
83
+ const baseUrl = useBaseUrl();
84
+ return useQuery({
85
+ queryKey: ['agent-events', agentId, params],
86
+ queryFn: () => events.listEvents(token, agentId, params ?? {}, baseUrl),
87
+ enabled: !!token && !!agentId && enabled,
88
+ staleTime: 10_000,
89
+ });
90
+ }
91
+ export function useAgentEvent(agentId, eventId) {
92
+ const token = useAuthToken();
93
+ const baseUrl = useBaseUrl();
94
+ return useQuery({
95
+ queryKey: ['agent-events', agentId, eventId],
96
+ queryFn: () => events.getEvent(token, agentId, eventId, baseUrl),
97
+ enabled: !!token && !!agentId && !!eventId,
98
+ staleTime: 10_000,
99
+ });
100
+ }
101
+ export function useCreateAgentEvent() {
102
+ const token = useAuthToken();
103
+ const baseUrl = useBaseUrl();
104
+ const queryClient = useQueryClient();
105
+ return useMutation({
106
+ mutationFn: (payload) => events.createEvent(token, payload, baseUrl),
107
+ onSuccess: () => {
108
+ queryClient.invalidateQueries({ queryKey: ['agent-events'] });
109
+ },
110
+ });
111
+ }
112
+ export function useUpdateAgentEvent(agentId) {
113
+ const token = useAuthToken();
114
+ const baseUrl = useBaseUrl();
115
+ const queryClient = useQueryClient();
116
+ return useMutation({
117
+ mutationFn: ({ eventId, payload, }) => events.updateEvent(token, agentId, eventId, payload, baseUrl),
118
+ onSuccess: (_, variables) => {
119
+ queryClient.invalidateQueries({ queryKey: ['agent-events'] });
120
+ queryClient.invalidateQueries({
121
+ queryKey: ['agent-events', agentId, variables.eventId],
122
+ });
123
+ },
124
+ });
125
+ }
126
+ export function useDeleteAgentEvent(agentId) {
127
+ const token = useAuthToken();
128
+ const baseUrl = useBaseUrl();
129
+ const queryClient = useQueryClient();
130
+ return useMutation({
131
+ mutationFn: (payload) => {
132
+ const eventId = typeof payload === 'string' ? payload : payload.eventId;
133
+ const eventAgentId = typeof payload === 'string' ? undefined : payload.eventAgentId;
134
+ const resolvedAgentId = eventAgentId || agentId;
135
+ return events.deleteEvent(token, resolvedAgentId, eventId, baseUrl);
136
+ },
137
+ onSuccess: () => {
138
+ queryClient.invalidateQueries({ queryKey: ['agent-events'] });
139
+ },
140
+ });
141
+ }
142
+ export function useMarkEventRead(agentId) {
143
+ const token = useAuthToken();
144
+ const baseUrl = useBaseUrl();
145
+ const queryClient = useQueryClient();
146
+ return useMutation({
147
+ mutationFn: ({ eventId, eventAgentId, }) => {
148
+ const resolvedAgentId = eventAgentId || agentId;
149
+ return events.markEventRead(token, resolvedAgentId, eventId, baseUrl);
150
+ },
151
+ onSuccess: () => {
152
+ queryClient.invalidateQueries({ queryKey: ['agent-events'] });
153
+ },
154
+ });
155
+ }
156
+ export function useMarkEventUnread(agentId) {
157
+ const token = useAuthToken();
158
+ const baseUrl = useBaseUrl();
159
+ const queryClient = useQueryClient();
160
+ return useMutation({
161
+ mutationFn: ({ eventId, eventAgentId, }) => {
162
+ const resolvedAgentId = eventAgentId || agentId;
163
+ return events.markEventUnread(token, resolvedAgentId, eventId, baseUrl);
164
+ },
165
+ onSuccess: () => {
166
+ queryClient.invalidateQueries({ queryKey: ['agent-events'] });
167
+ },
168
+ });
169
+ }
170
+ // ─── Composite hook ──────────────────────────────────────────────────
171
+ export function useNotifications(agentId, filters, eventParams, eventId) {
172
+ const notificationsQuery = useFilteredNotifications(filters);
173
+ const unreadCountQuery = useUnreadNotificationCount();
174
+ const markRead = useMarkNotificationRead();
175
+ const markAllRead = useMarkAllNotificationsRead();
176
+ const eventsQuery = useAgentEvents(agentId, eventParams);
177
+ const eventQuery = useAgentEvent(agentId, eventId);
178
+ const createEvent = useCreateAgentEvent();
179
+ const updateEvent = useUpdateAgentEvent(agentId);
180
+ const deleteEvent = useDeleteAgentEvent(agentId);
181
+ const markEventRead = useMarkEventRead(agentId);
182
+ const markEventUnread = useMarkEventUnread(agentId);
183
+ return useMemo(() => ({
184
+ notificationsQuery,
185
+ unreadCountQuery,
186
+ markRead,
187
+ markAllRead,
188
+ eventsQuery,
189
+ eventQuery,
190
+ createEvent,
191
+ updateEvent,
192
+ deleteEvent,
193
+ markEventRead,
194
+ markEventUnread,
195
+ }), [
196
+ notificationsQuery,
197
+ unreadCountQuery,
198
+ markRead,
199
+ markAllRead,
200
+ eventsQuery,
201
+ eventQuery,
202
+ createEvent,
203
+ updateEvent,
204
+ deleteEvent,
205
+ markEventRead,
206
+ markEventUnread,
207
+ ]);
208
+ }
@@ -0,0 +1,12 @@
1
+ import type { SandboxStatusData } from '../types/context';
2
+ /**
3
+ * Hook to poll sandbox execution status from the backend.
4
+ * Returns whether a sandbox is available and if code is currently executing.
5
+ */
6
+ export declare function useSandbox(enabled: boolean, configEndpoint?: string, authToken?: string): import("@tanstack/query-core").QueryObserverRefetchErrorResult<SandboxStatusData, Error> | import("@tanstack/query-core").QueryObserverSuccessResult<SandboxStatusData, Error> | import("@tanstack/query-core").QueryObserverLoadingErrorResult<SandboxStatusData, Error> | import("@tanstack/query-core").QueryObserverPendingResult<SandboxStatusData, Error> | import("@tanstack/query-core").QueryObserverPlaceholderResult<SandboxStatusData, Error> | {
7
+ data: undefined;
8
+ isLoading: boolean;
9
+ isError: boolean;
10
+ error: null;
11
+ refetch: () => Promise<any>;
12
+ };
@@ -0,0 +1,49 @@
1
+ /*
2
+ * Copyright (c) 2025-2026 Datalayer, Inc.
3
+ * Distributed under the terms of the Modified BSD License.
4
+ */
5
+ import { useContext } from 'react';
6
+ import { useQuery, QueryClientContext } from '@tanstack/react-query';
7
+ import { getApiBaseFromConfig } from '../utils';
8
+ /**
9
+ * Hook to poll sandbox execution status from the backend.
10
+ * Returns whether a sandbox is available and if code is currently executing.
11
+ */
12
+ export function useSandbox(enabled, configEndpoint, authToken) {
13
+ const queryClient = useContext(QueryClientContext);
14
+ if (!queryClient) {
15
+ return {
16
+ data: undefined,
17
+ isLoading: false,
18
+ isError: false,
19
+ error: null,
20
+ refetch: () => Promise.resolve({}),
21
+ };
22
+ }
23
+ const statusUrl = configEndpoint
24
+ ? `${getApiBaseFromConfig(configEndpoint)}/configure/sandbox-status`
25
+ : undefined;
26
+ // eslint-disable-next-line react-hooks/rules-of-hooks
27
+ return useQuery({
28
+ queryKey: ['sandbox-status', statusUrl],
29
+ queryFn: async () => {
30
+ if (!statusUrl) {
31
+ throw new Error('No sandbox status URL available');
32
+ }
33
+ const headers = { 'Content-Type': 'application/json' };
34
+ if (authToken) {
35
+ headers['Authorization'] = `Bearer ${authToken}`;
36
+ }
37
+ const response = await fetch(statusUrl, { headers });
38
+ if (!response.ok) {
39
+ throw new Error(`Sandbox status fetch failed: ${response.statusText}`);
40
+ }
41
+ return response.json();
42
+ },
43
+ enabled: enabled && !!statusUrl,
44
+ refetchInterval: query => (query.state.status === 'error' ? false : 2_000),
45
+ refetchOnMount: 'always',
46
+ staleTime: 0,
47
+ retry: 1,
48
+ });
49
+ }
@@ -0,0 +1,13 @@
1
+ import type { SkillsResponse } from '../types/skills';
2
+ /**
3
+ * Hook to fetch available skills from backend.
4
+ */
5
+ export declare function useSkills(enabled: boolean, baseEndpoint?: string, authToken?: string): import("@tanstack/query-core").QueryObserverRefetchErrorResult<SkillsResponse, Error> | import("@tanstack/query-core").QueryObserverSuccessResult<SkillsResponse, Error> | import("@tanstack/query-core").QueryObserverLoadingErrorResult<SkillsResponse, Error> | import("@tanstack/query-core").QueryObserverPendingResult<SkillsResponse, Error> | import("@tanstack/query-core").QueryObserverPlaceholderResult<SkillsResponse, Error> | {
6
+ data: undefined;
7
+ isLoading: boolean;
8
+ isError: boolean;
9
+ error: null;
10
+ refetch: () => Promise<{
11
+ data: undefined;
12
+ }>;
13
+ };
@@ -0,0 +1,46 @@
1
+ /*
2
+ * Copyright (c) 2025-2026 Datalayer, Inc.
3
+ * Distributed under the terms of the Modified BSD License.
4
+ */
5
+ import { useContext } from 'react';
6
+ import { useQuery, QueryClientContext } from '@tanstack/react-query';
7
+ /**
8
+ * Hook to fetch available skills from backend.
9
+ */
10
+ export function useSkills(enabled, baseEndpoint, authToken) {
11
+ const queryClient = useContext(QueryClientContext);
12
+ if (!queryClient) {
13
+ return {
14
+ data: undefined,
15
+ isLoading: false,
16
+ isError: false,
17
+ error: null,
18
+ refetch: () => Promise.resolve({ data: undefined }),
19
+ };
20
+ }
21
+ // eslint-disable-next-line react-hooks/rules-of-hooks
22
+ return useQuery({
23
+ queryFn: async () => {
24
+ if (!baseEndpoint) {
25
+ return { skills: [], total: 0 };
26
+ }
27
+ // Derive skills endpoint from config endpoint.
28
+ const skillsEndpoint = baseEndpoint.replace('/configure', '/skills');
29
+ const headers = {
30
+ 'Content-Type': 'application/json',
31
+ };
32
+ if (authToken) {
33
+ headers['Authorization'] = `Bearer ${authToken}`;
34
+ }
35
+ const response = await fetch(skillsEndpoint, { headers });
36
+ if (!response.ok) {
37
+ throw new Error(`Skills fetch failed: ${response.statusText}`);
38
+ }
39
+ return response.json();
40
+ },
41
+ queryKey: ['skills', baseEndpoint || 'jupyter'],
42
+ enabled,
43
+ staleTime: 5 * 60 * 1000,
44
+ retry: 1,
45
+ });
46
+ }
@@ -0,0 +1,45 @@
1
+ import type { ToolApprovalFilters } from '../types/tool-approvals';
2
+ export declare function useToolApprovalsQuery(filters?: ToolApprovalFilters): import("@tanstack/react-query").UseQueryResult<import("../types").ToolApproval[], Error>;
3
+ export declare function usePendingApprovalCount(): import("@tanstack/react-query").UseQueryResult<{
4
+ count: number;
5
+ }, Error>;
6
+ export declare function useApproveToolRequest(): import("@tanstack/react-query").UseMutationResult<void, Error, {
7
+ id: string;
8
+ note?: string;
9
+ }, unknown>;
10
+ export declare function useRejectToolRequest(): import("@tanstack/react-query").UseMutationResult<void, Error, {
11
+ id: string;
12
+ note?: string;
13
+ }, unknown>;
14
+ export declare function useMarkToolApprovalRead(): import("@tanstack/react-query").UseMutationResult<void, Error, {
15
+ id: string;
16
+ }, unknown>;
17
+ export declare function useMarkToolApprovalUnread(): import("@tanstack/react-query").UseMutationResult<void, Error, {
18
+ id: string;
19
+ }, unknown>;
20
+ export declare function useDeleteToolApproval(): import("@tanstack/react-query").UseMutationResult<void, Error, {
21
+ id: string;
22
+ }, unknown>;
23
+ export declare function useToolApprovals(filters?: ToolApprovalFilters): {
24
+ approvalsQuery: import("@tanstack/react-query").UseQueryResult<import("../types").ToolApproval[], Error>;
25
+ pendingCountQuery: import("@tanstack/react-query").UseQueryResult<{
26
+ count: number;
27
+ }, Error>;
28
+ approve: import("@tanstack/react-query").UseMutationResult<void, Error, {
29
+ id: string;
30
+ note?: string;
31
+ }, unknown>;
32
+ reject: import("@tanstack/react-query").UseMutationResult<void, Error, {
33
+ id: string;
34
+ note?: string;
35
+ }, unknown>;
36
+ markRead: import("@tanstack/react-query").UseMutationResult<void, Error, {
37
+ id: string;
38
+ }, unknown>;
39
+ markUnread: import("@tanstack/react-query").UseMutationResult<void, Error, {
40
+ id: string;
41
+ }, unknown>;
42
+ remove: import("@tanstack/react-query").UseMutationResult<void, Error, {
43
+ id: string;
44
+ }, unknown>;
45
+ };
@@ -0,0 +1,126 @@
1
+ /*
2
+ * Copyright (c) 2025-2026 Datalayer, Inc.
3
+ * Distributed under the terms of the Modified BSD License.
4
+ */
5
+ /**
6
+ * Agent tool approval hooks.
7
+ *
8
+ * @module hooks/useToolApprovals
9
+ */
10
+ import { useMemo } from 'react';
11
+ import { useQuery, useMutation, useQueryClient } from '@tanstack/react-query';
12
+ import { useCoreStore, useIAMStore } from '@datalayer/core/lib/state';
13
+ import { DEFAULT_SERVICE_URLS } from '@datalayer/core/lib/api/constants';
14
+ import { toolApprovals } from '../api';
15
+ // ─── Auth helpers ────────────────────────────────────────────────────
16
+ function useDashboardAuthToken() {
17
+ const token = useIAMStore((s) => s.token);
18
+ return token ?? '';
19
+ }
20
+ function useDashboardBaseUrl() {
21
+ const config = useCoreStore((s) => s.configuration);
22
+ return config?.aiagentsRunUrl ?? DEFAULT_SERVICE_URLS.AI_AGENTS;
23
+ }
24
+ // ─── Base hooks ──────────────────────────────────────────────────────
25
+ export function useToolApprovalsQuery(filters) {
26
+ const token = useDashboardAuthToken();
27
+ const baseUrl = useDashboardBaseUrl();
28
+ return useQuery({
29
+ queryKey: ['tool-approvals', filters],
30
+ queryFn: () => toolApprovals.getToolApprovals(token, filters, baseUrl),
31
+ enabled: !!token,
32
+ staleTime: 10_000,
33
+ });
34
+ }
35
+ export function usePendingApprovalCount() {
36
+ const token = useDashboardAuthToken();
37
+ const baseUrl = useDashboardBaseUrl();
38
+ return useQuery({
39
+ queryKey: ['tool-approvals', 'pending-count'],
40
+ queryFn: () => toolApprovals.getPendingApprovalCount(token, baseUrl),
41
+ enabled: !!token,
42
+ staleTime: 5_000,
43
+ });
44
+ }
45
+ export function useApproveToolRequest() {
46
+ const token = useDashboardAuthToken();
47
+ const baseUrl = useDashboardBaseUrl();
48
+ const queryClient = useQueryClient();
49
+ return useMutation({
50
+ mutationFn: ({ id, note }) => toolApprovals.approveToolRequest(token, id, note, baseUrl),
51
+ onSuccess: () => {
52
+ queryClient.invalidateQueries({ queryKey: ['tool-approvals'] });
53
+ },
54
+ });
55
+ }
56
+ export function useRejectToolRequest() {
57
+ const token = useDashboardAuthToken();
58
+ const baseUrl = useDashboardBaseUrl();
59
+ const queryClient = useQueryClient();
60
+ return useMutation({
61
+ mutationFn: ({ id, note }) => toolApprovals.rejectToolRequest(token, id, note, baseUrl),
62
+ onSuccess: () => {
63
+ queryClient.invalidateQueries({ queryKey: ['tool-approvals'] });
64
+ },
65
+ });
66
+ }
67
+ export function useMarkToolApprovalRead() {
68
+ const token = useDashboardAuthToken();
69
+ const baseUrl = useDashboardBaseUrl();
70
+ const queryClient = useQueryClient();
71
+ return useMutation({
72
+ mutationFn: ({ id }) => toolApprovals.markToolApprovalRead(token, id, baseUrl),
73
+ onSuccess: () => {
74
+ queryClient.invalidateQueries({ queryKey: ['tool-approvals'] });
75
+ },
76
+ });
77
+ }
78
+ export function useMarkToolApprovalUnread() {
79
+ const token = useDashboardAuthToken();
80
+ const baseUrl = useDashboardBaseUrl();
81
+ const queryClient = useQueryClient();
82
+ return useMutation({
83
+ mutationFn: ({ id }) => toolApprovals.markToolApprovalUnread(token, id, baseUrl),
84
+ onSuccess: () => {
85
+ queryClient.invalidateQueries({ queryKey: ['tool-approvals'] });
86
+ },
87
+ });
88
+ }
89
+ export function useDeleteToolApproval() {
90
+ const token = useDashboardAuthToken();
91
+ const baseUrl = useDashboardBaseUrl();
92
+ const queryClient = useQueryClient();
93
+ return useMutation({
94
+ mutationFn: ({ id }) => toolApprovals.deleteToolApproval(token, id, baseUrl),
95
+ onSuccess: () => {
96
+ queryClient.invalidateQueries({ queryKey: ['tool-approvals'] });
97
+ },
98
+ });
99
+ }
100
+ // ─── Composite hook ──────────────────────────────────────────────────
101
+ export function useToolApprovals(filters) {
102
+ const approvalsQuery = useToolApprovalsQuery(filters);
103
+ const pendingQuery = usePendingApprovalCount();
104
+ const approve = useApproveToolRequest();
105
+ const reject = useRejectToolRequest();
106
+ const markRead = useMarkToolApprovalRead();
107
+ const markUnread = useMarkToolApprovalUnread();
108
+ const remove = useDeleteToolApproval();
109
+ return useMemo(() => ({
110
+ approvalsQuery,
111
+ pendingCountQuery: pendingQuery,
112
+ approve,
113
+ reject,
114
+ markRead,
115
+ markUnread,
116
+ remove,
117
+ }), [
118
+ approvalsQuery,
119
+ pendingQuery,
120
+ approve,
121
+ reject,
122
+ markRead,
123
+ markUnread,
124
+ remove,
125
+ ]);
126
+ }