@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.
- package/README.md +106 -99
- package/build/module/docs/assets/navigation.js +1 -1
- package/build/module/docs/assets/navigation.js.map +1 -1
- package/build/module/docs/assets/search.js +1 -1
- package/build/module/docs/assets/search.js.map +1 -1
- package/build/module/src/ai/anthropic/api.d.ts +13 -19
- package/build/module/src/ai/anthropic/api.js +165 -72
- package/build/module/src/ai/anthropic/api.js.map +1 -1
- package/build/module/src/ai/anthropic/info.d.ts +2 -2
- package/build/module/src/ai/anthropic/info.js +7 -7
- package/build/module/src/ai/anthropic/info.js.map +1 -1
- package/build/module/src/ai/anthropic/types.d.ts +63 -52
- package/build/module/src/ai/anthropic/types.js +8 -12
- package/build/module/src/ai/anthropic/types.js.map +1 -1
- package/build/module/src/ai/azure-openai/api.d.ts +12 -28
- package/build/module/src/ai/azure-openai/api.js +6 -22
- package/build/module/src/ai/azure-openai/api.js.map +1 -1
- package/build/module/src/ai/balance.d.ts +9 -12
- package/build/module/src/ai/balance.js +1 -1
- package/build/module/src/ai/balance.js.map +1 -1
- package/build/module/src/ai/base.d.ts +26 -29
- package/build/module/src/ai/base.js +37 -55
- package/build/module/src/ai/base.js.map +1 -1
- package/build/module/src/ai/cohere/api.d.ts +16 -30
- package/build/module/src/ai/cohere/api.js +26 -32
- package/build/module/src/ai/cohere/api.js.map +1 -1
- package/build/module/src/ai/cohere/info.d.ts +2 -2
- package/build/module/src/ai/cohere/info.js +10 -10
- package/build/module/src/ai/cohere/info.js.map +1 -1
- package/build/module/src/ai/cohere/types.d.ts +13 -13
- package/build/module/src/ai/cohere/types.js +14 -14
- package/build/module/src/ai/cohere/types.js.map +1 -1
- package/build/module/src/ai/deepseek/api.d.ts +10 -22
- package/build/module/src/ai/deepseek/api.js +14 -24
- package/build/module/src/ai/deepseek/api.js.map +1 -1
- package/build/module/src/ai/deepseek/info.d.ts +2 -2
- package/build/module/src/ai/deepseek/info.js +4 -4
- package/build/module/src/ai/deepseek/info.js.map +1 -1
- package/build/module/src/ai/deepseek/types.d.ts +1 -1
- package/build/module/src/ai/deepseek/types.js +5 -5
- package/build/module/src/ai/deepseek/types.js.map +1 -1
- package/build/module/src/ai/google-gemini/api.d.ts +18 -20
- package/build/module/src/ai/google-gemini/api.js +93 -67
- package/build/module/src/ai/google-gemini/api.js.map +1 -1
- package/build/module/src/ai/google-gemini/info.d.ts +3 -3
- package/build/module/src/ai/google-gemini/info.js +6 -6
- package/build/module/src/ai/google-gemini/info.js.map +1 -1
- package/build/module/src/ai/google-gemini/types.d.ts +41 -48
- package/build/module/src/ai/google-gemini/types.js +25 -37
- package/build/module/src/ai/google-gemini/types.js.map +1 -1
- package/build/module/src/ai/groq/api.d.ts +9 -18
- package/build/module/src/ai/groq/api.js +11 -17
- package/build/module/src/ai/groq/api.js.map +1 -1
- package/build/module/src/ai/groq/types.d.ts +6 -0
- package/build/module/src/ai/groq/types.js +8 -0
- package/build/module/src/ai/groq/types.js.map +1 -0
- package/build/module/src/ai/huggingface/api.d.ts +13 -27
- package/build/module/src/ai/huggingface/api.js +12 -24
- package/build/module/src/ai/huggingface/api.js.map +1 -1
- package/build/module/src/ai/huggingface/info.d.ts +2 -2
- package/build/module/src/ai/huggingface/info.js +1 -1
- package/build/module/src/ai/huggingface/info.js.map +1 -1
- package/build/module/src/ai/huggingface/types.d.ts +7 -15
- package/build/module/src/ai/huggingface/types.js +4 -8
- package/build/module/src/ai/huggingface/types.js.map +1 -1
- package/build/module/src/ai/index.d.ts +13 -13
- package/build/module/src/ai/index.js +23 -23
- package/build/module/src/ai/index.js.map +1 -1
- package/build/module/src/ai/mistral/api.d.ts +10 -22
- package/build/module/src/ai/mistral/api.js +14 -24
- package/build/module/src/ai/mistral/api.js.map +1 -1
- package/build/module/src/ai/mistral/info.d.ts +2 -2
- package/build/module/src/ai/mistral/info.js +7 -7
- package/build/module/src/ai/mistral/info.js.map +1 -1
- package/build/module/src/ai/mistral/types.d.ts +2 -10
- package/build/module/src/ai/mistral/types.js +12 -20
- package/build/module/src/ai/mistral/types.js.map +1 -1
- package/build/module/src/ai/ollama/api.d.ts +13 -16
- package/build/module/src/ai/ollama/api.js +17 -13
- package/build/module/src/ai/ollama/api.js.map +1 -1
- package/build/module/src/ai/openai/api.d.ts +19 -40
- package/build/module/src/ai/openai/api.js +90 -135
- package/build/module/src/ai/openai/api.js.map +1 -1
- package/build/module/src/ai/openai/info.d.ts +2 -2
- package/build/module/src/ai/openai/info.js +9 -9
- package/build/module/src/ai/openai/info.js.map +1 -1
- package/build/module/src/ai/openai/types.d.ts +29 -56
- package/build/module/src/ai/openai/types.js +17 -34
- package/build/module/src/ai/openai/types.js.map +1 -1
- package/build/module/src/ai/together/api.d.ts +9 -17
- package/build/module/src/ai/together/api.js +9 -15
- package/build/module/src/ai/together/api.js.map +1 -1
- package/build/module/src/ai/together/info.d.ts +2 -2
- package/build/module/src/ai/together/info.js +1 -1
- package/build/module/src/ai/together/info.js.map +1 -1
- package/build/module/src/ai/types.d.ts +112 -25
- package/build/module/src/ai/util.d.ts +4 -4
- package/build/module/src/ai/util.js +6 -3
- package/build/module/src/ai/util.js.map +1 -1
- package/build/module/src/db/base.d.ts +14 -15
- package/build/module/src/db/base.js +26 -29
- package/build/module/src/db/base.js.map +1 -1
- package/build/module/src/db/cloudflare.d.ts +10 -9
- package/build/module/src/db/cloudflare.js +2 -2
- package/build/module/src/db/cloudflare.js.map +1 -1
- package/build/module/src/db/index.d.ts +6 -7
- package/build/module/src/db/index.js +9 -9
- package/build/module/src/db/index.js.map +1 -1
- package/build/module/src/db/memory.d.ts +13 -11
- package/build/module/src/db/memory.js +2 -2
- package/build/module/src/db/memory.js.map +1 -1
- package/build/module/src/db/pinecone.d.ts +10 -9
- package/build/module/src/db/pinecone.js +2 -2
- package/build/module/src/db/pinecone.js.map +1 -1
- package/build/module/src/db/types.d.ts +9 -9
- package/build/module/src/db/weaviate.d.ts +10 -9
- package/build/module/src/db/weaviate.js +2 -2
- package/build/module/src/db/weaviate.js.map +1 -1
- package/build/module/src/docs/manager.d.ts +16 -16
- package/build/module/src/docs/manager.js +1 -1
- package/build/module/src/docs/manager.js.map +1 -1
- package/build/module/src/docs/reranker.d.ts +6 -6
- package/build/module/src/docs/reranker.js +3 -3
- package/build/module/src/docs/reranker.js.map +1 -1
- package/build/module/src/docs/rewriter.d.ts +3 -3
- package/build/module/src/docs/rewriter.js +2 -2
- package/build/module/src/docs/rewriter.js.map +1 -1
- package/build/module/src/docs/tika.d.ts +5 -8
- package/build/module/src/docs/tika.js +2 -2
- package/build/module/src/docs/tika.js.map +1 -1
- package/build/module/src/dsp/asserts.d.ts +8 -8
- package/build/module/src/dsp/asserts.js +9 -6
- package/build/module/src/dsp/asserts.js.map +1 -1
- package/build/module/src/dsp/eval.d.ts +9 -28
- package/build/module/src/dsp/eval.js +10 -5
- package/build/module/src/dsp/eval.js.map +1 -1
- package/build/module/src/dsp/evaluate.d.ts +8 -8
- package/build/module/src/dsp/evaluate.js +1 -1
- package/build/module/src/dsp/evaluate.js.map +1 -1
- package/build/module/src/dsp/extract.d.ts +6 -6
- package/build/module/src/dsp/extract.js +1 -0
- package/build/module/src/dsp/extract.js.map +1 -1
- package/build/module/src/dsp/generate.d.ts +24 -25
- package/build/module/src/dsp/generate.js +24 -23
- package/build/module/src/dsp/generate.js.map +1 -1
- package/build/module/src/dsp/generate.test.js +4 -4
- package/build/module/src/dsp/generate.test.js.map +1 -1
- package/build/module/src/dsp/index.d.ts +4 -4
- package/build/module/src/dsp/loader.d.ts +5 -5
- package/build/module/src/dsp/loader.js +1 -1
- package/build/module/src/dsp/loader.js.map +1 -1
- package/build/module/src/dsp/optimize.d.ts +12 -12
- package/build/module/src/dsp/optimize.js +5 -5
- package/build/module/src/dsp/optimize.js.map +1 -1
- package/build/module/src/dsp/parser.d.ts +2 -1
- package/build/module/src/dsp/parser.js +178 -73
- package/build/module/src/dsp/parser.js.map +1 -1
- package/build/module/src/dsp/program.d.ts +38 -37
- package/build/module/src/dsp/program.js +4 -53
- package/build/module/src/dsp/program.js.map +1 -1
- package/build/module/src/dsp/prompt.d.ts +36 -13
- package/build/module/src/dsp/prompt.js +123 -51
- package/build/module/src/dsp/prompt.js.map +1 -1
- package/build/module/src/dsp/registry.d.ts +1 -1
- package/build/module/src/dsp/registry.js +1 -1
- package/build/module/src/dsp/registry.js.map +1 -1
- package/build/module/src/dsp/router.d.ts +7 -7
- package/build/module/src/dsp/router.js +4 -4
- package/build/module/src/dsp/router.js.map +1 -1
- package/build/module/src/dsp/sig.d.ts +13 -13
- package/build/module/src/dsp/sig.js +7 -3
- package/build/module/src/dsp/sig.js.map +1 -1
- package/build/module/src/dsp/sig.test.js +4 -1
- package/build/module/src/dsp/sig.test.js.map +1 -1
- package/build/module/src/dsp/stopwords.js +1 -0
- package/build/module/src/dsp/stopwords.js.map +1 -1
- package/build/module/src/dsp/strutil.d.ts +10 -8
- package/build/module/src/dsp/strutil.js +14 -6
- package/build/module/src/dsp/strutil.js.map +1 -1
- package/build/module/src/dsp/util.d.ts +4 -0
- package/build/module/src/dsp/util.js +62 -0
- package/build/module/src/dsp/util.js.map +1 -1
- package/build/module/src/examples/agent.js +7 -5
- package/build/module/src/examples/agent.js.map +1 -1
- package/build/module/src/examples/chain-of-thought.js +5 -3
- package/build/module/src/examples/chain-of-thought.js.map +1 -1
- package/build/module/src/examples/customer-support.js +6 -4
- package/build/module/src/examples/customer-support.js.map +1 -1
- package/build/module/src/examples/fibonacci.js +6 -4
- package/build/module/src/examples/fibonacci.js.map +1 -1
- package/build/module/src/examples/food-search.js +8 -8
- package/build/module/src/examples/food-search.js.map +1 -1
- package/build/module/src/examples/marketing.js +5 -3
- package/build/module/src/examples/marketing.js.map +1 -1
- package/build/module/src/examples/multi-modal.d.ts +1 -0
- package/build/module/src/examples/multi-modal.js +16 -0
- package/build/module/src/examples/multi-modal.js.map +1 -0
- package/build/module/src/examples/qna-tune.js +9 -7
- package/build/module/src/examples/qna-tune.js.map +1 -1
- package/build/module/src/examples/qna-use-tuned.js +8 -6
- package/build/module/src/examples/qna-use-tuned.js.map +1 -1
- package/build/module/src/examples/rag-docs.js +7 -5
- package/build/module/src/examples/rag-docs.js.map +1 -1
- package/build/module/src/examples/rag.js +6 -4
- package/build/module/src/examples/rag.js.map +1 -1
- package/build/module/src/examples/react.js +5 -3
- package/build/module/src/examples/react.js.map +1 -1
- package/build/module/src/examples/routing.js +9 -7
- package/build/module/src/examples/routing.js.map +1 -1
- package/build/module/src/examples/smart-home.d.ts +10 -0
- package/build/module/src/examples/smart-home.js +106 -0
- package/build/module/src/examples/smart-home.js.map +1 -0
- package/build/module/src/examples/streaming1.js +7 -5
- package/build/module/src/examples/streaming1.js.map +1 -1
- package/build/module/src/examples/streaming2.js +9 -7
- package/build/module/src/examples/streaming2.js.map +1 -1
- package/build/module/src/examples/summarize.js +5 -5
- package/build/module/src/examples/summarize.js.map +1 -1
- package/build/module/src/examples/vectordb.js +6 -4
- package/build/module/src/examples/vectordb.js.map +1 -1
- package/build/module/src/funcs/code.d.ts +3 -4
- package/build/module/src/funcs/code.js +15 -15
- package/build/module/src/funcs/code.js.map +1 -1
- package/build/module/src/funcs/embed.d.ts +3 -5
- package/build/module/src/funcs/embed.js +2 -2
- package/build/module/src/funcs/embed.js.map +1 -1
- package/build/module/src/funcs/functions.d.ts +18 -0
- package/build/module/src/funcs/functions.js +55 -0
- package/build/module/src/funcs/functions.js.map +1 -0
- package/build/module/src/funcs/index.d.ts +1 -0
- package/build/module/src/funcs/index.js +1 -0
- package/build/module/src/funcs/index.js.map +1 -1
- package/build/module/src/funcs/jsonschema.d.ts +2 -0
- package/build/module/src/funcs/jsonschema.js +97 -0
- package/build/module/src/funcs/jsonschema.js.map +1 -0
- package/build/module/src/index.d.ts +1 -2
- package/build/module/src/index.js +1 -2
- package/build/module/src/index.js.map +1 -1
- package/build/module/src/mem/index.d.ts +2 -0
- package/build/module/src/mem/index.js +3 -0
- package/build/module/src/mem/index.js.map +1 -0
- package/build/module/src/mem/memory.d.ts +16 -0
- package/build/module/src/mem/memory.js +74 -0
- package/build/module/src/mem/memory.js.map +1 -0
- package/build/module/src/mem/types.d.ts +10 -0
- package/build/module/src/mem/types.js +2 -0
- package/build/module/src/mem/types.js.map +1 -0
- package/build/module/src/prompts/agent.d.ts +14 -15
- package/build/module/src/prompts/agent.js +10 -11
- package/build/module/src/prompts/agent.js.map +1 -1
- package/build/module/src/prompts/cot.d.ts +6 -6
- package/build/module/src/prompts/cot.js +4 -4
- package/build/module/src/prompts/cot.js.map +1 -1
- package/build/module/src/prompts/prompts.test.js +9 -5
- package/build/module/src/prompts/prompts.test.js.map +1 -1
- package/build/module/src/prompts/rag.d.ts +7 -7
- package/build/module/src/prompts/rag.js +6 -6
- package/build/module/src/prompts/rag.js.map +1 -1
- package/build/module/src/prompts/react.d.ts +7 -7
- package/build/module/src/prompts/react.js +4 -4
- package/build/module/src/prompts/react.js.map +1 -1
- package/build/module/src/text/functions.d.ts +3 -20
- package/build/module/src/text/functions.js.map +1 -1
- package/build/module/src/text/jsonschema.d.ts +1 -2
- package/build/module/src/text/memory.d.ts +8 -8
- package/build/module/src/text/memory.js.map +1 -1
- package/build/module/src/text/types.d.ts +14 -14
- package/build/module/src/trace/index.d.ts +50 -50
- package/build/module/src/trace/index.js +30 -30
- package/build/module/src/trace/index.js.map +1 -1
- package/build/module/src/types/index.d.ts +9 -39
- package/build/module/src/types/index.js +1 -1
- package/build/module/src/types/index.js.map +1 -1
- package/build/module/src/util/apicall.d.ts +2 -5
- package/build/module/src/util/apicall.js +4 -3
- package/build/module/src/util/apicall.js.map +1 -1
- package/build/module/src/util/log.js +1 -0
- package/build/module/src/util/log.js.map +1 -1
- package/build/module/src/util/other.js +1 -0
- package/build/module/src/util/other.js.map +1 -1
- package/build/module/src/util/stream.d.ts +0 -1
- package/build/module/src/util/stream.js +1 -0
- package/build/module/src/util/stream.js.map +1 -1
- package/build/module/src/util/transform.d.ts +0 -1
- package/build/module/src/util/transform.js +2 -0
- package/build/module/src/util/transform.js.map +1 -1
- package/package.json +2 -4
|
@@ -1,47 +1,40 @@
|
|
|
1
|
-
|
|
2
|
-
|
|
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
|
|
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
|
|
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
|
|
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
|
|
57
|
+
export type AxGoogleGeminiToolFunctionDeclaration = {
|
|
65
58
|
name: string;
|
|
66
59
|
description?: string;
|
|
67
60
|
parameters?: object;
|
|
68
61
|
};
|
|
69
|
-
export type
|
|
70
|
-
functionDeclarations:
|
|
62
|
+
export type AxGoogleGeminiTool = {
|
|
63
|
+
functionDeclarations: AxGoogleGeminiToolFunctionDeclaration[];
|
|
71
64
|
};
|
|
72
|
-
export type
|
|
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
|
|
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
|
|
87
|
-
category:
|
|
88
|
-
threshold:
|
|
79
|
+
export type AxGoogleGeminiSafetySettings = {
|
|
80
|
+
category: AxGoogleGeminiSafetyCategory;
|
|
81
|
+
threshold: AxGoogleGeminiSafetyThreshold;
|
|
89
82
|
}[];
|
|
90
|
-
export type
|
|
91
|
-
contents:
|
|
92
|
-
tools?:
|
|
93
|
-
tool_config?:
|
|
94
|
-
systemInstruction?:
|
|
95
|
-
generationConfig:
|
|
96
|
-
safetySettings?:
|
|
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
|
|
91
|
+
export type AxGoogleGeminiChatResponse = {
|
|
99
92
|
candidates: {
|
|
100
|
-
content:
|
|
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
|
|
116
|
+
export type AxGoogleGeminiChatResponseDelta = AxGoogleGeminiChatResponse;
|
|
124
117
|
/**
|
|
125
|
-
*
|
|
118
|
+
* AxGoogleGeminiConfig: Configuration options for Google Gemini API
|
|
126
119
|
* @export
|
|
127
120
|
*/
|
|
128
|
-
export type
|
|
129
|
-
model:
|
|
130
|
-
embedModel:
|
|
131
|
-
safetySettings?:
|
|
121
|
+
export type AxGoogleGeminiConfig = AxModelConfig & {
|
|
122
|
+
model: AxGoogleGeminiModel | string;
|
|
123
|
+
embedModel: AxGoogleGeminiEmbedModels;
|
|
124
|
+
safetySettings?: AxGoogleGeminiSafetySettings;
|
|
132
125
|
};
|
|
133
126
|
/**
|
|
134
|
-
*
|
|
127
|
+
* AxGoogleGeminiEmbedRequest: Structure for making an embedding request to the Google Gemini API.
|
|
135
128
|
* @export
|
|
136
129
|
*/
|
|
137
|
-
export type
|
|
130
|
+
export type AxGoogleGeminiBatchEmbedRequest = {
|
|
138
131
|
requests: {
|
|
139
132
|
model: string;
|
|
140
133
|
text: string;
|
|
141
134
|
}[];
|
|
142
135
|
};
|
|
143
136
|
/**
|
|
144
|
-
*
|
|
137
|
+
* AxGoogleGeminiEmbedResponse: Structure for handling responses from the Google Gemini API embedding requests.
|
|
145
138
|
* @export
|
|
146
139
|
*/
|
|
147
|
-
export type
|
|
140
|
+
export type AxGoogleGeminiBatchEmbedResponse = {
|
|
148
141
|
embeddings: {
|
|
149
142
|
value: number[];
|
|
150
143
|
}[];
|
|
@@ -1,38 +1,26 @@
|
|
|
1
|
-
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
(
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
|
|
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":"
|
|
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
|
|
2
|
-
import {
|
|
3
|
-
import type {
|
|
4
|
-
type
|
|
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<
|
|
13
|
-
options?: Readonly<
|
|
7
|
+
config: Readonly<AxAxGroqAIConfig>;
|
|
8
|
+
options?: Readonly<AxAIServiceOptions>;
|
|
14
9
|
}
|
|
15
|
-
|
|
16
|
-
|
|
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 {
|
|
2
|
-
import {
|
|
3
|
-
|
|
4
|
-
|
|
5
|
-
|
|
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
|
-
|
|
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('
|
|
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":"
|
|
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,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 {
|
|
5
|
-
import type {
|
|
6
|
-
import { type
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
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<
|
|
20
|
-
options?: Readonly<
|
|
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<
|
|
29
|
-
getModelConfig():
|
|
30
|
-
generateChatReq: (req: Readonly<
|
|
31
|
-
generateChatResp: (resp: Readonly<
|
|
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 {
|
|
2
|
-
import {
|
|
3
|
-
import {
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
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
|
-
|
|
14
|
-
|
|
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 =
|
|
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:
|
|
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":"
|
|
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 {
|
|
1
|
+
import type { AxModelInfo } from '../types.js';
|
|
2
2
|
/**
|
|
3
3
|
* HuggingFace: Model information
|
|
4
4
|
* @export
|
|
5
5
|
*/
|
|
6
|
-
export declare const
|
|
6
|
+
export declare const axModelInfoHuggingFace: AxModelInfo[];
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"info.js","sourceRoot":"","sources":["../../../../../src/ai/huggingface/info.ts"],"names":[],"mappings":"AAEA;;;GAGG;AACH,MAAM,CAAC,MAAM,
|
|
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 {
|
|
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
|
-
|
|
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
|
|
22
|
-
model:
|
|
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
|
|
32
|
+
export type AxHuggingFaceResponse = {
|
|
41
33
|
generated_text: string;
|
|
42
34
|
};
|
|
@@ -1,9 +1,5 @@
|
|
|
1
|
-
|
|
2
|
-
|
|
3
|
-
|
|
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
|
|
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 {
|
|
2
|
-
import { type
|
|
3
|
-
import {
|
|
4
|
-
import { type
|
|
5
|
-
import {
|
|
6
|
-
import { type
|
|
7
|
-
import {
|
|
8
|
-
import { type
|
|
9
|
-
import { type
|
|
10
|
-
import {
|
|
11
|
-
import { type
|
|
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
|
|
28
|
-
export declare const
|
|
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;
|