@useatlas/create 0.0.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (515) hide show
  1. package/README.md +231 -0
  2. package/index.ts +829 -0
  3. package/package.json +38 -0
  4. package/templates/docker/.env.example +67 -0
  5. package/templates/docker/Dockerfile +52 -0
  6. package/templates/docker/bin/__tests__/benchmark.test.ts +598 -0
  7. package/templates/docker/bin/__tests__/duckdb-ingest.test.ts +171 -0
  8. package/templates/docker/bin/__tests__/eval.test.ts +434 -0
  9. package/templates/docker/bin/__tests__/matview-partition.test.ts +615 -0
  10. package/templates/docker/bin/__tests__/multi-source.test.ts +113 -0
  11. package/templates/docker/bin/__tests__/plugin-cli.test.ts +322 -0
  12. package/templates/docker/bin/__tests__/profiler-heuristics.test.ts +608 -0
  13. package/templates/docker/bin/__tests__/query.test.ts +240 -0
  14. package/templates/docker/bin/__tests__/schema-drift.test.ts +542 -0
  15. package/templates/docker/bin/__tests__/view-yaml-generation.test.ts +146 -0
  16. package/templates/docker/bin/atlas.ts +5044 -0
  17. package/templates/docker/bin/benchmark.ts +695 -0
  18. package/templates/docker/bin/enrich.ts +559 -0
  19. package/templates/docker/bin/eval.ts +770 -0
  20. package/templates/docker/bin/smoke.ts +438 -0
  21. package/templates/docker/data/.gitkeep +0 -0
  22. package/templates/docker/data/cybersec.sql +1961 -0
  23. package/templates/docker/data/demo-semantic/catalog.yml +40 -0
  24. package/templates/docker/data/demo-semantic/entities/accounts.yml +170 -0
  25. package/templates/docker/data/demo-semantic/entities/companies.yml +207 -0
  26. package/templates/docker/data/demo-semantic/entities/people.yml +145 -0
  27. package/templates/docker/data/demo-semantic/glossary.yml +22 -0
  28. package/templates/docker/data/demo-semantic/metrics/accounts.yml +38 -0
  29. package/templates/docker/data/demo-semantic/metrics/companies.yml +89 -0
  30. package/templates/docker/data/demo.sql +373 -0
  31. package/templates/docker/data/ecommerce.sql +1690 -0
  32. package/templates/docker/data/init-demo-db.sql +8 -0
  33. package/templates/docker/docker-compose.yml +34 -0
  34. package/templates/docker/docs/deploy.md +390 -0
  35. package/templates/docker/eslint.config.mjs +18 -0
  36. package/templates/docker/gitignore +5 -0
  37. package/templates/docker/next.config.ts +9 -0
  38. package/templates/docker/package.json +59 -0
  39. package/templates/docker/postcss.config.mjs +8 -0
  40. package/templates/docker/public/.gitkeep +0 -0
  41. package/templates/docker/public/favicon.svg +4 -0
  42. package/templates/docker/railway.json +13 -0
  43. package/templates/docker/render.yaml +34 -0
  44. package/templates/docker/semantic/catalog.yml +5 -0
  45. package/templates/docker/semantic/entities/.gitkeep +0 -0
  46. package/templates/docker/semantic/glossary.yml +6 -0
  47. package/templates/docker/semantic/metrics/.gitkeep +0 -0
  48. package/templates/docker/sidecar/Dockerfile +28 -0
  49. package/templates/docker/sidecar/railway.json +14 -0
  50. package/templates/docker/sidecar/server.ts +188 -0
  51. package/templates/docker/src/api/__tests__/actions.test.ts +683 -0
  52. package/templates/docker/src/api/__tests__/admin.test.ts +820 -0
  53. package/templates/docker/src/api/__tests__/auth.test.ts +165 -0
  54. package/templates/docker/src/api/__tests__/chat.test.ts +376 -0
  55. package/templates/docker/src/api/__tests__/conversations.test.ts +555 -0
  56. package/templates/docker/src/api/__tests__/cors.test.ts +135 -0
  57. package/templates/docker/src/api/__tests__/health-plugin.test.ts +169 -0
  58. package/templates/docker/src/api/__tests__/health.test.ts +261 -0
  59. package/templates/docker/src/api/__tests__/query.test.ts +891 -0
  60. package/templates/docker/src/api/__tests__/scheduled-tasks.test.ts +601 -0
  61. package/templates/docker/src/api/__tests__/slack.test.ts +847 -0
  62. package/templates/docker/src/api/index.ts +117 -0
  63. package/templates/docker/src/api/routes/actions.ts +274 -0
  64. package/templates/docker/src/api/routes/admin.ts +757 -0
  65. package/templates/docker/src/api/routes/auth.ts +48 -0
  66. package/templates/docker/src/api/routes/chat.ts +465 -0
  67. package/templates/docker/src/api/routes/conversations.ts +266 -0
  68. package/templates/docker/src/api/routes/health.ts +287 -0
  69. package/templates/docker/src/api/routes/openapi.ts +390 -0
  70. package/templates/docker/src/api/routes/query.ts +318 -0
  71. package/templates/docker/src/api/routes/scheduled-tasks.ts +467 -0
  72. package/templates/docker/src/api/routes/slack.ts +611 -0
  73. package/templates/docker/src/api/server.ts +226 -0
  74. package/templates/docker/src/app/api/[...route]/route.ts +33 -0
  75. package/templates/docker/src/app/error.tsx +24 -0
  76. package/templates/docker/src/app/globals.css +126 -0
  77. package/templates/docker/src/app/layout.tsx +19 -0
  78. package/templates/docker/src/app/page.tsx +14 -0
  79. package/templates/docker/src/global.d.ts +1 -0
  80. package/templates/docker/src/lib/__tests__/agent-cache.test.ts +437 -0
  81. package/templates/docker/src/lib/__tests__/agent-dialect.test.ts +114 -0
  82. package/templates/docker/src/lib/__tests__/agent-health-annotations.test.ts +164 -0
  83. package/templates/docker/src/lib/__tests__/agent-integration.test.ts +514 -0
  84. package/templates/docker/src/lib/__tests__/config-actions.test.ts +166 -0
  85. package/templates/docker/src/lib/__tests__/config.test.ts +1063 -0
  86. package/templates/docker/src/lib/__tests__/conversations.test.ts +589 -0
  87. package/templates/docker/src/lib/__tests__/errors.test.ts +256 -0
  88. package/templates/docker/src/lib/__tests__/logger.test.ts +200 -0
  89. package/templates/docker/src/lib/__tests__/providers.test.ts +99 -0
  90. package/templates/docker/src/lib/__tests__/rls.test.ts +435 -0
  91. package/templates/docker/src/lib/__tests__/scheduled-task-types.test.ts +124 -0
  92. package/templates/docker/src/lib/__tests__/scheduled-tasks.test.ts +550 -0
  93. package/templates/docker/src/lib/__tests__/semantic-index.test.ts +547 -0
  94. package/templates/docker/src/lib/__tests__/semantic-multisource.test.ts +544 -0
  95. package/templates/docker/src/lib/__tests__/semantic.test.ts +363 -0
  96. package/templates/docker/src/lib/__tests__/startup-actions.test.ts +452 -0
  97. package/templates/docker/src/lib/__tests__/startup.test.ts +465 -0
  98. package/templates/docker/src/lib/__tests__/tracing.test.ts +28 -0
  99. package/templates/docker/src/lib/action-types.ts +95 -0
  100. package/templates/docker/src/lib/agent-query.ts +178 -0
  101. package/templates/docker/src/lib/agent.ts +505 -0
  102. package/templates/docker/src/lib/api-url.ts +2 -0
  103. package/templates/docker/src/lib/auth/__tests__/audit.test.ts +418 -0
  104. package/templates/docker/src/lib/auth/__tests__/byot-integration.test.ts +222 -0
  105. package/templates/docker/src/lib/auth/__tests__/byot.test.ts +366 -0
  106. package/templates/docker/src/lib/auth/__tests__/detect.test.ts +190 -0
  107. package/templates/docker/src/lib/auth/__tests__/managed.test.ts +173 -0
  108. package/templates/docker/src/lib/auth/__tests__/middleware.test.ts +456 -0
  109. package/templates/docker/src/lib/auth/__tests__/migrate.test.ts +201 -0
  110. package/templates/docker/src/lib/auth/__tests__/permissions.test.ts +225 -0
  111. package/templates/docker/src/lib/auth/__tests__/server.test.ts +34 -0
  112. package/templates/docker/src/lib/auth/__tests__/simple-key.test.ts +176 -0
  113. package/templates/docker/src/lib/auth/__tests__/types.test.ts +44 -0
  114. package/templates/docker/src/lib/auth/audit.ts +89 -0
  115. package/templates/docker/src/lib/auth/byot.ts +158 -0
  116. package/templates/docker/src/lib/auth/client.ts +35 -0
  117. package/templates/docker/src/lib/auth/detect.ts +83 -0
  118. package/templates/docker/src/lib/auth/managed.ts +73 -0
  119. package/templates/docker/src/lib/auth/middleware.ts +208 -0
  120. package/templates/docker/src/lib/auth/migrate.ts +111 -0
  121. package/templates/docker/src/lib/auth/permissions.ts +156 -0
  122. package/templates/docker/src/lib/auth/server.ts +142 -0
  123. package/templates/docker/src/lib/auth/simple-key.ts +92 -0
  124. package/templates/docker/src/lib/auth/types.ts +49 -0
  125. package/templates/docker/src/lib/config.ts +704 -0
  126. package/templates/docker/src/lib/conversation-types.ts +29 -0
  127. package/templates/docker/src/lib/conversations.ts +270 -0
  128. package/templates/docker/src/lib/db/__tests__/connection.test.ts +69 -0
  129. package/templates/docker/src/lib/db/__tests__/duckdb.test.ts +141 -0
  130. package/templates/docker/src/lib/db/__tests__/internal.test.ts +387 -0
  131. package/templates/docker/src/lib/db/__tests__/registry-health.test.ts +207 -0
  132. package/templates/docker/src/lib/db/__tests__/registry-pool-limits.test.ts +156 -0
  133. package/templates/docker/src/lib/db/__tests__/registry.test.ts +595 -0
  134. package/templates/docker/src/lib/db/__tests__/salesforce.test.ts +339 -0
  135. package/templates/docker/src/lib/db/__tests__/snowflake.test.ts +217 -0
  136. package/templates/docker/src/lib/db/__tests__/source-rate-limit.test.ts +130 -0
  137. package/templates/docker/src/lib/db/connection.ts +753 -0
  138. package/templates/docker/src/lib/db/duckdb.ts +122 -0
  139. package/templates/docker/src/lib/db/internal.ts +273 -0
  140. package/templates/docker/src/lib/db/salesforce.ts +342 -0
  141. package/templates/docker/src/lib/db/source-rate-limit.ts +191 -0
  142. package/templates/docker/src/lib/errors.ts +154 -0
  143. package/templates/docker/src/lib/logger.ts +98 -0
  144. package/templates/docker/src/lib/plugins/__tests__/hooks-integration.test.ts +202 -0
  145. package/templates/docker/src/lib/plugins/__tests__/hooks.test.ts +529 -0
  146. package/templates/docker/src/lib/plugins/__tests__/migrate.test.ts +521 -0
  147. package/templates/docker/src/lib/plugins/__tests__/registry.test.ts +346 -0
  148. package/templates/docker/src/lib/plugins/__tests__/tools.test.ts +49 -0
  149. package/templates/docker/src/lib/plugins/__tests__/wiring.test.ts +585 -0
  150. package/templates/docker/src/lib/plugins/hooks.ts +162 -0
  151. package/templates/docker/src/lib/plugins/index.ts +9 -0
  152. package/templates/docker/src/lib/plugins/migrate.ts +309 -0
  153. package/templates/docker/src/lib/plugins/registry.ts +231 -0
  154. package/templates/docker/src/lib/plugins/tools.ts +39 -0
  155. package/templates/docker/src/lib/plugins/wiring.ts +291 -0
  156. package/templates/docker/src/lib/providers.ts +102 -0
  157. package/templates/docker/src/lib/rls.ts +321 -0
  158. package/templates/docker/src/lib/scheduled-task-types.ts +132 -0
  159. package/templates/docker/src/lib/scheduled-tasks.ts +475 -0
  160. package/templates/docker/src/lib/scheduler/__tests__/delivery.test.ts +192 -0
  161. package/templates/docker/src/lib/scheduler/__tests__/engine.test.ts +248 -0
  162. package/templates/docker/src/lib/scheduler/__tests__/format-email.test.ts +96 -0
  163. package/templates/docker/src/lib/scheduler/__tests__/format-slack.test.ts +78 -0
  164. package/templates/docker/src/lib/scheduler/__tests__/format-webhook.test.ts +78 -0
  165. package/templates/docker/src/lib/scheduler/delivery.ts +248 -0
  166. package/templates/docker/src/lib/scheduler/engine.ts +317 -0
  167. package/templates/docker/src/lib/scheduler/executor.ts +73 -0
  168. package/templates/docker/src/lib/scheduler/format-email.ts +109 -0
  169. package/templates/docker/src/lib/scheduler/format-slack.ts +35 -0
  170. package/templates/docker/src/lib/scheduler/format-webhook.ts +37 -0
  171. package/templates/docker/src/lib/scheduler/index.ts +7 -0
  172. package/templates/docker/src/lib/security.ts +11 -0
  173. package/templates/docker/src/lib/semantic-index.ts +503 -0
  174. package/templates/docker/src/lib/semantic.ts +387 -0
  175. package/templates/docker/src/lib/sidecar-types.ts +16 -0
  176. package/templates/docker/src/lib/slack/__tests__/api.test.ts +160 -0
  177. package/templates/docker/src/lib/slack/__tests__/format.test.ts +237 -0
  178. package/templates/docker/src/lib/slack/__tests__/store.test.ts +188 -0
  179. package/templates/docker/src/lib/slack/__tests__/threads.test.ts +112 -0
  180. package/templates/docker/src/lib/slack/__tests__/verify.test.ts +111 -0
  181. package/templates/docker/src/lib/slack/api.ts +102 -0
  182. package/templates/docker/src/lib/slack/format.ts +209 -0
  183. package/templates/docker/src/lib/slack/store.ts +107 -0
  184. package/templates/docker/src/lib/slack/threads.ts +64 -0
  185. package/templates/docker/src/lib/slack/verify.ts +71 -0
  186. package/templates/docker/src/lib/startup.ts +730 -0
  187. package/templates/docker/src/lib/tools/__tests__/action-permissions.test.ts +594 -0
  188. package/templates/docker/src/lib/tools/__tests__/custom-validation.test.ts +238 -0
  189. package/templates/docker/src/lib/tools/__tests__/explore-backend.test.ts +267 -0
  190. package/templates/docker/src/lib/tools/__tests__/explore-nsjail.test.ts +492 -0
  191. package/templates/docker/src/lib/tools/__tests__/explore-plugin.test.ts +374 -0
  192. package/templates/docker/src/lib/tools/__tests__/explore-sdk-compat.test.ts +82 -0
  193. package/templates/docker/src/lib/tools/__tests__/explore-sidecar.test.ts +208 -0
  194. package/templates/docker/src/lib/tools/__tests__/registry-actions.test.ts +144 -0
  195. package/templates/docker/src/lib/tools/__tests__/registry.test.ts +235 -0
  196. package/templates/docker/src/lib/tools/__tests__/salesforce-tool.test.ts +154 -0
  197. package/templates/docker/src/lib/tools/__tests__/soql-validation.test.ts +303 -0
  198. package/templates/docker/src/lib/tools/__tests__/sql-audit.test.ts +225 -0
  199. package/templates/docker/src/lib/tools/__tests__/sql-connection-whitelist.test.ts +98 -0
  200. package/templates/docker/src/lib/tools/__tests__/sql-duckdb.test.ts +233 -0
  201. package/templates/docker/src/lib/tools/__tests__/sql-ratelimit.test.ts +225 -0
  202. package/templates/docker/src/lib/tools/__tests__/sql.test.ts +1012 -0
  203. package/templates/docker/src/lib/tools/actions/__tests__/audit.test.ts +211 -0
  204. package/templates/docker/src/lib/tools/actions/__tests__/email.test.ts +378 -0
  205. package/templates/docker/src/lib/tools/actions/__tests__/handler.test.ts +681 -0
  206. package/templates/docker/src/lib/tools/actions/__tests__/jira.test.ts +427 -0
  207. package/templates/docker/src/lib/tools/actions/audit.ts +47 -0
  208. package/templates/docker/src/lib/tools/actions/email.ts +191 -0
  209. package/templates/docker/src/lib/tools/actions/handler.ts +591 -0
  210. package/templates/docker/src/lib/tools/actions/index.ts +23 -0
  211. package/templates/docker/src/lib/tools/actions/jira.ts +220 -0
  212. package/templates/docker/src/lib/tools/explore-nsjail.ts +343 -0
  213. package/templates/docker/src/lib/tools/explore-sandbox.ts +264 -0
  214. package/templates/docker/src/lib/tools/explore-sidecar.ts +163 -0
  215. package/templates/docker/src/lib/tools/explore.ts +379 -0
  216. package/templates/docker/src/lib/tools/registry.ts +221 -0
  217. package/templates/docker/src/lib/tools/salesforce.ts +138 -0
  218. package/templates/docker/src/lib/tools/soql-validation.ts +172 -0
  219. package/templates/docker/src/lib/tools/sql.ts +680 -0
  220. package/templates/docker/src/lib/tracing.ts +40 -0
  221. package/templates/docker/src/lib/utils.ts +6 -0
  222. package/templates/docker/src/test-setup.ts +38 -0
  223. package/templates/docker/src/types/vercel-sandbox.d.ts +54 -0
  224. package/templates/docker/src/ui/components/actions/action-approval-card.tsx +295 -0
  225. package/templates/docker/src/ui/components/actions/action-status-badge.tsx +50 -0
  226. package/templates/docker/src/ui/components/admin/admin-layout.tsx +26 -0
  227. package/templates/docker/src/ui/components/admin/admin-sidebar.tsx +96 -0
  228. package/templates/docker/src/ui/components/admin/empty-state.tsx +24 -0
  229. package/templates/docker/src/ui/components/admin/entity-detail.tsx +233 -0
  230. package/templates/docker/src/ui/components/admin/entity-list.tsx +96 -0
  231. package/templates/docker/src/ui/components/admin/error-banner.tsx +22 -0
  232. package/templates/docker/src/ui/components/admin/feature-disabled.tsx +44 -0
  233. package/templates/docker/src/ui/components/admin/health-badge.tsx +30 -0
  234. package/templates/docker/src/ui/components/admin/loading-state.tsx +14 -0
  235. package/templates/docker/src/ui/components/admin/stat-card.tsx +32 -0
  236. package/templates/docker/src/ui/components/atlas-chat.tsx +370 -0
  237. package/templates/docker/src/ui/components/chart/chart-detection.ts +261 -0
  238. package/templates/docker/src/ui/components/chart/result-chart.tsx +375 -0
  239. package/templates/docker/src/ui/components/chat/api-key-bar.tsx +66 -0
  240. package/templates/docker/src/ui/components/chat/copy-button.tsx +25 -0
  241. package/templates/docker/src/ui/components/chat/data-table.tsx +102 -0
  242. package/templates/docker/src/ui/components/chat/error-banner.tsx +32 -0
  243. package/templates/docker/src/ui/components/chat/explore-card.tsx +41 -0
  244. package/templates/docker/src/ui/components/chat/loading-card.tsx +10 -0
  245. package/templates/docker/src/ui/components/chat/managed-auth-card.tsx +116 -0
  246. package/templates/docker/src/ui/components/chat/markdown.tsx +72 -0
  247. package/templates/docker/src/ui/components/chat/sql-block.tsx +30 -0
  248. package/templates/docker/src/ui/components/chat/sql-result-card.tsx +144 -0
  249. package/templates/docker/src/ui/components/chat/starter-prompts.ts +6 -0
  250. package/templates/docker/src/ui/components/chat/tool-part.tsx +40 -0
  251. package/templates/docker/src/ui/components/chat/typing-indicator.tsx +19 -0
  252. package/templates/docker/src/ui/components/conversations/conversation-item.tsx +120 -0
  253. package/templates/docker/src/ui/components/conversations/conversation-list.tsx +66 -0
  254. package/templates/docker/src/ui/components/conversations/conversation-sidebar.tsx +78 -0
  255. package/templates/docker/src/ui/components/conversations/delete-confirmation.tsx +27 -0
  256. package/templates/docker/src/ui/context.tsx +78 -0
  257. package/templates/docker/src/ui/hooks/use-admin-fetch.ts +104 -0
  258. package/templates/docker/src/ui/hooks/use-conversations.ts +184 -0
  259. package/templates/docker/src/ui/hooks/use-dark-mode.ts +17 -0
  260. package/templates/docker/src/ui/lib/action-types.ts +63 -0
  261. package/templates/docker/src/ui/lib/helpers.ts +104 -0
  262. package/templates/docker/src/ui/lib/types.ts +145 -0
  263. package/templates/docker/tsconfig.json +41 -0
  264. package/templates/docker/vercel.json +3 -0
  265. package/templates/nextjs-standalone/.env.example +68 -0
  266. package/templates/nextjs-standalone/bin/__tests__/benchmark.test.ts +598 -0
  267. package/templates/nextjs-standalone/bin/__tests__/duckdb-ingest.test.ts +171 -0
  268. package/templates/nextjs-standalone/bin/__tests__/eval.test.ts +434 -0
  269. package/templates/nextjs-standalone/bin/__tests__/matview-partition.test.ts +615 -0
  270. package/templates/nextjs-standalone/bin/__tests__/multi-source.test.ts +113 -0
  271. package/templates/nextjs-standalone/bin/__tests__/plugin-cli.test.ts +322 -0
  272. package/templates/nextjs-standalone/bin/__tests__/profiler-heuristics.test.ts +608 -0
  273. package/templates/nextjs-standalone/bin/__tests__/query.test.ts +240 -0
  274. package/templates/nextjs-standalone/bin/__tests__/schema-drift.test.ts +542 -0
  275. package/templates/nextjs-standalone/bin/__tests__/view-yaml-generation.test.ts +146 -0
  276. package/templates/nextjs-standalone/bin/atlas.ts +5044 -0
  277. package/templates/nextjs-standalone/bin/benchmark.ts +695 -0
  278. package/templates/nextjs-standalone/bin/enrich.ts +559 -0
  279. package/templates/nextjs-standalone/bin/eval.ts +770 -0
  280. package/templates/nextjs-standalone/bin/smoke.ts +438 -0
  281. package/templates/nextjs-standalone/data/.gitkeep +0 -0
  282. package/templates/nextjs-standalone/data/cybersec.sql +1961 -0
  283. package/templates/nextjs-standalone/data/demo-semantic/catalog.yml +40 -0
  284. package/templates/nextjs-standalone/data/demo-semantic/entities/accounts.yml +170 -0
  285. package/templates/nextjs-standalone/data/demo-semantic/entities/companies.yml +207 -0
  286. package/templates/nextjs-standalone/data/demo-semantic/entities/people.yml +145 -0
  287. package/templates/nextjs-standalone/data/demo-semantic/glossary.yml +22 -0
  288. package/templates/nextjs-standalone/data/demo-semantic/metrics/accounts.yml +38 -0
  289. package/templates/nextjs-standalone/data/demo-semantic/metrics/companies.yml +89 -0
  290. package/templates/nextjs-standalone/data/demo.sql +373 -0
  291. package/templates/nextjs-standalone/data/ecommerce.sql +1690 -0
  292. package/templates/nextjs-standalone/data/init-demo-db.sql +8 -0
  293. package/templates/nextjs-standalone/docs/deploy.md +390 -0
  294. package/templates/nextjs-standalone/eslint.config.mjs +18 -0
  295. package/templates/nextjs-standalone/gitignore +5 -0
  296. package/templates/nextjs-standalone/next.config.ts +10 -0
  297. package/templates/nextjs-standalone/package.json +63 -0
  298. package/templates/nextjs-standalone/postcss.config.mjs +8 -0
  299. package/templates/nextjs-standalone/semantic/catalog.yml +5 -0
  300. package/templates/nextjs-standalone/semantic/entities/.gitkeep +0 -0
  301. package/templates/nextjs-standalone/semantic/glossary.yml +6 -0
  302. package/templates/nextjs-standalone/semantic/metrics/.gitkeep +0 -0
  303. package/templates/nextjs-standalone/src/api/__tests__/actions.test.ts +683 -0
  304. package/templates/nextjs-standalone/src/api/__tests__/admin.test.ts +820 -0
  305. package/templates/nextjs-standalone/src/api/__tests__/auth.test.ts +165 -0
  306. package/templates/nextjs-standalone/src/api/__tests__/chat.test.ts +376 -0
  307. package/templates/nextjs-standalone/src/api/__tests__/conversations.test.ts +555 -0
  308. package/templates/nextjs-standalone/src/api/__tests__/cors.test.ts +135 -0
  309. package/templates/nextjs-standalone/src/api/__tests__/health-plugin.test.ts +169 -0
  310. package/templates/nextjs-standalone/src/api/__tests__/health.test.ts +261 -0
  311. package/templates/nextjs-standalone/src/api/__tests__/query.test.ts +891 -0
  312. package/templates/nextjs-standalone/src/api/__tests__/scheduled-tasks.test.ts +601 -0
  313. package/templates/nextjs-standalone/src/api/__tests__/slack.test.ts +847 -0
  314. package/templates/nextjs-standalone/src/api/index.ts +117 -0
  315. package/templates/nextjs-standalone/src/api/routes/actions.ts +274 -0
  316. package/templates/nextjs-standalone/src/api/routes/admin.ts +757 -0
  317. package/templates/nextjs-standalone/src/api/routes/auth.ts +48 -0
  318. package/templates/nextjs-standalone/src/api/routes/chat.ts +465 -0
  319. package/templates/nextjs-standalone/src/api/routes/conversations.ts +266 -0
  320. package/templates/nextjs-standalone/src/api/routes/health.ts +287 -0
  321. package/templates/nextjs-standalone/src/api/routes/openapi.ts +390 -0
  322. package/templates/nextjs-standalone/src/api/routes/query.ts +318 -0
  323. package/templates/nextjs-standalone/src/api/routes/scheduled-tasks.ts +467 -0
  324. package/templates/nextjs-standalone/src/api/routes/slack.ts +611 -0
  325. package/templates/nextjs-standalone/src/api/server.ts +226 -0
  326. package/templates/nextjs-standalone/src/app/api/[...route]/route.ts +33 -0
  327. package/templates/nextjs-standalone/src/app/error.tsx +24 -0
  328. package/templates/nextjs-standalone/src/app/global-error.tsx +68 -0
  329. package/templates/nextjs-standalone/src/app/globals.css +126 -0
  330. package/templates/nextjs-standalone/src/app/layout.tsx +19 -0
  331. package/templates/nextjs-standalone/src/app/page.tsx +14 -0
  332. package/templates/nextjs-standalone/src/lib/__tests__/agent-cache.test.ts +437 -0
  333. package/templates/nextjs-standalone/src/lib/__tests__/agent-dialect.test.ts +114 -0
  334. package/templates/nextjs-standalone/src/lib/__tests__/agent-health-annotations.test.ts +164 -0
  335. package/templates/nextjs-standalone/src/lib/__tests__/agent-integration.test.ts +514 -0
  336. package/templates/nextjs-standalone/src/lib/__tests__/config-actions.test.ts +166 -0
  337. package/templates/nextjs-standalone/src/lib/__tests__/config.test.ts +1063 -0
  338. package/templates/nextjs-standalone/src/lib/__tests__/conversations.test.ts +589 -0
  339. package/templates/nextjs-standalone/src/lib/__tests__/errors.test.ts +256 -0
  340. package/templates/nextjs-standalone/src/lib/__tests__/logger.test.ts +200 -0
  341. package/templates/nextjs-standalone/src/lib/__tests__/providers.test.ts +99 -0
  342. package/templates/nextjs-standalone/src/lib/__tests__/rls.test.ts +435 -0
  343. package/templates/nextjs-standalone/src/lib/__tests__/scheduled-task-types.test.ts +124 -0
  344. package/templates/nextjs-standalone/src/lib/__tests__/scheduled-tasks.test.ts +550 -0
  345. package/templates/nextjs-standalone/src/lib/__tests__/semantic-index.test.ts +547 -0
  346. package/templates/nextjs-standalone/src/lib/__tests__/semantic-multisource.test.ts +544 -0
  347. package/templates/nextjs-standalone/src/lib/__tests__/semantic.test.ts +363 -0
  348. package/templates/nextjs-standalone/src/lib/__tests__/startup-actions.test.ts +452 -0
  349. package/templates/nextjs-standalone/src/lib/__tests__/startup.test.ts +465 -0
  350. package/templates/nextjs-standalone/src/lib/__tests__/tracing.test.ts +28 -0
  351. package/templates/nextjs-standalone/src/lib/action-types.ts +95 -0
  352. package/templates/nextjs-standalone/src/lib/agent-query.ts +178 -0
  353. package/templates/nextjs-standalone/src/lib/agent.ts +505 -0
  354. package/templates/nextjs-standalone/src/lib/api-url.ts +3 -0
  355. package/templates/nextjs-standalone/src/lib/auth/__tests__/audit.test.ts +418 -0
  356. package/templates/nextjs-standalone/src/lib/auth/__tests__/byot-integration.test.ts +222 -0
  357. package/templates/nextjs-standalone/src/lib/auth/__tests__/byot.test.ts +366 -0
  358. package/templates/nextjs-standalone/src/lib/auth/__tests__/detect.test.ts +190 -0
  359. package/templates/nextjs-standalone/src/lib/auth/__tests__/managed.test.ts +173 -0
  360. package/templates/nextjs-standalone/src/lib/auth/__tests__/middleware.test.ts +456 -0
  361. package/templates/nextjs-standalone/src/lib/auth/__tests__/migrate.test.ts +201 -0
  362. package/templates/nextjs-standalone/src/lib/auth/__tests__/permissions.test.ts +225 -0
  363. package/templates/nextjs-standalone/src/lib/auth/__tests__/server.test.ts +34 -0
  364. package/templates/nextjs-standalone/src/lib/auth/__tests__/simple-key.test.ts +176 -0
  365. package/templates/nextjs-standalone/src/lib/auth/__tests__/types.test.ts +44 -0
  366. package/templates/nextjs-standalone/src/lib/auth/audit.ts +89 -0
  367. package/templates/nextjs-standalone/src/lib/auth/byot.ts +158 -0
  368. package/templates/nextjs-standalone/src/lib/auth/client.ts +23 -0
  369. package/templates/nextjs-standalone/src/lib/auth/detect.ts +83 -0
  370. package/templates/nextjs-standalone/src/lib/auth/managed.ts +73 -0
  371. package/templates/nextjs-standalone/src/lib/auth/middleware.ts +208 -0
  372. package/templates/nextjs-standalone/src/lib/auth/migrate.ts +111 -0
  373. package/templates/nextjs-standalone/src/lib/auth/permissions.ts +156 -0
  374. package/templates/nextjs-standalone/src/lib/auth/server.ts +142 -0
  375. package/templates/nextjs-standalone/src/lib/auth/simple-key.ts +92 -0
  376. package/templates/nextjs-standalone/src/lib/auth/types.ts +49 -0
  377. package/templates/nextjs-standalone/src/lib/config.ts +704 -0
  378. package/templates/nextjs-standalone/src/lib/conversation-types.ts +29 -0
  379. package/templates/nextjs-standalone/src/lib/conversations.ts +270 -0
  380. package/templates/nextjs-standalone/src/lib/db/__tests__/connection.test.ts +69 -0
  381. package/templates/nextjs-standalone/src/lib/db/__tests__/duckdb.test.ts +141 -0
  382. package/templates/nextjs-standalone/src/lib/db/__tests__/internal.test.ts +387 -0
  383. package/templates/nextjs-standalone/src/lib/db/__tests__/registry-health.test.ts +207 -0
  384. package/templates/nextjs-standalone/src/lib/db/__tests__/registry-pool-limits.test.ts +156 -0
  385. package/templates/nextjs-standalone/src/lib/db/__tests__/registry.test.ts +595 -0
  386. package/templates/nextjs-standalone/src/lib/db/__tests__/salesforce.test.ts +339 -0
  387. package/templates/nextjs-standalone/src/lib/db/__tests__/snowflake.test.ts +217 -0
  388. package/templates/nextjs-standalone/src/lib/db/__tests__/source-rate-limit.test.ts +130 -0
  389. package/templates/nextjs-standalone/src/lib/db/connection.ts +753 -0
  390. package/templates/nextjs-standalone/src/lib/db/duckdb.ts +122 -0
  391. package/templates/nextjs-standalone/src/lib/db/internal.ts +273 -0
  392. package/templates/nextjs-standalone/src/lib/db/salesforce.ts +342 -0
  393. package/templates/nextjs-standalone/src/lib/db/source-rate-limit.ts +191 -0
  394. package/templates/nextjs-standalone/src/lib/errors.ts +154 -0
  395. package/templates/nextjs-standalone/src/lib/logger.ts +98 -0
  396. package/templates/nextjs-standalone/src/lib/plugins/__tests__/hooks-integration.test.ts +202 -0
  397. package/templates/nextjs-standalone/src/lib/plugins/__tests__/hooks.test.ts +529 -0
  398. package/templates/nextjs-standalone/src/lib/plugins/__tests__/migrate.test.ts +521 -0
  399. package/templates/nextjs-standalone/src/lib/plugins/__tests__/registry.test.ts +346 -0
  400. package/templates/nextjs-standalone/src/lib/plugins/__tests__/tools.test.ts +49 -0
  401. package/templates/nextjs-standalone/src/lib/plugins/__tests__/wiring.test.ts +585 -0
  402. package/templates/nextjs-standalone/src/lib/plugins/hooks.ts +162 -0
  403. package/templates/nextjs-standalone/src/lib/plugins/index.ts +9 -0
  404. package/templates/nextjs-standalone/src/lib/plugins/migrate.ts +309 -0
  405. package/templates/nextjs-standalone/src/lib/plugins/registry.ts +231 -0
  406. package/templates/nextjs-standalone/src/lib/plugins/tools.ts +39 -0
  407. package/templates/nextjs-standalone/src/lib/plugins/wiring.ts +291 -0
  408. package/templates/nextjs-standalone/src/lib/providers.ts +102 -0
  409. package/templates/nextjs-standalone/src/lib/rls.ts +321 -0
  410. package/templates/nextjs-standalone/src/lib/scheduled-task-types.ts +132 -0
  411. package/templates/nextjs-standalone/src/lib/scheduled-tasks.ts +475 -0
  412. package/templates/nextjs-standalone/src/lib/scheduler/__tests__/delivery.test.ts +192 -0
  413. package/templates/nextjs-standalone/src/lib/scheduler/__tests__/engine.test.ts +248 -0
  414. package/templates/nextjs-standalone/src/lib/scheduler/__tests__/format-email.test.ts +96 -0
  415. package/templates/nextjs-standalone/src/lib/scheduler/__tests__/format-slack.test.ts +78 -0
  416. package/templates/nextjs-standalone/src/lib/scheduler/__tests__/format-webhook.test.ts +78 -0
  417. package/templates/nextjs-standalone/src/lib/scheduler/delivery.ts +248 -0
  418. package/templates/nextjs-standalone/src/lib/scheduler/engine.ts +317 -0
  419. package/templates/nextjs-standalone/src/lib/scheduler/executor.ts +73 -0
  420. package/templates/nextjs-standalone/src/lib/scheduler/format-email.ts +109 -0
  421. package/templates/nextjs-standalone/src/lib/scheduler/format-slack.ts +35 -0
  422. package/templates/nextjs-standalone/src/lib/scheduler/format-webhook.ts +37 -0
  423. package/templates/nextjs-standalone/src/lib/scheduler/index.ts +7 -0
  424. package/templates/nextjs-standalone/src/lib/security.ts +11 -0
  425. package/templates/nextjs-standalone/src/lib/semantic-index.ts +503 -0
  426. package/templates/nextjs-standalone/src/lib/semantic.ts +387 -0
  427. package/templates/nextjs-standalone/src/lib/sidecar-types.ts +16 -0
  428. package/templates/nextjs-standalone/src/lib/slack/__tests__/api.test.ts +160 -0
  429. package/templates/nextjs-standalone/src/lib/slack/__tests__/format.test.ts +237 -0
  430. package/templates/nextjs-standalone/src/lib/slack/__tests__/store.test.ts +188 -0
  431. package/templates/nextjs-standalone/src/lib/slack/__tests__/threads.test.ts +112 -0
  432. package/templates/nextjs-standalone/src/lib/slack/__tests__/verify.test.ts +111 -0
  433. package/templates/nextjs-standalone/src/lib/slack/api.ts +102 -0
  434. package/templates/nextjs-standalone/src/lib/slack/format.ts +209 -0
  435. package/templates/nextjs-standalone/src/lib/slack/store.ts +107 -0
  436. package/templates/nextjs-standalone/src/lib/slack/threads.ts +64 -0
  437. package/templates/nextjs-standalone/src/lib/slack/verify.ts +71 -0
  438. package/templates/nextjs-standalone/src/lib/startup.ts +730 -0
  439. package/templates/nextjs-standalone/src/lib/tools/__tests__/action-permissions.test.ts +594 -0
  440. package/templates/nextjs-standalone/src/lib/tools/__tests__/custom-validation.test.ts +238 -0
  441. package/templates/nextjs-standalone/src/lib/tools/__tests__/explore-backend.test.ts +267 -0
  442. package/templates/nextjs-standalone/src/lib/tools/__tests__/explore-nsjail.test.ts +492 -0
  443. package/templates/nextjs-standalone/src/lib/tools/__tests__/explore-plugin.test.ts +374 -0
  444. package/templates/nextjs-standalone/src/lib/tools/__tests__/explore-sdk-compat.test.ts +82 -0
  445. package/templates/nextjs-standalone/src/lib/tools/__tests__/explore-sidecar.test.ts +208 -0
  446. package/templates/nextjs-standalone/src/lib/tools/__tests__/registry-actions.test.ts +144 -0
  447. package/templates/nextjs-standalone/src/lib/tools/__tests__/registry.test.ts +235 -0
  448. package/templates/nextjs-standalone/src/lib/tools/__tests__/salesforce-tool.test.ts +154 -0
  449. package/templates/nextjs-standalone/src/lib/tools/__tests__/soql-validation.test.ts +303 -0
  450. package/templates/nextjs-standalone/src/lib/tools/__tests__/sql-audit.test.ts +225 -0
  451. package/templates/nextjs-standalone/src/lib/tools/__tests__/sql-connection-whitelist.test.ts +98 -0
  452. package/templates/nextjs-standalone/src/lib/tools/__tests__/sql-duckdb.test.ts +233 -0
  453. package/templates/nextjs-standalone/src/lib/tools/__tests__/sql-ratelimit.test.ts +225 -0
  454. package/templates/nextjs-standalone/src/lib/tools/__tests__/sql.test.ts +1012 -0
  455. package/templates/nextjs-standalone/src/lib/tools/actions/__tests__/audit.test.ts +211 -0
  456. package/templates/nextjs-standalone/src/lib/tools/actions/__tests__/email.test.ts +378 -0
  457. package/templates/nextjs-standalone/src/lib/tools/actions/__tests__/handler.test.ts +681 -0
  458. package/templates/nextjs-standalone/src/lib/tools/actions/__tests__/jira.test.ts +427 -0
  459. package/templates/nextjs-standalone/src/lib/tools/actions/audit.ts +47 -0
  460. package/templates/nextjs-standalone/src/lib/tools/actions/email.ts +191 -0
  461. package/templates/nextjs-standalone/src/lib/tools/actions/handler.ts +591 -0
  462. package/templates/nextjs-standalone/src/lib/tools/actions/index.ts +23 -0
  463. package/templates/nextjs-standalone/src/lib/tools/actions/jira.ts +220 -0
  464. package/templates/nextjs-standalone/src/lib/tools/explore-nsjail.ts +343 -0
  465. package/templates/nextjs-standalone/src/lib/tools/explore-sandbox.ts +264 -0
  466. package/templates/nextjs-standalone/src/lib/tools/explore-sidecar.ts +163 -0
  467. package/templates/nextjs-standalone/src/lib/tools/explore.ts +379 -0
  468. package/templates/nextjs-standalone/src/lib/tools/registry.ts +221 -0
  469. package/templates/nextjs-standalone/src/lib/tools/salesforce.ts +138 -0
  470. package/templates/nextjs-standalone/src/lib/tools/soql-validation.ts +172 -0
  471. package/templates/nextjs-standalone/src/lib/tools/sql.ts +680 -0
  472. package/templates/nextjs-standalone/src/lib/tracing.ts +40 -0
  473. package/templates/nextjs-standalone/src/lib/utils.ts +6 -0
  474. package/templates/nextjs-standalone/src/test-setup.ts +38 -0
  475. package/templates/nextjs-standalone/src/ui/components/actions/action-approval-card.tsx +295 -0
  476. package/templates/nextjs-standalone/src/ui/components/actions/action-status-badge.tsx +50 -0
  477. package/templates/nextjs-standalone/src/ui/components/admin/admin-layout.tsx +26 -0
  478. package/templates/nextjs-standalone/src/ui/components/admin/admin-sidebar.tsx +96 -0
  479. package/templates/nextjs-standalone/src/ui/components/admin/empty-state.tsx +24 -0
  480. package/templates/nextjs-standalone/src/ui/components/admin/entity-detail.tsx +233 -0
  481. package/templates/nextjs-standalone/src/ui/components/admin/entity-list.tsx +96 -0
  482. package/templates/nextjs-standalone/src/ui/components/admin/error-banner.tsx +22 -0
  483. package/templates/nextjs-standalone/src/ui/components/admin/feature-disabled.tsx +44 -0
  484. package/templates/nextjs-standalone/src/ui/components/admin/health-badge.tsx +30 -0
  485. package/templates/nextjs-standalone/src/ui/components/admin/loading-state.tsx +14 -0
  486. package/templates/nextjs-standalone/src/ui/components/admin/stat-card.tsx +32 -0
  487. package/templates/nextjs-standalone/src/ui/components/atlas-chat.tsx +370 -0
  488. package/templates/nextjs-standalone/src/ui/components/chart/chart-detection.ts +261 -0
  489. package/templates/nextjs-standalone/src/ui/components/chart/result-chart.tsx +375 -0
  490. package/templates/nextjs-standalone/src/ui/components/chat/api-key-bar.tsx +66 -0
  491. package/templates/nextjs-standalone/src/ui/components/chat/copy-button.tsx +25 -0
  492. package/templates/nextjs-standalone/src/ui/components/chat/data-table.tsx +102 -0
  493. package/templates/nextjs-standalone/src/ui/components/chat/error-banner.tsx +32 -0
  494. package/templates/nextjs-standalone/src/ui/components/chat/explore-card.tsx +41 -0
  495. package/templates/nextjs-standalone/src/ui/components/chat/loading-card.tsx +10 -0
  496. package/templates/nextjs-standalone/src/ui/components/chat/managed-auth-card.tsx +116 -0
  497. package/templates/nextjs-standalone/src/ui/components/chat/markdown.tsx +72 -0
  498. package/templates/nextjs-standalone/src/ui/components/chat/sql-block.tsx +30 -0
  499. package/templates/nextjs-standalone/src/ui/components/chat/sql-result-card.tsx +144 -0
  500. package/templates/nextjs-standalone/src/ui/components/chat/starter-prompts.ts +6 -0
  501. package/templates/nextjs-standalone/src/ui/components/chat/tool-part.tsx +40 -0
  502. package/templates/nextjs-standalone/src/ui/components/chat/typing-indicator.tsx +19 -0
  503. package/templates/nextjs-standalone/src/ui/components/conversations/conversation-item.tsx +120 -0
  504. package/templates/nextjs-standalone/src/ui/components/conversations/conversation-list.tsx +66 -0
  505. package/templates/nextjs-standalone/src/ui/components/conversations/conversation-sidebar.tsx +78 -0
  506. package/templates/nextjs-standalone/src/ui/components/conversations/delete-confirmation.tsx +27 -0
  507. package/templates/nextjs-standalone/src/ui/context.tsx +78 -0
  508. package/templates/nextjs-standalone/src/ui/hooks/use-admin-fetch.ts +104 -0
  509. package/templates/nextjs-standalone/src/ui/hooks/use-conversations.ts +184 -0
  510. package/templates/nextjs-standalone/src/ui/hooks/use-dark-mode.ts +17 -0
  511. package/templates/nextjs-standalone/src/ui/lib/action-types.ts +63 -0
  512. package/templates/nextjs-standalone/src/ui/lib/helpers.ts +104 -0
  513. package/templates/nextjs-standalone/src/ui/lib/types.ts +145 -0
  514. package/templates/nextjs-standalone/tsconfig.json +32 -0
  515. package/templates/nextjs-standalone/vercel.json +4 -0
@@ -0,0 +1,171 @@
1
+ /**
2
+ * Integration tests for DuckDB CSV/Parquet ingestion via the CLI.
3
+ *
4
+ * Tests the ingestIntoDuckDB, listDuckDBObjects, and profileDuckDB functions
5
+ * using real DuckDB instances with temporary files.
6
+ */
7
+ import { describe, it, expect, afterEach } from "bun:test";
8
+ import * as fs from "fs";
9
+ import * as path from "path";
10
+ import * as os from "os";
11
+ import { ingestIntoDuckDB, listDuckDBObjects, profileDuckDB } from "../atlas";
12
+
13
+ let tmpDir: string;
14
+
15
+ function createTmpDir(): string {
16
+ const dir = fs.mkdtempSync(path.join(os.tmpdir(), "atlas-duckdb-test-"));
17
+ tmpDir = dir;
18
+ return dir;
19
+ }
20
+
21
+ function cleanTmpDir(): void {
22
+ if (tmpDir && fs.existsSync(tmpDir)) {
23
+ fs.rmSync(tmpDir, { recursive: true, force: true });
24
+ }
25
+ }
26
+
27
+ afterEach(cleanTmpDir);
28
+
29
+ describe("ingestIntoDuckDB", () => {
30
+ it("ingests a CSV file and creates a table", async () => {
31
+ const dir = createTmpDir();
32
+ const csvPath = path.join(dir, "sales.csv");
33
+ fs.writeFileSync(csvPath, "id,product,amount\n1,Widget,100\n2,Gadget,250\n3,Widget,75\n");
34
+
35
+ const dbPath = path.join(dir, "test.duckdb");
36
+ const tables = await ingestIntoDuckDB(dbPath, [{ path: csvPath, format: "csv" }]);
37
+
38
+ expect(tables).toEqual(["sales"]);
39
+ expect(fs.existsSync(dbPath)).toBe(true);
40
+ });
41
+
42
+ it("ingests multiple CSV files", async () => {
43
+ const dir = createTmpDir();
44
+ const csv1 = path.join(dir, "products.csv");
45
+ const csv2 = path.join(dir, "orders.csv");
46
+ fs.writeFileSync(csv1, "id,name,price\n1,Widget,9.99\n2,Gadget,19.99\n");
47
+ fs.writeFileSync(csv2, "id,product_id,quantity\n1,1,5\n2,2,3\n");
48
+
49
+ const dbPath = path.join(dir, "test.duckdb");
50
+ const tables = await ingestIntoDuckDB(dbPath, [
51
+ { path: csv1, format: "csv" },
52
+ { path: csv2, format: "csv" },
53
+ ]);
54
+
55
+ expect(tables).toContain("products");
56
+ expect(tables).toContain("orders");
57
+ expect(tables).toHaveLength(2);
58
+ });
59
+
60
+ it("sanitizes table names from file stems", async () => {
61
+ const dir = createTmpDir();
62
+ const csvPath = path.join(dir, "my-sales data (2024).csv");
63
+ fs.writeFileSync(csvPath, "id,amount\n1,100\n");
64
+
65
+ const dbPath = path.join(dir, "test.duckdb");
66
+ const tables = await ingestIntoDuckDB(dbPath, [{ path: csvPath, format: "csv" }]);
67
+
68
+ // Non-identifier chars replaced with underscores
69
+ expect(tables[0]).toMatch(/^[a-z_][a-z0-9_]*$/);
70
+ });
71
+
72
+ it("throws for missing file", async () => {
73
+ const dir = createTmpDir();
74
+ const dbPath = path.join(dir, "test.duckdb");
75
+
76
+ await expect(
77
+ ingestIntoDuckDB(dbPath, [{ path: "/nonexistent/file.csv", format: "csv" }])
78
+ ).rejects.toThrow("File not found");
79
+ });
80
+
81
+ it("ingests a Parquet file", async () => {
82
+ const dir = createTmpDir();
83
+ // Create a Parquet file using DuckDB
84
+ // eslint-disable-next-line @typescript-eslint/no-require-imports
85
+ const { DuckDBInstance } = require("@duckdb/node-api");
86
+ const inst = await DuckDBInstance.create(":memory:");
87
+ const c = await inst.connect();
88
+ const parquetPath = path.join(dir, "data.parquet");
89
+ // eslint-disable-next-line @typescript-eslint/no-explicit-any
90
+ await (c as any).run(`COPY (SELECT 1 AS id, 'test' AS name) TO '${parquetPath.replace(/'/g, "''")}' (FORMAT PARQUET)`);
91
+ // eslint-disable-next-line @typescript-eslint/no-explicit-any
92
+ (c as any).disconnectSync();
93
+ // eslint-disable-next-line @typescript-eslint/no-explicit-any
94
+ (inst as any).closeSync();
95
+
96
+ const dbPath = path.join(dir, "test.duckdb");
97
+ const tables = await ingestIntoDuckDB(dbPath, [{ path: parquetPath, format: "parquet" }]);
98
+ expect(tables).toEqual(["data"]);
99
+ });
100
+ });
101
+
102
+ describe("listDuckDBObjects", () => {
103
+ it("lists tables in a DuckDB database", async () => {
104
+ const dir = createTmpDir();
105
+ const csvPath = path.join(dir, "data.csv");
106
+ fs.writeFileSync(csvPath, "x,y\n1,2\n3,4\n");
107
+
108
+ const dbPath = path.join(dir, "test.duckdb");
109
+ await ingestIntoDuckDB(dbPath, [{ path: csvPath, format: "csv" }]);
110
+
111
+ const objects = await listDuckDBObjects(dbPath);
112
+ expect(objects).toHaveLength(1);
113
+ expect(objects[0].name).toBe("data");
114
+ expect(objects[0].type).toBe("table");
115
+ });
116
+ });
117
+
118
+ describe("profileDuckDB", () => {
119
+ it("profiles a table with correct row count and columns", async () => {
120
+ const dir = createTmpDir();
121
+ const csvPath = path.join(dir, "employees.csv");
122
+ fs.writeFileSync(
123
+ csvPath,
124
+ "id,name,department,salary\n" +
125
+ "1,Alice,Engineering,120000\n" +
126
+ "2,Bob,Sales,95000\n" +
127
+ "3,Charlie,Engineering,115000\n" +
128
+ "4,Diana,Marketing,88000\n" +
129
+ "5,Eve,Sales,92000\n"
130
+ );
131
+
132
+ const dbPath = path.join(dir, "test.duckdb");
133
+ await ingestIntoDuckDB(dbPath, [{ path: csvPath, format: "csv" }]);
134
+
135
+ const profiles = await profileDuckDB(dbPath);
136
+ expect(profiles).toHaveLength(1);
137
+
138
+ const profile = profiles[0];
139
+ expect(profile.table_name).toBe("employees");
140
+ expect(profile.row_count).toBe(5);
141
+ expect(profile.columns.length).toBe(4);
142
+
143
+ const nameCol = profile.columns.find((c) => c.name === "name");
144
+ expect(nameCol).toBeDefined();
145
+ expect(nameCol!.unique_count).toBe(5);
146
+
147
+ const deptCol = profile.columns.find((c) => c.name === "department");
148
+ expect(deptCol).toBeDefined();
149
+ // 3 unique departments with 5 rows = 60% cardinality, but <=10 unique → enum-like
150
+ expect(deptCol!.is_enum_like).toBe(true);
151
+ expect(deptCol!.sample_values.length).toBeGreaterThan(0);
152
+ });
153
+
154
+ it("respects filterTables parameter", async () => {
155
+ const dir = createTmpDir();
156
+ const csv1 = path.join(dir, "a.csv");
157
+ const csv2 = path.join(dir, "b.csv");
158
+ fs.writeFileSync(csv1, "x\n1\n2\n");
159
+ fs.writeFileSync(csv2, "y\n3\n4\n");
160
+
161
+ const dbPath = path.join(dir, "test.duckdb");
162
+ await ingestIntoDuckDB(dbPath, [
163
+ { path: csv1, format: "csv" },
164
+ { path: csv2, format: "csv" },
165
+ ]);
166
+
167
+ const profiles = await profileDuckDB(dbPath, ["a"]);
168
+ expect(profiles).toHaveLength(1);
169
+ expect(profiles[0].table_name).toBe("a");
170
+ });
171
+ });
@@ -0,0 +1,434 @@
1
+ import { afterEach, describe, expect, test } from "bun:test";
2
+ import * as fs from "fs";
3
+ import * as path from "path";
4
+ import * as os from "os";
5
+ import {
6
+ loadEvalCases,
7
+ validateCase,
8
+ filterCases,
9
+ computeSummary,
10
+ detectRegressions,
11
+ type EvalCase,
12
+ type EvalResult,
13
+ } from "../eval";
14
+
15
+ // ---------------------------------------------------------------------------
16
+ // Helper: create a temp directory with YAML case files
17
+ // ---------------------------------------------------------------------------
18
+
19
+ function writeTempCases(
20
+ cases: Record<string, string>[],
21
+ schemaDir: string = "simple",
22
+ ): string {
23
+ const tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "eval-test-"));
24
+ const schemaDirPath = path.join(tmpDir, schemaDir);
25
+ fs.mkdirSync(schemaDirPath, { recursive: true });
26
+
27
+ for (const c of cases) {
28
+ const id = c.id ?? "test-001";
29
+ fs.writeFileSync(
30
+ path.join(schemaDirPath, `${id}.yml`),
31
+ Object.entries(c)
32
+ .map(([k, v]) => {
33
+ if (k === "tags") return `tags: [${v}]`;
34
+ if (k === "gold_sql") return `gold_sql: |\n ${v}`;
35
+ return `${k}: ${v}`;
36
+ })
37
+ .join("\n"),
38
+ );
39
+ }
40
+
41
+ return tmpDir;
42
+ }
43
+
44
+ // ---------------------------------------------------------------------------
45
+ // validateCase
46
+ // ---------------------------------------------------------------------------
47
+
48
+ describe("validateCase", () => {
49
+ test("rejects missing required field", () => {
50
+ expect(() =>
51
+ validateCase({ question: "test", schema: "simple", difficulty: "simple", category: "filter", gold_sql: "SELECT 1" }, "test.yml"),
52
+ ).toThrow('Missing required field "id"');
53
+ });
54
+
55
+ test("rejects invalid schema", () => {
56
+ expect(() =>
57
+ validateCase(
58
+ { id: "t-001", question: "test", schema: "invalid", difficulty: "simple", category: "filter", gold_sql: "SELECT 1" },
59
+ "test.yml",
60
+ ),
61
+ ).toThrow('Invalid schema "invalid"');
62
+ });
63
+
64
+ test("rejects invalid difficulty", () => {
65
+ expect(() =>
66
+ validateCase(
67
+ { id: "t-001", question: "test", schema: "simple", difficulty: "impossible", category: "filter", gold_sql: "SELECT 1" },
68
+ "test.yml",
69
+ ),
70
+ ).toThrow('Invalid difficulty "impossible"');
71
+ });
72
+
73
+ test("rejects empty id", () => {
74
+ expect(() =>
75
+ validateCase(
76
+ { id: " ", question: "test", schema: "simple", difficulty: "simple", category: "filter", gold_sql: "SELECT 1" },
77
+ "test.yml",
78
+ ),
79
+ ).toThrow("Invalid id");
80
+ });
81
+
82
+ test("rejects empty question", () => {
83
+ expect(() =>
84
+ validateCase(
85
+ { id: "t-001", question: "", schema: "simple", difficulty: "simple", category: "filter", gold_sql: "SELECT 1" },
86
+ "test.yml",
87
+ ),
88
+ ).toThrow("question");
89
+ });
90
+
91
+ test("rejects whitespace-only gold_sql", () => {
92
+ expect(() =>
93
+ validateCase(
94
+ { id: "t-001", question: "Q?", schema: "simple", difficulty: "simple", category: "filter", gold_sql: " " },
95
+ "test.yml",
96
+ ),
97
+ ).toThrow("gold_sql");
98
+ });
99
+
100
+ test("rejects numeric id (YAML auto-casts numbers)", () => {
101
+ expect(() =>
102
+ validateCase(
103
+ { id: 42, question: "Q?", schema: "simple", difficulty: "simple", category: "filter", gold_sql: "SELECT 1" },
104
+ "test.yml",
105
+ ),
106
+ ).toThrow("Invalid id");
107
+ });
108
+
109
+ test("accepts valid case", () => {
110
+ expect(() =>
111
+ validateCase(
112
+ { id: "t-001", question: "How many?", schema: "simple", difficulty: "simple", category: "filter", gold_sql: "SELECT 1" },
113
+ "test.yml",
114
+ ),
115
+ ).not.toThrow();
116
+ });
117
+ });
118
+
119
+ // ---------------------------------------------------------------------------
120
+ // loadEvalCases
121
+ // ---------------------------------------------------------------------------
122
+
123
+ describe("loadEvalCases", () => {
124
+ let tmpDir: string;
125
+
126
+ afterEach(() => {
127
+ if (tmpDir && fs.existsSync(tmpDir)) {
128
+ fs.rmSync(tmpDir, { recursive: true });
129
+ }
130
+ });
131
+
132
+ test("loads cases from YAML files", () => {
133
+ tmpDir = writeTempCases([
134
+ {
135
+ id: "sp-001",
136
+ question: "How many companies?",
137
+ schema: "simple",
138
+ difficulty: "simple",
139
+ category: "aggregation",
140
+ tags: "companies, count",
141
+ gold_sql: "SELECT COUNT(*) FROM companies",
142
+ },
143
+ ]);
144
+
145
+ const cases = loadEvalCases(tmpDir);
146
+ expect(cases).toHaveLength(1);
147
+ expect(cases[0].id).toBe("sp-001");
148
+ expect(cases[0].question).toBe("How many companies?");
149
+ expect(cases[0].schema).toBe("simple");
150
+ expect(cases[0].difficulty).toBe("simple");
151
+ expect(cases[0].category).toBe("aggregation");
152
+ expect(cases[0].gold_sql).toBe("SELECT COUNT(*) FROM companies");
153
+ });
154
+
155
+ test("loads from multiple schema directories", () => {
156
+ tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "eval-test-"));
157
+
158
+ // Simple case
159
+ const simpleDir = path.join(tmpDir, "simple");
160
+ fs.mkdirSync(simpleDir, { recursive: true });
161
+ fs.writeFileSync(
162
+ path.join(simpleDir, "sp-001.yml"),
163
+ 'id: sp-001\nquestion: "Q1"\nschema: simple\ndifficulty: simple\ncategory: filter\ntags: []\ngold_sql: "SELECT 1"',
164
+ );
165
+
166
+ // Cybersec case
167
+ const cybersecDir = path.join(tmpDir, "cybersec");
168
+ fs.mkdirSync(cybersecDir, { recursive: true });
169
+ fs.writeFileSync(
170
+ path.join(cybersecDir, "cs-001.yml"),
171
+ 'id: cs-001\nquestion: "Q2"\nschema: cybersec\ndifficulty: medium\ncategory: join\ntags: []\ngold_sql: "SELECT 2"',
172
+ );
173
+
174
+ const cases = loadEvalCases(tmpDir);
175
+ expect(cases).toHaveLength(2);
176
+ const schemas = cases.map(c => c.schema).sort();
177
+ expect(schemas).toEqual(["cybersec", "simple"]);
178
+ });
179
+
180
+ test("throws on missing directory", () => {
181
+ expect(() => loadEvalCases("/nonexistent/path")).toThrow("Eval cases directory not found");
182
+ });
183
+
184
+ test("returns empty for schema directory with no YAML files", () => {
185
+ tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "eval-test-"));
186
+ fs.mkdirSync(path.join(tmpDir, "simple"), { recursive: true });
187
+ const cases = loadEvalCases(tmpDir);
188
+ expect(cases).toHaveLength(0);
189
+ });
190
+
191
+ test("defaults tags to empty array when missing from YAML", () => {
192
+ tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "eval-test-"));
193
+ const dir = path.join(tmpDir, "simple");
194
+ fs.mkdirSync(dir, { recursive: true });
195
+ fs.writeFileSync(
196
+ path.join(dir, "sp-001.yml"),
197
+ 'id: sp-001\nquestion: "Q1"\nschema: simple\ndifficulty: simple\ncategory: filter\ngold_sql: "SELECT 1"',
198
+ );
199
+ const cases = loadEvalCases(tmpDir);
200
+ expect(cases).toHaveLength(1);
201
+ expect(cases[0].tags).toEqual([]);
202
+ });
203
+
204
+ test("rejects duplicate case IDs across files", () => {
205
+ tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "eval-test-"));
206
+ const dir = path.join(tmpDir, "simple");
207
+ fs.mkdirSync(dir, { recursive: true });
208
+ const yaml = 'id: dup-001\nquestion: "Q"\nschema: simple\ndifficulty: simple\ncategory: filter\ntags: []\ngold_sql: "SELECT 1"';
209
+ fs.writeFileSync(path.join(dir, "a.yml"), yaml);
210
+ fs.writeFileSync(path.join(dir, "b.yml"), yaml);
211
+ expect(() => loadEvalCases(tmpDir)).toThrow('Duplicate eval case id "dup-001"');
212
+ });
213
+
214
+ test("throws on invalid case", () => {
215
+ tmpDir = writeTempCases([
216
+ {
217
+ id: "bad-001",
218
+ question: "Missing schema",
219
+ difficulty: "simple",
220
+ category: "filter",
221
+ gold_sql: "SELECT 1",
222
+ } as unknown as Record<string, string>,
223
+ ]);
224
+
225
+ expect(() => loadEvalCases(tmpDir)).toThrow('Missing required field "schema"');
226
+ });
227
+ });
228
+
229
+ // ---------------------------------------------------------------------------
230
+ // filterCases
231
+ // ---------------------------------------------------------------------------
232
+
233
+ describe("filterCases", () => {
234
+ const cases: EvalCase[] = [
235
+ { id: "sp-001", question: "Q1", schema: "simple", difficulty: "simple", category: "aggregation", tags: [], gold_sql: "SELECT 1" },
236
+ { id: "sp-002", question: "Q2", schema: "simple", difficulty: "medium", category: "join", tags: [], gold_sql: "SELECT 2" },
237
+ { id: "cs-001", question: "Q3", schema: "cybersec", difficulty: "simple", category: "filter", tags: [], gold_sql: "SELECT 3" },
238
+ { id: "cs-002", question: "Q4", schema: "cybersec", difficulty: "complex", category: "aggregation", tags: [], gold_sql: "SELECT 4" },
239
+ { id: "ec-001", question: "Q5", schema: "ecommerce", difficulty: "medium", category: "timeseries", tags: [], gold_sql: "SELECT 5" },
240
+ { id: "sp-skip", question: "Q6", schema: "simple", difficulty: "simple", category: "filter", tags: [], gold_sql: "SELECT 6", skip: true },
241
+ ];
242
+
243
+ test("filters by schema", () => {
244
+ const result = filterCases(cases, { schema: "cybersec" });
245
+ expect(result).toHaveLength(2);
246
+ expect(result.every(c => c.schema === "cybersec")).toBe(true);
247
+ });
248
+
249
+ test("filters by category", () => {
250
+ const result = filterCases(cases, { category: "aggregation" });
251
+ expect(result).toHaveLength(2);
252
+ expect(result.every(c => c.category === "aggregation")).toBe(true);
253
+ });
254
+
255
+ test("filters by difficulty", () => {
256
+ const result = filterCases(cases, { difficulty: "simple" });
257
+ expect(result).toHaveLength(2);
258
+ expect(result.every(c => c.difficulty === "simple")).toBe(true);
259
+ });
260
+
261
+ test("filters by id", () => {
262
+ const result = filterCases(cases, { id: "cs-001" });
263
+ expect(result).toHaveLength(1);
264
+ expect(result[0].id).toBe("cs-001");
265
+ });
266
+
267
+ test("applies limit", () => {
268
+ const result = filterCases(cases, { limit: 2 });
269
+ expect(result).toHaveLength(2);
270
+ });
271
+
272
+ test("excludes skipped cases", () => {
273
+ const result = filterCases(cases, {});
274
+ expect(result.find(c => c.id === "sp-skip")).toBeUndefined();
275
+ expect(result).toHaveLength(5);
276
+ });
277
+
278
+ test("combines filters", () => {
279
+ const result = filterCases(cases, { schema: "simple", difficulty: "simple" });
280
+ expect(result).toHaveLength(1);
281
+ expect(result[0].id).toBe("sp-001");
282
+ });
283
+
284
+ test("limit of 0 returns all cases (no slicing)", () => {
285
+ const result = filterCases(cases, { limit: 0 });
286
+ expect(result).toHaveLength(5);
287
+ });
288
+
289
+ test("returns empty on no match", () => {
290
+ const result = filterCases(cases, { schema: "nonexistent" });
291
+ expect(result).toHaveLength(0);
292
+ });
293
+ });
294
+
295
+ // ---------------------------------------------------------------------------
296
+ // computeSummary
297
+ // ---------------------------------------------------------------------------
298
+
299
+ describe("computeSummary", () => {
300
+ const results: EvalResult[] = [
301
+ { id: "sp-001", schema: "simple", question: "Q1", category: "aggregation", difficulty: "simple", tags: [], gold_sql: "", predicted_sql: "SELECT 1", match: true, error: null, latency_ms: 1000, tokens: 500, steps: 3 },
302
+ { id: "sp-002", schema: "simple", question: "Q2", category: "join", difficulty: "medium", tags: [], gold_sql: "", predicted_sql: "SELECT 2", match: false, error: null, latency_ms: 2000, tokens: 800, steps: 5 },
303
+ { id: "cs-001", schema: "cybersec", question: "Q3", category: "filter", difficulty: "simple", tags: [], gold_sql: "", predicted_sql: null, match: false, error: "timeout", latency_ms: 30000, tokens: 0, steps: 0 },
304
+ { id: "cs-002", schema: "cybersec", question: "Q4", category: "aggregation", difficulty: "complex", tags: [], gold_sql: "", predicted_sql: "SELECT 4", match: true, error: null, latency_ms: 3000, tokens: 1200, steps: 7 },
305
+ ];
306
+
307
+ test("computes overall accuracy", () => {
308
+ const summary = computeSummary(results);
309
+ expect(summary.total).toBe(4);
310
+ expect(summary.correct).toBe(2);
311
+ expect(summary.errors).toBe(1);
312
+ expect(summary.accuracy).toBe(50);
313
+ });
314
+
315
+ test("computes per-schema breakdown", () => {
316
+ const summary = computeSummary(results);
317
+ expect(summary.bySchema.get("simple")).toEqual({ total: 2, correct: 1 });
318
+ expect(summary.bySchema.get("cybersec")).toEqual({ total: 2, correct: 1 });
319
+ });
320
+
321
+ test("computes per-category breakdown", () => {
322
+ const summary = computeSummary(results);
323
+ expect(summary.byCategory.get("aggregation")).toEqual({ total: 2, correct: 2 });
324
+ expect(summary.byCategory.get("join")).toEqual({ total: 1, correct: 0 });
325
+ expect(summary.byCategory.get("filter")).toEqual({ total: 1, correct: 0 });
326
+ });
327
+
328
+ test("computes per-difficulty breakdown", () => {
329
+ const summary = computeSummary(results);
330
+ expect(summary.byDifficulty.get("simple")).toEqual({ total: 2, correct: 1 });
331
+ expect(summary.byDifficulty.get("medium")).toEqual({ total: 1, correct: 0 });
332
+ expect(summary.byDifficulty.get("complex")).toEqual({ total: 1, correct: 1 });
333
+ });
334
+
335
+ test("computes token and latency stats", () => {
336
+ const summary = computeSummary(results);
337
+ expect(summary.totalTokens).toBe(2500);
338
+ expect(summary.totalLatencyMs).toBe(36000);
339
+ expect(summary.avgLatencyMs).toBe(9000);
340
+ });
341
+
342
+ test("handles empty results", () => {
343
+ const summary = computeSummary([]);
344
+ expect(summary.total).toBe(0);
345
+ expect(summary.correct).toBe(0);
346
+ expect(summary.accuracy).toBe(0);
347
+ expect(summary.avgLatencyMs).toBe(0);
348
+ });
349
+ });
350
+
351
+ // ---------------------------------------------------------------------------
352
+ // detectRegressions
353
+ // ---------------------------------------------------------------------------
354
+
355
+ describe("detectRegressions", () => {
356
+ const baseline: EvalResult[] = [
357
+ { id: "sp-001", schema: "simple", question: "Q1", category: "agg", difficulty: "simple", tags: [], gold_sql: "", predicted_sql: "S1", match: true, error: null, latency_ms: 1000, tokens: 500, steps: 3 },
358
+ { id: "sp-002", schema: "simple", question: "Q2", category: "join", difficulty: "medium", tags: [], gold_sql: "", predicted_sql: "S2", match: false, error: null, latency_ms: 2000, tokens: 800, steps: 5 },
359
+ { id: "cs-001", schema: "cybersec", question: "Q3", category: "filter", difficulty: "simple", tags: [], gold_sql: "", predicted_sql: "S3", match: true, error: null, latency_ms: 1500, tokens: 600, steps: 4 },
360
+ ];
361
+
362
+ test("detects regression (was pass, now fail)", () => {
363
+ const current: EvalResult[] = [
364
+ { ...baseline[0], match: false, error: "wrong result" }, // sp-001: was PASS → now FAIL
365
+ { ...baseline[1], match: false }, // sp-002: was FAIL → still FAIL
366
+ { ...baseline[2], match: true }, // cs-001: was PASS → still PASS
367
+ ];
368
+
369
+ const report = detectRegressions(current, baseline);
370
+ expect(report.regressions).toHaveLength(1);
371
+ expect(report.regressions[0].id).toBe("sp-001");
372
+ expect(report.newPasses).toHaveLength(0);
373
+ expect(report.newCases).toHaveLength(0);
374
+ expect(report.stable).toBe(2);
375
+ });
376
+
377
+ test("detects new pass (was fail, now pass)", () => {
378
+ const current: EvalResult[] = [
379
+ { ...baseline[0], match: true }, // sp-001: still PASS
380
+ { ...baseline[1], match: true }, // sp-002: was FAIL → now PASS
381
+ { ...baseline[2], match: true }, // cs-001: still PASS
382
+ ];
383
+
384
+ const report = detectRegressions(current, baseline);
385
+ expect(report.regressions).toHaveLength(0);
386
+ expect(report.newPasses).toHaveLength(1);
387
+ expect(report.newPasses[0].id).toBe("sp-002");
388
+ expect(report.stable).toBe(2);
389
+ });
390
+
391
+ test("detects new cases (not in baseline)", () => {
392
+ const current: EvalResult[] = [
393
+ { ...baseline[0], match: true },
394
+ { ...baseline[1], match: false },
395
+ { ...baseline[2], match: true },
396
+ { id: "ec-001", schema: "ecommerce", question: "New Q", category: "agg", difficulty: "simple", tags: [], gold_sql: "", predicted_sql: "S4", match: true, error: null, latency_ms: 1000, tokens: 500, steps: 3 },
397
+ ];
398
+
399
+ const report = detectRegressions(current, baseline);
400
+ expect(report.newCases).toHaveLength(1);
401
+ expect(report.newCases[0].id).toBe("ec-001");
402
+ expect(report.stable).toBe(3);
403
+ });
404
+
405
+ test("handles empty baseline", () => {
406
+ const current: EvalResult[] = [
407
+ { ...baseline[0], match: true },
408
+ ];
409
+
410
+ const report = detectRegressions(current, []);
411
+ expect(report.newCases).toHaveLength(1);
412
+ expect(report.regressions).toHaveLength(0);
413
+ expect(report.stable).toBe(0);
414
+ });
415
+
416
+ test("handles empty current", () => {
417
+ const report = detectRegressions([], baseline);
418
+ expect(report.regressions).toHaveLength(0);
419
+ expect(report.newPasses).toHaveLength(0);
420
+ expect(report.newCases).toHaveLength(0);
421
+ expect(report.stable).toBe(0);
422
+ });
423
+
424
+ test("baseline cases absent from current are silently ignored", () => {
425
+ const current: EvalResult[] = [
426
+ { ...baseline[0], match: true },
427
+ ];
428
+ const report = detectRegressions(current, baseline);
429
+ expect(report.regressions).toHaveLength(0);
430
+ expect(report.newPasses).toHaveLength(0);
431
+ expect(report.newCases).toHaveLength(0);
432
+ expect(report.stable).toBe(1);
433
+ });
434
+ });