@ax-llm/ax 9.0.26 → 9.0.29
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.
- package/ai/anthropic/api.d.ts +3 -2
- package/ai/anthropic/api.js +5 -3
- package/ai/anthropic/api.js.map +1 -1
- package/ai/azure-openai/api.d.ts +3 -2
- package/ai/azure-openai/api.js +2 -2
- package/ai/azure-openai/api.js.map +1 -1
- package/ai/balance.d.ts +5 -5
- package/ai/balance.js +4 -0
- package/ai/balance.js.map +1 -1
- package/ai/base.d.ts +18 -13
- package/ai/base.js +66 -31
- package/ai/base.js.map +1 -1
- package/ai/cohere/api.d.ts +3 -2
- package/ai/cohere/api.js +5 -4
- package/ai/cohere/api.js.map +1 -1
- package/ai/deepseek/api.d.ts +3 -1
- package/ai/deepseek/api.js +3 -2
- package/ai/deepseek/api.js.map +1 -1
- package/ai/google-gemini/api.d.ts +3 -2
- package/ai/google-gemini/api.js +9 -5
- package/ai/google-gemini/api.js.map +1 -1
- package/ai/groq/api.d.ts +3 -1
- package/ai/groq/api.js +3 -2
- package/ai/groq/api.js.map +1 -1
- package/ai/huggingface/api.d.ts +3 -2
- package/ai/huggingface/api.js +4 -3
- package/ai/huggingface/api.js.map +1 -1
- package/ai/mistral/api.d.ts +3 -1
- package/ai/mistral/api.js +4 -3
- package/ai/mistral/api.js.map +1 -1
- package/ai/ollama/api.d.ts +2 -1
- package/ai/ollama/api.js +3 -2
- package/ai/ollama/api.js.map +1 -1
- package/ai/openai/api.d.ts +3 -2
- package/ai/openai/api.js +9 -5
- package/ai/openai/api.js.map +1 -1
- package/ai/together/api.d.ts +2 -1
- package/ai/together/api.js +3 -2
- package/ai/together/api.js.map +1 -1
- package/ai/types.d.ts +4 -4
- package/ai/wrap.d.ts +5 -12
- package/ai/wrap.js +15 -29
- package/ai/wrap.js.map +1 -1
- package/cjs/ai/anthropic/api.d.ts +3 -2
- package/cjs/ai/anthropic/api.js +5 -3
- package/cjs/ai/anthropic/api.js.map +1 -1
- package/cjs/ai/azure-openai/api.d.ts +3 -2
- package/cjs/ai/azure-openai/api.js +2 -2
- package/cjs/ai/azure-openai/api.js.map +1 -1
- package/cjs/ai/balance.d.ts +5 -5
- package/cjs/ai/balance.js +4 -0
- package/cjs/ai/balance.js.map +1 -1
- package/cjs/ai/base.d.ts +18 -13
- package/cjs/ai/base.js +66 -31
- package/cjs/ai/base.js.map +1 -1
- package/cjs/ai/cohere/api.d.ts +3 -2
- package/cjs/ai/cohere/api.js +5 -4
- package/cjs/ai/cohere/api.js.map +1 -1
- package/cjs/ai/deepseek/api.d.ts +3 -1
- package/cjs/ai/deepseek/api.js +3 -2
- package/cjs/ai/deepseek/api.js.map +1 -1
- package/cjs/ai/google-gemini/api.d.ts +3 -2
- package/cjs/ai/google-gemini/api.js +9 -5
- package/cjs/ai/google-gemini/api.js.map +1 -1
- package/cjs/ai/groq/api.d.ts +3 -1
- package/cjs/ai/groq/api.js +3 -2
- package/cjs/ai/groq/api.js.map +1 -1
- package/cjs/ai/huggingface/api.d.ts +3 -2
- package/cjs/ai/huggingface/api.js +4 -3
- package/cjs/ai/huggingface/api.js.map +1 -1
- package/cjs/ai/mistral/api.d.ts +3 -1
- package/cjs/ai/mistral/api.js +3 -2
- package/cjs/ai/mistral/api.js.map +1 -1
- package/cjs/ai/ollama/api.d.ts +2 -1
- package/cjs/ai/ollama/api.js +3 -2
- package/cjs/ai/ollama/api.js.map +1 -1
- package/cjs/ai/openai/api.d.ts +3 -2
- package/cjs/ai/openai/api.js +9 -5
- package/cjs/ai/openai/api.js.map +1 -1
- package/cjs/ai/together/api.d.ts +2 -1
- package/cjs/ai/together/api.js +3 -2
- package/cjs/ai/together/api.js.map +1 -1
- package/cjs/ai/types.d.ts +4 -4
- package/cjs/ai/wrap.d.ts +5 -12
- package/cjs/ai/wrap.js +15 -29
- package/cjs/ai/wrap.js.map +1 -1
- package/package.json +1 -1
package/cjs/ai/cohere/api.d.ts
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
import type { API } from '../../util/apicall.js';
|
|
2
2
|
import { AxBaseAI } from '../base.js';
|
|
3
3
|
import type { AxAIPromptConfig, AxAIServiceOptions, AxChatRequest, AxChatResponse, AxEmbedRequest, AxEmbedResponse, AxModelConfig } from '../types.js';
|
|
4
|
-
import { type AxAICohereChatRequest, type AxAICohereChatResponse, type AxAICohereChatResponseDelta, type AxAICohereConfig, type AxAICohereEmbedRequest, type AxAICohereEmbedResponse } from './types.js';
|
|
4
|
+
import { type AxAICohereChatRequest, type AxAICohereChatResponse, type AxAICohereChatResponseDelta, type AxAICohereConfig, AxAICohereEmbedModel, type AxAICohereEmbedRequest, type AxAICohereEmbedResponse, AxAICohereModel } from './types.js';
|
|
5
5
|
export declare const axAICohereDefaultConfig: () => AxAICohereConfig;
|
|
6
6
|
export declare const axAICohereCreativeConfig: () => AxAICohereConfig;
|
|
7
7
|
export interface AxAICohereArgs {
|
|
@@ -9,10 +9,11 @@ export interface AxAICohereArgs {
|
|
|
9
9
|
apiKey: string;
|
|
10
10
|
config?: Readonly<Partial<AxAICohereConfig>>;
|
|
11
11
|
options?: Readonly<AxAIServiceOptions>;
|
|
12
|
+
modelMap?: Record<string, AxAICohereModel | AxAICohereEmbedModel | string>;
|
|
12
13
|
}
|
|
13
14
|
export declare class AxAICohere extends AxBaseAI<AxAICohereChatRequest, AxAICohereEmbedRequest, AxAICohereChatResponse, AxAICohereChatResponseDelta, AxAICohereEmbedResponse> {
|
|
14
15
|
private config;
|
|
15
|
-
constructor({ apiKey, config, options }: Readonly<Omit<AxAICohereArgs, 'name'>>);
|
|
16
|
+
constructor({ apiKey, config, options, modelMap }: Readonly<Omit<AxAICohereArgs, 'name'>>);
|
|
16
17
|
getModelConfig(): AxModelConfig;
|
|
17
18
|
generateChatReq: (req: Readonly<AxChatRequest>, _config: Readonly<AxAIPromptConfig>) => [API, AxAICohereChatRequest];
|
|
18
19
|
generateEmbedReq: (req: Readonly<AxEmbedRequest>) => [API, AxAICohereEmbedRequest];
|
package/cjs/ai/cohere/api.js
CHANGED
|
@@ -18,7 +18,7 @@ const axAICohereCreativeConfig = () => structuredClone({
|
|
|
18
18
|
exports.axAICohereCreativeConfig = axAICohereCreativeConfig;
|
|
19
19
|
class AxAICohere extends base_js_1.AxBaseAI {
|
|
20
20
|
config;
|
|
21
|
-
constructor({ apiKey, config, options }) {
|
|
21
|
+
constructor({ apiKey, config, options, modelMap }) {
|
|
22
22
|
if (!apiKey || apiKey === '') {
|
|
23
23
|
throw new Error('Cohere API key not set');
|
|
24
24
|
}
|
|
@@ -33,7 +33,8 @@ class AxAICohere extends base_js_1.AxBaseAI {
|
|
|
33
33
|
modelInfo: info_js_1.axModelInfoCohere,
|
|
34
34
|
models: { model: _config.model },
|
|
35
35
|
supportFor: { functions: true, streaming: true },
|
|
36
|
-
options
|
|
36
|
+
options,
|
|
37
|
+
modelMap
|
|
37
38
|
});
|
|
38
39
|
this.config = _config;
|
|
39
40
|
}
|
|
@@ -53,7 +54,7 @@ class AxAICohere extends base_js_1.AxBaseAI {
|
|
|
53
54
|
generateChatReq = (req,
|
|
54
55
|
// eslint-disable-next-line @typescript-eslint/no-unused-vars
|
|
55
56
|
_config) => {
|
|
56
|
-
const model =
|
|
57
|
+
const model = this.config.model;
|
|
57
58
|
const lastChatMsg = req.chatPrompt.at(-1);
|
|
58
59
|
const restOfChat = req.chatPrompt.slice(0, -1);
|
|
59
60
|
let message;
|
|
@@ -114,7 +115,7 @@ class AxAICohere extends base_js_1.AxBaseAI {
|
|
|
114
115
|
return [apiConfig, reqValue];
|
|
115
116
|
};
|
|
116
117
|
generateEmbedReq = (req) => {
|
|
117
|
-
const model =
|
|
118
|
+
const model = this.config.embedModel;
|
|
118
119
|
if (!model) {
|
|
119
120
|
throw new Error('Embed model not set');
|
|
120
121
|
}
|
package/cjs/ai/cohere/api.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../ai/cohere/api.ts"],"names":[],"mappings":";;;AACA,wCAIoB;AAWpB,uCAA8C;AAC9C,yCASoB;AAEb,MAAM,uBAAuB,GAAG,GAAqB,EAAE,CAC5D,eAAe,CAAC;IACd,KAAK,EAAE,0BAAe,CAAC,YAAY;IACnC,UAAU,EAAE,+BAAoB,CAAC,eAAe;IAChD,GAAG,IAAA,+BAAqB,GAAE;CAC3B,CAAC,CAAC;AALQ,QAAA,uBAAuB,2BAK/B;AAEE,MAAM,wBAAwB,GAAG,GAAqB,EAAE,CAC7D,eAAe,CAAC;IACd,KAAK,EAAE,0BAAe,CAAC,QAAQ;IAC/B,UAAU,EAAE,+BAAoB,CAAC,eAAe;IAChD,GAAG,IAAA,uCAA6B,GAAE;CACnC,CAAC,CAAC;AALQ,QAAA,wBAAwB,4BAKhC;
|
|
1
|
+
{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../ai/cohere/api.ts"],"names":[],"mappings":";;;AACA,wCAIoB;AAWpB,uCAA8C;AAC9C,yCASoB;AAEb,MAAM,uBAAuB,GAAG,GAAqB,EAAE,CAC5D,eAAe,CAAC;IACd,KAAK,EAAE,0BAAe,CAAC,YAAY;IACnC,UAAU,EAAE,+BAAoB,CAAC,eAAe;IAChD,GAAG,IAAA,+BAAqB,GAAE;CAC3B,CAAC,CAAC;AALQ,QAAA,uBAAuB,2BAK/B;AAEE,MAAM,wBAAwB,GAAG,GAAqB,EAAE,CAC7D,eAAe,CAAC;IACd,KAAK,EAAE,0BAAe,CAAC,QAAQ;IAC/B,UAAU,EAAE,+BAAoB,CAAC,eAAe;IAChD,GAAG,IAAA,uCAA6B,GAAE;CACnC,CAAC,CAAC;AALQ,QAAA,wBAAwB,4BAKhC;AAUL,MAAa,UAAW,SAAQ,kBAM/B;IACS,MAAM,CAAmB;IAEjC,YAAY,EACV,MAAM,EACN,MAAM,EACN,OAAO,EACP,QAAQ,EAC+B;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,IAAA,+BAAuB,GAAE;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,2BAAiB;YAC5B,MAAM,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE;YAChC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE;YAChD,OAAO;YACP,QAAQ;SACT,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,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;QAEhC,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,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;QAErC,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;AAjQD,gCAiQC;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"}
|
package/cjs/ai/deepseek/api.d.ts
CHANGED
|
@@ -1,6 +1,7 @@
|
|
|
1
1
|
import { AxAIOpenAI } from '../openai/api.js';
|
|
2
2
|
import type { AxAIOpenAIConfig } from '../openai/types.js';
|
|
3
3
|
import type { AxAIServiceOptions } from '../types.js';
|
|
4
|
+
import { AxAIDeepSeekModel } from './types.js';
|
|
4
5
|
type DeepSeekConfig = AxAIOpenAIConfig;
|
|
5
6
|
export declare const axAIDeepSeekDefaultConfig: () => DeepSeekConfig;
|
|
6
7
|
export declare const axAIDeepSeekCodeConfig: () => DeepSeekConfig;
|
|
@@ -9,8 +10,9 @@ export interface AxAIDeepSeekArgs {
|
|
|
9
10
|
apiKey: string;
|
|
10
11
|
config?: Readonly<Partial<DeepSeekConfig>>;
|
|
11
12
|
options?: Readonly<AxAIServiceOptions>;
|
|
13
|
+
modelMap?: Record<string, AxAIDeepSeekModel | string>;
|
|
12
14
|
}
|
|
13
15
|
export declare class AxAIDeepSeek extends AxAIOpenAI {
|
|
14
|
-
constructor({ apiKey, config, options }: Readonly<Omit<AxAIDeepSeekArgs, 'name'>>);
|
|
16
|
+
constructor({ apiKey, config, options, modelMap }: Readonly<Omit<AxAIDeepSeekArgs, 'name'>>);
|
|
15
17
|
}
|
|
16
18
|
export {};
|
package/cjs/ai/deepseek/api.js
CHANGED
|
@@ -16,7 +16,7 @@ const axAIDeepSeekCodeConfig = () => structuredClone({
|
|
|
16
16
|
});
|
|
17
17
|
exports.axAIDeepSeekCodeConfig = axAIDeepSeekCodeConfig;
|
|
18
18
|
class AxAIDeepSeek extends api_js_1.AxAIOpenAI {
|
|
19
|
-
constructor({ apiKey, config, options }) {
|
|
19
|
+
constructor({ apiKey, config, options, modelMap }) {
|
|
20
20
|
if (!apiKey || apiKey === '') {
|
|
21
21
|
throw new Error('DeepSeek API key not set');
|
|
22
22
|
}
|
|
@@ -29,7 +29,8 @@ class AxAIDeepSeek extends api_js_1.AxAIOpenAI {
|
|
|
29
29
|
config: _config,
|
|
30
30
|
options,
|
|
31
31
|
apiURL: 'https://api.deepseek.com',
|
|
32
|
-
modelInfo: info_js_1.axModelInfoDeepSeek
|
|
32
|
+
modelInfo: info_js_1.axModelInfoDeepSeek,
|
|
33
|
+
modelMap
|
|
33
34
|
});
|
|
34
35
|
super.setName('DeepSeek');
|
|
35
36
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../ai/deepseek/api.ts"],"names":[],"mappings":";;;AAAA,wCAGoB;AACpB,6CAA8C;AAI9C,uCAAgD;AAChD,yCAA+C;AAIxC,MAAM,yBAAyB,GAAG,GAAmB,EAAE,CAC5D,eAAe,CAAC;IACd,KAAK,EAAE,4BAAiB,CAAC,YAAY;IACrC,GAAG,IAAA,+BAAqB,GAAE;CAC3B,CAAC,CAAC;AAJQ,QAAA,yBAAyB,6BAIjC;AAEE,MAAM,sBAAsB,GAAG,GAAmB,EAAE,CACzD,eAAe,CAAC;IACd,KAAK,EAAE,4BAAiB,CAAC,aAAa;IACtC,GAAG,IAAA,uCAA6B,GAAE;CACnC,CAAC,CAAC;AAJQ,QAAA,sBAAsB,0BAI9B;
|
|
1
|
+
{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../ai/deepseek/api.ts"],"names":[],"mappings":";;;AAAA,wCAGoB;AACpB,6CAA8C;AAI9C,uCAAgD;AAChD,yCAA+C;AAIxC,MAAM,yBAAyB,GAAG,GAAmB,EAAE,CAC5D,eAAe,CAAC;IACd,KAAK,EAAE,4BAAiB,CAAC,YAAY;IACrC,GAAG,IAAA,+BAAqB,GAAE;CAC3B,CAAC,CAAC;AAJQ,QAAA,yBAAyB,6BAIjC;AAEE,MAAM,sBAAsB,GAAG,GAAmB,EAAE,CACzD,eAAe,CAAC;IACd,KAAK,EAAE,4BAAiB,CAAC,aAAa;IACtC,GAAG,IAAA,uCAA6B,GAAE;CACnC,CAAC,CAAC;AAJQ,QAAA,sBAAsB,0BAI9B;AAUL,MAAa,YAAa,SAAQ,mBAAU;IAC1C,YAAY,EACV,MAAM,EACN,MAAM,EACN,OAAO,EACP,QAAQ,EACiC;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,IAAA,iCAAyB,GAAE;YAC9B,GAAG,MAAM;SACV,CAAC;QACF,KAAK,CAAC;YACJ,MAAM;YACN,MAAM,EAAE,OAAO;YACf,OAAO;YACP,MAAM,EAAE,0BAA0B;YAClC,SAAS,EAAE,6BAAmB;YAC9B,QAAQ;SACT,CAAC,CAAC;QAEH,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAC5B,CAAC;CACF;AAzBD,oCAyBC"}
|
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
import type { API } from '../../util/apicall.js';
|
|
2
2
|
import { AxBaseAI } from '../base.js';
|
|
3
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';
|
|
4
|
+
import { type AxAIGoogleGeminiBatchEmbedRequest, type AxAIGoogleGeminiBatchEmbedResponse, type AxAIGoogleGeminiChatRequest, type AxAIGoogleGeminiChatResponse, type AxAIGoogleGeminiChatResponseDelta, type AxAIGoogleGeminiConfig, AxAIGoogleGeminiEmbedModel, AxAIGoogleGeminiModel } from './types.js';
|
|
5
5
|
/**
|
|
6
6
|
* AxAIGoogleGemini: Default Model options for text generation
|
|
7
7
|
* @export
|
|
@@ -17,6 +17,7 @@ export interface AxAIGoogleGeminiArgs {
|
|
|
17
17
|
options?: Readonly<AxAIServiceOptions & {
|
|
18
18
|
codeExecution?: boolean;
|
|
19
19
|
}>;
|
|
20
|
+
modelMap?: Record<string, AxAIGoogleGeminiModel | AxAIGoogleGeminiEmbedModel | string>;
|
|
20
21
|
}
|
|
21
22
|
/**
|
|
22
23
|
* AxAIGoogleGemini: AI Service
|
|
@@ -26,7 +27,7 @@ export declare class AxAIGoogleGemini extends AxBaseAI<AxAIGoogleGeminiChatReque
|
|
|
26
27
|
private options?;
|
|
27
28
|
private config;
|
|
28
29
|
private apiKey;
|
|
29
|
-
constructor({ apiKey, projectId, region, config, options }: Readonly<Omit<AxAIGoogleGeminiArgs, 'name'>>);
|
|
30
|
+
constructor({ apiKey, projectId, region, config, options, modelMap }: Readonly<Omit<AxAIGoogleGeminiArgs, 'name'>>);
|
|
30
31
|
getModelConfig(): AxModelConfig;
|
|
31
32
|
generateChatReq: (req: Readonly<AxChatRequest>) => [API, AxAIGoogleGeminiChatRequest];
|
|
32
33
|
generateEmbedReq: (req: Readonly<AxEmbedRequest>) => [API, AxAIGoogleGeminiBatchEmbedRequest];
|
|
@@ -48,7 +48,7 @@ class AxAIGoogleGemini extends base_js_1.AxBaseAI {
|
|
|
48
48
|
options;
|
|
49
49
|
config;
|
|
50
50
|
apiKey;
|
|
51
|
-
constructor({ apiKey, projectId, region, config, options }) {
|
|
51
|
+
constructor({ apiKey, projectId, region, config, options, modelMap }) {
|
|
52
52
|
if (!apiKey || apiKey === '') {
|
|
53
53
|
throw new Error('GoogleGemini AI API key not set');
|
|
54
54
|
}
|
|
@@ -65,9 +65,13 @@ class AxAIGoogleGemini extends base_js_1.AxBaseAI {
|
|
|
65
65
|
apiURL,
|
|
66
66
|
headers: {},
|
|
67
67
|
modelInfo: info_js_1.axModelInfoGoogleGemini,
|
|
68
|
-
models: {
|
|
68
|
+
models: {
|
|
69
|
+
model: _config.model,
|
|
70
|
+
embedModel: _config.embedModel
|
|
71
|
+
},
|
|
69
72
|
options,
|
|
70
|
-
supportFor: { functions: true, streaming: true }
|
|
73
|
+
supportFor: { functions: true, streaming: true },
|
|
74
|
+
modelMap
|
|
71
75
|
});
|
|
72
76
|
this.options = options;
|
|
73
77
|
this.config = _config;
|
|
@@ -83,7 +87,7 @@ class AxAIGoogleGemini extends base_js_1.AxBaseAI {
|
|
|
83
87
|
};
|
|
84
88
|
}
|
|
85
89
|
generateChatReq = (req) => {
|
|
86
|
-
const model =
|
|
90
|
+
const model = this.config.model;
|
|
87
91
|
const stream = req.modelConfig?.stream ?? this.config.stream;
|
|
88
92
|
if (!req.chatPrompt || req.chatPrompt.length === 0) {
|
|
89
93
|
throw new Error('Chat prompt is empty');
|
|
@@ -228,7 +232,7 @@ class AxAIGoogleGemini extends base_js_1.AxBaseAI {
|
|
|
228
232
|
return [apiConfig, reqValue];
|
|
229
233
|
};
|
|
230
234
|
generateEmbedReq = (req) => {
|
|
231
|
-
const model =
|
|
235
|
+
const model = this.config.embedModel;
|
|
232
236
|
if (!model) {
|
|
233
237
|
throw new Error('Embed model not set');
|
|
234
238
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../ai/google-gemini/api.ts"],"names":[],"mappings":";;;AACA,wCAIoB;AAYpB,uCAAoD;AACpD,yCAYoB;AAEpB,MAAM,cAAc,GAAmC;IACrD;QACE,QAAQ,EAAE,yCAA8B,CAAC,sBAAsB;QAC/D,SAAS,EAAE,0CAA+B,CAAC,SAAS;KACrD;IACD;QACE,QAAQ,EAAE,yCAA8B,CAAC,sBAAsB;QAC/D,SAAS,EAAE,0CAA+B,CAAC,SAAS;KACrD;IACD;QACE,QAAQ,EAAE,yCAA8B,CAAC,4BAA4B;QACrE,SAAS,EAAE,0CAA+B,CAAC,SAAS;KACrD;IACD;QACE,QAAQ,EAAE,yCAA8B,CAAC,4BAA4B;QACrE,SAAS,EAAE,0CAA+B,CAAC,SAAS;KACrD;CACF,CAAC;AAEF;;;GAGG;AACI,MAAM,6BAA6B,GAAG,GAA2B,EAAE,CACxE,eAAe,CAAC;IACd,KAAK,EAAE,gCAAqB,CAAC,WAAW;IACxC,UAAU,EAAE,qCAA0B,CAAC,YAAY;IACnD,cAAc;IACd,GAAG,IAAA,+BAAqB,GAAE;CAC3B,CAAC,CAAC;AANQ,QAAA,6BAA6B,iCAMrC;AAEE,MAAM,qCAAqC,GAChD,GAA2B,EAAE,CAC3B,eAAe,CAAC;IACd,KAAK,EAAE,gCAAqB,CAAC,aAAa;IAC1C,UAAU,EAAE,qCAA0B,CAAC,YAAY;IACnD,cAAc;IACd,GAAG,IAAA,uCAA6B,GAAE;CACnC,CAAC,CAAC;AAPM,QAAA,qCAAqC,yCAO3C;
|
|
1
|
+
{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../ai/google-gemini/api.ts"],"names":[],"mappings":";;;AACA,wCAIoB;AAYpB,uCAAoD;AACpD,yCAYoB;AAEpB,MAAM,cAAc,GAAmC;IACrD;QACE,QAAQ,EAAE,yCAA8B,CAAC,sBAAsB;QAC/D,SAAS,EAAE,0CAA+B,CAAC,SAAS;KACrD;IACD;QACE,QAAQ,EAAE,yCAA8B,CAAC,sBAAsB;QAC/D,SAAS,EAAE,0CAA+B,CAAC,SAAS;KACrD;IACD;QACE,QAAQ,EAAE,yCAA8B,CAAC,4BAA4B;QACrE,SAAS,EAAE,0CAA+B,CAAC,SAAS;KACrD;IACD;QACE,QAAQ,EAAE,yCAA8B,CAAC,4BAA4B;QACrE,SAAS,EAAE,0CAA+B,CAAC,SAAS;KACrD;CACF,CAAC;AAEF;;;GAGG;AACI,MAAM,6BAA6B,GAAG,GAA2B,EAAE,CACxE,eAAe,CAAC;IACd,KAAK,EAAE,gCAAqB,CAAC,WAAW;IACxC,UAAU,EAAE,qCAA0B,CAAC,YAAY;IACnD,cAAc;IACd,GAAG,IAAA,+BAAqB,GAAE;CAC3B,CAAC,CAAC;AANQ,QAAA,6BAA6B,iCAMrC;AAEE,MAAM,qCAAqC,GAChD,GAA2B,EAAE,CAC3B,eAAe,CAAC;IACd,KAAK,EAAE,gCAAqB,CAAC,aAAa;IAC1C,UAAU,EAAE,qCAA0B,CAAC,YAAY;IACnD,cAAc;IACd,GAAG,IAAA,uCAA6B,GAAE;CACnC,CAAC,CAAC;AAPM,QAAA,qCAAqC,yCAO3C;AAeP;;;GAGG;AACH,MAAa,gBAAiB,SAAQ,kBAMrC;IACS,OAAO,CAAmC;IAC1C,MAAM,CAAyB;IAC/B,MAAM,CAAS;IAEvB,YAAY,EACV,MAAM,EACN,SAAS,EACT,MAAM,EACN,MAAM,EACN,OAAO,EACP,QAAQ,EACqC;QAC7C,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACrD,CAAC;QAED,IAAI,MAAM,GAAG,kDAAkD,CAAC;QAEhE,IAAI,SAAS,IAAI,MAAM,EAAE,CAAC;YACxB,MAAM,GAAG,gBAAgB,MAAM,0CAA0C,SAAS,wCAAwC,CAAC;QAC7H,CAAC;QAED,MAAM,OAAO,GAAG;YACd,GAAG,IAAA,qCAA6B,GAAE;YAClC,GAAG,MAAM;SACV,CAAC;QAEF,KAAK,CAAC;YACJ,IAAI,EAAE,gBAAgB;YACtB,MAAM;YACN,OAAO,EAAE,EAAE;YACX,SAAS,EAAE,iCAAuB;YAClC,MAAM,EAAE;gBACN,KAAK,EAAE,OAAO,CAAC,KAA8B;gBAC7C,UAAU,EAAE,OAAO,CAAC,UAAwC;aAC7D;YACD,OAAO;YACP,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE;YAChD,QAAQ;SACT,CAAC,CAAC;QACH,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC;QACtB,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;SACD,CAAC;IACrB,CAAC;IAEQ,eAAe,GAAG,CACzB,GAA4B,EACQ,EAAE;QACtC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;QAChC,MAAM,MAAM,GAAG,GAAG,CAAC,WAAW,EAAE,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;QAE7D,IAAI,CAAC,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnD,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC1C,CAAC;QAED,MAAM,SAAS,GAAG;YAChB,IAAI,EAAE,MAAM;gBACV,CAAC,CAAC,WAAW,KAAK,sCAAsC,IAAI,CAAC,MAAM,EAAE;gBACrE,CAAC,CAAC,WAAW,KAAK,wBAAwB,IAAI,CAAC,MAAM,EAAE;SAC1D,CAAC;QAEF,MAAM,aAAa,GAAG,GAAG,CAAC,UAAU;aACjC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;aAClC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QAEzB,MAAM,iBAAiB,GACrB,aAAa,CAAC,MAAM,GAAG,CAAC;YACtB,CAAC,CAAC;gBACE,IAAI,EAAE,MAAe;gBACrB,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;aAC3C;YACH,CAAC,CAAC,SAAS,CAAC;QAEhB,MAAM,QAAQ,GAA4C,GAAG,CAAC,UAAU;aACrE,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;aAClC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE;YACd,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;gBACjB,KAAK,MAAM,CAAC,CAAC,CAAC;oBACZ,MAAM,KAAK,GAGE,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC;wBACrC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;4BACvB,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;gCACf,KAAK,MAAM;oCACT,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;gCAC1B,KAAK,OAAO;oCACV,OAAO;wCACL,UAAU,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,EAAE;qCACpD,CAAC;gCACJ;oCACE,MAAM,IAAI,KAAK,CACb,kDAAkD,CAAC,GAAG,CACvD,CAAC;4BACN,CAAC;wBACH,CAAC,CAAC;wBACJ,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;oBAC5B,OAAO;wBACL,IAAI,EAAE,MAAe;wBACrB,KAAK;qBACN,CAAC;gBACJ,CAAC;gBAED,KAAK,WAAW,CAAC,CAAC,CAAC;oBACjB,IAAI,SAAS,IAAI,GAAG,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;wBACxD,MAAM,KAAK,GAGE,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;wBACrC,OAAO;4BACL,IAAI,EAAE,OAAgB;4BACtB,KAAK;yBACN,CAAC;oBACJ,CAAC;oBAED,IAAI,KAAK,GAGI,EAAE,CAAC;oBAEhB,IAAI,eAAe,IAAI,GAAG,EAAE,CAAC;wBAC3B,KAAK;4BACH,GAAG,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gCAC3B,MAAM,IAAI,GACR,OAAO,CAAC,CAAC,QAAQ,CAAC,MAAM,KAAK,QAAQ;oCACnC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC;oCAC/B,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC;gCACxB,OAAO;oCACL,YAAY,EAAE;wCACZ,IAAI,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI;wCACrB,IAAI,EAAE,IAAI;qCACX;iCACF,CAAC;4BACJ,CAAC,CAAC,IAAI,EAAE,CAAC;oBACb,CAAC;oBAED,OAAO;wBACL,IAAI,EAAE,OAAgB;wBACtB,KAAK;qBACN,CAAC;gBACJ,CAAC;gBAED,KAAK,UAAU,CAAC,CAAC,CAAC;oBAChB,IAAI,CAAC,CAAC,YAAY,IAAI,GAAG,CAAC,EAAE,CAAC;wBAC3B,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,GAAG,CAAC,CAAC;oBACnE,CAAC;oBACD,MAAM,KAAK,GAGE;wBACX;4BACE,gBAAgB,EAAE;gCAChB,IAAI,EAAE,GAAG,CAAC,UAAU;gCACpB,QAAQ,EAAE,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE;6BACjC;yBACF;qBACF,CAAC;oBAEF,OAAO;wBACL,IAAI,EAAE,UAAmB;wBACzB,KAAK;qBACN,CAAC;gBACJ,CAAC;gBAED;oBACE,MAAM,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC;YACpC,CAAC;QACH,CAAC,CAAC,CAAC;QAEL,IAAI,KAAK,GAAqD,EAAE,CAAC;QAEjE,IAAI,GAAG,CAAC,SAAS,IAAI,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9C,KAAK,CAAC,IAAI,CAAC,EAAE,oBAAoB,EAAE,GAAG,CAAC,SAAS,EAAE,CAAC,CAAC;QACtD,CAAC;QAED,IAAI,IAAI,CAAC,OAAO,EAAE,aAAa,EAAE,CAAC;YAChC,KAAK,CAAC,IAAI,CAAC,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC,CAAC;QACpC,CAAC;QAED,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,KAAK,GAAG,SAAS,CAAC;QACpB,CAAC;QAED,gEAAgE;QAChE,IAAI,WAAW,CAAC;QAChB,IAAI,GAAG,CAAC,YAAY,EAAE,CAAC;YACrB,IAAI,GAAG,CAAC,YAAY,KAAK,MAAM,EAAE,CAAC;gBAChC,WAAW,GAAG,EAAE,uBAAuB,EAAE,EAAE,IAAI,EAAE,MAAe,EAAE,EAAE,CAAC;YACvE,CAAC;iBAAM,IAAI,GAAG,CAAC,YAAY,KAAK,MAAM,EAAE,CAAC;gBACvC,WAAW,GAAG,EAAE,uBAAuB,EAAE,EAAE,IAAI,EAAE,MAAe,EAAE,EAAE,CAAC;YACvE,CAAC;iBAAM,IAAI,GAAG,CAAC,YAAY,KAAK,UAAU,EAAE,CAAC;gBAC3C,WAAW,GAAG;oBACZ,uBAAuB,EAAE,EAAE,IAAI,EAAE,KAAc,EAAE;iBAClD,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,WAAW,GAAG;oBACZ,uBAAuB,EAAE;wBACvB,IAAI,EAAE,KAAc;wBACpB,sBAAsB,EAAE,CAAC,GAAG,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC;qBACzD;iBACF,CAAC;YACJ,CAAC;QACH,CAAC;QAED,MAAM,gBAAgB,GAAG;YACvB,eAAe,EAAE,GAAG,CAAC,WAAW,EAAE,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS;YACpE,WAAW,EAAE,GAAG,CAAC,WAAW,EAAE,WAAW,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW;YACpE,IAAI,EAAE,GAAG,CAAC,WAAW,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI;YAC/C,IAAI,EAAE,GAAG,CAAC,WAAW,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI;YAC/C,cAAc,EAAE,CAAC;YACjB,aAAa,EAAE,GAAG,CAAC,WAAW,EAAE,aAAa,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa;SAC3E,CAAC;QAEF,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC;QAElD,MAAM,QAAQ,GAAgC;YAC5C,QAAQ;YACR,KAAK;YACL,WAAW;YACX,iBAAiB;YACjB,gBAAgB;YAChB,cAAc;SACf,CAAC;QAEF,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAC/B,CAAC,CAAC;IAEO,gBAAgB,GAAG,CAC1B,GAA6B,EACa,EAAE;QAC5C,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;QAErC,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,KAAK,uBAAuB,IAAI,CAAC,MAAM,EAAE;SAC3D,CAAC;QAEF,MAAM,QAAQ,GAAsC;YAClD,QAAQ,EAAE,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;SACrD,CAAC;QAEF,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAC/B,CAAC,CAAC;IAEO,gBAAgB,GAAG,CAC1B,IAA4C,EAC5B,EAAE;QAClB,MAAM,OAAO,GAA2B,IAAI,CAAC,UAAU,EAAE,GAAG,CAC1D,CAAC,SAAS,EAAE,EAAE;YACZ,MAAM,MAAM,GAAyB,EAAE,CAAC;YAExC,QAAQ,SAAS,CAAC,YAAY,EAAE,CAAC;gBAC/B,KAAK,YAAY;oBACf,MAAM,CAAC,YAAY,GAAG,QAAQ,CAAC;oBAC/B,MAAM;gBACR,KAAK,MAAM;oBACT,MAAM,CAAC,YAAY,GAAG,MAAM,CAAC;oBAC7B,MAAM;gBACR,KAAK,QAAQ;oBACX,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;gBAC3C,KAAK,YAAY;oBACf,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;YACjD,CAAC;YAED,KAAK,MAAM,IAAI,IAAI,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;gBAC3C,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;oBACnB,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC;oBAC3B,SAAS;gBACX,CAAC;gBACD,IAAI,cAAc,IAAI,IAAI,EAAE,CAAC;oBAC3B,MAAM,CAAC,aAAa,GAAG;wBACrB;4BACE,EAAE,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI;4BAC1B,IAAI,EAAE,UAAU;4BAChB,QAAQ,EAAE;gCACR,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI;gCAC5B,MAAM,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI;6BAC/B;yBACF;qBACF,CAAC;gBACJ,CAAC;YACH,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC,CACF,CAAC;QAEF,IAAI,UAAoC,CAAC;QACzC,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,UAAU,GAAG;gBACX,WAAW,EAAE,IAAI,CAAC,aAAa,CAAC,eAAe;gBAC/C,YAAY,EAAE,IAAI,CAAC,aAAa,CAAC,gBAAgB;gBACjD,gBAAgB,EAAE,IAAI,CAAC,aAAa,CAAC,oBAAoB;aAC1D,CAAC;QACJ,CAAC;QACD,OAAO;YACL,OAAO;YACP,UAAU;SACX,CAAC;IACJ,CAAC,CAAC;IAEO,sBAAsB,GAAG,CAChC,IAAiD,EACjC,EAAE;QAClB,OAAO,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;IACrC,CAAC,CAAC;IAEO,iBAAiB,GAAG,CAC3B,IAAkD,EACjC,EAAE;QACnB,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAEvE,OAAO;YACL,UAAU;SACX,CAAC;IACJ,CAAC,CAAC;CACH;AAnVD,4CAmVC"}
|
package/cjs/ai/groq/api.d.ts
CHANGED
|
@@ -1,6 +1,7 @@
|
|
|
1
1
|
import { AxAIOpenAI } from '../openai/api.js';
|
|
2
2
|
import type { AxAIOpenAIConfig } from '../openai/types.js';
|
|
3
3
|
import type { AxAIServiceOptions } from '../types.js';
|
|
4
|
+
import { AxAIGroqModel } from './types.js';
|
|
4
5
|
type AxAIGroqAIConfig = AxAIOpenAIConfig;
|
|
5
6
|
export interface AxAIGroqArgs {
|
|
6
7
|
name: 'groq';
|
|
@@ -9,9 +10,10 @@ export interface AxAIGroqArgs {
|
|
|
9
10
|
options?: Readonly<AxAIServiceOptions> & {
|
|
10
11
|
tokensPerMinute?: number;
|
|
11
12
|
};
|
|
13
|
+
modelMap?: Record<string, AxAIGroqModel>;
|
|
12
14
|
}
|
|
13
15
|
export declare class AxAIGroq extends AxAIOpenAI {
|
|
14
|
-
constructor({ apiKey, config, options }: Readonly<Omit<AxAIGroqArgs, 'groq'>>);
|
|
16
|
+
constructor({ apiKey, config, options, modelMap }: Readonly<Omit<AxAIGroqArgs, 'groq'>>);
|
|
15
17
|
setOptions: (options: Readonly<AxAIServiceOptions>) => void;
|
|
16
18
|
private newRateLimiter;
|
|
17
19
|
}
|
package/cjs/ai/groq/api.js
CHANGED
|
@@ -10,7 +10,7 @@ const axAIGroqDefaultConfig = () => structuredClone({
|
|
|
10
10
|
...(0, base_js_1.axBaseAIDefaultConfig)()
|
|
11
11
|
});
|
|
12
12
|
class AxAIGroq extends api_js_1.AxAIOpenAI {
|
|
13
|
-
constructor({ apiKey, config, options }) {
|
|
13
|
+
constructor({ apiKey, config, options, modelMap }) {
|
|
14
14
|
if (!apiKey || apiKey === '') {
|
|
15
15
|
throw new Error('Groq API key not set');
|
|
16
16
|
}
|
|
@@ -27,7 +27,8 @@ class AxAIGroq extends api_js_1.AxAIOpenAI {
|
|
|
27
27
|
config: _config,
|
|
28
28
|
options: _options,
|
|
29
29
|
apiURL: 'https://api.groq.com/openai/v1',
|
|
30
|
-
modelInfo: []
|
|
30
|
+
modelInfo: [],
|
|
31
|
+
modelMap
|
|
31
32
|
});
|
|
32
33
|
super.setName('Groq');
|
|
33
34
|
this.setOptions(_options);
|
package/cjs/ai/groq/api.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../ai/groq/api.ts"],"names":[],"mappings":";;;AAAA,4DAAmE;AACnE,wCAAmD;AACnD,6CAA8C;AAI9C,yCAA2C;AAI3C,MAAM,qBAAqB,GAAG,GAAqB,EAAE,CACnD,eAAe,CAAC;IACd,KAAK,EAAE,wBAAa,CAAC,UAAU;IAC/B,GAAG,IAAA,+BAAqB,GAAE;CAC3B,CAAC,CAAC;
|
|
1
|
+
{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../ai/groq/api.ts"],"names":[],"mappings":";;;AAAA,4DAAmE;AACnE,wCAAmD;AACnD,6CAA8C;AAI9C,yCAA2C;AAI3C,MAAM,qBAAqB,GAAG,GAAqB,EAAE,CACnD,eAAe,CAAC;IACd,KAAK,EAAE,wBAAa,CAAC,UAAU;IAC/B,GAAG,IAAA,+BAAqB,GAAE;CAC3B,CAAC,CAAC;AAUL,MAAa,QAAS,SAAQ,mBAAU;IACtC,YAAY,EACV,MAAM,EACN,MAAM,EACN,OAAO,EACP,QAAQ,EAC6B;QACrC,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC1C,CAAC;QACD,MAAM,OAAO,GAAG;YACd,GAAG,qBAAqB,EAAE;YAC1B,GAAG,MAAM;SACV,CAAC;QAEF,MAAM,QAAQ,GAAG;YACf,GAAG,OAAO;YACV,cAAc,EAAE,KAAK;SACtB,CAAC;QAEF,KAAK,CAAC;YACJ,MAAM;YACN,MAAM,EAAE,OAAO;YACf,OAAO,EAAE,QAAQ;YACjB,MAAM,EAAE,gCAAgC;YACxC,SAAS,EAAE,EAAE;YACb,QAAQ;SACT,CAAC,CAAC;QAEH,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACtB,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;IAC5B,CAAC;IAEQ,UAAU,GAAG,CAAC,OAAqC,EAAE,EAAE;QAC9D,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACjD,KAAK,CAAC,UAAU,CAAC,EAAE,GAAG,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;IAChD,CAAC,CAAC;IAEM,cAAc,GAAG,CAAC,OAA0C,EAAE,EAAE;QACtE,IAAI,OAAO,EAAE,WAAW,EAAE,CAAC;YACzB,OAAO,OAAO,CAAC,WAAW,CAAC;QAC7B,CAAC;QAED,MAAM,YAAY,GAAG,OAAO,EAAE,eAAe,IAAI,IAAI,CAAC;QACtD,MAAM,EAAE,GAAG,IAAI,uCAAuB,CAAC,YAAY,EAAE,YAAY,GAAG,EAAE,EAAE;YACtE,KAAK,EAAE,OAAO,EAAE,KAAK;SACtB,CAAC,CAAC;QAEH,MAAM,MAAM,GAA0B,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE;YACzD,MAAM,WAAW,GAAG,IAAI,CAAC,UAAU,EAAE,WAAW,IAAI,CAAC,CAAC;YACtD,MAAM,EAAE,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YAC9B,OAAO,MAAM,IAAI,EAAE,CAAC;QACtB,CAAC,CAAC;QAEF,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;CACH;AAxDD,4BAwDC"}
|
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
import type { API } from '../../util/apicall.js';
|
|
2
2
|
import { AxBaseAI } from '../base.js';
|
|
3
3
|
import type { AxAIPromptConfig, AxAIServiceOptions, AxChatRequest, AxChatResponse, AxModelConfig } from '../types.js';
|
|
4
|
-
import { type AxAIHuggingFaceConfig, type AxAIHuggingFaceRequest, type AxAIHuggingFaceResponse } from './types.js';
|
|
4
|
+
import { type AxAIHuggingFaceConfig, AxAIHuggingFaceModel, type AxAIHuggingFaceRequest, type AxAIHuggingFaceResponse } from './types.js';
|
|
5
5
|
export declare const axAIHuggingFaceDefaultConfig: () => AxAIHuggingFaceConfig;
|
|
6
6
|
export declare const axAIHuggingFaceCreativeConfig: () => AxAIHuggingFaceConfig;
|
|
7
7
|
export interface AxAIHuggingFaceArgs {
|
|
@@ -9,10 +9,11 @@ export interface AxAIHuggingFaceArgs {
|
|
|
9
9
|
apiKey: string;
|
|
10
10
|
config?: Readonly<Partial<AxAIHuggingFaceConfig>>;
|
|
11
11
|
options?: Readonly<AxAIServiceOptions>;
|
|
12
|
+
modelMap?: Record<string, AxAIHuggingFaceModel>;
|
|
12
13
|
}
|
|
13
14
|
export declare class AxAIHuggingFace extends AxBaseAI<AxAIHuggingFaceRequest, unknown, AxAIHuggingFaceResponse, unknown, unknown> {
|
|
14
15
|
private config;
|
|
15
|
-
constructor({ apiKey, config, options }: Readonly<Omit<AxAIHuggingFaceArgs, 'name'>>);
|
|
16
|
+
constructor({ apiKey, config, options, modelMap }: Readonly<Omit<AxAIHuggingFaceArgs, 'name'>>);
|
|
16
17
|
getModelConfig(): AxModelConfig;
|
|
17
18
|
generateChatReq: (req: Readonly<AxChatRequest>, _config: Readonly<AxAIPromptConfig>) => [API, AxAIHuggingFaceRequest];
|
|
18
19
|
generateChatResp: (resp: Readonly<AxAIHuggingFaceResponse>) => AxChatResponse;
|
|
@@ -16,7 +16,7 @@ const axAIHuggingFaceCreativeConfig = () => structuredClone({
|
|
|
16
16
|
exports.axAIHuggingFaceCreativeConfig = axAIHuggingFaceCreativeConfig;
|
|
17
17
|
class AxAIHuggingFace extends base_js_1.AxBaseAI {
|
|
18
18
|
config;
|
|
19
|
-
constructor({ apiKey, config, options }) {
|
|
19
|
+
constructor({ apiKey, config, options, modelMap }) {
|
|
20
20
|
if (!apiKey || apiKey === '') {
|
|
21
21
|
throw new Error('HuggingFace API key not set');
|
|
22
22
|
}
|
|
@@ -31,7 +31,8 @@ class AxAIHuggingFace extends base_js_1.AxBaseAI {
|
|
|
31
31
|
modelInfo: info_js_1.axModelInfoHuggingFace,
|
|
32
32
|
models: { model: _config.model },
|
|
33
33
|
options,
|
|
34
|
-
supportFor: { functions: false, streaming: false }
|
|
34
|
+
supportFor: { functions: false, streaming: false },
|
|
35
|
+
modelMap
|
|
35
36
|
});
|
|
36
37
|
this.config = _config;
|
|
37
38
|
}
|
|
@@ -49,7 +50,7 @@ class AxAIHuggingFace extends base_js_1.AxBaseAI {
|
|
|
49
50
|
generateChatReq = (req,
|
|
50
51
|
// eslint-disable-next-line @typescript-eslint/no-unused-vars
|
|
51
52
|
_config) => {
|
|
52
|
-
const model =
|
|
53
|
+
const model = this.config.model;
|
|
53
54
|
const functionsList = req.functions
|
|
54
55
|
? `Functions:\n${JSON.stringify(req.functions, null, 2)}\n`
|
|
55
56
|
: '';
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../ai/huggingface/api.ts"],"names":[],"mappings":";;;AACA,wCAIoB;AASpB,uCAAmD;AACnD,yCAKoB;AAEb,MAAM,4BAA4B,GAAG,GAA0B,EAAE,CACtE,eAAe,CAAC;IACd,KAAK,EAAE,+BAAoB,CAAC,mBAAmB;IAC/C,GAAG,IAAA,+BAAqB,GAAE;CAC3B,CAAC,CAAC;AAJQ,QAAA,4BAA4B,gCAIpC;AAEE,MAAM,6BAA6B,GAAG,GAA0B,EAAE,CACvE,eAAe,CAAC;IACd,KAAK,EAAE,+BAAoB,CAAC,mBAAmB;IAC/C,GAAG,IAAA,uCAA6B,GAAE;CACnC,CAAC,CAAC;AAJQ,QAAA,6BAA6B,iCAIrC;
|
|
1
|
+
{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../ai/huggingface/api.ts"],"names":[],"mappings":";;;AACA,wCAIoB;AASpB,uCAAmD;AACnD,yCAKoB;AAEb,MAAM,4BAA4B,GAAG,GAA0B,EAAE,CACtE,eAAe,CAAC;IACd,KAAK,EAAE,+BAAoB,CAAC,mBAAmB;IAC/C,GAAG,IAAA,+BAAqB,GAAE;CAC3B,CAAC,CAAC;AAJQ,QAAA,4BAA4B,gCAIpC;AAEE,MAAM,6BAA6B,GAAG,GAA0B,EAAE,CACvE,eAAe,CAAC;IACd,KAAK,EAAE,+BAAoB,CAAC,mBAAmB;IAC/C,GAAG,IAAA,uCAA6B,GAAE;CACnC,CAAC,CAAC;AAJQ,QAAA,6BAA6B,iCAIrC;AAUL,MAAa,eAAgB,SAAQ,kBAMpC;IACS,MAAM,CAAwB;IAEtC,YAAY,EACV,MAAM,EACN,MAAM,EACN,OAAO,EACP,QAAQ,EACoC;QAC5C,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;QACjD,CAAC;QACD,MAAM,OAAO,GAAG;YACd,GAAG,IAAA,oCAA4B,GAAE;YACjC,GAAG,MAAM;SACV,CAAC;QACF,KAAK,CAAC;YACJ,IAAI,EAAE,aAAa;YACnB,MAAM,EAAE,sCAAsC;YAC9C,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE;YAC9C,SAAS,EAAE,gCAAsB;YACjC,MAAM,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE;YAChC,OAAO;YACP,UAAU,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE;YAClD,QAAQ;SACT,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,CAAC,EAAE,MAAM,CAAC,CAAC;YACX,eAAe,EAAE,MAAM,CAAC,eAAe;SACvB,CAAC;IACrB,CAAC;IAEQ,eAAe,GAAG,CACzB,GAA4B;IAC5B,6DAA6D;IAC7D,OAAmC,EACJ,EAAE;QACjC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;QAEhC,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,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;gBACjB,KAAK,MAAM;oBACT,OAAO,SAAS,GAAG,CAAC,OAAO,EAAE,CAAC;gBAChC,KAAK,QAAQ;oBACX,OAAO,WAAW,GAAG,CAAC,OAAO,EAAE,CAAC;gBAClC,KAAK,UAAU;oBACb,OAAO,oBAAoB,GAAG,CAAC,MAAM,EAAE,CAAC;gBAC1C,KAAK,WAAW,CAAC,CAAC,CAAC;oBACjB,MAAM,EAAE,GAAG,GAAG,CAAC,aAAa;wBAC1B,EAAE,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE;wBACX,MAAM,IAAI,GACR,OAAO,EAAE,CAAC,QAAQ,CAAC,MAAM,KAAK,QAAQ;4BACpC,CAAC,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM;4BACpB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;wBAEzC,OAAO,GAAG,EAAE,CAAC,QAAQ,CAAC,IAAI,IAAI,IAAI,GAAG,CAAC;oBACxC,CAAC,CAAC;yBACD,IAAI,CAAC,IAAI,CAAC,CAAC;oBACd,IAAI,EAAE,EAAE,CAAC;wBACP,OAAO,cAAc,GAAG,CAAC,OAAO,kBAAkB,EAAE,EAAE,CAAC;oBACzD,CAAC;oBACD,OAAO,cAAc,GAAG,CAAC,OAAO,EAAE,CAAC;gBACrC,CAAC;gBACD;oBACE,MAAM,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC;YACpC,CAAC;YAED,uCAAuC;QACzC,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,GAA2B;YACvC,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,IAAuC,EACvB,EAAE;QAClB,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,OAAO,EAAE,IAAI,CAAC,cAAc;iBAC7B;aACF;SACF,CAAC;IACJ,CAAC,CAAC;CACH;AApID,0CAoIC"}
|
package/cjs/ai/mistral/api.d.ts
CHANGED
|
@@ -1,6 +1,7 @@
|
|
|
1
1
|
import { AxAIOpenAI } from '../openai/api.js';
|
|
2
2
|
import type { AxAIOpenAIConfig } from '../openai/types.js';
|
|
3
3
|
import type { AxAIServiceOptions } from '../types.js';
|
|
4
|
+
import { AxAIMistralEmbedModels, AxAIMistralModel } from './types.js';
|
|
4
5
|
type MistralConfig = AxAIOpenAIConfig;
|
|
5
6
|
export declare const axAIMistralDefaultConfig: () => MistralConfig;
|
|
6
7
|
export declare const axAIMistralBestConfig: () => AxAIOpenAIConfig;
|
|
@@ -9,8 +10,9 @@ export interface AxAIMistralArgs {
|
|
|
9
10
|
apiKey: string;
|
|
10
11
|
config?: Readonly<Partial<MistralConfig>>;
|
|
11
12
|
options?: Readonly<AxAIServiceOptions>;
|
|
13
|
+
modelMap?: Record<string, AxAIMistralModel | AxAIMistralEmbedModels | string>;
|
|
12
14
|
}
|
|
13
15
|
export declare class AxAIMistral extends AxAIOpenAI {
|
|
14
|
-
constructor({ apiKey, config, options }: Readonly<Omit<AxAIMistralArgs, 'name'>>);
|
|
16
|
+
constructor({ apiKey, config, options, modelMap }: Readonly<Omit<AxAIMistralArgs, 'name'>>);
|
|
15
17
|
}
|
|
16
18
|
export {};
|
package/cjs/ai/mistral/api.js
CHANGED
|
@@ -16,7 +16,7 @@ const axAIMistralBestConfig = () => structuredClone({
|
|
|
16
16
|
});
|
|
17
17
|
exports.axAIMistralBestConfig = axAIMistralBestConfig;
|
|
18
18
|
class AxAIMistral extends api_js_1.AxAIOpenAI {
|
|
19
|
-
constructor({ apiKey, config, options }) {
|
|
19
|
+
constructor({ apiKey, config, options, modelMap }) {
|
|
20
20
|
if (!apiKey || apiKey === '') {
|
|
21
21
|
throw new Error('Mistral API key not set');
|
|
22
22
|
}
|
|
@@ -29,7 +29,8 @@ class AxAIMistral extends api_js_1.AxAIOpenAI {
|
|
|
29
29
|
config: _config,
|
|
30
30
|
options,
|
|
31
31
|
apiURL: 'https://api.mistral.ai/v1',
|
|
32
|
-
modelInfo: info_js_1.axModelInfoMistral
|
|
32
|
+
modelInfo: info_js_1.axModelInfoMistral,
|
|
33
|
+
modelMap
|
|
33
34
|
});
|
|
34
35
|
super.setName('Mistral');
|
|
35
36
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../ai/mistral/api.ts"],"names":[],"mappings":";;;AAAA,wCAAmD;AACnD,6CAA8C;AAI9C,uCAA+C;AAC/C,
|
|
1
|
+
{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../ai/mistral/api.ts"],"names":[],"mappings":";;;AAAA,wCAAmD;AACnD,6CAA8C;AAI9C,uCAA+C;AAC/C,yCAAsE;AAI/D,MAAM,wBAAwB,GAAG,GAAkB,EAAE,CAC1D,eAAe,CAAC;IACd,KAAK,EAAE,2BAAgB,CAAC,YAAY;IACpC,GAAG,IAAA,+BAAqB,GAAE;CAC3B,CAAC,CAAC;AAJQ,QAAA,wBAAwB,4BAIhC;AAEE,MAAM,qBAAqB,GAAG,GAAqB,EAAE,CAC1D,eAAe,CAAC;IACd,GAAG,IAAA,gCAAwB,GAAE;IAC7B,KAAK,EAAE,2BAAgB,CAAC,YAAY;CACrC,CAAC,CAAC;AAJQ,QAAA,qBAAqB,yBAI7B;AAUL,MAAa,WAAY,SAAQ,mBAAU;IACzC,YAAY,EACV,MAAM,EACN,MAAM,EACN,OAAO,EACP,QAAQ,EACgC;QACxC,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QAC7C,CAAC;QACD,MAAM,OAAO,GAAG;YACd,GAAG,IAAA,gCAAwB,GAAE;YAC7B,GAAG,MAAM;SACV,CAAC;QACF,KAAK,CAAC;YACJ,MAAM;YACN,MAAM,EAAE,OAAO;YACf,OAAO;YACP,MAAM,EAAE,2BAA2B;YACnC,SAAS,EAAE,4BAAkB;YAC7B,QAAQ;SACT,CAAC,CAAC;QAEH,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAC3B,CAAC;CACF;AAzBD,kCAyBC"}
|
package/cjs/ai/ollama/api.d.ts
CHANGED
|
@@ -12,11 +12,12 @@ export type AxAIOllamaArgs = {
|
|
|
12
12
|
apiKey?: string;
|
|
13
13
|
config?: Readonly<Partial<AxAIOllamaAIConfig>>;
|
|
14
14
|
options?: Readonly<AxAIServiceOptions>;
|
|
15
|
+
modelMap?: Record<string, string>;
|
|
15
16
|
};
|
|
16
17
|
/**
|
|
17
18
|
* OllamaAI: AI Service
|
|
18
19
|
* @export
|
|
19
20
|
*/
|
|
20
21
|
export declare class AxAIOllama extends AxAIOpenAI {
|
|
21
|
-
constructor({ apiKey, url, config, options }: Readonly<Omit<AxAIOllamaArgs, 'name'>>);
|
|
22
|
+
constructor({ apiKey, url, config, options, modelMap }: Readonly<Omit<AxAIOllamaArgs, 'name'>>);
|
|
22
23
|
}
|
package/cjs/ai/ollama/api.js
CHANGED
|
@@ -20,7 +20,7 @@ exports.axAIOllamaDefaultCreativeConfig = axAIOllamaDefaultCreativeConfig;
|
|
|
20
20
|
* @export
|
|
21
21
|
*/
|
|
22
22
|
class AxAIOllama extends api_js_1.AxAIOpenAI {
|
|
23
|
-
constructor({ apiKey = 'not-set', url = 'http://localhost:11434', config, options }) {
|
|
23
|
+
constructor({ apiKey = 'not-set', url = 'http://localhost:11434', config, options, modelMap }) {
|
|
24
24
|
const _config = {
|
|
25
25
|
...(0, exports.axAIOllamaDefaultConfig)(),
|
|
26
26
|
...config
|
|
@@ -29,7 +29,8 @@ class AxAIOllama extends api_js_1.AxAIOpenAI {
|
|
|
29
29
|
apiKey,
|
|
30
30
|
options,
|
|
31
31
|
config: _config,
|
|
32
|
-
apiURL: new URL('/v1', url).href
|
|
32
|
+
apiURL: new URL('/v1', url).href,
|
|
33
|
+
modelMap
|
|
33
34
|
});
|
|
34
35
|
super.setName('Ollama');
|
|
35
36
|
}
|
package/cjs/ai/ollama/api.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../ai/ollama/api.ts"],"names":[],"mappings":";;;AAAA,wCAGoB;AACpB,6CAA8C;AAMvC,MAAM,uBAAuB,GAAG,GAAuB,EAAE,CAC9D,eAAe,CAAC;IACd,GAAG,IAAA,+BAAqB,GAAE;IAC1B,KAAK,EAAE,cAAc;IACrB,UAAU,EAAE,YAAY;CACzB,CAAC,CAAC;AALQ,QAAA,uBAAuB,2BAK/B;AAEE,MAAM,+BAA+B,GAAG,GAAuB,EAAE,CACtE,eAAe,CAAC;IACd,GAAG,IAAA,uCAA6B,GAAE;IAClC,KAAK,EAAE,cAAc;IACrB,UAAU,EAAE,YAAY;CACzB,CAAC,CAAC;AALQ,QAAA,+BAA+B,mCAKvC;
|
|
1
|
+
{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../ai/ollama/api.ts"],"names":[],"mappings":";;;AAAA,wCAGoB;AACpB,6CAA8C;AAMvC,MAAM,uBAAuB,GAAG,GAAuB,EAAE,CAC9D,eAAe,CAAC;IACd,GAAG,IAAA,+BAAqB,GAAE;IAC1B,KAAK,EAAE,cAAc;IACrB,UAAU,EAAE,YAAY;CACzB,CAAC,CAAC;AALQ,QAAA,uBAAuB,2BAK/B;AAEE,MAAM,+BAA+B,GAAG,GAAuB,EAAE,CACtE,eAAe,CAAC;IACd,GAAG,IAAA,uCAA6B,GAAE;IAClC,KAAK,EAAE,cAAc;IACrB,UAAU,EAAE,YAAY;CACzB,CAAC,CAAC;AALQ,QAAA,+BAA+B,mCAKvC;AAaL;;;GAGG;AACH,MAAa,UAAW,SAAQ,mBAAU;IACxC,YAAY,EACV,MAAM,GAAG,SAAS,EAClB,GAAG,GAAG,wBAAwB,EAC9B,MAAM,EACN,OAAO,EACP,QAAQ,EAC+B;QACvC,MAAM,OAAO,GAAG;YACd,GAAG,IAAA,+BAAuB,GAAE;YAC5B,GAAG,MAAM;SACV,CAAC;QACF,KAAK,CAAC;YACJ,MAAM;YACN,OAAO;YACP,MAAM,EAAE,OAAO;YACf,MAAM,EAAE,IAAI,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI;YAChC,QAAQ;SACT,CAAC,CAAC;QAEH,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAC1B,CAAC;CACF;AAtBD,gCAsBC"}
|
package/cjs/ai/openai/api.d.ts
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
import type { API } from '../../util/apicall.js';
|
|
2
2
|
import { AxBaseAI } from '../base.js';
|
|
3
3
|
import type { AxAIPromptConfig, AxAIServiceOptions, AxChatRequest, AxChatResponse, AxEmbedRequest, AxEmbedResponse, AxModelConfig, AxModelInfo } from '../types.js';
|
|
4
|
-
import { type AxAIOpenAIChatRequest, type AxAIOpenAIChatResponse, type AxAIOpenAIChatResponseDelta, type AxAIOpenAIConfig, type AxAIOpenAIEmbedRequest, type AxAIOpenAIEmbedResponse } from './types.js';
|
|
4
|
+
import { type AxAIOpenAIChatRequest, type AxAIOpenAIChatResponse, type AxAIOpenAIChatResponseDelta, type AxAIOpenAIConfig, AxAIOpenAIEmbedModel, type AxAIOpenAIEmbedRequest, type AxAIOpenAIEmbedResponse, AxAIOpenAIModel } from './types.js';
|
|
5
5
|
export declare const axAIOpenAIDefaultConfig: () => AxAIOpenAIConfig;
|
|
6
6
|
export declare const axAIOpenAIBestConfig: () => AxAIOpenAIConfig;
|
|
7
7
|
export declare const axAIOpenAICreativeConfig: () => AxAIOpenAIConfig;
|
|
@@ -15,11 +15,12 @@ export interface AxAIOpenAIArgs {
|
|
|
15
15
|
streamingUsage?: boolean;
|
|
16
16
|
}>;
|
|
17
17
|
modelInfo?: Readonly<AxModelInfo[]>;
|
|
18
|
+
modelMap?: Record<string, AxAIOpenAIModel | AxAIOpenAIEmbedModel | string>;
|
|
18
19
|
}
|
|
19
20
|
export declare class AxAIOpenAI extends AxBaseAI<AxAIOpenAIChatRequest, AxAIOpenAIEmbedRequest, AxAIOpenAIChatResponse, AxAIOpenAIChatResponseDelta, AxAIOpenAIEmbedResponse> {
|
|
20
21
|
private config;
|
|
21
22
|
private streamingUsage;
|
|
22
|
-
constructor({ apiKey, config, options, apiURL, modelInfo }: Readonly<Omit<AxAIOpenAIArgs, 'name'>>);
|
|
23
|
+
constructor({ apiKey, config, options, apiURL, modelInfo, modelMap }: Readonly<Omit<AxAIOpenAIArgs, 'name'>>);
|
|
23
24
|
getModelConfig(): AxModelConfig;
|
|
24
25
|
generateChatReq: (req: Readonly<AxChatRequest>, _config: Readonly<AxAIPromptConfig>) => [API, AxAIOpenAIChatRequest];
|
|
25
26
|
generateEmbedReq: (req: Readonly<AxEmbedRequest>) => [API, AxAIOpenAIEmbedRequest];
|
package/cjs/ai/openai/api.js
CHANGED
|
@@ -29,7 +29,7 @@ exports.axAIOpenAIFastConfig = axAIOpenAIFastConfig;
|
|
|
29
29
|
class AxAIOpenAI extends base_js_1.AxBaseAI {
|
|
30
30
|
config;
|
|
31
31
|
streamingUsage;
|
|
32
|
-
constructor({ apiKey, config, options, apiURL, modelInfo = info_js_1.axModelInfoOpenAI }) {
|
|
32
|
+
constructor({ apiKey, config, options, apiURL, modelInfo = info_js_1.axModelInfoOpenAI, modelMap }) {
|
|
33
33
|
if (!apiKey || apiKey === '') {
|
|
34
34
|
throw new Error('OpenAI API key not set');
|
|
35
35
|
}
|
|
@@ -42,9 +42,13 @@ class AxAIOpenAI extends base_js_1.AxBaseAI {
|
|
|
42
42
|
apiURL: apiURL ? apiURL : 'https://api.openai.com/v1',
|
|
43
43
|
headers: { Authorization: `Bearer ${apiKey}` },
|
|
44
44
|
modelInfo,
|
|
45
|
-
models: {
|
|
45
|
+
models: {
|
|
46
|
+
model: _config.model,
|
|
47
|
+
embedModel: _config.embedModel
|
|
48
|
+
},
|
|
46
49
|
options,
|
|
47
|
-
supportFor: { functions: true, streaming: true }
|
|
50
|
+
supportFor: { functions: true, streaming: true },
|
|
51
|
+
modelMap
|
|
48
52
|
});
|
|
49
53
|
this.config = _config;
|
|
50
54
|
this.streamingUsage = options?.streamingUsage ?? true;
|
|
@@ -65,7 +69,7 @@ class AxAIOpenAI extends base_js_1.AxBaseAI {
|
|
|
65
69
|
generateChatReq = (req,
|
|
66
70
|
// eslint-disable-next-line @typescript-eslint/no-unused-vars
|
|
67
71
|
_config) => {
|
|
68
|
-
const model =
|
|
72
|
+
const model = this.config.model;
|
|
69
73
|
if (!req.chatPrompt || req.chatPrompt.length === 0) {
|
|
70
74
|
throw new Error('Chat prompt is empty');
|
|
71
75
|
}
|
|
@@ -109,7 +113,7 @@ class AxAIOpenAI extends base_js_1.AxBaseAI {
|
|
|
109
113
|
return [apiConfig, reqValue];
|
|
110
114
|
};
|
|
111
115
|
generateEmbedReq = (req) => {
|
|
112
|
-
const model =
|
|
116
|
+
const model = this.config.embedModel;
|
|
113
117
|
if (!model) {
|
|
114
118
|
throw new Error('Embed model not set');
|
|
115
119
|
}
|
package/cjs/ai/openai/api.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../ai/openai/api.ts"],"names":[],"mappings":";;;AACA,wCAIoB;AAapB,uCAA8C;AAC9C,yCASoB;AAEb,MAAM,uBAAuB,GAAG,GAAqB,EAAE,CAC5D,eAAe,CAAC;IACd,KAAK,EAAE,0BAAe,CAAC,KAAK;IAC5B,UAAU,EAAE,+BAAoB,CAAC,mBAAmB;IACpD,GAAG,IAAA,+BAAqB,GAAE;CAC3B,CAAC,CAAC;AALQ,QAAA,uBAAuB,2BAK/B;AAEE,MAAM,oBAAoB,GAAG,GAAqB,EAAE,CACzD,eAAe,CAAC;IACd,GAAG,IAAA,+BAAuB,GAAE;IAC5B,KAAK,EAAE,0BAAe,CAAC,SAAS;CACjC,CAAC,CAAC;AAJQ,QAAA,oBAAoB,wBAI5B;AAEE,MAAM,wBAAwB,GAAG,GAAqB,EAAE,CAC7D,eAAe,CAAC;IACd,KAAK,EAAE,0BAAe,CAAC,SAAS;IAChC,UAAU,EAAE,+BAAoB,CAAC,mBAAmB;IACpD,GAAG,IAAA,uCAA6B,GAAE;CACnC,CAAC,CAAC;AALQ,QAAA,wBAAwB,4BAKhC;AAEE,MAAM,oBAAoB,GAAG,GAAqB,EAAE,CAAC,CAAC;IAC3D,GAAG,IAAA,+BAAuB,GAAE;IAC5B,KAAK,EAAE,0BAAe,CAAC,KAAK;CAC7B,CAAC,CAAC;AAHU,QAAA,oBAAoB,wBAG9B;
|
|
1
|
+
{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../ai/openai/api.ts"],"names":[],"mappings":";;;AACA,wCAIoB;AAapB,uCAA8C;AAC9C,yCASoB;AAEb,MAAM,uBAAuB,GAAG,GAAqB,EAAE,CAC5D,eAAe,CAAC;IACd,KAAK,EAAE,0BAAe,CAAC,KAAK;IAC5B,UAAU,EAAE,+BAAoB,CAAC,mBAAmB;IACpD,GAAG,IAAA,+BAAqB,GAAE;CAC3B,CAAC,CAAC;AALQ,QAAA,uBAAuB,2BAK/B;AAEE,MAAM,oBAAoB,GAAG,GAAqB,EAAE,CACzD,eAAe,CAAC;IACd,GAAG,IAAA,+BAAuB,GAAE;IAC5B,KAAK,EAAE,0BAAe,CAAC,SAAS;CACjC,CAAC,CAAC;AAJQ,QAAA,oBAAoB,wBAI5B;AAEE,MAAM,wBAAwB,GAAG,GAAqB,EAAE,CAC7D,eAAe,CAAC;IACd,KAAK,EAAE,0BAAe,CAAC,SAAS;IAChC,UAAU,EAAE,+BAAoB,CAAC,mBAAmB;IACpD,GAAG,IAAA,uCAA6B,GAAE;CACnC,CAAC,CAAC;AALQ,QAAA,wBAAwB,4BAKhC;AAEE,MAAM,oBAAoB,GAAG,GAAqB,EAAE,CAAC,CAAC;IAC3D,GAAG,IAAA,+BAAuB,GAAE;IAC5B,KAAK,EAAE,0BAAe,CAAC,KAAK;CAC7B,CAAC,CAAC;AAHU,QAAA,oBAAoB,wBAG9B;AAYH,MAAa,UAAW,SAAQ,kBAM/B;IACS,MAAM,CAAmB;IACzB,cAAc,CAAU;IAEhC,YAAY,EACV,MAAM,EACN,MAAM,EACN,OAAO,EACP,MAAM,EACN,SAAS,GAAG,2BAAiB,EAC7B,QAAQ,EAC+B;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,IAAA,+BAAuB,GAAE;YAC5B,GAAG,MAAM;SACV,CAAC;QAEF,KAAK,CAAC;YACJ,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,2BAA2B;YACrD,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE;YAC9C,SAAS;YACT,MAAM,EAAE;gBACN,KAAK,EAAE,OAAO,CAAC,KAAe;gBAC9B,UAAU,EAAE,OAAO,CAAC,UAAoB;aACzC;YACD,OAAO;YACP,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE;YAChD,QAAQ;SACT,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC;QACtB,IAAI,CAAC,cAAc,GAAG,OAAO,EAAE,cAAc,IAAI,IAAI,CAAC;IACxD,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,eAAe,EAAE,MAAM,CAAC,eAAe;YACvC,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;YACzC,aAAa,EAAE,MAAM,CAAC,aAAa;YACnC,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,CAAC,EAAE,MAAM,CAAC,CAAC;YACX,MAAM,EAAE,MAAM,CAAC,MAAM;SACtB,CAAC;IACJ,CAAC;IAEQ,eAAe,GAAG,CACzB,GAA4B;IAC5B,6DAA6D;IAC7D,OAAmC,EACL,EAAE;QAChC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;QAEhC,IAAI,CAAC,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnD,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC1C,CAAC;QAED,MAAM,SAAS,GAAG;YAChB,IAAI,EAAE,mBAAmB;SAC1B,CAAC;QAEF,MAAM,KAAK,GAAG,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACvC,IAAI,EAAE,UAAmB;YACzB,QAAQ,EAAE;gBACR,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;gBAC1B,UAAU,EAAE,CAAC,CAAC,UAAU;aACzB;SACF,CAAC,CAAC,CAAC;QAEJ,MAAM,WAAW,GACf,CAAC,GAAG,CAAC,YAAY,IAAI,GAAG,CAAC,SAAS,IAAI,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC;YAC5D,CAAC,CAAC,MAAM;YACR,CAAC,CAAC,GAAG,CAAC,YAAY,CAAC;QAEvB,MAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;QAErC,MAAM,gBAAgB,GACpB,GAAG,CAAC,WAAW,EAAE,gBAAgB,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC;QAEpE,MAAM,MAAM,GAAG,GAAG,CAAC,WAAW,EAAE,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;QAE7D,MAAM,QAAQ,GAA0B;YACtC,KAAK;YACL,QAAQ;YACR,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,cAAc;gBAC1C,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,cAAc,EAAE;gBACvC,CAAC,CAAC,SAAS;YACb,KAAK;YACL,WAAW,EAAE,WAAW;YACxB,UAAU,EAAE,GAAG,CAAC,WAAW,EAAE,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,GAAG;YACtE,WAAW,EAAE,GAAG,CAAC,WAAW,EAAE,WAAW,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW;YACpE,KAAK,EAAE,GAAG,CAAC,WAAW,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC;YACrD,CAAC,EAAE,GAAG,CAAC,WAAW,EAAE,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,CAAC;YACtC,IAAI,EAAE,GAAG,CAAC,WAAW,EAAE,aAAa,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI;YACxD,gBAAgB,EACd,GAAG,CAAC,WAAW,EAAE,eAAe,IAAI,IAAI,CAAC,MAAM,CAAC,eAAe;YACjE,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS;YACjC,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACpE,GAAG,CAAC,MAAM,IAAI,IAAI,CAAC,cAAc;gBAC/B,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,EAAE;gBAC3D,CAAC,CAAC,EAAE,CAAC;SACR,CAAC;QAEF,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAC/B,CAAC,CAAC;IAEO,gBAAgB,GAAG,CAC1B,GAA6B,EACE,EAAE;QACjC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;QAErC,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,aAAa;SACpB,CAAC;QAEF,MAAM,QAAQ,GAAG;YACf,KAAK,EAAE,KAAK;YACZ,KAAK,EAAE,GAAG,CAAC,KAAK;SACjB,CAAC;QAEF,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAC/B,CAAC,CAAC;IAEO,gBAAgB,GAAG,CAC1B,IAAsC,EACtB,EAAE;QAClB,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC;QAE3C,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,KAAK,CAAC;QACd,CAAC;QAED,MAAM,UAAU,GAAG,KAAK;YACtB,CAAC,CAAC;gBACE,YAAY,EAAE,KAAK,CAAC,aAAa;gBACjC,gBAAgB,EAAE,KAAK,CAAC,iBAAiB;gBACzC,WAAW,EAAE,KAAK,CAAC,YAAY;aAChC;YACH,CAAC,CAAC,SAAS,CAAC;QAEd,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;YACrC,MAAM,YAAY,GAAG,eAAe,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YAE3D,MAAM,aAAa,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,GAAG,CAClD,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;gBAClD,EAAE,EAAE,EAAE;gBACN,IAAI,EAAE,UAAmB;gBACzB,QAAQ,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;aAC3B,CAAC,CACH,CAAC;YAEF,OAAO;gBACL,EAAE,EAAE,GAAG,MAAM,CAAC,KAAK,EAAE;gBACrB,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO;gBAC/B,aAAa;gBACb,YAAY;aACb,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,OAAO;YACL,UAAU;YACV,OAAO;YACP,QAAQ,EAAE,EAAE;SACb,CAAC;IACJ,CAAC,CAAC;IAEO,sBAAsB,GAAG,CAChC,IAA2C,EAC3C,KAAa,EACG,EAAE;QAClB,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;QAEpC,MAAM,UAAU,GAAG,KAAK;YACtB,CAAC,CAAC;gBACE,YAAY,EAAE,KAAK,CAAC,aAAa;gBACjC,gBAAgB,EAAE,KAAK,CAAC,iBAAiB;gBACzC,WAAW,EAAE,KAAK,CAAC,YAAY;aAChC;YACH,CAAC,CAAC,SAAS,CAAC;QAEd,MAAM,MAAM,GAAG,KAEd,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;YACvB,MAAM,CAAC,UAAU,GAAG,EAAE,CAAC;QACzB,CAAC;QAED,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG;QACzB,gEAAgE;QAChE,CAAC,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE;YAC1D,MAAM,YAAY,GAAG,eAAe,CAAC,aAAa,CAAC,CAAC;YAEpD,MAAM,aAAa,GAAG,UAAU;gBAC9B,EAAE,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE;gBAClE,IACE,OAAO,GAAG,KAAK,QAAQ;oBACvB,OAAO,KAAK,KAAK,QAAQ;oBACzB,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,EACzB,CAAC;oBACD,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,GAAG,GAAG,CAAC;gBACjC,CAAC;gBAED,MAAM,EAAE,GAAG,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;gBACpC,IAAI,CAAC,EAAE,EAAE,CAAC;oBACR,OAAO,IAAI,CAAC;gBACd,CAAC;gBAED,OAAO;oBACL,EAAE;oBACF,IAAI,EAAE,UAAmB;oBACzB,QAAQ,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;iBAC3B,CAAC;YACJ,CAAC,CAAC;iBACD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC;YAE7B,OAAO;gBACL,OAAO;gBACP,IAAI,EAAE,IAAI;gBACV,aAAa;gBACb,YAAY;gBACZ,EAAE;aACH,CAAC;QACJ,CAAC,CACF,CAAC;QAEF,OAAO;YACL,OAAO;YACP,UAAU;SACX,CAAC;IACJ,CAAC,CAAC;IAEO,iBAAiB,GAAG,CAC3B,IAAuC,EACtB,EAAE;QACnB,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC;QAE7B,MAAM,UAAU,GAAG,KAAK;YACtB,CAAC,CAAC;gBACE,YAAY,EAAE,KAAK,CAAC,aAAa;gBACjC,gBAAgB,EAAE,KAAK,CAAC,iBAAiB;gBACzC,WAAW,EAAE,KAAK,CAAC,YAAY;aAChC;YACH,CAAC,CAAC,SAAS,CAAC;QAEd,OAAO;YACL,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;YACxC,UAAU;SACX,CAAC;IACJ,CAAC,CAAC;CACH;AA9QD,gCA8QC;AAED,MAAM,eAAe,GAAG,CACtB,YAAmE,EAC7B,EAAE;IACxC,QAAQ,YAAY,EAAE,CAAC;QACrB,KAAK,MAAM;YACT,OAAO,MAAe,CAAC;QACzB,KAAK,QAAQ;YACX,OAAO,QAAiB,CAAC;QAC3B,KAAK,gBAAgB;YACnB,OAAO,OAAgB,CAAC;QAC1B,KAAK,YAAY;YACf,OAAO,eAAwB,CAAC;IACpC,CAAC;AACH,CAAC,CAAC;AAEF,SAAS,cAAc,CACrB,GAA4B;IAE5B,OAAO,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QAChC,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;YACjB,KAAK,QAAQ;gBACX,OAAO,EAAE,IAAI,EAAE,QAAiB,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;YAC3D,KAAK,MAAM;gBACT,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC/B,OAAO;wBACL,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,GAAG,CAAC,IAAI;wBACd,OAAO,EAAE,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;4BAC7B,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;gCACf,KAAK,MAAM;oCACT,OAAO,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;gCACjD,KAAK,OAAO,CAAC,CAAC,CAAC;oCACb,MAAM,GAAG,GAAG,QAAQ,CAAC,CAAC,QAAQ,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC;oCACnD,OAAO;wCACL,IAAI,EAAE,WAAoB;wCAC1B,SAAS,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,IAAI,MAAM,EAAE;qCACjD,CAAC;gCACJ,CAAC;gCACD;oCACE,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;4BAC5C,CAAC;wBACH,CAAC,CAAC;qBACH,CAAC;gBACJ,CAAC;gBACD,OAAO,EAAE,IAAI,EAAE,MAAe,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,CAAC;YACzE,KAAK,WAAW;gBACd,OAAO;oBACL,IAAI,EAAE,WAAoB;oBAC1B,OAAO,EAAE,GAAG,CAAC,OAAiB;oBAC9B,IAAI,EAAE,GAAG,CAAC,IAAI;oBACd,UAAU,EAAE,GAAG,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;wBACzC,EAAE,EAAE,CAAC,CAAC,EAAE;wBACR,IAAI,EAAE,UAAmB;wBACzB,QAAQ,EAAE;4BACR,IAAI,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI;4BACrB,SAAS,EACP,OAAO,CAAC,CAAC,QAAQ,CAAC,MAAM,KAAK,QAAQ;gCACnC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC;gCACnC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM;yBACxB;qBACF,CAAC,CAAC;iBACJ,CAAC;YACJ,KAAK,UAAU;gBACb,OAAO;oBACL,IAAI,EAAE,MAAe;oBACrB,OAAO,EAAE,GAAG,CAAC,MAAM;oBACnB,YAAY,EAAE,GAAG,CAAC,UAAU;iBAC7B,CAAC;YACJ;gBACE,MAAM,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC;QACpC,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC"}
|
package/cjs/ai/together/api.d.ts
CHANGED
|
@@ -8,8 +8,9 @@ export interface AxAITogetherArgs {
|
|
|
8
8
|
apiKey: string;
|
|
9
9
|
config?: Readonly<Partial<TogetherAIConfig>>;
|
|
10
10
|
options?: Readonly<AxAIServiceOptions>;
|
|
11
|
+
modelMap?: Record<string, string>;
|
|
11
12
|
}
|
|
12
13
|
export declare class AxAITogether extends AxAIOpenAI {
|
|
13
|
-
constructor({ apiKey, config, options }: Readonly<Omit<AxAITogetherArgs, 'name'>>);
|
|
14
|
+
constructor({ apiKey, config, options, modelMap }: Readonly<Omit<AxAITogetherArgs, 'name'>>);
|
|
14
15
|
}
|
|
15
16
|
export {};
|
package/cjs/ai/together/api.js
CHANGED
|
@@ -11,7 +11,7 @@ const axAITogetherDefaultConfig = () => structuredClone({
|
|
|
11
11
|
});
|
|
12
12
|
exports.axAITogetherDefaultConfig = axAITogetherDefaultConfig;
|
|
13
13
|
class AxAITogether extends api_js_1.AxAIOpenAI {
|
|
14
|
-
constructor({ apiKey, config, options }) {
|
|
14
|
+
constructor({ apiKey, config, options, modelMap }) {
|
|
15
15
|
if (!apiKey || apiKey === '') {
|
|
16
16
|
throw new Error('Together API key not set');
|
|
17
17
|
}
|
|
@@ -24,7 +24,8 @@ class AxAITogether extends api_js_1.AxAIOpenAI {
|
|
|
24
24
|
config: _config,
|
|
25
25
|
options,
|
|
26
26
|
apiURL: 'https://api.together.xyz/v1',
|
|
27
|
-
modelInfo: info_js_1.axModelInfoTogether
|
|
27
|
+
modelInfo: info_js_1.axModelInfoTogether,
|
|
28
|
+
modelMap
|
|
28
29
|
});
|
|
29
30
|
super.setName('Together');
|
|
30
31
|
}
|