@ax-llm/ax 9.0.21 → 9.0.22

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 (570) hide show
  1. package/ai/anthropic/api.d.ts +19 -0
  2. package/ai/anthropic/api.js +326 -0
  3. package/ai/anthropic/api.js.map +1 -0
  4. package/ai/anthropic/index.d.ts +2 -0
  5. package/ai/anthropic/index.js +3 -0
  6. package/ai/anthropic/index.js.map +1 -0
  7. package/ai/anthropic/info.d.ts +2 -0
  8. package/ai/anthropic/info.js +40 -0
  9. package/ai/anthropic/info.js.map +1 -0
  10. package/ai/anthropic/types.d.ts +171 -0
  11. package/ai/anthropic/types.js +10 -0
  12. package/ai/anthropic/types.js.map +1 -0
  13. package/ai/azure-openai/api.d.ts +19 -0
  14. package/ai/azure-openai/api.js +30 -0
  15. package/ai/azure-openai/api.js.map +1 -0
  16. package/ai/azure-openai/index.d.ts +1 -0
  17. package/ai/azure-openai/index.js +2 -0
  18. package/ai/azure-openai/index.js.map +1 -0
  19. package/ai/balance.d.ts +23 -0
  20. package/ai/balance.js +87 -0
  21. package/ai/balance.js.map +1 -0
  22. package/ai/base.d.ts +55 -0
  23. package/ai/base.js +343 -0
  24. package/ai/base.js.map +1 -0
  25. package/ai/cohere/api.d.ts +22 -0
  26. package/ai/cohere/api.js +264 -0
  27. package/ai/cohere/api.js.map +1 -0
  28. package/ai/cohere/index.d.ts +2 -0
  29. package/ai/cohere/index.js +3 -0
  30. package/ai/cohere/index.js.map +1 -0
  31. package/ai/cohere/info.d.ts +2 -0
  32. package/ai/cohere/info.js +52 -0
  33. package/ai/cohere/info.js.map +1 -0
  34. package/ai/cohere/types.d.ts +102 -0
  35. package/ai/cohere/types.js +23 -0
  36. package/ai/cohere/types.js.map +1 -0
  37. package/ai/deepseek/api.d.ts +16 -0
  38. package/ai/deepseek/api.js +32 -0
  39. package/ai/deepseek/api.js.map +1 -0
  40. package/ai/deepseek/index.d.ts +2 -0
  41. package/ai/deepseek/index.js +3 -0
  42. package/ai/deepseek/index.js.map +1 -0
  43. package/ai/deepseek/info.d.ts +2 -0
  44. package/ai/deepseek/info.js +16 -0
  45. package/ai/deepseek/info.js.map +1 -0
  46. package/ai/deepseek/types.d.ts +8 -0
  47. package/ai/deepseek/types.js +10 -0
  48. package/ai/deepseek/types.js.map +1 -0
  49. package/ai/google-gemini/api.d.ts +36 -0
  50. package/ai/google-gemini/api.js +299 -0
  51. package/ai/google-gemini/api.js.map +1 -0
  52. package/ai/google-gemini/index.d.ts +2 -0
  53. package/ai/google-gemini/index.js +3 -0
  54. package/ai/google-gemini/index.js.map +1 -0
  55. package/ai/google-gemini/info.d.ts +6 -0
  56. package/ai/google-gemini/info.js +29 -0
  57. package/ai/google-gemini/info.js.map +1 -0
  58. package/ai/google-gemini/types.d.ts +146 -0
  59. package/ai/google-gemini/types.js +27 -0
  60. package/ai/google-gemini/types.js.map +1 -0
  61. package/ai/groq/api.d.ts +18 -0
  62. package/ai/groq/api.js +52 -0
  63. package/ai/groq/api.js.map +1 -0
  64. package/ai/groq/index.d.ts +1 -0
  65. package/ai/groq/index.js +2 -0
  66. package/ai/groq/index.js.map +1 -0
  67. package/ai/groq/types.d.ts +6 -0
  68. package/ai/groq/types.js +8 -0
  69. package/ai/groq/types.js.map +1 -0
  70. package/ai/huggingface/api.d.ts +19 -0
  71. package/ai/huggingface/api.js +115 -0
  72. package/ai/huggingface/api.js.map +1 -0
  73. package/ai/huggingface/index.d.ts +2 -0
  74. package/ai/huggingface/index.js +3 -0
  75. package/ai/huggingface/index.js.map +1 -0
  76. package/ai/huggingface/info.d.ts +6 -0
  77. package/ai/huggingface/info.js +6 -0
  78. package/ai/huggingface/info.js.map +1 -0
  79. package/ai/huggingface/types.d.ts +34 -0
  80. package/ai/huggingface/types.js +5 -0
  81. package/ai/huggingface/types.js.map +1 -0
  82. package/ai/index.d.ts +16 -0
  83. package/ai/index.js +16 -0
  84. package/ai/index.js.map +1 -0
  85. package/ai/mistral/api.d.ts +16 -0
  86. package/ai/mistral/api.js +32 -0
  87. package/ai/mistral/api.js.map +1 -0
  88. package/ai/mistral/index.d.ts +2 -0
  89. package/ai/mistral/index.js +3 -0
  90. package/ai/mistral/index.js.map +1 -0
  91. package/ai/mistral/info.d.ts +2 -0
  92. package/ai/mistral/info.js +34 -0
  93. package/ai/mistral/info.js.map +1 -0
  94. package/ai/mistral/types.d.ts +10 -0
  95. package/ai/mistral/types.js +13 -0
  96. package/ai/mistral/types.js.map +1 -0
  97. package/ai/ollama/api.d.ts +22 -0
  98. package/ai/ollama/api.js +32 -0
  99. package/ai/ollama/api.js.map +1 -0
  100. package/ai/ollama/index.d.ts +1 -0
  101. package/ai/ollama/index.js +2 -0
  102. package/ai/ollama/index.js.map +1 -0
  103. package/ai/openai/api.d.ts +29 -0
  104. package/ai/openai/api.js +283 -0
  105. package/ai/openai/api.js.map +1 -0
  106. package/ai/openai/index.d.ts +2 -0
  107. package/ai/openai/index.js +3 -0
  108. package/ai/openai/index.js.map +1 -0
  109. package/ai/openai/info.d.ts +6 -0
  110. package/ai/openai/info.js +50 -0
  111. package/ai/openai/info.js.map +1 -0
  112. package/ai/openai/types.d.ts +165 -0
  113. package/ai/openai/types.js +18 -0
  114. package/ai/openai/types.js.map +1 -0
  115. package/ai/together/api.d.ts +15 -0
  116. package/ai/together/api.js +28 -0
  117. package/ai/together/api.js.map +1 -0
  118. package/ai/together/index.d.ts +1 -0
  119. package/ai/together/index.js +2 -0
  120. package/ai/together/index.js.map +1 -0
  121. package/ai/together/info.d.ts +2 -0
  122. package/ai/together/info.js +2 -0
  123. package/ai/together/info.js.map +1 -0
  124. package/ai/types.d.ts +159 -0
  125. package/ai/types.js +2 -0
  126. package/ai/types.js.map +1 -0
  127. package/ai/util.d.ts +12 -0
  128. package/ai/util.js +63 -0
  129. package/ai/util.js.map +1 -0
  130. package/ai/wrap.d.ts +47 -0
  131. package/ai/wrap.js +95 -0
  132. package/ai/wrap.js.map +1 -0
  133. package/cjs/ai/anthropic/api.d.ts +19 -0
  134. package/cjs/ai/anthropic/api.js +331 -0
  135. package/cjs/ai/anthropic/api.js.map +1 -0
  136. package/cjs/ai/anthropic/index.d.ts +2 -0
  137. package/cjs/ai/anthropic/index.js +19 -0
  138. package/cjs/ai/anthropic/index.js.map +1 -0
  139. package/cjs/ai/anthropic/info.d.ts +2 -0
  140. package/cjs/ai/anthropic/info.js +43 -0
  141. package/cjs/ai/anthropic/info.js.map +1 -0
  142. package/cjs/ai/anthropic/types.d.ts +171 -0
  143. package/cjs/ai/anthropic/types.js +13 -0
  144. package/cjs/ai/anthropic/types.js.map +1 -0
  145. package/cjs/ai/azure-openai/api.d.ts +19 -0
  146. package/cjs/ai/azure-openai/api.js +34 -0
  147. package/cjs/ai/azure-openai/api.js.map +1 -0
  148. package/cjs/ai/azure-openai/index.d.ts +1 -0
  149. package/cjs/ai/azure-openai/index.js +18 -0
  150. package/cjs/ai/azure-openai/index.js.map +1 -0
  151. package/cjs/ai/balance.d.ts +23 -0
  152. package/cjs/ai/balance.js +91 -0
  153. package/cjs/ai/balance.js.map +1 -0
  154. package/cjs/ai/base.d.ts +55 -0
  155. package/cjs/ai/base.js +349 -0
  156. package/cjs/ai/base.js.map +1 -0
  157. package/cjs/ai/cohere/api.d.ts +22 -0
  158. package/cjs/ai/cohere/api.js +270 -0
  159. package/cjs/ai/cohere/api.js.map +1 -0
  160. package/cjs/ai/cohere/index.d.ts +2 -0
  161. package/cjs/ai/cohere/index.js +19 -0
  162. package/cjs/ai/cohere/index.js.map +1 -0
  163. package/cjs/ai/cohere/info.d.ts +2 -0
  164. package/cjs/ai/cohere/info.js +55 -0
  165. package/cjs/ai/cohere/info.js.map +1 -0
  166. package/cjs/ai/cohere/types.d.ts +102 -0
  167. package/cjs/ai/cohere/types.js +26 -0
  168. package/cjs/ai/cohere/types.js.map +1 -0
  169. package/cjs/ai/deepseek/api.d.ts +16 -0
  170. package/cjs/ai/deepseek/api.js +38 -0
  171. package/cjs/ai/deepseek/api.js.map +1 -0
  172. package/cjs/ai/deepseek/index.d.ts +2 -0
  173. package/cjs/ai/deepseek/index.js +19 -0
  174. package/cjs/ai/deepseek/index.js.map +1 -0
  175. package/cjs/ai/deepseek/info.d.ts +2 -0
  176. package/cjs/ai/deepseek/info.js +19 -0
  177. package/cjs/ai/deepseek/info.js.map +1 -0
  178. package/cjs/ai/deepseek/types.d.ts +8 -0
  179. package/cjs/ai/deepseek/types.js +13 -0
  180. package/cjs/ai/deepseek/types.js.map +1 -0
  181. package/cjs/ai/google-gemini/api.d.ts +36 -0
  182. package/cjs/ai/google-gemini/api.js +305 -0
  183. package/cjs/ai/google-gemini/api.js.map +1 -0
  184. package/cjs/ai/google-gemini/index.d.ts +2 -0
  185. package/cjs/ai/google-gemini/index.js +19 -0
  186. package/cjs/ai/google-gemini/index.js.map +1 -0
  187. package/cjs/ai/google-gemini/info.d.ts +6 -0
  188. package/cjs/ai/google-gemini/info.js +32 -0
  189. package/cjs/ai/google-gemini/info.js.map +1 -0
  190. package/cjs/ai/google-gemini/types.d.ts +146 -0
  191. package/cjs/ai/google-gemini/types.js +30 -0
  192. package/cjs/ai/google-gemini/types.js.map +1 -0
  193. package/cjs/ai/groq/api.d.ts +18 -0
  194. package/cjs/ai/groq/api.js +56 -0
  195. package/cjs/ai/groq/api.js.map +1 -0
  196. package/cjs/ai/groq/index.d.ts +1 -0
  197. package/cjs/ai/groq/index.js +18 -0
  198. package/cjs/ai/groq/index.js.map +1 -0
  199. package/cjs/ai/groq/types.d.ts +6 -0
  200. package/cjs/ai/groq/types.js +11 -0
  201. package/cjs/ai/groq/types.js.map +1 -0
  202. package/cjs/ai/huggingface/api.d.ts +19 -0
  203. package/cjs/ai/huggingface/api.js +121 -0
  204. package/cjs/ai/huggingface/api.js.map +1 -0
  205. package/cjs/ai/huggingface/index.d.ts +2 -0
  206. package/cjs/ai/huggingface/index.js +19 -0
  207. package/cjs/ai/huggingface/index.js.map +1 -0
  208. package/cjs/ai/huggingface/info.d.ts +6 -0
  209. package/cjs/ai/huggingface/info.js +9 -0
  210. package/cjs/ai/huggingface/info.js.map +1 -0
  211. package/cjs/ai/huggingface/types.d.ts +34 -0
  212. package/cjs/ai/huggingface/types.js +8 -0
  213. package/cjs/ai/huggingface/types.js.map +1 -0
  214. package/cjs/ai/index.d.ts +16 -0
  215. package/cjs/ai/index.js +32 -0
  216. package/cjs/ai/index.js.map +1 -0
  217. package/cjs/ai/mistral/api.d.ts +16 -0
  218. package/cjs/ai/mistral/api.js +38 -0
  219. package/cjs/ai/mistral/api.js.map +1 -0
  220. package/cjs/ai/mistral/index.d.ts +2 -0
  221. package/cjs/ai/mistral/index.js +19 -0
  222. package/cjs/ai/mistral/index.js.map +1 -0
  223. package/cjs/ai/mistral/info.d.ts +2 -0
  224. package/cjs/ai/mistral/info.js +37 -0
  225. package/cjs/ai/mistral/info.js.map +1 -0
  226. package/cjs/ai/mistral/types.d.ts +10 -0
  227. package/cjs/ai/mistral/types.js +16 -0
  228. package/cjs/ai/mistral/types.js.map +1 -0
  229. package/cjs/ai/ollama/api.d.ts +22 -0
  230. package/cjs/ai/ollama/api.js +38 -0
  231. package/cjs/ai/ollama/api.js.map +1 -0
  232. package/cjs/ai/ollama/index.d.ts +1 -0
  233. package/cjs/ai/ollama/index.js +18 -0
  234. package/cjs/ai/ollama/index.js.map +1 -0
  235. package/cjs/ai/openai/api.d.ts +29 -0
  236. package/cjs/ai/openai/api.js +291 -0
  237. package/cjs/ai/openai/api.js.map +1 -0
  238. package/cjs/ai/openai/index.d.ts +2 -0
  239. package/cjs/ai/openai/index.js +19 -0
  240. package/cjs/ai/openai/index.js.map +1 -0
  241. package/cjs/ai/openai/info.d.ts +6 -0
  242. package/cjs/ai/openai/info.js +53 -0
  243. package/cjs/ai/openai/info.js.map +1 -0
  244. package/cjs/ai/openai/types.d.ts +165 -0
  245. package/cjs/ai/openai/types.js +21 -0
  246. package/cjs/ai/openai/types.js.map +1 -0
  247. package/cjs/ai/together/api.d.ts +15 -0
  248. package/cjs/ai/together/api.js +33 -0
  249. package/cjs/ai/together/api.js.map +1 -0
  250. package/cjs/ai/together/index.d.ts +1 -0
  251. package/cjs/ai/together/index.js +18 -0
  252. package/cjs/ai/together/index.js.map +1 -0
  253. package/cjs/ai/together/info.d.ts +2 -0
  254. package/cjs/ai/together/info.js +5 -0
  255. package/cjs/ai/together/info.js.map +1 -0
  256. package/cjs/ai/types.d.ts +159 -0
  257. package/cjs/ai/types.js +3 -0
  258. package/cjs/ai/types.js.map +1 -0
  259. package/cjs/ai/util.d.ts +12 -0
  260. package/cjs/ai/util.js +71 -0
  261. package/cjs/ai/util.js.map +1 -0
  262. package/cjs/ai/wrap.d.ts +47 -0
  263. package/cjs/ai/wrap.js +99 -0
  264. package/cjs/ai/wrap.js.map +1 -0
  265. package/cjs/db/base.d.ts +23 -0
  266. package/cjs/db/base.js +91 -0
  267. package/cjs/db/base.js.map +1 -0
  268. package/cjs/db/cloudflare.d.ts +21 -0
  269. package/cjs/db/cloudflare.js +102 -0
  270. package/cjs/db/cloudflare.js.map +1 -0
  271. package/cjs/db/index.d.ts +6 -0
  272. package/cjs/db/index.js +23 -0
  273. package/cjs/db/index.js.map +1 -0
  274. package/cjs/db/memory.d.ts +21 -0
  275. package/cjs/db/memory.js +103 -0
  276. package/cjs/db/memory.js.map +1 -0
  277. package/cjs/db/pinecone.d.ts +21 -0
  278. package/cjs/db/pinecone.js +75 -0
  279. package/cjs/db/pinecone.js.map +1 -0
  280. package/cjs/db/types.d.ts +35 -0
  281. package/cjs/db/types.js +4 -0
  282. package/cjs/db/types.js.map +1 -0
  283. package/cjs/db/weaviate.d.ts +21 -0
  284. package/cjs/db/weaviate.js +131 -0
  285. package/cjs/db/weaviate.js.map +1 -0
  286. package/cjs/db/wrap.d.ts +13 -0
  287. package/cjs/db/wrap.js +39 -0
  288. package/cjs/db/wrap.js.map +1 -0
  289. package/cjs/docs/index.d.ts +4 -0
  290. package/cjs/docs/index.js +21 -0
  291. package/cjs/docs/index.js.map +1 -0
  292. package/cjs/docs/manager.d.ts +45 -0
  293. package/cjs/docs/manager.js +162 -0
  294. package/cjs/docs/manager.js.map +1 -0
  295. package/cjs/docs/reranker.d.ts +7 -0
  296. package/cjs/docs/reranker.js +30 -0
  297. package/cjs/docs/reranker.js.map +1 -0
  298. package/cjs/docs/rewriter.d.ts +4 -0
  299. package/cjs/docs/rewriter.js +13 -0
  300. package/cjs/docs/rewriter.js.map +1 -0
  301. package/cjs/docs/tika.d.ts +18 -0
  302. package/cjs/docs/tika.js +50 -0
  303. package/cjs/docs/tika.js.map +1 -0
  304. package/cjs/dsp/asserts.d.ts +32 -0
  305. package/cjs/dsp/asserts.js +98 -0
  306. package/cjs/dsp/asserts.js.map +1 -0
  307. package/cjs/dsp/eval.d.ts +43 -0
  308. package/cjs/dsp/eval.js +128 -0
  309. package/cjs/dsp/eval.js.map +1 -0
  310. package/cjs/dsp/evaluate.d.ts +12 -0
  311. package/cjs/dsp/evaluate.js +36 -0
  312. package/cjs/dsp/evaluate.js.map +1 -0
  313. package/cjs/dsp/extract.d.ts +25 -0
  314. package/cjs/dsp/extract.js +164 -0
  315. package/cjs/dsp/extract.js.map +1 -0
  316. package/cjs/dsp/generate.d.ts +55 -0
  317. package/cjs/dsp/generate.js +314 -0
  318. package/cjs/dsp/generate.js.map +1 -0
  319. package/cjs/dsp/index.d.ts +15 -0
  320. package/cjs/dsp/index.js +27 -0
  321. package/cjs/dsp/index.js.map +1 -0
  322. package/cjs/dsp/loader.d.ts +30 -0
  323. package/cjs/dsp/loader.js +85 -0
  324. package/cjs/dsp/loader.js.map +1 -0
  325. package/cjs/dsp/optimize.d.ts +27 -0
  326. package/cjs/dsp/optimize.js +99 -0
  327. package/cjs/dsp/optimize.js.map +1 -0
  328. package/cjs/dsp/parser.d.ts +87 -0
  329. package/cjs/dsp/parser.js +1545 -0
  330. package/cjs/dsp/parser.js.map +1 -0
  331. package/cjs/dsp/program.d.ts +73 -0
  332. package/cjs/dsp/program.js +120 -0
  333. package/cjs/dsp/program.js.map +1 -0
  334. package/cjs/dsp/prompt.d.ts +48 -0
  335. package/cjs/dsp/prompt.js +263 -0
  336. package/cjs/dsp/prompt.js.map +1 -0
  337. package/cjs/dsp/registry.d.ts +6 -0
  338. package/cjs/dsp/registry.js +19 -0
  339. package/cjs/dsp/registry.js.map +1 -0
  340. package/cjs/dsp/router.d.ts +26 -0
  341. package/cjs/dsp/router.js +76 -0
  342. package/cjs/dsp/router.js.map +1 -0
  343. package/cjs/dsp/sig.d.ts +37 -0
  344. package/cjs/dsp/sig.js +197 -0
  345. package/cjs/dsp/sig.js.map +1 -0
  346. package/cjs/dsp/stopwords.d.ts +1 -0
  347. package/cjs/dsp/stopwords.js +1167 -0
  348. package/cjs/dsp/stopwords.js.map +1 -0
  349. package/cjs/dsp/strutil.d.ts +11 -0
  350. package/cjs/dsp/strutil.js +63 -0
  351. package/cjs/dsp/strutil.js.map +1 -0
  352. package/cjs/dsp/util.d.ts +5 -0
  353. package/cjs/dsp/util.js +82 -0
  354. package/cjs/dsp/util.js.map +1 -0
  355. package/cjs/funcs/code.d.ts +9 -0
  356. package/cjs/funcs/code.js +90 -0
  357. package/cjs/funcs/code.js.map +1 -0
  358. package/cjs/funcs/embed.d.ts +6 -0
  359. package/cjs/funcs/embed.js +33 -0
  360. package/cjs/funcs/embed.js.map +1 -0
  361. package/cjs/funcs/functions.d.ts +16 -0
  362. package/cjs/funcs/functions.js +62 -0
  363. package/cjs/funcs/functions.js.map +1 -0
  364. package/cjs/funcs/index.d.ts +3 -0
  365. package/cjs/funcs/index.js +20 -0
  366. package/cjs/funcs/index.js.map +1 -0
  367. package/cjs/funcs/jsonschema.d.ts +2 -0
  368. package/cjs/funcs/jsonschema.js +101 -0
  369. package/cjs/funcs/jsonschema.js.map +1 -0
  370. package/cjs/index.d.ts +9 -0
  371. package/cjs/index.js +34 -0
  372. package/cjs/index.js.map +1 -0
  373. package/cjs/mem/index.d.ts +2 -0
  374. package/cjs/mem/index.js +19 -0
  375. package/cjs/mem/index.js.map +1 -0
  376. package/cjs/mem/memory.d.ts +16 -0
  377. package/cjs/mem/memory.js +81 -0
  378. package/cjs/mem/memory.js.map +1 -0
  379. package/cjs/mem/types.d.ts +10 -0
  380. package/cjs/mem/types.js +3 -0
  381. package/cjs/mem/types.js.map +1 -0
  382. package/cjs/prompts/agent.d.ts +34 -0
  383. package/cjs/prompts/agent.js +110 -0
  384. package/cjs/prompts/agent.js.map +1 -0
  385. package/cjs/prompts/cot.d.ts +9 -0
  386. package/cjs/prompts/cot.js +21 -0
  387. package/cjs/prompts/cot.js.map +1 -0
  388. package/cjs/prompts/index.d.ts +4 -0
  389. package/cjs/prompts/index.js +21 -0
  390. package/cjs/prompts/index.js.map +1 -0
  391. package/cjs/prompts/rag.d.ts +23 -0
  392. package/cjs/prompts/rag.js +33 -0
  393. package/cjs/prompts/rag.js.map +1 -0
  394. package/cjs/prompts/react.d.ts +8 -0
  395. package/cjs/prompts/react.js +26 -0
  396. package/cjs/prompts/react.js.map +1 -0
  397. package/cjs/trace/index.d.ts +171 -0
  398. package/cjs/trace/index.js +70 -0
  399. package/cjs/trace/index.js.map +1 -0
  400. package/cjs/util/apicall.d.ts +18 -0
  401. package/cjs/util/apicall.js +57 -0
  402. package/cjs/util/apicall.js.map +1 -0
  403. package/cjs/util/log.d.ts +14 -0
  404. package/cjs/util/log.js +35 -0
  405. package/cjs/util/log.js.map +1 -0
  406. package/cjs/util/other.d.ts +1 -0
  407. package/cjs/util/other.js +8 -0
  408. package/cjs/util/other.js.map +1 -0
  409. package/cjs/util/rate-limit.d.ts +24 -0
  410. package/cjs/util/rate-limit.js +53 -0
  411. package/cjs/util/rate-limit.js.map +1 -0
  412. package/cjs/util/stream.d.ts +9 -0
  413. package/cjs/util/stream.js +33 -0
  414. package/cjs/util/stream.js.map +1 -0
  415. package/cjs/util/transform.d.ts +7 -0
  416. package/cjs/util/transform.js +64 -0
  417. package/cjs/util/transform.js.map +1 -0
  418. package/db/base.d.ts +23 -0
  419. package/db/base.js +87 -0
  420. package/db/base.js.map +1 -0
  421. package/db/cloudflare.d.ts +21 -0
  422. package/db/cloudflare.js +98 -0
  423. package/db/cloudflare.js.map +1 -0
  424. package/db/index.d.ts +6 -0
  425. package/db/index.js +7 -0
  426. package/db/index.js.map +1 -0
  427. package/db/memory.d.ts +21 -0
  428. package/db/memory.js +99 -0
  429. package/db/memory.js.map +1 -0
  430. package/db/pinecone.d.ts +21 -0
  431. package/db/pinecone.js +71 -0
  432. package/db/pinecone.js.map +1 -0
  433. package/db/types.d.ts +35 -0
  434. package/db/types.js +3 -0
  435. package/db/types.js.map +1 -0
  436. package/db/weaviate.d.ts +21 -0
  437. package/db/weaviate.js +127 -0
  438. package/db/weaviate.js.map +1 -0
  439. package/db/wrap.d.ts +13 -0
  440. package/db/wrap.js +35 -0
  441. package/db/wrap.js.map +1 -0
  442. package/docs/index.d.ts +4 -0
  443. package/docs/index.js +5 -0
  444. package/docs/index.js.map +1 -0
  445. package/docs/manager.d.ts +45 -0
  446. package/docs/manager.js +159 -0
  447. package/docs/manager.js.map +1 -0
  448. package/docs/reranker.d.ts +7 -0
  449. package/docs/reranker.js +26 -0
  450. package/docs/reranker.js.map +1 -0
  451. package/docs/rewriter.d.ts +4 -0
  452. package/docs/rewriter.js +9 -0
  453. package/docs/rewriter.js.map +1 -0
  454. package/docs/tika.d.ts +18 -0
  455. package/docs/tika.js +46 -0
  456. package/docs/tika.js.map +1 -0
  457. package/dsp/asserts.d.ts +32 -0
  458. package/dsp/asserts.js +91 -0
  459. package/dsp/asserts.js.map +1 -0
  460. package/dsp/eval.d.ts +43 -0
  461. package/dsp/eval.js +125 -0
  462. package/dsp/eval.js.map +1 -0
  463. package/dsp/evaluate.d.ts +12 -0
  464. package/dsp/evaluate.js +32 -0
  465. package/dsp/evaluate.js.map +1 -0
  466. package/dsp/extract.d.ts +25 -0
  467. package/dsp/extract.js +153 -0
  468. package/dsp/extract.js.map +1 -0
  469. package/dsp/generate.d.ts +55 -0
  470. package/dsp/generate.js +311 -0
  471. package/dsp/generate.js.map +1 -0
  472. package/dsp/index.d.ts +15 -0
  473. package/dsp/index.js +11 -0
  474. package/dsp/index.js.map +1 -0
  475. package/dsp/loader.d.ts +30 -0
  476. package/dsp/loader.js +81 -0
  477. package/dsp/loader.js.map +1 -0
  478. package/dsp/optimize.d.ts +27 -0
  479. package/dsp/optimize.js +95 -0
  480. package/dsp/optimize.js.map +1 -0
  481. package/dsp/parser.d.ts +87 -0
  482. package/dsp/parser.js +1542 -0
  483. package/dsp/parser.js.map +1 -0
  484. package/dsp/program.d.ts +73 -0
  485. package/dsp/program.js +116 -0
  486. package/dsp/program.js.map +1 -0
  487. package/dsp/prompt.d.ts +48 -0
  488. package/dsp/prompt.js +260 -0
  489. package/dsp/prompt.js.map +1 -0
  490. package/dsp/registry.d.ts +6 -0
  491. package/dsp/registry.js +15 -0
  492. package/dsp/registry.js.map +1 -0
  493. package/dsp/router.d.ts +26 -0
  494. package/dsp/router.js +71 -0
  495. package/dsp/router.js.map +1 -0
  496. package/dsp/sig.d.ts +37 -0
  497. package/dsp/sig.js +193 -0
  498. package/dsp/sig.js.map +1 -0
  499. package/dsp/stopwords.d.ts +1 -0
  500. package/dsp/stopwords.js +1164 -0
  501. package/dsp/stopwords.js.map +1 -0
  502. package/dsp/strutil.d.ts +11 -0
  503. package/dsp/strutil.js +60 -0
  504. package/dsp/strutil.js.map +1 -0
  505. package/dsp/util.d.ts +5 -0
  506. package/dsp/util.js +76 -0
  507. package/dsp/util.js.map +1 -0
  508. package/funcs/code.d.ts +9 -0
  509. package/funcs/code.js +63 -0
  510. package/funcs/code.js.map +1 -0
  511. package/funcs/embed.d.ts +6 -0
  512. package/funcs/embed.js +29 -0
  513. package/funcs/embed.js.map +1 -0
  514. package/funcs/functions.d.ts +16 -0
  515. package/funcs/functions.js +55 -0
  516. package/funcs/functions.js.map +1 -0
  517. package/funcs/index.d.ts +3 -0
  518. package/funcs/index.js +4 -0
  519. package/funcs/index.js.map +1 -0
  520. package/funcs/jsonschema.d.ts +2 -0
  521. package/funcs/jsonschema.js +97 -0
  522. package/funcs/jsonschema.js.map +1 -0
  523. package/index.d.ts +9 -0
  524. package/index.js.map +1 -0
  525. package/mem/index.d.ts +2 -0
  526. package/mem/index.js +3 -0
  527. package/mem/index.js.map +1 -0
  528. package/mem/memory.d.ts +16 -0
  529. package/mem/memory.js +77 -0
  530. package/mem/memory.js.map +1 -0
  531. package/mem/types.d.ts +10 -0
  532. package/mem/types.js +2 -0
  533. package/mem/types.js.map +1 -0
  534. package/package.json +4 -7
  535. package/prompts/agent.d.ts +34 -0
  536. package/prompts/agent.js +108 -0
  537. package/prompts/agent.js.map +1 -0
  538. package/prompts/cot.d.ts +9 -0
  539. package/prompts/cot.js +17 -0
  540. package/prompts/cot.js.map +1 -0
  541. package/prompts/index.d.ts +4 -0
  542. package/prompts/index.js +5 -0
  543. package/prompts/index.js.map +1 -0
  544. package/prompts/rag.d.ts +23 -0
  545. package/prompts/rag.js +30 -0
  546. package/prompts/rag.js.map +1 -0
  547. package/prompts/react.d.ts +8 -0
  548. package/prompts/react.js +22 -0
  549. package/prompts/react.js.map +1 -0
  550. package/trace/index.d.ts +171 -0
  551. package/trace/index.js +67 -0
  552. package/trace/index.js.map +1 -0
  553. package/util/apicall.d.ts +18 -0
  554. package/util/apicall.js +50 -0
  555. package/util/apicall.js.map +1 -0
  556. package/util/log.d.ts +14 -0
  557. package/util/log.js +31 -0
  558. package/util/log.js.map +1 -0
  559. package/util/other.d.ts +1 -0
  560. package/util/other.js +5 -0
  561. package/util/other.js.map +1 -0
  562. package/util/rate-limit.d.ts +24 -0
  563. package/util/rate-limit.js +49 -0
  564. package/util/rate-limit.js.map +1 -0
  565. package/util/stream.d.ts +9 -0
  566. package/util/stream.js +29 -0
  567. package/util/stream.js.map +1 -0
  568. package/util/transform.d.ts +7 -0
  569. package/util/transform.js +59 -0
  570. package/util/transform.js.map +1 -0
@@ -0,0 +1,70 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.AxSpanKind = exports.AxSpanStatusCode = exports.AxSpanKindValues = exports.AxLLMRequestTypeValues = exports.axSpanEvents = exports.axSpanAttributes = void 0;
4
+ exports.axSpanAttributes = {
5
+ // LLM
6
+ LLM_SYSTEM: 'gen_ai.system',
7
+ LLM_REQUEST_MODEL: 'gen_ai.request.model',
8
+ LLM_REQUEST_MAX_TOKENS: 'gen_ai.request.max_tokens',
9
+ LLM_REQUEST_TEMPERATURE: 'gen_ai.request.temperature',
10
+ LLM_REQUEST_TOP_K: 'gen_ai.request.top_k',
11
+ LLM_REQUEST_FREQUENCY_PENALTY: 'gen_ai.request.frequency_penalty',
12
+ LLM_REQUEST_PRESENCE_PENALTY: 'gen_ai.request.presence_penalty',
13
+ LLM_REQUEST_STOP_SEQUENCES: 'gen_ai.request.stop_sequences',
14
+ LLM_REQUEST_LLM_IS_STREAMING: 'gen_ai.request.llm_is_streaming',
15
+ LLM_REQUEST_TOP_P: 'gen_ai.request.top_p',
16
+ LLM_USAGE_PROMPT_TOKENS: 'gen_ai.usage.prompt_tokens',
17
+ LLM_USAGE_COMPLETION_TOKENS: 'gen_ai.usage.completion_tokens',
18
+ // Vector DB
19
+ DB_SYSTEM: 'db.system',
20
+ DB_TABLE: 'db.table',
21
+ DB_NAMESPACE: 'db.namespace',
22
+ DB_ID: 'db.id',
23
+ DB_QUERY_TEXT: 'db.query.text',
24
+ DB_VECTOR: 'db.vector',
25
+ DB_OPERATION_NAME: 'db.operation.name',
26
+ DB_VECTOR_QUERY_TOP_K: 'db.vector.query.top_k',
27
+ DB_QUERY_EMBEDDINGS: 'db.query.embeddings',
28
+ DB_QUERY_RESULT: 'db.query.result',
29
+ // Query Embeddings
30
+ DB_QUERY_EMBEDDINGS_VECTOR: 'db.query.embeddings.vector',
31
+ // Query Result (canonical format)
32
+ DB_QUERY_RESULT_ID: 'db.query.result.id',
33
+ DB_QUERY_RESULT_SCORE: 'db.query.result.score',
34
+ DB_QUERY_RESULT_DISTANCE: 'db.query.result.distance',
35
+ DB_QUERY_RESULT_METADATA: 'db.query.result.metadata',
36
+ DB_QUERY_RESULT_VECTOR: 'db.query.result.vector',
37
+ DB_QUERY_RESULT_DOCUMENT: 'db.query.result.document'
38
+ };
39
+ exports.axSpanEvents = {
40
+ LLM_PROMPT: 'gen_ai.prompt'
41
+ };
42
+ var AxLLMRequestTypeValues;
43
+ (function (AxLLMRequestTypeValues) {
44
+ AxLLMRequestTypeValues["COMPLETION"] = "completion";
45
+ AxLLMRequestTypeValues["CHAT"] = "chat";
46
+ AxLLMRequestTypeValues["RERANK"] = "rerank";
47
+ AxLLMRequestTypeValues["UNKNOWN"] = "unknown";
48
+ })(AxLLMRequestTypeValues || (exports.AxLLMRequestTypeValues = AxLLMRequestTypeValues = {}));
49
+ var AxSpanKindValues;
50
+ (function (AxSpanKindValues) {
51
+ AxSpanKindValues["WORKFLOW"] = "workflow";
52
+ AxSpanKindValues["TASK"] = "task";
53
+ AxSpanKindValues["AGENT"] = "agent";
54
+ AxSpanKindValues["TOOL"] = "tool";
55
+ AxSpanKindValues["UNKNOWN"] = "unknown";
56
+ })(AxSpanKindValues || (exports.AxSpanKindValues = AxSpanKindValues = {}));
57
+ var AxSpanStatusCode;
58
+ (function (AxSpanStatusCode) {
59
+ AxSpanStatusCode["OK"] = "OK";
60
+ AxSpanStatusCode["ERROR"] = "ERROR";
61
+ })(AxSpanStatusCode || (exports.AxSpanStatusCode = AxSpanStatusCode = {}));
62
+ var AxSpanKind;
63
+ (function (AxSpanKind) {
64
+ AxSpanKind["INTERNAL"] = "INTERNAL";
65
+ AxSpanKind["SERVER"] = "SERVER";
66
+ AxSpanKind["CLIENT"] = "CLIENT";
67
+ AxSpanKind["PRODUCER"] = "PRODUCER";
68
+ AxSpanKind["CONSUMER"] = "CONSUMER";
69
+ })(AxSpanKind || (exports.AxSpanKind = AxSpanKind = {}));
70
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../trace/index.ts"],"names":[],"mappings":";;;AAAa,QAAA,gBAAgB,GAAG;IAC9B,MAAM;IACN,UAAU,EAAE,eAAe;IAC3B,iBAAiB,EAAE,sBAAsB;IACzC,sBAAsB,EAAE,2BAA2B;IACnD,uBAAuB,EAAE,4BAA4B;IACrD,iBAAiB,EAAE,sBAAsB;IACzC,6BAA6B,EAAE,kCAAkC;IACjE,4BAA4B,EAAE,iCAAiC;IAC/D,0BAA0B,EAAE,+BAA+B;IAC3D,4BAA4B,EAAE,iCAAiC;IAC/D,iBAAiB,EAAE,sBAAsB;IAEzC,uBAAuB,EAAE,4BAA4B;IACrD,2BAA2B,EAAE,gCAAgC;IAE7D,YAAY;IACZ,SAAS,EAAE,WAAW;IACtB,QAAQ,EAAE,UAAU;IACpB,YAAY,EAAE,cAAc;IAC5B,KAAK,EAAE,OAAO;IACd,aAAa,EAAE,eAAe;IAC9B,SAAS,EAAE,WAAW;IACtB,iBAAiB,EAAE,mBAAmB;IACtC,qBAAqB,EAAE,uBAAuB;IAE9C,mBAAmB,EAAE,qBAAqB;IAC1C,eAAe,EAAE,iBAAiB;IAElC,mBAAmB;IACnB,0BAA0B,EAAE,4BAA4B;IAExD,kCAAkC;IAClC,kBAAkB,EAAE,oBAAoB;IACxC,qBAAqB,EAAE,uBAAuB;IAC9C,wBAAwB,EAAE,0BAA0B;IACpD,wBAAwB,EAAE,0BAA0B;IACpD,sBAAsB,EAAE,wBAAwB;IAChD,wBAAwB,EAAE,0BAA0B;CACrD,CAAC;AAEW,QAAA,YAAY,GAAG;IAC1B,UAAU,EAAE,eAAe;CAC5B,CAAC;AAEF,IAAY,sBAKX;AALD,WAAY,sBAAsB;IAChC,mDAAyB,CAAA;IACzB,uCAAa,CAAA;IACb,2CAAiB,CAAA;IACjB,6CAAmB,CAAA;AACrB,CAAC,EALW,sBAAsB,sCAAtB,sBAAsB,QAKjC;AAED,IAAY,gBAMX;AAND,WAAY,gBAAgB;IAC1B,yCAAqB,CAAA;IACrB,iCAAa,CAAA;IACb,mCAAe,CAAA;IACf,iCAAa,CAAA;IACb,uCAAmB,CAAA;AACrB,CAAC,EANW,gBAAgB,gCAAhB,gBAAgB,QAM3B;AAuBD,IAAY,gBAGX;AAHD,WAAY,gBAAgB;IAC1B,6BAAS,CAAA;IACT,mCAAe,CAAA;AACjB,CAAC,EAHW,gBAAgB,gCAAhB,gBAAgB,QAG3B;AA2DD,IAAY,UAMX;AAND,WAAY,UAAU;IACpB,mCAAqB,CAAA;IACrB,+BAAiB,CAAA;IACjB,+BAAiB,CAAA;IACjB,mCAAqB,CAAA;IACrB,mCAAqB,CAAA;AACvB,CAAC,EANW,UAAU,0BAAV,UAAU,QAMrB"}
@@ -0,0 +1,18 @@
1
+ import { type ReadableStream } from 'stream/web';
2
+ import type { AxSpan } from '../trace/index.js';
3
+ /**
4
+ * Util: API details
5
+ * @export
6
+ */
7
+ export type API = {
8
+ name?: string;
9
+ headers?: Record<string, string>;
10
+ put?: boolean;
11
+ };
12
+ export declare const apiCall: <TRequest = unknown, TResponse = unknown>(api: Readonly<API & {
13
+ url: string | URL;
14
+ stream?: boolean;
15
+ debug?: boolean;
16
+ fetch?: typeof fetch;
17
+ span?: AxSpan;
18
+ }>, json: TRequest) => Promise<TResponse | ReadableStream<TResponse>>;
@@ -0,0 +1,57 @@
1
+ "use strict";
2
+ var __importDefault = (this && this.__importDefault) || function (mod) {
3
+ return (mod && mod.__esModule) ? mod : { "default": mod };
4
+ };
5
+ Object.defineProperty(exports, "__esModule", { value: true });
6
+ exports.apiCall = void 0;
7
+ const path_1 = __importDefault(require("path"));
8
+ const web_1 = require("stream/web");
9
+ const stream_js_1 = require("./stream.js");
10
+ const transform_js_1 = require("./transform.js");
11
+ const textDecoderStream = web_1.TextDecoderStream ?? stream_js_1.TextDecoderStreamPolyfill;
12
+ // eslint-disable-next-line @typescript-eslint/naming-convention
13
+ const apiCall = async (api, json) => {
14
+ const baseUrl = new URL(process.env.PROXY ?? api.url);
15
+ const apiPath = path_1.default.join(baseUrl.pathname, api.name ?? '/', baseUrl.search);
16
+ const apiUrl = new URL(apiPath, baseUrl);
17
+ if (api.span?.isRecording()) {
18
+ api.span.setAttributes({
19
+ 'http.request.method': api.put ? 'PUT' : 'POST',
20
+ 'url.full': apiUrl.href
21
+ });
22
+ }
23
+ let res;
24
+ try {
25
+ res = await (api.fetch ?? fetch)(apiUrl, {
26
+ method: api.put ? 'PUT' : 'POST',
27
+ headers: {
28
+ 'Content-Type': 'application/json',
29
+ ...api.headers
30
+ },
31
+ body: JSON.stringify(json)
32
+ });
33
+ if (res.status >= 400) {
34
+ const respBody = JSON.stringify(await res.json(), null, 2);
35
+ throw new Error(`${res.status}, ${res.statusText}\n:Response Body: ${respBody}`);
36
+ }
37
+ if (!api.stream) {
38
+ return (await res.json());
39
+ }
40
+ if (!res.body) {
41
+ throw new Error('Response body is null');
42
+ }
43
+ const st = res.body
44
+ .pipeThrough(new textDecoderStream())
45
+ .pipeThrough(new transform_js_1.JSONStringifyStream());
46
+ return st;
47
+ }
48
+ catch (e) {
49
+ if (api.span?.isRecording()) {
50
+ api.span.recordAxSpanException(e);
51
+ }
52
+ const reqBody = JSON.stringify(json, null, 2);
53
+ throw new Error(`API Error: ${apiUrl.href}, ${e}\nRequest Body: ${reqBody}`);
54
+ }
55
+ };
56
+ exports.apiCall = apiCall;
57
+ //# sourceMappingURL=apicall.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"apicall.js","sourceRoot":"","sources":["../../../util/apicall.ts"],"names":[],"mappings":";;;;;;AAAA,gDAAwB;AACxB,oCAGoB;AAIpB,2CAAwD;AACxD,iDAAqD;AAarD,MAAM,iBAAiB,GAAG,uBAAuB,IAAI,qCAAyB,CAAC;AAE/E,gEAAgE;AACzD,MAAM,OAAO,GAAG,KAAK,EAC1B,GAQC,EACD,IAAc,EACkC,EAAE;IAClD,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;IACtD,MAAM,OAAO,GAAG,cAAI,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,IAAI,IAAI,GAAG,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;IAC7E,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAEzC,IAAI,GAAG,CAAC,IAAI,EAAE,WAAW,EAAE,EAAE,CAAC;QAC5B,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC;YACrB,qBAAqB,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM;YAC/C,UAAU,EAAE,MAAM,CAAC,IAAI;SACxB,CAAC,CAAC;IACL,CAAC;IAED,IAAI,GAAqD,CAAC;IAE1D,IAAI,CAAC;QACH,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,IAAI,KAAK,CAAC,CAAC,MAAM,EAAE;YACvC,MAAM,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM;YAChC,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,GAAG,GAAG,CAAC,OAAO;aACf;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CAAC,CAAC;QAEH,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC;YACtB,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YAC3D,MAAM,IAAI,KAAK,CACb,GAAG,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,UAAU,qBAAqB,QAAQ,EAAE,CAChE,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAc,CAAC;QACzC,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;QAED,MAAM,EAAE,GAAG,GAAG,CAAC,IAAI;aAChB,WAAW,CAAC,IAAI,iBAAiB,EAAE,CAAC;aACpC,WAAW,CAAC,IAAI,kCAAmB,EAAa,CAAC,CAAC;QAErD,OAAO,EAAE,CAAC;IACZ,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,IAAI,GAAG,CAAC,IAAI,EAAE,WAAW,EAAE,EAAE,CAAC;YAC5B,GAAG,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAU,CAAC,CAAC;QAC7C,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAC9C,MAAM,IAAI,KAAK,CACb,cAAc,MAAM,CAAC,IAAI,KAAK,CAAC,mBAAmB,OAAO,EAAE,CAC5D,CAAC;IACJ,CAAC;AACH,CAAC,CAAC;AAhEW,QAAA,OAAO,WAgElB"}
@@ -0,0 +1,14 @@
1
+ export declare class ColorLog {
2
+ private readonly ANSI_WHITE_BRIGHT;
3
+ private readonly ANSI_GREEN_BRIGHT;
4
+ private readonly ANSI_BLUE_BRIGHT;
5
+ private readonly ANSI_YELLOW;
6
+ private readonly ANSI_RED;
7
+ private readonly ANSI_RESET;
8
+ private colorize;
9
+ whiteBright(text: string): string;
10
+ greenBright(text: string): string;
11
+ blueBright(text: string): string;
12
+ yellow(text: string): string;
13
+ red(text: string): string;
14
+ }
@@ -0,0 +1,35 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.ColorLog = void 0;
4
+ // eslint-disable-next-line @typescript-eslint/naming-convention
5
+ class ColorLog {
6
+ // ANSI escape codes for different colors
7
+ ANSI_WHITE_BRIGHT = '\x1b[97m';
8
+ ANSI_GREEN_BRIGHT = '\x1b[92m';
9
+ ANSI_BLUE_BRIGHT = '\x1b[94m';
10
+ ANSI_YELLOW = '\x1b[93m';
11
+ ANSI_RED = '\x1b[91m';
12
+ ANSI_RESET = '\x1b[0m';
13
+ // Method to wrap text with the specified ANSI color code
14
+ colorize(text, colorCode) {
15
+ return `${colorCode}${text}${this.ANSI_RESET}`;
16
+ }
17
+ // Public methods to colorize text in various colors
18
+ whiteBright(text) {
19
+ return this.colorize(text, this.ANSI_WHITE_BRIGHT);
20
+ }
21
+ greenBright(text) {
22
+ return this.colorize(text, this.ANSI_GREEN_BRIGHT);
23
+ }
24
+ blueBright(text) {
25
+ return this.colorize(text, this.ANSI_BLUE_BRIGHT);
26
+ }
27
+ yellow(text) {
28
+ return this.colorize(text, this.ANSI_YELLOW);
29
+ }
30
+ red(text) {
31
+ return this.colorize(text, this.ANSI_RED);
32
+ }
33
+ }
34
+ exports.ColorLog = ColorLog;
35
+ //# sourceMappingURL=log.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"log.js","sourceRoot":"","sources":["../../../util/log.ts"],"names":[],"mappings":";;;AAAA,gEAAgE;AAChE,MAAa,QAAQ;IACnB,yCAAyC;IACxB,iBAAiB,GAAG,UAAU,CAAC;IAC/B,iBAAiB,GAAG,UAAU,CAAC;IAC/B,gBAAgB,GAAG,UAAU,CAAC;IAC9B,WAAW,GAAG,UAAU,CAAC;IACzB,QAAQ,GAAG,UAAU,CAAC;IACtB,UAAU,GAAG,SAAS,CAAC;IAExC,yDAAyD;IACjD,QAAQ,CAAC,IAAY,EAAE,SAAiB;QAC9C,OAAO,GAAG,SAAS,GAAG,IAAI,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;IACjD,CAAC;IAED,oDAAoD;IAC7C,WAAW,CAAC,IAAY;QAC7B,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACrD,CAAC;IAEM,WAAW,CAAC,IAAY;QAC7B,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACrD,CAAC;IAEM,UAAU,CAAC,IAAY;QAC5B,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;IACpD,CAAC;IAEM,MAAM,CAAC,IAAY;QACxB,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;IAC/C,CAAC;IAEM,GAAG,CAAC,IAAY;QACrB,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC5C,CAAC;CACF;AAlCD,4BAkCC"}
@@ -0,0 +1 @@
1
+ export declare function sleep(ms: number): Promise<unknown>;
@@ -0,0 +1,8 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.sleep = sleep;
4
+ // eslint-disable-next-line @typescript-eslint/naming-convention
5
+ async function sleep(ms) {
6
+ return new Promise((resolve) => setTimeout(resolve, ms));
7
+ }
8
+ //# sourceMappingURL=other.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"other.js","sourceRoot":"","sources":["../../../util/other.ts"],"names":[],"mappings":";;AACA,sBAEC;AAHD,gEAAgE;AACzD,KAAK,UAAU,KAAK,CAAC,EAAU;IACpC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3D,CAAC"}
@@ -0,0 +1,24 @@
1
+ export interface AxRateLimiterTokenUsageOptions {
2
+ debug?: boolean;
3
+ }
4
+ export declare class AxRateLimiterTokenUsage {
5
+ private options?;
6
+ private maxTokens;
7
+ private refillRate;
8
+ private currentTokens;
9
+ private lastRefillTime;
10
+ constructor(maxTokens: number, refillRate: number, options?: Readonly<AxRateLimiterTokenUsageOptions>);
11
+ private refillTokens;
12
+ private waitUntilTokensAvailable;
13
+ acquire(tokens: number): Promise<void>;
14
+ }
15
+ /**
16
+ * Example usage of the rate limiter. Limits to 5800 tokens per minute.
17
+ const rateLimiter = new AxRateLimiterTokenUsage(5800, 5800 / 60);
18
+
19
+ const axRateLimiterFunction = async (func, info) => {
20
+ const totalTokens = info.modelUsage?.totalTokens || 0;
21
+ await rateLimiter.acquire(totalTokens);
22
+ return func();
23
+ };
24
+ **/
@@ -0,0 +1,53 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.AxRateLimiterTokenUsage = void 0;
4
+ const log_js_1 = require("./log.js");
5
+ const colorLog = new log_js_1.ColorLog();
6
+ class AxRateLimiterTokenUsage {
7
+ options;
8
+ maxTokens;
9
+ refillRate;
10
+ currentTokens;
11
+ lastRefillTime;
12
+ constructor(maxTokens, refillRate, options) {
13
+ this.maxTokens = maxTokens;
14
+ this.refillRate = refillRate;
15
+ this.currentTokens = maxTokens;
16
+ this.lastRefillTime = Date.now();
17
+ this.options = options;
18
+ }
19
+ refillTokens() {
20
+ const now = Date.now();
21
+ const timeElapsed = (now - this.lastRefillTime) / 1000; // Convert ms to seconds
22
+ const tokensToAdd = timeElapsed * this.refillRate;
23
+ this.currentTokens = Math.min(this.maxTokens, this.currentTokens + tokensToAdd);
24
+ this.lastRefillTime = now;
25
+ }
26
+ async waitUntilTokensAvailable(tokens) {
27
+ this.refillTokens();
28
+ if (this.currentTokens >= tokens) {
29
+ this.currentTokens -= tokens;
30
+ return;
31
+ }
32
+ if (this.options?.debug) {
33
+ console.log(colorLog.red(`Rate limiter: Waiting for ${tokens - this.currentTokens} tokens`));
34
+ }
35
+ await new Promise((resolve) => setTimeout(resolve, 100)); // Wait for 100ms before checking again
36
+ return this.waitUntilTokensAvailable(tokens); // Recursive call
37
+ }
38
+ async acquire(tokens) {
39
+ await this.waitUntilTokensAvailable(tokens);
40
+ }
41
+ }
42
+ exports.AxRateLimiterTokenUsage = AxRateLimiterTokenUsage;
43
+ /**
44
+ * Example usage of the rate limiter. Limits to 5800 tokens per minute.
45
+ const rateLimiter = new AxRateLimiterTokenUsage(5800, 5800 / 60);
46
+
47
+ const axRateLimiterFunction = async (func, info) => {
48
+ const totalTokens = info.modelUsage?.totalTokens || 0;
49
+ await rateLimiter.acquire(totalTokens);
50
+ return func();
51
+ };
52
+ **/
53
+ //# sourceMappingURL=rate-limit.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"rate-limit.js","sourceRoot":"","sources":["../../../util/rate-limit.ts"],"names":[],"mappings":";;;AAAA,qCAAoC;AAEpC,MAAM,QAAQ,GAAG,IAAI,iBAAQ,EAAE,CAAC;AAMhC,MAAa,uBAAuB;IAC1B,OAAO,CAA4C;IACnD,SAAS,CAAS;IAClB,UAAU,CAAS;IACnB,aAAa,CAAS;IACtB,cAAc,CAAS;IAE/B,YACE,SAAiB,EACjB,UAAkB,EAClB,OAAkD;QAElD,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;QAC/B,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACjC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAEO,YAAY;QAClB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,WAAW,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,IAAI,CAAC,CAAC,wBAAwB;QAChF,MAAM,WAAW,GAAG,WAAW,GAAG,IAAI,CAAC,UAAU,CAAC;QAClD,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,GAAG,CAC3B,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,aAAa,GAAG,WAAW,CACjC,CAAC;QACF,IAAI,CAAC,cAAc,GAAG,GAAG,CAAC;IAC5B,CAAC;IAEO,KAAK,CAAC,wBAAwB,CAAC,MAAc;QACnD,IAAI,CAAC,YAAY,EAAE,CAAC;QACpB,IAAI,IAAI,CAAC,aAAa,IAAI,MAAM,EAAE,CAAC;YACjC,IAAI,CAAC,aAAa,IAAI,MAAM,CAAC;YAC7B,OAAO;QACT,CAAC;QACD,IAAI,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;YACxB,OAAO,CAAC,GAAG,CACT,QAAQ,CAAC,GAAG,CACV,6BAA6B,MAAM,GAAG,IAAI,CAAC,aAAa,SAAS,CAClE,CACF,CAAC;QACJ,CAAC;QACD,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,uCAAuC;QACjG,OAAO,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,CAAC,CAAC,iBAAiB;IACjE,CAAC;IAEM,KAAK,CAAC,OAAO,CAAC,MAAc;QACjC,MAAM,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,CAAC;IAC9C,CAAC;CACF;AAlDD,0DAkDC;AAED;;;;;;;;;GASG"}
@@ -0,0 +1,9 @@
1
+ import { TransformStream } from 'stream/web';
2
+ export interface TextDecoderCommon {
3
+ readonly encoding: string;
4
+ readonly fatal: boolean;
5
+ readonly ignoreBOM: boolean;
6
+ }
7
+ export declare class TextDecoderStreamPolyfill extends TransformStream<ArrayBuffer | Uint8Array, string> {
8
+ constructor();
9
+ }
@@ -0,0 +1,33 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.TextDecoderStreamPolyfill = void 0;
4
+ const web_1 = require("stream/web");
5
+ class TextDecodeTransformer {
6
+ decoder;
7
+ constructor() {
8
+ this.decoder = new TextDecoder();
9
+ }
10
+ transform(chunk, controller) {
11
+ if (!(chunk instanceof ArrayBuffer || ArrayBuffer.isView(chunk))) {
12
+ throw new TypeError('Input data must be a BufferSource');
13
+ }
14
+ const text = this.decoder.decode(chunk, { stream: true });
15
+ if (text.length !== 0) {
16
+ controller.enqueue(text);
17
+ }
18
+ }
19
+ flush(controller) {
20
+ const text = this.decoder.decode();
21
+ if (text.length !== 0) {
22
+ controller.enqueue(text);
23
+ }
24
+ }
25
+ }
26
+ // eslint-disable-next-line @typescript-eslint/naming-convention
27
+ class TextDecoderStreamPolyfill extends web_1.TransformStream {
28
+ constructor() {
29
+ super(new TextDecodeTransformer());
30
+ }
31
+ }
32
+ exports.TextDecoderStreamPolyfill = TextDecoderStreamPolyfill;
33
+ //# sourceMappingURL=stream.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"stream.js","sourceRoot":"","sources":["../../../util/stream.ts"],"names":[],"mappings":";;;AAAA,oCAIoB;AASpB,MAAM,qBAAqB;IAGjB,OAAO,CAAC;IAEhB;QACE,IAAI,CAAC,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;IACnC,CAAC;IAED,SAAS,CACP,KAA+B,EAC/B,UAAoD;QAEpD,IAAI,CAAC,CAAC,KAAK,YAAY,WAAW,IAAI,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;YACjE,MAAM,IAAI,SAAS,CAAC,mCAAmC,CAAC,CAAC;QAC3D,CAAC;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1D,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtB,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAoD;QACxD,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;QACnC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtB,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;CACF;AAED,gEAAgE;AAChE,MAAa,yBAA0B,SAAQ,qBAG9C;IACC;QACE,KAAK,CAAC,IAAI,qBAAqB,EAAE,CAAC,CAAC;IACrC,CAAC;CACF;AAPD,8DAOC"}
@@ -0,0 +1,7 @@
1
+ import { TransformStream } from 'stream/web';
2
+ export declare class JSONStringifyStream<O> extends TransformStream<string, O> {
3
+ constructor();
4
+ }
5
+ export declare class RespTransformStream<I, O> extends TransformStream<I, O> {
6
+ constructor(transformFn: (arg0: I) => O, doneCallback?: (args0: readonly O[]) => Promise<void>);
7
+ }
@@ -0,0 +1,64 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.RespTransformStream = exports.JSONStringifyStream = void 0;
4
+ const web_1 = require("stream/web");
5
+ class JSONTransformer {
6
+ async transform(obj, controller) {
7
+ (obj.split('\n') ?? [])
8
+ .map(extractJson)
9
+ .map((v) => {
10
+ try {
11
+ return v && v.length > 0 ? JSON.parse(v) : null;
12
+ }
13
+ catch (e) {
14
+ return null;
15
+ }
16
+ })
17
+ .filter((v) => v)
18
+ .forEach((v) => v && controller.enqueue(v));
19
+ }
20
+ }
21
+ // eslint-disable-next-line @typescript-eslint/naming-convention
22
+ class JSONStringifyStream extends web_1.TransformStream {
23
+ constructor() {
24
+ super(new JSONTransformer());
25
+ }
26
+ }
27
+ exports.JSONStringifyStream = JSONStringifyStream;
28
+ class TypeTransformer {
29
+ buffer;
30
+ doneCallback;
31
+ transformFn;
32
+ constructor(transformFn, doneCallback) {
33
+ this.transformFn = transformFn;
34
+ this.doneCallback = doneCallback;
35
+ this.buffer = doneCallback ? [] : undefined;
36
+ }
37
+ async transform(obj, controller) {
38
+ const val = this.transformFn(obj);
39
+ if (val) {
40
+ controller.enqueue(val);
41
+ this.buffer?.push(val);
42
+ }
43
+ }
44
+ async flush(controller) {
45
+ await this.doneCallback?.(this.buffer ?? []);
46
+ controller.terminate();
47
+ }
48
+ }
49
+ // eslint-disable-next-line @typescript-eslint/naming-convention
50
+ class RespTransformStream extends web_1.TransformStream {
51
+ constructor(transformFn, doneCallback) {
52
+ super(new TypeTransformer(transformFn, doneCallback));
53
+ }
54
+ }
55
+ exports.RespTransformStream = RespTransformStream;
56
+ const extractJson = (str) => {
57
+ const startIndex = str.indexOf('{');
58
+ const endIndex = str.lastIndexOf('}');
59
+ if (startIndex !== -1 && endIndex !== -1) {
60
+ return str.substring(startIndex, endIndex + 1);
61
+ }
62
+ return null;
63
+ };
64
+ //# sourceMappingURL=transform.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"transform.js","sourceRoot":"","sources":["../../../util/transform.ts"],"names":[],"mappings":";;;AAAA,oCAIoB;AAEpB,MAAM,eAAe;IACnB,KAAK,CAAC,SAAS,CACb,GAAW,EACX,UAA+C;QAE/C,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;aACpB,GAAG,CAAC,WAAW,CAAC;aAChB,GAAG,CAAI,CAAC,CAAC,EAAE,EAAE;YACZ,IAAI,CAAC;gBACH,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC5D,CAAC;YAAC,OAAO,CAAU,EAAE,CAAC;gBACpB,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC,CAAC;aACD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;aAChB,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC;CACF;AAED,gEAAgE;AAChE,MAAa,mBAAuB,SAAQ,qBAA0B;IACpE;QACE,KAAK,CAAC,IAAI,eAAe,EAAK,CAAC,CAAC;IAClC,CAAC;CACF;AAJD,kDAIC;AAED,MAAM,eAAe;IACX,MAAM,CAAO;IACb,YAAY,CAA0C;IACtD,WAAW,CAAiB;IAEpC,YACE,WAA2B,EAC3B,YAAqD;QAErD,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,GAAM,EAAE,UAA+C;QACrE,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QAClC,IAAI,GAAG,EAAE,CAAC;YACR,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACxB,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,UAA+C;QACzD,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;QAC7C,UAAU,CAAC,SAAS,EAAE,CAAC;IACzB,CAAC;CACF;AAED,gEAAgE;AAChE,MAAa,mBAA0B,SAAQ,qBAAqB;IAClE,YACE,WAA2B,EAC3B,YAAqD;QAErD,KAAK,CAAC,IAAI,eAAe,CAAO,WAAW,EAAE,YAAY,CAAC,CAAC,CAAC;IAC9D,CAAC;CACF;AAPD,kDAOC;AAED,MAAM,WAAW,GAAG,CAAC,GAAW,EAAiB,EAAE;IACjD,MAAM,UAAU,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACpC,MAAM,QAAQ,GAAG,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IAEtC,IAAI,UAAU,KAAK,CAAC,CAAC,IAAI,QAAQ,KAAK,CAAC,CAAC,EAAE,CAAC;QACzC,OAAO,GAAG,CAAC,SAAS,CAAC,UAAU,EAAE,QAAQ,GAAG,CAAC,CAAC,CAAC;IACjD,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC,CAAC"}
package/db/base.d.ts ADDED
@@ -0,0 +1,23 @@
1
+ import { type AxSpan, type AxTracer } from '../trace/index.js';
2
+ import type { AxDBQueryRequest, AxDBQueryResponse, AxDBService, AxDBUpsertRequest, AxDBUpsertResponse } from './types.js';
3
+ export interface AxDBBaseArgs {
4
+ fetch?: typeof fetch;
5
+ tracer?: AxTracer;
6
+ }
7
+ export interface AxDBBaseOpOptions {
8
+ span?: AxSpan;
9
+ }
10
+ export declare class AxDBBase implements AxDBService {
11
+ protected name: string;
12
+ protected fetch?: typeof fetch;
13
+ private tracer?;
14
+ _upsert?: (req: Readonly<AxDBUpsertRequest>, update?: boolean, options?: Readonly<AxDBBaseOpOptions>) => Promise<AxDBUpsertResponse>;
15
+ _batchUpsert?: (batchReq: Readonly<AxDBUpsertRequest[]>, update?: boolean, options?: Readonly<AxDBBaseOpOptions>) => Promise<AxDBUpsertResponse>;
16
+ _query?: (req: Readonly<AxDBQueryRequest>, options?: Readonly<AxDBBaseOpOptions>) => Promise<AxDBQueryResponse>;
17
+ constructor({ name, fetch, tracer }: Readonly<AxDBBaseArgs & {
18
+ name: string;
19
+ }>);
20
+ upsert(req: Readonly<AxDBUpsertRequest>, update?: boolean): Promise<AxDBUpsertResponse>;
21
+ batchUpsert(req: Readonly<AxDBUpsertRequest[]>, update?: boolean): Promise<AxDBUpsertResponse>;
22
+ query(req: Readonly<AxDBQueryRequest>): Promise<AxDBQueryResponse>;
23
+ }
package/db/base.js ADDED
@@ -0,0 +1,87 @@
1
+ import { axSpanAttributes, AxSpanKind } from '../trace/index.js';
2
+ export class AxDBBase {
3
+ name;
4
+ fetch;
5
+ tracer;
6
+ _upsert;
7
+ _batchUpsert;
8
+ _query;
9
+ constructor({ name, fetch, tracer }) {
10
+ this.name = name;
11
+ this.fetch = fetch;
12
+ this.tracer = tracer;
13
+ }
14
+ async upsert(req, update) {
15
+ if (!this._upsert) {
16
+ throw new Error('upsert() not implemented');
17
+ }
18
+ if (!this.tracer) {
19
+ return await this._upsert(req, update);
20
+ }
21
+ return await this.tracer?.startActiveSpan('DB Upsert Request', {
22
+ kind: AxSpanKind.SERVER,
23
+ attributes: {
24
+ [axSpanAttributes.DB_SYSTEM]: this.name,
25
+ [axSpanAttributes.DB_OPERATION_NAME]: 'upsert',
26
+ [axSpanAttributes.DB_TABLE]: req.table,
27
+ [axSpanAttributes.DB_NAMESPACE]: req.namespace,
28
+ [axSpanAttributes.DB_OPERATION_NAME]: update ? 'update' : 'insert'
29
+ }
30
+ }, async (span) => {
31
+ const res = await this._upsert(req, update, { span });
32
+ span.end();
33
+ return res;
34
+ });
35
+ }
36
+ async batchUpsert(req, update) {
37
+ if (!this._batchUpsert) {
38
+ throw new Error('batchUpsert() not implemented');
39
+ }
40
+ if (req.length == 0) {
41
+ throw new Error('Batch request is empty');
42
+ }
43
+ if (!req[0]) {
44
+ throw new Error('Batch request is invalid first element is undefined');
45
+ }
46
+ if (!this.tracer) {
47
+ return await this._batchUpsert(req, update);
48
+ }
49
+ return await this.tracer?.startActiveSpan('DB Batch Upsert Request', {
50
+ kind: AxSpanKind.SERVER,
51
+ attributes: {
52
+ [axSpanAttributes.DB_SYSTEM]: this.name,
53
+ [axSpanAttributes.DB_OPERATION_NAME]: 'upsert',
54
+ [axSpanAttributes.DB_TABLE]: req[0].table,
55
+ [axSpanAttributes.DB_NAMESPACE]: req[0].namespace,
56
+ [axSpanAttributes.DB_OPERATION_NAME]: update ? 'update' : 'insert'
57
+ }
58
+ }, async (span) => {
59
+ const res = await this._batchUpsert(req, update, { span });
60
+ span.end();
61
+ return res;
62
+ });
63
+ }
64
+ async query(req) {
65
+ if (!this._query) {
66
+ throw new Error('query() not implemented');
67
+ }
68
+ if (!this.tracer) {
69
+ return await this._query(req);
70
+ }
71
+ return await this.tracer?.startActiveSpan('DB Query Request', {
72
+ kind: AxSpanKind.SERVER,
73
+ attributes: {
74
+ [axSpanAttributes.DB_SYSTEM]: this.name,
75
+ [axSpanAttributes.DB_OPERATION_NAME]: 'upsert',
76
+ [axSpanAttributes.DB_TABLE]: req.table,
77
+ [axSpanAttributes.DB_NAMESPACE]: req.namespace,
78
+ [axSpanAttributes.DB_OPERATION_NAME]: 'query'
79
+ }
80
+ }, async (span) => {
81
+ const res = await this._query(req, { span });
82
+ span.end();
83
+ return res;
84
+ });
85
+ }
86
+ }
87
+ //# sourceMappingURL=base.js.map
package/db/base.js.map ADDED
@@ -0,0 +1 @@
1
+ {"version":3,"file":"base.js","sourceRoot":"","sources":["../../db/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,gBAAgB,EAChB,UAAU,EAEX,MAAM,mBAAmB,CAAC;AAmB3B,MAAM,OAAO,QAAQ;IACT,IAAI,CAAS;IACb,KAAK,CAAgB;IACvB,MAAM,CAAY;IAE1B,OAAO,CAI0B;IAEjC,YAAY,CAIqB;IAEjC,MAAM,CAG0B;IAEhC,YAAY,EACV,IAAI,EACJ,KAAK,EACL,MAAM,EACoC;QAC1C,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,MAAM,CACV,GAAgC,EAChC,MAAgB;QAEhB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAC9C,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACzC,CAAC;QAED,OAAO,MAAM,IAAI,CAAC,MAAM,EAAE,eAAe,CACvC,mBAAmB,EACnB;YACE,IAAI,EAAE,UAAU,CAAC,MAAM;YACvB,UAAU,EAAE;gBACV,CAAC,gBAAgB,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI;gBACvC,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,QAAQ;gBAC9C,CAAC,gBAAgB,CAAC,QAAQ,CAAC,EAAE,GAAG,CAAC,KAAK;gBACtC,CAAC,gBAAgB,CAAC,YAAY,CAAC,EAAE,GAAG,CAAC,SAAS;gBAC9C,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ;aACnE;SACF,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAQ,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;YACvD,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,GAAG,CAAC;QACb,CAAC,CACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW,CACf,GAAkC,EAClC,MAAgB;QAEhB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;QACnD,CAAC;QACD,IAAI,GAAG,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAC5C,CAAC;QACD,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;QACzE,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,OAAO,MAAM,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAC9C,CAAC;QAED,OAAO,MAAM,IAAI,CAAC,MAAM,EAAE,eAAe,CACvC,yBAAyB,EACzB;YACE,IAAI,EAAE,UAAU,CAAC,MAAM;YACvB,UAAU,EAAE;gBACV,CAAC,gBAAgB,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI;gBACvC,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,QAAQ;gBAC9C,CAAC,gBAAgB,CAAC,QAAQ,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK;gBACzC,CAAC,gBAAgB,CAAC,YAAY,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS;gBACjD,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ;aACnE;SACF,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,YAAa,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;YAC5D,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,GAAG,CAAC;QACb,CAAC,CACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,GAA+B;QACzC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QAC7C,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAChC,CAAC;QAED,OAAO,MAAM,IAAI,CAAC,MAAM,EAAE,eAAe,CACvC,kBAAkB,EAClB;YACE,IAAI,EAAE,UAAU,CAAC,MAAM;YACvB,UAAU,EAAE;gBACV,CAAC,gBAAgB,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI;gBACvC,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,QAAQ;gBAC9C,CAAC,gBAAgB,CAAC,QAAQ,CAAC,EAAE,GAAG,CAAC,KAAK;gBACtC,CAAC,gBAAgB,CAAC,YAAY,CAAC,EAAE,GAAG,CAAC,SAAS;gBAC9C,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,OAAO;aAC9C;SACF,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAO,CAAC,GAAG,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;YAC9C,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,GAAG,CAAC;QACb,CAAC,CACF,CAAC;IACJ,CAAC;CACF"}
@@ -0,0 +1,21 @@
1
+ import { AxDBBase, type AxDBBaseArgs, type AxDBBaseOpOptions } from './base.js';
2
+ import type { AxDBQueryRequest, AxDBQueryResponse, AxDBUpsertRequest, AxDBUpsertResponse } from './types.js';
3
+ export type AxDBCloudflareOpOptions = AxDBBaseOpOptions;
4
+ export interface AxDBCloudflareArgs extends AxDBBaseArgs {
5
+ name: 'cloudflare';
6
+ apiKey: string;
7
+ accountId: string;
8
+ fetch?: typeof fetch;
9
+ }
10
+ /**
11
+ * Cloudflare: DB Service
12
+ * @export
13
+ */
14
+ export declare class AxDBCloudflare extends AxDBBase {
15
+ private apiKey;
16
+ private accountId;
17
+ constructor({ apiKey, accountId, fetch, tracer }: Readonly<Omit<AxDBCloudflareArgs, 'name'>>);
18
+ _upsert: (req: Readonly<AxDBUpsertRequest>, update?: boolean, options?: Readonly<AxDBCloudflareOpOptions>) => Promise<AxDBUpsertResponse>;
19
+ batchUpsert: (batchReq: Readonly<AxDBUpsertRequest[]>, update?: boolean, options?: Readonly<AxDBCloudflareOpOptions>) => Promise<AxDBUpsertResponse>;
20
+ query: (req: Readonly<AxDBQueryRequest>, options?: Readonly<AxDBCloudflareOpOptions>) => Promise<AxDBQueryResponse>;
21
+ }