modelfusion 0.0.44

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 (405) hide show
  1. package/LICENSE +21 -0
  2. package/README.md +429 -0
  3. package/composed-function/index.cjs +22 -0
  4. package/composed-function/index.d.ts +6 -0
  5. package/composed-function/index.js +6 -0
  6. package/composed-function/summarize/SummarizationFunction.cjs +2 -0
  7. package/composed-function/summarize/SummarizationFunction.d.ts +4 -0
  8. package/composed-function/summarize/SummarizationFunction.js +1 -0
  9. package/composed-function/summarize/summarizeRecursively.cjs +19 -0
  10. package/composed-function/summarize/summarizeRecursively.d.ts +11 -0
  11. package/composed-function/summarize/summarizeRecursively.js +15 -0
  12. package/composed-function/summarize/summarizeRecursivelyWithTextGenerationAndTokenSplitting.cjs +29 -0
  13. package/composed-function/summarize/summarizeRecursivelyWithTextGenerationAndTokenSplitting.d.ts +24 -0
  14. package/composed-function/summarize/summarizeRecursivelyWithTextGenerationAndTokenSplitting.js +25 -0
  15. package/composed-function/use-tool/NoSuchToolError.cjs +17 -0
  16. package/composed-function/use-tool/NoSuchToolError.d.ts +4 -0
  17. package/composed-function/use-tool/NoSuchToolError.js +13 -0
  18. package/composed-function/use-tool/Tool.cjs +43 -0
  19. package/composed-function/use-tool/Tool.d.ts +15 -0
  20. package/composed-function/use-tool/Tool.js +39 -0
  21. package/composed-function/use-tool/useTool.cjs +59 -0
  22. package/composed-function/use-tool/useTool.d.ts +36 -0
  23. package/composed-function/use-tool/useTool.js +54 -0
  24. package/cost/Cost.cjs +38 -0
  25. package/cost/Cost.d.ts +16 -0
  26. package/cost/Cost.js +34 -0
  27. package/cost/CostCalculator.cjs +2 -0
  28. package/cost/CostCalculator.d.ts +8 -0
  29. package/cost/CostCalculator.js +1 -0
  30. package/cost/calculateCost.cjs +28 -0
  31. package/cost/calculateCost.d.ts +7 -0
  32. package/cost/calculateCost.js +24 -0
  33. package/cost/index.cjs +19 -0
  34. package/cost/index.d.ts +3 -0
  35. package/cost/index.js +3 -0
  36. package/index.cjs +25 -0
  37. package/index.d.ts +9 -0
  38. package/index.js +9 -0
  39. package/model-function/AbstractModel.cjs +22 -0
  40. package/model-function/AbstractModel.d.ts +12 -0
  41. package/model-function/AbstractModel.js +18 -0
  42. package/model-function/FunctionOptions.cjs +2 -0
  43. package/model-function/FunctionOptions.d.ts +6 -0
  44. package/model-function/FunctionOptions.js +1 -0
  45. package/model-function/Model.cjs +2 -0
  46. package/model-function/Model.d.ts +23 -0
  47. package/model-function/Model.js +1 -0
  48. package/model-function/ModelCallEvent.cjs +2 -0
  49. package/model-function/ModelCallEvent.d.ts +18 -0
  50. package/model-function/ModelCallEvent.js +1 -0
  51. package/model-function/ModelCallEventSource.cjs +42 -0
  52. package/model-function/ModelCallEventSource.d.ts +13 -0
  53. package/model-function/ModelCallEventSource.js +38 -0
  54. package/model-function/ModelCallObserver.cjs +2 -0
  55. package/model-function/ModelCallObserver.d.ts +5 -0
  56. package/model-function/ModelCallObserver.js +1 -0
  57. package/model-function/ModelInformation.cjs +2 -0
  58. package/model-function/ModelInformation.d.ts +4 -0
  59. package/model-function/ModelInformation.js +1 -0
  60. package/model-function/SuccessfulModelCall.cjs +22 -0
  61. package/model-function/SuccessfulModelCall.d.ts +9 -0
  62. package/model-function/SuccessfulModelCall.js +18 -0
  63. package/model-function/embed-text/TextEmbeddingEvent.cjs +2 -0
  64. package/model-function/embed-text/TextEmbeddingEvent.d.ts +23 -0
  65. package/model-function/embed-text/TextEmbeddingEvent.js +1 -0
  66. package/model-function/embed-text/TextEmbeddingModel.cjs +2 -0
  67. package/model-function/embed-text/TextEmbeddingModel.d.ts +18 -0
  68. package/model-function/embed-text/TextEmbeddingModel.js +1 -0
  69. package/model-function/embed-text/embedText.cjs +90 -0
  70. package/model-function/embed-text/embedText.d.ts +33 -0
  71. package/model-function/embed-text/embedText.js +85 -0
  72. package/model-function/executeCall.cjs +60 -0
  73. package/model-function/executeCall.d.ts +27 -0
  74. package/model-function/executeCall.js +56 -0
  75. package/model-function/generate-image/ImageGenerationEvent.cjs +2 -0
  76. package/model-function/generate-image/ImageGenerationEvent.d.ts +22 -0
  77. package/model-function/generate-image/ImageGenerationEvent.js +1 -0
  78. package/model-function/generate-image/ImageGenerationModel.cjs +2 -0
  79. package/model-function/generate-image/ImageGenerationModel.d.ts +8 -0
  80. package/model-function/generate-image/ImageGenerationModel.js +1 -0
  81. package/model-function/generate-image/generateImage.cjs +63 -0
  82. package/model-function/generate-image/generateImage.d.ts +23 -0
  83. package/model-function/generate-image/generateImage.js +59 -0
  84. package/model-function/generate-json/GenerateJsonModel.cjs +2 -0
  85. package/model-function/generate-json/GenerateJsonModel.d.ts +10 -0
  86. package/model-function/generate-json/GenerateJsonModel.js +1 -0
  87. package/model-function/generate-json/GenerateJsonOrTextModel.cjs +2 -0
  88. package/model-function/generate-json/GenerateJsonOrTextModel.d.ts +18 -0
  89. package/model-function/generate-json/GenerateJsonOrTextModel.js +1 -0
  90. package/model-function/generate-json/JsonGenerationEvent.cjs +2 -0
  91. package/model-function/generate-json/JsonGenerationEvent.d.ts +22 -0
  92. package/model-function/generate-json/JsonGenerationEvent.js +1 -0
  93. package/model-function/generate-json/NoSuchSchemaError.cjs +17 -0
  94. package/model-function/generate-json/NoSuchSchemaError.d.ts +4 -0
  95. package/model-function/generate-json/NoSuchSchemaError.js +13 -0
  96. package/model-function/generate-json/SchemaDefinition.cjs +2 -0
  97. package/model-function/generate-json/SchemaDefinition.d.ts +6 -0
  98. package/model-function/generate-json/SchemaDefinition.js +1 -0
  99. package/model-function/generate-json/SchemaValidationError.cjs +36 -0
  100. package/model-function/generate-json/SchemaValidationError.d.ts +11 -0
  101. package/model-function/generate-json/SchemaValidationError.js +32 -0
  102. package/model-function/generate-json/generateJson.cjs +61 -0
  103. package/model-function/generate-json/generateJson.d.ts +9 -0
  104. package/model-function/generate-json/generateJson.js +57 -0
  105. package/model-function/generate-json/generateJsonOrText.cjs +74 -0
  106. package/model-function/generate-json/generateJsonOrText.d.ts +25 -0
  107. package/model-function/generate-json/generateJsonOrText.js +70 -0
  108. package/model-function/generate-text/AsyncQueue.cjs +66 -0
  109. package/model-function/generate-text/AsyncQueue.d.ts +17 -0
  110. package/model-function/generate-text/AsyncQueue.js +62 -0
  111. package/model-function/generate-text/DeltaEvent.cjs +2 -0
  112. package/model-function/generate-text/DeltaEvent.d.ts +7 -0
  113. package/model-function/generate-text/DeltaEvent.js +1 -0
  114. package/model-function/generate-text/TextDeltaEventSource.cjs +54 -0
  115. package/model-function/generate-text/TextDeltaEventSource.d.ts +5 -0
  116. package/model-function/generate-text/TextDeltaEventSource.js +46 -0
  117. package/model-function/generate-text/TextGenerationEvent.cjs +2 -0
  118. package/model-function/generate-text/TextGenerationEvent.d.ts +22 -0
  119. package/model-function/generate-text/TextGenerationEvent.js +1 -0
  120. package/model-function/generate-text/TextGenerationModel.cjs +2 -0
  121. package/model-function/generate-text/TextGenerationModel.d.ts +42 -0
  122. package/model-function/generate-text/TextGenerationModel.js +1 -0
  123. package/model-function/generate-text/TextStreamingEvent.cjs +2 -0
  124. package/model-function/generate-text/TextStreamingEvent.d.ts +22 -0
  125. package/model-function/generate-text/TextStreamingEvent.js +1 -0
  126. package/model-function/generate-text/extractTextDeltas.cjs +23 -0
  127. package/model-function/generate-text/extractTextDeltas.d.ts +7 -0
  128. package/model-function/generate-text/extractTextDeltas.js +19 -0
  129. package/model-function/generate-text/generateText.cjs +67 -0
  130. package/model-function/generate-text/generateText.d.ts +20 -0
  131. package/model-function/generate-text/generateText.js +63 -0
  132. package/model-function/generate-text/parseEventSourceReadableStream.cjs +30 -0
  133. package/model-function/generate-text/parseEventSourceReadableStream.d.ts +8 -0
  134. package/model-function/generate-text/parseEventSourceReadableStream.js +26 -0
  135. package/model-function/generate-text/streamText.cjs +115 -0
  136. package/model-function/generate-text/streamText.d.ts +11 -0
  137. package/model-function/generate-text/streamText.js +111 -0
  138. package/model-function/index.cjs +47 -0
  139. package/model-function/index.d.ts +31 -0
  140. package/model-function/index.js +31 -0
  141. package/model-function/tokenize-text/Tokenizer.cjs +2 -0
  142. package/model-function/tokenize-text/Tokenizer.d.ts +19 -0
  143. package/model-function/tokenize-text/Tokenizer.js +1 -0
  144. package/model-function/tokenize-text/countTokens.cjs +10 -0
  145. package/model-function/tokenize-text/countTokens.d.ts +5 -0
  146. package/model-function/tokenize-text/countTokens.js +6 -0
  147. package/model-function/transcribe-audio/TranscriptionEvent.cjs +2 -0
  148. package/model-function/transcribe-audio/TranscriptionEvent.d.ts +22 -0
  149. package/model-function/transcribe-audio/TranscriptionEvent.js +1 -0
  150. package/model-function/transcribe-audio/TranscriptionModel.cjs +2 -0
  151. package/model-function/transcribe-audio/TranscriptionModel.d.ts +8 -0
  152. package/model-function/transcribe-audio/TranscriptionModel.js +1 -0
  153. package/model-function/transcribe-audio/transcribe.cjs +62 -0
  154. package/model-function/transcribe-audio/transcribe.d.ts +22 -0
  155. package/model-function/transcribe-audio/transcribe.js +58 -0
  156. package/model-provider/automatic1111/Automatic1111Error.cjs +39 -0
  157. package/model-provider/automatic1111/Automatic1111Error.d.ts +31 -0
  158. package/model-provider/automatic1111/Automatic1111Error.js +31 -0
  159. package/model-provider/automatic1111/Automatic1111ImageGenerationModel.cjs +76 -0
  160. package/model-provider/automatic1111/Automatic1111ImageGenerationModel.d.ts +54 -0
  161. package/model-provider/automatic1111/Automatic1111ImageGenerationModel.js +72 -0
  162. package/model-provider/automatic1111/index.cjs +20 -0
  163. package/model-provider/automatic1111/index.d.ts +2 -0
  164. package/model-provider/automatic1111/index.js +2 -0
  165. package/model-provider/cohere/CohereError.cjs +36 -0
  166. package/model-provider/cohere/CohereError.d.ts +22 -0
  167. package/model-provider/cohere/CohereError.js +28 -0
  168. package/model-provider/cohere/CohereTextEmbeddingModel.cjs +172 -0
  169. package/model-provider/cohere/CohereTextEmbeddingModel.d.ts +119 -0
  170. package/model-provider/cohere/CohereTextEmbeddingModel.js +165 -0
  171. package/model-provider/cohere/CohereTextGenerationModel.cjs +283 -0
  172. package/model-provider/cohere/CohereTextGenerationModel.d.ts +203 -0
  173. package/model-provider/cohere/CohereTextGenerationModel.js +276 -0
  174. package/model-provider/cohere/CohereTokenizer.cjs +136 -0
  175. package/model-provider/cohere/CohereTokenizer.d.ts +118 -0
  176. package/model-provider/cohere/CohereTokenizer.js +129 -0
  177. package/model-provider/cohere/index.cjs +22 -0
  178. package/model-provider/cohere/index.d.ts +4 -0
  179. package/model-provider/cohere/index.js +4 -0
  180. package/model-provider/huggingface/HuggingFaceError.cjs +52 -0
  181. package/model-provider/huggingface/HuggingFaceError.d.ts +22 -0
  182. package/model-provider/huggingface/HuggingFaceError.js +44 -0
  183. package/model-provider/huggingface/HuggingFaceTextGenerationModel.cjs +174 -0
  184. package/model-provider/huggingface/HuggingFaceTextGenerationModel.d.ts +75 -0
  185. package/model-provider/huggingface/HuggingFaceTextGenerationModel.js +167 -0
  186. package/model-provider/huggingface/index.cjs +20 -0
  187. package/model-provider/huggingface/index.d.ts +2 -0
  188. package/model-provider/huggingface/index.js +2 -0
  189. package/model-provider/index.cjs +22 -0
  190. package/model-provider/index.d.ts +6 -0
  191. package/model-provider/index.js +6 -0
  192. package/model-provider/llamacpp/LlamaCppError.cjs +52 -0
  193. package/model-provider/llamacpp/LlamaCppError.d.ts +22 -0
  194. package/model-provider/llamacpp/LlamaCppError.js +44 -0
  195. package/model-provider/llamacpp/LlamaCppTextEmbeddingModel.cjs +96 -0
  196. package/model-provider/llamacpp/LlamaCppTextEmbeddingModel.d.ts +40 -0
  197. package/model-provider/llamacpp/LlamaCppTextEmbeddingModel.js +89 -0
  198. package/model-provider/llamacpp/LlamaCppTextGenerationModel.cjs +245 -0
  199. package/model-provider/llamacpp/LlamaCppTextGenerationModel.d.ts +399 -0
  200. package/model-provider/llamacpp/LlamaCppTextGenerationModel.js +238 -0
  201. package/model-provider/llamacpp/LlamaCppTokenizer.cjs +64 -0
  202. package/model-provider/llamacpp/LlamaCppTokenizer.d.ts +38 -0
  203. package/model-provider/llamacpp/LlamaCppTokenizer.js +57 -0
  204. package/model-provider/llamacpp/index.cjs +22 -0
  205. package/model-provider/llamacpp/index.d.ts +4 -0
  206. package/model-provider/llamacpp/index.js +4 -0
  207. package/model-provider/openai/OpenAICostCalculator.cjs +71 -0
  208. package/model-provider/openai/OpenAICostCalculator.d.ts +6 -0
  209. package/model-provider/openai/OpenAICostCalculator.js +67 -0
  210. package/model-provider/openai/OpenAIError.cjs +50 -0
  211. package/model-provider/openai/OpenAIError.d.ts +47 -0
  212. package/model-provider/openai/OpenAIError.js +42 -0
  213. package/model-provider/openai/OpenAIImageGenerationModel.cjs +124 -0
  214. package/model-provider/openai/OpenAIImageGenerationModel.d.ts +113 -0
  215. package/model-provider/openai/OpenAIImageGenerationModel.js +119 -0
  216. package/model-provider/openai/OpenAIModelSettings.cjs +2 -0
  217. package/model-provider/openai/OpenAIModelSettings.d.ts +8 -0
  218. package/model-provider/openai/OpenAIModelSettings.js +1 -0
  219. package/model-provider/openai/OpenAITextEmbeddingModel.cjs +171 -0
  220. package/model-provider/openai/OpenAITextEmbeddingModel.d.ts +122 -0
  221. package/model-provider/openai/OpenAITextEmbeddingModel.js +162 -0
  222. package/model-provider/openai/OpenAITextGenerationModel.cjs +326 -0
  223. package/model-provider/openai/OpenAITextGenerationModel.d.ts +254 -0
  224. package/model-provider/openai/OpenAITextGenerationModel.js +317 -0
  225. package/model-provider/openai/OpenAITranscriptionModel.cjs +195 -0
  226. package/model-provider/openai/OpenAITranscriptionModel.d.ts +196 -0
  227. package/model-provider/openai/OpenAITranscriptionModel.js +187 -0
  228. package/model-provider/openai/TikTokenTokenizer.cjs +86 -0
  229. package/model-provider/openai/TikTokenTokenizer.d.ts +35 -0
  230. package/model-provider/openai/TikTokenTokenizer.js +82 -0
  231. package/model-provider/openai/chat/OpenAIChatMessage.cjs +24 -0
  232. package/model-provider/openai/chat/OpenAIChatMessage.d.ts +26 -0
  233. package/model-provider/openai/chat/OpenAIChatMessage.js +21 -0
  234. package/model-provider/openai/chat/OpenAIChatModel.cjs +288 -0
  235. package/model-provider/openai/chat/OpenAIChatModel.d.ts +344 -0
  236. package/model-provider/openai/chat/OpenAIChatModel.js +279 -0
  237. package/model-provider/openai/chat/OpenAIChatPrompt.cjs +143 -0
  238. package/model-provider/openai/chat/OpenAIChatPrompt.d.ts +108 -0
  239. package/model-provider/openai/chat/OpenAIChatPrompt.js +135 -0
  240. package/model-provider/openai/chat/OpenAIChatStreamIterable.cjs +112 -0
  241. package/model-provider/openai/chat/OpenAIChatStreamIterable.d.ts +19 -0
  242. package/model-provider/openai/chat/OpenAIChatStreamIterable.js +105 -0
  243. package/model-provider/openai/chat/countOpenAIChatMessageTokens.cjs +28 -0
  244. package/model-provider/openai/chat/countOpenAIChatMessageTokens.d.ts +20 -0
  245. package/model-provider/openai/chat/countOpenAIChatMessageTokens.js +23 -0
  246. package/model-provider/openai/index.cjs +31 -0
  247. package/model-provider/openai/index.d.ts +13 -0
  248. package/model-provider/openai/index.js +12 -0
  249. package/model-provider/stability/StabilityError.cjs +36 -0
  250. package/model-provider/stability/StabilityError.d.ts +22 -0
  251. package/model-provider/stability/StabilityError.js +28 -0
  252. package/model-provider/stability/StabilityImageGenerationModel.cjs +133 -0
  253. package/model-provider/stability/StabilityImageGenerationModel.d.ts +95 -0
  254. package/model-provider/stability/StabilityImageGenerationModel.js +129 -0
  255. package/model-provider/stability/index.cjs +20 -0
  256. package/model-provider/stability/index.d.ts +2 -0
  257. package/model-provider/stability/index.js +2 -0
  258. package/package.json +87 -0
  259. package/prompt/InstructionPrompt.cjs +2 -0
  260. package/prompt/InstructionPrompt.d.ts +7 -0
  261. package/prompt/InstructionPrompt.js +1 -0
  262. package/prompt/Llama2PromptMapping.cjs +56 -0
  263. package/prompt/Llama2PromptMapping.d.ts +10 -0
  264. package/prompt/Llama2PromptMapping.js +51 -0
  265. package/prompt/OpenAIChatPromptMapping.cjs +62 -0
  266. package/prompt/OpenAIChatPromptMapping.d.ts +6 -0
  267. package/prompt/OpenAIChatPromptMapping.js +57 -0
  268. package/prompt/PromptMapping.cjs +2 -0
  269. package/prompt/PromptMapping.d.ts +7 -0
  270. package/prompt/PromptMapping.js +1 -0
  271. package/prompt/PromptMappingTextGenerationModel.cjs +88 -0
  272. package/prompt/PromptMappingTextGenerationModel.d.ts +26 -0
  273. package/prompt/PromptMappingTextGenerationModel.js +84 -0
  274. package/prompt/TextPromptMapping.cjs +50 -0
  275. package/prompt/TextPromptMapping.d.ts +14 -0
  276. package/prompt/TextPromptMapping.js +45 -0
  277. package/prompt/chat/ChatPrompt.cjs +2 -0
  278. package/prompt/chat/ChatPrompt.d.ts +33 -0
  279. package/prompt/chat/ChatPrompt.js +1 -0
  280. package/prompt/chat/trimChatPrompt.cjs +50 -0
  281. package/prompt/chat/trimChatPrompt.d.ts +19 -0
  282. package/prompt/chat/trimChatPrompt.js +46 -0
  283. package/prompt/chat/validateChatPrompt.cjs +36 -0
  284. package/prompt/chat/validateChatPrompt.d.ts +8 -0
  285. package/prompt/chat/validateChatPrompt.js +31 -0
  286. package/prompt/index.cjs +25 -0
  287. package/prompt/index.d.ts +9 -0
  288. package/prompt/index.js +9 -0
  289. package/run/ConsoleLogger.cjs +12 -0
  290. package/run/ConsoleLogger.d.ts +6 -0
  291. package/run/ConsoleLogger.js +8 -0
  292. package/run/DefaultRun.cjs +78 -0
  293. package/run/DefaultRun.d.ts +24 -0
  294. package/run/DefaultRun.js +74 -0
  295. package/run/IdMetadata.cjs +2 -0
  296. package/run/IdMetadata.d.ts +7 -0
  297. package/run/IdMetadata.js +1 -0
  298. package/run/Run.cjs +2 -0
  299. package/run/Run.d.ts +27 -0
  300. package/run/Run.js +1 -0
  301. package/run/RunFunction.cjs +2 -0
  302. package/run/RunFunction.d.ts +13 -0
  303. package/run/RunFunction.js +1 -0
  304. package/run/Vector.cjs +2 -0
  305. package/run/Vector.d.ts +5 -0
  306. package/run/Vector.js +1 -0
  307. package/run/index.cjs +22 -0
  308. package/run/index.d.ts +6 -0
  309. package/run/index.js +6 -0
  310. package/text-chunk/TextChunk.cjs +2 -0
  311. package/text-chunk/TextChunk.d.ts +3 -0
  312. package/text-chunk/TextChunk.js +1 -0
  313. package/text-chunk/index.cjs +22 -0
  314. package/text-chunk/index.d.ts +6 -0
  315. package/text-chunk/index.js +6 -0
  316. package/text-chunk/retrieve-text-chunks/TextChunkRetriever.cjs +2 -0
  317. package/text-chunk/retrieve-text-chunks/TextChunkRetriever.d.ts +8 -0
  318. package/text-chunk/retrieve-text-chunks/TextChunkRetriever.js +1 -0
  319. package/text-chunk/retrieve-text-chunks/retrieveTextChunks.cjs +10 -0
  320. package/text-chunk/retrieve-text-chunks/retrieveTextChunks.d.ts +6 -0
  321. package/text-chunk/retrieve-text-chunks/retrieveTextChunks.js +6 -0
  322. package/text-chunk/split/SplitFunction.cjs +2 -0
  323. package/text-chunk/split/SplitFunction.d.ts +4 -0
  324. package/text-chunk/split/SplitFunction.js +1 -0
  325. package/text-chunk/split/splitOnSeparator.cjs +12 -0
  326. package/text-chunk/split/splitOnSeparator.d.ts +8 -0
  327. package/text-chunk/split/splitOnSeparator.js +7 -0
  328. package/text-chunk/split/splitRecursively.cjs +41 -0
  329. package/text-chunk/split/splitRecursively.d.ts +22 -0
  330. package/text-chunk/split/splitRecursively.js +33 -0
  331. package/util/DurationMeasurement.cjs +42 -0
  332. package/util/DurationMeasurement.d.ts +5 -0
  333. package/util/DurationMeasurement.js +38 -0
  334. package/util/ErrorHandler.cjs +2 -0
  335. package/util/ErrorHandler.d.ts +1 -0
  336. package/util/ErrorHandler.js +1 -0
  337. package/util/SafeResult.cjs +2 -0
  338. package/util/SafeResult.d.ts +8 -0
  339. package/util/SafeResult.js +1 -0
  340. package/util/api/AbortError.cjs +9 -0
  341. package/util/api/AbortError.d.ts +3 -0
  342. package/util/api/AbortError.js +5 -0
  343. package/util/api/ApiCallError.cjs +45 -0
  344. package/util/api/ApiCallError.d.ts +15 -0
  345. package/util/api/ApiCallError.js +41 -0
  346. package/util/api/RetryError.cjs +24 -0
  347. package/util/api/RetryError.d.ts +10 -0
  348. package/util/api/RetryError.js +20 -0
  349. package/util/api/RetryFunction.cjs +2 -0
  350. package/util/api/RetryFunction.d.ts +1 -0
  351. package/util/api/RetryFunction.js +1 -0
  352. package/util/api/ThrottleFunction.cjs +2 -0
  353. package/util/api/ThrottleFunction.d.ts +1 -0
  354. package/util/api/ThrottleFunction.js +1 -0
  355. package/util/api/callWithRetryAndThrottle.cjs +7 -0
  356. package/util/api/callWithRetryAndThrottle.d.ts +7 -0
  357. package/util/api/callWithRetryAndThrottle.js +3 -0
  358. package/util/api/postToApi.cjs +103 -0
  359. package/util/api/postToApi.d.ts +29 -0
  360. package/util/api/postToApi.js +96 -0
  361. package/util/api/retryNever.cjs +8 -0
  362. package/util/api/retryNever.d.ts +4 -0
  363. package/util/api/retryNever.js +4 -0
  364. package/util/api/retryWithExponentialBackoff.cjs +48 -0
  365. package/util/api/retryWithExponentialBackoff.d.ts +10 -0
  366. package/util/api/retryWithExponentialBackoff.js +44 -0
  367. package/util/api/throttleMaxConcurrency.cjs +65 -0
  368. package/util/api/throttleMaxConcurrency.d.ts +7 -0
  369. package/util/api/throttleMaxConcurrency.js +61 -0
  370. package/util/api/throttleUnlimitedConcurrency.cjs +8 -0
  371. package/util/api/throttleUnlimitedConcurrency.d.ts +5 -0
  372. package/util/api/throttleUnlimitedConcurrency.js +4 -0
  373. package/util/cosineSimilarity.cjs +26 -0
  374. package/util/cosineSimilarity.d.ts +11 -0
  375. package/util/cosineSimilarity.js +22 -0
  376. package/util/index.cjs +26 -0
  377. package/util/index.d.ts +10 -0
  378. package/util/index.js +10 -0
  379. package/util/never.cjs +6 -0
  380. package/util/never.d.ts +1 -0
  381. package/util/never.js +2 -0
  382. package/util/runSafe.cjs +15 -0
  383. package/util/runSafe.d.ts +2 -0
  384. package/util/runSafe.js +11 -0
  385. package/vector-index/VectorIndex.cjs +2 -0
  386. package/vector-index/VectorIndex.d.ts +18 -0
  387. package/vector-index/VectorIndex.js +1 -0
  388. package/vector-index/VectorIndexSimilarTextChunkRetriever.cjs +57 -0
  389. package/vector-index/VectorIndexSimilarTextChunkRetriever.d.ts +20 -0
  390. package/vector-index/VectorIndexSimilarTextChunkRetriever.js +53 -0
  391. package/vector-index/VectorIndexTextChunkStore.cjs +77 -0
  392. package/vector-index/VectorIndexTextChunkStore.d.ts +35 -0
  393. package/vector-index/VectorIndexTextChunkStore.js +73 -0
  394. package/vector-index/index.cjs +22 -0
  395. package/vector-index/index.d.ts +6 -0
  396. package/vector-index/index.js +6 -0
  397. package/vector-index/memory/MemoryVectorIndex.cjs +63 -0
  398. package/vector-index/memory/MemoryVectorIndex.d.ts +31 -0
  399. package/vector-index/memory/MemoryVectorIndex.js +56 -0
  400. package/vector-index/pinecone/PineconeVectorIndex.cjs +66 -0
  401. package/vector-index/pinecone/PineconeVectorIndex.d.ts +29 -0
  402. package/vector-index/pinecone/PineconeVectorIndex.js +62 -0
  403. package/vector-index/upsertTextChunks.cjs +15 -0
  404. package/vector-index/upsertTextChunks.d.ts +11 -0
  405. package/vector-index/upsertTextChunks.js +11 -0
@@ -0,0 +1,8 @@
1
+ export type SafeResult<OUTPUT> = {
2
+ ok: true;
3
+ output: OUTPUT;
4
+ } | {
5
+ ok: false;
6
+ isAborted?: boolean;
7
+ error?: unknown;
8
+ };
@@ -0,0 +1 @@
1
+ export {};
@@ -0,0 +1,9 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.AbortError = void 0;
4
+ class AbortError extends Error {
5
+ constructor(message = "Aborted") {
6
+ super(message);
7
+ }
8
+ }
9
+ exports.AbortError = AbortError;
@@ -0,0 +1,3 @@
1
+ export declare class AbortError extends Error {
2
+ constructor(message?: string);
3
+ }
@@ -0,0 +1,5 @@
1
+ export class AbortError extends Error {
2
+ constructor(message = "Aborted") {
3
+ super(message);
4
+ }
5
+ }
@@ -0,0 +1,45 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.ApiCallError = void 0;
4
+ class ApiCallError extends Error {
5
+ constructor({ message, url, requestBodyValues, statusCode, cause, isRetryable = statusCode === 429 || statusCode >= 500, }) {
6
+ super(message);
7
+ Object.defineProperty(this, "url", {
8
+ enumerable: true,
9
+ configurable: true,
10
+ writable: true,
11
+ value: void 0
12
+ });
13
+ Object.defineProperty(this, "requestBodyValues", {
14
+ enumerable: true,
15
+ configurable: true,
16
+ writable: true,
17
+ value: void 0
18
+ });
19
+ Object.defineProperty(this, "statusCode", {
20
+ enumerable: true,
21
+ configurable: true,
22
+ writable: true,
23
+ value: void 0
24
+ });
25
+ Object.defineProperty(this, "cause", {
26
+ enumerable: true,
27
+ configurable: true,
28
+ writable: true,
29
+ value: void 0
30
+ });
31
+ Object.defineProperty(this, "isRetryable", {
32
+ enumerable: true,
33
+ configurable: true,
34
+ writable: true,
35
+ value: void 0
36
+ });
37
+ this.name = "ApiCallError";
38
+ this.url = url;
39
+ this.requestBodyValues = requestBodyValues;
40
+ this.statusCode = statusCode;
41
+ this.cause = cause;
42
+ this.isRetryable = isRetryable;
43
+ }
44
+ }
45
+ exports.ApiCallError = ApiCallError;
@@ -0,0 +1,15 @@
1
+ export declare class ApiCallError extends Error {
2
+ readonly url: string;
3
+ readonly requestBodyValues: unknown;
4
+ readonly statusCode: number;
5
+ readonly cause?: unknown;
6
+ readonly isRetryable: boolean;
7
+ constructor({ message, url, requestBodyValues, statusCode, cause, isRetryable, }: {
8
+ message: string;
9
+ url: string;
10
+ requestBodyValues: unknown;
11
+ statusCode: number;
12
+ cause?: unknown;
13
+ isRetryable?: boolean;
14
+ });
15
+ }
@@ -0,0 +1,41 @@
1
+ export class ApiCallError extends Error {
2
+ constructor({ message, url, requestBodyValues, statusCode, cause, isRetryable = statusCode === 429 || statusCode >= 500, }) {
3
+ super(message);
4
+ Object.defineProperty(this, "url", {
5
+ enumerable: true,
6
+ configurable: true,
7
+ writable: true,
8
+ value: void 0
9
+ });
10
+ Object.defineProperty(this, "requestBodyValues", {
11
+ enumerable: true,
12
+ configurable: true,
13
+ writable: true,
14
+ value: void 0
15
+ });
16
+ Object.defineProperty(this, "statusCode", {
17
+ enumerable: true,
18
+ configurable: true,
19
+ writable: true,
20
+ value: void 0
21
+ });
22
+ Object.defineProperty(this, "cause", {
23
+ enumerable: true,
24
+ configurable: true,
25
+ writable: true,
26
+ value: void 0
27
+ });
28
+ Object.defineProperty(this, "isRetryable", {
29
+ enumerable: true,
30
+ configurable: true,
31
+ writable: true,
32
+ value: void 0
33
+ });
34
+ this.name = "ApiCallError";
35
+ this.url = url;
36
+ this.requestBodyValues = requestBodyValues;
37
+ this.statusCode = statusCode;
38
+ this.cause = cause;
39
+ this.isRetryable = isRetryable;
40
+ }
41
+ }
@@ -0,0 +1,24 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.RetryError = void 0;
4
+ class RetryError extends Error {
5
+ constructor({ message, reason, errors, }) {
6
+ super(message);
7
+ Object.defineProperty(this, "errors", {
8
+ enumerable: true,
9
+ configurable: true,
10
+ writable: true,
11
+ value: void 0
12
+ });
13
+ Object.defineProperty(this, "reason", {
14
+ enumerable: true,
15
+ configurable: true,
16
+ writable: true,
17
+ value: void 0
18
+ });
19
+ this.name = "RetryError";
20
+ this.reason = reason;
21
+ this.errors = errors;
22
+ }
23
+ }
24
+ exports.RetryError = RetryError;
@@ -0,0 +1,10 @@
1
+ export type RetryErrorReason = "maxTriesExceeded" | "errorNotRetryable" | "abort";
2
+ export declare class RetryError extends Error {
3
+ readonly errors: Array<unknown>;
4
+ readonly reason: RetryErrorReason;
5
+ constructor({ message, reason, errors, }: {
6
+ message: string;
7
+ reason: RetryErrorReason;
8
+ errors: Array<unknown>;
9
+ });
10
+ }
@@ -0,0 +1,20 @@
1
+ export class RetryError extends Error {
2
+ constructor({ message, reason, errors, }) {
3
+ super(message);
4
+ Object.defineProperty(this, "errors", {
5
+ enumerable: true,
6
+ configurable: true,
7
+ writable: true,
8
+ value: void 0
9
+ });
10
+ Object.defineProperty(this, "reason", {
11
+ enumerable: true,
12
+ configurable: true,
13
+ writable: true,
14
+ value: void 0
15
+ });
16
+ this.name = "RetryError";
17
+ this.reason = reason;
18
+ this.errors = errors;
19
+ }
20
+ }
@@ -0,0 +1,2 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
@@ -0,0 +1 @@
1
+ export type RetryFunction = <OUTPUT>(fn: () => PromiseLike<OUTPUT>) => PromiseLike<OUTPUT>;
@@ -0,0 +1 @@
1
+ export {};
@@ -0,0 +1,2 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
@@ -0,0 +1 @@
1
+ export type ThrottleFunction = <OUTPUT>(fn: () => PromiseLike<OUTPUT>) => PromiseLike<OUTPUT>;
@@ -0,0 +1 @@
1
+ export {};
@@ -0,0 +1,7 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.callWithRetryAndThrottle = void 0;
4
+ const retryWithExponentialBackoff_js_1 = require("./retryWithExponentialBackoff.cjs");
5
+ const throttleUnlimitedConcurrency_js_1 = require("./throttleUnlimitedConcurrency.cjs");
6
+ const callWithRetryAndThrottle = async ({ retry = (0, retryWithExponentialBackoff_js_1.retryWithExponentialBackoff)(), throttle = (0, throttleUnlimitedConcurrency_js_1.throttleUnlimitedConcurrency)(), call, }) => retry(async () => throttle(call));
7
+ exports.callWithRetryAndThrottle = callWithRetryAndThrottle;
@@ -0,0 +1,7 @@
1
+ import { RetryFunction } from "./RetryFunction.js";
2
+ import { ThrottleFunction } from "./ThrottleFunction.js";
3
+ export declare const callWithRetryAndThrottle: <OUTPUT>({ retry, throttle, call, }: {
4
+ retry?: RetryFunction | undefined;
5
+ throttle?: ThrottleFunction | undefined;
6
+ call: () => PromiseLike<OUTPUT>;
7
+ }) => Promise<OUTPUT>;
@@ -0,0 +1,3 @@
1
+ import { retryWithExponentialBackoff } from "./retryWithExponentialBackoff.js";
2
+ import { throttleUnlimitedConcurrency } from "./throttleUnlimitedConcurrency.js";
3
+ export const callWithRetryAndThrottle = async ({ retry = retryWithExponentialBackoff(), throttle = throttleUnlimitedConcurrency(), call, }) => retry(async () => throttle(call));
@@ -0,0 +1,103 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.postToApi = exports.postJsonToApi = exports.createTextResponseHandler = exports.createJsonResponseHandler = void 0;
4
+ const ApiCallError_js_1 = require("./ApiCallError.cjs");
5
+ const createJsonResponseHandler = (responseSchema) => async ({ response, url, requestBodyValues }) => {
6
+ const parsedResult = responseSchema.safeParse(await response.json());
7
+ if (!parsedResult.success) {
8
+ throw new ApiCallError_js_1.ApiCallError({
9
+ message: "Invalid JSON response",
10
+ cause: parsedResult.error,
11
+ statusCode: response.status,
12
+ url,
13
+ requestBodyValues,
14
+ });
15
+ }
16
+ return parsedResult.data;
17
+ };
18
+ exports.createJsonResponseHandler = createJsonResponseHandler;
19
+ const createTextResponseHandler = () => async ({ response }) => response.text();
20
+ exports.createTextResponseHandler = createTextResponseHandler;
21
+ const postJsonToApi = async ({ url, apiKey, body, failedResponseHandler, successfulResponseHandler, abortSignal, }) => (0, exports.postToApi)({
22
+ url,
23
+ apiKey,
24
+ contentType: "application/json",
25
+ body: {
26
+ content: JSON.stringify(body),
27
+ values: body,
28
+ },
29
+ failedResponseHandler,
30
+ successfulResponseHandler,
31
+ abortSignal,
32
+ });
33
+ exports.postJsonToApi = postJsonToApi;
34
+ const postToApi = async ({ url, apiKey, contentType, body, successfulResponseHandler, failedResponseHandler, abortSignal, }) => {
35
+ try {
36
+ const headers = {};
37
+ if (apiKey !== undefined) {
38
+ headers["Authorization"] = `Bearer ${apiKey}`;
39
+ }
40
+ if (contentType !== null) {
41
+ headers["Content-Type"] = contentType;
42
+ }
43
+ const response = await fetch(url, {
44
+ method: "POST",
45
+ headers,
46
+ body: body.content,
47
+ signal: abortSignal,
48
+ });
49
+ if (!response.ok) {
50
+ try {
51
+ throw await failedResponseHandler({
52
+ response,
53
+ url,
54
+ requestBodyValues: body.values,
55
+ });
56
+ }
57
+ catch (error) {
58
+ if (error instanceof Error) {
59
+ if (error.name === "AbortError" || error instanceof ApiCallError_js_1.ApiCallError) {
60
+ throw error;
61
+ }
62
+ }
63
+ throw new ApiCallError_js_1.ApiCallError({
64
+ message: "Failed to process error response",
65
+ cause: error,
66
+ statusCode: response.status,
67
+ url,
68
+ requestBodyValues: body.values,
69
+ });
70
+ }
71
+ }
72
+ try {
73
+ return await successfulResponseHandler({
74
+ response,
75
+ url,
76
+ requestBodyValues: body.values,
77
+ });
78
+ }
79
+ catch (error) {
80
+ if (error instanceof Error) {
81
+ if (error.name === "AbortError" || error instanceof ApiCallError_js_1.ApiCallError) {
82
+ throw error;
83
+ }
84
+ }
85
+ throw new ApiCallError_js_1.ApiCallError({
86
+ message: "Failed to process successful response",
87
+ cause: error,
88
+ statusCode: response.status,
89
+ url,
90
+ requestBodyValues: body.values,
91
+ });
92
+ }
93
+ }
94
+ catch (error) {
95
+ if (error instanceof Error) {
96
+ if (error.name === "AbortError") {
97
+ throw error;
98
+ }
99
+ }
100
+ throw error;
101
+ }
102
+ };
103
+ exports.postToApi = postToApi;
@@ -0,0 +1,29 @@
1
+ import { z } from "zod";
2
+ import { ApiCallError } from "./ApiCallError.js";
3
+ export type ResponseHandler<T> = (options: {
4
+ url: string;
5
+ requestBodyValues: unknown;
6
+ response: Response;
7
+ }) => PromiseLike<T>;
8
+ export declare const createJsonResponseHandler: <T>(responseSchema: z.ZodType<T, z.ZodTypeDef, T>) => ResponseHandler<T>;
9
+ export declare const createTextResponseHandler: () => ResponseHandler<string>;
10
+ export declare const postJsonToApi: <T>({ url, apiKey, body, failedResponseHandler, successfulResponseHandler, abortSignal, }: {
11
+ url: string;
12
+ apiKey?: string | undefined;
13
+ body: unknown;
14
+ failedResponseHandler: ResponseHandler<ApiCallError>;
15
+ successfulResponseHandler: ResponseHandler<T>;
16
+ abortSignal?: AbortSignal | undefined;
17
+ }) => Promise<T>;
18
+ export declare const postToApi: <T>({ url, apiKey, contentType, body, successfulResponseHandler, failedResponseHandler, abortSignal, }: {
19
+ url: string;
20
+ apiKey?: string | undefined;
21
+ contentType: string | null;
22
+ body: {
23
+ content: string | FormData;
24
+ values: unknown;
25
+ };
26
+ failedResponseHandler: ResponseHandler<Error>;
27
+ successfulResponseHandler: ResponseHandler<T>;
28
+ abortSignal?: AbortSignal | undefined;
29
+ }) => Promise<T>;
@@ -0,0 +1,96 @@
1
+ import { ApiCallError } from "./ApiCallError.js";
2
+ export const createJsonResponseHandler = (responseSchema) => async ({ response, url, requestBodyValues }) => {
3
+ const parsedResult = responseSchema.safeParse(await response.json());
4
+ if (!parsedResult.success) {
5
+ throw new ApiCallError({
6
+ message: "Invalid JSON response",
7
+ cause: parsedResult.error,
8
+ statusCode: response.status,
9
+ url,
10
+ requestBodyValues,
11
+ });
12
+ }
13
+ return parsedResult.data;
14
+ };
15
+ export const createTextResponseHandler = () => async ({ response }) => response.text();
16
+ export const postJsonToApi = async ({ url, apiKey, body, failedResponseHandler, successfulResponseHandler, abortSignal, }) => postToApi({
17
+ url,
18
+ apiKey,
19
+ contentType: "application/json",
20
+ body: {
21
+ content: JSON.stringify(body),
22
+ values: body,
23
+ },
24
+ failedResponseHandler,
25
+ successfulResponseHandler,
26
+ abortSignal,
27
+ });
28
+ export const postToApi = async ({ url, apiKey, contentType, body, successfulResponseHandler, failedResponseHandler, abortSignal, }) => {
29
+ try {
30
+ const headers = {};
31
+ if (apiKey !== undefined) {
32
+ headers["Authorization"] = `Bearer ${apiKey}`;
33
+ }
34
+ if (contentType !== null) {
35
+ headers["Content-Type"] = contentType;
36
+ }
37
+ const response = await fetch(url, {
38
+ method: "POST",
39
+ headers,
40
+ body: body.content,
41
+ signal: abortSignal,
42
+ });
43
+ if (!response.ok) {
44
+ try {
45
+ throw await failedResponseHandler({
46
+ response,
47
+ url,
48
+ requestBodyValues: body.values,
49
+ });
50
+ }
51
+ catch (error) {
52
+ if (error instanceof Error) {
53
+ if (error.name === "AbortError" || error instanceof ApiCallError) {
54
+ throw error;
55
+ }
56
+ }
57
+ throw new ApiCallError({
58
+ message: "Failed to process error response",
59
+ cause: error,
60
+ statusCode: response.status,
61
+ url,
62
+ requestBodyValues: body.values,
63
+ });
64
+ }
65
+ }
66
+ try {
67
+ return await successfulResponseHandler({
68
+ response,
69
+ url,
70
+ requestBodyValues: body.values,
71
+ });
72
+ }
73
+ catch (error) {
74
+ if (error instanceof Error) {
75
+ if (error.name === "AbortError" || error instanceof ApiCallError) {
76
+ throw error;
77
+ }
78
+ }
79
+ throw new ApiCallError({
80
+ message: "Failed to process successful response",
81
+ cause: error,
82
+ statusCode: response.status,
83
+ url,
84
+ requestBodyValues: body.values,
85
+ });
86
+ }
87
+ }
88
+ catch (error) {
89
+ if (error instanceof Error) {
90
+ if (error.name === "AbortError") {
91
+ throw error;
92
+ }
93
+ }
94
+ throw error;
95
+ }
96
+ };
@@ -0,0 +1,8 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.retryNever = void 0;
4
+ /**
5
+ * The `retryNever` strategy never retries a failed API call.
6
+ */
7
+ const retryNever = () => async (f) => f();
8
+ exports.retryNever = retryNever;
@@ -0,0 +1,4 @@
1
+ /**
2
+ * The `retryNever` strategy never retries a failed API call.
3
+ */
4
+ export declare const retryNever: () => <OUTPUT>(f: () => PromiseLike<OUTPUT>) => Promise<OUTPUT>;
@@ -0,0 +1,4 @@
1
+ /**
2
+ * The `retryNever` strategy never retries a failed API call.
3
+ */
4
+ export const retryNever = () => async (f) => f();
@@ -0,0 +1,48 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.retryWithExponentialBackoff = void 0;
4
+ const ApiCallError_js_1 = require("./ApiCallError.cjs");
5
+ const RetryError_js_1 = require("./RetryError.cjs");
6
+ /**
7
+ * The `retryWithExponentialBackoff` strategy retries a failed API call with an exponential backoff.
8
+ * You can configure the maximum number of tries, the initial delay, and the backoff factor.
9
+ */
10
+ const retryWithExponentialBackoff = ({ maxTries = 3, initialDelayInMs = 2000, backoffFactor = 2, } = {}) => async (f) => _retryWithExponentialBackoff(f, {
11
+ maxTries,
12
+ delay: initialDelayInMs,
13
+ backoffFactor,
14
+ });
15
+ exports.retryWithExponentialBackoff = retryWithExponentialBackoff;
16
+ async function _retryWithExponentialBackoff(f, { maxTries, delay, backoffFactor, }, errors = []) {
17
+ try {
18
+ return await f();
19
+ }
20
+ catch (error) {
21
+ const newErrors = [...errors, error];
22
+ const tryNumber = newErrors.length;
23
+ if (tryNumber > maxTries) {
24
+ throw new RetryError_js_1.RetryError({
25
+ message: `Failed after ${tryNumber} tries.`,
26
+ reason: "maxTriesExceeded",
27
+ errors: newErrors,
28
+ });
29
+ }
30
+ if (error instanceof Error) {
31
+ if (error.name === "AbortError") {
32
+ throw error;
33
+ }
34
+ if (error instanceof ApiCallError_js_1.ApiCallError &&
35
+ error.isRetryable &&
36
+ tryNumber < maxTries) {
37
+ await new Promise((resolve) => setTimeout(resolve, delay));
38
+ return _retryWithExponentialBackoff(f, { maxTries, delay: backoffFactor * delay, backoffFactor }, newErrors);
39
+ }
40
+ }
41
+ const errorMessage = error instanceof Error ? error.message : String(error);
42
+ throw new RetryError_js_1.RetryError({
43
+ message: `Failed after ${tryNumber} attempt(s) with non-retryable error: '${errorMessage}'`,
44
+ reason: "errorNotRetryable",
45
+ errors: newErrors,
46
+ });
47
+ }
48
+ }
@@ -0,0 +1,10 @@
1
+ import { RetryFunction } from "./RetryFunction.js";
2
+ /**
3
+ * The `retryWithExponentialBackoff` strategy retries a failed API call with an exponential backoff.
4
+ * You can configure the maximum number of tries, the initial delay, and the backoff factor.
5
+ */
6
+ export declare const retryWithExponentialBackoff: ({ maxTries, initialDelayInMs, backoffFactor, }?: {
7
+ maxTries?: number | undefined;
8
+ initialDelayInMs?: number | undefined;
9
+ backoffFactor?: number | undefined;
10
+ }) => RetryFunction;
@@ -0,0 +1,44 @@
1
+ import { ApiCallError } from "./ApiCallError.js";
2
+ import { RetryError } from "./RetryError.js";
3
+ /**
4
+ * The `retryWithExponentialBackoff` strategy retries a failed API call with an exponential backoff.
5
+ * You can configure the maximum number of tries, the initial delay, and the backoff factor.
6
+ */
7
+ export const retryWithExponentialBackoff = ({ maxTries = 3, initialDelayInMs = 2000, backoffFactor = 2, } = {}) => async (f) => _retryWithExponentialBackoff(f, {
8
+ maxTries,
9
+ delay: initialDelayInMs,
10
+ backoffFactor,
11
+ });
12
+ async function _retryWithExponentialBackoff(f, { maxTries, delay, backoffFactor, }, errors = []) {
13
+ try {
14
+ return await f();
15
+ }
16
+ catch (error) {
17
+ const newErrors = [...errors, error];
18
+ const tryNumber = newErrors.length;
19
+ if (tryNumber > maxTries) {
20
+ throw new RetryError({
21
+ message: `Failed after ${tryNumber} tries.`,
22
+ reason: "maxTriesExceeded",
23
+ errors: newErrors,
24
+ });
25
+ }
26
+ if (error instanceof Error) {
27
+ if (error.name === "AbortError") {
28
+ throw error;
29
+ }
30
+ if (error instanceof ApiCallError &&
31
+ error.isRetryable &&
32
+ tryNumber < maxTries) {
33
+ await new Promise((resolve) => setTimeout(resolve, delay));
34
+ return _retryWithExponentialBackoff(f, { maxTries, delay: backoffFactor * delay, backoffFactor }, newErrors);
35
+ }
36
+ }
37
+ const errorMessage = error instanceof Error ? error.message : String(error);
38
+ throw new RetryError({
39
+ message: `Failed after ${tryNumber} attempt(s) with non-retryable error: '${errorMessage}'`,
40
+ reason: "errorNotRetryable",
41
+ errors: newErrors,
42
+ });
43
+ }
44
+ }
@@ -0,0 +1,65 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.throttleMaxConcurrency = void 0;
4
+ class MaxConcurrencyThrottler {
5
+ constructor({ maxConcurrentCalls }) {
6
+ Object.defineProperty(this, "maxConcurrentCalls", {
7
+ enumerable: true,
8
+ configurable: true,
9
+ writable: true,
10
+ value: void 0
11
+ });
12
+ Object.defineProperty(this, "activeCallCount", {
13
+ enumerable: true,
14
+ configurable: true,
15
+ writable: true,
16
+ value: void 0
17
+ });
18
+ Object.defineProperty(this, "callQueue", {
19
+ enumerable: true,
20
+ configurable: true,
21
+ writable: true,
22
+ value: void 0
23
+ });
24
+ this.maxConcurrentCalls = maxConcurrentCalls;
25
+ this.activeCallCount = 0;
26
+ this.callQueue = [];
27
+ }
28
+ async run(fn) {
29
+ return new Promise((resolve, reject) => {
30
+ const tryExecute = async () => {
31
+ if (this.activeCallCount >= this.maxConcurrentCalls)
32
+ return;
33
+ // mark as active and remove from queue:
34
+ this.activeCallCount++;
35
+ const idx = this.callQueue.indexOf(tryExecute);
36
+ if (idx !== -1)
37
+ this.callQueue.splice(idx, 1);
38
+ try {
39
+ resolve(await fn());
40
+ }
41
+ catch (error) {
42
+ reject(error);
43
+ }
44
+ finally {
45
+ this.activeCallCount--;
46
+ if (this.callQueue.length > 0) {
47
+ this.callQueue[0]();
48
+ }
49
+ }
50
+ };
51
+ this.callQueue.push(tryExecute);
52
+ if (this.activeCallCount < this.maxConcurrentCalls) {
53
+ tryExecute();
54
+ }
55
+ });
56
+ }
57
+ }
58
+ /**
59
+ * The `throttleMaxConcurrency` strategy limits the number of parallel API calls.
60
+ */
61
+ function throttleMaxConcurrency({ maxConcurrentCalls, }) {
62
+ const throttler = new MaxConcurrencyThrottler({ maxConcurrentCalls });
63
+ return (fn) => throttler.run(fn);
64
+ }
65
+ exports.throttleMaxConcurrency = throttleMaxConcurrency;