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
@@ -0,0 +1,302 @@
1
+ /**
2
+ * Tests for tagged template literal support
3
+ *
4
+ * Every function should support:
5
+ * 1. Tagged template syntax: fn`prompt ${variable}`
6
+ * 2. Objects/arrays auto-convert to YAML in templates
7
+ * 3. Options chaining: fn`prompt`({ model: '...' })
8
+ */
9
+
10
+ import { describe, it, expect } from 'vitest'
11
+ import { stringify } from 'yaml'
12
+
13
+ // ============================================================================
14
+ // Template Parsing Tests (Unit Tests - No AI calls)
15
+ // ============================================================================
16
+
17
+ /**
18
+ * Parse a tagged template literal into a prompt string
19
+ * Objects and arrays are converted to YAML for readability
20
+ */
21
+ function parseTemplate(strings: TemplateStringsArray, ...values: unknown[]): string {
22
+ return strings.reduce((result, str, i) => {
23
+ const value = values[i]
24
+ if (value === undefined) {
25
+ return result + str
26
+ }
27
+
28
+ // Convert objects/arrays to YAML
29
+ if (typeof value === 'object' && value !== null) {
30
+ const yaml = stringify(value).trim()
31
+ return result + str + '\n' + yaml
32
+ }
33
+
34
+ // Primitives use toString()
35
+ return result + str + String(value)
36
+ }, '')
37
+ }
38
+
39
+ /**
40
+ * Create a template function that supports both call styles
41
+ */
42
+ function createTemplateFunction<T>(
43
+ handler: (prompt: string, options?: Record<string, unknown>) => Promise<T>
44
+ ): {
45
+ (strings: TemplateStringsArray, ...values: unknown[]): Promise<T> & { (options?: Record<string, unknown>): Promise<T> }
46
+ (prompt: string, options?: Record<string, unknown>): Promise<T>
47
+ } {
48
+ function templateFn(promptOrStrings: string | TemplateStringsArray, ...args: unknown[]) {
49
+ // Tagged template literal
50
+ if (Array.isArray(promptOrStrings) && 'raw' in promptOrStrings) {
51
+ const prompt = parseTemplate(promptOrStrings as TemplateStringsArray, ...args)
52
+
53
+ // Return a thenable that also accepts options
54
+ const promise = handler(prompt)
55
+
56
+ // Allow chaining with options: fn`prompt`({ model: '...' })
57
+ // Create a function that also implements then/catch for Promise interface
58
+ const chainable = (options?: Record<string, unknown>) => handler(prompt, options)
59
+
60
+ // Make it thenable by adding then/catch from the original promise
61
+ ;(chainable as unknown as Promise<T>).then = promise.then.bind(promise)
62
+ ;(chainable as unknown as Promise<T>).catch = promise.catch.bind(promise)
63
+
64
+ return chainable
65
+ }
66
+
67
+ // Regular function call
68
+ return handler(promptOrStrings as string, args[0] as Record<string, unknown> | undefined)
69
+ }
70
+
71
+ return templateFn as ReturnType<typeof createTemplateFunction<T>>
72
+ }
73
+
74
+ describe('template parsing', () => {
75
+ it('parses simple string interpolation', () => {
76
+ const topic = 'TypeScript'
77
+ const result = parseTemplate`Write about ${topic}`
78
+ expect(result).toBe('Write about TypeScript')
79
+ })
80
+
81
+ it('parses multiple interpolations', () => {
82
+ const topic = 'TypeScript'
83
+ const audience = 'beginners'
84
+ const result = parseTemplate`Write about ${topic} for ${audience}`
85
+ expect(result).toBe('Write about TypeScript for beginners')
86
+ })
87
+
88
+ it('converts objects to YAML', () => {
89
+ const context = { topic: 'TypeScript', level: 'beginner' }
90
+ const result = parseTemplate`Write a post about ${{ context }}`
91
+
92
+ expect(result).toContain('context:')
93
+ expect(result).toContain('topic: TypeScript')
94
+ expect(result).toContain('level: beginner')
95
+ })
96
+
97
+ it('converts arrays to YAML lists', () => {
98
+ const topics = ['React', 'Vue', 'Angular']
99
+ const result = parseTemplate`Compare ${topics}`
100
+
101
+ expect(result).toContain('- React')
102
+ expect(result).toContain('- Vue')
103
+ expect(result).toContain('- Angular')
104
+ })
105
+
106
+ it('handles nested objects', () => {
107
+ const brand = {
108
+ hero: 'developers',
109
+ problem: {
110
+ internal: 'complexity',
111
+ external: 'time constraints',
112
+ },
113
+ }
114
+ const result = parseTemplate`Create a story for ${{ brand }}`
115
+
116
+ expect(result).toContain('brand:')
117
+ expect(result).toContain('hero: developers')
118
+ expect(result).toContain('problem:')
119
+ expect(result).toContain('internal: complexity')
120
+ expect(result).toContain('external: time constraints')
121
+ })
122
+
123
+ it('handles numbers and booleans', () => {
124
+ const count = 10
125
+ const active = true
126
+ const result = parseTemplate`Generate ${count} items, active: ${active}`
127
+
128
+ expect(result).toBe('Generate 10 items, active: true')
129
+ })
130
+
131
+ it('handles null and undefined gracefully', () => {
132
+ const value = null
133
+ const result = parseTemplate`Value is ${value}`
134
+ expect(result).toContain('Value is')
135
+ })
136
+
137
+ it('preserves template structure with objects inline', () => {
138
+ const requirements = {
139
+ pages: ['home', 'about', 'contact'],
140
+ features: ['dark mode', 'responsive'],
141
+ }
142
+ const result = parseTemplate`marketing site${{ requirements }}`
143
+
144
+ expect(result).toContain('marketing site')
145
+ expect(result).toContain('requirements:')
146
+ expect(result).toContain('pages:')
147
+ expect(result).toContain('- home')
148
+ expect(result).toContain('features:')
149
+ expect(result).toContain('- dark mode')
150
+ })
151
+ })
152
+
153
+ describe('template function creation', () => {
154
+ it('creates a function that handles tagged templates', async () => {
155
+ const mockHandler = async (prompt: string) => `Processed: ${prompt}`
156
+ const fn = createTemplateFunction(mockHandler)
157
+
158
+ const result = await fn`Hello ${'world'}`
159
+ expect(result).toBe('Processed: Hello world')
160
+ })
161
+
162
+ it('creates a function that handles regular calls', async () => {
163
+ const mockHandler = async (prompt: string) => `Processed: ${prompt}`
164
+ const fn = createTemplateFunction(mockHandler)
165
+
166
+ const result = await fn('Hello world')
167
+ expect(result).toBe('Processed: Hello world')
168
+ })
169
+
170
+ it('supports options chaining on tagged templates', async () => {
171
+ let capturedOptions: Record<string, unknown> | undefined
172
+ const mockHandler = async (prompt: string, options?: Record<string, unknown>) => {
173
+ capturedOptions = options
174
+ return `Processed: ${prompt}`
175
+ }
176
+ const fn = createTemplateFunction(mockHandler)
177
+
178
+ const result = await fn`Hello world`({ model: 'claude-opus-4-5' })
179
+
180
+ expect(result).toBe('Processed: Hello world')
181
+ expect(capturedOptions).toEqual({ model: 'claude-opus-4-5' })
182
+ })
183
+
184
+ it('passes options in regular calls', async () => {
185
+ let capturedOptions: Record<string, unknown> | undefined
186
+ const mockHandler = async (prompt: string, options?: Record<string, unknown>) => {
187
+ capturedOptions = options
188
+ return `Processed: ${prompt}`
189
+ }
190
+ const fn = createTemplateFunction(mockHandler)
191
+
192
+ await fn('Hello world', { model: 'gpt-5-1', temperature: 0.7 })
193
+
194
+ expect(capturedOptions).toEqual({ model: 'gpt-5-1', temperature: 0.7 })
195
+ })
196
+
197
+ it('converts complex objects to YAML in templates', async () => {
198
+ let capturedPrompt = ''
199
+ const mockHandler = async (prompt: string) => {
200
+ capturedPrompt = prompt
201
+ return 'ok'
202
+ }
203
+ const fn = createTemplateFunction(mockHandler)
204
+
205
+ const brand = { hero: 'developers', guide: 'ai-functions' }
206
+ await fn`Create story for ${{ brand }}`
207
+
208
+ expect(capturedPrompt).toContain('brand:')
209
+ expect(capturedPrompt).toContain('hero: developers')
210
+ expect(capturedPrompt).toContain('guide: ai-functions')
211
+ })
212
+ })
213
+
214
+ describe('options parameter', () => {
215
+ it('supports model option', async () => {
216
+ let capturedOptions: Record<string, unknown> | undefined
217
+ const mockHandler = async (_prompt: string, options?: Record<string, unknown>) => {
218
+ capturedOptions = options
219
+ return 'ok'
220
+ }
221
+ const fn = createTemplateFunction(mockHandler)
222
+
223
+ await fn`test`({ model: 'claude-opus-4-5' })
224
+
225
+ expect(capturedOptions?.model).toBe('claude-opus-4-5')
226
+ })
227
+
228
+ it('supports thinking option', async () => {
229
+ let capturedOptions: Record<string, unknown> | undefined
230
+ const mockHandler = async (_prompt: string, options?: Record<string, unknown>) => {
231
+ capturedOptions = options
232
+ return 'ok'
233
+ }
234
+ const fn = createTemplateFunction(mockHandler)
235
+
236
+ await fn`complex analysis`({ thinking: 'high' })
237
+
238
+ expect(capturedOptions?.thinking).toBe('high')
239
+ })
240
+
241
+ it('supports thinking as token budget number', async () => {
242
+ let capturedOptions: Record<string, unknown> | undefined
243
+ const mockHandler = async (_prompt: string, options?: Record<string, unknown>) => {
244
+ capturedOptions = options
245
+ return 'ok'
246
+ }
247
+ const fn = createTemplateFunction(mockHandler)
248
+
249
+ await fn`complex analysis`({ thinking: 10000 })
250
+
251
+ expect(capturedOptions?.thinking).toBe(10000)
252
+ })
253
+
254
+ it('supports temperature option', async () => {
255
+ let capturedOptions: Record<string, unknown> | undefined
256
+ const mockHandler = async (_prompt: string, options?: Record<string, unknown>) => {
257
+ capturedOptions = options
258
+ return 'ok'
259
+ }
260
+ const fn = createTemplateFunction(mockHandler)
261
+
262
+ await fn`creative writing`({ temperature: 0.9 })
263
+
264
+ expect(capturedOptions?.temperature).toBe(0.9)
265
+ })
266
+
267
+ it('supports maxTokens option', async () => {
268
+ let capturedOptions: Record<string, unknown> | undefined
269
+ const mockHandler = async (_prompt: string, options?: Record<string, unknown>) => {
270
+ capturedOptions = options
271
+ return 'ok'
272
+ }
273
+ const fn = createTemplateFunction(mockHandler)
274
+
275
+ await fn`long article`({ maxTokens: 4000 })
276
+
277
+ expect(capturedOptions?.maxTokens).toBe(4000)
278
+ })
279
+
280
+ it('supports multiple options together', async () => {
281
+ let capturedOptions: Record<string, unknown> | undefined
282
+ const mockHandler = async (_prompt: string, options?: Record<string, unknown>) => {
283
+ capturedOptions = options
284
+ return 'ok'
285
+ }
286
+ const fn = createTemplateFunction(mockHandler)
287
+
288
+ await fn`complex task`({
289
+ model: 'claude-opus-4-5',
290
+ thinking: 'high',
291
+ temperature: 0.7,
292
+ maxTokens: 8000,
293
+ })
294
+
295
+ expect(capturedOptions).toEqual({
296
+ model: 'claude-opus-4-5',
297
+ thinking: 'high',
298
+ temperature: 0.7,
299
+ maxTokens: 8000,
300
+ })
301
+ })
302
+ })
package/tsconfig.json ADDED
@@ -0,0 +1,10 @@
1
+ {
2
+ "extends": "../../tsconfig.base.json",
3
+ "compilerOptions": {
4
+ "rootDir": "src",
5
+ "outDir": "dist",
6
+ "lib": ["ES2022", "DOM", "DOM.Iterable"]
7
+ },
8
+ "include": ["src/**/*"],
9
+ "exclude": ["node_modules", "dist", "**/*.test.ts", "**/*.spec.ts"]
10
+ }
@@ -0,0 +1,42 @@
1
+ import { defineConfig } from 'vitest/config'
2
+ import { config } from 'dotenv'
3
+ import { resolve } from 'path'
4
+ import { existsSync } from 'fs'
5
+
6
+ // Load .env from current directory, parent directories, and root
7
+ // This supports primitives being used as a submodule
8
+ const envPaths = [
9
+ resolve(process.cwd(), '.env'),
10
+ resolve(process.cwd(), '..', '.env'),
11
+ resolve(process.cwd(), '..', '..', '.env'),
12
+ resolve(process.cwd(), '..', '..', '..', '.env'),
13
+ ]
14
+
15
+ for (const envPath of envPaths) {
16
+ if (existsSync(envPath)) {
17
+ config({ path: envPath })
18
+ }
19
+ }
20
+
21
+ export default defineConfig({
22
+ test: {
23
+ globals: false,
24
+ environment: 'node',
25
+ include: ['test/**/*.test.ts'],
26
+ // Exclude runtime-specific tests (use separate configs)
27
+ exclude: [
28
+ 'test/rpc/workers/**',
29
+ 'test/rpc/bun/**',
30
+ 'node_modules/**',
31
+ ],
32
+ testTimeout: 60000, // AI calls can take time
33
+ hookTimeout: 30000,
34
+ // Run tests sequentially to avoid rate limiting
35
+ pool: 'forks',
36
+ poolOptions: {
37
+ forks: {
38
+ singleFork: true,
39
+ },
40
+ },
41
+ },
42
+ })
package/LICENSE DELETED
@@ -1,21 +0,0 @@
1
- MIT License
2
-
3
- Copyright (c) 2024 AI Primitives
4
-
5
- Permission is hereby granted, free of charge, to any person obtaining a copy
6
- of this software and associated documentation files (the "Software"), to deal
7
- in the Software without restriction, including without limitation the rights
8
- to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
- copies of the Software, and to permit persons to whom the Software is
10
- furnished to do so, subject to the following conditions:
11
-
12
- The above copyright notice and this permission notice shall be included in all
13
- copies or substantial portions of the Software.
14
-
15
- THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
- IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
- FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
- AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
- LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
- OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
- SOFTWARE.
package/bin/cli.js DELETED
@@ -1,5 +0,0 @@
1
- #!/usr/bin/env node
2
-
3
- import { cli } from '../dist/cli/index.js'
4
-
5
- cli()
@@ -1,10 +0,0 @@
1
- interface CliOptions {
2
- version?: boolean;
3
- help?: boolean;
4
- }
5
- export declare function parseArgs(args: string[]): CliOptions;
6
- export declare function showHelp(): void;
7
- export declare function showVersion(): void;
8
- export declare function cli(args?: string[]): void;
9
- export {};
10
- //# sourceMappingURL=index.d.ts.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/cli/index.ts"],"names":[],"mappings":"AAEA,UAAU,UAAU;IAClB,OAAO,CAAC,EAAE,OAAO,CAAA;IACjB,IAAI,CAAC,EAAE,OAAO,CAAA;CACf;AAED,wBAAgB,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,UAAU,CAYpD;AAED,wBAAgB,QAAQ,IAAI,IAAI,CAQ/B;AAED,wBAAgB,WAAW,IAAI,IAAI,CAElC;AAED,wBAAgB,GAAG,CAAC,IAAI,GAAE,MAAM,EAA0B,GAAG,IAAI,CAUhE"}
package/dist/cli/index.js DELETED
@@ -1,38 +0,0 @@
1
- import { version } from '../../package.json';
2
- export function parseArgs(args) {
3
- const options = {};
4
- for (const arg of args) {
5
- if (arg === '-v' || arg === '--version') {
6
- options.version = true;
7
- }
8
- else if (arg === '-h' || arg === '--help') {
9
- options.help = true;
10
- }
11
- }
12
- return options;
13
- }
14
- export function showHelp() {
15
- console.log(`
16
- Usage: ai-functions [options]
17
-
18
- Options:
19
- -v, --version Show version number
20
- -h, --help Show help
21
- `);
22
- }
23
- export function showVersion() {
24
- console.log(`v${version}`);
25
- }
26
- export function cli(args = process.argv.slice(2)) {
27
- const options = parseArgs(args);
28
- if (options.version) {
29
- showVersion();
30
- }
31
- else if (options.help) {
32
- showHelp();
33
- }
34
- else {
35
- showHelp();
36
- }
37
- }
38
- //# sourceMappingURL=index.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/cli/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAA;AAO5C,MAAM,UAAU,SAAS,CAAC,IAAc;IACtC,MAAM,OAAO,GAAe,EAAE,CAAA;IAE9B,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IAAI,GAAG,KAAK,IAAI,IAAI,GAAG,KAAK,WAAW,EAAE,CAAC;YACxC,OAAO,CAAC,OAAO,GAAG,IAAI,CAAA;QACxB,CAAC;aAAM,IAAI,GAAG,KAAK,IAAI,IAAI,GAAG,KAAK,QAAQ,EAAE,CAAC;YAC5C,OAAO,CAAC,IAAI,GAAG,IAAI,CAAA;QACrB,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAA;AAChB,CAAC;AAED,MAAM,UAAU,QAAQ;IACtB,OAAO,CAAC,GAAG,CAAC;;;;;;CAMb,CAAC,CAAA;AACF,CAAC;AAED,MAAM,UAAU,WAAW;IACzB,OAAO,CAAC,GAAG,CAAC,IAAI,OAAO,EAAE,CAAC,CAAA;AAC5B,CAAC;AAED,MAAM,UAAU,GAAG,CAAC,OAAiB,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;IACxD,MAAM,OAAO,GAAG,SAAS,CAAC,IAAI,CAAC,CAAA;IAE/B,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACpB,WAAW,EAAE,CAAA;IACf,CAAC;SAAM,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;QACxB,QAAQ,EAAE,CAAA;IACZ,CAAC;SAAM,CAAC;QACN,QAAQ,EAAE,CAAA;IACZ,CAAC;AACH,CAAC"}
@@ -1,2 +0,0 @@
1
- export {};
2
- //# sourceMappingURL=index.test.d.ts.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"index.test.d.ts","sourceRoot":"","sources":["../../src/cli/index.test.ts"],"names":[],"mappings":""}
@@ -1,35 +0,0 @@
1
- import { describe, it, expect, vi } from 'vitest';
2
- import { parseArgs, showHelp, showVersion, cli } from './index';
3
- describe('CLI', () => {
4
- it('should parse version flag correctly', () => {
5
- const options = parseArgs(['-v']);
6
- expect(options.version).toBe(true);
7
- const longOptions = parseArgs(['--version']);
8
- expect(longOptions.version).toBe(true);
9
- });
10
- it('should parse help flag correctly', () => {
11
- const options = parseArgs(['-h']);
12
- expect(options.help).toBe(true);
13
- const longOptions = parseArgs(['--help']);
14
- expect(longOptions.help).toBe(true);
15
- });
16
- it('should show version number', () => {
17
- const consoleSpy = vi.spyOn(console, 'log');
18
- showVersion();
19
- expect(consoleSpy).toHaveBeenCalled();
20
- consoleSpy.mockRestore();
21
- });
22
- it('should show help message', () => {
23
- const consoleSpy = vi.spyOn(console, 'log');
24
- showHelp();
25
- expect(consoleSpy).toHaveBeenCalled();
26
- consoleSpy.mockRestore();
27
- });
28
- it('should default to help when no arguments provided', () => {
29
- const consoleSpy = vi.spyOn(console, 'log');
30
- cli([]);
31
- expect(consoleSpy).toHaveBeenCalled();
32
- consoleSpy.mockRestore();
33
- });
34
- });
35
- //# sourceMappingURL=index.test.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"index.test.js","sourceRoot":"","sources":["../../src/cli/index.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AACjD,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,EAAE,MAAM,SAAS,CAAA;AAE/D,QAAQ,CAAC,KAAK,EAAE,GAAG,EAAE;IACnB,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,OAAO,GAAG,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAA;QACjC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAElC,MAAM,WAAW,GAAG,SAAS,CAAC,CAAC,WAAW,CAAC,CAAC,CAAA;QAC5C,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACxC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,OAAO,GAAG,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAA;QACjC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAE/B,MAAM,WAAW,GAAG,SAAS,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAA;QACzC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACrC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAA;QAC3C,WAAW,EAAE,CAAA;QACb,MAAM,CAAC,UAAU,CAAC,CAAC,gBAAgB,EAAE,CAAA;QACrC,UAAU,CAAC,WAAW,EAAE,CAAA;IAC1B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAClC,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAA;QAC3C,QAAQ,EAAE,CAAA;QACV,MAAM,CAAC,UAAU,CAAC,CAAC,gBAAgB,EAAE,CAAA;QACrC,UAAU,CAAC,WAAW,EAAE,CAAA;IAC1B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAA;QAC3C,GAAG,CAAC,EAAE,CAAC,CAAA;QACP,MAAM,CAAC,UAAU,CAAC,CAAC,gBAAgB,EAAE,CAAA;QACrC,UAAU,CAAC,WAAW,EAAE,CAAA;IAC1B,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
@@ -1,10 +0,0 @@
1
- export declare const MODEL_BEST = "gpt-4o";
2
- export declare const MODEL_FAST = "gpt-4o-mini";
3
- export declare const DEFAULT_MODEL = "gpt-4o";
4
- export declare const FALLBACK_MODEL = "gpt-4o-mini";
5
- export declare const MODELS: {
6
- readonly BEST: "gpt-4o";
7
- readonly FAST: "gpt-4o-mini";
8
- };
9
- export type ModelType = typeof MODELS[keyof typeof MODELS];
10
- //# sourceMappingURL=models.d.ts.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../../src/constants/models.ts"],"names":[],"mappings":"AACA,eAAO,MAAM,UAAU,WAAW,CAAA;AAClC,eAAO,MAAM,UAAU,gBAAgB,CAAA;AAGvC,eAAO,MAAM,aAAa,WAAa,CAAA;AACvC,eAAO,MAAM,cAAc,gBAAa,CAAA;AAGxC,eAAO,MAAM,MAAM;;;CAGT,CAAA;AAGV,MAAM,MAAM,SAAS,GAAG,OAAO,MAAM,CAAC,MAAM,OAAO,MAAM,CAAC,CAAA"}
@@ -1,12 +0,0 @@
1
- // Model identifiers
2
- export const MODEL_BEST = 'gpt-4o';
3
- export const MODEL_FAST = 'gpt-4o-mini';
4
- // Default model configuration
5
- export const DEFAULT_MODEL = MODEL_BEST;
6
- export const FALLBACK_MODEL = MODEL_FAST;
7
- // Model constants for external use
8
- export const MODELS = {
9
- BEST: MODEL_BEST,
10
- FAST: MODEL_FAST
11
- };
12
- //# sourceMappingURL=models.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"models.js","sourceRoot":"","sources":["../../src/constants/models.ts"],"names":[],"mappings":"AAAA,oBAAoB;AACpB,MAAM,CAAC,MAAM,UAAU,GAAG,QAAQ,CAAA;AAClC,MAAM,CAAC,MAAM,UAAU,GAAG,aAAa,CAAA;AAEvC,8BAA8B;AAC9B,MAAM,CAAC,MAAM,aAAa,GAAG,UAAU,CAAA;AACvC,MAAM,CAAC,MAAM,cAAc,GAAG,UAAU,CAAA;AAExC,mCAAmC;AACnC,MAAM,CAAC,MAAM,MAAM,GAAG;IACpB,IAAI,EAAE,UAAU;IAChB,IAAI,EAAE,UAAU;CACR,CAAA"}
@@ -1,3 +0,0 @@
1
- export * from './model';
2
- export * from './schema';
3
- //# sourceMappingURL=index.d.ts.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/converters/index.ts"],"names":[],"mappings":"AAAA,cAAc,SAAS,CAAA;AACvB,cAAc,UAAU,CAAA"}
@@ -1,3 +0,0 @@
1
- export * from './model';
2
- export * from './schema';
3
- //# sourceMappingURL=index.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/converters/index.ts"],"names":[],"mappings":"AAAA,cAAc,SAAS,CAAA;AACvB,cAAc,UAAU,CAAA"}
@@ -1,4 +0,0 @@
1
- import { Model } from '../types';
2
- export declare function getOpenAIClient(): import("@ai-sdk/openai").OpenAIProvider;
3
- export declare function convertToOpenAIModel(model: Model): any;
4
- //# sourceMappingURL=model.d.ts.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"model.d.ts","sourceRoot":"","sources":["../../src/converters/model.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,EAAE,MAAM,UAAU,CAAA;AAKhC,wBAAgB,eAAe,4CAO9B;AAED,wBAAgB,oBAAoB,CAAC,KAAK,EAAE,KAAK,OAMhD"}
@@ -1,19 +0,0 @@
1
- import { createOpenAI } from '@ai-sdk/openai';
2
- // Create a singleton OpenAI client
3
- let clientInstance;
4
- export function getOpenAIClient() {
5
- if (!clientInstance) {
6
- clientInstance = createOpenAI({
7
- baseURL: process.env.AI_GATEWAY,
8
- });
9
- }
10
- return clientInstance;
11
- }
12
- export function convertToOpenAIModel(model) {
13
- if (typeof model === 'string') {
14
- const client = getOpenAIClient();
15
- return client(model);
16
- }
17
- return model;
18
- }
19
- //# sourceMappingURL=model.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"model.js","sourceRoot":"","sources":["../../src/converters/model.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAA;AAG7C,mCAAmC;AACnC,IAAI,cAA+C,CAAA;AAEnD,MAAM,UAAU,eAAe;IAC7B,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,cAAc,GAAG,YAAY,CAAC;YAC5B,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,UAAU;SAChC,CAAC,CAAA;IACJ,CAAC;IACD,OAAO,cAAc,CAAA;AACvB,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,KAAY;IAC/C,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,eAAe,EAAE,CAAA;QAChC,OAAO,MAAM,CAAC,KAAK,CAAC,CAAA;IACtB,CAAC;IACD,OAAO,KAAK,CAAA;AACd,CAAC"}
@@ -1,4 +0,0 @@
1
- import { z } from 'zod';
2
- import { SimpleSchema } from '../types';
3
- export declare function convertToZodSchema(schema: SimpleSchema): z.ZodObject<any>;
4
- //# sourceMappingURL=schema.d.ts.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"schema.d.ts","sourceRoot":"","sources":["../../src/converters/schema.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AACvB,OAAO,EAAE,YAAY,EAAqB,MAAM,UAAU,CAAA;AAiB1D,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,YAAY,GAAG,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAQzE"}
@@ -1,25 +0,0 @@
1
- import { z } from 'zod';
2
- function convertValue(value) {
3
- if (Array.isArray(value)) {
4
- return z.array(z.string().describe(value[0] || ''));
5
- }
6
- else if (typeof value === 'string') {
7
- if (value.includes('|')) {
8
- const enumValues = value.split('|').map(v => v.trim());
9
- return z.enum(enumValues);
10
- }
11
- return z.string().describe(value);
12
- }
13
- else if (typeof value === 'object') {
14
- return convertToZodSchema(value);
15
- }
16
- return z.string();
17
- }
18
- export function convertToZodSchema(schema) {
19
- const zodSchema = {};
20
- for (const [key, value] of Object.entries(schema)) {
21
- zodSchema[key] = convertValue(value);
22
- }
23
- return z.object(zodSchema);
24
- }
25
- //# sourceMappingURL=schema.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"schema.js","sourceRoot":"","sources":["../../src/converters/schema.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAGvB,SAAS,YAAY,CAAC,KAAwB;IAC5C,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAA;IACrD,CAAC;SAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACrC,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAA;YACtD,OAAO,CAAC,CAAC,IAAI,CAAC,UAAmC,CAAC,CAAA;QACpD,CAAC;QACD,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAA;IACnC,CAAC;SAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACrC,OAAO,kBAAkB,CAAC,KAAK,CAAC,CAAA;IAClC,CAAC;IACD,OAAO,CAAC,CAAC,MAAM,EAAE,CAAA;AACnB,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,MAAoB;IACrD,MAAM,SAAS,GAAiC,EAAE,CAAA;IAElD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAClD,SAAS,CAAC,GAAG,CAAC,GAAG,YAAY,CAAC,KAAK,CAAC,CAAA;IACtC,CAAC;IAED,OAAO,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAA;AAC5B,CAAC"}
@@ -1,5 +0,0 @@
1
- import type { GenerateTextResult, StreamingResult } from 'ai';
2
- export declare function createJsonResponse(result: GenerateTextResult): Response;
3
- export declare function createStreamResponse(result: StreamingResult): Response;
4
- export declare function createTextResponse(result: GenerateTextResult): Response;
5
- //# sourceMappingURL=responses.d.ts.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"responses.d.ts","sourceRoot":"","sources":["../../src/core/responses.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAyB,eAAe,EAAE,MAAM,IAAI,CAAA;AAEpF,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,kBAAkB,GAAG,QAAQ,CAIvE;AAED,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,eAAe,GAAG,QAAQ,CAItE;AAED,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,kBAAkB,GAAG,QAAQ,CAIvE"}
@@ -1,16 +0,0 @@
1
- export function createJsonResponse(result) {
2
- return new Response(JSON.stringify(result), {
3
- headers: { 'Content-Type': 'application/json' }
4
- });
5
- }
6
- export function createStreamResponse(result) {
7
- return new Response(result.experimental_stream, {
8
- headers: { 'Content-Type': 'text/event-stream' }
9
- });
10
- }
11
- export function createTextResponse(result) {
12
- return new Response(result.text, {
13
- headers: { 'Content-Type': 'text/plain' },
14
- });
15
- }
16
- //# sourceMappingURL=responses.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"responses.js","sourceRoot":"","sources":["../../src/core/responses.ts"],"names":[],"mappings":"AAEA,MAAM,UAAU,kBAAkB,CAAC,MAA0B;IAC3D,OAAO,IAAI,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE;QAC1C,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;KAChD,CAAC,CAAA;AACJ,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,MAAuB;IAC1D,OAAO,IAAI,QAAQ,CAAC,MAAM,CAAC,mBAAgD,EAAE;QAC3E,OAAO,EAAE,EAAE,cAAc,EAAE,mBAAmB,EAAE;KACjD,CAAC,CAAA;AACJ,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,MAA0B;IAC3D,OAAO,IAAI,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE;QAC/B,OAAO,EAAE,EAAE,cAAc,EAAE,YAAY,EAAE;KAC1C,CAAC,CAAA;AACJ,CAAC"}
@@ -1,2 +0,0 @@
1
- export {};
2
- //# sourceMappingURL=responses.test.d.ts.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"responses.test.d.ts","sourceRoot":"","sources":["../../src/core/responses.test.ts"],"names":[],"mappings":""}