@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
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
import type { AxChatRequest, AxChatResponseResult } from '../ai/types.js';
|
|
2
|
+
import type { AxAIMemory } from './types.js';
|
|
3
|
+
export declare class AxMemory implements AxAIMemory {
|
|
4
|
+
private data;
|
|
5
|
+
private sdata;
|
|
6
|
+
private limit;
|
|
7
|
+
constructor(limit?: number);
|
|
8
|
+
add(value: Readonly<AxChatRequest['chatPrompt'][0] | AxChatRequest['chatPrompt']>, sessionId?: string): void;
|
|
9
|
+
addResult({ content, name, functionCalls }: Readonly<AxChatResponseResult>, sessionId?: string): void;
|
|
10
|
+
updateResult({ content, name, functionCalls }: Readonly<AxChatResponseResult>, sessionId?: string): void;
|
|
11
|
+
history(sessionId?: string): AxChatRequest['chatPrompt'];
|
|
12
|
+
peek(sessionId?: string): AxChatRequest['chatPrompt'];
|
|
13
|
+
getLast(sessionId?: string): AxChatRequest['chatPrompt'][0] | undefined;
|
|
14
|
+
reset(sessionId?: string): void;
|
|
15
|
+
private get;
|
|
16
|
+
}
|
|
@@ -0,0 +1,74 @@
|
|
|
1
|
+
export class AxMemory {
|
|
2
|
+
data = [];
|
|
3
|
+
sdata = new Map();
|
|
4
|
+
limit;
|
|
5
|
+
constructor(limit = 50) {
|
|
6
|
+
if (limit <= 0) {
|
|
7
|
+
throw Error("argument 'last' must be greater than 0");
|
|
8
|
+
}
|
|
9
|
+
this.limit = limit;
|
|
10
|
+
}
|
|
11
|
+
add(value, sessionId) {
|
|
12
|
+
const d = this.get(sessionId);
|
|
13
|
+
let n = 0;
|
|
14
|
+
if (Array.isArray(value)) {
|
|
15
|
+
n = d.push(...structuredClone(value));
|
|
16
|
+
}
|
|
17
|
+
else {
|
|
18
|
+
n = d.push({
|
|
19
|
+
...structuredClone(value)
|
|
20
|
+
});
|
|
21
|
+
}
|
|
22
|
+
if (d.length > this.limit) {
|
|
23
|
+
d.splice(0, this.limit + n - this.limit);
|
|
24
|
+
}
|
|
25
|
+
}
|
|
26
|
+
addResult({ content, name, functionCalls }, sessionId) {
|
|
27
|
+
this.add({ content, name, role: 'assistant', functionCalls }, sessionId);
|
|
28
|
+
}
|
|
29
|
+
updateResult({ content, name, functionCalls }, sessionId) {
|
|
30
|
+
const item = this.get(sessionId);
|
|
31
|
+
const lastItem = item.at(-1);
|
|
32
|
+
if (!lastItem || lastItem.role !== 'assistant') {
|
|
33
|
+
this.addResult({ content, name, functionCalls }, sessionId);
|
|
34
|
+
return;
|
|
35
|
+
}
|
|
36
|
+
if ('content' in lastItem && content) {
|
|
37
|
+
lastItem.content = content;
|
|
38
|
+
}
|
|
39
|
+
if ('name' in lastItem && name) {
|
|
40
|
+
lastItem.name = name;
|
|
41
|
+
}
|
|
42
|
+
if ('functionCalls' in lastItem && functionCalls) {
|
|
43
|
+
lastItem.functionCalls = functionCalls;
|
|
44
|
+
}
|
|
45
|
+
}
|
|
46
|
+
history(sessionId) {
|
|
47
|
+
return this.get(sessionId);
|
|
48
|
+
}
|
|
49
|
+
peek(sessionId) {
|
|
50
|
+
return this.get(sessionId);
|
|
51
|
+
}
|
|
52
|
+
getLast(sessionId) {
|
|
53
|
+
const d = this.get(sessionId);
|
|
54
|
+
return d.at(-1);
|
|
55
|
+
}
|
|
56
|
+
reset(sessionId) {
|
|
57
|
+
if (!sessionId) {
|
|
58
|
+
this.data = [];
|
|
59
|
+
}
|
|
60
|
+
else {
|
|
61
|
+
this.sdata.set(sessionId, []);
|
|
62
|
+
}
|
|
63
|
+
}
|
|
64
|
+
get(sessionId) {
|
|
65
|
+
if (!sessionId) {
|
|
66
|
+
return this.data;
|
|
67
|
+
}
|
|
68
|
+
if (!this.sdata.has(sessionId)) {
|
|
69
|
+
this.sdata.set(sessionId, []);
|
|
70
|
+
}
|
|
71
|
+
return this.sdata.get(sessionId) || [];
|
|
72
|
+
}
|
|
73
|
+
}
|
|
74
|
+
//# sourceMappingURL=memory.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory.js","sourceRoot":"","sources":["../../../../src/mem/memory.ts"],"names":[],"mappings":"AAMA,MAAM,OAAO,QAAQ;IACX,IAAI,GAAgC,EAAE,CAAC;IACvC,KAAK,GAAG,IAAI,GAAG,EAAuC,CAAC;IACvD,KAAK,CAAS;IAEtB,YAAY,KAAK,GAAG,EAAE;QACpB,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;YACf,MAAM,KAAK,CAAC,wCAAwC,CAAC,CAAC;QACxD,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,GAAG,CACD,KAEC,EACD,SAAkB;QAElB,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC9B,IAAI,CAAC,GAAG,CAAC,CAAC;QAEV,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzB,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC;QACxC,CAAC;aAAM,CAAC;YACN,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;gBACT,GAAG,eAAe,CAAC,KAAK,CAAC;aACQ,CAAC,CAAC;QACvC,CAAC;QACD,IAAI,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;YAC1B,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAED,SAAS,CACP,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAkC,EAChE,SAAkB;QAElB,IAAI,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,aAAa,EAAE,EAAE,SAAS,CAAC,CAAC;IAC3E,CAAC;IAED,YAAY,CACV,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAkC,EAChE,SAAkB;QAElB,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAEjC,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAE1B,CAAC;QACF,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC/C,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,SAAS,CAAC,CAAC;YAC5D,OAAO;QACT,CAAC;QAED,IAAI,SAAS,IAAI,QAAQ,IAAI,OAAO,EAAE,CAAC;YACrC,QAAQ,CAAC,OAAO,GAAG,OAAO,CAAC;QAC7B,CAAC;QACD,IAAI,MAAM,IAAI,QAAQ,IAAI,IAAI,EAAE,CAAC;YAC/B,QAAQ,CAAC,IAAI,GAAG,IAAI,CAAC;QACvB,CAAC;QACD,IAAI,eAAe,IAAI,QAAQ,IAAI,aAAa,EAAE,CAAC;YACjD,QAAQ,CAAC,aAAa,GAAG,aAAa,CAAC;QACzC,CAAC;IACH,CAAC;IAED,OAAO,CAAC,SAAkB;QACxB,OAAO,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC7B,CAAC;IAED,IAAI,CAAC,SAAkB;QACrB,OAAO,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC7B,CAAC;IAED,OAAO,CAAC,SAAkB;QACxB,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC9B,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,SAAkB;QACtB,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,IAAI,CAAC,IAAI,GAAG,EAAE,CAAC;QACjB,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAEO,GAAG,CAAC,SAAkB;QAC5B,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,OAAO,IAAI,CAAC,IAAI,CAAC;QACnB,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAChC,CAAC;QAED,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;IACzC,CAAC;CACF"}
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
import type { AxChatRequest, AxChatResponseResult } from '../ai/index.js';
|
|
2
|
+
export interface AxAIMemory {
|
|
3
|
+
add(result: Readonly<AxChatRequest['chatPrompt'] | AxChatRequest['chatPrompt'][0]>, sessionId?: string): void;
|
|
4
|
+
addResult(result: Readonly<AxChatResponseResult>, sessionId?: string): void;
|
|
5
|
+
updateResult(result: Readonly<AxChatResponseResult>, sessionId?: string): void;
|
|
6
|
+
history(sessionId?: string): AxChatRequest['chatPrompt'];
|
|
7
|
+
peek(sessionId?: string): AxChatRequest['chatPrompt'];
|
|
8
|
+
reset(sessionId?: string): void;
|
|
9
|
+
getLast(sessionId?: string): AxChatRequest['chatPrompt'][0] | undefined;
|
|
10
|
+
}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../src/mem/types.ts"],"names":[],"mappings":""}
|
|
@@ -1,24 +1,23 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import { type
|
|
3
|
-
import { type
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
getFunction(): AITextFunction;
|
|
1
|
+
import type { AxAIService, AxFunction } from '../ai/types.js';
|
|
2
|
+
import { type AxGenerateOptions, AxSignature } from '../dsp/index.js';
|
|
3
|
+
import { type AxGenIn, type AxGenOut, AxProgram, type AxProgramForwardOptions, type AxTunable, type AxUsable } from '../dsp/program.js';
|
|
4
|
+
export interface AxAgentic extends AxTunable, AxUsable {
|
|
5
|
+
getFunction(): AxFunction;
|
|
7
6
|
}
|
|
8
|
-
export type
|
|
9
|
-
export declare class
|
|
7
|
+
export type AxAgentOptions = Omit<AxGenerateOptions, 'functions' | 'functionCall'>;
|
|
8
|
+
export declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut> extends AxProgram<IN, OUT> implements AxAgentic {
|
|
10
9
|
private gen;
|
|
11
10
|
private name;
|
|
12
11
|
private description;
|
|
13
12
|
private subAgentList?;
|
|
14
13
|
private func;
|
|
15
|
-
constructor(ai:
|
|
14
|
+
constructor(ai: AxAIService, { name, description, signature, agents, functions }: Readonly<{
|
|
16
15
|
name: string;
|
|
17
16
|
description: string;
|
|
18
|
-
signature:
|
|
19
|
-
agents?:
|
|
20
|
-
functions?:
|
|
21
|
-
}>, options?: Readonly<
|
|
22
|
-
getFunction():
|
|
23
|
-
forward(values: IN, options?: Readonly<
|
|
17
|
+
signature: AxSignature | string;
|
|
18
|
+
agents?: AxAgentic[];
|
|
19
|
+
functions?: AxFunction[];
|
|
20
|
+
}>, options?: Readonly<AxAgentOptions>);
|
|
21
|
+
getFunction(): AxFunction;
|
|
22
|
+
forward(values: IN, options?: Readonly<AxProgramForwardOptions>): Promise<OUT>;
|
|
24
23
|
}
|
|
@@ -1,10 +1,9 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import {
|
|
3
|
-
import {} from '../
|
|
4
|
-
import {
|
|
5
|
-
import {
|
|
6
|
-
|
|
7
|
-
export class Agent extends Program {
|
|
1
|
+
import { AxSignature } from '../dsp/index.js';
|
|
2
|
+
import { AxProgram } from '../dsp/program.js';
|
|
3
|
+
import { AxSpanKind } from '../trace/index.js';
|
|
4
|
+
import { AxChainOfThought } from './cot.js';
|
|
5
|
+
import { AxReAct } from './react.js';
|
|
6
|
+
export class AxAgent extends AxProgram {
|
|
8
7
|
gen;
|
|
9
8
|
name;
|
|
10
9
|
description;
|
|
@@ -12,7 +11,7 @@ export class Agent extends Program {
|
|
|
12
11
|
func;
|
|
13
12
|
constructor(ai, { name, description, signature, agents, functions }, options) {
|
|
14
13
|
super();
|
|
15
|
-
const sig = new
|
|
14
|
+
const sig = new AxSignature(signature);
|
|
16
15
|
const funcs = [
|
|
17
16
|
...(functions ?? []),
|
|
18
17
|
...(agents?.map((a) => a.getFunction()) ?? [])
|
|
@@ -24,8 +23,8 @@ export class Agent extends Program {
|
|
|
24
23
|
};
|
|
25
24
|
this.gen =
|
|
26
25
|
funcs.length > 0
|
|
27
|
-
? new
|
|
28
|
-
: new
|
|
26
|
+
? new AxReAct(ai, sig, opt)
|
|
27
|
+
: new AxChainOfThought(ai, sig, opt);
|
|
29
28
|
this.name = name;
|
|
30
29
|
this.description = description;
|
|
31
30
|
this.subAgentList = agents?.map((a) => a.getFunction().name).join(', ');
|
|
@@ -53,7 +52,7 @@ export class Agent extends Program {
|
|
|
53
52
|
['agent.subAgents']: this.subAgentList ?? 'none'
|
|
54
53
|
};
|
|
55
54
|
return await options?.tracer.startActiveSpan('Agent', {
|
|
56
|
-
kind:
|
|
55
|
+
kind: AxSpanKind.SERVER,
|
|
57
56
|
attributes
|
|
58
57
|
}, async (span) => {
|
|
59
58
|
const res = await this.gen.forward(values, options);
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../../../src/prompts/agent.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../../../src/prompts/agent.ts"],"names":[],"mappings":"AACA,OAAO,EAA0B,WAAW,EAAE,MAAM,iBAAiB,CAAC;AACtE,OAAO,EAGL,SAAS,EAIV,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAE/C,OAAO,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAC5C,OAAO,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAWrC,MAAM,OAAO,OACX,SAAQ,SAAkB;IAGlB,GAAG,CAAqB;IAExB,IAAI,CAAS;IACb,WAAW,CAAS;IACpB,YAAY,CAAU;IACtB,IAAI,CAAa;IAEzB,YACE,EAAe,EACf,EACE,IAAI,EACJ,WAAW,EACX,SAAS,EACT,MAAM,EACN,SAAS,EAOT,EACF,OAAkC;QAElC,KAAK,EAAE,CAAC;QAER,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC,SAAS,CAAC,CAAC;QAEvC,MAAM,KAAK,GAAiB;YAC1B,GAAG,CAAC,SAAS,IAAI,EAAE,CAAC;YACpB,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC;SAC/C,CAAC;QAEF,MAAM,GAAG,GAAG;YACV,cAAc,EAAE,OAAO,EAAE,cAAc;YACvC,OAAO,EAAE,OAAO,EAAE,OAAO;YACzB,SAAS,EAAE,KAAK;SACjB,CAAC;QAEF,IAAI,CAAC,GAAG;YACN,KAAK,CAAC,MAAM,GAAG,CAAC;gBACd,CAAC,CAAC,IAAI,OAAO,CAAU,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC;gBACpC,CAAC,CAAC,IAAI,gBAAgB,CAAU,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAElD,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,YAAY,GAAG,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxE,IAAI,CAAC,IAAI,GAAG;YACV,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,GAAG,CAAC,YAAY,EAAE;YAC9B,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO;SACzB,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAExB,KAAK,MAAM,KAAK,IAAI,MAAM,IAAI,EAAE,EAAE,CAAC;YACjC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;IAEM,WAAW;QAChB,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAEe,KAAK,CAAC,OAAO,CAC3B,MAAU,EACV,OAA2C;QAE3C,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;YACrB,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACjD,CAAC;QAED,MAAM,UAAU,GAAG;YACjB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,IAAI;YACzB,CAAC,mBAAmB,CAAC,EAAE,IAAI,CAAC,WAAW;YACvC,CAAC,iBAAiB,CAAC,EAAE,IAAI,CAAC,YAAY,IAAI,MAAM;SACjD,CAAC;QAEF,OAAO,MAAM,OAAO,EAAE,MAAM,CAAC,eAAe,CAC1C,OAAO,EACP;YACE,IAAI,EAAE,UAAU,CAAC,MAAM;YACvB,UAAU;SACX,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACpD,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,GAAG,CAAC;QACb,CAAC,CACF,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -1,9 +1,9 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import
|
|
3
|
-
import {
|
|
4
|
-
import
|
|
5
|
-
export declare class
|
|
1
|
+
import type { AxAIService } from '../ai/types.js';
|
|
2
|
+
import { AxGenerate, type AxGenerateOptions } from '../dsp/generate.js';
|
|
3
|
+
import type { AxGenIn, AxGenOut } from '../dsp/program.js';
|
|
4
|
+
import { AxSignature } from '../dsp/sig.js';
|
|
5
|
+
export declare class AxChainOfThought<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGenOut> extends AxGenerate<IN, OUT & {
|
|
6
6
|
reason: string;
|
|
7
7
|
}> {
|
|
8
|
-
constructor(ai:
|
|
8
|
+
constructor(ai: AxAIService, signature: Readonly<AxSignature | string>, options?: Readonly<AxGenerateOptions>);
|
|
9
9
|
}
|
|
@@ -1,8 +1,8 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import {
|
|
3
|
-
export class
|
|
1
|
+
import { AxGenerate } from '../dsp/generate.js';
|
|
2
|
+
import { AxSignature } from '../dsp/sig.js';
|
|
3
|
+
export class AxChainOfThought extends AxGenerate {
|
|
4
4
|
constructor(ai, signature, options) {
|
|
5
|
-
const sig = new
|
|
5
|
+
const sig = new AxSignature(signature);
|
|
6
6
|
const description = `Let's work this out in a step by step way in order to ensure we have the right answer.`;
|
|
7
7
|
sig.setOutputFields([
|
|
8
8
|
{
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"cot.js","sourceRoot":"","sources":["../../../../src/prompts/cot.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"cot.js","sourceRoot":"","sources":["../../../../src/prompts/cot.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAA0B,MAAM,oBAAoB,CAAC;AAExE,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAE5C,MAAM,OAAO,gBAGX,SAAQ,UAAwC;IAChD,YACE,EAAe,EACf,SAAyC,EACzC,OAAqC;QAErC,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC,SAAS,CAAC,CAAC;QACvC,MAAM,WAAW,GAAG,wFAAwF,CAAC;QAE7G,GAAG,CAAC,eAAe,CAAC;YAClB;gBACE,IAAI,EAAE,QAAQ;gBACd,WAAW;aACZ;YACD,GAAG,GAAG,CAAC,eAAe,EAAE;SACzB,CAAC,CAAC;QAEH,KAAK,CAAC,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;IAC1B,CAAC;CACF"}
|
|
@@ -1,7 +1,8 @@
|
|
|
1
1
|
import { Readable } from 'stream';
|
|
2
2
|
import test from 'ava';
|
|
3
|
-
import {
|
|
4
|
-
import {
|
|
3
|
+
import { axAI } from '../ai/index.js';
|
|
4
|
+
import { AxSignature } from '../dsp/sig.js';
|
|
5
|
+
import { AxChainOfThought } from './cot.js';
|
|
5
6
|
const text = `The technological singularity—or simply the singularity[1]—is a hypothetical future point in time at which technological growth becomes uncontrollable and irreversible.`;
|
|
6
7
|
const examples = [
|
|
7
8
|
{
|
|
@@ -35,9 +36,9 @@ const mockFetch = async () => {
|
|
|
35
36
|
};
|
|
36
37
|
test('generate prompt', async (t) => {
|
|
37
38
|
const options = { fetch: mockFetch };
|
|
38
|
-
const ai =
|
|
39
|
-
// const ai =
|
|
40
|
-
const gen = new
|
|
39
|
+
const ai = axAI('openai', { apiKey: 'no-key', options });
|
|
40
|
+
// const ai = axAI('ollama', { model: 'nous-hermes2' });
|
|
41
|
+
const gen = new AxChainOfThought(ai, `text -> shortSummary "summarize in 5 to 10 words"`);
|
|
41
42
|
gen.setExamples(examples);
|
|
42
43
|
const res = await gen.forward({ text });
|
|
43
44
|
t.deepEqual(res, {
|
|
@@ -45,4 +46,7 @@ test('generate prompt', async (t) => {
|
|
|
45
46
|
shortSummary: 'More blah blah blah'
|
|
46
47
|
});
|
|
47
48
|
});
|
|
49
|
+
test('generate prompt: invalid signature', async (t) => {
|
|
50
|
+
t.throws(() => new AxSignature(`text -> output:image`));
|
|
51
|
+
});
|
|
48
52
|
//# sourceMappingURL=prompts.test.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"prompts.test.js","sourceRoot":"","sources":["../../../../src/prompts/prompts.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AAElC,OAAO,IAAI,MAAM,KAAK,CAAC;AAEvB,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"prompts.test.js","sourceRoot":"","sources":["../../../../src/prompts/prompts.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AAElC,OAAO,IAAI,MAAM,KAAK,CAAC;AAEvB,OAAO,EAAE,IAAI,EAAqB,MAAM,gBAAgB,CAAC;AACzD,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAE5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAE5C,MAAM,IAAI,GAAG,0KAA0K,CAAC;AAExL,MAAM,QAAQ,GAAG;IACf;QACE,IAAI,EAAE,0TAA0T;QAChU,YAAY,EACV,+GAA+G;KAClH;IACD;QACE,IAAI,EAAE,gZAAgZ;QACtZ,YAAY,EACV,+HAA+H;KAClI;CACF,CAAC;AAEF,MAAM,SAAS,GAAG,KAAK,IAAuB,EAAE;IAC9C,MAAM,OAAO,GAAG;QACd,OAAO,EAAE;YACP;gBACE,OAAO,EAAE;oBACP,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,4DAA4D;iBACtE;aACF;SACF;KACF,CAAC;IAEF,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,OAAO,CAAC;YACN,EAAE,EAAE,IAAI;YACR,MAAM,EAAE,GAAG;YACX,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YAC5C,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;SACtC,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAEF,IAAI,CAAC,iBAAiB,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;IAClC,MAAM,OAAO,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;IACrC,MAAM,EAAE,GAAG,IAAI,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAkB,CAAC,CAAC;IAEzE,wDAAwD;IAExD,MAAM,GAAG,GAAG,IAAI,gBAAgB,CAC9B,EAAE,EACF,mDAAmD,CACpD,CAAC;IACF,GAAG,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IAE1B,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;IAExC,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE;QACf,MAAM,EAAE,gBAAgB;QACxB,YAAY,EAAE,qBAAqB;KACpC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,oCAAoC,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;IACrD,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,WAAW,CAAC,sBAAsB,CAAC,CAAC,CAAC;AAC1D,CAAC,CAAC,CAAC"}
|
|
@@ -1,8 +1,8 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import { type
|
|
3
|
-
import type
|
|
4
|
-
import {
|
|
5
|
-
export declare class
|
|
1
|
+
import type { AxAIService } from '../ai/types.js';
|
|
2
|
+
import { type AxGenerateOptions } from '../dsp/index.js';
|
|
3
|
+
import { type AxProgramForwardOptions } from '../dsp/program.js';
|
|
4
|
+
import { AxChainOfThought } from './cot.js';
|
|
5
|
+
export declare class AxRAG extends AxChainOfThought<{
|
|
6
6
|
context: string[];
|
|
7
7
|
question: string;
|
|
8
8
|
}, {
|
|
@@ -11,12 +11,12 @@ export declare class RAG extends ChainOfThought<{
|
|
|
11
11
|
private genQuery;
|
|
12
12
|
private queryFn;
|
|
13
13
|
private maxHops;
|
|
14
|
-
constructor(ai:
|
|
14
|
+
constructor(ai: AxAIService, queryFn: (query: string) => Promise<string>, options: Readonly<AxGenerateOptions & {
|
|
15
15
|
maxHops?: number;
|
|
16
16
|
}>);
|
|
17
17
|
forward({ question }: Readonly<{
|
|
18
18
|
question: string;
|
|
19
|
-
}>, options?: Readonly<
|
|
19
|
+
}>, options?: Readonly<AxProgramForwardOptions>): Promise<{
|
|
20
20
|
answer: string;
|
|
21
21
|
reason: string;
|
|
22
22
|
}>;
|
|
@@ -1,7 +1,7 @@
|
|
|
1
|
-
import {
|
|
1
|
+
import { AxGenerate, AxSignature, axStringUtil } from '../dsp/index.js';
|
|
2
2
|
import {} from '../dsp/program.js';
|
|
3
|
-
import {
|
|
4
|
-
export class
|
|
3
|
+
import { AxChainOfThought } from './cot.js';
|
|
4
|
+
export class AxRAG extends AxChainOfThought {
|
|
5
5
|
genQuery;
|
|
6
6
|
queryFn;
|
|
7
7
|
maxHops;
|
|
@@ -9,8 +9,8 @@ export class RAG extends ChainOfThought {
|
|
|
9
9
|
const sig = '"Answer questions with short factoid answers." context:string[] "may contain relevant facts", question -> answer';
|
|
10
10
|
super(ai, sig, options);
|
|
11
11
|
this.maxHops = options?.maxHops ?? 3;
|
|
12
|
-
const qsig = new
|
|
13
|
-
this.genQuery = new
|
|
12
|
+
const qsig = new AxSignature('"Write a simple search query that will help answer a complex question." context?:string[] "may contain relevant facts", question -> query "question to further our understanding"');
|
|
13
|
+
this.genQuery = new AxGenerate(ai, qsig);
|
|
14
14
|
this.queryFn = queryFn;
|
|
15
15
|
this.register(this.genQuery);
|
|
16
16
|
}
|
|
@@ -22,7 +22,7 @@ export class RAG extends ChainOfThought {
|
|
|
22
22
|
question
|
|
23
23
|
}, options);
|
|
24
24
|
const val = await this.queryFn(query);
|
|
25
|
-
context = dedup([...context, val]);
|
|
25
|
+
context = axStringUtil.dedup([...context, val]);
|
|
26
26
|
}
|
|
27
27
|
return super.forward({ context, question }, options);
|
|
28
28
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"rag.js","sourceRoot":"","sources":["../../../../src/prompts/rag.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"rag.js","sourceRoot":"","sources":["../../../../src/prompts/rag.ts"],"names":[],"mappings":"AACA,OAAO,EACL,UAAU,EAEV,WAAW,EACX,YAAY,EACb,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAgC,MAAM,mBAAmB,CAAC;AAEjE,OAAO,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAE5C,MAAM,OAAO,KAAM,SAAQ,gBAG1B;IACS,QAAQ,CAGd;IACM,OAAO,CAAqC;IAC5C,OAAO,CAAS;IAExB,YACE,EAAe,EACf,OAA2C,EAC3C,OAA2D;QAE3D,MAAM,GAAG,GACP,kHAAkH,CAAC;QACrH,KAAK,CAAC,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;QAExB,IAAI,CAAC,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,CAAC,CAAC;QAErC,MAAM,IAAI,GAAG,IAAI,WAAW,CAC1B,mLAAmL,CACpL,CAAC;QACF,IAAI,CAAC,QAAQ,GAAG,IAAI,UAAU,CAG5B,EAAE,EAAE,IAAI,CAAC,CAAC;QACZ,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC/B,CAAC;IAEe,KAAK,CAAC,OAAO,CAC3B,EAAE,QAAQ,EAAkC,EAC5C,OAA2C;QAE3C,IAAI,OAAO,GAAa,EAAE,CAAC;QAE3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;YACtC,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAC3C;gBACE,OAAO;gBACP,QAAQ;aACT,EACD,OAAO,CACR,CAAC;YACF,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACtC,OAAO,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,GAAG,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QAClD,CAAC;QAED,OAAO,KAAK,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,OAAO,CAAC,CAAC;IACvD,CAAC;CACF"}
|
|
@@ -1,8 +1,8 @@
|
|
|
1
|
-
import type {
|
|
2
|
-
import type {
|
|
3
|
-
import {
|
|
4
|
-
import
|
|
5
|
-
import {
|
|
6
|
-
export declare class
|
|
7
|
-
constructor(ai:
|
|
1
|
+
import type { AxAIService } from '../ai/index.js';
|
|
2
|
+
import type { AxGenerateOptions } from '../dsp/generate.js';
|
|
3
|
+
import type { AxGenIn, AxGenOut } from '../dsp/program.js';
|
|
4
|
+
import { AxSignature } from '../dsp/sig.js';
|
|
5
|
+
import { AxChainOfThought } from './cot.js';
|
|
6
|
+
export declare class AxReAct<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGenOut> extends AxChainOfThought<IN, OUT> {
|
|
7
|
+
constructor(ai: AxAIService, signature: Readonly<AxSignature | string>, options: Readonly<AxGenerateOptions>);
|
|
8
8
|
}
|
|
@@ -1,13 +1,13 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import {
|
|
3
|
-
export class
|
|
1
|
+
import { AxSignature } from '../dsp/sig.js';
|
|
2
|
+
import { AxChainOfThought } from './cot.js';
|
|
3
|
+
export class AxReAct extends AxChainOfThought {
|
|
4
4
|
constructor(ai, signature, options) {
|
|
5
5
|
if (!options?.functions || options.functions.length === 0) {
|
|
6
6
|
throw new Error('No functions provided');
|
|
7
7
|
}
|
|
8
8
|
const functions = [...options.functions];
|
|
9
9
|
const funcList = functions.map((f) => `'${f.name}'`).join(', ');
|
|
10
|
-
const sig = new
|
|
10
|
+
const sig = new AxSignature(signature);
|
|
11
11
|
sig.setDescription(`Use the provided functions ${funcList} to complete the task and return the result if any.`);
|
|
12
12
|
// sig.addInputField({
|
|
13
13
|
// name: 'observation',
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"react.js","sourceRoot":"","sources":["../../../../src/prompts/react.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"react.js","sourceRoot":"","sources":["../../../../src/prompts/react.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAE5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAE5C,MAAM,OAAO,OAGX,SAAQ,gBAAyB;IACjC,YACE,EAAe,EACf,SAAyC,EACzC,OAAoC;QAEpC,IAAI,CAAC,OAAO,EAAE,SAAS,IAAI,OAAO,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1D,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;QAED,MAAM,SAAS,GAAG,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;QACzC,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEhE,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC,SAAS,CAAC,CAAC;QACvC,GAAG,CAAC,cAAc,CAChB,8BAA8B,QAAQ,qDAAqD,CAC5F,CAAC;QAEF,sBAAsB;QACtB,yBAAyB;QACzB,wDAAwD;QACxD,sBAAsB;QACtB,4CAA4C;QAC5C,MAAM;QAEN,GAAG,CAAC,eAAe,CACjB,GAAG,CAAC,eAAe,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAC/D,CAAC;QAEF,KAAK,CAAC,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;IAC1B,CAAC;CACF"}
|
|
@@ -1,25 +1,8 @@
|
|
|
1
|
-
import type {
|
|
1
|
+
import type { ChatResponseFunctionCall } from '../ai/types.js';
|
|
2
2
|
import type { AIServiceActionOptions, FunctionExec } from './types.js';
|
|
3
|
-
export type AITextFunctionHandler = (args?: any, extra?: Readonly<AIServiceActionOptions>) => unknown;
|
|
4
|
-
export type FunctionJSONSchema = {
|
|
5
|
-
type: string;
|
|
6
|
-
properties?: Record<string, FunctionJSONSchema & {
|
|
7
|
-
enum?: string[];
|
|
8
|
-
default?: unknown;
|
|
9
|
-
description: string;
|
|
10
|
-
}>;
|
|
11
|
-
required?: string[];
|
|
12
|
-
items?: FunctionJSONSchema;
|
|
13
|
-
};
|
|
14
|
-
export type AITextFunction = {
|
|
15
|
-
name: string;
|
|
16
|
-
description: string;
|
|
17
|
-
parameters?: FunctionJSONSchema;
|
|
18
|
-
func?: AITextFunctionHandler;
|
|
19
|
-
};
|
|
20
3
|
export declare class FunctionProcessor {
|
|
21
4
|
private funcList;
|
|
22
|
-
constructor(funcList: readonly
|
|
5
|
+
constructor(funcList: readonly Function[]);
|
|
23
6
|
private executeFunction;
|
|
24
|
-
execute: (func: Readonly<
|
|
7
|
+
execute: (func: Readonly<ChatResponseFunctionCall>, options?: Readonly<AIServiceActionOptions>) => Promise<FunctionExec>;
|
|
25
8
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"functions.js","sourceRoot":"","sources":["../../../../src/text/functions.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAI1B,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;
|
|
1
|
+
{"version":3,"file":"functions.js","sourceRoot":"","sources":["../../../../src/text/functions.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAI1B,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAGrD,MAAM,OAAO,iBAAiB;IACpB,QAAQ,CAAsB;IAEtC,YAAY,QAA6B;QACvC,QAAQ;aACL,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC;aAC3B,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,CAAC,UAAW,CAAC,CAAC,CAAC;QACrD,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAEO,eAAe,GAAG,KAAK,EAC7B,MAA0B,EAC1B,IAAwC,EACxC,OAA0C,EACnB,EAAE;QACzB,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,wBAAwB,MAAM,CAAC,IAAI,kBAAkB,CAAC,CAAC;QACzE,CAAC;QAED,IAAI,MAAM,CAAC;QACX,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtC,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;YACvB,MAAM,GAAG,GACP,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;gBACtB,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC;gBAC5B,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAE1B,OAAO;gBACL,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC;aACrC,CAAC;QACJ,CAAC;QAED,MAAM,GAAG,GACP,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;YACtB,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC;YACpC,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEhC,OAAO;YACL,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC;SACrC,CAAC;IACJ,CAAC,CAAC;IAEK,OAAO,GAAG,KAAK,EACpB,IAAwC,EACxC,OAA0C,EACnB,EAAE;QACzB,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAC/B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAC7C,CAAC;QACF,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,sBAAsB,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC;QACtD,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,2BAA2B,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3D,CAAC;QAED,+BAA+B;QAC/B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;QAEnE,sCAAsC;QACtC,0DAA0D;QAC1D,sDAAsD;QACtD,IAAI;QACJ,OAAO,QAAQ,CAAC;IAClB,CAAC,CAAC;CACH"}
|
|
@@ -1,2 +1 @@
|
|
|
1
|
-
|
|
2
|
-
export declare const validateJSONSchema: (schema: Readonly<FunctionJSONSchema>) => void;
|
|
1
|
+
export declare const validateJSONSchema: (schema: FunctionJSONSchema) => void;
|
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import type {
|
|
2
|
-
import type {
|
|
1
|
+
import type { ChatResponseResult } from '../ai/types.js';
|
|
2
|
+
import type { ChatRequest } from '../types/index.js';
|
|
3
3
|
import type { AIMemory } from './types.js';
|
|
4
4
|
/**
|
|
5
5
|
* A memory class to store ai interactions
|
|
@@ -10,12 +10,12 @@ export declare class Memory implements AIMemory {
|
|
|
10
10
|
private sdata;
|
|
11
11
|
private limit;
|
|
12
12
|
constructor(limit?: number);
|
|
13
|
-
add(value: Readonly<
|
|
14
|
-
addResult({ content, name, functionCalls }: Readonly<
|
|
15
|
-
updateResult({ content, name, functionCalls }: Readonly<
|
|
16
|
-
history(sessionId?: string):
|
|
17
|
-
peek(sessionId?: string):
|
|
18
|
-
getLast(sessionId?: string):
|
|
13
|
+
add(value: Readonly<ChatRequest['chatPrompt'][0] | ChatRequest['chatPrompt']>, sessionId?: string): void;
|
|
14
|
+
addResult({ content, name, functionCalls }: Readonly<ChatResponseResult>, sessionId?: string): void;
|
|
15
|
+
updateResult({ content, name, functionCalls }: Readonly<ChatResponseResult>, sessionId?: string): void;
|
|
16
|
+
history(sessionId?: string): ChatRequest['chatPrompt'];
|
|
17
|
+
peek(sessionId?: string): ChatRequest['chatPrompt'];
|
|
18
|
+
getLast(sessionId?: string): ChatRequest['chatPrompt'][0] | undefined;
|
|
19
19
|
reset(sessionId?: string): void;
|
|
20
20
|
private get;
|
|
21
21
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"memory.js","sourceRoot":"","sources":["../../../../src/text/memory.ts"],"names":[],"mappings":"AAKA;;;GAGG;AACH,MAAM,OAAO,MAAM;IACT,IAAI,
|
|
1
|
+
{"version":3,"file":"memory.js","sourceRoot":"","sources":["../../../../src/text/memory.ts"],"names":[],"mappings":"AAKA;;;GAGG;AACH,MAAM,OAAO,MAAM;IACT,IAAI,GAA8B,EAAE,CAAC;IACrC,KAAK,GAAG,IAAI,GAAG,EAAqC,CAAC;IACrD,KAAK,CAAS;IAEtB,YAAY,KAAK,GAAG,EAAE;QACpB,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;YACf,MAAM,KAAK,CAAC,wCAAwC,CAAC,CAAC;QACxD,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,GAAG,CACD,KAAyE,EACzE,SAAkB;QAElB,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC9B,IAAI,CAAC,GAAG,CAAC,CAAC;QACV,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzB,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC;QACxC,CAAC;aAAM,CAAC;YACN,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;gBACT,GAAG,eAAe,CAAC,KAAK,CAAC;aACM,CAAC,CAAC;QACrC,CAAC;QACD,IAAI,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;YAC1B,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAED,SAAS,CACP,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAgC,EAC9D,SAAkB;QAElB,IAAI,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,aAAa,EAAE,EAAE,SAAS,CAAC,CAAC;IAC3E,CAAC;IAED,YAAY,CACV,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAgC,EAC9D,SAAkB;QAElB,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACjC,IAAI,SAAS,IAAI,IAAI,IAAI,OAAO,EAAE,CAAC;YACjC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACzB,CAAC;QACD,IAAI,MAAM,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;YAC3B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACnB,CAAC;QACD,IAAI,eAAe,IAAI,IAAI,IAAI,aAAa,EAAE,CAAC;YAC7C,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACrC,CAAC;IACH,CAAC;IAED,OAAO,CAAC,SAAkB;QACxB,OAAO,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC7B,CAAC;IAED,IAAI,CAAC,SAAkB;QACrB,OAAO,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC7B,CAAC;IAED,OAAO,CAAC,SAAkB;QACxB,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC9B,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,SAAkB;QACtB,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,IAAI,CAAC,IAAI,GAAG,EAAE,CAAC;QACjB,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAEO,GAAG,CAAC,SAAkB;QAC5B,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,OAAO,IAAI,CAAC,IAAI,CAAC;QACnB,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAChC,CAAC;QAED,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;IACzC,CAAC;CACF"}
|
|
@@ -1,28 +1,28 @@
|
|
|
1
1
|
/// <reference types="node" />
|
|
2
2
|
/// <reference types="node" />
|
|
3
3
|
import type { ReadableStream } from 'stream/web';
|
|
4
|
-
import type {
|
|
4
|
+
import type { ChatResponse, ChatResponseResult, EmbedResponse, ModelConfig, ModelInfo, RateLimiterFunction } from '../ai/types.js';
|
|
5
5
|
import type { Tracer } from '../trace/index.js';
|
|
6
|
-
import type {
|
|
6
|
+
import type { ChatRequest, EmbedRequest } from '../types/index.js';
|
|
7
7
|
export type FunctionExec = {
|
|
8
8
|
id?: string;
|
|
9
9
|
name: string;
|
|
10
10
|
args?: any;
|
|
11
11
|
result?: string;
|
|
12
12
|
};
|
|
13
|
-
export type
|
|
13
|
+
export type AIChatResponse<T> = {
|
|
14
14
|
prompt: string;
|
|
15
15
|
sessionId?: string;
|
|
16
16
|
value(): T;
|
|
17
17
|
};
|
|
18
18
|
export interface AIMemory {
|
|
19
|
-
add(result: Readonly<
|
|
20
|
-
addResult(result: Readonly<
|
|
21
|
-
updateResult(result: Readonly<
|
|
22
|
-
history(sessionId?: string):
|
|
23
|
-
peek(sessionId?: string):
|
|
19
|
+
add(result: Readonly<ChatRequest['chatPrompt'] | ChatRequest['chatPrompt'][0]>, sessionId?: string): void;
|
|
20
|
+
addResult(result: Readonly<ChatResponseResult>, sessionId?: string): void;
|
|
21
|
+
updateResult(result: Readonly<ChatResponseResult>, sessionId?: string): void;
|
|
22
|
+
history(sessionId?: string): ChatRequest['chatPrompt'];
|
|
23
|
+
peek(sessionId?: string): ChatRequest['chatPrompt'];
|
|
24
24
|
reset(sessionId?: string): void;
|
|
25
|
-
getLast(sessionId?: string):
|
|
25
|
+
getLast(sessionId?: string): ChatRequest['chatPrompt'][0] | undefined;
|
|
26
26
|
}
|
|
27
27
|
export type AIPromptConfig = {
|
|
28
28
|
stream?: boolean;
|
|
@@ -42,16 +42,16 @@ export type AIServiceActionOptions = {
|
|
|
42
42
|
};
|
|
43
43
|
export interface AIService {
|
|
44
44
|
getName(): string;
|
|
45
|
-
getModelInfo(): Readonly<
|
|
45
|
+
getModelInfo(): Readonly<ModelInfo & {
|
|
46
46
|
provider: string;
|
|
47
47
|
}>;
|
|
48
|
-
getEmbedModelInfo(): Readonly<
|
|
49
|
-
getModelConfig(): Readonly<
|
|
48
|
+
getEmbedModelInfo(): Readonly<ModelInfo> | undefined;
|
|
49
|
+
getModelConfig(): Readonly<ModelConfig>;
|
|
50
50
|
getFeatures(): {
|
|
51
51
|
functions: boolean;
|
|
52
52
|
streaming: boolean;
|
|
53
53
|
};
|
|
54
|
-
chat(req: Readonly<
|
|
55
|
-
embed(req: Readonly<
|
|
54
|
+
chat(req: Readonly<ChatRequest>, options?: Readonly<AIPromptConfig & AIServiceActionOptions>): Promise<ChatResponse | ReadableStream<ChatResponse>>;
|
|
55
|
+
embed(req: Readonly<EmbedRequest>, options?: Readonly<AIServiceActionOptions & AIServiceActionOptions>): Promise<EmbedResponse>;
|
|
56
56
|
setOptions(options: Readonly<AIServiceOptions>): void;
|
|
57
57
|
}
|