@ax-llm/ax 9.0.19 → 9.0.20

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 (310) hide show
  1. package/package.json +33 -85
  2. package/CHANGELOG.md +0 -184
  3. package/LICENSE +0 -201
  4. package/README.md +0 -569
  5. package/build/module/docs/assets/icons.d.ts +0 -1
  6. package/build/module/docs/assets/icons.js +0 -19
  7. package/build/module/docs/assets/icons.js.map +0 -1
  8. package/build/module/docs/assets/main.d.ts +0 -1
  9. package/build/module/docs/assets/main.js +0 -629
  10. package/build/module/docs/assets/main.js.map +0 -1
  11. package/build/module/docs/assets/navigation.d.ts +0 -1
  12. package/build/module/docs/assets/navigation.js +0 -3
  13. package/build/module/docs/assets/navigation.js.map +0 -1
  14. package/build/module/docs/assets/search.d.ts +0 -1
  15. package/build/module/docs/assets/search.js +0 -3
  16. package/build/module/docs/assets/search.js.map +0 -1
  17. package/build/module/src/ai/anthropic/api.d.ts +0 -19
  18. package/build/module/src/ai/anthropic/api.js +0 -326
  19. package/build/module/src/ai/anthropic/api.js.map +0 -1
  20. package/build/module/src/ai/anthropic/index.d.ts +0 -2
  21. package/build/module/src/ai/anthropic/index.js +0 -3
  22. package/build/module/src/ai/anthropic/index.js.map +0 -1
  23. package/build/module/src/ai/anthropic/info.d.ts +0 -2
  24. package/build/module/src/ai/anthropic/info.js +0 -40
  25. package/build/module/src/ai/anthropic/info.js.map +0 -1
  26. package/build/module/src/ai/anthropic/types.d.ts +0 -171
  27. package/build/module/src/ai/anthropic/types.js +0 -10
  28. package/build/module/src/ai/anthropic/types.js.map +0 -1
  29. package/build/module/src/ai/azure-openai/api.d.ts +0 -19
  30. package/build/module/src/ai/azure-openai/api.js +0 -30
  31. package/build/module/src/ai/azure-openai/api.js.map +0 -1
  32. package/build/module/src/ai/azure-openai/index.d.ts +0 -1
  33. package/build/module/src/ai/azure-openai/index.js +0 -2
  34. package/build/module/src/ai/azure-openai/index.js.map +0 -1
  35. package/build/module/src/ai/balance.d.ts +0 -23
  36. package/build/module/src/ai/balance.js +0 -87
  37. package/build/module/src/ai/balance.js.map +0 -1
  38. package/build/module/src/ai/base.d.ts +0 -55
  39. package/build/module/src/ai/base.js +0 -343
  40. package/build/module/src/ai/base.js.map +0 -1
  41. package/build/module/src/ai/cohere/api.d.ts +0 -22
  42. package/build/module/src/ai/cohere/api.js +0 -264
  43. package/build/module/src/ai/cohere/api.js.map +0 -1
  44. package/build/module/src/ai/cohere/index.d.ts +0 -2
  45. package/build/module/src/ai/cohere/index.js +0 -3
  46. package/build/module/src/ai/cohere/index.js.map +0 -1
  47. package/build/module/src/ai/cohere/info.d.ts +0 -2
  48. package/build/module/src/ai/cohere/info.js +0 -52
  49. package/build/module/src/ai/cohere/info.js.map +0 -1
  50. package/build/module/src/ai/cohere/types.d.ts +0 -102
  51. package/build/module/src/ai/cohere/types.js +0 -23
  52. package/build/module/src/ai/cohere/types.js.map +0 -1
  53. package/build/module/src/ai/deepseek/api.d.ts +0 -16
  54. package/build/module/src/ai/deepseek/api.js +0 -32
  55. package/build/module/src/ai/deepseek/api.js.map +0 -1
  56. package/build/module/src/ai/deepseek/index.d.ts +0 -2
  57. package/build/module/src/ai/deepseek/index.js +0 -3
  58. package/build/module/src/ai/deepseek/index.js.map +0 -1
  59. package/build/module/src/ai/deepseek/info.d.ts +0 -2
  60. package/build/module/src/ai/deepseek/info.js +0 -16
  61. package/build/module/src/ai/deepseek/info.js.map +0 -1
  62. package/build/module/src/ai/deepseek/types.d.ts +0 -8
  63. package/build/module/src/ai/deepseek/types.js +0 -10
  64. package/build/module/src/ai/deepseek/types.js.map +0 -1
  65. package/build/module/src/ai/google-gemini/api.d.ts +0 -36
  66. package/build/module/src/ai/google-gemini/api.js +0 -299
  67. package/build/module/src/ai/google-gemini/api.js.map +0 -1
  68. package/build/module/src/ai/google-gemini/index.d.ts +0 -2
  69. package/build/module/src/ai/google-gemini/index.js +0 -3
  70. package/build/module/src/ai/google-gemini/index.js.map +0 -1
  71. package/build/module/src/ai/google-gemini/info.d.ts +0 -6
  72. package/build/module/src/ai/google-gemini/info.js +0 -29
  73. package/build/module/src/ai/google-gemini/info.js.map +0 -1
  74. package/build/module/src/ai/google-gemini/types.d.ts +0 -146
  75. package/build/module/src/ai/google-gemini/types.js +0 -27
  76. package/build/module/src/ai/google-gemini/types.js.map +0 -1
  77. package/build/module/src/ai/groq/api.d.ts +0 -18
  78. package/build/module/src/ai/groq/api.js +0 -52
  79. package/build/module/src/ai/groq/api.js.map +0 -1
  80. package/build/module/src/ai/groq/index.d.ts +0 -1
  81. package/build/module/src/ai/groq/index.js +0 -2
  82. package/build/module/src/ai/groq/index.js.map +0 -1
  83. package/build/module/src/ai/groq/types.d.ts +0 -6
  84. package/build/module/src/ai/groq/types.js +0 -8
  85. package/build/module/src/ai/groq/types.js.map +0 -1
  86. package/build/module/src/ai/huggingface/api.d.ts +0 -19
  87. package/build/module/src/ai/huggingface/api.js +0 -115
  88. package/build/module/src/ai/huggingface/api.js.map +0 -1
  89. package/build/module/src/ai/huggingface/index.d.ts +0 -2
  90. package/build/module/src/ai/huggingface/index.js +0 -3
  91. package/build/module/src/ai/huggingface/index.js.map +0 -1
  92. package/build/module/src/ai/huggingface/info.d.ts +0 -6
  93. package/build/module/src/ai/huggingface/info.js +0 -6
  94. package/build/module/src/ai/huggingface/info.js.map +0 -1
  95. package/build/module/src/ai/huggingface/types.d.ts +0 -34
  96. package/build/module/src/ai/huggingface/types.js +0 -5
  97. package/build/module/src/ai/huggingface/types.js.map +0 -1
  98. package/build/module/src/ai/index.d.ts +0 -16
  99. package/build/module/src/ai/index.js +0 -16
  100. package/build/module/src/ai/index.js.map +0 -1
  101. package/build/module/src/ai/mistral/api.d.ts +0 -16
  102. package/build/module/src/ai/mistral/api.js +0 -32
  103. package/build/module/src/ai/mistral/api.js.map +0 -1
  104. package/build/module/src/ai/mistral/index.d.ts +0 -2
  105. package/build/module/src/ai/mistral/index.js +0 -3
  106. package/build/module/src/ai/mistral/index.js.map +0 -1
  107. package/build/module/src/ai/mistral/info.d.ts +0 -2
  108. package/build/module/src/ai/mistral/info.js +0 -34
  109. package/build/module/src/ai/mistral/info.js.map +0 -1
  110. package/build/module/src/ai/mistral/types.d.ts +0 -10
  111. package/build/module/src/ai/mistral/types.js +0 -13
  112. package/build/module/src/ai/mistral/types.js.map +0 -1
  113. package/build/module/src/ai/ollama/api.d.ts +0 -22
  114. package/build/module/src/ai/ollama/api.js +0 -32
  115. package/build/module/src/ai/ollama/api.js.map +0 -1
  116. package/build/module/src/ai/ollama/index.d.ts +0 -1
  117. package/build/module/src/ai/ollama/index.js +0 -2
  118. package/build/module/src/ai/ollama/index.js.map +0 -1
  119. package/build/module/src/ai/openai/api.d.ts +0 -29
  120. package/build/module/src/ai/openai/api.js +0 -283
  121. package/build/module/src/ai/openai/api.js.map +0 -1
  122. package/build/module/src/ai/openai/index.d.ts +0 -2
  123. package/build/module/src/ai/openai/index.js +0 -3
  124. package/build/module/src/ai/openai/index.js.map +0 -1
  125. package/build/module/src/ai/openai/info.d.ts +0 -6
  126. package/build/module/src/ai/openai/info.js +0 -50
  127. package/build/module/src/ai/openai/info.js.map +0 -1
  128. package/build/module/src/ai/openai/types.d.ts +0 -165
  129. package/build/module/src/ai/openai/types.js +0 -18
  130. package/build/module/src/ai/openai/types.js.map +0 -1
  131. package/build/module/src/ai/together/api.d.ts +0 -15
  132. package/build/module/src/ai/together/api.js +0 -28
  133. package/build/module/src/ai/together/api.js.map +0 -1
  134. package/build/module/src/ai/together/index.d.ts +0 -1
  135. package/build/module/src/ai/together/index.js +0 -2
  136. package/build/module/src/ai/together/index.js.map +0 -1
  137. package/build/module/src/ai/together/info.d.ts +0 -2
  138. package/build/module/src/ai/together/info.js +0 -2
  139. package/build/module/src/ai/together/info.js.map +0 -1
  140. package/build/module/src/ai/types.d.ts +0 -159
  141. package/build/module/src/ai/types.js +0 -2
  142. package/build/module/src/ai/types.js.map +0 -1
  143. package/build/module/src/ai/util.d.ts +0 -12
  144. package/build/module/src/ai/util.js +0 -63
  145. package/build/module/src/ai/util.js.map +0 -1
  146. package/build/module/src/ai/wrap.d.ts +0 -47
  147. package/build/module/src/ai/wrap.js +0 -95
  148. package/build/module/src/ai/wrap.js.map +0 -1
  149. package/build/module/src/db/base.d.ts +0 -23
  150. package/build/module/src/db/base.js +0 -87
  151. package/build/module/src/db/base.js.map +0 -1
  152. package/build/module/src/db/cloudflare.d.ts +0 -21
  153. package/build/module/src/db/cloudflare.js +0 -98
  154. package/build/module/src/db/cloudflare.js.map +0 -1
  155. package/build/module/src/db/index.d.ts +0 -6
  156. package/build/module/src/db/index.js +0 -7
  157. package/build/module/src/db/index.js.map +0 -1
  158. package/build/module/src/db/memory.d.ts +0 -21
  159. package/build/module/src/db/memory.js +0 -99
  160. package/build/module/src/db/memory.js.map +0 -1
  161. package/build/module/src/db/pinecone.d.ts +0 -21
  162. package/build/module/src/db/pinecone.js +0 -71
  163. package/build/module/src/db/pinecone.js.map +0 -1
  164. package/build/module/src/db/types.d.ts +0 -35
  165. package/build/module/src/db/types.js +0 -3
  166. package/build/module/src/db/types.js.map +0 -1
  167. package/build/module/src/db/weaviate.d.ts +0 -21
  168. package/build/module/src/db/weaviate.js +0 -127
  169. package/build/module/src/db/weaviate.js.map +0 -1
  170. package/build/module/src/db/wrap.d.ts +0 -13
  171. package/build/module/src/db/wrap.js +0 -35
  172. package/build/module/src/db/wrap.js.map +0 -1
  173. package/build/module/src/docs/index.d.ts +0 -4
  174. package/build/module/src/docs/index.js +0 -5
  175. package/build/module/src/docs/index.js.map +0 -1
  176. package/build/module/src/docs/manager.d.ts +0 -45
  177. package/build/module/src/docs/manager.js +0 -159
  178. package/build/module/src/docs/manager.js.map +0 -1
  179. package/build/module/src/docs/reranker.d.ts +0 -7
  180. package/build/module/src/docs/reranker.js +0 -26
  181. package/build/module/src/docs/reranker.js.map +0 -1
  182. package/build/module/src/docs/rewriter.d.ts +0 -4
  183. package/build/module/src/docs/rewriter.js +0 -9
  184. package/build/module/src/docs/rewriter.js.map +0 -1
  185. package/build/module/src/docs/tika.d.ts +0 -18
  186. package/build/module/src/docs/tika.js +0 -46
  187. package/build/module/src/docs/tika.js.map +0 -1
  188. package/build/module/src/dsp/asserts.d.ts +0 -32
  189. package/build/module/src/dsp/asserts.js +0 -91
  190. package/build/module/src/dsp/asserts.js.map +0 -1
  191. package/build/module/src/dsp/eval.d.ts +0 -43
  192. package/build/module/src/dsp/eval.js +0 -125
  193. package/build/module/src/dsp/eval.js.map +0 -1
  194. package/build/module/src/dsp/evaluate.d.ts +0 -12
  195. package/build/module/src/dsp/evaluate.js +0 -32
  196. package/build/module/src/dsp/evaluate.js.map +0 -1
  197. package/build/module/src/dsp/extract.d.ts +0 -25
  198. package/build/module/src/dsp/extract.js +0 -153
  199. package/build/module/src/dsp/extract.js.map +0 -1
  200. package/build/module/src/dsp/generate.d.ts +0 -55
  201. package/build/module/src/dsp/generate.js +0 -314
  202. package/build/module/src/dsp/generate.js.map +0 -1
  203. package/build/module/src/dsp/generate.test.d.ts +0 -1
  204. package/build/module/src/dsp/generate.test.js +0 -41
  205. package/build/module/src/dsp/generate.test.js.map +0 -1
  206. package/build/module/src/dsp/index.d.ts +0 -15
  207. package/build/module/src/dsp/index.js +0 -11
  208. package/build/module/src/dsp/index.js.map +0 -1
  209. package/build/module/src/dsp/loader.d.ts +0 -30
  210. package/build/module/src/dsp/loader.js +0 -81
  211. package/build/module/src/dsp/loader.js.map +0 -1
  212. package/build/module/src/dsp/optimize.d.ts +0 -27
  213. package/build/module/src/dsp/optimize.js +0 -95
  214. package/build/module/src/dsp/optimize.js.map +0 -1
  215. package/build/module/src/dsp/parser.d.ts +0 -87
  216. package/build/module/src/dsp/parser.js +0 -1542
  217. package/build/module/src/dsp/parser.js.map +0 -1
  218. package/build/module/src/dsp/program.d.ts +0 -73
  219. package/build/module/src/dsp/program.js +0 -116
  220. package/build/module/src/dsp/program.js.map +0 -1
  221. package/build/module/src/dsp/prompt.d.ts +0 -48
  222. package/build/module/src/dsp/prompt.js +0 -260
  223. package/build/module/src/dsp/prompt.js.map +0 -1
  224. package/build/module/src/dsp/registry.d.ts +0 -6
  225. package/build/module/src/dsp/registry.js +0 -15
  226. package/build/module/src/dsp/registry.js.map +0 -1
  227. package/build/module/src/dsp/router.d.ts +0 -26
  228. package/build/module/src/dsp/router.js +0 -71
  229. package/build/module/src/dsp/router.js.map +0 -1
  230. package/build/module/src/dsp/sig.d.ts +0 -37
  231. package/build/module/src/dsp/sig.js +0 -194
  232. package/build/module/src/dsp/sig.js.map +0 -1
  233. package/build/module/src/dsp/sig.test.d.ts +0 -1
  234. package/build/module/src/dsp/sig.test.js +0 -28
  235. package/build/module/src/dsp/sig.test.js.map +0 -1
  236. package/build/module/src/dsp/stopwords.d.ts +0 -1
  237. package/build/module/src/dsp/stopwords.js +0 -1164
  238. package/build/module/src/dsp/stopwords.js.map +0 -1
  239. package/build/module/src/dsp/strutil.d.ts +0 -11
  240. package/build/module/src/dsp/strutil.js +0 -60
  241. package/build/module/src/dsp/strutil.js.map +0 -1
  242. package/build/module/src/dsp/util.d.ts +0 -5
  243. package/build/module/src/dsp/util.js +0 -76
  244. package/build/module/src/dsp/util.js.map +0 -1
  245. package/build/module/src/funcs/code.d.ts +0 -9
  246. package/build/module/src/funcs/code.js +0 -63
  247. package/build/module/src/funcs/code.js.map +0 -1
  248. package/build/module/src/funcs/embed.d.ts +0 -6
  249. package/build/module/src/funcs/embed.js +0 -29
  250. package/build/module/src/funcs/embed.js.map +0 -1
  251. package/build/module/src/funcs/functions.d.ts +0 -16
  252. package/build/module/src/funcs/functions.js +0 -55
  253. package/build/module/src/funcs/functions.js.map +0 -1
  254. package/build/module/src/funcs/index.d.ts +0 -3
  255. package/build/module/src/funcs/index.js +0 -4
  256. package/build/module/src/funcs/index.js.map +0 -1
  257. package/build/module/src/funcs/jsonschema.d.ts +0 -2
  258. package/build/module/src/funcs/jsonschema.js +0 -97
  259. package/build/module/src/funcs/jsonschema.js.map +0 -1
  260. package/build/module/src/index.d.ts +0 -9
  261. package/build/module/src/index.js.map +0 -1
  262. package/build/module/src/mem/index.d.ts +0 -2
  263. package/build/module/src/mem/index.js +0 -3
  264. package/build/module/src/mem/index.js.map +0 -1
  265. package/build/module/src/mem/memory.d.ts +0 -16
  266. package/build/module/src/mem/memory.js +0 -77
  267. package/build/module/src/mem/memory.js.map +0 -1
  268. package/build/module/src/mem/types.d.ts +0 -10
  269. package/build/module/src/mem/types.js +0 -2
  270. package/build/module/src/mem/types.js.map +0 -1
  271. package/build/module/src/prompts/agent.d.ts +0 -34
  272. package/build/module/src/prompts/agent.js +0 -108
  273. package/build/module/src/prompts/agent.js.map +0 -1
  274. package/build/module/src/prompts/cot.d.ts +0 -9
  275. package/build/module/src/prompts/cot.js +0 -17
  276. package/build/module/src/prompts/cot.js.map +0 -1
  277. package/build/module/src/prompts/index.d.ts +0 -4
  278. package/build/module/src/prompts/index.js +0 -5
  279. package/build/module/src/prompts/index.js.map +0 -1
  280. package/build/module/src/prompts/prompts.test.d.ts +0 -1
  281. package/build/module/src/prompts/prompts.test.js +0 -54
  282. package/build/module/src/prompts/prompts.test.js.map +0 -1
  283. package/build/module/src/prompts/rag.d.ts +0 -23
  284. package/build/module/src/prompts/rag.js +0 -30
  285. package/build/module/src/prompts/rag.js.map +0 -1
  286. package/build/module/src/prompts/react.d.ts +0 -8
  287. package/build/module/src/prompts/react.js +0 -22
  288. package/build/module/src/prompts/react.js.map +0 -1
  289. package/build/module/src/trace/index.d.ts +0 -171
  290. package/build/module/src/trace/index.js +0 -67
  291. package/build/module/src/trace/index.js.map +0 -1
  292. package/build/module/src/util/apicall.d.ts +0 -18
  293. package/build/module/src/util/apicall.js +0 -50
  294. package/build/module/src/util/apicall.js.map +0 -1
  295. package/build/module/src/util/log.d.ts +0 -14
  296. package/build/module/src/util/log.js +0 -31
  297. package/build/module/src/util/log.js.map +0 -1
  298. package/build/module/src/util/other.d.ts +0 -1
  299. package/build/module/src/util/other.js +0 -5
  300. package/build/module/src/util/other.js.map +0 -1
  301. package/build/module/src/util/rate-limit.d.ts +0 -24
  302. package/build/module/src/util/rate-limit.js +0 -49
  303. package/build/module/src/util/rate-limit.js.map +0 -1
  304. package/build/module/src/util/stream.d.ts +0 -9
  305. package/build/module/src/util/stream.js +0 -29
  306. package/build/module/src/util/stream.js.map +0 -1
  307. package/build/module/src/util/transform.d.ts +0 -7
  308. package/build/module/src/util/transform.js +0 -59
  309. package/build/module/src/util/transform.js.map +0 -1
  310. /package/{build/module/src/index.js → index.js} +0 -0
@@ -1,264 +0,0 @@
1
- import { AxBaseAI, axBaseAIDefaultConfig, axBaseAIDefaultCreativeConfig } from '../base.js';
2
- import { axModelInfoCohere } from './info.js';
3
- import { AxAICohereEmbedModel, AxAICohereModel } from './types.js';
4
- export const axAICohereDefaultConfig = () => structuredClone({
5
- model: AxAICohereModel.CommandRPlus,
6
- embedModel: AxAICohereEmbedModel.EmbedEnglishV30,
7
- ...axBaseAIDefaultConfig()
8
- });
9
- export const axAICohereCreativeConfig = () => structuredClone({
10
- model: AxAICohereModel.CommandR,
11
- embedModel: AxAICohereEmbedModel.EmbedEnglishV30,
12
- ...axBaseAIDefaultCreativeConfig()
13
- });
14
- export class AxAICohere extends AxBaseAI {
15
- config;
16
- constructor({ apiKey, config, options }) {
17
- if (!apiKey || apiKey === '') {
18
- throw new Error('Cohere API key not set');
19
- }
20
- const _config = {
21
- ...axAICohereDefaultConfig(),
22
- ...config
23
- };
24
- super({
25
- name: 'Cohere',
26
- apiURL: 'https://api.cohere.ai',
27
- headers: { Authorization: `Bearer ${apiKey}` },
28
- modelInfo: axModelInfoCohere,
29
- models: { model: _config.model },
30
- supportFor: { functions: true, streaming: true },
31
- options
32
- });
33
- this.config = _config;
34
- }
35
- getModelConfig() {
36
- const { config } = this;
37
- return {
38
- maxTokens: config.maxTokens,
39
- temperature: config.temperature,
40
- topP: config.topP,
41
- topK: config.topK,
42
- frequencyPenalty: config.frequencyPenalty,
43
- presencePenalty: config.presencePenalty,
44
- endSequences: config.endSequences,
45
- stopSequences: config.stopSequences
46
- };
47
- }
48
- generateChatReq = (req,
49
- // eslint-disable-next-line @typescript-eslint/no-unused-vars
50
- _config) => {
51
- const model = req.model ?? this.config.model;
52
- const lastChatMsg = req.chatPrompt.at(-1);
53
- const restOfChat = req.chatPrompt.slice(0, -1);
54
- let message;
55
- if (lastChatMsg &&
56
- lastChatMsg.role === 'user' &&
57
- typeof lastChatMsg.content === 'string') {
58
- message = lastChatMsg?.content;
59
- }
60
- const chatHistory = createHistory(restOfChat);
61
- const tools = req.functions?.map((v) => {
62
- const props = {};
63
- if (v.parameters?.properties) {
64
- for (const [key, value] of Object.entries(v.parameters.properties)) {
65
- props[key] = {
66
- description: value.description,
67
- type: value.type,
68
- required: v.parameters.required?.includes(key) ?? false
69
- };
70
- }
71
- }
72
- return {
73
- name: v.name,
74
- description: v.description,
75
- parameter_definitions: props
76
- };
77
- });
78
- // eslint-disable-next-line @typescript-eslint/naming-convention
79
- const tool_results = req.chatPrompt
80
- .filter((chat) => chat.role === 'function')
81
- .map((chat) => {
82
- const fn = tools?.find((t) => t.name === chat.functionId);
83
- if (!fn) {
84
- throw new Error('Function not found');
85
- }
86
- return {
87
- call: { name: fn.name, parameters: fn.parameter_definitions },
88
- outputs: [{ result: chat.result ?? '' }]
89
- };
90
- });
91
- const apiConfig = {
92
- name: '/v1/chat'
93
- };
94
- const reqValue = {
95
- message,
96
- model,
97
- tools,
98
- ...(tool_results && !message ? { tool_results } : {}),
99
- chat_history: chatHistory,
100
- max_tokens: req.modelConfig?.maxTokens ?? this.config.maxTokens,
101
- temperature: req.modelConfig?.temperature ?? this.config.temperature,
102
- k: req.modelConfig?.topK ?? this.config.topK,
103
- p: req.modelConfig?.topP ?? this.config.topP,
104
- frequency_penalty: req.modelConfig?.frequencyPenalty ?? this.config.frequencyPenalty,
105
- presence_penalty: req.modelConfig?.presencePenalty ?? this.config.presencePenalty,
106
- end_sequences: this.config.endSequences,
107
- stop_sequences: req.modelConfig?.stopSequences ?? this.config.stopSequences
108
- };
109
- return [apiConfig, reqValue];
110
- };
111
- generateEmbedReq = (req) => {
112
- const model = req.embedModel ?? this.config.embedModel;
113
- if (!model) {
114
- throw new Error('Embed model not set');
115
- }
116
- if (!req.texts || req.texts.length === 0) {
117
- throw new Error('Embed texts is empty');
118
- }
119
- const apiConfig = {
120
- name: '/v1/embed'
121
- };
122
- const reqValue = {
123
- model,
124
- texts: req.texts ?? [],
125
- input_type: 'classification',
126
- truncate: ''
127
- };
128
- return [apiConfig, reqValue];
129
- };
130
- generateChatResp = (resp) => {
131
- const modelUsage = resp.meta.billed_units
132
- ? {
133
- promptTokens: resp.meta.billed_units.input_tokens,
134
- completionTokens: resp.meta.billed_units.output_tokens,
135
- totalTokens: resp.meta.billed_units.input_tokens +
136
- resp.meta.billed_units.output_tokens
137
- }
138
- : undefined;
139
- let finishReason;
140
- if ('finish_reason' in resp) {
141
- switch (resp.finish_reason) {
142
- case 'COMPLETE':
143
- finishReason = 'stop';
144
- break;
145
- case 'MAX_TOKENS':
146
- finishReason = 'length';
147
- break;
148
- case 'ERROR':
149
- throw new Error('Finish reason: ERROR');
150
- case 'ERROR_TOXIC':
151
- throw new Error('Finish reason: CONTENT_FILTER');
152
- default:
153
- finishReason = 'stop';
154
- break;
155
- }
156
- }
157
- let functionCalls;
158
- if ('tool_calls' in resp) {
159
- functionCalls = resp.tool_calls?.map((v) => {
160
- return {
161
- id: v.name,
162
- type: 'function',
163
- function: { name: v.name, params: v.parameters }
164
- };
165
- });
166
- }
167
- const results = [
168
- {
169
- id: resp.generation_id,
170
- content: resp.text,
171
- functionCalls,
172
- finishReason
173
- }
174
- ];
175
- return {
176
- results,
177
- modelUsage,
178
- remoteId: resp.response_id
179
- };
180
- };
181
- generateChatStreamResp = (resp, state) => {
182
- const ss = state;
183
- if (resp.event_type === 'stream-start') {
184
- ss.generation_id = resp.generation_id;
185
- }
186
- const { results } = this.generateChatResp(resp);
187
- const result = results[0];
188
- if (!result) {
189
- throw new Error('No result');
190
- }
191
- result.id = ss.generation_id ?? '';
192
- return { results };
193
- };
194
- generateEmbedResp = (resp) => {
195
- return {
196
- remoteId: resp.id,
197
- embeddings: resp.embeddings
198
- };
199
- };
200
- }
201
- function createHistory(chatPrompt) {
202
- return chatPrompt.map((chat) => {
203
- let message = '';
204
- if (chat.role === 'system' ||
205
- chat.role === 'assistant' ||
206
- chat.role === 'user') {
207
- if (typeof chat.content === 'string') {
208
- message = chat.content;
209
- }
210
- else {
211
- throw new Error('Multi-modal content not supported');
212
- }
213
- }
214
- switch (chat.role) {
215
- case 'user':
216
- return { role: 'USER', message };
217
- case 'system':
218
- return { role: 'SYSTEM', message };
219
- case 'assistant': {
220
- const toolCalls = createToolCall(chat.functionCalls);
221
- return {
222
- role: 'CHATBOT',
223
- message,
224
- tool_calls: toolCalls
225
- };
226
- }
227
- case 'function': {
228
- const functionCalls = chatPrompt
229
- .map((v) => {
230
- if (v.role === 'assistant') {
231
- return v.functionCalls?.find((f) => f.id === chat.functionId);
232
- }
233
- return undefined;
234
- })
235
- .filter((v) => v !== undefined);
236
- const call = createToolCall(functionCalls)?.at(0);
237
- if (!call) {
238
- throw new Error('Function call not found');
239
- }
240
- const outputs = [{ result: chat.result }];
241
- return {
242
- role: 'TOOL',
243
- tool_results: [
244
- {
245
- call,
246
- outputs
247
- }
248
- ]
249
- };
250
- }
251
- default:
252
- throw new Error('Unknown role');
253
- }
254
- });
255
- }
256
- function createToolCall(functionCalls) {
257
- return functionCalls?.map((v) => {
258
- const parameters = typeof v.function.params === 'string'
259
- ? JSON.parse(v.function.params)
260
- : v.function.params;
261
- return { name: v.function.name, parameters };
262
- });
263
- }
264
- //# sourceMappingURL=api.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../../src/ai/cohere/api.ts"],"names":[],"mappings":"AACA,OAAO,EACL,QAAQ,EACR,qBAAqB,EACrB,6BAA6B,EAC9B,MAAM,YAAY,CAAC;AAWpB,OAAO,EAAE,iBAAiB,EAAE,MAAM,WAAW,CAAC;AAC9C,OAAO,EAKL,oBAAoB,EAGpB,eAAe,EAChB,MAAM,YAAY,CAAC;AAEpB,MAAM,CAAC,MAAM,uBAAuB,GAAG,GAAqB,EAAE,CAC5D,eAAe,CAAC;IACd,KAAK,EAAE,eAAe,CAAC,YAAY;IACnC,UAAU,EAAE,oBAAoB,CAAC,eAAe;IAChD,GAAG,qBAAqB,EAAE;CAC3B,CAAC,CAAC;AAEL,MAAM,CAAC,MAAM,wBAAwB,GAAG,GAAqB,EAAE,CAC7D,eAAe,CAAC;IACd,KAAK,EAAE,eAAe,CAAC,QAAQ;IAC/B,UAAU,EAAE,oBAAoB,CAAC,eAAe;IAChD,GAAG,6BAA6B,EAAE;CACnC,CAAC,CAAC;AASL,MAAM,OAAO,UAAW,SAAQ,QAM/B;IACS,MAAM,CAAmB;IAEjC,YAAY,EACV,MAAM,EACN,MAAM,EACN,OAAO,EACgC;QACvC,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAC5C,CAAC;QACD,MAAM,OAAO,GAAG;YACd,GAAG,uBAAuB,EAAE;YAC5B,GAAG,MAAM;SACV,CAAC;QACF,KAAK,CAAC;YACJ,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,uBAAuB;YAC/B,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE;YAC9C,SAAS,EAAE,iBAAiB;YAC5B,MAAM,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE;YAChC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE;YAChD,OAAO;SACR,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC;IACxB,CAAC;IAEQ,cAAc;QACrB,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;QACxB,OAAO;YACL,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;YACzC,eAAe,EAAE,MAAM,CAAC,eAAe;YACvC,YAAY,EAAE,MAAM,CAAC,YAAY;YACjC,aAAa,EAAE,MAAM,CAAC,aAAa;SACnB,CAAC;IACrB,CAAC;IAEQ,eAAe,GAAG,CACzB,GAA4B;IAC5B,6DAA6D;IAC7D,OAAmC,EACL,EAAE;QAChC,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;QAE7C,MAAM,WAAW,GAAG,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,UAAU,GAAG,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAE/C,IAAI,OAAqD,CAAC;QAE1D,IACE,WAAW;YACX,WAAW,CAAC,IAAI,KAAK,MAAM;YAC3B,OAAO,WAAW,CAAC,OAAO,KAAK,QAAQ,EACvC,CAAC;YACD,OAAO,GAAG,WAAW,EAAE,OAAO,CAAC;QACjC,CAAC;QAED,MAAM,WAAW,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;QAM9C,MAAM,KAAK,GAAmC,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACrE,MAAM,KAAK,GAA8B,EAAE,CAAC;YAC5C,IAAI,CAAC,CAAC,UAAU,EAAE,UAAU,EAAE,CAAC;gBAC7B,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;oBACnE,KAAK,CAAC,GAAG,CAAC,GAAG;wBACX,WAAW,EAAE,KAAK,CAAC,WAAW;wBAC9B,IAAI,EAAE,KAAK,CAAC,IAAI;wBAChB,QAAQ,EAAE,CAAC,CAAC,UAAU,CAAC,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,IAAI,KAAK;qBACxD,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,OAAO;gBACL,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;gBAC1B,qBAAqB,EAAE,KAAK;aAC7B,CAAC;QACJ,CAAC,CAAC,CAAC;QAIH,gEAAgE;QAChE,MAAM,YAAY,GAChB,GAAG,CAAC,UACL;aACE,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,UAAU,CAAC;aAC1C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YACZ,MAAM,EAAE,GAAG,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,UAAU,CAAC,CAAC;YAC1D,IAAI,CAAC,EAAE,EAAE,CAAC;gBACR,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;YACxC,CAAC;YACD,OAAO;gBACL,IAAI,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,UAAU,EAAE,EAAE,CAAC,qBAAqB,EAAE;gBAC7D,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,EAAE,EAAE,CAAC;aACzC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEL,MAAM,SAAS,GAAG;YAChB,IAAI,EAAE,UAAU;SACjB,CAAC;QAEF,MAAM,QAAQ,GAA0B;YACtC,OAAO;YACP,KAAK;YACL,KAAK;YACL,GAAG,CAAC,YAAY,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACrD,YAAY,EAAE,WAAW;YACzB,UAAU,EAAE,GAAG,CAAC,WAAW,EAAE,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS;YAC/D,WAAW,EAAE,GAAG,CAAC,WAAW,EAAE,WAAW,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW;YACpE,CAAC,EAAE,GAAG,CAAC,WAAW,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI;YAC5C,CAAC,EAAE,GAAG,CAAC,WAAW,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI;YAC5C,iBAAiB,EACf,GAAG,CAAC,WAAW,EAAE,gBAAgB,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB;YACnE,gBAAgB,EACd,GAAG,CAAC,WAAW,EAAE,eAAe,IAAI,IAAI,CAAC,MAAM,CAAC,eAAe;YACjE,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;YACvC,cAAc,EACZ,GAAG,CAAC,WAAW,EAAE,aAAa,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa;SAC9D,CAAC;QAEF,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAC/B,CAAC,CAAC;IAEO,gBAAgB,GAAG,CAC1B,GAA6B,EACE,EAAE;QACjC,MAAM,KAAK,GAAG,GAAG,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;QAEvD,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;QACzC,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzC,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC1C,CAAC;QAED,MAAM,SAAS,GAAG;YAChB,IAAI,EAAE,WAAW;SAClB,CAAC;QAEF,MAAM,QAAQ,GAAG;YACf,KAAK;YACL,KAAK,EAAE,GAAG,CAAC,KAAK,IAAI,EAAE;YACtB,UAAU,EAAE,gBAAgB;YAC5B,QAAQ,EAAE,EAAE;SACb,CAAC;QAEF,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAC/B,CAAC,CAAC;IAEO,gBAAgB,GAAG,CAC1B,IAAsC,EACtB,EAAE;QAClB,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY;YACvC,CAAC,CAAC;gBACE,YAAY,EAAE,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,YAAY;gBACjD,gBAAgB,EAAE,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,aAAa;gBACtD,WAAW,EACT,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,YAAY;oBACnC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,aAAa;aACvC;YACH,CAAC,CAAC,SAAS,CAAC;QAEd,IAAI,YAA0D,CAAC;QAC/D,IAAI,eAAe,IAAI,IAAI,EAAE,CAAC;YAC5B,QAAQ,IAAI,CAAC,aAAa,EAAE,CAAC;gBAC3B,KAAK,UAAU;oBACb,YAAY,GAAG,MAAM,CAAC;oBACtB,MAAM;gBACR,KAAK,YAAY;oBACf,YAAY,GAAG,QAAQ,CAAC;oBACxB,MAAM;gBACR,KAAK,OAAO;oBACV,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;gBAC1C,KAAK,aAAa;oBAChB,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;gBACnD;oBACE,YAAY,GAAG,MAAM,CAAC;oBACtB,MAAM;YACV,CAAC;QACH,CAAC;QAED,IAAI,aAA4D,CAAC;QAEjE,IAAI,YAAY,IAAI,IAAI,EAAE,CAAC;YACzB,aAAa,GAAG,IAAI,CAAC,UAAU,EAAE,GAAG,CAClC,CAAC,CAAC,EAAiE,EAAE;gBACnE,OAAO;oBACL,EAAE,EAAE,CAAC,CAAC,IAAI;oBACV,IAAI,EAAE,UAAmB;oBACzB,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,UAAU,EAAE;iBACjD,CAAC;YACJ,CAAC,CACF,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAA8B;YACzC;gBACE,EAAE,EAAE,IAAI,CAAC,aAAa;gBACtB,OAAO,EAAE,IAAI,CAAC,IAAI;gBAClB,aAAa;gBACb,YAAY;aACb;SACF,CAAC;QAEF,OAAO;YACL,OAAO;YACP,UAAU;YACV,QAAQ,EAAE,IAAI,CAAC,WAAW;SAC3B,CAAC;IACJ,CAAC,CAAC;IAEO,sBAAsB,GAAG,CAChC,IAA2C,EAC3C,KAAa,EACG,EAAE;QAClB,MAAM,EAAE,GAAG,KAEV,CAAC;QAEF,IAAI,IAAI,CAAC,UAAU,KAAK,cAAc,EAAE,CAAC;YACvC,EAAE,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,CAAC;QAED,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAChD,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAC1B,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;QAC/B,CAAC;QAED,MAAM,CAAC,EAAE,GAAG,EAAE,CAAC,aAAa,IAAI,EAAE,CAAC;QACnC,OAAO,EAAE,OAAO,EAAE,CAAC;IACrB,CAAC,CAAC;IAEO,iBAAiB,GAAG,CAC3B,IAAuC,EACtB,EAAE;QACnB,OAAO;YACL,QAAQ,EAAE,IAAI,CAAC,EAAE;YACjB,UAAU,EAAE,IAAI,CAAC,UAAU;SAC5B,CAAC;IACJ,CAAC,CAAC;CACH;AACD,SAAS,aAAa,CACpB,UAAiD;IAEjD,OAAO,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QAC7B,IAAI,OAAO,GAAW,EAAE,CAAC;QAEzB,IACE,IAAI,CAAC,IAAI,KAAK,QAAQ;YACtB,IAAI,CAAC,IAAI,KAAK,WAAW;YACzB,IAAI,CAAC,IAAI,KAAK,MAAM,EACpB,CAAC;YACD,IAAI,OAAO,IAAI,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;gBACrC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;YACzB,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;QAED,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,MAAM;gBACT,OAAO,EAAE,IAAI,EAAE,MAAe,EAAE,OAAO,EAAE,CAAC;YAC5C,KAAK,QAAQ;gBACX,OAAO,EAAE,IAAI,EAAE,QAAiB,EAAE,OAAO,EAAE,CAAC;YAC9C,KAAK,WAAW,CAAC,CAAC,CAAC;gBACjB,MAAM,SAAS,GAAG,cAAc,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;gBACrD,OAAO;oBACL,IAAI,EAAE,SAAkB;oBACxB,OAAO;oBACP,UAAU,EAAE,SAAS;iBACtB,CAAC;YACJ,CAAC;YACD,KAAK,UAAU,CAAC,CAAC,CAAC;gBAChB,MAAM,aAAa,GAAG,UAAU;qBAC7B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;oBACT,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;wBAC3B,OAAO,CAAC,CAAC,aAAa,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,UAAU,CAAC,CAAC;oBAChE,CAAC;oBACD,OAAO,SAAS,CAAC;gBACnB,CAAC,CAAC;qBACD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;gBAElC,MAAM,IAAI,GAAG,cAAc,CAAC,aAAa,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;gBAElD,IAAI,CAAC,IAAI,EAAE,CAAC;oBACV,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;gBAC7C,CAAC;gBAED,MAAM,OAAO,GAAG,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;gBAC1C,OAAO;oBACL,IAAI,EAAE,MAAe;oBACrB,YAAY,EAAE;wBACZ;4BACE,IAAI;4BACJ,OAAO;yBACR;qBACF;iBACF,CAAC;YACJ,CAAC;YACD;gBACE,MAAM,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC;QACpC,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC;AACD,SAAS,cAAc,CACrB,aAKC;IAED,OAAO,aAAa,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QAC9B,MAAM,UAAU,GACd,OAAO,CAAC,CAAC,QAAQ,CAAC,MAAM,KAAK,QAAQ;YACnC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC;YAC/B,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC;QACxB,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI,EAAE,UAAU,EAAE,CAAC;IAC/C,CAAC,CAAC,CAAC;AACL,CAAC"}
@@ -1,2 +0,0 @@
1
- export * from './api.js';
2
- export * from './types.js';
@@ -1,3 +0,0 @@
1
- export * from './api.js';
2
- export * from './types.js';
3
- //# sourceMappingURL=index.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/ai/cohere/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAC;AACzB,cAAc,YAAY,CAAC"}
@@ -1,2 +0,0 @@
1
- import type { AxModelInfo } from '../types.js';
2
- export declare const axModelInfoCohere: AxModelInfo[];
@@ -1,52 +0,0 @@
1
- import { AxAICohereEmbedModel, AxAICohereModel } from './types.js';
2
- export const axModelInfoCohere = [
3
- {
4
- name: AxAICohereModel.CommandRPlus,
5
- currency: 'usd',
6
- promptTokenCostPer1M: 3.0,
7
- completionTokenCostPer1M: 15
8
- },
9
- {
10
- name: AxAICohereModel.CommandR,
11
- currency: 'usd',
12
- promptTokenCostPer1M: 0.5,
13
- completionTokenCostPer1M: 1.5
14
- },
15
- {
16
- name: AxAICohereModel.Command,
17
- currency: 'usd',
18
- promptTokenCostPer1M: 0.5,
19
- completionTokenCostPer1M: 1.5
20
- },
21
- {
22
- name: AxAICohereModel.CommandLight,
23
- currency: 'usd',
24
- promptTokenCostPer1M: 0.3,
25
- completionTokenCostPer1M: 0.6
26
- },
27
- {
28
- name: AxAICohereEmbedModel.EmbedEnglishLightV30,
29
- currency: 'usd',
30
- promptTokenCostPer1M: 0.1,
31
- completionTokenCostPer1M: 0.1
32
- },
33
- {
34
- name: AxAICohereEmbedModel.EmbedEnglishV30,
35
- currency: 'usd',
36
- promptTokenCostPer1M: 0.1,
37
- completionTokenCostPer1M: 0.1
38
- },
39
- {
40
- name: AxAICohereEmbedModel.EmbedMultiLingualV30,
41
- currency: 'usd',
42
- promptTokenCostPer1M: 0.1,
43
- completionTokenCostPer1M: 0.1
44
- },
45
- {
46
- name: AxAICohereEmbedModel.EmbedMultiLingualLightV30,
47
- currency: 'usd',
48
- promptTokenCostPer1M: 0.1,
49
- completionTokenCostPer1M: 0.1
50
- }
51
- ];
52
- //# sourceMappingURL=info.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"info.js","sourceRoot":"","sources":["../../../../../src/ai/cohere/info.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,oBAAoB,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAEnE,MAAM,CAAC,MAAM,iBAAiB,GAAkB;IAC9C;QACE,IAAI,EAAE,eAAe,CAAC,YAAY;QAClC,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,EAAE;KAC7B;IACD;QACE,IAAI,EAAE,eAAe,CAAC,QAAQ;QAC9B,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,eAAe,CAAC,OAAO;QAC7B,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,eAAe,CAAC,YAAY;QAClC,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,oBAAoB,CAAC,oBAAoB;QAC/C,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,oBAAoB,CAAC,eAAe;QAC1C,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,oBAAoB,CAAC,oBAAoB;QAC/C,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,oBAAoB,CAAC,yBAAyB;QACpD,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;CACF,CAAC"}
@@ -1,102 +0,0 @@
1
- import type { AxModelConfig } from '../types.js';
2
- /**
3
- * Cohere: Models for text generation
4
- * @export
5
- */
6
- export declare enum AxAICohereModel {
7
- CommandRPlus = "command-r-plus",
8
- CommandR = "command-r",
9
- Command = "command",
10
- CommandLight = "command-light"
11
- }
12
- /**
13
- * Cohere: Models for use in embeddings
14
- * @export
15
- */
16
- export declare enum AxAICohereEmbedModel {
17
- EmbedEnglishV30 = "embed-english-v3.0",
18
- EmbedEnglishLightV30 = "embed-english-light-v3.0",
19
- EmbedMultiLingualV30 = "embed-multilingual-v3.0",
20
- EmbedMultiLingualLightV30 = "embed-multilingual-light-v3.0"
21
- }
22
- /**
23
- * Cohere: Model options for text generation
24
- * @export
25
- */
26
- export type AxAICohereConfig = AxModelConfig & {
27
- model: AxAICohereModel;
28
- embedModel: AxAICohereEmbedModel;
29
- };
30
- export type AxAICohereChatResponseToolCalls = {
31
- name: string;
32
- parameters?: object;
33
- }[];
34
- export type AxAICohereChatRequestToolResults = {
35
- call: AxAICohereChatResponseToolCalls[0];
36
- outputs: object[];
37
- }[];
38
- export type AxAICohereChatRequest = {
39
- message?: string;
40
- preamble?: string;
41
- chat_history: ({
42
- role: 'CHATBOT';
43
- message: string;
44
- tool_calls?: AxAICohereChatResponseToolCalls;
45
- } | {
46
- role: 'SYSTEM';
47
- message: string;
48
- } | {
49
- role: 'USER';
50
- message: string;
51
- } | {
52
- role: 'TOOL';
53
- message?: string;
54
- tool_results: AxAICohereChatRequestToolResults;
55
- })[];
56
- model: AxAICohereModel | string;
57
- max_tokens?: number;
58
- temperature?: number;
59
- k?: number;
60
- p?: number;
61
- frequency_penalty?: number;
62
- presence_penalty?: number;
63
- end_sequences?: readonly string[];
64
- stop_sequences?: string[];
65
- tools?: {
66
- name: string;
67
- description: string;
68
- parameter_definitions: Record<string, {
69
- description: string;
70
- type: string;
71
- required: boolean;
72
- }>;
73
- }[];
74
- tool_results?: AxAICohereChatRequestToolResults;
75
- };
76
- export type AxAICohereChatResponse = {
77
- response_id: string;
78
- meta: {
79
- billed_units: {
80
- input_tokens: number;
81
- output_tokens: number;
82
- };
83
- };
84
- generation_id: string;
85
- text: string;
86
- finish_reason: 'COMPLETE' | 'ERROR' | 'ERROR_TOXIC' | 'ERROR_LIMIT' | 'USER_CANCEL' | 'MAX_TOKENS';
87
- tool_calls: AxAICohereChatResponseToolCalls;
88
- };
89
- export type AxAICohereChatResponseDelta = AxAICohereChatResponse & {
90
- event_type: 'stream-start' | 'text-generation' | 'tool-calls-generation' | 'stream-end';
91
- };
92
- export type AxAICohereEmbedRequest = {
93
- texts: readonly string[];
94
- model: AxAICohereModel | string;
95
- truncate: string;
96
- };
97
- export type AxAICohereEmbedResponse = {
98
- id: string;
99
- texts: string[];
100
- model: string;
101
- embeddings: number[][];
102
- };
@@ -1,23 +0,0 @@
1
- /**
2
- * Cohere: Models for text generation
3
- * @export
4
- */
5
- export var AxAICohereModel;
6
- (function (AxAICohereModel) {
7
- AxAICohereModel["CommandRPlus"] = "command-r-plus";
8
- AxAICohereModel["CommandR"] = "command-r";
9
- AxAICohereModel["Command"] = "command";
10
- AxAICohereModel["CommandLight"] = "command-light";
11
- })(AxAICohereModel || (AxAICohereModel = {}));
12
- /**
13
- * Cohere: Models for use in embeddings
14
- * @export
15
- */
16
- export var AxAICohereEmbedModel;
17
- (function (AxAICohereEmbedModel) {
18
- AxAICohereEmbedModel["EmbedEnglishV30"] = "embed-english-v3.0";
19
- AxAICohereEmbedModel["EmbedEnglishLightV30"] = "embed-english-light-v3.0";
20
- AxAICohereEmbedModel["EmbedMultiLingualV30"] = "embed-multilingual-v3.0";
21
- AxAICohereEmbedModel["EmbedMultiLingualLightV30"] = "embed-multilingual-light-v3.0";
22
- })(AxAICohereEmbedModel || (AxAICohereEmbedModel = {}));
23
- //# sourceMappingURL=types.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../../src/ai/cohere/types.ts"],"names":[],"mappings":"AAEA;;;GAGG;AACH,MAAM,CAAN,IAAY,eAKX;AALD,WAAY,eAAe;IACzB,kDAA+B,CAAA;IAC/B,yCAAsB,CAAA;IACtB,sCAAmB,CAAA;IACnB,iDAA8B,CAAA;AAChC,CAAC,EALW,eAAe,KAAf,eAAe,QAK1B;AAED;;;GAGG;AACH,MAAM,CAAN,IAAY,oBAKX;AALD,WAAY,oBAAoB;IAC9B,8DAAsC,CAAA;IACtC,yEAAiD,CAAA;IACjD,wEAAgD,CAAA;IAChD,mFAA2D,CAAA;AAC7D,CAAC,EALW,oBAAoB,KAApB,oBAAoB,QAK/B"}
@@ -1,16 +0,0 @@
1
- import { AxAIOpenAI } from '../openai/api.js';
2
- import type { AxAIOpenAIConfig } from '../openai/types.js';
3
- import type { AxAIServiceOptions } from '../types.js';
4
- type DeepSeekConfig = AxAIOpenAIConfig;
5
- export declare const axAIDeepSeekDefaultConfig: () => DeepSeekConfig;
6
- export declare const axAIDeepSeekCodeConfig: () => DeepSeekConfig;
7
- export interface AxAIDeepSeekArgs {
8
- name: 'deepseek';
9
- apiKey: string;
10
- config?: Readonly<DeepSeekConfig>;
11
- options?: Readonly<AxAIServiceOptions>;
12
- }
13
- export declare class AxAIDeepSeek extends AxAIOpenAI {
14
- constructor({ apiKey, config, options }: Readonly<Omit<AxAIDeepSeekArgs, 'name'>>);
15
- }
16
- export {};
@@ -1,32 +0,0 @@
1
- import { axBaseAIDefaultConfig, axBaseAIDefaultCreativeConfig } from '../base.js';
2
- import { AxAIOpenAI } from '../openai/api.js';
3
- import { axModelInfoDeepSeek } from './info.js';
4
- import { AxAIDeepSeekModel } from './types.js';
5
- export const axAIDeepSeekDefaultConfig = () => structuredClone({
6
- model: AxAIDeepSeekModel.DeepSeekChat,
7
- ...axBaseAIDefaultConfig()
8
- });
9
- export const axAIDeepSeekCodeConfig = () => structuredClone({
10
- model: AxAIDeepSeekModel.DeepSeekCoder,
11
- ...axBaseAIDefaultCreativeConfig()
12
- });
13
- export class AxAIDeepSeek extends AxAIOpenAI {
14
- constructor({ apiKey, config, options }) {
15
- if (!apiKey || apiKey === '') {
16
- throw new Error('DeepSeek API key not set');
17
- }
18
- const _config = {
19
- ...axAIDeepSeekDefaultConfig(),
20
- ...config
21
- };
22
- super({
23
- apiKey,
24
- config: _config,
25
- options,
26
- apiURL: 'https://api.deepseek.com',
27
- modelInfo: axModelInfoDeepSeek
28
- });
29
- super.setName('DeepSeek');
30
- }
31
- }
32
- //# sourceMappingURL=api.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../../src/ai/deepseek/api.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,qBAAqB,EACrB,6BAA6B,EAC9B,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAI9C,OAAO,EAAE,mBAAmB,EAAE,MAAM,WAAW,CAAC;AAChD,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAI/C,MAAM,CAAC,MAAM,yBAAyB,GAAG,GAAmB,EAAE,CAC5D,eAAe,CAAC;IACd,KAAK,EAAE,iBAAiB,CAAC,YAAY;IACrC,GAAG,qBAAqB,EAAE;CAC3B,CAAC,CAAC;AAEL,MAAM,CAAC,MAAM,sBAAsB,GAAG,GAAmB,EAAE,CACzD,eAAe,CAAC;IACd,KAAK,EAAE,iBAAiB,CAAC,aAAa;IACtC,GAAG,6BAA6B,EAAE;CACnC,CAAC,CAAC;AASL,MAAM,OAAO,YAAa,SAAQ,UAAU;IAC1C,YAAY,EACV,MAAM,EACN,MAAM,EACN,OAAO,EACkC;QACzC,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAC9C,CAAC;QACD,MAAM,OAAO,GAAG;YACd,GAAG,yBAAyB,EAAE;YAC9B,GAAG,MAAM;SACV,CAAC;QACF,KAAK,CAAC;YACJ,MAAM;YACN,MAAM,EAAE,OAAO;YACf,OAAO;YACP,MAAM,EAAE,0BAA0B;YAClC,SAAS,EAAE,mBAAmB;SAC/B,CAAC,CAAC;QAEH,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAC5B,CAAC;CACF"}
@@ -1,2 +0,0 @@
1
- export * from './api.js';
2
- export * from './types.js';
@@ -1,3 +0,0 @@
1
- export * from './api.js';
2
- export * from './types.js';
3
- //# sourceMappingURL=index.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/ai/deepseek/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAC;AACzB,cAAc,YAAY,CAAC"}
@@ -1,2 +0,0 @@
1
- import type { AxModelInfo } from '../types.js';
2
- export declare const axModelInfoDeepSeek: AxModelInfo[];
@@ -1,16 +0,0 @@
1
- import { AxAIDeepSeekModel } from './types.js';
2
- export const axModelInfoDeepSeek = [
3
- {
4
- name: AxAIDeepSeekModel.DeepSeekChat,
5
- currency: 'USD',
6
- promptTokenCostPer1M: 0.14,
7
- completionTokenCostPer1M: 0.28
8
- },
9
- {
10
- name: AxAIDeepSeekModel.DeepSeekCoder,
11
- currency: 'USD',
12
- promptTokenCostPer1M: 0.14,
13
- completionTokenCostPer1M: 0.28
14
- }
15
- ];
16
- //# sourceMappingURL=info.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"info.js","sourceRoot":"","sources":["../../../../../src/ai/deepseek/info.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAE/C,MAAM,CAAC,MAAM,mBAAmB,GAAkB;IAChD;QACE,IAAI,EAAE,iBAAiB,CAAC,YAAY;QACpC,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,IAAI;QAC1B,wBAAwB,EAAE,IAAI;KAC/B;IACD;QACE,IAAI,EAAE,iBAAiB,CAAC,aAAa;QACrC,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,IAAI;QAC1B,wBAAwB,EAAE,IAAI;KAC/B;CACF,CAAC"}
@@ -1,8 +0,0 @@
1
- /**
2
- * DeepSeek: Models for text generation
3
- * @export
4
- */
5
- export declare enum AxAIDeepSeekModel {
6
- DeepSeekChat = "deepseek-chat",
7
- DeepSeekCoder = "deepseek-coder"
8
- }
@@ -1,10 +0,0 @@
1
- /**
2
- * DeepSeek: Models for text generation
3
- * @export
4
- */
5
- export var AxAIDeepSeekModel;
6
- (function (AxAIDeepSeekModel) {
7
- AxAIDeepSeekModel["DeepSeekChat"] = "deepseek-chat";
8
- AxAIDeepSeekModel["DeepSeekCoder"] = "deepseek-coder";
9
- })(AxAIDeepSeekModel || (AxAIDeepSeekModel = {}));
10
- //# sourceMappingURL=types.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../../src/ai/deepseek/types.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,MAAM,CAAN,IAAY,iBAGX;AAHD,WAAY,iBAAiB;IAC3B,mDAA8B,CAAA;IAC9B,qDAAgC,CAAA;AAClC,CAAC,EAHW,iBAAiB,KAAjB,iBAAiB,QAG5B"}
@@ -1,36 +0,0 @@
1
- import type { API } from '../../util/apicall.js';
2
- import { AxBaseAI } from '../base.js';
3
- import type { AxAIServiceOptions, AxChatRequest, AxChatResponse, AxEmbedRequest, AxEmbedResponse, AxModelConfig } from '../types.js';
4
- import { type AxAIGoogleGeminiBatchEmbedRequest, type AxAIGoogleGeminiBatchEmbedResponse, type AxAIGoogleGeminiChatRequest, type AxAIGoogleGeminiChatResponse, type AxAIGoogleGeminiChatResponseDelta, type AxAIGoogleGeminiConfig } from './types.js';
5
- /**
6
- * AxAIGoogleGemini: Default Model options for text generation
7
- * @export
8
- */
9
- export declare const axAIGoogleGeminiDefaultConfig: () => AxAIGoogleGeminiConfig;
10
- export declare const axAIGoogleGeminiDefaultCreativeConfig: () => AxAIGoogleGeminiConfig;
11
- export interface AxAIGoogleGeminiArgs {
12
- name: 'google-gemini';
13
- apiKey: string;
14
- projectId?: string;
15
- region?: string;
16
- config?: Readonly<AxAIGoogleGeminiConfig>;
17
- options?: Readonly<AxAIServiceOptions & {
18
- codeExecution?: boolean;
19
- }>;
20
- }
21
- /**
22
- * AxAIGoogleGemini: AI Service
23
- * @export
24
- */
25
- export declare class AxAIGoogleGemini extends AxBaseAI<AxAIGoogleGeminiChatRequest, AxAIGoogleGeminiBatchEmbedRequest, AxAIGoogleGeminiChatResponse, AxAIGoogleGeminiChatResponseDelta, AxAIGoogleGeminiBatchEmbedResponse> {
26
- private options?;
27
- private config;
28
- private apiKey;
29
- constructor({ apiKey, projectId, region, config, options }: Readonly<Omit<AxAIGoogleGeminiArgs, 'name'>>);
30
- getModelConfig(): AxModelConfig;
31
- generateChatReq: (req: Readonly<AxChatRequest>) => [API, AxAIGoogleGeminiChatRequest];
32
- generateEmbedReq: (req: Readonly<AxEmbedRequest>) => [API, AxAIGoogleGeminiBatchEmbedRequest];
33
- generateChatResp: (resp: Readonly<AxAIGoogleGeminiChatResponse>) => AxChatResponse;
34
- generateChatStreamResp: (resp: Readonly<AxAIGoogleGeminiChatResponseDelta>) => AxChatResponse;
35
- generateEmbedResp: (resp: Readonly<AxAIGoogleGeminiBatchEmbedResponse>) => AxEmbedResponse;
36
- }