@ax-llm/ax 9.0.0 → 9.0.3

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 (287) hide show
  1. package/README.md +106 -99
  2. package/build/module/docs/assets/navigation.js +1 -1
  3. package/build/module/docs/assets/navigation.js.map +1 -1
  4. package/build/module/docs/assets/search.js +1 -1
  5. package/build/module/docs/assets/search.js.map +1 -1
  6. package/build/module/src/ai/anthropic/api.d.ts +13 -19
  7. package/build/module/src/ai/anthropic/api.js +165 -72
  8. package/build/module/src/ai/anthropic/api.js.map +1 -1
  9. package/build/module/src/ai/anthropic/info.d.ts +2 -2
  10. package/build/module/src/ai/anthropic/info.js +7 -7
  11. package/build/module/src/ai/anthropic/info.js.map +1 -1
  12. package/build/module/src/ai/anthropic/types.d.ts +63 -52
  13. package/build/module/src/ai/anthropic/types.js +8 -12
  14. package/build/module/src/ai/anthropic/types.js.map +1 -1
  15. package/build/module/src/ai/azure-openai/api.d.ts +12 -28
  16. package/build/module/src/ai/azure-openai/api.js +6 -22
  17. package/build/module/src/ai/azure-openai/api.js.map +1 -1
  18. package/build/module/src/ai/balance.d.ts +9 -12
  19. package/build/module/src/ai/balance.js +1 -1
  20. package/build/module/src/ai/balance.js.map +1 -1
  21. package/build/module/src/ai/base.d.ts +26 -29
  22. package/build/module/src/ai/base.js +37 -55
  23. package/build/module/src/ai/base.js.map +1 -1
  24. package/build/module/src/ai/cohere/api.d.ts +16 -30
  25. package/build/module/src/ai/cohere/api.js +26 -32
  26. package/build/module/src/ai/cohere/api.js.map +1 -1
  27. package/build/module/src/ai/cohere/info.d.ts +2 -2
  28. package/build/module/src/ai/cohere/info.js +10 -10
  29. package/build/module/src/ai/cohere/info.js.map +1 -1
  30. package/build/module/src/ai/cohere/types.d.ts +13 -13
  31. package/build/module/src/ai/cohere/types.js +14 -14
  32. package/build/module/src/ai/cohere/types.js.map +1 -1
  33. package/build/module/src/ai/deepseek/api.d.ts +10 -22
  34. package/build/module/src/ai/deepseek/api.js +14 -24
  35. package/build/module/src/ai/deepseek/api.js.map +1 -1
  36. package/build/module/src/ai/deepseek/info.d.ts +2 -2
  37. package/build/module/src/ai/deepseek/info.js +4 -4
  38. package/build/module/src/ai/deepseek/info.js.map +1 -1
  39. package/build/module/src/ai/deepseek/types.d.ts +1 -1
  40. package/build/module/src/ai/deepseek/types.js +5 -5
  41. package/build/module/src/ai/deepseek/types.js.map +1 -1
  42. package/build/module/src/ai/google-gemini/api.d.ts +18 -20
  43. package/build/module/src/ai/google-gemini/api.js +93 -67
  44. package/build/module/src/ai/google-gemini/api.js.map +1 -1
  45. package/build/module/src/ai/google-gemini/info.d.ts +3 -3
  46. package/build/module/src/ai/google-gemini/info.js +6 -6
  47. package/build/module/src/ai/google-gemini/info.js.map +1 -1
  48. package/build/module/src/ai/google-gemini/types.d.ts +41 -48
  49. package/build/module/src/ai/google-gemini/types.js +25 -37
  50. package/build/module/src/ai/google-gemini/types.js.map +1 -1
  51. package/build/module/src/ai/groq/api.d.ts +9 -18
  52. package/build/module/src/ai/groq/api.js +11 -17
  53. package/build/module/src/ai/groq/api.js.map +1 -1
  54. package/build/module/src/ai/groq/types.d.ts +6 -0
  55. package/build/module/src/ai/groq/types.js +8 -0
  56. package/build/module/src/ai/groq/types.js.map +1 -0
  57. package/build/module/src/ai/huggingface/api.d.ts +13 -27
  58. package/build/module/src/ai/huggingface/api.js +12 -24
  59. package/build/module/src/ai/huggingface/api.js.map +1 -1
  60. package/build/module/src/ai/huggingface/info.d.ts +2 -2
  61. package/build/module/src/ai/huggingface/info.js +1 -1
  62. package/build/module/src/ai/huggingface/info.js.map +1 -1
  63. package/build/module/src/ai/huggingface/types.d.ts +7 -15
  64. package/build/module/src/ai/huggingface/types.js +4 -8
  65. package/build/module/src/ai/huggingface/types.js.map +1 -1
  66. package/build/module/src/ai/index.d.ts +13 -13
  67. package/build/module/src/ai/index.js +23 -23
  68. package/build/module/src/ai/index.js.map +1 -1
  69. package/build/module/src/ai/mistral/api.d.ts +10 -22
  70. package/build/module/src/ai/mistral/api.js +14 -24
  71. package/build/module/src/ai/mistral/api.js.map +1 -1
  72. package/build/module/src/ai/mistral/info.d.ts +2 -2
  73. package/build/module/src/ai/mistral/info.js +7 -7
  74. package/build/module/src/ai/mistral/info.js.map +1 -1
  75. package/build/module/src/ai/mistral/types.d.ts +2 -10
  76. package/build/module/src/ai/mistral/types.js +12 -20
  77. package/build/module/src/ai/mistral/types.js.map +1 -1
  78. package/build/module/src/ai/ollama/api.d.ts +13 -16
  79. package/build/module/src/ai/ollama/api.js +17 -13
  80. package/build/module/src/ai/ollama/api.js.map +1 -1
  81. package/build/module/src/ai/openai/api.d.ts +19 -40
  82. package/build/module/src/ai/openai/api.js +90 -135
  83. package/build/module/src/ai/openai/api.js.map +1 -1
  84. package/build/module/src/ai/openai/info.d.ts +2 -2
  85. package/build/module/src/ai/openai/info.js +9 -9
  86. package/build/module/src/ai/openai/info.js.map +1 -1
  87. package/build/module/src/ai/openai/types.d.ts +29 -56
  88. package/build/module/src/ai/openai/types.js +17 -34
  89. package/build/module/src/ai/openai/types.js.map +1 -1
  90. package/build/module/src/ai/together/api.d.ts +9 -17
  91. package/build/module/src/ai/together/api.js +9 -15
  92. package/build/module/src/ai/together/api.js.map +1 -1
  93. package/build/module/src/ai/together/info.d.ts +2 -2
  94. package/build/module/src/ai/together/info.js +1 -1
  95. package/build/module/src/ai/together/info.js.map +1 -1
  96. package/build/module/src/ai/types.d.ts +112 -25
  97. package/build/module/src/ai/util.d.ts +4 -4
  98. package/build/module/src/ai/util.js +6 -3
  99. package/build/module/src/ai/util.js.map +1 -1
  100. package/build/module/src/db/base.d.ts +14 -15
  101. package/build/module/src/db/base.js +26 -29
  102. package/build/module/src/db/base.js.map +1 -1
  103. package/build/module/src/db/cloudflare.d.ts +10 -9
  104. package/build/module/src/db/cloudflare.js +2 -2
  105. package/build/module/src/db/cloudflare.js.map +1 -1
  106. package/build/module/src/db/index.d.ts +6 -7
  107. package/build/module/src/db/index.js +9 -9
  108. package/build/module/src/db/index.js.map +1 -1
  109. package/build/module/src/db/memory.d.ts +13 -11
  110. package/build/module/src/db/memory.js +2 -2
  111. package/build/module/src/db/memory.js.map +1 -1
  112. package/build/module/src/db/pinecone.d.ts +10 -9
  113. package/build/module/src/db/pinecone.js +2 -2
  114. package/build/module/src/db/pinecone.js.map +1 -1
  115. package/build/module/src/db/types.d.ts +9 -9
  116. package/build/module/src/db/weaviate.d.ts +10 -9
  117. package/build/module/src/db/weaviate.js +2 -2
  118. package/build/module/src/db/weaviate.js.map +1 -1
  119. package/build/module/src/docs/manager.d.ts +16 -16
  120. package/build/module/src/docs/manager.js +1 -1
  121. package/build/module/src/docs/manager.js.map +1 -1
  122. package/build/module/src/docs/reranker.d.ts +6 -6
  123. package/build/module/src/docs/reranker.js +3 -3
  124. package/build/module/src/docs/reranker.js.map +1 -1
  125. package/build/module/src/docs/rewriter.d.ts +3 -3
  126. package/build/module/src/docs/rewriter.js +2 -2
  127. package/build/module/src/docs/rewriter.js.map +1 -1
  128. package/build/module/src/docs/tika.d.ts +5 -8
  129. package/build/module/src/docs/tika.js +2 -2
  130. package/build/module/src/docs/tika.js.map +1 -1
  131. package/build/module/src/dsp/asserts.d.ts +8 -8
  132. package/build/module/src/dsp/asserts.js +9 -6
  133. package/build/module/src/dsp/asserts.js.map +1 -1
  134. package/build/module/src/dsp/eval.d.ts +9 -28
  135. package/build/module/src/dsp/eval.js +10 -5
  136. package/build/module/src/dsp/eval.js.map +1 -1
  137. package/build/module/src/dsp/evaluate.d.ts +8 -8
  138. package/build/module/src/dsp/evaluate.js +1 -1
  139. package/build/module/src/dsp/evaluate.js.map +1 -1
  140. package/build/module/src/dsp/extract.d.ts +6 -6
  141. package/build/module/src/dsp/extract.js +1 -0
  142. package/build/module/src/dsp/extract.js.map +1 -1
  143. package/build/module/src/dsp/generate.d.ts +24 -25
  144. package/build/module/src/dsp/generate.js +24 -23
  145. package/build/module/src/dsp/generate.js.map +1 -1
  146. package/build/module/src/dsp/generate.test.js +4 -4
  147. package/build/module/src/dsp/generate.test.js.map +1 -1
  148. package/build/module/src/dsp/index.d.ts +4 -4
  149. package/build/module/src/dsp/loader.d.ts +5 -5
  150. package/build/module/src/dsp/loader.js +1 -1
  151. package/build/module/src/dsp/loader.js.map +1 -1
  152. package/build/module/src/dsp/optimize.d.ts +12 -12
  153. package/build/module/src/dsp/optimize.js +5 -5
  154. package/build/module/src/dsp/optimize.js.map +1 -1
  155. package/build/module/src/dsp/parser.d.ts +2 -1
  156. package/build/module/src/dsp/parser.js +178 -73
  157. package/build/module/src/dsp/parser.js.map +1 -1
  158. package/build/module/src/dsp/program.d.ts +38 -37
  159. package/build/module/src/dsp/program.js +4 -53
  160. package/build/module/src/dsp/program.js.map +1 -1
  161. package/build/module/src/dsp/prompt.d.ts +36 -13
  162. package/build/module/src/dsp/prompt.js +123 -51
  163. package/build/module/src/dsp/prompt.js.map +1 -1
  164. package/build/module/src/dsp/registry.d.ts +1 -1
  165. package/build/module/src/dsp/registry.js +1 -1
  166. package/build/module/src/dsp/registry.js.map +1 -1
  167. package/build/module/src/dsp/router.d.ts +7 -7
  168. package/build/module/src/dsp/router.js +4 -4
  169. package/build/module/src/dsp/router.js.map +1 -1
  170. package/build/module/src/dsp/sig.d.ts +13 -13
  171. package/build/module/src/dsp/sig.js +7 -3
  172. package/build/module/src/dsp/sig.js.map +1 -1
  173. package/build/module/src/dsp/sig.test.js +4 -1
  174. package/build/module/src/dsp/sig.test.js.map +1 -1
  175. package/build/module/src/dsp/stopwords.js +1 -0
  176. package/build/module/src/dsp/stopwords.js.map +1 -1
  177. package/build/module/src/dsp/strutil.d.ts +10 -8
  178. package/build/module/src/dsp/strutil.js +14 -6
  179. package/build/module/src/dsp/strutil.js.map +1 -1
  180. package/build/module/src/dsp/util.d.ts +4 -0
  181. package/build/module/src/dsp/util.js +62 -0
  182. package/build/module/src/dsp/util.js.map +1 -1
  183. package/build/module/src/examples/agent.js +7 -5
  184. package/build/module/src/examples/agent.js.map +1 -1
  185. package/build/module/src/examples/chain-of-thought.js +5 -3
  186. package/build/module/src/examples/chain-of-thought.js.map +1 -1
  187. package/build/module/src/examples/customer-support.js +6 -4
  188. package/build/module/src/examples/customer-support.js.map +1 -1
  189. package/build/module/src/examples/fibonacci.js +6 -4
  190. package/build/module/src/examples/fibonacci.js.map +1 -1
  191. package/build/module/src/examples/food-search.js +8 -8
  192. package/build/module/src/examples/food-search.js.map +1 -1
  193. package/build/module/src/examples/marketing.js +5 -3
  194. package/build/module/src/examples/marketing.js.map +1 -1
  195. package/build/module/src/examples/multi-modal.d.ts +1 -0
  196. package/build/module/src/examples/multi-modal.js +16 -0
  197. package/build/module/src/examples/multi-modal.js.map +1 -0
  198. package/build/module/src/examples/qna-tune.js +9 -7
  199. package/build/module/src/examples/qna-tune.js.map +1 -1
  200. package/build/module/src/examples/qna-use-tuned.js +8 -6
  201. package/build/module/src/examples/qna-use-tuned.js.map +1 -1
  202. package/build/module/src/examples/rag-docs.js +7 -5
  203. package/build/module/src/examples/rag-docs.js.map +1 -1
  204. package/build/module/src/examples/rag.js +6 -4
  205. package/build/module/src/examples/rag.js.map +1 -1
  206. package/build/module/src/examples/react.js +5 -3
  207. package/build/module/src/examples/react.js.map +1 -1
  208. package/build/module/src/examples/routing.js +9 -7
  209. package/build/module/src/examples/routing.js.map +1 -1
  210. package/build/module/src/examples/smart-home.d.ts +10 -0
  211. package/build/module/src/examples/smart-home.js +106 -0
  212. package/build/module/src/examples/smart-home.js.map +1 -0
  213. package/build/module/src/examples/streaming1.js +7 -5
  214. package/build/module/src/examples/streaming1.js.map +1 -1
  215. package/build/module/src/examples/streaming2.js +9 -7
  216. package/build/module/src/examples/streaming2.js.map +1 -1
  217. package/build/module/src/examples/summarize.js +5 -5
  218. package/build/module/src/examples/summarize.js.map +1 -1
  219. package/build/module/src/examples/vectordb.js +6 -4
  220. package/build/module/src/examples/vectordb.js.map +1 -1
  221. package/build/module/src/funcs/code.d.ts +3 -4
  222. package/build/module/src/funcs/code.js +15 -15
  223. package/build/module/src/funcs/code.js.map +1 -1
  224. package/build/module/src/funcs/embed.d.ts +3 -5
  225. package/build/module/src/funcs/embed.js +2 -2
  226. package/build/module/src/funcs/embed.js.map +1 -1
  227. package/build/module/src/funcs/functions.d.ts +18 -0
  228. package/build/module/src/funcs/functions.js +55 -0
  229. package/build/module/src/funcs/functions.js.map +1 -0
  230. package/build/module/src/funcs/index.d.ts +1 -0
  231. package/build/module/src/funcs/index.js +1 -0
  232. package/build/module/src/funcs/index.js.map +1 -1
  233. package/build/module/src/funcs/jsonschema.d.ts +2 -0
  234. package/build/module/src/funcs/jsonschema.js +97 -0
  235. package/build/module/src/funcs/jsonschema.js.map +1 -0
  236. package/build/module/src/index.d.ts +1 -2
  237. package/build/module/src/index.js +1 -2
  238. package/build/module/src/index.js.map +1 -1
  239. package/build/module/src/mem/index.d.ts +2 -0
  240. package/build/module/src/mem/index.js +3 -0
  241. package/build/module/src/mem/index.js.map +1 -0
  242. package/build/module/src/mem/memory.d.ts +16 -0
  243. package/build/module/src/mem/memory.js +74 -0
  244. package/build/module/src/mem/memory.js.map +1 -0
  245. package/build/module/src/mem/types.d.ts +10 -0
  246. package/build/module/src/mem/types.js +2 -0
  247. package/build/module/src/mem/types.js.map +1 -0
  248. package/build/module/src/prompts/agent.d.ts +14 -15
  249. package/build/module/src/prompts/agent.js +10 -11
  250. package/build/module/src/prompts/agent.js.map +1 -1
  251. package/build/module/src/prompts/cot.d.ts +6 -6
  252. package/build/module/src/prompts/cot.js +4 -4
  253. package/build/module/src/prompts/cot.js.map +1 -1
  254. package/build/module/src/prompts/prompts.test.js +9 -5
  255. package/build/module/src/prompts/prompts.test.js.map +1 -1
  256. package/build/module/src/prompts/rag.d.ts +7 -7
  257. package/build/module/src/prompts/rag.js +6 -6
  258. package/build/module/src/prompts/rag.js.map +1 -1
  259. package/build/module/src/prompts/react.d.ts +7 -7
  260. package/build/module/src/prompts/react.js +4 -4
  261. package/build/module/src/prompts/react.js.map +1 -1
  262. package/build/module/src/text/functions.d.ts +3 -20
  263. package/build/module/src/text/functions.js.map +1 -1
  264. package/build/module/src/text/jsonschema.d.ts +1 -2
  265. package/build/module/src/text/memory.d.ts +8 -8
  266. package/build/module/src/text/memory.js.map +1 -1
  267. package/build/module/src/text/types.d.ts +14 -14
  268. package/build/module/src/trace/index.d.ts +50 -50
  269. package/build/module/src/trace/index.js +30 -30
  270. package/build/module/src/trace/index.js.map +1 -1
  271. package/build/module/src/types/index.d.ts +9 -39
  272. package/build/module/src/types/index.js +1 -1
  273. package/build/module/src/types/index.js.map +1 -1
  274. package/build/module/src/util/apicall.d.ts +2 -5
  275. package/build/module/src/util/apicall.js +4 -3
  276. package/build/module/src/util/apicall.js.map +1 -1
  277. package/build/module/src/util/log.js +1 -0
  278. package/build/module/src/util/log.js.map +1 -1
  279. package/build/module/src/util/other.js +1 -0
  280. package/build/module/src/util/other.js.map +1 -1
  281. package/build/module/src/util/stream.d.ts +0 -1
  282. package/build/module/src/util/stream.js +1 -0
  283. package/build/module/src/util/stream.js.map +1 -1
  284. package/build/module/src/util/transform.d.ts +0 -1
  285. package/build/module/src/util/transform.js +2 -0
  286. package/build/module/src/util/transform.js.map +1 -1
  287. package/package.json +2 -4
@@ -1,47 +1,40 @@
1
- /**
2
- * apiURLGoogleGemini: Base URL for Google Gemini API calls
3
- * @export
4
- */
5
- import type { TextModelConfig } from '../types.js';
6
- /**
7
- * GoogleGeminiModel: Enum for specifying the model version
8
- * @export
9
- */
10
- export declare enum GoogleGeminiModel {
1
+ import type { AxModelConfig } from '../types.js';
2
+ export declare enum AxGoogleGeminiModel {
11
3
  Gemini1Pro = "gemini-1.0-pro",
12
4
  Gemini15Flash = "gemini-1.5-flash",
13
5
  Gemini15Pro = "gemini-1.5-pro"
14
6
  }
15
- /**
16
- * GoogleGeminiEmbedModels: Enum for specifying embedding models
17
- * @export
18
- */
19
- export declare enum GoogleGeminiEmbedModels {
7
+ export declare enum AxGoogleGeminiEmbedModels {
20
8
  Embedding001 = "embedding-001"
21
9
  }
22
- export declare enum GoogleGeminiSafetyCategory {
10
+ export declare enum AxGoogleGeminiSafetyCategory {
23
11
  HarmCategoryHarassment = "HARM_CATEGORY_HARASSMENT",
24
12
  HarmCategoryHateSpeech = "HARM_CATEGORY_HATE_SPEECH",
25
13
  HarmCategorySexuallyExplicit = "HARM_CATEGORY_SEXUALLY_EXPLICIT",
26
14
  HarmCategoryDangerousContent = "HARM_CATEGORY_DANGEROUS_CONTENT"
27
15
  }
28
- export declare enum GoogleGeminiSafetyThreshold {
16
+ export declare enum AxGoogleGeminiSafetyThreshold {
29
17
  BlockNone = "BLOCK_NONE",
30
18
  BlockOnlyHigh = "BLOCK_ONLY_HIGH",
31
19
  BlockMediumAndAbove = "BLOCK_MEDIUM_AND_ABOVE",
32
20
  BlockLowAndAbove = "BLOCK_LOW_AND_ABOVE",
33
21
  BlockDefault = "HARM_BLOCK_THRESHOLD_UNSPECIFIED"
34
22
  }
35
- export type GoogleGeminiContent = {
23
+ export type AxGoogleGeminiContent = {
36
24
  role: 'user';
37
- parts: {
25
+ parts: ({
38
26
  text: string;
39
- }[] | {
27
+ } | {
28
+ inlineData: {
29
+ mimeType: string;
30
+ data: string;
31
+ };
32
+ } | {
40
33
  fileData: {
41
34
  mimeType: string;
42
35
  fileUri: string;
43
36
  };
44
- }[];
37
+ })[];
45
38
  } | {
46
39
  role: 'model';
47
40
  parts: {
@@ -61,21 +54,21 @@ export type GoogleGeminiContent = {
61
54
  };
62
55
  }[];
63
56
  };
64
- export type GoogleGeminiToolFunctionDeclaration = {
57
+ export type AxGoogleGeminiToolFunctionDeclaration = {
65
58
  name: string;
66
59
  description?: string;
67
60
  parameters?: object;
68
61
  };
69
- export type GoogleGeminiTool = {
70
- functionDeclarations: GoogleGeminiToolFunctionDeclaration[];
62
+ export type AxGoogleGeminiTool = {
63
+ functionDeclarations: AxGoogleGeminiToolFunctionDeclaration[];
71
64
  };
72
- export type GoogleGeminiToolConfig = {
65
+ export type AxGoogleGeminiToolConfig = {
73
66
  function_calling_config: {
74
67
  mode: 'ANY' | 'NONE' | 'AUTO';
75
68
  allowed_function_names?: string[];
76
69
  };
77
70
  };
78
- export type GoogleGeminiGenerationConfig = {
71
+ export type AxGoogleGeminiGenerationConfig = {
79
72
  temperature?: number;
80
73
  topP?: number;
81
74
  topK?: number;
@@ -83,21 +76,21 @@ export type GoogleGeminiGenerationConfig = {
83
76
  maxOutputTokens?: number;
84
77
  stopSequences?: readonly string[];
85
78
  };
86
- export type GoogleGeminiSafetySettings = {
87
- category: GoogleGeminiSafetyCategory;
88
- threshold: GoogleGeminiSafetyThreshold;
79
+ export type AxGoogleGeminiSafetySettings = {
80
+ category: AxGoogleGeminiSafetyCategory;
81
+ threshold: AxGoogleGeminiSafetyThreshold;
89
82
  }[];
90
- export type GoogleGeminiChatRequest = {
91
- contents: GoogleGeminiContent[];
92
- tools?: GoogleGeminiTool[];
93
- tool_config?: GoogleGeminiToolConfig;
94
- systemInstruction?: GoogleGeminiContent;
95
- generationConfig: GoogleGeminiGenerationConfig;
96
- safetySettings?: GoogleGeminiSafetySettings;
83
+ export type AxGoogleGeminiChatRequest = {
84
+ contents: AxGoogleGeminiContent[];
85
+ tools?: AxGoogleGeminiTool[];
86
+ tool_config?: AxGoogleGeminiToolConfig;
87
+ systemInstruction?: AxGoogleGeminiContent;
88
+ generationConfig: AxGoogleGeminiGenerationConfig;
89
+ safetySettings?: AxGoogleGeminiSafetySettings;
97
90
  };
98
- export type GoogleGeminiChatResponse = {
91
+ export type AxGoogleGeminiChatResponse = {
99
92
  candidates: {
100
- content: GoogleGeminiContent;
93
+ content: AxGoogleGeminiContent;
101
94
  finishReason: 'STOP' | 'MAX_TOKENS' | 'SAFETY' | 'RECITATION' | 'OTHER';
102
95
  citationMetadata: {
103
96
  citations: {
@@ -120,31 +113,31 @@ export type GoogleGeminiChatResponse = {
120
113
  totalTokenCount: number;
121
114
  };
122
115
  };
123
- export type GoogleGeminiChatResponseDelta = GoogleGeminiChatResponse;
116
+ export type AxGoogleGeminiChatResponseDelta = AxGoogleGeminiChatResponse;
124
117
  /**
125
- * GoogleGeminiConfig: Configuration options for Google Gemini API
118
+ * AxGoogleGeminiConfig: Configuration options for Google Gemini API
126
119
  * @export
127
120
  */
128
- export type GoogleGeminiConfig = TextModelConfig & {
129
- model: GoogleGeminiModel | string;
130
- embedModel: GoogleGeminiEmbedModels;
131
- safetySettings?: GoogleGeminiSafetySettings;
121
+ export type AxGoogleGeminiConfig = AxModelConfig & {
122
+ model: AxGoogleGeminiModel | string;
123
+ embedModel: AxGoogleGeminiEmbedModels;
124
+ safetySettings?: AxGoogleGeminiSafetySettings;
132
125
  };
133
126
  /**
134
- * GoogleGeminiEmbedRequest: Structure for making an embedding request to the Google Gemini API.
127
+ * AxGoogleGeminiEmbedRequest: Structure for making an embedding request to the Google Gemini API.
135
128
  * @export
136
129
  */
137
- export type GoogleGeminiBatchEmbedRequest = {
130
+ export type AxGoogleGeminiBatchEmbedRequest = {
138
131
  requests: {
139
132
  model: string;
140
133
  text: string;
141
134
  }[];
142
135
  };
143
136
  /**
144
- * GoogleGeminiEmbedResponse: Structure for handling responses from the Google Gemini API embedding requests.
137
+ * AxGoogleGeminiEmbedResponse: Structure for handling responses from the Google Gemini API embedding requests.
145
138
  * @export
146
139
  */
147
- export type GoogleGeminiBatchEmbedResponse = {
140
+ export type AxGoogleGeminiBatchEmbedResponse = {
148
141
  embeddings: {
149
142
  value: number[];
150
143
  }[];
@@ -1,38 +1,26 @@
1
- /**
2
- * apiURLGoogleGemini: Base URL for Google Gemini API calls
3
- * @export
4
- */
5
- /**
6
- * GoogleGeminiModel: Enum for specifying the model version
7
- * @export
8
- */
9
- export var GoogleGeminiModel;
10
- (function (GoogleGeminiModel) {
11
- GoogleGeminiModel["Gemini1Pro"] = "gemini-1.0-pro";
12
- GoogleGeminiModel["Gemini15Flash"] = "gemini-1.5-flash";
13
- GoogleGeminiModel["Gemini15Pro"] = "gemini-1.5-pro";
14
- })(GoogleGeminiModel || (GoogleGeminiModel = {}));
15
- /**
16
- * GoogleGeminiEmbedModels: Enum for specifying embedding models
17
- * @export
18
- */
19
- export var GoogleGeminiEmbedModels;
20
- (function (GoogleGeminiEmbedModels) {
21
- GoogleGeminiEmbedModels["Embedding001"] = "embedding-001";
22
- })(GoogleGeminiEmbedModels || (GoogleGeminiEmbedModels = {}));
23
- export var GoogleGeminiSafetyCategory;
24
- (function (GoogleGeminiSafetyCategory) {
25
- GoogleGeminiSafetyCategory["HarmCategoryHarassment"] = "HARM_CATEGORY_HARASSMENT";
26
- GoogleGeminiSafetyCategory["HarmCategoryHateSpeech"] = "HARM_CATEGORY_HATE_SPEECH";
27
- GoogleGeminiSafetyCategory["HarmCategorySexuallyExplicit"] = "HARM_CATEGORY_SEXUALLY_EXPLICIT";
28
- GoogleGeminiSafetyCategory["HarmCategoryDangerousContent"] = "HARM_CATEGORY_DANGEROUS_CONTENT";
29
- })(GoogleGeminiSafetyCategory || (GoogleGeminiSafetyCategory = {}));
30
- export var GoogleGeminiSafetyThreshold;
31
- (function (GoogleGeminiSafetyThreshold) {
32
- GoogleGeminiSafetyThreshold["BlockNone"] = "BLOCK_NONE";
33
- GoogleGeminiSafetyThreshold["BlockOnlyHigh"] = "BLOCK_ONLY_HIGH";
34
- GoogleGeminiSafetyThreshold["BlockMediumAndAbove"] = "BLOCK_MEDIUM_AND_ABOVE";
35
- GoogleGeminiSafetyThreshold["BlockLowAndAbove"] = "BLOCK_LOW_AND_ABOVE";
36
- GoogleGeminiSafetyThreshold["BlockDefault"] = "HARM_BLOCK_THRESHOLD_UNSPECIFIED";
37
- })(GoogleGeminiSafetyThreshold || (GoogleGeminiSafetyThreshold = {}));
1
+ export var AxGoogleGeminiModel;
2
+ (function (AxGoogleGeminiModel) {
3
+ AxGoogleGeminiModel["Gemini1Pro"] = "gemini-1.0-pro";
4
+ AxGoogleGeminiModel["Gemini15Flash"] = "gemini-1.5-flash";
5
+ AxGoogleGeminiModel["Gemini15Pro"] = "gemini-1.5-pro";
6
+ })(AxGoogleGeminiModel || (AxGoogleGeminiModel = {}));
7
+ export var AxGoogleGeminiEmbedModels;
8
+ (function (AxGoogleGeminiEmbedModels) {
9
+ AxGoogleGeminiEmbedModels["Embedding001"] = "embedding-001";
10
+ })(AxGoogleGeminiEmbedModels || (AxGoogleGeminiEmbedModels = {}));
11
+ export var AxGoogleGeminiSafetyCategory;
12
+ (function (AxGoogleGeminiSafetyCategory) {
13
+ AxGoogleGeminiSafetyCategory["HarmCategoryHarassment"] = "HARM_CATEGORY_HARASSMENT";
14
+ AxGoogleGeminiSafetyCategory["HarmCategoryHateSpeech"] = "HARM_CATEGORY_HATE_SPEECH";
15
+ AxGoogleGeminiSafetyCategory["HarmCategorySexuallyExplicit"] = "HARM_CATEGORY_SEXUALLY_EXPLICIT";
16
+ AxGoogleGeminiSafetyCategory["HarmCategoryDangerousContent"] = "HARM_CATEGORY_DANGEROUS_CONTENT";
17
+ })(AxGoogleGeminiSafetyCategory || (AxGoogleGeminiSafetyCategory = {}));
18
+ export var AxGoogleGeminiSafetyThreshold;
19
+ (function (AxGoogleGeminiSafetyThreshold) {
20
+ AxGoogleGeminiSafetyThreshold["BlockNone"] = "BLOCK_NONE";
21
+ AxGoogleGeminiSafetyThreshold["BlockOnlyHigh"] = "BLOCK_ONLY_HIGH";
22
+ AxGoogleGeminiSafetyThreshold["BlockMediumAndAbove"] = "BLOCK_MEDIUM_AND_ABOVE";
23
+ AxGoogleGeminiSafetyThreshold["BlockLowAndAbove"] = "BLOCK_LOW_AND_ABOVE";
24
+ AxGoogleGeminiSafetyThreshold["BlockDefault"] = "HARM_BLOCK_THRESHOLD_UNSPECIFIED";
25
+ })(AxGoogleGeminiSafetyThreshold || (AxGoogleGeminiSafetyThreshold = {}));
38
26
  //# sourceMappingURL=types.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../../src/ai/google-gemini/types.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAIH;;;GAGG;AACH,MAAM,CAAN,IAAY,iBAIX;AAJD,WAAY,iBAAiB;IAC3B,kDAA6B,CAAA;IAC7B,uDAAkC,CAAA;IAClC,mDAA8B,CAAA;AAChC,CAAC,EAJW,iBAAiB,KAAjB,iBAAiB,QAI5B;AAED;;;GAGG;AACH,MAAM,CAAN,IAAY,uBAEX;AAFD,WAAY,uBAAuB;IACjC,yDAA8B,CAAA;AAChC,CAAC,EAFW,uBAAuB,KAAvB,uBAAuB,QAElC;AAED,MAAM,CAAN,IAAY,0BAKX;AALD,WAAY,0BAA0B;IACpC,iFAAmD,CAAA;IACnD,kFAAoD,CAAA;IACpD,8FAAgE,CAAA;IAChE,8FAAgE,CAAA;AAClE,CAAC,EALW,0BAA0B,KAA1B,0BAA0B,QAKrC;AAED,MAAM,CAAN,IAAY,2BAMX;AAND,WAAY,2BAA2B;IACrC,uDAAwB,CAAA;IACxB,gEAAiC,CAAA;IACjC,6EAA8C,CAAA;IAC9C,uEAAwC,CAAA;IACxC,gFAAiD,CAAA;AACnD,CAAC,EANW,2BAA2B,KAA3B,2BAA2B,QAMtC"}
1
+ {"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../../src/ai/google-gemini/types.ts"],"names":[],"mappings":"AAEA,MAAM,CAAN,IAAY,mBAIX;AAJD,WAAY,mBAAmB;IAC7B,oDAA6B,CAAA;IAC7B,yDAAkC,CAAA;IAClC,qDAA8B,CAAA;AAChC,CAAC,EAJW,mBAAmB,KAAnB,mBAAmB,QAI9B;AAED,MAAM,CAAN,IAAY,yBAEX;AAFD,WAAY,yBAAyB;IACnC,2DAA8B,CAAA;AAChC,CAAC,EAFW,yBAAyB,KAAzB,yBAAyB,QAEpC;AAED,MAAM,CAAN,IAAY,4BAKX;AALD,WAAY,4BAA4B;IACtC,mFAAmD,CAAA;IACnD,oFAAoD,CAAA;IACpD,gGAAgE,CAAA;IAChE,gGAAgE,CAAA;AAClE,CAAC,EALW,4BAA4B,KAA5B,4BAA4B,QAKvC;AAED,MAAM,CAAN,IAAY,6BAMX;AAND,WAAY,6BAA6B;IACvC,yDAAwB,CAAA;IACxB,kEAAiC,CAAA;IACjC,+EAA8C,CAAA;IAC9C,yEAAwC,CAAA;IACxC,kFAAiD,CAAA;AACnD,CAAC,EANW,6BAA6B,KAA7B,6BAA6B,QAMxC"}
@@ -1,22 +1,13 @@
1
- import type { AIServiceOptions } from '../../text/types.js';
2
- import { OpenAI } from '../openai/api.js';
3
- import type { OpenAIConfig } from '../openai/types.js';
4
- type GroqAIConfig = OpenAIConfig;
5
- /**
6
- * GroqAI: Default Model options for text generation
7
- * @export
8
- */
9
- export declare const GroqDefaultConfig: () => GroqAIConfig;
10
- export interface GroqArgs {
1
+ import { AxOpenAI } from '../openai/api.js';
2
+ import type { AxOpenAIConfig } from '../openai/types.js';
3
+ import type { AxAIServiceOptions } from '../types.js';
4
+ type AxAxGroqAIConfig = AxOpenAIConfig;
5
+ export interface AxAxGroqArgs {
11
6
  apiKey: string;
12
- config: Readonly<GroqAIConfig>;
13
- options?: Readonly<AIServiceOptions>;
7
+ config: Readonly<AxAxGroqAIConfig>;
8
+ options?: Readonly<AxAIServiceOptions>;
14
9
  }
15
- /**
16
- * GroqAI: AI Service
17
- * @export
18
- */
19
- export declare class Groq extends OpenAI {
20
- constructor({ apiKey, config, options }: Readonly<GroqArgs>);
10
+ export declare class AxGroq extends AxOpenAI {
11
+ constructor({ apiKey, config, options }: Readonly<AxAxGroqArgs>);
21
12
  }
22
13
  export {};
@@ -1,19 +1,12 @@
1
- import { BaseAIDefaultConfig } from '../base.js';
2
- import { OpenAI } from '../openai/api.js';
3
- /**
4
- * GroqAI: Default Model options for text generation
5
- * @export
6
- */
7
- export const GroqDefaultConfig = () => structuredClone({
8
- model: 'llama3-70b-8192',
9
- ...BaseAIDefaultConfig()
1
+ import { axBaseAIDefaultConfig } from '../base.js';
2
+ import { AxOpenAI } from '../openai/api.js';
3
+ import { AxGroqModel } from './types.js';
4
+ const axGroqDefaultConfig = () => structuredClone({
5
+ model: AxGroqModel.Llama3_70B,
6
+ ...axBaseAIDefaultConfig()
10
7
  });
11
- /**
12
- * GroqAI: AI Service
13
- * @export
14
- */
15
- export class Groq extends OpenAI {
16
- constructor({ apiKey, config = GroqDefaultConfig(), options }) {
8
+ export class AxGroq extends AxOpenAI {
9
+ constructor({ apiKey, config = axGroqDefaultConfig(), options }) {
17
10
  if (!apiKey || apiKey === '') {
18
11
  throw new Error('Groq API key not set');
19
12
  }
@@ -21,9 +14,10 @@ export class Groq extends OpenAI {
21
14
  apiKey,
22
15
  config,
23
16
  options: { ...options, streamingUsage: false },
24
- apiURL: 'https://api.groq.com/openai/v1'
17
+ apiURL: 'https://api.groq.com/openai/v1',
18
+ modelInfo: []
25
19
  });
26
- super.setName('Groq');
20
+ super.setName('AxGroq');
27
21
  }
28
22
  }
29
23
  //# sourceMappingURL=api.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../../src/ai/groq/api.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AACjD,OAAO,EAAE,MAAM,EAAE,MAAM,kBAAkB,CAAC;AAK1C;;;GAGG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAG,GAAiB,EAAE,CAClD,eAAe,CAAC;IACd,KAAK,EAAE,iBAAiB;IACxB,GAAG,mBAAmB,EAAE;CACzB,CAAC,CAAC;AAQL;;;GAGG;AACH,MAAM,OAAO,IAAK,SAAQ,MAAM;IAC9B,YAAY,EACV,MAAM,EACN,MAAM,GAAG,iBAAiB,EAAE,EAC5B,OAAO,EACY;QACnB,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC1C,CAAC;QACD,KAAK,CAAC;YACJ,MAAM;YACN,MAAM;YACN,OAAO,EAAE,EAAE,GAAG,OAAO,EAAE,cAAc,EAAE,KAAK,EAAE;YAC9C,MAAM,EAAE,gCAAgC;SACzC,CAAC,CAAC;QAEH,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACxB,CAAC;CACF"}
1
+ {"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../../src/ai/groq/api.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,qBAAqB,EAAE,MAAM,YAAY,CAAC;AACnD,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAI5C,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAIzC,MAAM,mBAAmB,GAAG,GAAqB,EAAE,CACjD,eAAe,CAAC;IACd,KAAK,EAAE,WAAW,CAAC,UAAU;IAC7B,GAAG,qBAAqB,EAAE;CAC3B,CAAC,CAAC;AAQL,MAAM,OAAO,MAAO,SAAQ,QAAQ;IAClC,YAAY,EACV,MAAM,EACN,MAAM,GAAG,mBAAmB,EAAE,EAC9B,OAAO,EACgB;QACvB,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC1C,CAAC;QACD,KAAK,CAAC;YACJ,MAAM;YACN,MAAM;YACN,OAAO,EAAE,EAAE,GAAG,OAAO,EAAE,cAAc,EAAE,KAAK,EAAE;YAC9C,MAAM,EAAE,gCAAgC;YACxC,SAAS,EAAE,EAAE;SACd,CAAC,CAAC;QAEH,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAC1B,CAAC;CACF"}
@@ -0,0 +1,6 @@
1
+ export declare enum AxGroqModel {
2
+ Llama3_8B = "llama3-8b-8192",
3
+ Llama3_70B = "llama3-70b-8192",
4
+ Mixtral_8x7B = "mixtral-8x7b-32768",
5
+ Gemma_7B = "gemma-7b-it"
6
+ }
@@ -0,0 +1,8 @@
1
+ export var AxGroqModel;
2
+ (function (AxGroqModel) {
3
+ AxGroqModel["Llama3_8B"] = "llama3-8b-8192";
4
+ AxGroqModel["Llama3_70B"] = "llama3-70b-8192";
5
+ AxGroqModel["Mixtral_8x7B"] = "mixtral-8x7b-32768";
6
+ AxGroqModel["Gemma_7B"] = "gemma-7b-it";
7
+ })(AxGroqModel || (AxGroqModel = {}));
8
+ //# sourceMappingURL=types.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../../src/ai/groq/types.ts"],"names":[],"mappings":"AAAA,MAAM,CAAN,IAAY,WAKX;AALD,WAAY,WAAW;IACrB,2CAA4B,CAAA;IAC5B,6CAA8B,CAAA;IAC9B,kDAAmC,CAAA;IACnC,uCAAwB,CAAA;AAC1B,CAAC,EALW,WAAW,KAAX,WAAW,QAKtB"}
@@ -1,32 +1,18 @@
1
- import type { AIPromptConfig, AIServiceOptions } from '../../text/types.js';
2
- import type { AITextChatRequest } from '../../types/index.js';
3
1
  import type { API } from '../../util/apicall.js';
4
- import { BaseAI } from '../base.js';
5
- import type { TextModelConfig, TextResponse } from '../types.js';
6
- import { type HuggingFaceConfig, type HuggingFaceRequest, type HuggingFaceResponse } from './types.js';
7
- /**
8
- * HuggingFace: Default Model options for text generation
9
- * @export
10
- */
11
- export declare const HuggingFaceDefaultConfig: () => HuggingFaceConfig;
12
- /**
13
- * HuggingFace: Default model options for more creative text generation
14
- * @export
15
- */
16
- export declare const HuggingFaceCreativeConfig: () => HuggingFaceConfig;
17
- export interface HuggingFaceArgs {
2
+ import { AxBaseAI } from '../base.js';
3
+ import type { AxAIPromptConfig, AxAIServiceOptions, AxChatRequest, AxChatResponse, AxModelConfig } from '../types.js';
4
+ import { type AxHuggingFaceConfig, type AxHuggingFaceRequest, type AxHuggingFaceResponse } from './types.js';
5
+ export declare const axHuggingFaceDefaultConfig: () => AxHuggingFaceConfig;
6
+ export declare const axHuggingFaceCreativeConfig: () => AxHuggingFaceConfig;
7
+ export interface AxHuggingFaceArgs {
18
8
  apiKey: string;
19
- config?: Readonly<HuggingFaceConfig>;
20
- options?: Readonly<AIServiceOptions>;
9
+ config?: Readonly<AxHuggingFaceConfig>;
10
+ options?: Readonly<AxAIServiceOptions>;
21
11
  }
22
- /**
23
- * HuggingFace: AI Service
24
- * @export
25
- */
26
- export declare class HuggingFace extends BaseAI<HuggingFaceRequest, unknown, HuggingFaceResponse, unknown, unknown> {
12
+ export declare class AxHuggingFace extends AxBaseAI<AxHuggingFaceRequest, unknown, AxHuggingFaceResponse, unknown, unknown> {
27
13
  private config;
28
- constructor({ apiKey, config, options }: Readonly<HuggingFaceArgs>);
29
- getModelConfig(): TextModelConfig;
30
- generateChatReq: (req: Readonly<AITextChatRequest>, _config: Readonly<AIPromptConfig>) => [API, HuggingFaceRequest];
31
- generateChatResp: (resp: Readonly<HuggingFaceResponse>) => TextResponse;
14
+ constructor({ apiKey, config, options }: Readonly<AxHuggingFaceArgs>);
15
+ getModelConfig(): AxModelConfig;
16
+ generateChatReq: (req: Readonly<AxChatRequest>, _config: Readonly<AxAIPromptConfig>) => [API, AxHuggingFaceRequest];
17
+ generateChatResp: (resp: Readonly<AxHuggingFaceResponse>) => AxChatResponse;
32
18
  }
@@ -1,29 +1,17 @@
1
- import { BaseAI, BaseAIDefaultConfig, BaseAIDefaultCreativeConfig } from '../base.js';
2
- import { modelInfoHuggingFace } from './info.js';
3
- import { HuggingFaceModel } from './types.js';
4
- /**
5
- * HuggingFace: Default Model options for text generation
6
- * @export
7
- */
8
- export const HuggingFaceDefaultConfig = () => structuredClone({
9
- model: HuggingFaceModel.MetaLlama270BChatHF,
10
- ...BaseAIDefaultConfig()
1
+ import { AxBaseAI, axBaseAIDefaultConfig, axBaseAIDefaultCreativeConfig } from '../base.js';
2
+ import { axModelInfoHuggingFace } from './info.js';
3
+ import { AxHuggingFaceModel } from './types.js';
4
+ export const axHuggingFaceDefaultConfig = () => structuredClone({
5
+ model: AxHuggingFaceModel.MetaLlama270BChatHF,
6
+ ...axBaseAIDefaultConfig()
11
7
  });
12
- /**
13
- * HuggingFace: Default model options for more creative text generation
14
- * @export
15
- */
16
- export const HuggingFaceCreativeConfig = () => structuredClone({
17
- model: HuggingFaceModel.MetaLlama270BChatHF,
18
- ...BaseAIDefaultCreativeConfig()
8
+ export const axHuggingFaceCreativeConfig = () => structuredClone({
9
+ model: AxHuggingFaceModel.MetaLlama270BChatHF,
10
+ ...axBaseAIDefaultCreativeConfig()
19
11
  });
20
- /**
21
- * HuggingFace: AI Service
22
- * @export
23
- */
24
- export class HuggingFace extends BaseAI {
12
+ export class AxHuggingFace extends AxBaseAI {
25
13
  config;
26
- constructor({ apiKey, config = HuggingFaceDefaultConfig(), options }) {
14
+ constructor({ apiKey, config = axHuggingFaceDefaultConfig(), options }) {
27
15
  if (!apiKey || apiKey === '') {
28
16
  throw new Error('HuggingFace API key not set');
29
17
  }
@@ -31,7 +19,7 @@ export class HuggingFace extends BaseAI {
31
19
  name: 'HuggingFace',
32
20
  apiURL: 'https://api-inference.huggingface.co',
33
21
  headers: { Authorization: `Bearer ${apiKey}` },
34
- modelInfo: modelInfoHuggingFace,
22
+ modelInfo: axModelInfoHuggingFace,
35
23
  models: { model: config.model },
36
24
  options,
37
25
  supportFor: { functions: false, streaming: false }
@@ -1 +1 @@
1
- {"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../../src/ai/huggingface/api.ts"],"names":[],"mappings":"AAGA,OAAO,EACL,MAAM,EACN,mBAAmB,EACnB,2BAA2B,EAC5B,MAAM,YAAY,CAAC;AAGpB,OAAO,EAAE,oBAAoB,EAAE,MAAM,WAAW,CAAC;AACjD,OAAO,EAEL,gBAAgB,EAGjB,MAAM,YAAY,CAAC;AAEpB;;;GAGG;AACH,MAAM,CAAC,MAAM,wBAAwB,GAAG,GAAsB,EAAE,CAC9D,eAAe,CAAC;IACd,KAAK,EAAE,gBAAgB,CAAC,mBAAmB;IAC3C,GAAG,mBAAmB,EAAE;CACzB,CAAC,CAAC;AAEL;;;GAGG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,GAAsB,EAAE,CAC/D,eAAe,CAAC;IACd,KAAK,EAAE,gBAAgB,CAAC,mBAAmB;IAC3C,GAAG,2BAA2B,EAAE;CACjC,CAAC,CAAC;AAQL;;;GAGG;AACH,MAAM,OAAO,WAAY,SAAQ,MAMhC;IACS,MAAM,CAAoB;IAElC,YAAY,EACV,MAAM,EACN,MAAM,GAAG,wBAAwB,EAAE,EACnC,OAAO,EACmB;QAC1B,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;QACjD,CAAC;QACD,KAAK,CAAC;YACJ,IAAI,EAAE,aAAa;YACnB,MAAM,EAAE,sCAAsC;YAC9C,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE;YAC9C,SAAS,EAAE,oBAAoB;YAC/B,MAAM,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE;YAC/B,OAAO;YACP,UAAU,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE;SACnD,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,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,CAAC,EAAE,MAAM,CAAC,CAAC;YACX,eAAe,EAAE,MAAM,CAAC,eAAe;SACrB,CAAC;IACvB,CAAC;IAEQ,eAAe,GAAG,CACzB,GAAgC;IAChC,6DAA6D;IAC7D,OAAiC,EACN,EAAE;QAC7B,MAAM,KAAK,GAAG,GAAG,CAAC,SAAS,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;QAEvD,MAAM,aAAa,GAAG,GAAG,CAAC,SAAS;YACjC,CAAC,CAAC,eAAe,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI;YAC3D,CAAC,CAAC,EAAE,CAAC;QAEP,MAAM,MAAM,GAAG,GAAG,CAAC,UAAU;YAC3B,EAAE,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACZ,OAAO,GAAG,GAAG,CAAC,IAAI,KAAK,GAAG,CAAC,OAAO,EAAE,CAAC;QACvC,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,MAAM,MAAM,GAAG,GAAG,aAAa,IAAI,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC;QAEnD,MAAM,SAAS,GAAG;YAChB,IAAI,EAAE,SAAS;SAChB,CAAC;QAEF,MAAM,QAAQ,GAAuB;YACnC,KAAK;YACL,MAAM;YACN,UAAU,EAAE;gBACV,cAAc,EAAE,GAAG,CAAC,WAAW,EAAE,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS;gBACnE,kBAAkB,EAChB,GAAG,CAAC,WAAW,EAAE,eAAe,IAAI,IAAI,CAAC,MAAM,CAAC,eAAe;gBACjE,WAAW,EAAE,GAAG,CAAC,WAAW,EAAE,WAAW,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW;gBACpE,KAAK,EAAE,GAAG,CAAC,WAAW,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI;gBAChD,KAAK,EAAE,GAAG,CAAC,WAAW,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI;gBAChD,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc;gBAC5C,oBAAoB,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;gBACnC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBAC/B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;aAC9B;YACD,OAAO,EAAE;gBACP,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBAC/B,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;aACzC;SACF,CAAC;QAEF,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAC/B,CAAC,CAAC;IAEO,gBAAgB,GAAG,CAC1B,IAAmC,EACrB,EAAE;QAChB,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,OAAO,EAAE,IAAI,CAAC,cAAc;iBAC7B;aACF;SACF,CAAC;IACJ,CAAC,CAAC;CACH"}
1
+ {"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../../src/ai/huggingface/api.ts"],"names":[],"mappings":"AACA,OAAO,EACL,QAAQ,EACR,qBAAqB,EACrB,6BAA6B,EAC9B,MAAM,YAAY,CAAC;AASpB,OAAO,EAAE,sBAAsB,EAAE,MAAM,WAAW,CAAC;AACnD,OAAO,EAEL,kBAAkB,EAGnB,MAAM,YAAY,CAAC;AAEpB,MAAM,CAAC,MAAM,0BAA0B,GAAG,GAAwB,EAAE,CAClE,eAAe,CAAC;IACd,KAAK,EAAE,kBAAkB,CAAC,mBAAmB;IAC7C,GAAG,qBAAqB,EAAE;CAC3B,CAAC,CAAC;AAEL,MAAM,CAAC,MAAM,2BAA2B,GAAG,GAAwB,EAAE,CACnE,eAAe,CAAC;IACd,KAAK,EAAE,kBAAkB,CAAC,mBAAmB;IAC7C,GAAG,6BAA6B,EAAE;CACnC,CAAC,CAAC;AAQL,MAAM,OAAO,aAAc,SAAQ,QAMlC;IACS,MAAM,CAAsB;IAEpC,YAAY,EACV,MAAM,EACN,MAAM,GAAG,0BAA0B,EAAE,EACrC,OAAO,EACqB;QAC5B,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;QACjD,CAAC;QACD,KAAK,CAAC;YACJ,IAAI,EAAE,aAAa;YACnB,MAAM,EAAE,sCAAsC;YAC9C,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE;YAC9C,SAAS,EAAE,sBAAsB;YACjC,MAAM,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE;YAC/B,OAAO;YACP,UAAU,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE;SACnD,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,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,CAAC,EAAE,MAAM,CAAC,CAAC;YACX,eAAe,EAAE,MAAM,CAAC,eAAe;SACvB,CAAC;IACrB,CAAC;IAEQ,eAAe,GAAG,CACzB,GAA4B;IAC5B,6DAA6D;IAC7D,OAAmC,EACN,EAAE;QAC/B,MAAM,KAAK,GAAG,GAAG,CAAC,SAAS,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;QAEvD,MAAM,aAAa,GAAG,GAAG,CAAC,SAAS;YACjC,CAAC,CAAC,eAAe,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI;YAC3D,CAAC,CAAC,EAAE,CAAC;QAEP,MAAM,MAAM,GAAG,GAAG,CAAC,UAAU;YAC3B,EAAE,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACZ,OAAO,GAAG,GAAG,CAAC,IAAI,KAAK,GAAG,CAAC,OAAO,EAAE,CAAC;QACvC,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,MAAM,MAAM,GAAG,GAAG,aAAa,IAAI,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC;QAEnD,MAAM,SAAS,GAAG;YAChB,IAAI,EAAE,SAAS;SAChB,CAAC;QAEF,MAAM,QAAQ,GAAyB;YACrC,KAAK;YACL,MAAM;YACN,UAAU,EAAE;gBACV,cAAc,EAAE,GAAG,CAAC,WAAW,EAAE,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS;gBACnE,kBAAkB,EAChB,GAAG,CAAC,WAAW,EAAE,eAAe,IAAI,IAAI,CAAC,MAAM,CAAC,eAAe;gBACjE,WAAW,EAAE,GAAG,CAAC,WAAW,EAAE,WAAW,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW;gBACpE,KAAK,EAAE,GAAG,CAAC,WAAW,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI;gBAChD,KAAK,EAAE,GAAG,CAAC,WAAW,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI;gBAChD,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc;gBAC5C,oBAAoB,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;gBACnC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBAC/B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;aAC9B;YACD,OAAO,EAAE;gBACP,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBAC/B,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;aACzC;SACF,CAAC;QAEF,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAC/B,CAAC,CAAC;IAEO,gBAAgB,GAAG,CAC1B,IAAqC,EACrB,EAAE;QAClB,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,OAAO,EAAE,IAAI,CAAC,cAAc;iBAC7B;aACF;SACF,CAAC;IACJ,CAAC,CAAC;CACH"}
@@ -1,6 +1,6 @@
1
- import type { TextModelInfo } from '../types.js';
1
+ import type { AxModelInfo } from '../types.js';
2
2
  /**
3
3
  * HuggingFace: Model information
4
4
  * @export
5
5
  */
6
- export declare const modelInfoHuggingFace: TextModelInfo[];
6
+ export declare const axModelInfoHuggingFace: AxModelInfo[];
@@ -2,5 +2,5 @@
2
2
  * HuggingFace: Model information
3
3
  * @export
4
4
  */
5
- export const modelInfoHuggingFace = [];
5
+ export const axModelInfoHuggingFace = [];
6
6
  //# sourceMappingURL=info.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"info.js","sourceRoot":"","sources":["../../../../../src/ai/huggingface/info.ts"],"names":[],"mappings":"AAEA;;;GAGG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAoB,EAAE,CAAC"}
1
+ {"version":3,"file":"info.js","sourceRoot":"","sources":["../../../../../src/ai/huggingface/info.ts"],"names":[],"mappings":"AAEA;;;GAGG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAkB,EAAE,CAAC"}
@@ -1,25 +1,17 @@
1
- import type { TextModelConfig } from '../types.js';
2
- /**
3
- * HuggingFace: Models for text generation
4
- * @export
5
- */
6
- export declare enum HuggingFaceModel {
1
+ import type { AxModelConfig } from '../types.js';
2
+ export declare enum AxHuggingFaceModel {
7
3
  MetaLlama270BChatHF = "meta-llama/Llama-2-70b-chat-hf"
8
4
  }
9
- /**
10
- * HuggingFace: Model options for text generation
11
- * @export
12
- */
13
- export type HuggingFaceConfig = TextModelConfig & {
14
- model: HuggingFaceModel;
5
+ export type AxHuggingFaceConfig = AxModelConfig & {
6
+ model: AxHuggingFaceModel;
15
7
  returnFullText?: boolean;
16
8
  doSample?: boolean;
17
9
  maxTime?: number;
18
10
  useCache?: boolean;
19
11
  waitForModel?: boolean;
20
12
  };
21
- export type HuggingFaceRequest = {
22
- model: HuggingFaceModel | string;
13
+ export type AxHuggingFaceRequest = {
14
+ model: AxHuggingFaceModel | string;
23
15
  inputs: string;
24
16
  parameters: {
25
17
  max_new_tokens?: number;
@@ -37,6 +29,6 @@ export type HuggingFaceRequest = {
37
29
  wait_for_model?: boolean;
38
30
  };
39
31
  };
40
- export type HuggingFaceResponse = {
32
+ export type AxHuggingFaceResponse = {
41
33
  generated_text: string;
42
34
  };
@@ -1,9 +1,5 @@
1
- /**
2
- * HuggingFace: Models for text generation
3
- * @export
4
- */
5
- export var HuggingFaceModel;
6
- (function (HuggingFaceModel) {
7
- HuggingFaceModel["MetaLlama270BChatHF"] = "meta-llama/Llama-2-70b-chat-hf";
8
- })(HuggingFaceModel || (HuggingFaceModel = {}));
1
+ export var AxHuggingFaceModel;
2
+ (function (AxHuggingFaceModel) {
3
+ AxHuggingFaceModel["MetaLlama270BChatHF"] = "meta-llama/Llama-2-70b-chat-hf";
4
+ })(AxHuggingFaceModel || (AxHuggingFaceModel = {}));
9
5
  //# sourceMappingURL=types.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../../src/ai/huggingface/types.ts"],"names":[],"mappings":"AAEA;;;GAGG;AACH,MAAM,CAAN,IAAY,gBAEX;AAFD,WAAY,gBAAgB;IAC1B,0EAAsD,CAAA;AACxD,CAAC,EAFW,gBAAgB,KAAhB,gBAAgB,QAE3B"}
1
+ {"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../../src/ai/huggingface/types.ts"],"names":[],"mappings":"AAEA,MAAM,CAAN,IAAY,kBAEX;AAFD,WAAY,kBAAkB;IAC5B,4EAAsD,CAAA;AACxD,CAAC,EAFW,kBAAkB,KAAlB,kBAAkB,QAE7B"}
@@ -1,14 +1,14 @@
1
- import { Anthropic, type AnthropicArgs } from './anthropic/api.js';
2
- import { type AzureOpenAIArgs } from './azure-openai/api.js';
3
- import { Cohere, type CohereArgs } from './cohere/api.js';
4
- import { type DeepSeekArgs } from './deepseek/api.js';
5
- import { GoogleGemini } from './google-gemini/api.js';
6
- import { type GroqArgs } from './groq/api.js';
7
- import { HuggingFace, type HuggingFaceArgs } from './huggingface/api.js';
8
- import { type MistralArgs } from './mistral/api.js';
9
- import { type OllamaArgs } from './ollama/api.js';
10
- import { OpenAI, type OpenAIArgs } from './openai/api.js';
11
- import { type TogetherArgs } from './together/api.js';
1
+ import { AxAnthropic, type AxAnthropicArgs } from './anthropic/api.js';
2
+ import { type AxAzureOpenAIArgs } from './azure-openai/api.js';
3
+ import { AxCohere, type AxCohereArgs } from './cohere/api.js';
4
+ import { type AxDeepSeekArgs } from './deepseek/api.js';
5
+ import { AxGoogleGemini } from './google-gemini/api.js';
6
+ import { type AxAxGroqArgs } from './groq/api.js';
7
+ import { AxHuggingFace, type AxHuggingFaceArgs } from './huggingface/api.js';
8
+ import { type AxMistralArgs } from './mistral/api.js';
9
+ import { type AxOllamaArgs } from './ollama/api.js';
10
+ import { AxOpenAI, type AxOpenAIArgs } from './openai/api.js';
11
+ import { type AxTogetherArgs } from './together/api.js';
12
12
  export * from './openai/index.js';
13
13
  export * from './azure-openai/index.js';
14
14
  export * from './huggingface/index.js';
@@ -24,5 +24,5 @@ export * from './types.js';
24
24
  export * from './balance.js';
25
25
  export * from './base.js';
26
26
  export type { API } from '../util/apicall.js';
27
- export type AIName = 'openai' | 'azure-openai' | 'huggingface' | 'together' | 'cohere' | 'google-gemini' | 'anthropic' | 'groq' | 'mistral' | 'deepseek' | 'ollama';
28
- export declare const AI: (name: AIName, options: Readonly<AzureOpenAIArgs | OpenAIArgs | TogetherArgs | AnthropicArgs | CohereArgs | HuggingFaceArgs | GroqArgs | MistralArgs | DeepSeekArgs | OllamaArgs>) => Anthropic | OpenAI | Cohere | GoogleGemini | HuggingFace;
27
+ export type AxAIName = 'openai' | 'azure-openai' | 'huggingface' | 'together' | 'cohere' | 'google-gemini' | 'anthropic' | 'groq' | 'mistral' | 'deepseek' | 'ollama';
28
+ export declare const axAI: (name: AxAIName, options: Readonly<AxAzureOpenAIArgs | AxOpenAIArgs | AxTogetherArgs | AxAnthropicArgs | AxCohereArgs | AxHuggingFaceArgs | AxAxGroqArgs | AxMistralArgs | AxDeepSeekArgs | AxOllamaArgs>) => AxAnthropic | AxOpenAI | AxCohere | AxGoogleGemini | AxHuggingFace;