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,148 @@
1
+ import { useEffect } from 'react';
2
+ import { useUIStore } from '../store/uiStore';
3
+ import { useSettingsStore } from '../store/settingsStore';
4
+
5
+ interface KeyboardShortcutConfig {
6
+ onToggleMode?: () => void;
7
+ onFocusFilter?: () => void;
8
+ onCloseWindows?: () => void;
9
+ onToggleHelp?: () => void;
10
+ onToggleAgentDetails?: () => void;
11
+ }
12
+
13
+ /**
14
+ * Custom hook for managing global keyboard shortcuts
15
+ *
16
+ * Shortcuts:
17
+ * - Ctrl+M (or Cmd+M): Toggle between Agent View and Blackboard View
18
+ * - Ctrl+F (or Cmd+F): Focus the filter input
19
+ * - Ctrl+, (or Cmd+,): Toggle Settings Panel
20
+ * - Ctrl+Shift+F: Toggle Filters Panel
21
+ * - Ctrl+Shift+P: Toggle Publish Panel
22
+ * - Ctrl+Shift+D: Toggle Agent Details
23
+ * - Ctrl+/ (or Cmd+/): Toggle Keyboard Shortcuts Help
24
+ * - Esc: Close all panels and detail windows
25
+ */
26
+ export const useKeyboardShortcuts = (config: KeyboardShortcutConfig = {}) => {
27
+ useEffect(() => {
28
+ const handleKeyDown = (event: KeyboardEvent) => {
29
+ // Check for modifier key (Ctrl on Windows/Linux, Cmd on Mac)
30
+ const modifier = event.ctrlKey || event.metaKey;
31
+
32
+ // Ctrl+, or Cmd+,: Toggle Settings Panel
33
+ if (modifier && event.key === ',') {
34
+ event.preventDefault();
35
+ const showSettings = useSettingsStore.getState().ui.showSettings;
36
+ useSettingsStore.getState().setShowSettings(!showSettings);
37
+ return;
38
+ }
39
+
40
+ // Ctrl+Shift+F: Toggle Filters Panel
41
+ if (modifier && event.shiftKey && event.key === 'F') {
42
+ event.preventDefault();
43
+ const showFilters = useSettingsStore.getState().ui.showFilters;
44
+ useSettingsStore.getState().setShowFilters(!showFilters);
45
+ return;
46
+ }
47
+
48
+ // Ctrl+Shift+P: Toggle Publish/Controls Panel
49
+ if (modifier && event.shiftKey && event.key === 'P') {
50
+ event.preventDefault();
51
+ const showControls = useSettingsStore.getState().ui.showControls;
52
+ useSettingsStore.getState().setShowControls(!showControls);
53
+ return;
54
+ }
55
+
56
+ // Ctrl+Shift+D: Toggle Agent Details
57
+ if (modifier && event.shiftKey && event.key === 'D') {
58
+ event.preventDefault();
59
+ if (config.onToggleAgentDetails) {
60
+ config.onToggleAgentDetails();
61
+ }
62
+ return;
63
+ }
64
+
65
+ // Ctrl+/ or Cmd+/: Toggle Keyboard Shortcuts Help
66
+ if (modifier && event.key === '/') {
67
+ event.preventDefault();
68
+ if (config.onToggleHelp) {
69
+ config.onToggleHelp();
70
+ }
71
+ return;
72
+ }
73
+
74
+ // Ctrl+M or Cmd+M: Toggle mode
75
+ if (modifier && event.key === 'm') {
76
+ event.preventDefault();
77
+ if (config.onToggleMode) {
78
+ config.onToggleMode();
79
+ } else {
80
+ // Default behavior: toggle mode
81
+ const currentMode = useUIStore.getState().mode;
82
+ useUIStore.getState().setMode(currentMode === 'agent' ? 'blackboard' : 'agent');
83
+ }
84
+ return;
85
+ }
86
+
87
+ // Ctrl+F or Cmd+F: Focus filter (only if not in an input already)
88
+ if (modifier && event.key === 'f') {
89
+ const activeElement = document.activeElement;
90
+ const isInInput = activeElement?.tagName === 'INPUT' ||
91
+ activeElement?.tagName === 'TEXTAREA';
92
+
93
+ if (!isInInput) {
94
+ event.preventDefault();
95
+ if (config.onFocusFilter) {
96
+ config.onFocusFilter();
97
+ } else {
98
+ // Default behavior: focus correlation ID filter
99
+ const filterInput = document.querySelector<HTMLInputElement>(
100
+ 'input[placeholder*="correlation"], input[placeholder*="Correlation"]'
101
+ );
102
+ if (filterInput) {
103
+ filterInput.focus();
104
+ filterInput.select();
105
+ }
106
+ }
107
+ }
108
+ return;
109
+ }
110
+
111
+ // Esc: Close windows/dialogs/panels
112
+ if (event.key === 'Escape') {
113
+ if (config.onCloseWindows) {
114
+ config.onCloseWindows();
115
+ } else {
116
+ // Priority: Close panels first, then detail windows
117
+ const { showSettings, showFilters, showControls } = useSettingsStore.getState().ui;
118
+ const detailWindows = useUIStore.getState().detailWindows;
119
+
120
+ if (showSettings) {
121
+ event.preventDefault();
122
+ useSettingsStore.getState().setShowSettings(false);
123
+ } else if (showControls) {
124
+ event.preventDefault();
125
+ useSettingsStore.getState().setShowControls(false);
126
+ } else if (showFilters) {
127
+ event.preventDefault();
128
+ useSettingsStore.getState().setShowFilters(false);
129
+ } else if (detailWindows.size > 0) {
130
+ event.preventDefault();
131
+ // Close all detail windows
132
+ detailWindows.clear();
133
+ useUIStore.setState({ detailWindows: new Map() });
134
+ }
135
+ }
136
+ return;
137
+ }
138
+ };
139
+
140
+ // Add event listener
141
+ window.addEventListener('keydown', handleKeyDown);
142
+
143
+ // Cleanup
144
+ return () => {
145
+ window.removeEventListener('keydown', handleKeyDown);
146
+ };
147
+ }, [config]);
148
+ };
@@ -0,0 +1,442 @@
1
+ import { describe, it, expect, beforeEach, vi, afterEach } from 'vitest';
2
+ import { renderHook, waitFor } from '@testing-library/react';
3
+ import { useModulePersistence } from './useModulePersistence';
4
+ import { useModuleStore } from '../store/moduleStore';
5
+ import { indexedDBService } from '../services/indexeddb';
6
+ import type { ModuleInstance } from '../types/modules';
7
+
8
+ // Mock IndexedDB service
9
+ vi.mock('../services/indexeddb', () => ({
10
+ indexedDBService: {
11
+ getAllModuleInstances: vi.fn(),
12
+ saveModuleInstance: vi.fn(),
13
+ deleteModuleInstance: vi.fn(),
14
+ },
15
+ }));
16
+
17
+ describe('useModulePersistence', () => {
18
+ beforeEach(() => {
19
+ vi.clearAllMocks();
20
+ // Reset module store
21
+ useModuleStore.setState({ instances: new Map() });
22
+ });
23
+
24
+ afterEach(() => {
25
+ vi.useRealTimers();
26
+ });
27
+
28
+ describe('Module Loading', () => {
29
+ it('should load saved module instances on mount', async () => {
30
+ const savedInstances = [
31
+ {
32
+ instance_id: 'module-1',
33
+ type: 'eventLog',
34
+ position: { x: 100, y: 200 },
35
+ size: { width: 600, height: 400 },
36
+ visible: true,
37
+ created_at: '2025-10-03T00:00:00Z',
38
+ updated_at: '2025-10-03T00:00:00Z',
39
+ },
40
+ {
41
+ instance_id: 'module-2',
42
+ type: 'eventLog',
43
+ position: { x: 300, y: 400 },
44
+ size: { width: 800, height: 600 },
45
+ visible: true,
46
+ created_at: '2025-10-03T00:00:00Z',
47
+ updated_at: '2025-10-03T00:00:00Z',
48
+ },
49
+ ];
50
+
51
+ vi.mocked(indexedDBService.getAllModuleInstances).mockResolvedValue(savedInstances);
52
+
53
+ renderHook(() => useModulePersistence());
54
+
55
+ await waitFor(() => {
56
+ const instances = useModuleStore.getState().instances;
57
+ expect(instances.size).toBe(2);
58
+ expect(instances.get('module-1')).toEqual({
59
+ id: 'module-1',
60
+ type: 'eventLog',
61
+ position: { x: 100, y: 200 },
62
+ size: { width: 600, height: 400 },
63
+ visible: true,
64
+ });
65
+ expect(instances.get('module-2')).toEqual({
66
+ id: 'module-2',
67
+ type: 'eventLog',
68
+ position: { x: 300, y: 400 },
69
+ size: { width: 800, height: 600 },
70
+ visible: true,
71
+ });
72
+ });
73
+ });
74
+
75
+ it('should handle empty module instances on mount', async () => {
76
+ vi.mocked(indexedDBService.getAllModuleInstances).mockResolvedValue([]);
77
+
78
+ renderHook(() => useModulePersistence());
79
+
80
+ await waitFor(() => {
81
+ expect(indexedDBService.getAllModuleInstances).toHaveBeenCalledOnce();
82
+ });
83
+
84
+ const instances = useModuleStore.getState().instances;
85
+ expect(instances.size).toBe(0);
86
+ });
87
+
88
+ it('should handle errors when loading instances', async () => {
89
+ const consoleErrorSpy = vi.spyOn(console, 'error').mockImplementation(() => {});
90
+ vi.mocked(indexedDBService.getAllModuleInstances).mockRejectedValue(
91
+ new Error('Database error')
92
+ );
93
+
94
+ renderHook(() => useModulePersistence());
95
+
96
+ await waitFor(() => {
97
+ expect(consoleErrorSpy).toHaveBeenCalledWith(
98
+ '[ModulePersistence] Failed to load module instances:',
99
+ expect.any(Error)
100
+ );
101
+ });
102
+
103
+ consoleErrorSpy.mockRestore();
104
+ });
105
+ });
106
+
107
+ describe('Module Saving', () => {
108
+ it('should save new module instance when added', async () => {
109
+ vi.mocked(indexedDBService.getAllModuleInstances).mockResolvedValue([]);
110
+
111
+ renderHook(() => useModulePersistence());
112
+
113
+ await waitFor(() => {
114
+ expect(indexedDBService.getAllModuleInstances).toHaveBeenCalled();
115
+ });
116
+
117
+ const newInstance: ModuleInstance = {
118
+ id: 'new-module',
119
+ type: 'eventLog',
120
+ position: { x: 150, y: 250 },
121
+ size: { width: 700, height: 500 },
122
+ visible: true,
123
+ };
124
+
125
+ useModuleStore.getState().addModule(newInstance);
126
+
127
+ // Wait for debounced save (300ms + buffer)
128
+ await new Promise((resolve) => setTimeout(resolve, 400));
129
+
130
+ await waitFor(() => {
131
+ expect(indexedDBService.saveModuleInstance).toHaveBeenCalledWith(
132
+ expect.objectContaining({
133
+ instance_id: 'new-module',
134
+ type: 'eventLog',
135
+ position: { x: 150, y: 250 },
136
+ size: { width: 700, height: 500 },
137
+ visible: true,
138
+ })
139
+ );
140
+ });
141
+ });
142
+
143
+ it('should debounce saves when position changes rapidly', async () => {
144
+ vi.mocked(indexedDBService.getAllModuleInstances).mockResolvedValue([]);
145
+
146
+ renderHook(() => useModulePersistence());
147
+
148
+ await waitFor(() => {
149
+ expect(indexedDBService.getAllModuleInstances).toHaveBeenCalled();
150
+ });
151
+
152
+ const instance: ModuleInstance = {
153
+ id: 'drag-module',
154
+ type: 'eventLog',
155
+ position: { x: 100, y: 100 },
156
+ size: { width: 600, height: 400 },
157
+ visible: true,
158
+ };
159
+
160
+ useModuleStore.getState().addModule(instance);
161
+
162
+ // Wait for initial add to save
163
+ await new Promise((resolve) => setTimeout(resolve, 400));
164
+
165
+ // Rapidly update position (simulating drag)
166
+ useModuleStore.getState().updateModule('drag-module', { position: { x: 110, y: 110 } });
167
+ await new Promise((resolve) => setTimeout(resolve, 100));
168
+
169
+ useModuleStore.getState().updateModule('drag-module', { position: { x: 120, y: 120 } });
170
+ await new Promise((resolve) => setTimeout(resolve, 100));
171
+
172
+ useModuleStore.getState().updateModule('drag-module', { position: { x: 130, y: 130 } });
173
+
174
+ // Wait for debounced save
175
+ await new Promise((resolve) => setTimeout(resolve, 400));
176
+
177
+ await waitFor(() => {
178
+ // Should have saved at least twice: initial add + final position
179
+ expect(indexedDBService.saveModuleInstance).toHaveBeenCalled();
180
+ expect(indexedDBService.saveModuleInstance).toHaveBeenLastCalledWith(
181
+ expect.objectContaining({
182
+ position: { x: 130, y: 130 },
183
+ })
184
+ );
185
+ });
186
+ });
187
+
188
+ it('should save when size changes', async () => {
189
+ vi.mocked(indexedDBService.getAllModuleInstances).mockResolvedValue([]);
190
+
191
+ renderHook(() => useModulePersistence());
192
+
193
+ await waitFor(() => {
194
+ expect(indexedDBService.getAllModuleInstances).toHaveBeenCalled();
195
+ });
196
+
197
+ const instance: ModuleInstance = {
198
+ id: 'resize-module',
199
+ type: 'eventLog',
200
+ position: { x: 100, y: 100 },
201
+ size: { width: 600, height: 400 },
202
+ visible: true,
203
+ };
204
+
205
+ useModuleStore.getState().addModule(instance);
206
+ await new Promise((resolve) => setTimeout(resolve, 400));
207
+
208
+ // Update size
209
+ useModuleStore.getState().updateModule('resize-module', {
210
+ size: { width: 800, height: 600 },
211
+ });
212
+
213
+ await new Promise((resolve) => setTimeout(resolve, 400));
214
+
215
+ await waitFor(() => {
216
+ expect(indexedDBService.saveModuleInstance).toHaveBeenCalledWith(
217
+ expect.objectContaining({
218
+ size: { width: 800, height: 600 },
219
+ })
220
+ );
221
+ });
222
+ });
223
+
224
+ it('should save when visibility changes', async () => {
225
+ vi.mocked(indexedDBService.getAllModuleInstances).mockResolvedValue([]);
226
+
227
+ renderHook(() => useModulePersistence());
228
+
229
+ await waitFor(() => {
230
+ expect(indexedDBService.getAllModuleInstances).toHaveBeenCalled();
231
+ });
232
+
233
+ const instance: ModuleInstance = {
234
+ id: 'toggle-module',
235
+ type: 'eventLog',
236
+ position: { x: 100, y: 100 },
237
+ size: { width: 600, height: 400 },
238
+ visible: true,
239
+ };
240
+
241
+ useModuleStore.getState().addModule(instance);
242
+ await new Promise((resolve) => setTimeout(resolve, 400));
243
+
244
+ // Toggle visibility
245
+ useModuleStore.getState().toggleVisibility('toggle-module');
246
+
247
+ await new Promise((resolve) => setTimeout(resolve, 400));
248
+
249
+ await waitFor(() => {
250
+ expect(indexedDBService.saveModuleInstance).toHaveBeenCalledWith(
251
+ expect.objectContaining({
252
+ visible: false,
253
+ })
254
+ );
255
+ });
256
+ });
257
+
258
+ it('should handle save errors gracefully', async () => {
259
+ const consoleErrorSpy = vi.spyOn(console, 'error').mockImplementation(() => {});
260
+ vi.mocked(indexedDBService.getAllModuleInstances).mockResolvedValue([]);
261
+ vi.mocked(indexedDBService.saveModuleInstance).mockRejectedValue(
262
+ new Error('Save failed')
263
+ );
264
+
265
+ renderHook(() => useModulePersistence());
266
+
267
+ await waitFor(() => {
268
+ expect(indexedDBService.getAllModuleInstances).toHaveBeenCalled();
269
+ });
270
+
271
+ const instance: ModuleInstance = {
272
+ id: 'error-module',
273
+ type: 'eventLog',
274
+ position: { x: 100, y: 100 },
275
+ size: { width: 600, height: 400 },
276
+ visible: true,
277
+ };
278
+
279
+ useModuleStore.getState().addModule(instance);
280
+ await new Promise((resolve) => setTimeout(resolve, 400));
281
+
282
+ await waitFor(() => {
283
+ expect(consoleErrorSpy).toHaveBeenCalledWith(
284
+ expect.stringContaining('Failed to save instance'),
285
+ expect.any(Error)
286
+ );
287
+ });
288
+
289
+ consoleErrorSpy.mockRestore();
290
+ });
291
+ });
292
+
293
+ describe('Module Deletion', () => {
294
+ it('should delete module instance when removed', async () => {
295
+ vi.mocked(indexedDBService.getAllModuleInstances).mockResolvedValue([]);
296
+
297
+ renderHook(() => useModulePersistence());
298
+
299
+ await waitFor(() => {
300
+ expect(indexedDBService.getAllModuleInstances).toHaveBeenCalled();
301
+ });
302
+
303
+ const instance: ModuleInstance = {
304
+ id: 'delete-module',
305
+ type: 'eventLog',
306
+ position: { x: 100, y: 100 },
307
+ size: { width: 600, height: 400 },
308
+ visible: true,
309
+ };
310
+
311
+ useModuleStore.getState().addModule(instance);
312
+ await new Promise((resolve) => setTimeout(resolve, 400));
313
+
314
+ // Remove the instance
315
+ useModuleStore.getState().removeModule('delete-module');
316
+
317
+ await waitFor(() => {
318
+ expect(indexedDBService.deleteModuleInstance).toHaveBeenCalledWith('delete-module');
319
+ });
320
+ });
321
+
322
+ it('should cancel pending save when module is deleted', async () => {
323
+ vi.mocked(indexedDBService.getAllModuleInstances).mockResolvedValue([]);
324
+
325
+ renderHook(() => useModulePersistence());
326
+
327
+ await waitFor(() => {
328
+ expect(indexedDBService.getAllModuleInstances).toHaveBeenCalled();
329
+ });
330
+
331
+ const instance: ModuleInstance = {
332
+ id: 'cancel-module',
333
+ type: 'eventLog',
334
+ position: { x: 100, y: 100 },
335
+ size: { width: 600, height: 400 },
336
+ visible: true,
337
+ };
338
+
339
+ useModuleStore.getState().addModule(instance);
340
+
341
+ // Wait for initial add to save
342
+ await new Promise((resolve) => setTimeout(resolve, 400));
343
+
344
+ const initialCallCount = vi.mocked(indexedDBService.saveModuleInstance).mock.calls.length;
345
+
346
+ // Update position (triggers debounced save)
347
+ useModuleStore.getState().updateModule('cancel-module', { position: { x: 200, y: 200 } });
348
+
349
+ // Remove before debounce timer fires (100ms < 300ms debounce)
350
+ await new Promise((resolve) => setTimeout(resolve, 100));
351
+ useModuleStore.getState().removeModule('cancel-module');
352
+
353
+ // Wait for potential saves
354
+ await new Promise((resolve) => setTimeout(resolve, 400));
355
+
356
+ await waitFor(() => {
357
+ expect(indexedDBService.deleteModuleInstance).toHaveBeenCalledWith('cancel-module');
358
+ });
359
+
360
+ // save should not have been called again after the update (timer was cancelled)
361
+ const finalCallCount = vi.mocked(indexedDBService.saveModuleInstance).mock.calls.length;
362
+ expect(finalCallCount).toBe(initialCallCount);
363
+ });
364
+
365
+ it('should handle delete errors gracefully', async () => {
366
+ const consoleErrorSpy = vi.spyOn(console, 'error').mockImplementation(() => {});
367
+ vi.mocked(indexedDBService.getAllModuleInstances).mockResolvedValue([]);
368
+ vi.mocked(indexedDBService.deleteModuleInstance).mockRejectedValue(
369
+ new Error('Delete failed')
370
+ );
371
+
372
+ renderHook(() => useModulePersistence());
373
+
374
+ await waitFor(() => {
375
+ expect(indexedDBService.getAllModuleInstances).toHaveBeenCalled();
376
+ });
377
+
378
+ const instance: ModuleInstance = {
379
+ id: 'error-delete-module',
380
+ type: 'eventLog',
381
+ position: { x: 100, y: 100 },
382
+ size: { width: 600, height: 400 },
383
+ visible: true,
384
+ };
385
+
386
+ useModuleStore.getState().addModule(instance);
387
+ await new Promise((resolve) => setTimeout(resolve, 400));
388
+
389
+ useModuleStore.getState().removeModule('error-delete-module');
390
+
391
+ await waitFor(() => {
392
+ expect(consoleErrorSpy).toHaveBeenCalledWith(
393
+ expect.stringContaining('Failed to delete instance'),
394
+ expect.any(Error)
395
+ );
396
+ });
397
+
398
+ consoleErrorSpy.mockRestore();
399
+ });
400
+ });
401
+
402
+ describe('Timer Cleanup', () => {
403
+ it('should cleanup timers on unmount', async () => {
404
+ vi.mocked(indexedDBService.getAllModuleInstances).mockResolvedValue([]);
405
+
406
+ const { unmount } = renderHook(() => useModulePersistence());
407
+
408
+ await waitFor(() => {
409
+ expect(indexedDBService.getAllModuleInstances).toHaveBeenCalled();
410
+ });
411
+
412
+ const instance: ModuleInstance = {
413
+ id: 'cleanup-module',
414
+ type: 'eventLog',
415
+ position: { x: 100, y: 100 },
416
+ size: { width: 600, height: 400 },
417
+ visible: true,
418
+ };
419
+
420
+ useModuleStore.getState().addModule(instance);
421
+
422
+ // Wait for initial add to potentially save
423
+ await new Promise((resolve) => setTimeout(resolve, 400));
424
+
425
+ const callCountBeforeUpdate = vi.mocked(indexedDBService.saveModuleInstance).mock.calls.length;
426
+
427
+ // Update position (triggers debounced save)
428
+ useModuleStore.getState().updateModule('cleanup-module', { position: { x: 200, y: 200 } });
429
+
430
+ // Unmount before timer fires (100ms < 300ms debounce)
431
+ await new Promise((resolve) => setTimeout(resolve, 100));
432
+ unmount();
433
+
434
+ // Wait for potential timer - save should not be called after unmount
435
+ await new Promise((resolve) => setTimeout(resolve, 400));
436
+
437
+ // The update should NOT have saved because we unmounted before debounce completed
438
+ const finalCallCount = vi.mocked(indexedDBService.saveModuleInstance).mock.calls.length;
439
+ expect(finalCallCount).toBe(callCountBeforeUpdate);
440
+ });
441
+ });
442
+ });