ai-functions 0.3.0 → 0.4.0

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 (400) hide show
  1. package/.turbo/turbo-build.log +5 -0
  2. package/.turbo/turbo-test.log +105 -0
  3. package/README.md +190 -86
  4. package/TODO.md +138 -0
  5. package/dist/ai-promise.d.ts +219 -0
  6. package/dist/ai-promise.d.ts.map +1 -0
  7. package/dist/ai-promise.js +610 -0
  8. package/dist/ai-promise.js.map +1 -0
  9. package/dist/ai.d.ts +285 -0
  10. package/dist/ai.d.ts.map +1 -0
  11. package/dist/ai.js +842 -0
  12. package/dist/ai.js.map +1 -0
  13. package/dist/batch/anthropic.d.ts +23 -0
  14. package/dist/batch/anthropic.d.ts.map +1 -0
  15. package/dist/batch/anthropic.js +257 -0
  16. package/dist/batch/anthropic.js.map +1 -0
  17. package/dist/batch/bedrock.d.ts +64 -0
  18. package/dist/batch/bedrock.d.ts.map +1 -0
  19. package/dist/batch/bedrock.js +586 -0
  20. package/dist/batch/bedrock.js.map +1 -0
  21. package/dist/batch/cloudflare.d.ts +37 -0
  22. package/dist/batch/cloudflare.d.ts.map +1 -0
  23. package/dist/batch/cloudflare.js +289 -0
  24. package/dist/batch/cloudflare.js.map +1 -0
  25. package/dist/batch/google.d.ts +41 -0
  26. package/dist/batch/google.d.ts.map +1 -0
  27. package/dist/batch/google.js +360 -0
  28. package/dist/batch/google.js.map +1 -0
  29. package/dist/batch/index.d.ts +31 -0
  30. package/dist/batch/index.d.ts.map +1 -0
  31. package/dist/batch/index.js +31 -0
  32. package/dist/batch/index.js.map +1 -0
  33. package/dist/batch/memory.d.ts +44 -0
  34. package/dist/batch/memory.d.ts.map +1 -0
  35. package/dist/batch/memory.js +188 -0
  36. package/dist/batch/memory.js.map +1 -0
  37. package/dist/batch/openai.d.ts +37 -0
  38. package/dist/batch/openai.d.ts.map +1 -0
  39. package/dist/batch/openai.js +403 -0
  40. package/dist/batch/openai.js.map +1 -0
  41. package/dist/batch-map.d.ts +125 -0
  42. package/dist/batch-map.d.ts.map +1 -0
  43. package/dist/batch-map.js +406 -0
  44. package/dist/batch-map.js.map +1 -0
  45. package/dist/batch-queue.d.ts +273 -0
  46. package/dist/batch-queue.d.ts.map +1 -0
  47. package/dist/batch-queue.js +271 -0
  48. package/dist/batch-queue.js.map +1 -0
  49. package/dist/context.d.ts +133 -0
  50. package/dist/context.d.ts.map +1 -0
  51. package/dist/context.js +267 -0
  52. package/dist/context.js.map +1 -0
  53. package/dist/embeddings.d.ts +123 -0
  54. package/dist/embeddings.d.ts.map +1 -0
  55. package/dist/embeddings.js +170 -0
  56. package/dist/embeddings.js.map +1 -0
  57. package/dist/eval/index.d.ts +8 -0
  58. package/dist/eval/index.d.ts.map +1 -0
  59. package/dist/eval/index.js +8 -0
  60. package/dist/eval/index.js.map +1 -0
  61. package/dist/eval/models.d.ts +66 -0
  62. package/dist/eval/models.d.ts.map +1 -0
  63. package/dist/eval/models.js +120 -0
  64. package/dist/eval/models.js.map +1 -0
  65. package/dist/eval/runner.d.ts +64 -0
  66. package/dist/eval/runner.d.ts.map +1 -0
  67. package/dist/eval/runner.js +148 -0
  68. package/dist/eval/runner.js.map +1 -0
  69. package/dist/generate.d.ts +168 -0
  70. package/dist/generate.d.ts.map +1 -0
  71. package/dist/generate.js +174 -0
  72. package/dist/generate.js.map +1 -0
  73. package/dist/index.d.ts +29 -4
  74. package/dist/index.d.ts.map +1 -1
  75. package/dist/index.js +53 -52
  76. package/dist/index.js.map +1 -1
  77. package/dist/primitives.d.ts +292 -0
  78. package/dist/primitives.d.ts.map +1 -0
  79. package/dist/primitives.js +471 -0
  80. package/dist/primitives.js.map +1 -0
  81. package/dist/providers/cloudflare.d.ts +9 -0
  82. package/dist/providers/cloudflare.d.ts.map +1 -0
  83. package/dist/providers/cloudflare.js +9 -0
  84. package/dist/providers/cloudflare.js.map +1 -0
  85. package/dist/providers/index.d.ts +9 -0
  86. package/dist/providers/index.d.ts.map +1 -0
  87. package/dist/providers/index.js +9 -0
  88. package/dist/providers/index.js.map +1 -0
  89. package/dist/schema.d.ts +54 -0
  90. package/dist/schema.d.ts.map +1 -0
  91. package/dist/schema.js +109 -0
  92. package/dist/schema.js.map +1 -0
  93. package/dist/template.d.ts +73 -0
  94. package/dist/template.d.ts.map +1 -0
  95. package/dist/template.js +129 -0
  96. package/dist/template.js.map +1 -0
  97. package/dist/types.d.ts +474 -106
  98. package/dist/types.d.ts.map +1 -1
  99. package/dist/types.js +4 -8
  100. package/dist/types.js.map +1 -1
  101. package/evalite.config.ts +19 -0
  102. package/evals/README.md +212 -0
  103. package/evals/classification.eval.ts +108 -0
  104. package/evals/marketing.eval.ts +370 -0
  105. package/evals/math.eval.ts +94 -0
  106. package/evals/run-evals.ts +166 -0
  107. package/evals/structured-output.eval.ts +143 -0
  108. package/evals/writing.eval.ts +117 -0
  109. package/examples/batch-blog-posts.ts +160 -0
  110. package/package.json +57 -57
  111. package/src/ai-promise.ts +784 -0
  112. package/src/ai.ts +1183 -0
  113. package/src/batch/anthropic.ts +375 -0
  114. package/src/batch/bedrock.ts +801 -0
  115. package/src/batch/cloudflare.ts +421 -0
  116. package/src/batch/google.ts +491 -0
  117. package/src/batch/index.ts +31 -0
  118. package/src/batch/memory.ts +253 -0
  119. package/src/batch/openai.ts +557 -0
  120. package/src/batch-map.ts +534 -0
  121. package/src/batch-queue.ts +493 -0
  122. package/src/context.ts +332 -0
  123. package/src/embeddings.ts +244 -0
  124. package/src/eval/index.ts +8 -0
  125. package/src/eval/models.ts +158 -0
  126. package/src/eval/runner.ts +217 -0
  127. package/src/generate.ts +245 -0
  128. package/src/index.ts +154 -0
  129. package/src/primitives.ts +612 -0
  130. package/src/providers/cloudflare.ts +15 -0
  131. package/src/providers/index.ts +14 -0
  132. package/src/schema.ts +147 -0
  133. package/src/template.ts +209 -0
  134. package/src/types.ts +540 -0
  135. package/test/README.md +105 -0
  136. package/test/ai-proxy.test.ts +192 -0
  137. package/test/async-iterators.test.ts +327 -0
  138. package/test/batch-background.test.ts +482 -0
  139. package/test/batch-blog-posts.test.ts +387 -0
  140. package/test/blog-generation.test.ts +510 -0
  141. package/test/browse-read.test.ts +611 -0
  142. package/test/core-functions.test.ts +694 -0
  143. package/test/decide.test.ts +393 -0
  144. package/test/define.test.ts +274 -0
  145. package/test/e2e-bedrock-manual.ts +163 -0
  146. package/test/e2e-bedrock.test.ts +191 -0
  147. package/test/e2e-flex-gateway.ts +157 -0
  148. package/test/e2e-flex-manual.ts +183 -0
  149. package/test/e2e-flex.test.ts +209 -0
  150. package/test/e2e-google-manual.ts +178 -0
  151. package/test/e2e-google.test.ts +216 -0
  152. package/test/embeddings.test.ts +284 -0
  153. package/test/evals/define-function.eval.test.ts +379 -0
  154. package/test/evals/primitives.eval.test.ts +384 -0
  155. package/test/function-types.test.ts +492 -0
  156. package/test/generate-core.test.ts +319 -0
  157. package/test/generate.test.ts +163 -0
  158. package/test/implicit-batch.test.ts +422 -0
  159. package/test/schema.test.ts +109 -0
  160. package/test/tagged-templates.test.ts +302 -0
  161. package/tsconfig.json +10 -0
  162. package/vitest.config.ts +42 -0
  163. package/LICENSE +0 -21
  164. package/bin/cli.js +0 -5
  165. package/dist/cli/index.d.ts +0 -10
  166. package/dist/cli/index.d.ts.map +0 -1
  167. package/dist/cli/index.js +0 -38
  168. package/dist/cli/index.js.map +0 -1
  169. package/dist/cli/index.test.d.ts +0 -2
  170. package/dist/cli/index.test.d.ts.map +0 -1
  171. package/dist/cli/index.test.js +0 -35
  172. package/dist/cli/index.test.js.map +0 -1
  173. package/dist/constants/models.d.ts +0 -10
  174. package/dist/constants/models.d.ts.map +0 -1
  175. package/dist/constants/models.js +0 -12
  176. package/dist/constants/models.js.map +0 -1
  177. package/dist/converters/index.d.ts +0 -3
  178. package/dist/converters/index.d.ts.map +0 -1
  179. package/dist/converters/index.js +0 -3
  180. package/dist/converters/index.js.map +0 -1
  181. package/dist/converters/model.d.ts +0 -4
  182. package/dist/converters/model.d.ts.map +0 -1
  183. package/dist/converters/model.js +0 -19
  184. package/dist/converters/model.js.map +0 -1
  185. package/dist/converters/schema.d.ts +0 -4
  186. package/dist/converters/schema.d.ts.map +0 -1
  187. package/dist/converters/schema.js +0 -25
  188. package/dist/converters/schema.js.map +0 -1
  189. package/dist/core/responses.d.ts +0 -5
  190. package/dist/core/responses.d.ts.map +0 -1
  191. package/dist/core/responses.js +0 -16
  192. package/dist/core/responses.js.map +0 -1
  193. package/dist/core/responses.test.d.ts +0 -2
  194. package/dist/core/responses.test.d.ts.map +0 -1
  195. package/dist/core/responses.test.js +0 -31
  196. package/dist/core/responses.test.js.map +0 -1
  197. package/dist/errors.d.ts +0 -6
  198. package/dist/errors.d.ts.map +0 -1
  199. package/dist/errors.js +0 -9
  200. package/dist/errors.js.map +0 -1
  201. package/dist/examples/streaming.test.d.ts +0 -2
  202. package/dist/examples/streaming.test.d.ts.map +0 -1
  203. package/dist/examples/streaming.test.js +0 -176
  204. package/dist/examples/streaming.test.js.map +0 -1
  205. package/dist/factory/__tests__/index.test.d.ts +0 -2
  206. package/dist/factory/__tests__/index.test.d.ts.map +0 -1
  207. package/dist/factory/__tests__/index.test.js +0 -430
  208. package/dist/factory/__tests__/index.test.js.map +0 -1
  209. package/dist/factory/__tests__/list.test.d.ts +0 -2
  210. package/dist/factory/__tests__/list.test.d.ts.map +0 -1
  211. package/dist/factory/__tests__/list.test.js +0 -92
  212. package/dist/factory/__tests__/list.test.js.map +0 -1
  213. package/dist/factory/index.d.ts +0 -20
  214. package/dist/factory/index.d.ts.map +0 -1
  215. package/dist/factory/index.js +0 -287
  216. package/dist/factory/index.js.map +0 -1
  217. package/dist/factory/index.test.d.ts +0 -2
  218. package/dist/factory/index.test.d.ts.map +0 -1
  219. package/dist/factory/index.test.js +0 -287
  220. package/dist/factory/index.test.js.map +0 -1
  221. package/dist/factory/list.d.ts +0 -3
  222. package/dist/factory/list.d.ts.map +0 -1
  223. package/dist/factory/list.js +0 -221
  224. package/dist/factory/list.js.map +0 -1
  225. package/dist/factory/list.test.d.ts +0 -2
  226. package/dist/factory/list.test.d.ts.map +0 -1
  227. package/dist/factory/list.test.js +0 -84
  228. package/dist/factory/list.test.js.map +0 -1
  229. package/dist/generate/index.d.ts +0 -5
  230. package/dist/generate/index.d.ts.map +0 -1
  231. package/dist/generate/index.js +0 -17
  232. package/dist/generate/index.js.map +0 -1
  233. package/dist/index.test.d.ts +0 -2
  234. package/dist/index.test.d.ts.map +0 -1
  235. package/dist/index.test.js +0 -59
  236. package/dist/index.test.js.map +0 -1
  237. package/dist/list/await.d.ts +0 -3
  238. package/dist/list/await.d.ts.map +0 -1
  239. package/dist/list/await.js +0 -28
  240. package/dist/list/await.js.map +0 -1
  241. package/dist/list/constants.d.ts +0 -4
  242. package/dist/list/constants.d.ts.map +0 -1
  243. package/dist/list/constants.js +0 -5
  244. package/dist/list/constants.js.map +0 -1
  245. package/dist/list/create-function.d.ts +0 -3
  246. package/dist/list/create-function.d.ts.map +0 -1
  247. package/dist/list/create-function.js +0 -11
  248. package/dist/list/create-function.js.map +0 -1
  249. package/dist/list/index.d.ts +0 -4
  250. package/dist/list/index.d.ts.map +0 -1
  251. package/dist/list/index.js +0 -5
  252. package/dist/list/index.js.map +0 -1
  253. package/dist/list/prompt.d.ts +0 -3
  254. package/dist/list/prompt.d.ts.map +0 -1
  255. package/dist/list/prompt.js +0 -6
  256. package/dist/list/prompt.js.map +0 -1
  257. package/dist/list/schemas.d.ts +0 -4
  258. package/dist/list/schemas.d.ts.map +0 -1
  259. package/dist/list/schemas.js +0 -8
  260. package/dist/list/schemas.js.map +0 -1
  261. package/dist/list/stream.d.ts +0 -3
  262. package/dist/list/stream.d.ts.map +0 -1
  263. package/dist/list/stream.js +0 -33
  264. package/dist/list/stream.js.map +0 -1
  265. package/dist/list/types.d.ts +0 -11
  266. package/dist/list/types.d.ts.map +0 -1
  267. package/dist/list/types.js +0 -2
  268. package/dist/list/types.js.map +0 -1
  269. package/dist/list/validation.d.ts +0 -3
  270. package/dist/list/validation.d.ts.map +0 -1
  271. package/dist/list/validation.js +0 -12
  272. package/dist/list/validation.js.map +0 -1
  273. package/dist/providers/config.d.ts +0 -4
  274. package/dist/providers/config.d.ts.map +0 -1
  275. package/dist/providers/config.js +0 -21
  276. package/dist/providers/config.js.map +0 -1
  277. package/dist/providers/config.test.d.ts +0 -2
  278. package/dist/providers/config.test.d.ts.map +0 -1
  279. package/dist/providers/config.test.js +0 -37
  280. package/dist/providers/config.test.js.map +0 -1
  281. package/dist/proxy/constants.d.ts +0 -4
  282. package/dist/proxy/constants.d.ts.map +0 -1
  283. package/dist/proxy/constants.js +0 -5
  284. package/dist/proxy/constants.js.map +0 -1
  285. package/dist/proxy/create-function.d.ts +0 -4
  286. package/dist/proxy/create-function.d.ts.map +0 -1
  287. package/dist/proxy/create-function.js +0 -24
  288. package/dist/proxy/create-function.js.map +0 -1
  289. package/dist/proxy/create-proxy.d.ts +0 -2
  290. package/dist/proxy/create-proxy.d.ts.map +0 -1
  291. package/dist/proxy/create-proxy.js +0 -11
  292. package/dist/proxy/create-proxy.js.map +0 -1
  293. package/dist/proxy/function-generator.d.ts +0 -9
  294. package/dist/proxy/function-generator.d.ts.map +0 -1
  295. package/dist/proxy/function-generator.js +0 -29
  296. package/dist/proxy/function-generator.js.map +0 -1
  297. package/dist/proxy/index.d.ts +0 -4
  298. package/dist/proxy/index.d.ts.map +0 -1
  299. package/dist/proxy/index.js +0 -4
  300. package/dist/proxy/index.js.map +0 -1
  301. package/dist/proxy/prompt.d.ts +0 -2
  302. package/dist/proxy/prompt.d.ts.map +0 -1
  303. package/dist/proxy/prompt.js +0 -6
  304. package/dist/proxy/prompt.js.map +0 -1
  305. package/dist/proxy/types.d.ts +0 -7
  306. package/dist/proxy/types.d.ts.map +0 -1
  307. package/dist/proxy/types.js +0 -2
  308. package/dist/proxy/types.js.map +0 -1
  309. package/dist/queue/manager.d.ts +0 -5
  310. package/dist/queue/manager.d.ts.map +0 -1
  311. package/dist/queue/manager.js +0 -37
  312. package/dist/queue/manager.js.map +0 -1
  313. package/dist/queue/manager.test.d.ts +0 -2
  314. package/dist/queue/manager.test.d.ts.map +0 -1
  315. package/dist/queue/manager.test.js +0 -52
  316. package/dist/queue/manager.test.js.map +0 -1
  317. package/dist/schema-converter.d.ts +0 -4
  318. package/dist/schema-converter.d.ts.map +0 -1
  319. package/dist/schema-converter.js +0 -30
  320. package/dist/schema-converter.js.map +0 -1
  321. package/dist/stream/index.d.ts +0 -7
  322. package/dist/stream/index.d.ts.map +0 -1
  323. package/dist/stream/index.js +0 -23
  324. package/dist/stream/index.js.map +0 -1
  325. package/dist/streaming/utils.d.ts +0 -4
  326. package/dist/streaming/utils.d.ts.map +0 -1
  327. package/dist/streaming/utils.js +0 -131
  328. package/dist/streaming/utils.js.map +0 -1
  329. package/dist/streaming/utils.test.d.ts +0 -2
  330. package/dist/streaming/utils.test.d.ts.map +0 -1
  331. package/dist/streaming/utils.test.js +0 -84
  332. package/dist/streaming/utils.test.js.map +0 -1
  333. package/dist/templates/result.d.ts +0 -7
  334. package/dist/templates/result.d.ts.map +0 -1
  335. package/dist/templates/result.js +0 -40
  336. package/dist/templates/result.js.map +0 -1
  337. package/dist/templates/result.test.d.ts +0 -2
  338. package/dist/templates/result.test.d.ts.map +0 -1
  339. package/dist/templates/result.test.js +0 -75
  340. package/dist/templates/result.test.js.map +0 -1
  341. package/dist/test/setup.d.ts +0 -2
  342. package/dist/test/setup.d.ts.map +0 -1
  343. package/dist/test/setup.js +0 -21
  344. package/dist/test/setup.js.map +0 -1
  345. package/dist/test-types.d.ts +0 -13
  346. package/dist/test-types.d.ts.map +0 -1
  347. package/dist/test-types.js +0 -55
  348. package/dist/test-types.js.map +0 -1
  349. package/dist/types/index.d.ts +0 -4
  350. package/dist/types/index.d.ts.map +0 -1
  351. package/dist/types/index.js +0 -4
  352. package/dist/types/index.js.map +0 -1
  353. package/dist/types/list.d.ts +0 -10
  354. package/dist/types/list.d.ts.map +0 -1
  355. package/dist/types/list.js +0 -2
  356. package/dist/types/list.js.map +0 -1
  357. package/dist/types/model.d.ts +0 -7
  358. package/dist/types/model.d.ts.map +0 -1
  359. package/dist/types/model.js +0 -2
  360. package/dist/types/model.js.map +0 -1
  361. package/dist/types/options.d.ts +0 -25
  362. package/dist/types/options.d.ts.map +0 -1
  363. package/dist/types/options.js +0 -2
  364. package/dist/types/options.js.map +0 -1
  365. package/dist/types/schema.d.ts +0 -5
  366. package/dist/types/schema.d.ts.map +0 -1
  367. package/dist/types/schema.js +0 -2
  368. package/dist/types/schema.js.map +0 -1
  369. package/dist/utils/__tests__/request-handler.test.d.ts +0 -2
  370. package/dist/utils/__tests__/request-handler.test.d.ts.map +0 -1
  371. package/dist/utils/__tests__/request-handler.test.js +0 -134
  372. package/dist/utils/__tests__/request-handler.test.js.map +0 -1
  373. package/dist/utils/__tests__/schema.test.d.ts +0 -2
  374. package/dist/utils/__tests__/schema.test.d.ts.map +0 -1
  375. package/dist/utils/__tests__/schema.test.js +0 -49
  376. package/dist/utils/__tests__/schema.test.js.map +0 -1
  377. package/dist/utils/__tests__/stream-progress.test.d.ts +0 -2
  378. package/dist/utils/__tests__/stream-progress.test.d.ts.map +0 -1
  379. package/dist/utils/__tests__/stream-progress.test.js +0 -85
  380. package/dist/utils/__tests__/stream-progress.test.js.map +0 -1
  381. package/dist/utils/index.d.ts +0 -2
  382. package/dist/utils/index.d.ts.map +0 -1
  383. package/dist/utils/index.js +0 -2
  384. package/dist/utils/index.js.map +0 -1
  385. package/dist/utils/request-handler.d.ts +0 -17
  386. package/dist/utils/request-handler.d.ts.map +0 -1
  387. package/dist/utils/request-handler.js +0 -105
  388. package/dist/utils/request-handler.js.map +0 -1
  389. package/dist/utils/schema.d.ts +0 -11
  390. package/dist/utils/schema.d.ts.map +0 -1
  391. package/dist/utils/schema.js +0 -51
  392. package/dist/utils/schema.js.map +0 -1
  393. package/dist/utils/stream-progress.d.ts +0 -17
  394. package/dist/utils/stream-progress.d.ts.map +0 -1
  395. package/dist/utils/stream-progress.js +0 -86
  396. package/dist/utils/stream-progress.js.map +0 -1
  397. package/dist/utils/validation.d.ts +0 -3
  398. package/dist/utils/validation.d.ts.map +0 -1
  399. package/dist/utils/validation.js +0 -30
  400. package/dist/utils/validation.js.map +0 -1
@@ -1,2 +0,0 @@
1
- export {};
2
- //# sourceMappingURL=request-handler.test.d.ts.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"request-handler.test.d.ts","sourceRoot":"","sources":["../../../src/utils/__tests__/request-handler.test.ts"],"names":[],"mappings":""}
@@ -1,134 +0,0 @@
1
- import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
2
- import { createRequestHandler } from '../request-handler';
3
- import { AIRequestError } from '../../types';
4
- describe('RequestHandler', () => {
5
- beforeEach(() => {
6
- vi.useFakeTimers();
7
- });
8
- afterEach(() => {
9
- vi.useRealTimers();
10
- });
11
- it('should successfully execute an operation', async () => {
12
- const handler = createRequestHandler();
13
- const operation = vi.fn().mockResolvedValue('success');
14
- const promise = handler.execute(operation);
15
- await vi.runAllTimersAsync();
16
- const result = await promise;
17
- expect(result).toBe('success');
18
- expect(operation).toHaveBeenCalledTimes(1);
19
- });
20
- it('should retry on failure', async () => {
21
- const handler = createRequestHandler({
22
- requestHandling: {
23
- retry: {
24
- maxRetries: 2,
25
- initialDelay: 100,
26
- maxDelay: 1000,
27
- backoffFactor: 2,
28
- }
29
- }
30
- });
31
- const operation = vi.fn()
32
- .mockRejectedValueOnce(new Error('Temporary error'))
33
- .mockRejectedValueOnce(new Error('Another temporary error'))
34
- .mockResolvedValueOnce('success');
35
- const promise = handler.execute(operation);
36
- // Advance time for first retry
37
- await vi.advanceTimersByTimeAsync(100);
38
- // Advance time for second retry
39
- await vi.advanceTimersByTimeAsync(200);
40
- // Advance remaining timers
41
- await vi.runAllTimersAsync();
42
- const result = await promise;
43
- expect(result).toBe('success');
44
- expect(operation).toHaveBeenCalledTimes(3);
45
- });
46
- it('should respect rate limits', async () => {
47
- const handler = createRequestHandler({
48
- requestHandling: {
49
- rateLimit: {
50
- requestsPerMinute: 60,
51
- burstLimit: 2,
52
- }
53
- }
54
- });
55
- const operation = vi.fn().mockResolvedValue('success');
56
- // First two requests should be immediate
57
- const promise1 = handler.execute(operation);
58
- const promise2 = handler.execute(operation);
59
- await vi.runAllTimersAsync();
60
- await Promise.all([promise1, promise2]);
61
- // Third request should be delayed
62
- const startTime = Date.now();
63
- const promise3 = handler.execute(operation);
64
- await vi.advanceTimersByTimeAsync(1000);
65
- await vi.runAllTimersAsync();
66
- await promise3;
67
- const duration = Date.now() - startTime;
68
- expect(duration).toBeGreaterThanOrEqual(1000);
69
- expect(operation).toHaveBeenCalledTimes(3);
70
- });
71
- it('should handle timeouts', async () => {
72
- const timeout = 1000;
73
- const handler = createRequestHandler({
74
- requestHandling: {
75
- timeout,
76
- retry: {
77
- maxRetries: 0,
78
- initialDelay: 100,
79
- maxDelay: 1000,
80
- backoffFactor: 2
81
- }
82
- }
83
- });
84
- const operation = vi.fn(() => new Promise(resolve => {
85
- setTimeout(resolve, 2000, 'delayed result');
86
- }));
87
- const promise = handler.execute(operation);
88
- await vi.advanceTimersByTimeAsync(timeout);
89
- await vi.runAllTimersAsync();
90
- await expect(promise).rejects.toThrow(`Request timed out after ${timeout}ms`);
91
- });
92
- it('should not retry on non-retryable errors', async () => {
93
- const handler = createRequestHandler();
94
- const operation = vi.fn().mockRejectedValue(new AIRequestError('Non-retryable error', undefined, false));
95
- const promise = handler.execute(operation);
96
- await vi.runAllTimersAsync();
97
- await expect(promise).rejects.toThrow('Non-retryable error');
98
- expect(operation).toHaveBeenCalledTimes(1);
99
- });
100
- it('should respect max retries', async () => {
101
- const handler = createRequestHandler({
102
- requestHandling: {
103
- retry: {
104
- maxRetries: 2,
105
- initialDelay: 100,
106
- maxDelay: 1000,
107
- backoffFactor: 2,
108
- }
109
- }
110
- });
111
- const operation = vi.fn().mockRejectedValue(new Error('Persistent error'));
112
- const promise = handler.execute(operation);
113
- // Advance time for all retries
114
- await vi.advanceTimersByTimeAsync(100);
115
- await vi.advanceTimersByTimeAsync(200);
116
- await vi.advanceTimersByTimeAsync(400);
117
- await vi.runAllTimersAsync();
118
- await expect(promise).rejects.toThrow('Failed after 3 attempts');
119
- expect(operation).toHaveBeenCalledTimes(3);
120
- });
121
- it('should handle concurrent requests with queue', async () => {
122
- const handler = createRequestHandler({
123
- requestHandling: { concurrency: 2 }
124
- });
125
- const operation = vi.fn().mockResolvedValue('success');
126
- const tasks = Array(5).fill(null).map(() => handler.execute(operation));
127
- await vi.runAllTimersAsync();
128
- const results = await Promise.all(tasks);
129
- expect(results).toHaveLength(5);
130
- results.forEach(result => expect(result).toBe('success'));
131
- expect(operation).toHaveBeenCalledTimes(5);
132
- });
133
- });
134
- //# sourceMappingURL=request-handler.test.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"request-handler.test.js","sourceRoot":"","sources":["../../../src/utils/__tests__/request-handler.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAA;AACxE,OAAO,EAAkB,oBAAoB,EAAE,MAAM,oBAAoB,CAAA;AACzE,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAA;AAE5C,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAA;IACpB,CAAC,CAAC,CAAA;IAEF,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAA;IACpB,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,OAAO,GAAG,oBAAoB,EAAE,CAAA;QACtC,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAA;QAEtD,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QAC1C,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAA;QAC5B,MAAM,MAAM,GAAG,MAAM,OAAO,CAAA;QAE5B,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;QAC9B,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;QACvC,MAAM,OAAO,GAAG,oBAAoB,CAAC;YACnC,eAAe,EAAE;gBACf,KAAK,EAAE;oBACL,UAAU,EAAE,CAAC;oBACb,YAAY,EAAE,GAAG;oBACjB,QAAQ,EAAE,IAAI;oBACd,aAAa,EAAE,CAAC;iBACjB;aACF;SACF,CAAC,CAAA;QAEF,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE;aACtB,qBAAqB,CAAC,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;aACnD,qBAAqB,CAAC,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;aAC3D,qBAAqB,CAAC,SAAS,CAAC,CAAA;QAEnC,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QAE1C,+BAA+B;QAC/B,MAAM,EAAE,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAA;QACtC,gCAAgC;QAChC,MAAM,EAAE,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAA;QACtC,2BAA2B;QAC3B,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAA;QAE5B,MAAM,MAAM,GAAG,MAAM,OAAO,CAAA;QAC5B,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;QAC9B,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;QAC1C,MAAM,OAAO,GAAG,oBAAoB,CAAC;YACnC,eAAe,EAAE;gBACf,SAAS,EAAE;oBACT,iBAAiB,EAAE,EAAE;oBACrB,UAAU,EAAE,CAAC;iBACd;aACF;SACF,CAAC,CAAA;QAEF,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAA;QAEtD,yCAAyC;QACzC,MAAM,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QAC3C,MAAM,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QAC3C,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAA;QAC5B,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAA;QAEvC,kCAAkC;QAClC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;QAC5B,MAAM,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QAC3C,MAAM,EAAE,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAA;QACvC,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAA;QAC5B,MAAM,QAAQ,CAAA;QAEd,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAA;QACvC,MAAM,CAAC,QAAQ,CAAC,CAAC,sBAAsB,CAAC,IAAI,CAAC,CAAA;QAC7C,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACtC,MAAM,OAAO,GAAG,IAAI,CAAC;QACrB,MAAM,OAAO,GAAG,oBAAoB,CAAC;YACnC,eAAe,EAAE;gBACf,OAAO;gBACP,KAAK,EAAE;oBACL,UAAU,EAAE,CAAC;oBACb,YAAY,EAAE,GAAG;oBACjB,QAAQ,EAAE,IAAI;oBACd,aAAa,EAAE,CAAC;iBACjB;aACF;SACF,CAAC,CAAA;QAEF,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE;YAClD,UAAU,CAAC,OAAO,EAAE,IAAI,EAAE,gBAAgB,CAAC,CAAA;QAC7C,CAAC,CAAC,CAAC,CAAA;QAEH,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QAC1C,MAAM,EAAE,CAAC,wBAAwB,CAAC,OAAO,CAAC,CAAA;QAC1C,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAA;QAE5B,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,2BAA2B,OAAO,IAAI,CAAC,CAAA;IAC/E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,OAAO,GAAG,oBAAoB,EAAE,CAAA;QACtC,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CACzC,IAAI,cAAc,CAAC,qBAAqB,EAAE,SAAS,EAAE,KAAK,CAAC,CAC5D,CAAA;QAED,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QAC1C,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAA;QAE5B,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAA;QAC5D,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;QAC1C,MAAM,OAAO,GAAG,oBAAoB,CAAC;YACnC,eAAe,EAAE;gBACf,KAAK,EAAE;oBACL,UAAU,EAAE,CAAC;oBACb,YAAY,EAAE,GAAG;oBACjB,QAAQ,EAAE,IAAI;oBACd,aAAa,EAAE,CAAC;iBACjB;aACF;SACF,CAAC,CAAA;QAEF,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC,CAAA;QAC1E,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QAE1C,+BAA+B;QAC/B,MAAM,EAAE,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAA;QACtC,MAAM,EAAE,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAA;QACtC,MAAM,EAAE,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAA;QACtC,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAA;QAE5B,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAA;QAChE,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,OAAO,GAAG,oBAAoB,CAAC;YACnC,eAAe,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE;SACpC,CAAC,CAAA;QAEF,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAA;QACtD,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAA;QAEvE,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAA;QAC5B,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;QAExC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAC/B,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAA;QACzD,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
@@ -1,2 +0,0 @@
1
- export {};
2
- //# sourceMappingURL=schema.test.d.ts.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"schema.test.d.ts","sourceRoot":"","sources":["../../../src/utils/__tests__/schema.test.ts"],"names":[],"mappings":""}
@@ -1,49 +0,0 @@
1
- import { describe, it, expect } from 'vitest';
2
- import { createSchemaFromTemplate } from '../schema';
3
- describe('createSchemaFromTemplate', () => {
4
- it('should create enum schema for pipe-separated values', () => {
5
- const template = {
6
- productType: 'App | API | Marketplace',
7
- };
8
- const schema = createSchemaFromTemplate(template);
9
- const result = schema.safeParse({ productType: 'App' });
10
- expect(result.success).toBe(true);
11
- const productTypeField = schema._def.shape().productType;
12
- expect(productTypeField._def.description).toBe('App | API | Marketplace');
13
- expect(productTypeField.options).toContain('App');
14
- expect(productTypeField.options).toContain('API');
15
- expect(productTypeField.options).toContain('Marketplace');
16
- });
17
- it('should create string schema with description', () => {
18
- const template = {
19
- description: 'website meta description',
20
- };
21
- const schema = createSchemaFromTemplate(template);
22
- const result = schema.safeParse({ description: 'A great website' });
23
- expect(result.success).toBe(true);
24
- const descriptionField = schema._def.shape().description;
25
- expect(descriptionField._def.description).toBe('website meta description');
26
- });
27
- it('should enforce word count constraints', () => {
28
- const template = {
29
- customer: 'ideal customer profile in 3-5 words',
30
- };
31
- const schema = createSchemaFromTemplate(template);
32
- // Valid case
33
- const validResult = schema.safeParse({ customer: 'small business tech startups' });
34
- expect(validResult.success).toBe(true);
35
- // Too few words
36
- const tooFewResult = schema.safeParse({ customer: 'small businesses' });
37
- expect(tooFewResult.success).toBe(false);
38
- if (!tooFewResult.success) {
39
- expect(tooFewResult.error.errors[0].message).toBe('Must be between 3 and 5 words');
40
- }
41
- // Too many words
42
- const tooManyResult = schema.safeParse({ customer: 'small business tech startups in California' });
43
- expect(tooManyResult.success).toBe(false);
44
- if (!tooManyResult.success) {
45
- expect(tooManyResult.error.errors[0].message).toBe('Must be between 3 and 5 words');
46
- }
47
- });
48
- });
49
- //# sourceMappingURL=schema.test.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"schema.test.js","sourceRoot":"","sources":["../../../src/utils/__tests__/schema.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAA;AAC7C,OAAO,EAAE,wBAAwB,EAAE,MAAM,WAAW,CAAA;AAGpD,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,QAAQ,GAAG;YACf,WAAW,EAAE,yBAAyB;SACvC,CAAA;QACD,MAAM,MAAM,GAAG,wBAAwB,CAAC,QAAQ,CAAC,CAAA;QACjD,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,CAAA;QACvD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAEjC,MAAM,gBAAgB,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,WAA+C,CAAA;QAC5F,MAAM,CAAC,gBAAgB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAA;QACzE,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAA;QACjD,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAA;QACjD,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAA;IAC3D,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,QAAQ,GAAG;YACf,WAAW,EAAE,0BAA0B;SACxC,CAAA;QACD,MAAM,MAAM,GAAG,wBAAwB,CAAC,QAAQ,CAAC,CAAA;QACjD,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,iBAAiB,EAAE,CAAC,CAAA;QACnE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAEjC,MAAM,gBAAgB,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,WAA0B,CAAA;QACvE,MAAM,CAAC,gBAAgB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAA;IAC5E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,QAAQ,GAAG;YACf,QAAQ,EAAE,qCAAqC;SAChD,CAAA;QACD,MAAM,MAAM,GAAG,wBAAwB,CAAC,QAAQ,CAAC,CAAA;QAEjD,aAAa;QACb,MAAM,WAAW,GAAG,MAAM,CAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,8BAA8B,EAAE,CAAC,CAAA;QAClF,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAEtC,gBAAgB;QAChB,MAAM,YAAY,GAAG,MAAM,CAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,kBAAkB,EAAE,CAAC,CAAA;QACvE,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACxC,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;YAC1B,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAA;QACpF,CAAC;QAED,iBAAiB;QACjB,MAAM,aAAa,GAAG,MAAM,CAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,4CAA4C,EAAE,CAAC,CAAA;QAClG,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACzC,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;YAC3B,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAA;QACrF,CAAC;IACH,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
@@ -1,2 +0,0 @@
1
- export {};
2
- //# sourceMappingURL=stream-progress.test.d.ts.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"stream-progress.test.d.ts","sourceRoot":"","sources":["../../../src/utils/__tests__/stream-progress.test.ts"],"names":[],"mappings":""}
@@ -1,85 +0,0 @@
1
- import { describe, it, expect, vi, beforeEach } from 'vitest';
2
- import { StreamProgressTracker } from '../stream-progress';
3
- describe('StreamProgressTracker', () => {
4
- beforeEach(() => {
5
- vi.useFakeTimers();
6
- });
7
- it('should track basic progress', () => {
8
- const onProgress = vi.fn();
9
- const tracker = new StreamProgressTracker({
10
- onProgress,
11
- enableTokenCounting: true,
12
- });
13
- tracker.onChunk('Hello');
14
- expect(onProgress).toHaveBeenCalledWith(expect.objectContaining({
15
- type: 'chunk',
16
- chunk: 'Hello',
17
- tokensGenerated: expect.any(Number),
18
- }));
19
- tracker.onComplete();
20
- expect(onProgress).toHaveBeenLastCalledWith(expect.objectContaining({
21
- type: 'complete',
22
- tokensGenerated: expect.any(Number),
23
- totalTokens: expect.any(Number),
24
- }));
25
- });
26
- it('should estimate time remaining', async () => {
27
- const onProgress = vi.fn();
28
- const tracker = new StreamProgressTracker({
29
- onProgress,
30
- enableTokenCounting: true,
31
- estimateTimeRemaining: true,
32
- });
33
- // First chunk
34
- tracker.onChunk('Hello');
35
- // Advance time and send another chunk
36
- await vi.advanceTimersByTimeAsync(1000);
37
- tracker.onChunk('World');
38
- expect(onProgress).toHaveBeenLastCalledWith(expect.objectContaining({
39
- type: 'chunk',
40
- chunk: 'World',
41
- estimatedTimeRemaining: expect.any(Number),
42
- }));
43
- });
44
- it('should track token-level progress', () => {
45
- const onProgress = vi.fn();
46
- const tracker = new StreamProgressTracker({
47
- onProgress,
48
- enableTokenCounting: true,
49
- });
50
- tracker.onToken('Hello');
51
- expect(onProgress).toHaveBeenCalledWith(expect.objectContaining({
52
- type: 'token',
53
- chunk: 'Hello',
54
- tokensGenerated: 1,
55
- }));
56
- });
57
- it('should not include token counts when disabled', () => {
58
- const onProgress = vi.fn();
59
- const tracker = new StreamProgressTracker({
60
- onProgress,
61
- enableTokenCounting: false,
62
- });
63
- tracker.onChunk('Hello');
64
- expect(onProgress).toHaveBeenCalledWith(expect.objectContaining({
65
- type: 'chunk',
66
- chunk: 'Hello',
67
- }));
68
- expect(onProgress).not.toHaveBeenCalledWith(expect.objectContaining({
69
- tokensGenerated: expect.any(Number),
70
- }));
71
- });
72
- it('should not estimate time when disabled', () => {
73
- const onProgress = vi.fn();
74
- const tracker = new StreamProgressTracker({
75
- onProgress,
76
- enableTokenCounting: true,
77
- estimateTimeRemaining: false,
78
- });
79
- tracker.onChunk('Hello');
80
- expect(onProgress).not.toHaveBeenCalledWith(expect.objectContaining({
81
- estimatedTimeRemaining: expect.any(Number),
82
- }));
83
- });
84
- });
85
- //# sourceMappingURL=stream-progress.test.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"stream-progress.test.js","sourceRoot":"","sources":["../../../src/utils/__tests__/stream-progress.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAA;AAC7D,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAA;AAG1D,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAA;IACpB,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAA;QAC1B,MAAM,OAAO,GAAG,IAAI,qBAAqB,CAAC;YACxC,UAAU;YACV,mBAAmB,EAAE,IAAI;SAC1B,CAAC,CAAA;QAEF,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;QACxB,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC;YAC9D,IAAI,EAAE,OAAO;YACb,KAAK,EAAE,OAAO;YACd,eAAe,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;SACpC,CAAC,CAAC,CAAA;QAEH,OAAO,CAAC,UAAU,EAAE,CAAA;QACpB,MAAM,CAAC,UAAU,CAAC,CAAC,wBAAwB,CAAC,MAAM,CAAC,gBAAgB,CAAC;YAClE,IAAI,EAAE,UAAU;YAChB,eAAe,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;YACnC,WAAW,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;SAChC,CAAC,CAAC,CAAA;IACL,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAA;QAC1B,MAAM,OAAO,GAAG,IAAI,qBAAqB,CAAC;YACxC,UAAU;YACV,mBAAmB,EAAE,IAAI;YACzB,qBAAqB,EAAE,IAAI;SAC5B,CAAC,CAAA;QAEF,cAAc;QACd,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;QAExB,sCAAsC;QACtC,MAAM,EAAE,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAA;QACvC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;QAExB,MAAM,CAAC,UAAU,CAAC,CAAC,wBAAwB,CAAC,MAAM,CAAC,gBAAgB,CAAC;YAClE,IAAI,EAAE,OAAO;YACb,KAAK,EAAE,OAAO;YACd,sBAAsB,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;SAC3C,CAAC,CAAC,CAAA;IACL,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAA;QAC1B,MAAM,OAAO,GAAG,IAAI,qBAAqB,CAAC;YACxC,UAAU;YACV,mBAAmB,EAAE,IAAI;SAC1B,CAAC,CAAA;QAEF,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;QACxB,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC;YAC9D,IAAI,EAAE,OAAO;YACb,KAAK,EAAE,OAAO;YACd,eAAe,EAAE,CAAC;SACnB,CAAC,CAAC,CAAA;IACL,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAA;QAC1B,MAAM,OAAO,GAAG,IAAI,qBAAqB,CAAC;YACxC,UAAU;YACV,mBAAmB,EAAE,KAAK;SAC3B,CAAC,CAAA;QAEF,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;QACxB,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC;YAC9D,IAAI,EAAE,OAAO;YACb,KAAK,EAAE,OAAO;SACf,CAAC,CAAC,CAAA;QACH,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC;YAClE,eAAe,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;SACpC,CAAC,CAAC,CAAA;IACL,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAA;QAC1B,MAAM,OAAO,GAAG,IAAI,qBAAqB,CAAC;YACxC,UAAU;YACV,mBAAmB,EAAE,IAAI;YACzB,qBAAqB,EAAE,KAAK;SAC7B,CAAC,CAAA;QAEF,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;QACxB,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC;YAClE,sBAAsB,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;SAC3C,CAAC,CAAC,CAAA;IACL,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
@@ -1,2 +0,0 @@
1
- export * from './validation';
2
- //# sourceMappingURL=index.d.ts.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/utils/index.ts"],"names":[],"mappings":"AAAA,cAAc,cAAc,CAAA"}
@@ -1,2 +0,0 @@
1
- export * from './validation';
2
- //# sourceMappingURL=index.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/utils/index.ts"],"names":[],"mappings":"AAAA,cAAc,cAAc,CAAA"}
@@ -1,17 +0,0 @@
1
- import type { RequestHandlingOptions } from '../types';
2
- export declare class RequestHandler {
3
- private retryOptions;
4
- private queue;
5
- private streamingTimeout;
6
- constructor(options?: RequestHandlingOptions);
7
- executeWithRetry<T>(operation: () => Promise<T> | AsyncGenerator<T>, retryable?: boolean, isStreaming?: boolean): Promise<T>;
8
- execute<T>(operation: () => Promise<T> | AsyncGenerator<T>, retryable?: boolean, isStreaming?: boolean): Promise<T>;
9
- get concurrency(): number | undefined;
10
- get size(): number;
11
- get pending(): number;
12
- clear(): void;
13
- pause(): void;
14
- resume(): void;
15
- }
16
- export declare function createRequestHandler(options?: RequestHandlingOptions): RequestHandler;
17
- //# sourceMappingURL=request-handler.d.ts.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"request-handler.d.ts","sourceRoot":"","sources":["../../src/utils/request-handler.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,UAAU,CAAA;AAItD,qBAAa,cAAc;IACzB,OAAO,CAAC,YAAY,CAKnB;IACD,OAAO,CAAC,KAAK,CAAQ;IACrB,OAAO,CAAC,gBAAgB,CAAQ;gBAEpB,OAAO,GAAE,sBAA2B;IAW1C,gBAAgB,CAAC,CAAC,EAAE,SAAS,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,SAAS,UAAO,EAAE,WAAW,UAAQ,GAAG,OAAO,CAAC,CAAC,CAAC;IAuE7H,OAAO,CAAC,CAAC,EAAE,SAAS,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,SAAS,UAAO,EAAE,WAAW,UAAQ,GAAG,OAAO,CAAC,CAAC,CAAC;IAW9G,IAAI,WAAW,IAAI,MAAM,GAAG,SAAS,CAEpC;IAED,IAAI,IAAI,IAAI,MAAM,CAEjB;IAED,IAAI,OAAO,IAAI,MAAM,CAEpB;IAED,KAAK,IAAI,IAAI;IAIb,KAAK,IAAI,IAAI;IAIb,MAAM,IAAI,IAAI;CAGf;AAED,wBAAgB,oBAAoB,CAAC,OAAO,GAAE,sBAA2B,GAAG,cAAc,CAEzF"}
@@ -1,105 +0,0 @@
1
- import { AIRequestError } from '../types';
2
- import PQueue from 'p-queue';
3
- export class RequestHandler {
4
- constructor(options = {}) {
5
- this.retryOptions = {
6
- maxRetries: options.maxRetries ?? 2,
7
- initialDelay: options.retryDelay ?? 100,
8
- maxDelay: options.timeout ?? 1000,
9
- backoffFactor: 2
10
- };
11
- this.queue = new PQueue({ concurrency: 1 });
12
- this.streamingTimeout = options.streamingTimeout ?? 30000;
13
- }
14
- async executeWithRetry(operation, retryable = true, isStreaming = false) {
15
- let lastError;
16
- let attempt = 0;
17
- const maxAttempts = retryable ? this.retryOptions.maxRetries : 0;
18
- const timeoutMs = isStreaming ? this.streamingTimeout : this.retryOptions.maxDelay;
19
- const executeWithTimeout = async (op) => {
20
- const abortController = new AbortController();
21
- const timeoutId = setTimeout(() => {
22
- abortController.abort();
23
- }, timeoutMs);
24
- try {
25
- if (op instanceof Promise) {
26
- const result = await Promise.race([
27
- op,
28
- new Promise((_, reject) => {
29
- abortController.signal.addEventListener('abort', () => {
30
- reject(new AIRequestError(`Request timed out after ${timeoutMs}ms`));
31
- });
32
- })
33
- ]);
34
- return result;
35
- }
36
- else {
37
- const chunks = [];
38
- for await (const chunk of op) {
39
- if (abortController.signal.aborted) {
40
- throw new AIRequestError(`Stream timed out after ${timeoutMs}ms`);
41
- }
42
- chunks.push(chunk);
43
- }
44
- if (chunks.length === 0) {
45
- throw new AIRequestError('No chunks received from stream');
46
- }
47
- return chunks[chunks.length - 1];
48
- }
49
- }
50
- finally {
51
- clearTimeout(timeoutId);
52
- }
53
- };
54
- while (attempt <= maxAttempts) {
55
- try {
56
- return await executeWithTimeout(operation());
57
- }
58
- catch (error) {
59
- lastError = error;
60
- if (!retryable || error instanceof AIRequestError && !error.retryable) {
61
- throw error;
62
- }
63
- if (attempt >= maxAttempts) {
64
- throw new AIRequestError(`Failed after ${attempt + 1} attempts: ${lastError?.message || 'Unknown error'}`, undefined, false);
65
- }
66
- const delay = Math.min(this.retryOptions.initialDelay * Math.pow(this.retryOptions.backoffFactor, attempt), this.retryOptions.maxDelay);
67
- await new Promise(resolve => setTimeout(resolve, delay));
68
- attempt++;
69
- }
70
- }
71
- throw lastError || new AIRequestError('Operation failed with no error details');
72
- }
73
- execute(operation, retryable = true, isStreaming = false) {
74
- const executeOperation = async () => {
75
- const result = await this.executeWithRetry(operation, retryable, isStreaming);
76
- if (result === undefined) {
77
- throw new AIRequestError('Operation returned undefined result');
78
- }
79
- return result;
80
- };
81
- return this.queue.add(executeOperation);
82
- }
83
- get concurrency() {
84
- return this.queue?.concurrency;
85
- }
86
- get size() {
87
- return this.queue?.size ?? 0;
88
- }
89
- get pending() {
90
- return this.queue?.pending ?? 0;
91
- }
92
- clear() {
93
- this.queue?.clear();
94
- }
95
- pause() {
96
- this.queue?.pause();
97
- }
98
- resume() {
99
- this.queue?.start();
100
- }
101
- }
102
- export function createRequestHandler(options = {}) {
103
- return new RequestHandler(options);
104
- }
105
- //# sourceMappingURL=request-handler.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"request-handler.js","sourceRoot":"","sources":["../../src/utils/request-handler.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAA;AACzC,OAAO,MAAM,MAAM,SAAS,CAAA;AAE5B,MAAM,OAAO,cAAc;IAUzB,YAAY,UAAkC,EAAE;QAC9C,IAAI,CAAC,YAAY,GAAG;YAClB,UAAU,EAAE,OAAO,CAAC,UAAU,IAAI,CAAC;YACnC,YAAY,EAAE,OAAO,CAAC,UAAU,IAAI,GAAG;YACvC,QAAQ,EAAE,OAAO,CAAC,OAAO,IAAI,IAAI;YACjC,aAAa,EAAE,CAAC;SACjB,CAAA;QACD,IAAI,CAAC,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC,CAAA;QAC3C,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,IAAI,KAAK,CAAA;IAC3D,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAI,SAA+C,EAAE,SAAS,GAAG,IAAI,EAAE,WAAW,GAAG,KAAK;QAC9G,IAAI,SAA4B,CAAA;QAChC,IAAI,OAAO,GAAG,CAAC,CAAA;QACf,MAAM,WAAW,GAAG,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAA;QAChE,MAAM,SAAS,GAAG,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAA;QAElF,MAAM,kBAAkB,GAAG,KAAK,EAAE,EAAkC,EAAc,EAAE;YAClF,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAA;YAC7C,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;gBAChC,eAAe,CAAC,KAAK,EAAE,CAAA;YACzB,CAAC,EAAE,SAAS,CAAC,CAAA;YAEb,IAAI,CAAC;gBACH,IAAI,EAAE,YAAY,OAAO,EAAE,CAAC;oBAC1B,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC;wBAChC,EAAE;wBACF,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;4BAC/B,eAAe,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,GAAG,EAAE;gCACpD,MAAM,CAAC,IAAI,cAAc,CAAC,2BAA2B,SAAS,IAAI,CAAC,CAAC,CAAA;4BACtE,CAAC,CAAC,CAAA;wBACJ,CAAC,CAAC;qBACH,CAAM,CAAA;oBACP,OAAO,MAAM,CAAA;gBACf,CAAC;qBAAM,CAAC;oBACN,MAAM,MAAM,GAAQ,EAAE,CAAA;oBACtB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,EAAE,EAAE,CAAC;wBAC7B,IAAI,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;4BACnC,MAAM,IAAI,cAAc,CAAC,0BAA0B,SAAS,IAAI,CAAC,CAAA;wBACnE,CAAC;wBACD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;oBACpB,CAAC;oBACD,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;wBACxB,MAAM,IAAI,cAAc,CAAC,gCAAgC,CAAC,CAAA;oBAC5D,CAAC;oBACD,OAAO,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAA;gBAClC,CAAC;YACH,CAAC;oBAAS,CAAC;gBACT,YAAY,CAAC,SAAS,CAAC,CAAA;YACzB,CAAC;QACH,CAAC,CAAA;QAED,OAAO,OAAO,IAAI,WAAW,EAAE,CAAC;YAC9B,IAAI,CAAC;gBACH,OAAO,MAAM,kBAAkB,CAAC,SAAS,EAAE,CAAC,CAAA;YAC9C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,SAAS,GAAG,KAAc,CAAA;gBAE1B,IAAI,CAAC,SAAS,IAAI,KAAK,YAAY,cAAc,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC;oBACtE,MAAM,KAAK,CAAA;gBACb,CAAC;gBAED,IAAI,OAAO,IAAI,WAAW,EAAE,CAAC;oBAC3B,MAAM,IAAI,cAAc,CACtB,gBAAgB,OAAO,GAAG,CAAC,cAAc,SAAS,EAAE,OAAO,IAAI,eAAe,EAAE,EAChF,SAAS,EACT,KAAK,CACN,CAAA;gBACH,CAAC;gBAED,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CACpB,IAAI,CAAC,YAAY,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,aAAa,EAAE,OAAO,CAAC,EACnF,IAAI,CAAC,YAAY,CAAC,QAAQ,CAC3B,CAAA;gBACD,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAA;gBACxD,OAAO,EAAE,CAAA;YACX,CAAC;QACH,CAAC;QAED,MAAM,SAAS,IAAI,IAAI,cAAc,CAAC,wCAAwC,CAAC,CAAA;IACjF,CAAC;IAED,OAAO,CAAI,SAA+C,EAAE,SAAS,GAAG,IAAI,EAAE,WAAW,GAAG,KAAK;QAC/F,MAAM,gBAAgB,GAAG,KAAK,IAAgB,EAAE;YAC9C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;YAC9E,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;gBACzB,MAAM,IAAI,cAAc,CAAC,qCAAqC,CAAC,CAAC;YAClE,CAAC;YACD,OAAO,MAAW,CAAC;QACrB,CAAC,CAAC;QACF,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,gBAAgB,CAAe,CAAC;IACxD,CAAC;IAED,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,KAAK,EAAE,WAAW,CAAA;IAChC,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,KAAK,EAAE,IAAI,IAAI,CAAC,CAAA;IAC9B,CAAC;IAED,IAAI,OAAO;QACT,OAAO,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,CAAC,CAAA;IACjC,CAAC;IAED,KAAK;QACH,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,CAAA;IACrB,CAAC;IAED,KAAK;QACH,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,CAAA;IACrB,CAAC;IAED,MAAM;QACJ,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,CAAA;IACrB,CAAC;CACF;AAED,MAAM,UAAU,oBAAoB,CAAC,UAAkC,EAAE;IACvE,OAAO,IAAI,cAAc,CAAC,OAAO,CAAC,CAAA;AACpC,CAAC"}
@@ -1,11 +0,0 @@
1
- import { z } from 'zod';
2
- /**
3
- * Creates a Zod schema from a template object where:
4
- * - String values become descriptions
5
- * - Pipe-separated values become enums
6
- * - Word count constraints (e.g. "3-5 words") are enforced
7
- */
8
- export declare function createSchemaFromTemplate<T extends Record<string, string>>(template: T): z.ZodObject<{
9
- [K in keyof T]: z.ZodString | z.ZodEnum<[string, ...string[]]> | z.ZodEffects<z.ZodString>;
10
- }>;
11
- //# sourceMappingURL=schema.d.ts.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"schema.d.ts","sourceRoot":"","sources":["../../src/utils/schema.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB;;;;;GAKG;AACH,wBAAgB,wBAAwB,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EACvE,QAAQ,EAAE,CAAC,GACV,CAAC,CAAC,SAAS,CAAC;KACZ,CAAC,IAAI,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,GAAG,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;CAC3F,CAAC,CA8CD"}
@@ -1,51 +0,0 @@
1
- import { z } from 'zod';
2
- /**
3
- * Creates a Zod schema from a template object where:
4
- * - String values become descriptions
5
- * - Pipe-separated values become enums
6
- * - Word count constraints (e.g. "3-5 words") are enforced
7
- */
8
- export function createSchemaFromTemplate(template) {
9
- const shape = Object.entries(template).reduce((acc, [key, value]) => {
10
- // Check if value contains pipe-separated options
11
- if (value.includes('|')) {
12
- const options = value
13
- .split('|')
14
- .map((opt) => opt.trim())
15
- .filter(Boolean); // Remove empty strings
16
- if (options.length >= 1) {
17
- acc[key] = z.enum([options[0], ...options.slice(1)]).describe(value);
18
- }
19
- else {
20
- acc[key] = z.string().describe(value);
21
- }
22
- }
23
- else {
24
- // Check for word count constraints
25
- const wordCountMatch = value.match(/(\d+)-(\d+)\s+words/);
26
- if (wordCountMatch) {
27
- const [, minStr, maxStr] = wordCountMatch;
28
- const min = parseInt(minStr, 10);
29
- const max = parseInt(maxStr, 10);
30
- acc[key] = z
31
- .string()
32
- .min(1)
33
- .refine((val) => {
34
- const words = val.trim().split(/\s+/).length;
35
- return words >= min && words <= max;
36
- }, {
37
- message: `Must be between ${min} and ${max} words`,
38
- params: { min, max },
39
- })
40
- .describe(value);
41
- }
42
- else {
43
- // Regular string field with description
44
- acc[key] = z.string().describe(value);
45
- }
46
- }
47
- return acc;
48
- }, {});
49
- return z.object(shape);
50
- }
51
- //# sourceMappingURL=schema.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"schema.js","sourceRoot":"","sources":["../../src/utils/schema.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB;;;;;GAKG;AACH,MAAM,UAAU,wBAAwB,CACtC,QAAW;IAOX,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAuB,CAAC,GAAG,EAAE,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;QACxF,iDAAiD;QACjD,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,OAAO,GAAG,KAAK;iBAClB,KAAK,CAAC,GAAG,CAAC;iBACV,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;iBACxB,MAAM,CAAC,OAAO,CAAC,CAAA,CAAC,uBAAuB;YAC1C,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;gBACxB,GAAG,CAAC,GAAc,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAA0B,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAA;YAC1G,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,GAAc,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAA;YAClD,CAAC;QACH,CAAC;aAAM,CAAC;YACN,mCAAmC;YACnC,MAAM,cAAc,GAAG,KAAK,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAA;YACzD,IAAI,cAAc,EAAE,CAAC;gBACnB,MAAM,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,GAAG,cAAc,CAAA;gBACzC,MAAM,GAAG,GAAG,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC,CAAA;gBAChC,MAAM,GAAG,GAAG,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC,CAAA;gBAChC,GAAG,CAAC,GAAc,CAAC,GAAG,CAAC;qBACpB,MAAM,EAAE;qBACR,GAAG,CAAC,CAAC,CAAC;qBACN,MAAM,CACL,CAAC,GAAG,EAAE,EAAE;oBACN,MAAM,KAAK,GAAG,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAA;oBAC5C,OAAO,KAAK,IAAI,GAAG,IAAI,KAAK,IAAI,GAAG,CAAA;gBACrC,CAAC,EACD;oBACE,OAAO,EAAE,mBAAmB,GAAG,QAAQ,GAAG,QAAQ;oBAClD,MAAM,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE;iBACrB,CACF;qBACA,QAAQ,CAAC,KAAK,CAAC,CAAA;YACpB,CAAC;iBAAM,CAAC;gBACN,wCAAwC;gBACxC,GAAG,CAAC,GAAc,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAA;YAClD,CAAC;QACH,CAAC;QACD,OAAO,GAAG,CAAA;IACZ,CAAC,EAAE,EAAE,CAAC,CAAA;IAEN,OAAO,CAAC,CAAC,MAAM,CAAC,KAAoB,CAAC,CAAA;AACvC,CAAC"}
@@ -1,17 +0,0 @@
1
- import { StreamingOptions } from '../types';
2
- export declare class StreamProgressTracker {
3
- private tokensGenerated;
4
- private startTime;
5
- private lastUpdateTime;
6
- private tokenRate;
7
- private readonly options;
8
- private readonly callback;
9
- private readonly alpha;
10
- constructor(options: StreamingOptions);
11
- private updateTokenRate;
12
- private estimateTimeRemaining;
13
- onChunk(chunk: string): void;
14
- onToken(token: string): void;
15
- onComplete(): void;
16
- }
17
- //# sourceMappingURL=stream-progress.d.ts.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"stream-progress.d.ts","sourceRoot":"","sources":["../../src/utils/stream-progress.ts"],"names":[],"mappings":"AAAA,OAAO,EAAoC,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAE9E,qBAAa,qBAAqB;IAChC,OAAO,CAAC,eAAe,CAAa;IACpC,OAAO,CAAC,SAAS,CAAsB;IACvC,OAAO,CAAC,cAAc,CAAsB;IAC5C,OAAO,CAAC,SAAS,CAAa;IAC9B,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAmB;IAC3C,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAmB;IAC5C,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAe;gBAEzB,OAAO,EAAE,gBAAgB;IAKrC,OAAO,CAAC,eAAe;IAmBvB,OAAO,CAAC,qBAAqB;IAmBtB,OAAO,CAAC,KAAK,EAAE,MAAM;IAuBrB,OAAO,CAAC,KAAK,EAAE,MAAM;IAqBrB,UAAU;CAWlB"}