@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,22 +1,22 @@
|
|
|
1
|
-
import {
|
|
1
|
+
import { axSpanAttributes, AxSpanKind } from '../trace/index.js';
|
|
2
2
|
import { apiCall } from '../util/apicall.js';
|
|
3
3
|
import { ColorLog } from '../util/log.js';
|
|
4
4
|
import { RespTransformStream } from '../util/transform.js';
|
|
5
5
|
const colorLog = new ColorLog();
|
|
6
|
-
export const
|
|
6
|
+
export const axBaseAIDefaultConfig = () => structuredClone({
|
|
7
7
|
maxTokens: 500,
|
|
8
8
|
temperature: 0,
|
|
9
9
|
topK: 40,
|
|
10
10
|
frequencyPenalty: 0.2
|
|
11
11
|
});
|
|
12
|
-
export const
|
|
12
|
+
export const axBaseAIDefaultCreativeConfig = () => structuredClone({
|
|
13
13
|
maxTokens: 500,
|
|
14
14
|
temperature: 0.4,
|
|
15
15
|
topP: 0.7,
|
|
16
16
|
frequencyPenalty: 0.2,
|
|
17
17
|
presencePenalty: 0.2
|
|
18
18
|
});
|
|
19
|
-
export class
|
|
19
|
+
export class AxBaseAI {
|
|
20
20
|
generateChatReq;
|
|
21
21
|
generateEmbedReq;
|
|
22
22
|
generateChatResp;
|
|
@@ -97,19 +97,19 @@ export class BaseAI {
|
|
|
97
97
|
if (this.tracer) {
|
|
98
98
|
const mc = this.getModelConfig();
|
|
99
99
|
return await this.tracer?.startActiveSpan('Chat Request', {
|
|
100
|
-
kind:
|
|
100
|
+
kind: AxSpanKind.SERVER,
|
|
101
101
|
attributes: {
|
|
102
|
-
[
|
|
103
|
-
[
|
|
104
|
-
[
|
|
105
|
-
[
|
|
106
|
-
[
|
|
107
|
-
[
|
|
108
|
-
[
|
|
109
|
-
[
|
|
110
|
-
[
|
|
111
|
-
[
|
|
112
|
-
// [
|
|
102
|
+
[axSpanAttributes.LLM_SYSTEM]: this.name,
|
|
103
|
+
[axSpanAttributes.LLM_REQUEST_MODEL]: this.modelInfo.name,
|
|
104
|
+
[axSpanAttributes.LLM_REQUEST_MAX_TOKENS]: mc.maxTokens,
|
|
105
|
+
[axSpanAttributes.LLM_REQUEST_TEMPERATURE]: mc.temperature,
|
|
106
|
+
[axSpanAttributes.LLM_REQUEST_TOP_P]: mc.topP,
|
|
107
|
+
[axSpanAttributes.LLM_REQUEST_TOP_K]: mc.topK,
|
|
108
|
+
[axSpanAttributes.LLM_REQUEST_FREQUENCY_PENALTY]: mc.frequencyPenalty,
|
|
109
|
+
[axSpanAttributes.LLM_REQUEST_PRESENCE_PENALTY]: mc.presencePenalty,
|
|
110
|
+
[axSpanAttributes.LLM_REQUEST_STOP_SEQUENCES]: mc.stopSequences?.join(', '),
|
|
111
|
+
[axSpanAttributes.LLM_REQUEST_LLM_IS_STREAMING]: mc.stream
|
|
112
|
+
// [AxSpanAttributes.LLM_PROMPTS]: _req.chatPrompt
|
|
113
113
|
// ?.map((v) => v.content)
|
|
114
114
|
// .join('\n')
|
|
115
115
|
}
|
|
@@ -168,7 +168,7 @@ export class BaseAI {
|
|
|
168
168
|
};
|
|
169
169
|
// eslint-disable-next-line @typescript-eslint/no-unused-vars
|
|
170
170
|
const doneCb = async (_values) => {
|
|
171
|
-
// const res =
|
|
171
|
+
// const res = mergeChatResponses(values);
|
|
172
172
|
// if (span?.isRecording()) {
|
|
173
173
|
// setResponseAttr(res, span);
|
|
174
174
|
// }
|
|
@@ -200,10 +200,10 @@ export class BaseAI {
|
|
|
200
200
|
async embed(req, options) {
|
|
201
201
|
if (this.tracer) {
|
|
202
202
|
return await this.tracer?.startActiveSpan('Embed Request', {
|
|
203
|
-
kind:
|
|
203
|
+
kind: AxSpanKind.SERVER,
|
|
204
204
|
attributes: {
|
|
205
|
-
[
|
|
206
|
-
[
|
|
205
|
+
[axSpanAttributes.LLM_SYSTEM]: this.name,
|
|
206
|
+
[axSpanAttributes.LLM_REQUEST_MODEL]: this.modelInfo.name
|
|
207
207
|
}
|
|
208
208
|
}, async (span) => {
|
|
209
209
|
const res = await this._embed(req, options, span);
|
|
@@ -238,50 +238,29 @@ export class BaseAI {
|
|
|
238
238
|
if (span?.isRecording()) {
|
|
239
239
|
if (res.modelUsage) {
|
|
240
240
|
span.setAttributes({
|
|
241
|
-
[
|
|
242
|
-
[
|
|
241
|
+
[axSpanAttributes.LLM_USAGE_COMPLETION_TOKENS]: res.modelUsage.completionTokens ?? 0,
|
|
242
|
+
[axSpanAttributes.LLM_USAGE_PROMPT_TOKENS]: res.modelUsage.promptTokens
|
|
243
243
|
});
|
|
244
244
|
}
|
|
245
245
|
}
|
|
246
246
|
span?.end();
|
|
247
247
|
return res;
|
|
248
248
|
}
|
|
249
|
-
// async _transcribe(
|
|
250
|
-
// // eslint-disable-next-line @typescript-eslint/no-unused-vars
|
|
251
|
-
// _file: string,
|
|
252
|
-
// // eslint-disable-next-line @typescript-eslint/no-unused-vars
|
|
253
|
-
// _prompt?: string,
|
|
254
|
-
// // eslint-disable-next-line @typescript-eslint/no-unused-vars
|
|
255
|
-
// _options?: Readonly<AITranscribeConfig & AIServiceActionOptions>
|
|
256
|
-
// ): Promise<TranscriptResponse> {
|
|
257
|
-
// throw new Error('_transcribe not implemented');
|
|
258
|
-
// }
|
|
259
|
-
// async transcribe(
|
|
260
|
-
// file: string,
|
|
261
|
-
// prompt?: string,
|
|
262
|
-
// options?: Readonly<AITranscribeConfig & AIServiceActionOptions>
|
|
263
|
-
// ): Promise<TranscriptResponse> {
|
|
264
|
-
// const res = this.rt
|
|
265
|
-
// ? await this.rt<Promise<TranscriptResponse>>(
|
|
266
|
-
// async () => await this._transcribe(file, prompt, options)
|
|
267
|
-
// )
|
|
268
|
-
// : await this._transcribe(file, prompt, options);
|
|
269
|
-
// res.sessionId = options?.sessionId;
|
|
270
|
-
// return res;
|
|
271
|
-
// }
|
|
272
|
-
// async apiCallWithUpload<Request, Response, APIType extends API >(
|
|
273
|
-
// api: APIType,
|
|
274
|
-
// json: Request,
|
|
275
|
-
// file: string
|
|
276
|
-
// ): Promise<Response> {
|
|
277
|
-
// return apiCallWithUpload<Request, Response, APIType>(this.mergeAPIConfig<APIType>(api), json, file);
|
|
278
|
-
// }
|
|
279
249
|
buildHeaders(headers = {}) {
|
|
280
250
|
return { ...headers, ...this.headers };
|
|
281
251
|
}
|
|
282
252
|
}
|
|
283
253
|
const logChatRequest = (req) => {
|
|
284
|
-
const items = req.chatPrompt
|
|
254
|
+
const items = req.chatPrompt
|
|
255
|
+
?.map(({ content }) => {
|
|
256
|
+
if (typeof content === 'string') {
|
|
257
|
+
return content;
|
|
258
|
+
}
|
|
259
|
+
else {
|
|
260
|
+
return JSON.stringify(content, null, 2);
|
|
261
|
+
}
|
|
262
|
+
})
|
|
263
|
+
.join('\n');
|
|
285
264
|
if (items) {
|
|
286
265
|
console.log('==========');
|
|
287
266
|
console.log(colorLog.whiteBright(items));
|
|
@@ -301,6 +280,9 @@ const logResponse = (resp) => {
|
|
|
301
280
|
}
|
|
302
281
|
};
|
|
303
282
|
const logStreamingResponse = (resp) => {
|
|
283
|
+
if (!resp.results) {
|
|
284
|
+
return;
|
|
285
|
+
}
|
|
304
286
|
for (const r of resp.results) {
|
|
305
287
|
if (r.content) {
|
|
306
288
|
process.stdout.write(colorLog.greenBright(r.content));
|
|
@@ -320,8 +302,8 @@ const logStreamingResponse = (resp) => {
|
|
|
320
302
|
const setResponseAttr = (res, span) => {
|
|
321
303
|
if (res.modelUsage) {
|
|
322
304
|
span.setAttributes({
|
|
323
|
-
[
|
|
324
|
-
[
|
|
305
|
+
[axSpanAttributes.LLM_USAGE_COMPLETION_TOKENS]: res.modelUsage.completionTokens ?? 0,
|
|
306
|
+
[axSpanAttributes.LLM_USAGE_PROMPT_TOKENS]: res.modelUsage.promptTokens
|
|
325
307
|
});
|
|
326
308
|
}
|
|
327
309
|
};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../../src/ai/base.ts"],"names":[],"mappings":"AAQA,OAAO,EAAa,cAAc,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAMxE,OAAO,EAAY,OAAO,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAS3D,MAAM,QAAQ,GAAG,IAAI,QAAQ,EAAE,CAAC;AAYhC,MAAM,CAAC,MAAM,mBAAmB,GAAG,GAAoB,EAAE,CACvD,eAAe,CAAC;IACd,SAAS,EAAE,GAAG;IACd,WAAW,EAAE,CAAC;IACd,IAAI,EAAE,EAAE;IACR,gBAAgB,EAAE,GAAG;CACtB,CAAC,CAAC;AAEL,MAAM,CAAC,MAAM,2BAA2B,GAAG,GAAoB,EAAE,CAC/D,eAAe,CAAC;IACd,SAAS,EAAE,GAAG;IACd,WAAW,EAAE,GAAG;IAChB,IAAI,EAAE,GAAG;IACT,gBAAgB,EAAE,GAAG;IACrB,eAAe,EAAE,GAAG;CACrB,CAAC,CAAC;AAEL,MAAM,OAAO,MAAM;IAQjB,eAAe,CAGU;IACzB,gBAAgB,CAEU;IAC1B,gBAAgB,CAAmD;IACnE,sBAAsB,CAGJ;IAClB,iBAAiB,CAAqD;IAE9D,KAAK,GAAG,KAAK,CAAC;IAEd,EAAE,CAAmC;IACrC,KAAK,CAA6B;IAClC,MAAM,CAA8B;IAElC,MAAM,CAAS;IACf,IAAI,CAAS;IACb,OAAO,CAAyB;IAChC,SAAS,CAAgB;IACzB,cAAc,CAAiB;IAC/B,UAAU,CAA2B;IAE/C,YAAY,EACV,IAAI,EACJ,MAAM,EACN,OAAO,EACP,SAAS,EACT,MAAM,EACN,OAAO,GAAG,EAAE,EACZ,UAAU,EACW;QACrB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAE7B,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;QACtC,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,CAAC,CAAC;QAC1D,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI;YAClE,IAAI,EAAE,MAAM,CAAC,KAAK;YAClB,QAAQ,EAAE,KAAK;YACf,oBAAoB,EAAE,CAAC;YACvB,wBAAwB,EAAE,CAAC;SAC5B,CAAC;QAEF,IAAI,CAAC,cAAc,GAAG,SAAS;aAC5B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,UAAU,CAAC;aAC3C,EAAE,CAAC,CAAC,CAAC,CAAC;QAET,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IAC3B,CAAC;IAEM,OAAO,CAAC,IAAY;QACzB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;IAEM,SAAS,CAAC,MAAc;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAEM,UAAU,CAAC,OAA+B;QAC/C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,UAAU,CAAC,OAAmC;QAC5C,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC7B,CAAC;QAED,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;YACxB,IAAI,CAAC,EAAE,GAAG,OAAO,CAAC,WAAW,CAAC;QAChC,CAAC;QAED,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC7B,CAAC;QAED,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC/B,CAAC;IACH,CAAC;IAED,YAAY;QACV,OAAO,EAAE,GAAG,IAAI,CAAC,SAAS,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;IACpD,CAAC;IAED,iBAAiB;QACf,OAAO,IAAI,CAAC,cAAc;YACxB,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,cAAc,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE;YACjD,CAAC,CAAC,SAAS,CAAC;IAChB,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,cAAc;QACZ,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;IAED,KAAK,CAAC,IAAI,CACR,IAAiC,EACjC,OAA2D;QAE3D,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,EAAE,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;YACjC,OAAO,MAAM,IAAI,CAAC,MAAM,EAAE,eAAe,CACvC,cAAc,EACd;gBACE,IAAI,EAAE,QAAQ,CAAC,MAAM;gBACrB,UAAU,EAAE;oBACV,CAAC,cAAc,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,IAAI;oBACtC,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI;oBACvD,CAAC,cAAc,CAAC,sBAAsB,CAAC,EAAE,EAAE,CAAC,SAAS;oBACrD,CAAC,cAAc,CAAC,uBAAuB,CAAC,EAAE,EAAE,CAAC,WAAW;oBACxD,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,EAAE,CAAC,IAAI;oBAC3C,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,EAAE,CAAC,IAAI;oBAC3C,CAAC,cAAc,CAAC,6BAA6B,CAAC,EAAE,EAAE,CAAC,gBAAgB;oBACnE,CAAC,cAAc,CAAC,4BAA4B,CAAC,EAAE,EAAE,CAAC,eAAe;oBACjE,CAAC,cAAc,CAAC,0BAA0B,CAAC,EACzC,EAAE,CAAC,aAAa,EAAE,IAAI,CAAC,IAAI,CAAC;oBAC9B,CAAC,cAAc,CAAC,4BAA4B,CAAC,EAAE,EAAE,CAAC,MAAM;oBACxD,gDAAgD;oBAChD,4BAA4B;oBAC5B,gBAAgB;iBACjB;aACF,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;gBACb,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;gBAClD,IAAI,CAAC,GAAG,EAAE,CAAC;gBACX,OAAO,GAAG,CAAC;YACb,CAAC,CACF,CAAC;QACJ,CAAC;QACD,OAAO,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,KAAK,CACT,IAAiC,EACjC,OAA2D,EAC3D,IAAW;QAEX,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACrD,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC;QACnC,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,IAAI,CAAC,WAAW,EAAE,MAAM,CAAC;QAC3D,MAAM,SAAS,GACb,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;QAE3E,MAAM,GAAG,GAAG;YACV,GAAG,IAAI;YACP,SAAS;YACT,WAAW,EAAE,EAAE,GAAG,IAAI,CAAC,WAAW,EAAE,MAAM,EAAE;SACd,CAAC;QAEjC,MAAM,EAAE,GAAG,KAAK,IAAI,EAAE;YACpB,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,GAAG,KAAK,CAAC,GAAG,EAAE,OAAyB,CAAC,CAAC;YAEpE,MAAM,GAAG,GAAG,MAAM,OAAO,CACvB;gBACE,IAAI,EAAE,SAAS,CAAC,IAAI;gBACpB,GAAG,EAAE,IAAI,CAAC,MAAM;gBAChB,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC;gBAC7C,MAAM;gBACN,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,IAAI;aACL,EACD,QAAQ,CACT,CAAC;YACF,OAAO,GAAG,CAAC;QACb,CAAC,CAAC;QAEF,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,cAAc,CAAC,GAAG,CAAC,CAAC;QACtB,CAAC;QAED,MAAM,EAAE,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;QAEpD,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE,CAAC;gBACjC,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;YACtD,CAAC;YAED,MAAM,MAAM,GAAG,IAAI,CAAC,sBAAsB,CAAC;YAC3C,MAAM,aAAa,GACjB,CAAC,KAAa,EAAE,EAAE,CAAC,CAAC,IAAkC,EAAE,EAAE;gBACxD,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;gBAChC,GAAG,CAAC,SAAS,GAAG,OAAO,EAAE,SAAS,CAAC;gBAEnC,IAAI,IAAI,EAAE,WAAW,EAAE,EAAE,CAAC;oBACxB,eAAe,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;gBAC7B,CAAC;gBAED,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;oBACf,oBAAoB,CAAC,GAAG,CAAC,CAAC;gBAC5B,CAAC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC,CAAC;YAEJ,6DAA6D;YAC7D,MAAM,MAAM,GAAG,KAAK,EAAE,OAAgC,EAAE,EAAE;gBACxD,4CAA4C;gBAC5C,+BAA+B;gBAC/B,kCAAkC;gBAClC,MAAM;gBACN,iBAAiB;gBAEjB,sBAAsB;gBACtB,wBAAwB;gBACxB,MAAM;gBACN,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC7B,CAAC;YACH,CAAC,CAAC;YAEF,MAAM,EAAE,GAAI,EAAyC,CAAC,WAAW,CAC/D,IAAI,mBAAmB,CACrB,aAAa,CAAC,EAAE,CAAC,EACjB,MAAM,CACP,CACF,CAAC;YACF,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACtD,CAAC;QACD,MAAM,GAAG,GAAG,IAAI,CAAC,gBAAgB,CAAC,EAAmB,CAAC,CAAC;QACvD,GAAG,CAAC,SAAS,GAAG,OAAO,EAAE,SAAS,CAAC;QAEnC,IAAI,IAAI,EAAE,WAAW,EAAE,EAAE,CAAC;YACxB,eAAe,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAC7B,CAAC;QAED,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,WAAW,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;QAED,IAAI,EAAE,GAAG,EAAE,CAAC;QACZ,OAAO,GAAG,CAAC;IACb,CAAC;IAED,KAAK,CAAC,KAAK,CACT,GAAiC,EACjC,OAA0C;QAE1C,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO,MAAM,IAAI,CAAC,MAAM,EAAE,eAAe,CACvC,eAAe,EACf;gBACE,IAAI,EAAE,QAAQ,CAAC,MAAM;gBACrB,UAAU,EAAE;oBACV,CAAC,cAAc,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,IAAI;oBACtC,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI;iBACxD;aACF,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;gBACb,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;gBAClD,IAAI,CAAC,GAAG,EAAE,CAAC;gBACX,OAAO,GAAG,CAAC;YACb,CAAC,CACF,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,MAAM,CACV,GAAiC,EACjC,OAA0C,EAC1C,IAAW;QAEX,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACtD,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;QAED,MAAM,EAAE,GAAG,KAAK,IAAI,EAAE;YACpB,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,GAAG,IAAI,CAAC,gBAAiB,CAAC,GAAG,CAAC,CAAC;YAE1D,MAAM,GAAG,GAAG,MAAM,OAAO,CACvB;gBACE,IAAI,EAAE,SAAS,CAAC,IAAI;gBACpB,GAAG,EAAE,IAAI,CAAC,MAAM;gBAChB,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC;gBAC7C,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,IAAI;aACL,EACD,QAAQ,CACT,CAAC;YAEF,OAAO,GAAG,CAAC;QACb,CAAC,CAAC;QAEF,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;QACxE,MAAM,GAAG,GAAG,IAAI,CAAC,iBAAkB,CAAC,QAA0B,CAAC,CAAC;QAEhE,GAAG,CAAC,SAAS,GAAG,OAAO,EAAE,SAAS,CAAC;QAEnC,IAAI,IAAI,EAAE,WAAW,EAAE,EAAE,CAAC;YACxB,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;gBACnB,IAAI,CAAC,aAAa,CAAC;oBACjB,CAAC,cAAc,CAAC,2BAA2B,CAAC,EAC1C,GAAG,CAAC,UAAU,CAAC,gBAAgB,IAAI,CAAC;oBACtC,CAAC,cAAc,CAAC,uBAAuB,CAAC,EAAE,GAAG,CAAC,UAAU,CAAC,YAAY;iBACtE,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,IAAI,EAAE,GAAG,EAAE,CAAC;QACZ,OAAO,GAAG,CAAC;IACb,CAAC;IAED,qBAAqB;IACrB,kEAAkE;IAClE,mBAAmB;IACnB,kEAAkE;IAClE,sBAAsB;IACtB,kEAAkE;IAClE,qEAAqE;IACrE,mCAAmC;IACnC,oDAAoD;IACpD,IAAI;IAEJ,oBAAoB;IACpB,kBAAkB;IAClB,qBAAqB;IACrB,oEAAoE;IACpE,mCAAmC;IACnC,wBAAwB;IACxB,oDAAoD;IACpD,oEAAoE;IACpE,UAAU;IACV,uDAAuD;IAEvD,wCAAwC;IACxC,gBAAgB;IAChB,IAAI;IAEJ,oEAAoE;IACpE,kBAAkB;IAClB,mBAAmB;IACnB,iBAAiB;IACjB,yBAAyB;IACzB,uGAAuG;IACvG,IAAI;IAEI,YAAY,CAClB,UAAkC,EAAE;QAEpC,OAAO,EAAE,GAAG,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;IACzC,CAAC;CACF;AAED,MAAM,cAAc,GAAG,CAAC,GAAgC,EAAE,EAAE;IAC1D,MAAM,KAAK,GAAG,GAAG,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC/D,IAAI,KAAK,EAAE,CAAC;QACV,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QAC1B,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC;IAC3C,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,WAAW,GAAG,CAAC,IAA4B,EAAE,EAAE;IACnD,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;QAC7B,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC;YACd,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;QAC/C,CAAC;QACD,IAAI,CAAC,CAAC,aAAa,EAAE,CAAC;YACpB,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,CAAC;gBAChC,MAAM,GAAG,GAAG,GAAG,CAAC,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,CAAC,QAAQ,CAAC,SAAS,GAAG,CAAC;gBAC1D,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YACpD,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,oBAAoB,GAAG,CAAC,IAA4B,EAAE,EAAE;IAC5D,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;QAC7B,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC;YACd,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;QACxD,CAAC;QACD,IAAI,CAAC,CAAC,aAAa,EAAE,CAAC;YACpB,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,CAAC;gBAChC,IAAI,CAAC,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;oBACpB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;gBAC7D,CAAC;gBACD,IAAI,CAAC,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC;oBACzB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAmB,CAAC,CAAC,CAAC;gBACxE,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,eAAe,GAAG,CAAC,GAA2B,EAAE,IAAU,EAAE,EAAE;IAClE,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;QACnB,IAAI,CAAC,aAAa,CAAC;YACjB,CAAC,cAAc,CAAC,2BAA2B,CAAC,EAC1C,GAAG,CAAC,UAAU,CAAC,gBAAgB,IAAI,CAAC;YACtC,CAAC,cAAc,CAAC,uBAAuB,CAAC,EAAE,GAAG,CAAC,UAAU,CAAC,YAAY;SACtE,CAAC,CAAC;IACL,CAAC;AACH,CAAC,CAAC"}
|
|
1
|
+
{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../../src/ai/base.ts"],"names":[],"mappings":"AAEA,OAAO,EAAe,gBAAgB,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAC9E,OAAO,EAAY,OAAO,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAgB3D,MAAM,QAAQ,GAAG,IAAI,QAAQ,EAAE,CAAC;AAYhC,MAAM,CAAC,MAAM,qBAAqB,GAAG,GAAkB,EAAE,CACvD,eAAe,CAAC;IACd,SAAS,EAAE,GAAG;IACd,WAAW,EAAE,CAAC;IACd,IAAI,EAAE,EAAE;IACR,gBAAgB,EAAE,GAAG;CACtB,CAAC,CAAC;AAEL,MAAM,CAAC,MAAM,6BAA6B,GAAG,GAAkB,EAAE,CAC/D,eAAe,CAAC;IACd,SAAS,EAAE,GAAG;IACd,WAAW,EAAE,GAAG;IAChB,IAAI,EAAE,GAAG;IACT,gBAAgB,EAAE,GAAG;IACrB,eAAe,EAAE,GAAG;CACrB,CAAC,CAAC;AAEL,MAAM,OAAO,QAAQ;IAQnB,eAAe,CAGU;IACzB,gBAAgB,CAA2D;IAC3E,gBAAgB,CAAqD;IACrE,sBAAsB,CAGF;IACpB,iBAAiB,CAAuD;IAEhE,KAAK,GAAG,KAAK,CAAC;IAEd,EAAE,CAAqC;IACvC,KAAK,CAA+B;IACpC,MAAM,CAAgC;IAEpC,MAAM,CAAS;IACf,IAAI,CAAS;IACb,OAAO,CAAyB;IAChC,SAAS,CAAc;IACvB,cAAc,CAAe;IAC7B,UAAU,CAA6B;IAEjD,YAAY,EACV,IAAI,EACJ,MAAM,EACN,OAAO,EACP,SAAS,EACT,MAAM,EACN,OAAO,GAAG,EAAE,EACZ,UAAU,EACa;QACvB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAE7B,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;QACtC,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,CAAC,CAAC;QAC1D,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI;YAClE,IAAI,EAAE,MAAM,CAAC,KAAK;YAClB,QAAQ,EAAE,KAAK;YACf,oBAAoB,EAAE,CAAC;YACvB,wBAAwB,EAAE,CAAC;SAC5B,CAAC;QAEF,IAAI,CAAC,cAAc,GAAG,SAAS;aAC5B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,UAAU,CAAC;aAC3C,EAAE,CAAC,CAAC,CAAC,CAAC;QAET,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IAC3B,CAAC;IAEM,OAAO,CAAC,IAAY;QACzB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;IAEM,SAAS,CAAC,MAAc;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAEM,UAAU,CAAC,OAA+B;QAC/C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,UAAU,CAAC,OAAqC;QAC9C,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC7B,CAAC;QAED,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;YACxB,IAAI,CAAC,EAAE,GAAG,OAAO,CAAC,WAAW,CAAC;QAChC,CAAC;QAED,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC7B,CAAC;QAED,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC/B,CAAC;IACH,CAAC;IAED,YAAY;QACV,OAAO,EAAE,GAAG,IAAI,CAAC,SAAS,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;IACpD,CAAC;IAED,iBAAiB;QACf,OAAO,IAAI,CAAC,cAAc;YACxB,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,cAAc,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE;YACjD,CAAC,CAAC,SAAS,CAAC;IAChB,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,cAAc;QACZ,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;IAED,KAAK,CAAC,IAAI,CACR,IAA6B,EAC7B,OAA+D;QAE/D,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,EAAE,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;YACjC,OAAO,MAAM,IAAI,CAAC,MAAM,EAAE,eAAe,CACvC,cAAc,EACd;gBACE,IAAI,EAAE,UAAU,CAAC,MAAM;gBACvB,UAAU,EAAE;oBACV,CAAC,gBAAgB,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,IAAI;oBACxC,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI;oBACzD,CAAC,gBAAgB,CAAC,sBAAsB,CAAC,EAAE,EAAE,CAAC,SAAS;oBACvD,CAAC,gBAAgB,CAAC,uBAAuB,CAAC,EAAE,EAAE,CAAC,WAAW;oBAC1D,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,EAAE,CAAC,IAAI;oBAC7C,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,EAAE,CAAC,IAAI;oBAC7C,CAAC,gBAAgB,CAAC,6BAA6B,CAAC,EAC9C,EAAE,CAAC,gBAAgB;oBACrB,CAAC,gBAAgB,CAAC,4BAA4B,CAAC,EAAE,EAAE,CAAC,eAAe;oBACnE,CAAC,gBAAgB,CAAC,0BAA0B,CAAC,EAC3C,EAAE,CAAC,aAAa,EAAE,IAAI,CAAC,IAAI,CAAC;oBAC9B,CAAC,gBAAgB,CAAC,4BAA4B,CAAC,EAAE,EAAE,CAAC,MAAM;oBAC1D,kDAAkD;oBAClD,4BAA4B;oBAC5B,gBAAgB;iBACjB;aACF,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;gBACb,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;gBAClD,IAAI,CAAC,GAAG,EAAE,CAAC;gBACX,OAAO,GAAG,CAAC;YACb,CAAC,CACF,CAAC;QACJ,CAAC;QACD,OAAO,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,KAAK,CACT,IAA6B,EAC7B,OAA+D,EAC/D,IAAa;QAEb,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACrD,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC;QACnC,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,IAAI,CAAC,WAAW,EAAE,MAAM,CAAC;QAC3D,MAAM,SAAS,GACb,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;QAE3E,MAAM,GAAG,GAAG;YACV,GAAG,IAAI;YACP,SAAS;YACT,WAAW,EAAE,EAAE,GAAG,IAAI,CAAC,WAAW,EAAE,MAAM,EAAE;SAClB,CAAC;QAE7B,MAAM,EAAE,GAAG,KAAK,IAAI,EAAE;YACpB,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,GAAG,KAAK,CAAC,GAAG,EAAE,OAA2B,CAAC,CAAC;YAEtE,MAAM,GAAG,GAAG,MAAM,OAAO,CACvB;gBACE,IAAI,EAAE,SAAS,CAAC,IAAI;gBACpB,GAAG,EAAE,IAAI,CAAC,MAAM;gBAChB,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC;gBAC7C,MAAM;gBACN,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,IAAI;aACL,EACD,QAAQ,CACT,CAAC;YACF,OAAO,GAAG,CAAC;QACb,CAAC,CAAC;QAEF,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,cAAc,CAAC,GAAG,CAAC,CAAC;QACtB,CAAC;QAED,MAAM,EAAE,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;QAEpD,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE,CAAC;gBACjC,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;YACtD,CAAC;YAED,MAAM,MAAM,GAAG,IAAI,CAAC,sBAAsB,CAAC;YAC3C,MAAM,aAAa,GACjB,CAAC,KAAa,EAAE,EAAE,CAAC,CAAC,IAAkC,EAAE,EAAE;gBACxD,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;gBAChC,GAAG,CAAC,SAAS,GAAG,OAAO,EAAE,SAAS,CAAC;gBAEnC,IAAI,IAAI,EAAE,WAAW,EAAE,EAAE,CAAC;oBACxB,eAAe,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;gBAC7B,CAAC;gBAED,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;oBACf,oBAAoB,CAAC,GAAG,CAAC,CAAC;gBAC5B,CAAC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC,CAAC;YAEJ,6DAA6D;YAC7D,MAAM,MAAM,GAAG,KAAK,EAAE,OAAkC,EAAE,EAAE;gBAC1D,4CAA4C;gBAC5C,+BAA+B;gBAC/B,kCAAkC;gBAClC,MAAM;gBACN,iBAAiB;gBAEjB,sBAAsB;gBACtB,wBAAwB;gBACxB,MAAM;gBACN,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC7B,CAAC;YACH,CAAC,CAAC;YAEF,MAAM,EAAE,GAAI,EAAyC,CAAC,WAAW,CAC/D,IAAI,mBAAmB,CACrB,aAAa,CAAC,EAAE,CAAC,EACjB,MAAM,CACP,CACF,CAAC;YACF,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACtD,CAAC;QACD,MAAM,GAAG,GAAG,IAAI,CAAC,gBAAgB,CAAC,EAAmB,CAAC,CAAC;QACvD,GAAG,CAAC,SAAS,GAAG,OAAO,EAAE,SAAS,CAAC;QAEnC,IAAI,IAAI,EAAE,WAAW,EAAE,EAAE,CAAC;YACxB,eAAe,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAC7B,CAAC;QAED,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,WAAW,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;QAED,IAAI,EAAE,GAAG,EAAE,CAAC;QACZ,OAAO,GAAG,CAAC;IACb,CAAC;IAED,KAAK,CAAC,KAAK,CACT,GAA6B,EAC7B,OAA4C;QAE5C,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO,MAAM,IAAI,CAAC,MAAM,EAAE,eAAe,CACvC,eAAe,EACf;gBACE,IAAI,EAAE,UAAU,CAAC,MAAM;gBACvB,UAAU,EAAE;oBACV,CAAC,gBAAgB,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,IAAI;oBACxC,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI;iBAC1D;aACF,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;gBACb,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;gBAClD,IAAI,CAAC,GAAG,EAAE,CAAC;gBACX,OAAO,GAAG,CAAC;YACb,CAAC,CACF,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,MAAM,CACV,GAA6B,EAC7B,OAA4C,EAC5C,IAAa;QAEb,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACtD,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;QAED,MAAM,EAAE,GAAG,KAAK,IAAI,EAAE;YACpB,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,GAAG,IAAI,CAAC,gBAAiB,CAAC,GAAG,CAAC,CAAC;YAE1D,MAAM,GAAG,GAAG,MAAM,OAAO,CACvB;gBACE,IAAI,EAAE,SAAS,CAAC,IAAI;gBACpB,GAAG,EAAE,IAAI,CAAC,MAAM;gBAChB,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC;gBAC7C,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,IAAI;aACL,EACD,QAAQ,CACT,CAAC;YAEF,OAAO,GAAG,CAAC;QACb,CAAC,CAAC;QAEF,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;QACxE,MAAM,GAAG,GAAG,IAAI,CAAC,iBAAkB,CAAC,QAA0B,CAAC,CAAC;QAEhE,GAAG,CAAC,SAAS,GAAG,OAAO,EAAE,SAAS,CAAC;QAEnC,IAAI,IAAI,EAAE,WAAW,EAAE,EAAE,CAAC;YACxB,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;gBACnB,IAAI,CAAC,aAAa,CAAC;oBACjB,CAAC,gBAAgB,CAAC,2BAA2B,CAAC,EAC5C,GAAG,CAAC,UAAU,CAAC,gBAAgB,IAAI,CAAC;oBACtC,CAAC,gBAAgB,CAAC,uBAAuB,CAAC,EACxC,GAAG,CAAC,UAAU,CAAC,YAAY;iBAC9B,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,IAAI,EAAE,GAAG,EAAE,CAAC;QACZ,OAAO,GAAG,CAAC;IACb,CAAC;IAEO,YAAY,CAClB,UAAkC,EAAE;QAEpC,OAAO,EAAE,GAAG,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;IACzC,CAAC;CACF;AAED,MAAM,cAAc,GAAG,CAAC,GAA4B,EAAE,EAAE;IACtD,MAAM,KAAK,GAAG,GAAG,CAAC,UAAU;QAC1B,EAAE,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE;QACpB,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAChC,OAAO,OAAO,CAAC;QACjB,CAAC;aAAM,CAAC;YACN,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC,CAAC;SACD,IAAI,CAAC,IAAI,CAAC,CAAC;IAEd,IAAI,KAAK,EAAE,CAAC;QACV,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QAC1B,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC;IAC3C,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,WAAW,GAAG,CAAC,IAA8B,EAAE,EAAE;IACrD,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;QAC7B,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC;YACd,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;QAC/C,CAAC;QACD,IAAI,CAAC,CAAC,aAAa,EAAE,CAAC;YACpB,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,CAAC;gBAChC,MAAM,GAAG,GAAG,GAAG,CAAC,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,CAAC,QAAQ,CAAC,SAAS,GAAG,CAAC;gBAC1D,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YACpD,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,oBAAoB,GAAG,CAAC,IAA8B,EAAE,EAAE;IAC9D,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;QAClB,OAAO;IACT,CAAC;IACD,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;QAC7B,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC;YACd,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;QACxD,CAAC;QACD,IAAI,CAAC,CAAC,aAAa,EAAE,CAAC;YACpB,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,CAAC;gBAChC,IAAI,CAAC,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;oBACpB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;gBAC7D,CAAC;gBACD,IAAI,CAAC,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC;oBACzB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAmB,CAAC,CAAC,CAAC;gBACxE,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,eAAe,GAAG,CAAC,GAA6B,EAAE,IAAY,EAAE,EAAE;IACtE,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;QACnB,IAAI,CAAC,aAAa,CAAC;YACjB,CAAC,gBAAgB,CAAC,2BAA2B,CAAC,EAC5C,GAAG,CAAC,UAAU,CAAC,gBAAgB,IAAI,CAAC;YACtC,CAAC,gBAAgB,CAAC,uBAAuB,CAAC,EAAE,GAAG,CAAC,UAAU,CAAC,YAAY;SACxE,CAAC,CAAC;IACL,CAAC;AACH,CAAC,CAAC"}
|
|
@@ -1,35 +1,21 @@
|
|
|
1
|
-
import type { AIPromptConfig, AIServiceOptions } from '../../text/types.js';
|
|
2
|
-
import type { AITextChatRequest, AITextEmbedRequest } 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 CohereDefaultConfig: () => CohereConfig;
|
|
12
|
-
/**
|
|
13
|
-
* Cohere: Default model config for more creative text generation
|
|
14
|
-
* @export
|
|
15
|
-
*/
|
|
16
|
-
export declare const CohereCreativeConfig: () => CohereConfig;
|
|
17
|
-
export interface CohereArgs {
|
|
2
|
+
import { AxBaseAI } from '../base.js';
|
|
3
|
+
import type { AxAIPromptConfig, AxAIServiceOptions, AxChatRequest, AxChatResponse, AxEmbedRequest, AxEmbedResponse, AxModelConfig } from '../types.js';
|
|
4
|
+
import { type AxCohereChatRequest, type AxCohereChatResponse, type AxCohereChatResponseDelta, type AxCohereConfig, type AxCohereEmbedRequest, type AxCohereEmbedResponse } from './types.js';
|
|
5
|
+
export declare const axCohereDefaultConfig: () => AxCohereConfig;
|
|
6
|
+
export declare const axCohereCreativeConfig: () => AxCohereConfig;
|
|
7
|
+
export interface AxCohereArgs {
|
|
18
8
|
apiKey: string;
|
|
19
|
-
config: Readonly<
|
|
20
|
-
options?: Readonly<
|
|
9
|
+
config: Readonly<AxCohereConfig>;
|
|
10
|
+
options?: Readonly<AxAIServiceOptions>;
|
|
21
11
|
}
|
|
22
|
-
|
|
23
|
-
* Cohere: AI Service
|
|
24
|
-
* @export
|
|
25
|
-
*/
|
|
26
|
-
export declare class Cohere extends BaseAI<CohereChatRequest, CohereEmbedRequest, CohereChatResponse, CohereChatResponseDelta, CohereEmbedResponse> {
|
|
12
|
+
export declare class AxCohere extends AxBaseAI<AxCohereChatRequest, AxCohereEmbedRequest, AxCohereChatResponse, AxCohereChatResponseDelta, AxCohereEmbedResponse> {
|
|
27
13
|
private config;
|
|
28
|
-
constructor({ apiKey, config, options }: Readonly<
|
|
29
|
-
getModelConfig():
|
|
30
|
-
generateChatReq: (req: Readonly<
|
|
31
|
-
generateEmbedReq: (req: Readonly<
|
|
32
|
-
generateChatResp: (resp: Readonly<
|
|
33
|
-
generateChatStreamResp: (resp: Readonly<
|
|
34
|
-
generateEmbedResp: (resp: Readonly<
|
|
14
|
+
constructor({ apiKey, config, options }: Readonly<AxCohereArgs>);
|
|
15
|
+
getModelConfig(): AxModelConfig;
|
|
16
|
+
generateChatReq: (req: Readonly<AxChatRequest>, _config: Readonly<AxAIPromptConfig>) => [API, AxCohereChatRequest];
|
|
17
|
+
generateEmbedReq: (req: Readonly<AxEmbedRequest>) => [API, AxCohereEmbedRequest];
|
|
18
|
+
generateChatResp: (resp: Readonly<AxCohereChatResponse>) => AxChatResponse;
|
|
19
|
+
generateChatStreamResp: (resp: Readonly<AxCohereChatResponseDelta>, state: object) => AxChatResponse;
|
|
20
|
+
generateEmbedResp: (resp: Readonly<AxCohereEmbedResponse>) => AxEmbedResponse;
|
|
35
21
|
}
|
|
@@ -1,31 +1,19 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import {
|
|
3
|
-
import {
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
export const CohereDefaultConfig = () => structuredClone({
|
|
9
|
-
model: CohereModel.Command,
|
|
10
|
-
embedModel: CohereEmbedModel.EmbedEnglishV30,
|
|
11
|
-
...BaseAIDefaultConfig()
|
|
1
|
+
import { AxBaseAI, axBaseAIDefaultConfig, axBaseAIDefaultCreativeConfig } from '../base.js';
|
|
2
|
+
import { axModelInfoCohere } from './info.js';
|
|
3
|
+
import { AxCohereEmbedModel, AxCohereModel } from './types.js';
|
|
4
|
+
export const axCohereDefaultConfig = () => structuredClone({
|
|
5
|
+
model: AxCohereModel.Command,
|
|
6
|
+
embedModel: AxCohereEmbedModel.EmbedEnglishV30,
|
|
7
|
+
...axBaseAIDefaultConfig()
|
|
12
8
|
});
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
export const CohereCreativeConfig = () => structuredClone({
|
|
18
|
-
model: CohereModel.CommandR,
|
|
19
|
-
embedModel: CohereEmbedModel.EmbedEnglishV30,
|
|
20
|
-
...BaseAIDefaultCreativeConfig()
|
|
9
|
+
export const axCohereCreativeConfig = () => structuredClone({
|
|
10
|
+
model: AxCohereModel.CommandR,
|
|
11
|
+
embedModel: AxCohereEmbedModel.EmbedEnglishV30,
|
|
12
|
+
...axBaseAIDefaultCreativeConfig()
|
|
21
13
|
});
|
|
22
|
-
|
|
23
|
-
* Cohere: AI Service
|
|
24
|
-
* @export
|
|
25
|
-
*/
|
|
26
|
-
export class Cohere extends BaseAI {
|
|
14
|
+
export class AxCohere extends AxBaseAI {
|
|
27
15
|
config;
|
|
28
|
-
constructor({ apiKey, config =
|
|
16
|
+
constructor({ apiKey, config = axCohereDefaultConfig(), options }) {
|
|
29
17
|
if (!apiKey || apiKey === '') {
|
|
30
18
|
throw new Error('Cohere API key not set');
|
|
31
19
|
}
|
|
@@ -33,7 +21,7 @@ export class Cohere extends BaseAI {
|
|
|
33
21
|
name: 'Cohere',
|
|
34
22
|
apiURL: 'https://api.cohere.ai',
|
|
35
23
|
headers: { Authorization: `Bearer ${apiKey}` },
|
|
36
|
-
modelInfo:
|
|
24
|
+
modelInfo: axModelInfoCohere,
|
|
37
25
|
models: { model: config.model },
|
|
38
26
|
supportFor: { functions: false, streaming: true },
|
|
39
27
|
options
|
|
@@ -57,14 +45,19 @@ export class Cohere extends BaseAI {
|
|
|
57
45
|
// eslint-disable-next-line @typescript-eslint/no-unused-vars
|
|
58
46
|
_config) => {
|
|
59
47
|
const model = req.modelInfo?.name ?? this.config.model;
|
|
60
|
-
// const functionsList = req.functions
|
|
61
|
-
// ? `Functions:\n${JSON.stringify(req.functions, null, 2)}\n`
|
|
62
48
|
const lastChatMsg = req.chatPrompt.at(-1);
|
|
63
49
|
const restOfChat = req.chatPrompt.slice(0, -1);
|
|
50
|
+
if (Array.isArray(lastChatMsg?.content)) {
|
|
51
|
+
throw new Error('Chat prompt is empty');
|
|
52
|
+
}
|
|
64
53
|
const message = lastChatMsg?.content ?? '';
|
|
65
|
-
const chatHistory = restOfChat
|
|
66
|
-
|
|
67
|
-
|
|
54
|
+
const chatHistory = restOfChat.map((chat) => {
|
|
55
|
+
if (chat.role === 'function') {
|
|
56
|
+
throw new Error('Function calling not supported');
|
|
57
|
+
}
|
|
58
|
+
if (Array.isArray(chat.content)) {
|
|
59
|
+
throw new Error('Multi-modal content not supported');
|
|
60
|
+
}
|
|
68
61
|
let role;
|
|
69
62
|
switch (chat.role) {
|
|
70
63
|
case 'user':
|
|
@@ -99,12 +92,13 @@ export class Cohere extends BaseAI {
|
|
|
99
92
|
parameter_definitions: props
|
|
100
93
|
};
|
|
101
94
|
});
|
|
95
|
+
// eslint-disable-next-line @typescript-eslint/naming-convention
|
|
102
96
|
const tool_results = req.chatPrompt
|
|
103
97
|
.filter((chat) => chat.role === 'function')
|
|
104
98
|
.map((chat) => {
|
|
105
99
|
const fn = tools?.find((t) => t.name === chat.functionId);
|
|
106
100
|
if (!fn) {
|
|
107
|
-
throw new Error('
|
|
101
|
+
throw new Error('AxFunction not found');
|
|
108
102
|
}
|
|
109
103
|
return {
|
|
110
104
|
call: { name: fn.name, parameters: fn.parameter_definitions },
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../../src/ai/cohere/api.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../../src/ai/cohere/api.ts"],"names":[],"mappings":"AACA,OAAO,EACL,QAAQ,EACR,qBAAqB,EACrB,6BAA6B,EAC9B,MAAM,YAAY,CAAC;AAWpB,OAAO,EAAE,iBAAiB,EAAE,MAAM,WAAW,CAAC;AAC9C,OAAO,EAKL,kBAAkB,EAGlB,aAAa,EACd,MAAM,YAAY,CAAC;AAEpB,MAAM,CAAC,MAAM,qBAAqB,GAAG,GAAmB,EAAE,CACxD,eAAe,CAAC;IACd,KAAK,EAAE,aAAa,CAAC,OAAO;IAC5B,UAAU,EAAE,kBAAkB,CAAC,eAAe;IAC9C,GAAG,qBAAqB,EAAE;CAC3B,CAAC,CAAC;AAEL,MAAM,CAAC,MAAM,sBAAsB,GAAG,GAAmB,EAAE,CACzD,eAAe,CAAC;IACd,KAAK,EAAE,aAAa,CAAC,QAAQ;IAC7B,UAAU,EAAE,kBAAkB,CAAC,eAAe;IAC9C,GAAG,6BAA6B,EAAE;CACnC,CAAC,CAAC;AAQL,MAAM,OAAO,QAAS,SAAQ,QAM7B;IACS,MAAM,CAAiB;IAE/B,YAAY,EACV,MAAM,EACN,MAAM,GAAG,qBAAqB,EAAE,EAChC,OAAO,EACgB;QACvB,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAC5C,CAAC;QACD,KAAK,CAAC;YACJ,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,uBAAuB;YAC/B,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE;YAC9C,SAAS,EAAE,iBAAiB;YAC5B,MAAM,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE;YAC/B,UAAU,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE;YACjD,OAAO;SACR,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,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,EACP,EAAE;QAC9B,MAAM,KAAK,GAAG,GAAG,CAAC,SAAS,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;QAEvD,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,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE,OAAO,CAAC,EAAE,CAAC;YACxC,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC1C,CAAC;QAED,MAAM,OAAO,GAAmC,WAAW,EAAE,OAAO,IAAI,EAAE,CAAC;QAE3E,MAAM,WAAW,GAAwC,UAAU,CAAC,GAAG,CACrE,CAAC,IAAI,EAAE,EAAE;YACP,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC7B,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;YACpD,CAAC;YAED,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;gBAChC,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;YACvD,CAAC;YAED,IAAI,IAAoD,CAAC;YACzD,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;gBAClB,KAAK,MAAM;oBACT,IAAI,GAAG,MAAM,CAAC;oBACd,MAAM;gBACR,KAAK,QAAQ;oBACX,IAAI,GAAG,QAAQ,CAAC;oBAChB,MAAM;gBACR,KAAK,WAAW;oBACd,IAAI,GAAG,SAAS,CAAC;oBACjB,MAAM;gBACR;oBACE,IAAI,GAAG,MAAM,CAAC;oBACd,MAAM;YACV,CAAC;YACD,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;QAC/C,CAAC,CACF,CAAC;QAMF,MAAM,KAAK,GAAiC,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACnE,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;YACD,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,sBAAsB,CAAC,CAAC;YAC1C,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,OAAO,IAAI,EAAE,EAAE,CAAC;aAC1C,CAAC;QACJ,CAAC,CAAC,CAAC;QAEL,MAAM,SAAS,GAAG;YAChB,IAAI,EAAE,cAAc;SACrB,CAAC;QAEF,MAAM,QAAQ,GAAwB;YACpC,KAAK;YACL,OAAO;YACP,KAAK;YACL,YAAY;YACZ,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,EACA,EAAE;QAC/B,MAAM,KAAK,GAAG,GAAG,CAAC,cAAc,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;QAEjE,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,IAAoC,EACpB,EAAE;QAClB,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;gBACX,IAAI,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;oBACzB,OAAO;wBACL,EAAE,EAAE,CAAC,CAAC,IAAI;wBACV,IAAI,EAAE,UAAmB;wBACzB,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,UAAU,EAAE;qBAC/C,CAAC;gBACJ,CAAC,CAAC,IAAI,EAAE,CAAC;QACb,CAAC;QAED,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,EAAE,EAAE,IAAI,CAAC,aAAa;oBACtB,OAAO,EAAE,IAAI,CAAC,IAAI;oBAClB,aAAa;oBACb,YAAY;iBACb;aACF;SACF,CAAC;IACJ,CAAC,CAAC;IAEO,sBAAsB,GAAG,CAChC,IAAyC,EACzC,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,IAAqC,EACpB,EAAE;QACnB,OAAO;YACL,QAAQ,EAAE,IAAI,CAAC,EAAE;YACjB,UAAU,EAAE,IAAI,CAAC,UAAU;SAC5B,CAAC;IACJ,CAAC,CAAC;CACH"}
|
|
@@ -1,2 +1,2 @@
|
|
|
1
|
-
import type {
|
|
2
|
-
export declare const
|
|
1
|
+
import type { AxModelInfo } from '../types.js';
|
|
2
|
+
export declare const axModelInfoCohere: AxModelInfo[];
|
|
@@ -1,49 +1,49 @@
|
|
|
1
|
-
import {
|
|
2
|
-
export const
|
|
1
|
+
import { AxCohereEmbedModel, AxCohereModel } from './types.js';
|
|
2
|
+
export const axModelInfoCohere = [
|
|
3
3
|
{
|
|
4
|
-
name:
|
|
4
|
+
name: AxCohereModel.CommandRPlus,
|
|
5
5
|
currency: 'usd',
|
|
6
6
|
promptTokenCostPer1M: 3.0,
|
|
7
7
|
completionTokenCostPer1M: 15
|
|
8
8
|
},
|
|
9
9
|
{
|
|
10
|
-
name:
|
|
10
|
+
name: AxCohereModel.CommandR,
|
|
11
11
|
currency: 'usd',
|
|
12
12
|
promptTokenCostPer1M: 0.5,
|
|
13
13
|
completionTokenCostPer1M: 1.5
|
|
14
14
|
},
|
|
15
15
|
{
|
|
16
|
-
name:
|
|
16
|
+
name: AxCohereModel.Command,
|
|
17
17
|
currency: 'usd',
|
|
18
18
|
promptTokenCostPer1M: 0.5,
|
|
19
19
|
completionTokenCostPer1M: 1.5
|
|
20
20
|
},
|
|
21
21
|
{
|
|
22
|
-
name:
|
|
22
|
+
name: AxCohereModel.CommandLight,
|
|
23
23
|
currency: 'usd',
|
|
24
24
|
promptTokenCostPer1M: 0.3,
|
|
25
25
|
completionTokenCostPer1M: 0.6
|
|
26
26
|
},
|
|
27
27
|
{
|
|
28
|
-
name:
|
|
28
|
+
name: AxCohereEmbedModel.EmbedEnglishLightV30,
|
|
29
29
|
currency: 'usd',
|
|
30
30
|
promptTokenCostPer1M: 0.1,
|
|
31
31
|
completionTokenCostPer1M: 0.1
|
|
32
32
|
},
|
|
33
33
|
{
|
|
34
|
-
name:
|
|
34
|
+
name: AxCohereEmbedModel.EmbedEnglishV30,
|
|
35
35
|
currency: 'usd',
|
|
36
36
|
promptTokenCostPer1M: 0.1,
|
|
37
37
|
completionTokenCostPer1M: 0.1
|
|
38
38
|
},
|
|
39
39
|
{
|
|
40
|
-
name:
|
|
40
|
+
name: AxCohereEmbedModel.EmbedMultiLingualV30,
|
|
41
41
|
currency: 'usd',
|
|
42
42
|
promptTokenCostPer1M: 0.1,
|
|
43
43
|
completionTokenCostPer1M: 0.1
|
|
44
44
|
},
|
|
45
45
|
{
|
|
46
|
-
name:
|
|
46
|
+
name: AxCohereEmbedModel.EmbedMultiLingualLightV30,
|
|
47
47
|
currency: 'usd',
|
|
48
48
|
promptTokenCostPer1M: 0.1,
|
|
49
49
|
completionTokenCostPer1M: 0.1
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"info.js","sourceRoot":"","sources":["../../../../../src/ai/cohere/info.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"info.js","sourceRoot":"","sources":["../../../../../src/ai/cohere/info.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,kBAAkB,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAE/D,MAAM,CAAC,MAAM,iBAAiB,GAAkB;IAC9C;QACE,IAAI,EAAE,aAAa,CAAC,YAAY;QAChC,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,EAAE;KAC7B;IACD;QACE,IAAI,EAAE,aAAa,CAAC,QAAQ;QAC5B,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,aAAa,CAAC,OAAO;QAC3B,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,aAAa,CAAC,YAAY;QAChC,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,kBAAkB,CAAC,oBAAoB;QAC7C,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,kBAAkB,CAAC,eAAe;QACxC,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,kBAAkB,CAAC,oBAAoB;QAC7C,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,kBAAkB,CAAC,yBAAyB;QAClD,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;CACF,CAAC"}
|
|
@@ -1,9 +1,9 @@
|
|
|
1
|
-
import type {
|
|
1
|
+
import type { AxModelConfig } from '../types.js';
|
|
2
2
|
/**
|
|
3
3
|
* Cohere: Models for text generation
|
|
4
4
|
* @export
|
|
5
5
|
*/
|
|
6
|
-
export declare enum
|
|
6
|
+
export declare enum AxCohereModel {
|
|
7
7
|
CommandRPlus = "command-r-plus",
|
|
8
8
|
CommandR = "command-r",
|
|
9
9
|
Command = "command",
|
|
@@ -13,7 +13,7 @@ export declare enum CohereModel {
|
|
|
13
13
|
* Cohere: Models for use in embeddings
|
|
14
14
|
* @export
|
|
15
15
|
*/
|
|
16
|
-
export declare enum
|
|
16
|
+
export declare enum AxCohereEmbedModel {
|
|
17
17
|
EmbedEnglishV30 = "embed-english-v3.0",
|
|
18
18
|
EmbedEnglishLightV30 = "embed-english-light-v3.0",
|
|
19
19
|
EmbedMultiLingualV30 = "embed-multilingual-v3.0",
|
|
@@ -23,18 +23,18 @@ export declare enum CohereEmbedModel {
|
|
|
23
23
|
* Cohere: Model options for text generation
|
|
24
24
|
* @export
|
|
25
25
|
*/
|
|
26
|
-
export type
|
|
27
|
-
model:
|
|
28
|
-
embedModel:
|
|
26
|
+
export type AxCohereConfig = AxModelConfig & {
|
|
27
|
+
model: AxCohereModel;
|
|
28
|
+
embedModel: AxCohereEmbedModel;
|
|
29
29
|
};
|
|
30
|
-
export type
|
|
30
|
+
export type AxCohereChatRequest = {
|
|
31
31
|
message: string;
|
|
32
32
|
preamble?: string;
|
|
33
33
|
chat_history: {
|
|
34
34
|
role: 'CHATBOT' | 'SYSTEM' | 'USER';
|
|
35
35
|
message: string;
|
|
36
36
|
}[];
|
|
37
|
-
model:
|
|
37
|
+
model: AxCohereModel | string;
|
|
38
38
|
max_tokens?: number;
|
|
39
39
|
temperature?: number;
|
|
40
40
|
k?: number;
|
|
@@ -60,7 +60,7 @@ export type CohereChatRequest = {
|
|
|
60
60
|
outputs: object[];
|
|
61
61
|
}[];
|
|
62
62
|
};
|
|
63
|
-
export type
|
|
63
|
+
export type AxCohereChatResponse = {
|
|
64
64
|
generation_id: string;
|
|
65
65
|
text: string;
|
|
66
66
|
finish_reason: 'COMPLETE' | 'ERROR' | 'ERROR_TOXIC' | 'ERROR_LIMIT' | 'USER_CANCEL' | 'MAX_TOKENS';
|
|
@@ -69,15 +69,15 @@ export type CohereChatResponse = {
|
|
|
69
69
|
parameters: object;
|
|
70
70
|
}[];
|
|
71
71
|
};
|
|
72
|
-
export type
|
|
72
|
+
export type AxCohereChatResponseDelta = AxCohereChatResponse & {
|
|
73
73
|
event_type: 'stream-start' | 'text-generation' | 'tool-calls-generation' | 'stream-end';
|
|
74
74
|
};
|
|
75
|
-
export type
|
|
75
|
+
export type AxCohereEmbedRequest = {
|
|
76
76
|
texts: readonly string[];
|
|
77
|
-
model:
|
|
77
|
+
model: AxCohereModel | string;
|
|
78
78
|
truncate: string;
|
|
79
79
|
};
|
|
80
|
-
export type
|
|
80
|
+
export type AxCohereEmbedResponse = {
|
|
81
81
|
id: string;
|
|
82
82
|
texts: string[];
|
|
83
83
|
model: string;
|
|
@@ -2,22 +2,22 @@
|
|
|
2
2
|
* Cohere: Models for text generation
|
|
3
3
|
* @export
|
|
4
4
|
*/
|
|
5
|
-
export var
|
|
6
|
-
(function (
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
})(
|
|
5
|
+
export var AxCohereModel;
|
|
6
|
+
(function (AxCohereModel) {
|
|
7
|
+
AxCohereModel["CommandRPlus"] = "command-r-plus";
|
|
8
|
+
AxCohereModel["CommandR"] = "command-r";
|
|
9
|
+
AxCohereModel["Command"] = "command";
|
|
10
|
+
AxCohereModel["CommandLight"] = "command-light";
|
|
11
|
+
})(AxCohereModel || (AxCohereModel = {}));
|
|
12
12
|
/**
|
|
13
13
|
* Cohere: Models for use in embeddings
|
|
14
14
|
* @export
|
|
15
15
|
*/
|
|
16
|
-
export var
|
|
17
|
-
(function (
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
})(
|
|
16
|
+
export var AxCohereEmbedModel;
|
|
17
|
+
(function (AxCohereEmbedModel) {
|
|
18
|
+
AxCohereEmbedModel["EmbedEnglishV30"] = "embed-english-v3.0";
|
|
19
|
+
AxCohereEmbedModel["EmbedEnglishLightV30"] = "embed-english-light-v3.0";
|
|
20
|
+
AxCohereEmbedModel["EmbedMultiLingualV30"] = "embed-multilingual-v3.0";
|
|
21
|
+
AxCohereEmbedModel["EmbedMultiLingualLightV30"] = "embed-multilingual-light-v3.0";
|
|
22
|
+
})(AxCohereEmbedModel || (AxCohereEmbedModel = {}));
|
|
23
23
|
//# sourceMappingURL=types.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../../src/ai/cohere/types.ts"],"names":[],"mappings":"AAEA;;;GAGG;AACH,MAAM,CAAN,IAAY,
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../../src/ai/cohere/types.ts"],"names":[],"mappings":"AAEA;;;GAGG;AACH,MAAM,CAAN,IAAY,aAKX;AALD,WAAY,aAAa;IACvB,gDAA+B,CAAA;IAC/B,uCAAsB,CAAA;IACtB,oCAAmB,CAAA;IACnB,+CAA8B,CAAA;AAChC,CAAC,EALW,aAAa,KAAb,aAAa,QAKxB;AAED;;;GAGG;AACH,MAAM,CAAN,IAAY,kBAKX;AALD,WAAY,kBAAkB;IAC5B,4DAAsC,CAAA;IACtC,uEAAiD,CAAA;IACjD,sEAAgD,CAAA;IAChD,iFAA2D,CAAA;AAC7D,CAAC,EALW,kBAAkB,KAAlB,kBAAkB,QAK7B"}
|
|
@@ -1,27 +1,15 @@
|
|
|
1
|
-
import
|
|
2
|
-
import {
|
|
3
|
-
import type {
|
|
4
|
-
type DeepSeekConfig =
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
*/
|
|
9
|
-
export declare const DeepSeekDefaultConfig: () => DeepSeekConfig;
|
|
10
|
-
/**
|
|
11
|
-
* DeepSeek: Default Model options for text generation
|
|
12
|
-
* @export
|
|
13
|
-
*/
|
|
14
|
-
export declare const DeepSeekCodeConfig: () => DeepSeekConfig;
|
|
15
|
-
export interface DeepSeekArgs {
|
|
1
|
+
import { AxOpenAI } from '../openai/api.js';
|
|
2
|
+
import type { AxOpenAIConfig } from '../openai/types.js';
|
|
3
|
+
import type { AxAIServiceOptions } from '../types.js';
|
|
4
|
+
type DeepSeekConfig = AxOpenAIConfig;
|
|
5
|
+
export declare const axDeepSeekDefaultConfig: () => DeepSeekConfig;
|
|
6
|
+
export declare const axDeepSeekCodeConfig: () => DeepSeekConfig;
|
|
7
|
+
export interface AxDeepSeekArgs {
|
|
16
8
|
apiKey: string;
|
|
17
9
|
config: Readonly<DeepSeekConfig>;
|
|
18
|
-
options?: Readonly<
|
|
10
|
+
options?: Readonly<AxAIServiceOptions>;
|
|
19
11
|
}
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
* @export
|
|
23
|
-
*/
|
|
24
|
-
export declare class DeepSeek extends OpenAI {
|
|
25
|
-
constructor({ apiKey, config, options }: Readonly<DeepSeekArgs>);
|
|
12
|
+
export declare class AxDeepSeek extends AxOpenAI {
|
|
13
|
+
constructor({ apiKey, config, options }: Readonly<AxDeepSeekArgs>);
|
|
26
14
|
}
|
|
27
15
|
export {};
|