flock-core 0.4.542__py3-none-any.whl → 0.5.0__py3-none-any.whl

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.

Potentially problematic release.


This version of flock-core might be problematic. Click here for more details.

Files changed (501) hide show
  1. flock/__init__.py +12 -217
  2. flock/agent.py +1079 -0
  3. flock/api/themes.py +71 -0
  4. flock/artifacts.py +86 -0
  5. flock/cli.py +147 -0
  6. flock/components.py +189 -0
  7. flock/dashboard/__init__.py +30 -0
  8. flock/dashboard/collector.py +559 -0
  9. flock/dashboard/events.py +188 -0
  10. flock/dashboard/graph_builder.py +563 -0
  11. flock/dashboard/launcher.py +235 -0
  12. flock/dashboard/models/graph.py +156 -0
  13. flock/dashboard/service.py +991 -0
  14. flock/dashboard/static_v2/assets/index-DFRnI_mt.js +111 -0
  15. flock/dashboard/static_v2/assets/index-fPLNdmp1.css +1 -0
  16. flock/dashboard/static_v2/index.html +13 -0
  17. flock/dashboard/websocket.py +246 -0
  18. flock/engines/__init__.py +6 -0
  19. flock/engines/dspy_engine.py +932 -0
  20. flock/examples.py +131 -0
  21. flock/frontend/README.md +778 -0
  22. flock/frontend/docs/DESIGN_SYSTEM.md +1980 -0
  23. flock/frontend/index.html +12 -0
  24. flock/frontend/package-lock.json +4337 -0
  25. flock/frontend/package.json +48 -0
  26. flock/frontend/src/App.tsx +139 -0
  27. flock/frontend/src/__tests__/integration/graph-snapshot.test.tsx +647 -0
  28. flock/frontend/src/__tests__/integration/indexeddb-persistence.test.tsx +699 -0
  29. flock/frontend/src/components/common/BuildInfo.tsx +39 -0
  30. flock/frontend/src/components/common/EmptyState.module.css +115 -0
  31. flock/frontend/src/components/common/EmptyState.tsx +128 -0
  32. flock/frontend/src/components/common/ErrorBoundary.module.css +169 -0
  33. flock/frontend/src/components/common/ErrorBoundary.tsx +118 -0
  34. flock/frontend/src/components/common/KeyboardShortcutsDialog.css +251 -0
  35. flock/frontend/src/components/common/KeyboardShortcutsDialog.tsx +151 -0
  36. flock/frontend/src/components/common/LoadingSpinner.module.css +97 -0
  37. flock/frontend/src/components/common/LoadingSpinner.tsx +29 -0
  38. flock/frontend/src/components/controls/PublishControl.css +547 -0
  39. flock/frontend/src/components/controls/PublishControl.test.tsx +543 -0
  40. flock/frontend/src/components/controls/PublishControl.tsx +432 -0
  41. flock/frontend/src/components/details/DetailWindowContainer.tsx +58 -0
  42. flock/frontend/src/components/details/LiveOutputTab.test.tsx +792 -0
  43. flock/frontend/src/components/details/LiveOutputTab.tsx +220 -0
  44. flock/frontend/src/components/details/MessageDetailWindow.tsx +439 -0
  45. flock/frontend/src/components/details/MessageHistoryTab.tsx +374 -0
  46. flock/frontend/src/components/details/NodeDetailWindow.test.tsx +501 -0
  47. flock/frontend/src/components/details/NodeDetailWindow.tsx +218 -0
  48. flock/frontend/src/components/details/RunStatusTab.tsx +348 -0
  49. flock/frontend/src/components/details/tabs.test.tsx +1015 -0
  50. flock/frontend/src/components/filters/ArtifactTypeFilter.tsx +21 -0
  51. flock/frontend/src/components/filters/CorrelationIDFilter.module.css +102 -0
  52. flock/frontend/src/components/filters/CorrelationIDFilter.test.tsx +197 -0
  53. flock/frontend/src/components/filters/CorrelationIDFilter.tsx +121 -0
  54. flock/frontend/src/components/filters/FilterFlyout.module.css +104 -0
  55. flock/frontend/src/components/filters/FilterFlyout.tsx +80 -0
  56. flock/frontend/src/components/filters/FilterPills.module.css +220 -0
  57. flock/frontend/src/components/filters/FilterPills.test.tsx +189 -0
  58. flock/frontend/src/components/filters/FilterPills.tsx +143 -0
  59. flock/frontend/src/components/filters/ProducerFilter.tsx +21 -0
  60. flock/frontend/src/components/filters/SavedFiltersControl.module.css +60 -0
  61. flock/frontend/src/components/filters/SavedFiltersControl.test.tsx +158 -0
  62. flock/frontend/src/components/filters/SavedFiltersControl.tsx +159 -0
  63. flock/frontend/src/components/filters/TagFilter.tsx +21 -0
  64. flock/frontend/src/components/filters/TimeRangeFilter.module.css +115 -0
  65. flock/frontend/src/components/filters/TimeRangeFilter.test.tsx +154 -0
  66. flock/frontend/src/components/filters/TimeRangeFilter.tsx +110 -0
  67. flock/frontend/src/components/filters/VisibilityFilter.tsx +21 -0
  68. flock/frontend/src/components/graph/AgentNode.test.tsx +77 -0
  69. flock/frontend/src/components/graph/AgentNode.tsx +324 -0
  70. flock/frontend/src/components/graph/GraphCanvas.tsx +613 -0
  71. flock/frontend/src/components/graph/MessageFlowEdge.tsx +128 -0
  72. flock/frontend/src/components/graph/MessageNode.test.tsx +64 -0
  73. flock/frontend/src/components/graph/MessageNode.tsx +129 -0
  74. flock/frontend/src/components/graph/MiniMap.tsx +47 -0
  75. flock/frontend/src/components/graph/TransformEdge.tsx +123 -0
  76. flock/frontend/src/components/layout/DashboardLayout.css +420 -0
  77. flock/frontend/src/components/layout/DashboardLayout.tsx +287 -0
  78. flock/frontend/src/components/layout/Header.module.css +88 -0
  79. flock/frontend/src/components/layout/Header.tsx +52 -0
  80. flock/frontend/src/components/modules/HistoricalArtifactsModule.module.css +288 -0
  81. flock/frontend/src/components/modules/HistoricalArtifactsModule.tsx +450 -0
  82. flock/frontend/src/components/modules/HistoricalArtifactsModuleWrapper.tsx +13 -0
  83. flock/frontend/src/components/modules/JsonAttributeRenderer.tsx +140 -0
  84. flock/frontend/src/components/modules/ModuleRegistry.test.ts +333 -0
  85. flock/frontend/src/components/modules/ModuleRegistry.ts +93 -0
  86. flock/frontend/src/components/modules/ModuleWindow.tsx +223 -0
  87. flock/frontend/src/components/modules/TraceModuleJaeger.tsx +1971 -0
  88. flock/frontend/src/components/modules/TraceModuleJaegerWrapper.tsx +13 -0
  89. flock/frontend/src/components/modules/registerModules.ts +29 -0
  90. flock/frontend/src/components/settings/AdvancedSettings.tsx +175 -0
  91. flock/frontend/src/components/settings/AppearanceSettings.tsx +185 -0
  92. flock/frontend/src/components/settings/GraphSettings.tsx +110 -0
  93. flock/frontend/src/components/settings/MultiSelect.tsx +235 -0
  94. flock/frontend/src/components/settings/SettingsPanel.css +327 -0
  95. flock/frontend/src/components/settings/SettingsPanel.tsx +131 -0
  96. flock/frontend/src/components/settings/ThemeSelector.tsx +298 -0
  97. flock/frontend/src/components/settings/TracingSettings.tsx +404 -0
  98. flock/frontend/src/hooks/useKeyboardShortcuts.ts +148 -0
  99. flock/frontend/src/hooks/useModulePersistence.test.ts +442 -0
  100. flock/frontend/src/hooks/useModulePersistence.ts +154 -0
  101. flock/frontend/src/hooks/useModules.ts +157 -0
  102. flock/frontend/src/hooks/usePersistence.ts +141 -0
  103. flock/frontend/src/main.tsx +13 -0
  104. flock/frontend/src/services/api.ts +337 -0
  105. flock/frontend/src/services/graphService.test.ts +330 -0
  106. flock/frontend/src/services/graphService.ts +75 -0
  107. flock/frontend/src/services/indexeddb.test.ts +793 -0
  108. flock/frontend/src/services/indexeddb.ts +848 -0
  109. flock/frontend/src/services/layout.test.ts +437 -0
  110. flock/frontend/src/services/layout.ts +357 -0
  111. flock/frontend/src/services/themeApplicator.ts +140 -0
  112. flock/frontend/src/services/themeService.ts +77 -0
  113. flock/frontend/src/services/websocket.ts +650 -0
  114. flock/frontend/src/store/filterStore.test.ts +250 -0
  115. flock/frontend/src/store/filterStore.ts +272 -0
  116. flock/frontend/src/store/graphStore.test.ts +570 -0
  117. flock/frontend/src/store/graphStore.ts +462 -0
  118. flock/frontend/src/store/moduleStore.test.ts +253 -0
  119. flock/frontend/src/store/moduleStore.ts +75 -0
  120. flock/frontend/src/store/settingsStore.ts +188 -0
  121. flock/frontend/src/store/streamStore.ts +68 -0
  122. flock/frontend/src/store/uiStore.test.ts +54 -0
  123. flock/frontend/src/store/uiStore.ts +122 -0
  124. flock/frontend/src/store/wsStore.ts +34 -0
  125. flock/frontend/src/styles/index.css +15 -0
  126. flock/frontend/src/styles/scrollbar.css +47 -0
  127. flock/frontend/src/styles/variables.css +488 -0
  128. flock/frontend/src/test/setup.ts +1 -0
  129. flock/frontend/src/types/filters.ts +47 -0
  130. flock/frontend/src/types/graph.ts +95 -0
  131. flock/frontend/src/types/modules.ts +10 -0
  132. flock/frontend/src/types/theme.ts +55 -0
  133. flock/frontend/src/utils/artifacts.ts +24 -0
  134. flock/frontend/src/utils/mockData.ts +98 -0
  135. flock/frontend/src/utils/performance.ts +16 -0
  136. flock/frontend/src/vite-env.d.ts +17 -0
  137. flock/frontend/tsconfig.json +27 -0
  138. flock/frontend/tsconfig.node.json +11 -0
  139. flock/frontend/vite.config.ts +25 -0
  140. flock/frontend/vitest.config.ts +11 -0
  141. flock/{core/util → helper}/cli_helper.py +9 -5
  142. flock/{core/logging → logging}/__init__.py +2 -3
  143. flock/logging/auto_trace.py +159 -0
  144. flock/{core/logging → logging}/formatters/enum_builder.py +3 -4
  145. flock/{core/logging → logging}/formatters/theme_builder.py +19 -44
  146. flock/{core/logging → logging}/formatters/themed_formatter.py +69 -107
  147. flock/{core/logging → logging}/logging.py +78 -61
  148. flock/{core/logging → logging}/telemetry.py +66 -26
  149. flock/{core/logging → logging}/telemetry_exporter/base_exporter.py +2 -2
  150. flock/logging/telemetry_exporter/duckdb_exporter.py +216 -0
  151. flock/{core/logging → logging}/telemetry_exporter/file_exporter.py +13 -10
  152. flock/{core/logging → logging}/telemetry_exporter/sqlite_exporter.py +2 -3
  153. flock/logging/trace_and_logged.py +304 -0
  154. flock/mcp/__init__.py +91 -0
  155. flock/{core/mcp/mcp_client.py → mcp/client.py} +131 -158
  156. flock/{core/mcp/mcp_config.py → mcp/config.py} +86 -132
  157. flock/mcp/manager.py +286 -0
  158. flock/mcp/servers/sse/__init__.py +1 -1
  159. flock/mcp/servers/sse/flock_sse_server.py +16 -58
  160. flock/mcp/servers/stdio/__init__.py +1 -1
  161. flock/mcp/servers/stdio/flock_stdio_server.py +13 -53
  162. flock/mcp/servers/streamable_http/flock_streamable_http_server.py +22 -67
  163. flock/mcp/servers/websockets/flock_websocket_server.py +12 -45
  164. flock/{core/mcp/flock_mcp_tool_base.py → mcp/tool.py} +24 -78
  165. flock/mcp/types/__init__.py +42 -0
  166. flock/{core/mcp → mcp}/types/callbacks.py +12 -15
  167. flock/{core/mcp → mcp}/types/factories.py +7 -6
  168. flock/{core/mcp → mcp}/types/handlers.py +13 -18
  169. flock/{core/mcp → mcp}/types/types.py +70 -74
  170. flock/{core/mcp → mcp}/util/helpers.py +3 -3
  171. flock/orchestrator.py +970 -0
  172. flock/registry.py +148 -0
  173. flock/runtime.py +262 -0
  174. flock/service.py +277 -0
  175. flock/store.py +1214 -0
  176. flock/subscription.py +111 -0
  177. flock/themes/andromeda.toml +1 -1
  178. flock/themes/apple-system-colors.toml +1 -1
  179. flock/themes/arcoiris.toml +1 -1
  180. flock/themes/atomonelight.toml +1 -1
  181. flock/themes/ayu copy.toml +1 -1
  182. flock/themes/ayu-light.toml +1 -1
  183. flock/themes/belafonte-day.toml +1 -1
  184. flock/themes/belafonte-night.toml +1 -1
  185. flock/themes/blulocodark.toml +1 -1
  186. flock/themes/breeze.toml +1 -1
  187. flock/themes/broadcast.toml +1 -1
  188. flock/themes/brogrammer.toml +1 -1
  189. flock/themes/builtin-dark.toml +1 -1
  190. flock/themes/builtin-pastel-dark.toml +1 -1
  191. flock/themes/catppuccin-latte.toml +1 -1
  192. flock/themes/catppuccin-macchiato.toml +1 -1
  193. flock/themes/catppuccin-mocha.toml +1 -1
  194. flock/themes/cga.toml +1 -1
  195. flock/themes/chalk.toml +1 -1
  196. flock/themes/ciapre.toml +1 -1
  197. flock/themes/coffee-theme.toml +1 -1
  198. flock/themes/cyberpunkscarletprotocol.toml +1 -1
  199. flock/themes/dark+.toml +1 -1
  200. flock/themes/darkermatrix.toml +1 -1
  201. flock/themes/darkmatrix.toml +2 -2
  202. flock/themes/darkside.toml +1 -1
  203. flock/themes/deep.toml +2 -2
  204. flock/themes/desert.toml +1 -1
  205. flock/themes/django.toml +1 -1
  206. flock/themes/djangosmooth.toml +1 -1
  207. flock/themes/doomone.toml +1 -1
  208. flock/themes/dotgov.toml +1 -1
  209. flock/themes/dracula+.toml +1 -1
  210. flock/themes/duckbones.toml +1 -1
  211. flock/themes/encom.toml +1 -1
  212. flock/themes/espresso.toml +1 -1
  213. flock/themes/everblush.toml +1 -1
  214. flock/themes/fairyfloss.toml +1 -1
  215. flock/themes/fideloper.toml +1 -1
  216. flock/themes/fishtank.toml +1 -1
  217. flock/themes/flexoki-light.toml +1 -1
  218. flock/themes/floraverse.toml +1 -1
  219. flock/themes/framer.toml +1 -1
  220. flock/themes/galizur.toml +1 -1
  221. flock/themes/github.toml +1 -1
  222. flock/themes/grass.toml +1 -1
  223. flock/themes/grey-green.toml +1 -1
  224. flock/themes/gruvboxlight.toml +1 -1
  225. flock/themes/guezwhoz.toml +1 -1
  226. flock/themes/harper.toml +1 -1
  227. flock/themes/hax0r-blue.toml +1 -1
  228. flock/themes/hopscotch.256.toml +1 -1
  229. flock/themes/ic-green-ppl.toml +1 -1
  230. flock/themes/iceberg-dark.toml +1 -1
  231. flock/themes/japanesque.toml +1 -1
  232. flock/themes/jubi.toml +1 -1
  233. flock/themes/kibble.toml +1 -1
  234. flock/themes/kolorit.toml +1 -1
  235. flock/themes/kurokula.toml +1 -1
  236. flock/themes/materialdesigncolors.toml +1 -1
  237. flock/themes/matrix.toml +1 -1
  238. flock/themes/mellifluous.toml +1 -1
  239. flock/themes/midnight-in-mojave.toml +1 -1
  240. flock/themes/monokai-remastered.toml +1 -1
  241. flock/themes/monokai-soda.toml +1 -1
  242. flock/themes/neon.toml +1 -1
  243. flock/themes/neopolitan.toml +5 -5
  244. flock/themes/nord-light.toml +1 -1
  245. flock/themes/ocean.toml +1 -1
  246. flock/themes/onehalfdark.toml +1 -1
  247. flock/themes/onehalflight.toml +1 -1
  248. flock/themes/palenighthc.toml +1 -1
  249. flock/themes/paulmillr.toml +1 -1
  250. flock/themes/pencildark.toml +1 -1
  251. flock/themes/pnevma.toml +1 -1
  252. flock/themes/purple-rain.toml +1 -1
  253. flock/themes/purplepeter.toml +1 -1
  254. flock/themes/raycast-dark.toml +1 -1
  255. flock/themes/red-sands.toml +1 -1
  256. flock/themes/relaxed.toml +1 -1
  257. flock/themes/retro.toml +1 -1
  258. flock/themes/rose-pine.toml +1 -1
  259. flock/themes/royal.toml +1 -1
  260. flock/themes/ryuuko.toml +1 -1
  261. flock/themes/sakura.toml +1 -1
  262. flock/themes/scarlet-protocol.toml +1 -1
  263. flock/themes/seoulbones-dark.toml +1 -1
  264. flock/themes/shades-of-purple.toml +1 -1
  265. flock/themes/smyck.toml +1 -1
  266. flock/themes/softserver.toml +1 -1
  267. flock/themes/solarized-darcula.toml +1 -1
  268. flock/themes/square.toml +1 -1
  269. flock/themes/sugarplum.toml +1 -1
  270. flock/themes/thayer-bright.toml +1 -1
  271. flock/themes/tokyonight.toml +1 -1
  272. flock/themes/tomorrow.toml +1 -1
  273. flock/themes/ubuntu.toml +1 -1
  274. flock/themes/ultradark.toml +1 -1
  275. flock/themes/ultraviolent.toml +1 -1
  276. flock/themes/unikitty.toml +1 -1
  277. flock/themes/urple.toml +1 -1
  278. flock/themes/vesper.toml +1 -1
  279. flock/themes/vimbones.toml +1 -1
  280. flock/themes/wildcherry.toml +1 -1
  281. flock/themes/wilmersdorf.toml +1 -1
  282. flock/themes/wryan.toml +1 -1
  283. flock/themes/xcodedarkhc.toml +1 -1
  284. flock/themes/xcodelight.toml +1 -1
  285. flock/themes/zenbones-light.toml +1 -1
  286. flock/themes/zenwritten-dark.toml +1 -1
  287. flock/utilities.py +301 -0
  288. flock/utility/output_utility_component.py +226 -0
  289. flock/visibility.py +107 -0
  290. flock_core-0.5.0.dist-info/METADATA +964 -0
  291. flock_core-0.5.0.dist-info/RECORD +525 -0
  292. flock_core-0.5.0.dist-info/entry_points.txt +2 -0
  293. {flock_core-0.4.542.dist-info → flock_core-0.5.0.dist-info}/licenses/LICENSE +1 -1
  294. flock/adapter/__init__.py +0 -14
  295. flock/adapter/azure_adapter.py +0 -68
  296. flock/adapter/chroma_adapter.py +0 -73
  297. flock/adapter/faiss_adapter.py +0 -97
  298. flock/adapter/pinecone_adapter.py +0 -51
  299. flock/adapter/vector_base.py +0 -47
  300. flock/cli/assets/release_notes.md +0 -140
  301. flock/cli/config.py +0 -8
  302. flock/cli/constants.py +0 -36
  303. flock/cli/create_agent.py +0 -1
  304. flock/cli/create_flock.py +0 -280
  305. flock/cli/execute_flock.py +0 -620
  306. flock/cli/load_agent.py +0 -1
  307. flock/cli/load_examples.py +0 -1
  308. flock/cli/load_flock.py +0 -192
  309. flock/cli/load_release_notes.py +0 -20
  310. flock/cli/loaded_flock_cli.py +0 -254
  311. flock/cli/manage_agents.py +0 -459
  312. flock/cli/registry_management.py +0 -889
  313. flock/cli/runner.py +0 -41
  314. flock/cli/settings.py +0 -857
  315. flock/cli/utils.py +0 -135
  316. flock/cli/view_results.py +0 -29
  317. flock/cli/yaml_editor.py +0 -396
  318. flock/config.py +0 -56
  319. flock/core/__init__.py +0 -44
  320. flock/core/api/__init__.py +0 -10
  321. flock/core/api/custom_endpoint.py +0 -45
  322. flock/core/api/endpoints.py +0 -262
  323. flock/core/api/main.py +0 -162
  324. flock/core/api/models.py +0 -101
  325. flock/core/api/run_store.py +0 -224
  326. flock/core/api/runner.py +0 -44
  327. flock/core/api/service.py +0 -214
  328. flock/core/config/flock_agent_config.py +0 -11
  329. flock/core/config/scheduled_agent_config.py +0 -40
  330. flock/core/context/context.py +0 -214
  331. flock/core/context/context_manager.py +0 -40
  332. flock/core/context/context_vars.py +0 -11
  333. flock/core/evaluation/utils.py +0 -395
  334. flock/core/execution/batch_executor.py +0 -369
  335. flock/core/execution/evaluation_executor.py +0 -438
  336. flock/core/execution/local_executor.py +0 -31
  337. flock/core/execution/opik_executor.py +0 -103
  338. flock/core/execution/temporal_executor.py +0 -166
  339. flock/core/flock.py +0 -1003
  340. flock/core/flock_agent.py +0 -1258
  341. flock/core/flock_evaluator.py +0 -60
  342. flock/core/flock_factory.py +0 -513
  343. flock/core/flock_module.py +0 -207
  344. flock/core/flock_registry.py +0 -702
  345. flock/core/flock_router.py +0 -83
  346. flock/core/flock_scheduler.py +0 -166
  347. flock/core/flock_server_manager.py +0 -136
  348. flock/core/interpreter/python_interpreter.py +0 -689
  349. flock/core/logging/live_capture.py +0 -137
  350. flock/core/logging/trace_and_logged.py +0 -59
  351. flock/core/mcp/__init__.py +0 -1
  352. flock/core/mcp/flock_mcp_server.py +0 -640
  353. flock/core/mcp/mcp_client_manager.py +0 -201
  354. flock/core/mcp/types/__init__.py +0 -1
  355. flock/core/mixin/dspy_integration.py +0 -445
  356. flock/core/mixin/prompt_parser.py +0 -125
  357. flock/core/serialization/__init__.py +0 -13
  358. flock/core/serialization/callable_registry.py +0 -52
  359. flock/core/serialization/flock_serializer.py +0 -854
  360. flock/core/serialization/json_encoder.py +0 -41
  361. flock/core/serialization/secure_serializer.py +0 -175
  362. flock/core/serialization/serializable.py +0 -342
  363. flock/core/serialization/serialization_utils.py +0 -409
  364. flock/core/util/file_path_utils.py +0 -223
  365. flock/core/util/hydrator.py +0 -309
  366. flock/core/util/input_resolver.py +0 -141
  367. flock/core/util/loader.py +0 -59
  368. flock/core/util/splitter.py +0 -219
  369. flock/di.py +0 -41
  370. flock/evaluators/__init__.py +0 -1
  371. flock/evaluators/declarative/__init__.py +0 -1
  372. flock/evaluators/declarative/declarative_evaluator.py +0 -217
  373. flock/evaluators/memory/memory_evaluator.py +0 -90
  374. flock/evaluators/test/test_case_evaluator.py +0 -38
  375. flock/evaluators/zep/zep_evaluator.py +0 -59
  376. flock/modules/__init__.py +0 -1
  377. flock/modules/assertion/__init__.py +0 -1
  378. flock/modules/assertion/assertion_module.py +0 -286
  379. flock/modules/callback/__init__.py +0 -1
  380. flock/modules/callback/callback_module.py +0 -91
  381. flock/modules/enterprise_memory/README.md +0 -99
  382. flock/modules/enterprise_memory/enterprise_memory_module.py +0 -526
  383. flock/modules/mem0/__init__.py +0 -1
  384. flock/modules/mem0/mem0_module.py +0 -126
  385. flock/modules/mem0_async/__init__.py +0 -1
  386. flock/modules/mem0_async/async_mem0_module.py +0 -126
  387. flock/modules/memory/__init__.py +0 -1
  388. flock/modules/memory/memory_module.py +0 -429
  389. flock/modules/memory/memory_parser.py +0 -125
  390. flock/modules/memory/memory_storage.py +0 -736
  391. flock/modules/output/__init__.py +0 -1
  392. flock/modules/output/output_module.py +0 -196
  393. flock/modules/performance/__init__.py +0 -1
  394. flock/modules/performance/metrics_module.py +0 -678
  395. flock/modules/zep/__init__.py +0 -1
  396. flock/modules/zep/zep_module.py +0 -192
  397. flock/platform/docker_tools.py +0 -49
  398. flock/platform/jaeger_install.py +0 -86
  399. flock/routers/__init__.py +0 -1
  400. flock/routers/agent/__init__.py +0 -1
  401. flock/routers/agent/agent_router.py +0 -236
  402. flock/routers/agent/handoff_agent.py +0 -58
  403. flock/routers/conditional/conditional_router.py +0 -486
  404. flock/routers/default/__init__.py +0 -1
  405. flock/routers/default/default_router.py +0 -80
  406. flock/routers/feedback/feedback_router.py +0 -114
  407. flock/routers/list_generator/list_generator_router.py +0 -166
  408. flock/routers/llm/__init__.py +0 -1
  409. flock/routers/llm/llm_router.py +0 -365
  410. flock/tools/__init__.py +0 -0
  411. flock/tools/azure_tools.py +0 -781
  412. flock/tools/code_tools.py +0 -167
  413. flock/tools/file_tools.py +0 -149
  414. flock/tools/github_tools.py +0 -157
  415. flock/tools/markdown_tools.py +0 -205
  416. flock/tools/system_tools.py +0 -9
  417. flock/tools/text_tools.py +0 -810
  418. flock/tools/web_tools.py +0 -92
  419. flock/tools/zendesk_tools.py +0 -501
  420. flock/webapp/__init__.py +0 -1
  421. flock/webapp/app/__init__.py +0 -0
  422. flock/webapp/app/api/__init__.py +0 -0
  423. flock/webapp/app/api/agent_management.py +0 -237
  424. flock/webapp/app/api/execution.py +0 -503
  425. flock/webapp/app/api/flock_management.py +0 -125
  426. flock/webapp/app/api/registry_viewer.py +0 -29
  427. flock/webapp/app/chat.py +0 -662
  428. flock/webapp/app/config.py +0 -104
  429. flock/webapp/app/dependencies.py +0 -117
  430. flock/webapp/app/main.py +0 -1086
  431. flock/webapp/app/middleware.py +0 -113
  432. flock/webapp/app/models_ui.py +0 -7
  433. flock/webapp/app/services/__init__.py +0 -0
  434. flock/webapp/app/services/feedback_file_service.py +0 -363
  435. flock/webapp/app/services/flock_service.py +0 -345
  436. flock/webapp/app/services/sharing_models.py +0 -81
  437. flock/webapp/app/services/sharing_store.py +0 -597
  438. flock/webapp/app/templates/theme_mapper.html +0 -326
  439. flock/webapp/app/theme_mapper.py +0 -811
  440. flock/webapp/app/utils.py +0 -85
  441. flock/webapp/run.py +0 -219
  442. flock/webapp/static/css/chat.css +0 -301
  443. flock/webapp/static/css/components.css +0 -167
  444. flock/webapp/static/css/header.css +0 -39
  445. flock/webapp/static/css/layout.css +0 -281
  446. flock/webapp/static/css/sidebar.css +0 -127
  447. flock/webapp/static/css/two-pane.css +0 -48
  448. flock/webapp/templates/base.html +0 -389
  449. flock/webapp/templates/chat.html +0 -152
  450. flock/webapp/templates/chat_settings.html +0 -19
  451. flock/webapp/templates/flock_editor.html +0 -16
  452. flock/webapp/templates/index.html +0 -12
  453. flock/webapp/templates/partials/_agent_detail_form.html +0 -93
  454. flock/webapp/templates/partials/_agent_list.html +0 -18
  455. flock/webapp/templates/partials/_agent_manager_view.html +0 -51
  456. flock/webapp/templates/partials/_agent_tools_checklist.html +0 -14
  457. flock/webapp/templates/partials/_chat_container.html +0 -15
  458. flock/webapp/templates/partials/_chat_messages.html +0 -57
  459. flock/webapp/templates/partials/_chat_settings_form.html +0 -85
  460. flock/webapp/templates/partials/_create_flock_form.html +0 -50
  461. flock/webapp/templates/partials/_dashboard_flock_detail.html +0 -17
  462. flock/webapp/templates/partials/_dashboard_flock_file_list.html +0 -16
  463. flock/webapp/templates/partials/_dashboard_flock_properties_preview.html +0 -28
  464. flock/webapp/templates/partials/_dashboard_upload_flock_form.html +0 -16
  465. flock/webapp/templates/partials/_dynamic_input_form_content.html +0 -22
  466. flock/webapp/templates/partials/_env_vars_table.html +0 -23
  467. flock/webapp/templates/partials/_execution_form.html +0 -127
  468. flock/webapp/templates/partials/_execution_view_container.html +0 -28
  469. flock/webapp/templates/partials/_flock_file_list.html +0 -23
  470. flock/webapp/templates/partials/_flock_properties_form.html +0 -52
  471. flock/webapp/templates/partials/_flock_upload_form.html +0 -16
  472. flock/webapp/templates/partials/_header_flock_status.html +0 -5
  473. flock/webapp/templates/partials/_live_logs.html +0 -13
  474. flock/webapp/templates/partials/_load_manager_view.html +0 -49
  475. flock/webapp/templates/partials/_registry_table.html +0 -25
  476. flock/webapp/templates/partials/_registry_viewer_content.html +0 -70
  477. flock/webapp/templates/partials/_results_display.html +0 -78
  478. flock/webapp/templates/partials/_settings_env_content.html +0 -9
  479. flock/webapp/templates/partials/_settings_theme_content.html +0 -14
  480. flock/webapp/templates/partials/_settings_view.html +0 -36
  481. flock/webapp/templates/partials/_share_chat_link_snippet.html +0 -11
  482. flock/webapp/templates/partials/_share_link_snippet.html +0 -35
  483. flock/webapp/templates/partials/_sidebar.html +0 -74
  484. flock/webapp/templates/partials/_structured_data_view.html +0 -40
  485. flock/webapp/templates/partials/_theme_preview.html +0 -36
  486. flock/webapp/templates/registry_viewer.html +0 -84
  487. flock/webapp/templates/shared_run_page.html +0 -140
  488. flock/workflow/__init__.py +0 -0
  489. flock/workflow/activities.py +0 -237
  490. flock/workflow/agent_activities.py +0 -24
  491. flock/workflow/agent_execution_activity.py +0 -240
  492. flock/workflow/flock_workflow.py +0 -225
  493. flock/workflow/temporal_config.py +0 -96
  494. flock/workflow/temporal_setup.py +0 -60
  495. flock_core-0.4.542.dist-info/METADATA +0 -676
  496. flock_core-0.4.542.dist-info/RECORD +0 -572
  497. flock_core-0.4.542.dist-info/entry_points.txt +0 -2
  498. /flock/{core/logging → logging}/formatters/themes.py +0 -0
  499. /flock/{core/logging → logging}/span_middleware/baggage_span_processor.py +0 -0
  500. /flock/{core/mcp → mcp}/util/__init__.py +0 -0
  501. {flock_core-0.4.542.dist-info → flock_core-0.5.0.dist-info}/WHEEL +0 -0
@@ -0,0 +1,1015 @@
1
+ /**
2
+ * Unit tests for MessageHistoryTab and RunStatusTab components.
3
+ *
4
+ * Tests verify message history display, run status metrics, tab switching,
5
+ * and default tab preference handling.
6
+ */
7
+
8
+ import { describe, it, expect, beforeEach } from 'vitest';
9
+ import { render, screen, fireEvent } from '@testing-library/react';
10
+ import { useUIStore } from '../../store/uiStore';
11
+
12
+ // Message data types
13
+ interface MessageHistoryEntry {
14
+ id: string;
15
+ type: string;
16
+ direction: 'consumed' | 'published';
17
+ payload: any;
18
+ timestamp: number;
19
+ correlationId: string;
20
+ }
21
+
22
+ // Run status data types
23
+ interface RunStatusEntry {
24
+ runId: string;
25
+ startTime: number;
26
+ endTime: number;
27
+ duration: number;
28
+ status: 'completed' | 'error' | 'running';
29
+ metrics: {
30
+ tokensUsed?: number;
31
+ costUsd?: number;
32
+ artifactsProduced?: number;
33
+ };
34
+ errorMessage?: string;
35
+ }
36
+
37
+ // Mock MessageHistoryTab component
38
+ const MockMessageHistoryTab = ({
39
+ nodeId,
40
+ messages,
41
+ }: {
42
+ nodeId: string;
43
+ messages: MessageHistoryEntry[];
44
+ }) => {
45
+ const formatTimestamp = (timestamp: number) => {
46
+ return new Date(timestamp).toLocaleString();
47
+ };
48
+
49
+ const formatPayload = (payload: any) => {
50
+ try {
51
+ return JSON.stringify(payload, null, 2);
52
+ } catch {
53
+ return String(payload);
54
+ }
55
+ };
56
+
57
+ return (
58
+ <div data-testid={`message-history-${nodeId}`}>
59
+ {messages.length === 0 ? (
60
+ <div data-testid="empty-messages">No messages yet</div>
61
+ ) : (
62
+ <table data-testid="message-table">
63
+ <thead>
64
+ <tr>
65
+ <th>Time</th>
66
+ <th>Direction</th>
67
+ <th>Type</th>
68
+ <th>Correlation ID</th>
69
+ <th>Payload</th>
70
+ </tr>
71
+ </thead>
72
+ <tbody>
73
+ {messages.map((msg) => (
74
+ <tr key={msg.id} data-testid={`message-row-${msg.id}`}>
75
+ <td data-testid={`msg-time-${msg.id}`}>{formatTimestamp(msg.timestamp)}</td>
76
+ <td
77
+ data-testid={`msg-direction-${msg.id}`}
78
+ style={{
79
+ color: msg.direction === 'consumed' ? '#8be9fd' : '#50fa7b',
80
+ }}
81
+ >
82
+ {msg.direction === 'consumed' ? '↓ Consumed' : '↑ Published'}
83
+ </td>
84
+ <td data-testid={`msg-type-${msg.id}`}>{msg.type}</td>
85
+ <td data-testid={`msg-correlation-${msg.id}`}>{msg.correlationId}</td>
86
+ <td data-testid={`msg-payload-${msg.id}`}>
87
+ <pre style={{ fontSize: '10px', maxHeight: '100px', overflow: 'auto' }}>
88
+ {formatPayload(msg.payload)}
89
+ </pre>
90
+ </td>
91
+ </tr>
92
+ ))}
93
+ </tbody>
94
+ </table>
95
+ )}
96
+ </div>
97
+ );
98
+ };
99
+
100
+ // Mock RunStatusTab component
101
+ const MockRunStatusTab = ({
102
+ nodeId,
103
+ runs,
104
+ }: {
105
+ nodeId: string;
106
+ runs: RunStatusEntry[];
107
+ }) => {
108
+ const formatDuration = (ms: number) => {
109
+ if (ms < 1000) return `${ms}ms`;
110
+ return `${(ms / 1000).toFixed(2)}s`;
111
+ };
112
+
113
+ const formatTimestamp = (timestamp: number) => {
114
+ return new Date(timestamp).toLocaleString();
115
+ };
116
+
117
+ return (
118
+ <div data-testid={`run-status-${nodeId}`}>
119
+ {runs.length === 0 ? (
120
+ <div data-testid="empty-runs">No runs yet</div>
121
+ ) : (
122
+ <table data-testid="run-table">
123
+ <thead>
124
+ <tr>
125
+ <th>Run ID</th>
126
+ <th>Start Time</th>
127
+ <th>Duration</th>
128
+ <th>Status</th>
129
+ <th>Tokens</th>
130
+ <th>Cost</th>
131
+ <th>Artifacts</th>
132
+ </tr>
133
+ </thead>
134
+ <tbody>
135
+ {runs.map((run) => (
136
+ <tr key={run.runId} data-testid={`run-row-${run.runId}`}>
137
+ <td data-testid={`run-id-${run.runId}`}>{run.runId}</td>
138
+ <td data-testid={`run-start-${run.runId}`}>
139
+ {formatTimestamp(run.startTime)}
140
+ </td>
141
+ <td data-testid={`run-duration-${run.runId}`}>
142
+ {formatDuration(run.duration)}
143
+ </td>
144
+ <td
145
+ data-testid={`run-status-${run.runId}`}
146
+ style={{
147
+ color:
148
+ run.status === 'completed'
149
+ ? '#50fa7b'
150
+ : run.status === 'error'
151
+ ? '#ff5555'
152
+ : '#f1fa8c',
153
+ }}
154
+ >
155
+ {run.status}
156
+ </td>
157
+ <td data-testid={`run-tokens-${run.runId}`}>
158
+ {run.metrics.tokensUsed ?? '-'}
159
+ </td>
160
+ <td data-testid={`run-cost-${run.runId}`}>
161
+ {run.metrics.costUsd ? `$${run.metrics.costUsd.toFixed(4)}` : '-'}
162
+ </td>
163
+ <td data-testid={`run-artifacts-${run.runId}`}>
164
+ {run.metrics.artifactsProduced ?? '-'}
165
+ </td>
166
+ </tr>
167
+ ))}
168
+ </tbody>
169
+ </table>
170
+ )}
171
+ </div>
172
+ );
173
+ };
174
+
175
+ // Mock tab container component
176
+ const MockTabContainer = ({
177
+ nodeId,
178
+ messages,
179
+ runs,
180
+ }: {
181
+ nodeId: string;
182
+ messages: MessageHistoryEntry[];
183
+ runs: RunStatusEntry[];
184
+ }) => {
185
+ const window = useUIStore((state) => state.detailWindows.get(nodeId));
186
+ const updateDetailWindow = useUIStore((state) => state.updateDetailWindow);
187
+
188
+ if (!window) return null;
189
+
190
+ const activeTab = window.activeTab;
191
+
192
+ const handleTabClick = (tab: 'liveOutput' | 'messageHistory' | 'runStatus') => {
193
+ updateDetailWindow(nodeId, { activeTab: tab });
194
+ };
195
+
196
+ return (
197
+ <div data-testid={`tab-container-${nodeId}`}>
198
+ <div data-testid="tab-buttons">
199
+ <button
200
+ data-testid="tab-live-output"
201
+ onClick={() => handleTabClick('liveOutput')}
202
+ data-active={activeTab === 'liveOutput'}
203
+ style={{
204
+ fontWeight: activeTab === 'liveOutput' ? 'bold' : 'normal',
205
+ }}
206
+ >
207
+ Live Output
208
+ </button>
209
+ <button
210
+ data-testid="tab-message-history"
211
+ onClick={() => handleTabClick('messageHistory')}
212
+ data-active={activeTab === 'messageHistory'}
213
+ style={{
214
+ fontWeight: activeTab === 'messageHistory' ? 'bold' : 'normal',
215
+ }}
216
+ >
217
+ Message History
218
+ </button>
219
+ <button
220
+ data-testid="tab-run-status"
221
+ onClick={() => handleTabClick('runStatus')}
222
+ data-active={activeTab === 'runStatus'}
223
+ style={{
224
+ fontWeight: activeTab === 'runStatus' ? 'bold' : 'normal',
225
+ }}
226
+ >
227
+ Run Status
228
+ </button>
229
+ </div>
230
+
231
+ <div data-testid="tab-content">
232
+ {activeTab === 'liveOutput' && (
233
+ <div data-testid="live-output-content">Live Output Tab</div>
234
+ )}
235
+ {activeTab === 'messageHistory' && (
236
+ <MockMessageHistoryTab nodeId={nodeId} messages={messages} />
237
+ )}
238
+ {activeTab === 'runStatus' && (
239
+ <MockRunStatusTab nodeId={nodeId} runs={runs} />
240
+ )}
241
+ </div>
242
+ </div>
243
+ );
244
+ };
245
+
246
+ describe('MessageHistoryTab', () => {
247
+ beforeEach(() => {
248
+ useUIStore.setState({
249
+ mode: 'agent',
250
+ selectedNodeIds: new Set(),
251
+ detailWindows: new Map(),
252
+ defaultTab: 'liveOutput',
253
+ layoutDirection: 'TB',
254
+ autoLayoutEnabled: true,
255
+ });
256
+ });
257
+
258
+ describe('Display', () => {
259
+ it('should render empty state when no messages', () => {
260
+ render(<MockMessageHistoryTab nodeId="agent-1" messages={[]} />);
261
+
262
+ expect(screen.getByTestId('empty-messages')).toBeInTheDocument();
263
+ expect(screen.getByText('No messages yet')).toBeInTheDocument();
264
+ });
265
+
266
+ it('should display messages in table format', () => {
267
+ const messages: MessageHistoryEntry[] = [
268
+ {
269
+ id: 'msg-1',
270
+ type: 'Movie',
271
+ direction: 'consumed',
272
+ payload: { title: 'Inception' },
273
+ timestamp: Date.now(),
274
+ correlationId: 'corr-1',
275
+ },
276
+ ];
277
+
278
+ render(<MockMessageHistoryTab nodeId="agent-1" messages={messages} />);
279
+
280
+ expect(screen.getByTestId('message-table')).toBeInTheDocument();
281
+ expect(screen.getByTestId('message-row-msg-1')).toBeInTheDocument();
282
+ });
283
+
284
+ it('should display consumed messages with correct styling', () => {
285
+ const messages: MessageHistoryEntry[] = [
286
+ {
287
+ id: 'msg-1',
288
+ type: 'Input',
289
+ direction: 'consumed',
290
+ payload: { data: 'test' },
291
+ timestamp: Date.now(),
292
+ correlationId: 'corr-1',
293
+ },
294
+ ];
295
+
296
+ render(<MockMessageHistoryTab nodeId="agent-1" messages={messages} />);
297
+
298
+ const direction = screen.getByTestId('msg-direction-msg-1');
299
+ expect(direction).toHaveTextContent('↓ Consumed');
300
+ expect(direction).toHaveStyle({ color: '#8be9fd' });
301
+ });
302
+
303
+ it('should display published messages with correct styling', () => {
304
+ const messages: MessageHistoryEntry[] = [
305
+ {
306
+ id: 'msg-1',
307
+ type: 'Output',
308
+ direction: 'published',
309
+ payload: { result: 'success' },
310
+ timestamp: Date.now(),
311
+ correlationId: 'corr-1',
312
+ },
313
+ ];
314
+
315
+ render(<MockMessageHistoryTab nodeId="agent-1" messages={messages} />);
316
+
317
+ const direction = screen.getByTestId('msg-direction-msg-1');
318
+ expect(direction).toHaveTextContent('↑ Published');
319
+ expect(direction).toHaveStyle({ color: '#50fa7b' });
320
+ });
321
+
322
+ it('should display message type', () => {
323
+ const messages: MessageHistoryEntry[] = [
324
+ {
325
+ id: 'msg-1',
326
+ type: 'Movie',
327
+ direction: 'consumed',
328
+ payload: {},
329
+ timestamp: Date.now(),
330
+ correlationId: 'corr-1',
331
+ },
332
+ ];
333
+
334
+ render(<MockMessageHistoryTab nodeId="agent-1" messages={messages} />);
335
+
336
+ expect(screen.getByTestId('msg-type-msg-1')).toHaveTextContent('Movie');
337
+ });
338
+
339
+ it('should display correlation ID', () => {
340
+ const messages: MessageHistoryEntry[] = [
341
+ {
342
+ id: 'msg-1',
343
+ type: 'Test',
344
+ direction: 'consumed',
345
+ payload: {},
346
+ timestamp: Date.now(),
347
+ correlationId: 'corr-123-abc',
348
+ },
349
+ ];
350
+
351
+ render(<MockMessageHistoryTab nodeId="agent-1" messages={messages} />);
352
+
353
+ expect(screen.getByTestId('msg-correlation-msg-1')).toHaveTextContent('corr-123-abc');
354
+ });
355
+
356
+ it('should format and display payload as JSON', () => {
357
+ const messages: MessageHistoryEntry[] = [
358
+ {
359
+ id: 'msg-1',
360
+ type: 'Movie',
361
+ direction: 'consumed',
362
+ payload: { title: 'Inception', year: 2010 },
363
+ timestamp: Date.now(),
364
+ correlationId: 'corr-1',
365
+ },
366
+ ];
367
+
368
+ render(<MockMessageHistoryTab nodeId="agent-1" messages={messages} />);
369
+
370
+ const payload = screen.getByTestId('msg-payload-msg-1');
371
+ expect(payload.textContent).toContain('title');
372
+ expect(payload.textContent).toContain('Inception');
373
+ expect(payload.textContent).toContain('2010');
374
+ });
375
+
376
+ it('should display timestamp in readable format', () => {
377
+ const timestamp = Date.now();
378
+ const messages: MessageHistoryEntry[] = [
379
+ {
380
+ id: 'msg-1',
381
+ type: 'Test',
382
+ direction: 'consumed',
383
+ payload: {},
384
+ timestamp,
385
+ correlationId: 'corr-1',
386
+ },
387
+ ];
388
+
389
+ render(<MockMessageHistoryTab nodeId="agent-1" messages={messages} />);
390
+
391
+ const timeElement = screen.getByTestId('msg-time-msg-1');
392
+ expect(timeElement.textContent).toBeTruthy();
393
+ expect(timeElement.textContent).not.toBe(String(timestamp));
394
+ });
395
+ });
396
+
397
+ describe('Multiple Messages', () => {
398
+ it('should display multiple messages in order', () => {
399
+ const messages: MessageHistoryEntry[] = [
400
+ {
401
+ id: 'msg-1',
402
+ type: 'Input',
403
+ direction: 'consumed',
404
+ payload: { data: 1 },
405
+ timestamp: Date.now(),
406
+ correlationId: 'corr-1',
407
+ },
408
+ {
409
+ id: 'msg-2',
410
+ type: 'Output',
411
+ direction: 'published',
412
+ payload: { result: 1 },
413
+ timestamp: Date.now() + 1000,
414
+ correlationId: 'corr-2',
415
+ },
416
+ {
417
+ id: 'msg-3',
418
+ type: 'Input',
419
+ direction: 'consumed',
420
+ payload: { data: 2 },
421
+ timestamp: Date.now() + 2000,
422
+ correlationId: 'corr-3',
423
+ },
424
+ ];
425
+
426
+ render(<MockMessageHistoryTab nodeId="agent-1" messages={messages} />);
427
+
428
+ expect(screen.getByTestId('message-row-msg-1')).toBeInTheDocument();
429
+ expect(screen.getByTestId('message-row-msg-2')).toBeInTheDocument();
430
+ expect(screen.getByTestId('message-row-msg-3')).toBeInTheDocument();
431
+ });
432
+
433
+ it('should handle mixed consumed and published messages', () => {
434
+ const messages: MessageHistoryEntry[] = [
435
+ {
436
+ id: 'msg-1',
437
+ type: 'Movie',
438
+ direction: 'consumed',
439
+ payload: {},
440
+ timestamp: Date.now(),
441
+ correlationId: 'corr-1',
442
+ },
443
+ {
444
+ id: 'msg-2',
445
+ type: 'Tagline',
446
+ direction: 'published',
447
+ payload: {},
448
+ timestamp: Date.now() + 1000,
449
+ correlationId: 'corr-2',
450
+ },
451
+ ];
452
+
453
+ render(<MockMessageHistoryTab nodeId="agent-1" messages={messages} />);
454
+
455
+ expect(screen.getByTestId('msg-direction-msg-1')).toHaveTextContent('↓ Consumed');
456
+ expect(screen.getByTestId('msg-direction-msg-2')).toHaveTextContent('↑ Published');
457
+ });
458
+ });
459
+
460
+ describe('Edge Cases', () => {
461
+ it('should handle complex nested payload', () => {
462
+ const messages: MessageHistoryEntry[] = [
463
+ {
464
+ id: 'msg-1',
465
+ type: 'Complex',
466
+ direction: 'consumed',
467
+ payload: {
468
+ nested: {
469
+ deeply: {
470
+ value: 'test',
471
+ array: [1, 2, 3],
472
+ },
473
+ },
474
+ },
475
+ timestamp: Date.now(),
476
+ correlationId: 'corr-1',
477
+ },
478
+ ];
479
+
480
+ render(<MockMessageHistoryTab nodeId="agent-1" messages={messages} />);
481
+
482
+ const payload = screen.getByTestId('msg-payload-msg-1');
483
+ expect(payload.textContent).toContain('nested');
484
+ expect(payload.textContent).toContain('deeply');
485
+ });
486
+
487
+ it('should handle null payload', () => {
488
+ const messages: MessageHistoryEntry[] = [
489
+ {
490
+ id: 'msg-1',
491
+ type: 'Test',
492
+ direction: 'consumed',
493
+ payload: null,
494
+ timestamp: Date.now(),
495
+ correlationId: 'corr-1',
496
+ },
497
+ ];
498
+
499
+ expect(() => {
500
+ render(<MockMessageHistoryTab nodeId="agent-1" messages={messages} />);
501
+ }).not.toThrow();
502
+ });
503
+
504
+ it('should handle large payloads with scrolling', () => {
505
+ const largePayload = {
506
+ data: Array.from({ length: 100 }, (_, i) => ({ id: i, value: `Item ${i}` })),
507
+ };
508
+
509
+ const messages: MessageHistoryEntry[] = [
510
+ {
511
+ id: 'msg-1',
512
+ type: 'Large',
513
+ direction: 'consumed',
514
+ payload: largePayload,
515
+ timestamp: Date.now(),
516
+ correlationId: 'corr-1',
517
+ },
518
+ ];
519
+
520
+ render(<MockMessageHistoryTab nodeId="agent-1" messages={messages} />);
521
+
522
+ const payload = screen.getByTestId('msg-payload-msg-1');
523
+ const pre = payload.querySelector('pre');
524
+ expect(pre).toHaveStyle({ maxHeight: '100px', overflow: 'auto' });
525
+ });
526
+ });
527
+ });
528
+
529
+ describe('RunStatusTab', () => {
530
+ beforeEach(() => {
531
+ useUIStore.setState({
532
+ mode: 'agent',
533
+ selectedNodeIds: new Set(),
534
+ detailWindows: new Map(),
535
+ defaultTab: 'liveOutput',
536
+ layoutDirection: 'TB',
537
+ autoLayoutEnabled: true,
538
+ });
539
+ });
540
+
541
+ describe('Display', () => {
542
+ it('should render empty state when no runs', () => {
543
+ render(<MockRunStatusTab nodeId="agent-1" runs={[]} />);
544
+
545
+ expect(screen.getByTestId('empty-runs')).toBeInTheDocument();
546
+ expect(screen.getByText('No runs yet')).toBeInTheDocument();
547
+ });
548
+
549
+ it('should display runs in table format', () => {
550
+ const runs: RunStatusEntry[] = [
551
+ {
552
+ runId: 'run-1',
553
+ startTime: Date.now(),
554
+ endTime: Date.now() + 1000,
555
+ duration: 1000,
556
+ status: 'completed',
557
+ metrics: { tokensUsed: 100, costUsd: 0.01 },
558
+ },
559
+ ];
560
+
561
+ render(<MockRunStatusTab nodeId="agent-1" runs={runs} />);
562
+
563
+ expect(screen.getByTestId('run-table')).toBeInTheDocument();
564
+ expect(screen.getByTestId('run-row-run-1')).toBeInTheDocument();
565
+ });
566
+
567
+ it('should display run ID', () => {
568
+ const runs: RunStatusEntry[] = [
569
+ {
570
+ runId: 'run-abc-123',
571
+ startTime: Date.now(),
572
+ endTime: Date.now() + 1000,
573
+ duration: 1000,
574
+ status: 'completed',
575
+ metrics: {},
576
+ },
577
+ ];
578
+
579
+ render(<MockRunStatusTab nodeId="agent-1" runs={runs} />);
580
+
581
+ expect(screen.getByTestId('run-id-run-abc-123')).toHaveTextContent('run-abc-123');
582
+ });
583
+
584
+ it('should display start time in readable format', () => {
585
+ const startTime = Date.now();
586
+ const runs: RunStatusEntry[] = [
587
+ {
588
+ runId: 'run-1',
589
+ startTime,
590
+ endTime: startTime + 1000,
591
+ duration: 1000,
592
+ status: 'completed',
593
+ metrics: {},
594
+ },
595
+ ];
596
+
597
+ render(<MockRunStatusTab nodeId="agent-1" runs={runs} />);
598
+
599
+ const startElement = screen.getByTestId('run-start-run-1');
600
+ expect(startElement.textContent).toBeTruthy();
601
+ expect(startElement.textContent).not.toBe(String(startTime));
602
+ });
603
+
604
+ it('should format duration in milliseconds', () => {
605
+ const runs: RunStatusEntry[] = [
606
+ {
607
+ runId: 'run-1',
608
+ startTime: Date.now(),
609
+ endTime: Date.now() + 500,
610
+ duration: 500,
611
+ status: 'completed',
612
+ metrics: {},
613
+ },
614
+ ];
615
+
616
+ render(<MockRunStatusTab nodeId="agent-1" runs={runs} />);
617
+
618
+ expect(screen.getByTestId('run-duration-run-1')).toHaveTextContent('500ms');
619
+ });
620
+
621
+ it('should format duration in seconds', () => {
622
+ const runs: RunStatusEntry[] = [
623
+ {
624
+ runId: 'run-1',
625
+ startTime: Date.now(),
626
+ endTime: Date.now() + 2500,
627
+ duration: 2500,
628
+ status: 'completed',
629
+ metrics: {},
630
+ },
631
+ ];
632
+
633
+ render(<MockRunStatusTab nodeId="agent-1" runs={runs} />);
634
+
635
+ expect(screen.getByTestId('run-duration-run-1')).toHaveTextContent('2.50s');
636
+ });
637
+
638
+ it('should display completed status with green color', () => {
639
+ const runs: RunStatusEntry[] = [
640
+ {
641
+ runId: 'run-1',
642
+ startTime: Date.now(),
643
+ endTime: Date.now() + 1000,
644
+ duration: 1000,
645
+ status: 'completed',
646
+ metrics: {},
647
+ },
648
+ ];
649
+
650
+ render(<MockRunStatusTab nodeId="agent-1" runs={runs} />);
651
+
652
+ const status = screen.getByTestId('run-status-run-1');
653
+ expect(status).toHaveTextContent('completed');
654
+ expect(status).toHaveStyle({ color: '#50fa7b' });
655
+ });
656
+
657
+ it('should display error status with red color', () => {
658
+ const runs: RunStatusEntry[] = [
659
+ {
660
+ runId: 'run-1',
661
+ startTime: Date.now(),
662
+ endTime: Date.now() + 1000,
663
+ duration: 1000,
664
+ status: 'error',
665
+ metrics: {},
666
+ errorMessage: 'Test error',
667
+ },
668
+ ];
669
+
670
+ render(<MockRunStatusTab nodeId="agent-1" runs={runs} />);
671
+
672
+ const status = screen.getByTestId('run-status-run-1');
673
+ expect(status).toHaveTextContent('error');
674
+ expect(status).toHaveStyle({ color: '#ff5555' });
675
+ });
676
+
677
+ it('should display running status with yellow color', () => {
678
+ const runs: RunStatusEntry[] = [
679
+ {
680
+ runId: 'run-1',
681
+ startTime: Date.now(),
682
+ endTime: Date.now() + 1000,
683
+ duration: 1000,
684
+ status: 'running',
685
+ metrics: {},
686
+ },
687
+ ];
688
+
689
+ render(<MockRunStatusTab nodeId="agent-1" runs={runs} />);
690
+
691
+ const status = screen.getByTestId('run-status-run-1');
692
+ expect(status).toHaveTextContent('running');
693
+ expect(status).toHaveStyle({ color: '#f1fa8c' });
694
+ });
695
+ });
696
+
697
+ describe('Metrics', () => {
698
+ it('should display tokens used', () => {
699
+ const runs: RunStatusEntry[] = [
700
+ {
701
+ runId: 'run-1',
702
+ startTime: Date.now(),
703
+ endTime: Date.now() + 1000,
704
+ duration: 1000,
705
+ status: 'completed',
706
+ metrics: { tokensUsed: 500 },
707
+ },
708
+ ];
709
+
710
+ render(<MockRunStatusTab nodeId="agent-1" runs={runs} />);
711
+
712
+ expect(screen.getByTestId('run-tokens-run-1')).toHaveTextContent('500');
713
+ });
714
+
715
+ it('should display cost in USD', () => {
716
+ const runs: RunStatusEntry[] = [
717
+ {
718
+ runId: 'run-1',
719
+ startTime: Date.now(),
720
+ endTime: Date.now() + 1000,
721
+ duration: 1000,
722
+ status: 'completed',
723
+ metrics: { costUsd: 0.0123 },
724
+ },
725
+ ];
726
+
727
+ render(<MockRunStatusTab nodeId="agent-1" runs={runs} />);
728
+
729
+ expect(screen.getByTestId('run-cost-run-1')).toHaveTextContent('$0.0123');
730
+ });
731
+
732
+ it('should display artifacts produced', () => {
733
+ const runs: RunStatusEntry[] = [
734
+ {
735
+ runId: 'run-1',
736
+ startTime: Date.now(),
737
+ endTime: Date.now() + 1000,
738
+ duration: 1000,
739
+ status: 'completed',
740
+ metrics: { artifactsProduced: 3 },
741
+ },
742
+ ];
743
+
744
+ render(<MockRunStatusTab nodeId="agent-1" runs={runs} />);
745
+
746
+ expect(screen.getByTestId('run-artifacts-run-1')).toHaveTextContent('3');
747
+ });
748
+
749
+ it('should show dash for missing metrics', () => {
750
+ const runs: RunStatusEntry[] = [
751
+ {
752
+ runId: 'run-1',
753
+ startTime: Date.now(),
754
+ endTime: Date.now() + 1000,
755
+ duration: 1000,
756
+ status: 'completed',
757
+ metrics: {},
758
+ },
759
+ ];
760
+
761
+ render(<MockRunStatusTab nodeId="agent-1" runs={runs} />);
762
+
763
+ expect(screen.getByTestId('run-tokens-run-1')).toHaveTextContent('-');
764
+ expect(screen.getByTestId('run-cost-run-1')).toHaveTextContent('-');
765
+ expect(screen.getByTestId('run-artifacts-run-1')).toHaveTextContent('-');
766
+ });
767
+
768
+ it('should display all metrics together', () => {
769
+ const runs: RunStatusEntry[] = [
770
+ {
771
+ runId: 'run-1',
772
+ startTime: Date.now(),
773
+ endTime: Date.now() + 1000,
774
+ duration: 1000,
775
+ status: 'completed',
776
+ metrics: {
777
+ tokensUsed: 750,
778
+ costUsd: 0.025,
779
+ artifactsProduced: 5,
780
+ },
781
+ },
782
+ ];
783
+
784
+ render(<MockRunStatusTab nodeId="agent-1" runs={runs} />);
785
+
786
+ expect(screen.getByTestId('run-tokens-run-1')).toHaveTextContent('750');
787
+ expect(screen.getByTestId('run-cost-run-1')).toHaveTextContent('$0.0250');
788
+ expect(screen.getByTestId('run-artifacts-run-1')).toHaveTextContent('5');
789
+ });
790
+ });
791
+
792
+ describe('Multiple Runs', () => {
793
+ it('should display multiple runs in order', () => {
794
+ const runs: RunStatusEntry[] = [
795
+ {
796
+ runId: 'run-1',
797
+ startTime: Date.now(),
798
+ endTime: Date.now() + 1000,
799
+ duration: 1000,
800
+ status: 'completed',
801
+ metrics: {},
802
+ },
803
+ {
804
+ runId: 'run-2',
805
+ startTime: Date.now() + 2000,
806
+ endTime: Date.now() + 3000,
807
+ duration: 1000,
808
+ status: 'completed',
809
+ metrics: {},
810
+ },
811
+ {
812
+ runId: 'run-3',
813
+ startTime: Date.now() + 4000,
814
+ endTime: Date.now() + 5000,
815
+ duration: 1000,
816
+ status: 'running',
817
+ metrics: {},
818
+ },
819
+ ];
820
+
821
+ render(<MockRunStatusTab nodeId="agent-1" runs={runs} />);
822
+
823
+ expect(screen.getByTestId('run-row-run-1')).toBeInTheDocument();
824
+ expect(screen.getByTestId('run-row-run-2')).toBeInTheDocument();
825
+ expect(screen.getByTestId('run-row-run-3')).toBeInTheDocument();
826
+ });
827
+
828
+ it('should show runs with different statuses', () => {
829
+ const runs: RunStatusEntry[] = [
830
+ {
831
+ runId: 'run-1',
832
+ startTime: Date.now(),
833
+ endTime: Date.now() + 1000,
834
+ duration: 1000,
835
+ status: 'completed',
836
+ metrics: {},
837
+ },
838
+ {
839
+ runId: 'run-2',
840
+ startTime: Date.now() + 2000,
841
+ endTime: Date.now() + 3000,
842
+ duration: 1000,
843
+ status: 'error',
844
+ metrics: {},
845
+ },
846
+ ];
847
+
848
+ render(<MockRunStatusTab nodeId="agent-1" runs={runs} />);
849
+
850
+ expect(screen.getByTestId('run-status-run-1')).toHaveTextContent('completed');
851
+ expect(screen.getByTestId('run-status-run-2')).toHaveTextContent('error');
852
+ });
853
+ });
854
+ });
855
+
856
+ describe('Tab Switching', () => {
857
+ beforeEach(() => {
858
+ useUIStore.setState({
859
+ mode: 'agent',
860
+ selectedNodeIds: new Set(),
861
+ detailWindows: new Map(),
862
+ defaultTab: 'liveOutput',
863
+ layoutDirection: 'TB',
864
+ autoLayoutEnabled: true,
865
+ });
866
+ });
867
+
868
+ it('should switch between tabs', () => {
869
+ useUIStore.getState().openDetailWindow('agent-1');
870
+
871
+ const { rerender } = render(
872
+ <MockTabContainer nodeId="agent-1" messages={[]} runs={[]} />
873
+ );
874
+
875
+ // Initially on liveOutput
876
+ expect(screen.getByTestId('tab-live-output')).toHaveAttribute('data-active', 'true');
877
+ expect(screen.getByTestId('live-output-content')).toBeInTheDocument();
878
+
879
+ // Switch to message history
880
+ fireEvent.click(screen.getByTestId('tab-message-history'));
881
+ rerender(<MockTabContainer nodeId="agent-1" messages={[]} runs={[]} />);
882
+
883
+ expect(screen.getByTestId('tab-message-history')).toHaveAttribute('data-active', 'true');
884
+ expect(screen.getByTestId('message-history-agent-1')).toBeInTheDocument();
885
+
886
+ // Switch to run status
887
+ fireEvent.click(screen.getByTestId('tab-run-status'));
888
+ rerender(<MockTabContainer nodeId="agent-1" messages={[]} runs={[]} />);
889
+
890
+ expect(screen.getByTestId('tab-run-status')).toHaveAttribute('data-active', 'true');
891
+ expect(screen.getByTestId('run-status-agent-1')).toBeInTheDocument();
892
+ });
893
+
894
+ it('should highlight active tab', () => {
895
+ useUIStore.getState().openDetailWindow('agent-1');
896
+ render(<MockTabContainer nodeId="agent-1" messages={[]} runs={[]} />);
897
+
898
+ const liveOutputButton = screen.getByTestId('tab-live-output');
899
+ const messageHistoryButton = screen.getByTestId('tab-message-history');
900
+
901
+ expect(liveOutputButton).toHaveStyle({ fontWeight: 'bold' });
902
+ expect(messageHistoryButton).toHaveStyle({ fontWeight: 'normal' });
903
+
904
+ // Switch tab
905
+ fireEvent.click(messageHistoryButton);
906
+
907
+ expect(screen.getByTestId('tab-message-history')).toHaveStyle({ fontWeight: 'bold' });
908
+ });
909
+
910
+ it('should persist active tab in store', () => {
911
+ useUIStore.getState().openDetailWindow('agent-1');
912
+ render(<MockTabContainer nodeId="agent-1" messages={[]} runs={[]} />);
913
+
914
+ // Switch to message history
915
+ fireEvent.click(screen.getByTestId('tab-message-history'));
916
+
917
+ const window = useUIStore.getState().detailWindows.get('agent-1');
918
+ expect(window?.activeTab).toBe('messageHistory');
919
+ });
920
+
921
+ it('should maintain tab state across window updates', () => {
922
+ useUIStore.getState().openDetailWindow('agent-1');
923
+
924
+ // Set tab to messageHistory
925
+ useUIStore.getState().updateDetailWindow('agent-1', {
926
+ activeTab: 'messageHistory',
927
+ });
928
+
929
+ render(<MockTabContainer nodeId="agent-1" messages={[]} runs={[]} />);
930
+
931
+ expect(screen.getByTestId('tab-message-history')).toHaveAttribute('data-active', 'true');
932
+ expect(screen.getByTestId('message-history-agent-1')).toBeInTheDocument();
933
+
934
+ // Update window position (shouldn't affect tab)
935
+ useUIStore.getState().updateDetailWindow('agent-1', {
936
+ position: { x: 200, y: 200 },
937
+ });
938
+
939
+ const window = useUIStore.getState().detailWindows.get('agent-1');
940
+ expect(window?.activeTab).toBe('messageHistory');
941
+ });
942
+ });
943
+
944
+ describe('Default Tab Preference', () => {
945
+ beforeEach(() => {
946
+ useUIStore.setState({
947
+ mode: 'agent',
948
+ selectedNodeIds: new Set(),
949
+ detailWindows: new Map(),
950
+ defaultTab: 'liveOutput',
951
+ layoutDirection: 'TB',
952
+ autoLayoutEnabled: true,
953
+ });
954
+ });
955
+
956
+ it('should use liveOutput as default tab', () => {
957
+ useUIStore.setState({ defaultTab: 'liveOutput' });
958
+ useUIStore.getState().openDetailWindow('agent-1');
959
+
960
+ const window = useUIStore.getState().detailWindows.get('agent-1');
961
+ expect(window?.activeTab).toBe('liveOutput');
962
+ });
963
+
964
+ it('should use messageHistory as default tab when configured', () => {
965
+ useUIStore.setState({ defaultTab: 'messageHistory' });
966
+ useUIStore.getState().openDetailWindow('agent-1');
967
+
968
+ const window = useUIStore.getState().detailWindows.get('agent-1');
969
+ expect(window?.activeTab).toBe('messageHistory');
970
+ });
971
+
972
+ it('should use runStatus as default tab when configured', () => {
973
+ useUIStore.setState({ defaultTab: 'runStatus' });
974
+ useUIStore.getState().openDetailWindow('agent-1');
975
+
976
+ const window = useUIStore.getState().detailWindows.get('agent-1');
977
+ expect(window?.activeTab).toBe('runStatus');
978
+ });
979
+
980
+ it('should apply default tab to newly opened windows', () => {
981
+ useUIStore.setState({ defaultTab: 'messageHistory' });
982
+
983
+ useUIStore.getState().openDetailWindow('agent-1');
984
+ useUIStore.getState().openDetailWindow('agent-2');
985
+ useUIStore.getState().openDetailWindow('agent-3');
986
+
987
+ expect(
988
+ useUIStore.getState().detailWindows.get('agent-1')?.activeTab
989
+ ).toBe('messageHistory');
990
+ expect(
991
+ useUIStore.getState().detailWindows.get('agent-2')?.activeTab
992
+ ).toBe('messageHistory');
993
+ expect(
994
+ useUIStore.getState().detailWindows.get('agent-3')?.activeTab
995
+ ).toBe('messageHistory');
996
+ });
997
+
998
+ it('should respect default tab preference change', () => {
999
+ useUIStore.setState({ defaultTab: 'liveOutput' });
1000
+ useUIStore.getState().openDetailWindow('agent-1');
1001
+
1002
+ expect(
1003
+ useUIStore.getState().detailWindows.get('agent-1')?.activeTab
1004
+ ).toBe('liveOutput');
1005
+
1006
+ // Close and reopen with different preference
1007
+ useUIStore.getState().closeDetailWindow('agent-1');
1008
+ useUIStore.setState({ defaultTab: 'runStatus' });
1009
+ useUIStore.getState().openDetailWindow('agent-1');
1010
+
1011
+ expect(
1012
+ useUIStore.getState().detailWindows.get('agent-1')?.activeTab
1013
+ ).toBe('runStatus');
1014
+ });
1015
+ });