@fllf/agent-sdk 0.1.0
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 +240 -0
- package/dist/agent/Agent.d.ts +34 -0
- package/dist/agent/Agent.d.ts.map +1 -0
- package/dist/agent/Agent.js +168 -0
- package/dist/agent/Agent.js.map +1 -0
- package/dist/agent/createAgent.d.ts +4 -0
- package/dist/agent/createAgent.d.ts.map +1 -0
- package/dist/agent/createAgent.js +8 -0
- package/dist/agent/createAgent.js.map +1 -0
- package/dist/agent/index.d.ts +4 -0
- package/dist/agent/index.d.ts.map +1 -0
- package/dist/agent/index.js +8 -0
- package/dist/agent/index.js.map +1 -0
- package/dist/agent/types.d.ts +23 -0
- package/dist/agent/types.d.ts.map +1 -0
- package/dist/agent/types.js +3 -0
- package/dist/agent/types.js.map +1 -0
- package/dist/config/config.d.ts +35 -0
- package/dist/config/config.d.ts.map +1 -0
- package/dist/config/config.js +123 -0
- package/dist/config/config.js.map +1 -0
- package/dist/config/index.d.ts +3 -0
- package/dist/config/index.d.ts.map +1 -0
- package/dist/config/index.js +6 -0
- package/dist/config/index.js.map +1 -0
- package/dist/executors/base.d.ts +44 -0
- package/dist/executors/base.d.ts.map +1 -0
- package/dist/executors/base.js +3 -0
- package/dist/executors/base.js.map +1 -0
- package/dist/executors/index.d.ts +7 -0
- package/dist/executors/index.d.ts.map +1 -0
- package/dist/executors/index.js +10 -0
- package/dist/executors/index.js.map +1 -0
- package/dist/executors/rag-executor.d.ts +16 -0
- package/dist/executors/rag-executor.d.ts.map +1 -0
- package/dist/executors/rag-executor.js +120 -0
- package/dist/executors/rag-executor.js.map +1 -0
- package/dist/executors/simple-chat-executor.d.ts +5 -0
- package/dist/executors/simple-chat-executor.d.ts.map +1 -0
- package/dist/executors/simple-chat-executor.js +77 -0
- package/dist/executors/simple-chat-executor.js.map +1 -0
- package/dist/executors/tool-calling-executor.d.ts +10 -0
- package/dist/executors/tool-calling-executor.d.ts.map +1 -0
- package/dist/executors/tool-calling-executor.js +151 -0
- package/dist/executors/tool-calling-executor.js.map +1 -0
- package/dist/history/base.d.ts +16 -0
- package/dist/history/base.d.ts.map +1 -0
- package/dist/history/base.js +10 -0
- package/dist/history/base.js.map +1 -0
- package/dist/history/in-memory.d.ts +26 -0
- package/dist/history/in-memory.d.ts.map +1 -0
- package/dist/history/in-memory.js +88 -0
- package/dist/history/in-memory.js.map +1 -0
- package/dist/history/index.d.ts +5 -0
- package/dist/history/index.d.ts.map +1 -0
- package/dist/history/index.js +10 -0
- package/dist/history/index.js.map +1 -0
- package/dist/index.d.ts +10 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +26 -0
- package/dist/index.js.map +1 -0
- package/dist/llm/LLM.d.ts +23 -0
- package/dist/llm/LLM.d.ts.map +1 -0
- package/dist/llm/LLM.js +404 -0
- package/dist/llm/LLM.js.map +1 -0
- package/dist/llm/errors.d.ts +43 -0
- package/dist/llm/errors.d.ts.map +1 -0
- package/dist/llm/errors.js +128 -0
- package/dist/llm/errors.js.map +1 -0
- package/dist/llm/factory.d.ts +9 -0
- package/dist/llm/factory.d.ts.map +1 -0
- package/dist/llm/factory.js +28 -0
- package/dist/llm/factory.js.map +1 -0
- package/dist/llm/index.d.ts +6 -0
- package/dist/llm/index.d.ts.map +1 -0
- package/dist/llm/index.js +18 -0
- package/dist/llm/index.js.map +1 -0
- package/dist/llm/providers/base.d.ts +3 -0
- package/dist/llm/providers/base.d.ts.map +1 -0
- package/dist/llm/providers/base.js +3 -0
- package/dist/llm/providers/base.js.map +1 -0
- package/dist/llm/providers/local.d.ts +6 -0
- package/dist/llm/providers/local.d.ts.map +1 -0
- package/dist/llm/providers/local.js +19 -0
- package/dist/llm/providers/local.js.map +1 -0
- package/dist/llm/providers/openai-compatible.d.ts +13 -0
- package/dist/llm/providers/openai-compatible.d.ts.map +1 -0
- package/dist/llm/providers/openai-compatible.js +176 -0
- package/dist/llm/providers/openai-compatible.js.map +1 -0
- package/dist/llm/providers/openai.d.ts +6 -0
- package/dist/llm/providers/openai.d.ts.map +1 -0
- package/dist/llm/providers/openai.js +23 -0
- package/dist/llm/providers/openai.js.map +1 -0
- package/dist/llm/types.d.ts +116 -0
- package/dist/llm/types.d.ts.map +1 -0
- package/dist/llm/types.js +3 -0
- package/dist/llm/types.js.map +1 -0
- package/dist/messages/index.d.ts +3 -0
- package/dist/messages/index.d.ts.map +1 -0
- package/dist/messages/index.js +6 -0
- package/dist/messages/index.js.map +1 -0
- package/dist/messages/message.d.ts +25 -0
- package/dist/messages/message.d.ts.map +1 -0
- package/dist/messages/message.js +110 -0
- package/dist/messages/message.js.map +1 -0
- package/dist/messages/types.d.ts +20 -0
- package/dist/messages/types.d.ts.map +1 -0
- package/dist/messages/types.js +3 -0
- package/dist/messages/types.js.map +1 -0
- package/dist/observability/console-observer.d.ts +10 -0
- package/dist/observability/console-observer.d.ts.map +1 -0
- package/dist/observability/console-observer.js +28 -0
- package/dist/observability/console-observer.js.map +1 -0
- package/dist/observability/index.d.ts +5 -0
- package/dist/observability/index.d.ts.map +1 -0
- package/dist/observability/index.js +11 -0
- package/dist/observability/index.js.map +1 -0
- package/dist/observability/observer.d.ts +8 -0
- package/dist/observability/observer.d.ts.map +1 -0
- package/dist/observability/observer.js +46 -0
- package/dist/observability/observer.js.map +1 -0
- package/dist/observability/types.d.ts +34 -0
- package/dist/observability/types.d.ts.map +1 -0
- package/dist/observability/types.js +3 -0
- package/dist/observability/types.js.map +1 -0
- package/dist/rag/chunking/auto-chunker.d.ts +25 -0
- package/dist/rag/chunking/auto-chunker.d.ts.map +1 -0
- package/dist/rag/chunking/auto-chunker.js +101 -0
- package/dist/rag/chunking/auto-chunker.js.map +1 -0
- package/dist/rag/chunking/chunker.d.ts +44 -0
- package/dist/rag/chunking/chunker.d.ts.map +1 -0
- package/dist/rag/chunking/chunker.js +162 -0
- package/dist/rag/chunking/chunker.js.map +1 -0
- package/dist/rag/chunking/index.d.ts +6 -0
- package/dist/rag/chunking/index.d.ts.map +1 -0
- package/dist/rag/chunking/index.js +22 -0
- package/dist/rag/chunking/index.js.map +1 -0
- package/dist/rag/chunking/markdown-chunker.d.ts +13 -0
- package/dist/rag/chunking/markdown-chunker.d.ts.map +1 -0
- package/dist/rag/chunking/markdown-chunker.js +144 -0
- package/dist/rag/chunking/markdown-chunker.js.map +1 -0
- package/dist/rag/chunking/parent-child.d.ts +16 -0
- package/dist/rag/chunking/parent-child.d.ts.map +1 -0
- package/dist/rag/chunking/parent-child.js +76 -0
- package/dist/rag/chunking/parent-child.js.map +1 -0
- package/dist/rag/chunking/recursive-chunker.d.ts +11 -0
- package/dist/rag/chunking/recursive-chunker.d.ts.map +1 -0
- package/dist/rag/chunking/recursive-chunker.js +27 -0
- package/dist/rag/chunking/recursive-chunker.js.map +1 -0
- package/dist/rag/embeddings/embedder.d.ts +24 -0
- package/dist/rag/embeddings/embedder.d.ts.map +1 -0
- package/dist/rag/embeddings/embedder.js +73 -0
- package/dist/rag/embeddings/embedder.js.map +1 -0
- package/dist/rag/embeddings/embedding-cache.d.ts +40 -0
- package/dist/rag/embeddings/embedding-cache.d.ts.map +1 -0
- package/dist/rag/embeddings/embedding-cache.js +118 -0
- package/dist/rag/embeddings/embedding-cache.js.map +1 -0
- package/dist/rag/embeddings/fake-embedder.d.ts +20 -0
- package/dist/rag/embeddings/fake-embedder.d.ts.map +1 -0
- package/dist/rag/embeddings/fake-embedder.js +55 -0
- package/dist/rag/embeddings/fake-embedder.js.map +1 -0
- package/dist/rag/embeddings/index.d.ts +5 -0
- package/dist/rag/embeddings/index.d.ts.map +1 -0
- package/dist/rag/embeddings/index.js +21 -0
- package/dist/rag/embeddings/index.js.map +1 -0
- package/dist/rag/embeddings/openai-compatible-embedder.d.ts +46 -0
- package/dist/rag/embeddings/openai-compatible-embedder.d.ts.map +1 -0
- package/dist/rag/embeddings/openai-compatible-embedder.js +145 -0
- package/dist/rag/embeddings/openai-compatible-embedder.js.map +1 -0
- package/dist/rag/generation/context-builder.d.ts +22 -0
- package/dist/rag/generation/context-builder.d.ts.map +1 -0
- package/dist/rag/generation/context-builder.js +166 -0
- package/dist/rag/generation/context-builder.js.map +1 -0
- package/dist/rag/generation/generator.d.ts +25 -0
- package/dist/rag/generation/generator.d.ts.map +1 -0
- package/dist/rag/generation/generator.js +185 -0
- package/dist/rag/generation/generator.js.map +1 -0
- package/dist/rag/generation/index.d.ts +4 -0
- package/dist/rag/generation/index.d.ts.map +1 -0
- package/dist/rag/generation/index.js +20 -0
- package/dist/rag/generation/index.js.map +1 -0
- package/dist/rag/generation/verifier.d.ts +13 -0
- package/dist/rag/generation/verifier.d.ts.map +1 -0
- package/dist/rag/generation/verifier.js +43 -0
- package/dist/rag/generation/verifier.js.map +1 -0
- package/dist/rag/index.d.ts +9 -0
- package/dist/rag/index.d.ts.map +1 -0
- package/dist/rag/index.js +25 -0
- package/dist/rag/index.js.map +1 -0
- package/dist/rag/ingestion/index.d.ts +4 -0
- package/dist/rag/ingestion/index.d.ts.map +1 -0
- package/dist/rag/ingestion/index.js +20 -0
- package/dist/rag/ingestion/index.js.map +1 -0
- package/dist/rag/ingestion/loaders.d.ts +69 -0
- package/dist/rag/ingestion/loaders.d.ts.map +1 -0
- package/dist/rag/ingestion/loaders.js +653 -0
- package/dist/rag/ingestion/loaders.js.map +1 -0
- package/dist/rag/ingestion/metadata.d.ts +31 -0
- package/dist/rag/ingestion/metadata.d.ts.map +1 -0
- package/dist/rag/ingestion/metadata.js +81 -0
- package/dist/rag/ingestion/metadata.js.map +1 -0
- package/dist/rag/ingestion/normalizer.d.ts +20 -0
- package/dist/rag/ingestion/normalizer.d.ts.map +1 -0
- package/dist/rag/ingestion/normalizer.js +161 -0
- package/dist/rag/ingestion/normalizer.js.map +1 -0
- package/dist/rag/pipeline.d.ts +62 -0
- package/dist/rag/pipeline.d.ts.map +1 -0
- package/dist/rag/pipeline.js +75 -0
- package/dist/rag/pipeline.js.map +1 -0
- package/dist/rag/retrieval/dense-retriever.d.ts +17 -0
- package/dist/rag/retrieval/dense-retriever.d.ts.map +1 -0
- package/dist/rag/retrieval/dense-retriever.js +21 -0
- package/dist/rag/retrieval/dense-retriever.js.map +1 -0
- package/dist/rag/retrieval/fusion.d.ts +12 -0
- package/dist/rag/retrieval/fusion.d.ts.map +1 -0
- package/dist/rag/retrieval/fusion.js +54 -0
- package/dist/rag/retrieval/fusion.js.map +1 -0
- package/dist/rag/retrieval/http-rerank-model-client.d.ts +41 -0
- package/dist/rag/retrieval/http-rerank-model-client.d.ts.map +1 -0
- package/dist/rag/retrieval/http-rerank-model-client.js +130 -0
- package/dist/rag/retrieval/http-rerank-model-client.js.map +1 -0
- package/dist/rag/retrieval/hybrid-retriever.d.ts +22 -0
- package/dist/rag/retrieval/hybrid-retriever.d.ts.map +1 -0
- package/dist/rag/retrieval/hybrid-retriever.js +49 -0
- package/dist/rag/retrieval/hybrid-retriever.js.map +1 -0
- package/dist/rag/retrieval/index.d.ts +8 -0
- package/dist/rag/retrieval/index.d.ts.map +1 -0
- package/dist/rag/retrieval/index.js +24 -0
- package/dist/rag/retrieval/index.js.map +1 -0
- package/dist/rag/retrieval/reranker.d.ts +30 -0
- package/dist/rag/retrieval/reranker.d.ts.map +1 -0
- package/dist/rag/retrieval/reranker.js +85 -0
- package/dist/rag/retrieval/reranker.js.map +1 -0
- package/dist/rag/retrieval/sparse-retriever.d.ts +14 -0
- package/dist/rag/retrieval/sparse-retriever.d.ts.map +1 -0
- package/dist/rag/retrieval/sparse-retriever.js +18 -0
- package/dist/rag/retrieval/sparse-retriever.js.map +1 -0
- package/dist/rag/retrieval/types.d.ts +11 -0
- package/dist/rag/retrieval/types.d.ts.map +1 -0
- package/dist/rag/retrieval/types.js +10 -0
- package/dist/rag/retrieval/types.js.map +1 -0
- package/dist/rag/retrieval/utils.d.ts +8 -0
- package/dist/rag/retrieval/utils.d.ts.map +1 -0
- package/dist/rag/retrieval/utils.js +114 -0
- package/dist/rag/retrieval/utils.js.map +1 -0
- package/dist/rag/stores/in-memory-document-store.d.ts +14 -0
- package/dist/rag/stores/in-memory-document-store.d.ts.map +1 -0
- package/dist/rag/stores/in-memory-document-store.js +64 -0
- package/dist/rag/stores/in-memory-document-store.js.map +1 -0
- package/dist/rag/stores/in-memory-keyword-store.d.ts +10 -0
- package/dist/rag/stores/in-memory-keyword-store.d.ts.map +1 -0
- package/dist/rag/stores/in-memory-keyword-store.js +99 -0
- package/dist/rag/stores/in-memory-keyword-store.js.map +1 -0
- package/dist/rag/stores/in-memory-vector-store.d.ts +14 -0
- package/dist/rag/stores/in-memory-vector-store.d.ts.map +1 -0
- package/dist/rag/stores/in-memory-vector-store.js +61 -0
- package/dist/rag/stores/in-memory-vector-store.js.map +1 -0
- package/dist/rag/stores/index.d.ts +6 -0
- package/dist/rag/stores/index.d.ts.map +1 -0
- package/dist/rag/stores/index.js +22 -0
- package/dist/rag/stores/index.js.map +1 -0
- package/dist/rag/stores/postgres/index.d.ts +6 -0
- package/dist/rag/stores/postgres/index.d.ts.map +1 -0
- package/dist/rag/stores/postgres/index.js +22 -0
- package/dist/rag/stores/postgres/index.js.map +1 -0
- package/dist/rag/stores/postgres/pg-vector-store.d.ts +16 -0
- package/dist/rag/stores/postgres/pg-vector-store.d.ts.map +1 -0
- package/dist/rag/stores/postgres/pg-vector-store.js +77 -0
- package/dist/rag/stores/postgres/pg-vector-store.js.map +1 -0
- package/dist/rag/stores/postgres/postgres-document-store.d.ts +15 -0
- package/dist/rag/stores/postgres/postgres-document-store.d.ts.map +1 -0
- package/dist/rag/stores/postgres/postgres-document-store.js +174 -0
- package/dist/rag/stores/postgres/postgres-document-store.js.map +1 -0
- package/dist/rag/stores/postgres/postgres-keyword-store.d.ts +15 -0
- package/dist/rag/stores/postgres/postgres-keyword-store.d.ts.map +1 -0
- package/dist/rag/stores/postgres/postgres-keyword-store.js +163 -0
- package/dist/rag/stores/postgres/postgres-keyword-store.js.map +1 -0
- package/dist/rag/stores/postgres/schema.d.ts +8 -0
- package/dist/rag/stores/postgres/schema.d.ts.map +1 -0
- package/dist/rag/stores/postgres/schema.js +91 -0
- package/dist/rag/stores/postgres/schema.js.map +1 -0
- package/dist/rag/stores/postgres/sql.d.ts +23 -0
- package/dist/rag/stores/postgres/sql.d.ts.map +1 -0
- package/dist/rag/stores/postgres/sql.js +177 -0
- package/dist/rag/stores/postgres/sql.js.map +1 -0
- package/dist/rag/stores/postgres/types.d.ts +17 -0
- package/dist/rag/stores/postgres/types.d.ts.map +1 -0
- package/dist/rag/stores/postgres/types.js +3 -0
- package/dist/rag/stores/postgres/types.js.map +1 -0
- package/dist/rag/stores/types.d.ts +36 -0
- package/dist/rag/stores/types.d.ts.map +1 -0
- package/dist/rag/stores/types.js +3 -0
- package/dist/rag/stores/types.js.map +1 -0
- package/dist/rag/stores/utils.d.ts +11 -0
- package/dist/rag/stores/utils.d.ts.map +1 -0
- package/dist/rag/stores/utils.js +200 -0
- package/dist/rag/stores/utils.js.map +1 -0
- package/dist/rag/types.d.ts +174 -0
- package/dist/rag/types.d.ts.map +1 -0
- package/dist/rag/types.js +3 -0
- package/dist/rag/types.js.map +1 -0
- package/dist/tools/base.d.ts +28 -0
- package/dist/tools/base.d.ts.map +1 -0
- package/dist/tools/base.js +68 -0
- package/dist/tools/base.js.map +1 -0
- package/dist/tools/builtin/advancedSearchTool.d.ts +24 -0
- package/dist/tools/builtin/advancedSearchTool.d.ts.map +1 -0
- package/dist/tools/builtin/advancedSearchTool.js +134 -0
- package/dist/tools/builtin/advancedSearchTool.js.map +1 -0
- package/dist/tools/builtin/ragSearchTool.d.ts +29 -0
- package/dist/tools/builtin/ragSearchTool.d.ts.map +1 -0
- package/dist/tools/builtin/ragSearchTool.js +91 -0
- package/dist/tools/builtin/ragSearchTool.js.map +1 -0
- package/dist/tools/executor.d.ts +10 -0
- package/dist/tools/executor.d.ts.map +1 -0
- package/dist/tools/executor.js +86 -0
- package/dist/tools/executor.js.map +1 -0
- package/dist/tools/index.d.ts +9 -0
- package/dist/tools/index.d.ts.map +1 -0
- package/dist/tools/index.js +16 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/registry.d.ts +16 -0
- package/dist/tools/registry.d.ts.map +1 -0
- package/dist/tools/registry.js +35 -0
- package/dist/tools/registry.js.map +1 -0
- package/package.json +43 -0
|
@@ -0,0 +1,123 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.Config = void 0;
|
|
4
|
+
class Config {
|
|
5
|
+
defaultModel;
|
|
6
|
+
defaultProvider;
|
|
7
|
+
temperature;
|
|
8
|
+
maxTokens;
|
|
9
|
+
requestTimeoutMs;
|
|
10
|
+
maxRetries;
|
|
11
|
+
retryDelayMs;
|
|
12
|
+
retryMaxDelayMs;
|
|
13
|
+
debug;
|
|
14
|
+
logLevel;
|
|
15
|
+
maxHistoryLength;
|
|
16
|
+
constructor(options = {}) {
|
|
17
|
+
this.defaultModel = options.defaultModel;
|
|
18
|
+
this.defaultProvider = options.defaultProvider;
|
|
19
|
+
this.temperature = options.temperature ?? 0.7;
|
|
20
|
+
this.maxTokens = options.maxTokens;
|
|
21
|
+
this.requestTimeoutMs = options.requestTimeoutMs;
|
|
22
|
+
this.maxRetries = options.maxRetries;
|
|
23
|
+
this.retryDelayMs = options.retryDelayMs;
|
|
24
|
+
this.retryMaxDelayMs = options.retryMaxDelayMs;
|
|
25
|
+
this.debug = options.debug ?? false;
|
|
26
|
+
this.logLevel = options.logLevel ?? 'INFO';
|
|
27
|
+
this.maxHistoryLength = options.maxHistoryLength ?? 100;
|
|
28
|
+
}
|
|
29
|
+
/**
|
|
30
|
+
* 环境变量只负责覆盖运行参数。
|
|
31
|
+
* 如果没有显式 MODEL,就让 LLM provider 使用自己的默认模型。
|
|
32
|
+
*/
|
|
33
|
+
static fromEnv() {
|
|
34
|
+
const options = {};
|
|
35
|
+
if (process.env.MODEL) {
|
|
36
|
+
options.defaultModel = process.env.MODEL;
|
|
37
|
+
}
|
|
38
|
+
if (process.env.MODEL_PROVIDER) {
|
|
39
|
+
options.defaultProvider = process.env.MODEL_PROVIDER;
|
|
40
|
+
}
|
|
41
|
+
if (process.env.DEBUG !== undefined) {
|
|
42
|
+
options.debug = process.env.DEBUG.toLowerCase() === 'true';
|
|
43
|
+
}
|
|
44
|
+
if (process.env.LOG_LEVEL) {
|
|
45
|
+
options.logLevel = process.env.LOG_LEVEL;
|
|
46
|
+
}
|
|
47
|
+
if (process.env.TEMPERATURE !== undefined) {
|
|
48
|
+
const temperature = Number.parseFloat(process.env.TEMPERATURE);
|
|
49
|
+
if (!Number.isNaN(temperature)) {
|
|
50
|
+
options.temperature = temperature;
|
|
51
|
+
}
|
|
52
|
+
}
|
|
53
|
+
if (process.env.MAX_TOKENS !== undefined) {
|
|
54
|
+
const maxTokens = Number.parseInt(process.env.MAX_TOKENS, 10);
|
|
55
|
+
if (!Number.isNaN(maxTokens)) {
|
|
56
|
+
options.maxTokens = maxTokens;
|
|
57
|
+
}
|
|
58
|
+
}
|
|
59
|
+
if (process.env.REQUEST_TIMEOUT_MS !== undefined) {
|
|
60
|
+
const requestTimeoutMs = Number.parseInt(process.env.REQUEST_TIMEOUT_MS, 10);
|
|
61
|
+
if (!Number.isNaN(requestTimeoutMs)) {
|
|
62
|
+
options.requestTimeoutMs = requestTimeoutMs;
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
if (process.env.MAX_RETRIES !== undefined) {
|
|
66
|
+
const maxRetries = Number.parseInt(process.env.MAX_RETRIES, 10);
|
|
67
|
+
if (!Number.isNaN(maxRetries)) {
|
|
68
|
+
options.maxRetries = maxRetries;
|
|
69
|
+
}
|
|
70
|
+
}
|
|
71
|
+
if (process.env.RETRY_DELAY_MS !== undefined) {
|
|
72
|
+
const retryDelayMs = Number.parseInt(process.env.RETRY_DELAY_MS, 10);
|
|
73
|
+
if (!Number.isNaN(retryDelayMs)) {
|
|
74
|
+
options.retryDelayMs = retryDelayMs;
|
|
75
|
+
}
|
|
76
|
+
}
|
|
77
|
+
if (process.env.RETRY_MAX_DELAY_MS !== undefined) {
|
|
78
|
+
const retryMaxDelayMs = Number.parseInt(process.env.RETRY_MAX_DELAY_MS, 10);
|
|
79
|
+
if (!Number.isNaN(retryMaxDelayMs)) {
|
|
80
|
+
options.retryMaxDelayMs = retryMaxDelayMs;
|
|
81
|
+
}
|
|
82
|
+
}
|
|
83
|
+
if (process.env.MAX_HISTORY_LENGTH !== undefined) {
|
|
84
|
+
const maxHistoryLength = Number.parseInt(process.env.MAX_HISTORY_LENGTH, 10);
|
|
85
|
+
if (!Number.isNaN(maxHistoryLength)) {
|
|
86
|
+
options.maxHistoryLength = maxHistoryLength;
|
|
87
|
+
}
|
|
88
|
+
}
|
|
89
|
+
return new Config(options);
|
|
90
|
+
}
|
|
91
|
+
toJSON() {
|
|
92
|
+
const options = {
|
|
93
|
+
temperature: this.temperature,
|
|
94
|
+
debug: this.debug,
|
|
95
|
+
logLevel: this.logLevel,
|
|
96
|
+
maxHistoryLength: this.maxHistoryLength,
|
|
97
|
+
};
|
|
98
|
+
if (this.defaultModel !== undefined) {
|
|
99
|
+
options.defaultModel = this.defaultModel;
|
|
100
|
+
}
|
|
101
|
+
if (this.defaultProvider !== undefined) {
|
|
102
|
+
options.defaultProvider = this.defaultProvider;
|
|
103
|
+
}
|
|
104
|
+
if (this.maxTokens !== undefined) {
|
|
105
|
+
options.maxTokens = this.maxTokens;
|
|
106
|
+
}
|
|
107
|
+
if (this.requestTimeoutMs !== undefined) {
|
|
108
|
+
options.requestTimeoutMs = this.requestTimeoutMs;
|
|
109
|
+
}
|
|
110
|
+
if (this.maxRetries !== undefined) {
|
|
111
|
+
options.maxRetries = this.maxRetries;
|
|
112
|
+
}
|
|
113
|
+
if (this.retryDelayMs !== undefined) {
|
|
114
|
+
options.retryDelayMs = this.retryDelayMs;
|
|
115
|
+
}
|
|
116
|
+
if (this.retryMaxDelayMs !== undefined) {
|
|
117
|
+
options.retryMaxDelayMs = this.retryMaxDelayMs;
|
|
118
|
+
}
|
|
119
|
+
return options;
|
|
120
|
+
}
|
|
121
|
+
}
|
|
122
|
+
exports.Config = Config;
|
|
123
|
+
//# sourceMappingURL=config.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config.js","sourceRoot":"","sources":["../../src/config/config.ts"],"names":[],"mappings":";;;AAgBA,MAAa,MAAM;IACN,YAAY,CAAqB;IACjC,eAAe,CAAgC;IAC/C,WAAW,CAAS;IACpB,SAAS,CAAqB;IAC9B,gBAAgB,CAAqB;IACrC,UAAU,CAAqB;IAC/B,YAAY,CAAqB;IACjC,eAAe,CAAqB;IACpC,KAAK,CAAU;IACf,QAAQ,CAAS;IACjB,gBAAgB,CAAS;IAElC,YAAY,UAAyB,EAAE;QACnC,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;QACzC,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,eAAe,CAAC;QAC/C,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,GAAG,CAAC;QAC9C,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAC;QACjD,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;QACrC,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;QACzC,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,eAAe,CAAC;QAC/C,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,KAAK,CAAC;QACpC,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,MAAM,CAAC;QAC3C,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,IAAI,GAAG,CAAC;IAC5D,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,OAAO;QACV,MAAM,OAAO,GAAkB,EAAE,CAAC;QAElC,IAAI,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC;YACpB,OAAO,CAAC,YAAY,GAAG,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC;QAC7C,CAAC;QAED,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC;YAC7B,OAAO,CAAC,eAAe,GAAG,OAAO,CAAC,GAAG,CAAC,cAAmC,CAAC;QAC9E,CAAC;QAED,IAAI,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAClC,OAAO,CAAC,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,MAAM,CAAC;QAC/D,CAAC;QAED,IAAI,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC;YACxB,OAAO,CAAC,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC;QAC7C,CAAC;QAED,IAAI,OAAO,CAAC,GAAG,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;YACxC,MAAM,WAAW,GAAG,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;YAC/D,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,EAAE,CAAC;gBAC7B,OAAO,CAAC,WAAW,GAAG,WAAW,CAAC;YACtC,CAAC;QACL,CAAC;QAED,IAAI,OAAO,CAAC,GAAG,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;YACvC,MAAM,SAAS,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YAC9D,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC3B,OAAO,CAAC,SAAS,GAAG,SAAS,CAAC;YAClC,CAAC;QACL,CAAC;QAED,IAAI,OAAO,CAAC,GAAG,CAAC,kBAAkB,KAAK,SAAS,EAAE,CAAC;YAC/C,MAAM,gBAAgB,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC;YAC7E,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,EAAE,CAAC;gBAClC,OAAO,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;YAChD,CAAC;QACL,CAAC;QAED,IAAI,OAAO,CAAC,GAAG,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;YACxC,MAAM,UAAU,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;YAChE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC5B,OAAO,CAAC,UAAU,GAAG,UAAU,CAAC;YACpC,CAAC;QACL,CAAC;QAED,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,KAAK,SAAS,EAAE,CAAC;YAC3C,MAAM,YAAY,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;YACrE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,CAAC;gBAC9B,OAAO,CAAC,YAAY,GAAG,YAAY,CAAC;YACxC,CAAC;QACL,CAAC;QAED,IAAI,OAAO,CAAC,GAAG,CAAC,kBAAkB,KAAK,SAAS,EAAE,CAAC;YAC/C,MAAM,eAAe,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC;YAC5E,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,EAAE,CAAC;gBACjC,OAAO,CAAC,eAAe,GAAG,eAAe,CAAC;YAC9C,CAAC;QACL,CAAC;QAED,IAAI,OAAO,CAAC,GAAG,CAAC,kBAAkB,KAAK,SAAS,EAAE,CAAC;YAC/C,MAAM,gBAAgB,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC;YAC7E,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,EAAE,CAAC;gBAClC,OAAO,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;YAChD,CAAC;QACL,CAAC;QAED,OAAO,IAAI,MAAM,CAAC,OAAO,CAAC,CAAC;IAC/B,CAAC;IAED,MAAM;QACF,MAAM,OAAO,GAAkB;YAC3B,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;SAC1C,CAAC;QAEF,IAAI,IAAI,CAAC,YAAY,KAAK,SAAS,EAAE,CAAC;YAClC,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC;QAC7C,CAAC;QACD,IAAI,IAAI,CAAC,eAAe,KAAK,SAAS,EAAE,CAAC;YACrC,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC;QACnD,CAAC;QACD,IAAI,IAAI,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YAC/B,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;QACvC,CAAC;QACD,IAAI,IAAI,CAAC,gBAAgB,KAAK,SAAS,EAAE,CAAC;YACtC,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC;QACrD,CAAC;QACD,IAAI,IAAI,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;YAChC,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC;QACzC,CAAC;QACD,IAAI,IAAI,CAAC,YAAY,KAAK,SAAS,EAAE,CAAC;YAClC,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC;QAC7C,CAAC;QACD,IAAI,IAAI,CAAC,eAAe,KAAK,SAAS,EAAE,CAAC;YACrC,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC;QACnD,CAAC;QAED,OAAO,OAAO,CAAC;IACnB,CAAC;CACJ;AAtID,wBAsIC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/config/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAClC,YAAY,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC"}
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.Config = void 0;
|
|
4
|
+
var config_1 = require("./config");
|
|
5
|
+
Object.defineProperty(exports, "Config", { enumerable: true, get: function () { return config_1.Config; } });
|
|
6
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/config/index.ts"],"names":[],"mappings":";;;AAAA,mCAAkC;AAAzB,gGAAA,MAAM,OAAA"}
|
|
@@ -0,0 +1,44 @@
|
|
|
1
|
+
import type { Config } from '../config';
|
|
2
|
+
import type { MessageHistory } from '../history';
|
|
3
|
+
import type { ChatModel, TokenUsage } from '../llm';
|
|
4
|
+
import type { Message } from '../messages';
|
|
5
|
+
import type { Observer } from '../observability';
|
|
6
|
+
import type { ToolRegistry } from '../tools';
|
|
7
|
+
export interface AgentRunOptions {
|
|
8
|
+
sessionId?: string;
|
|
9
|
+
model?: string;
|
|
10
|
+
temperature?: number;
|
|
11
|
+
maxTokens?: number;
|
|
12
|
+
timeoutMs?: number;
|
|
13
|
+
maxRetries?: number;
|
|
14
|
+
retryDelayMs?: number;
|
|
15
|
+
retryMaxDelayMs?: number;
|
|
16
|
+
systemPrompt?: string;
|
|
17
|
+
runId?: string;
|
|
18
|
+
observer?: Observer;
|
|
19
|
+
signal?: AbortSignal;
|
|
20
|
+
metadata?: Record<string, unknown>;
|
|
21
|
+
}
|
|
22
|
+
export interface AgentResult {
|
|
23
|
+
output: string;
|
|
24
|
+
messages: Message[];
|
|
25
|
+
usage?: TokenUsage | undefined;
|
|
26
|
+
raw?: unknown;
|
|
27
|
+
}
|
|
28
|
+
export interface AgentContext {
|
|
29
|
+
runId: string;
|
|
30
|
+
agentName: string;
|
|
31
|
+
sessionId: string;
|
|
32
|
+
input: string;
|
|
33
|
+
options: AgentRunOptions;
|
|
34
|
+
llm: ChatModel;
|
|
35
|
+
history: MessageHistory;
|
|
36
|
+
tools: ToolRegistry;
|
|
37
|
+
observer: Observer | undefined;
|
|
38
|
+
systemPrompt: string | undefined;
|
|
39
|
+
config: Config;
|
|
40
|
+
}
|
|
41
|
+
export interface AgentExecutor {
|
|
42
|
+
run(context: AgentContext): Promise<AgentResult>;
|
|
43
|
+
}
|
|
44
|
+
//# sourceMappingURL=base.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../src/executors/base.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AACxC,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AACjD,OAAO,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpD,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AACjD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AAE7C,MAAM,WAAW,eAAe;IAC5B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,QAAQ,CAAC;IACpB,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED,MAAM,WAAW,WAAW;IACxB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,KAAK,CAAC,EAAE,UAAU,GAAG,SAAS,CAAC;IAC/B,GAAG,CAAC,EAAE,OAAO,CAAC;CACjB;AAED,MAAM,WAAW,YAAY;IACzB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,eAAe,CAAC;IACzB,GAAG,EAAE,SAAS,CAAC;IACf,OAAO,EAAE,cAAc,CAAC;IACxB,KAAK,EAAE,YAAY,CAAC;IACpB,QAAQ,EAAE,QAAQ,GAAG,SAAS,CAAC;IAC/B,YAAY,EAAE,MAAM,GAAG,SAAS,CAAC;IACjC,MAAM,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,aAAa;IAC1B,GAAG,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;CACpD"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/executors/base.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
export type { AgentContext, AgentExecutor, AgentResult, AgentRunOptions, } from './base';
|
|
2
|
+
export { RagExecutor } from './rag-executor';
|
|
3
|
+
export type { RagExecutorOptions } from './rag-executor';
|
|
4
|
+
export { SimpleChatExecutor } from './simple-chat-executor';
|
|
5
|
+
export { ToolCallingExecutor } from './tool-calling-executor';
|
|
6
|
+
export type { ToolCallingExecutorOptions } from './tool-calling-executor';
|
|
7
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/executors/index.ts"],"names":[],"mappings":"AAAA,YAAY,EACR,YAAY,EACZ,aAAa,EACb,WAAW,EACX,eAAe,GAClB,MAAM,QAAQ,CAAC;AAChB,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC7C,YAAY,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACzD,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAC5D,OAAO,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AAC9D,YAAY,EAAE,0BAA0B,EAAE,MAAM,yBAAyB,CAAC"}
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.ToolCallingExecutor = exports.SimpleChatExecutor = exports.RagExecutor = void 0;
|
|
4
|
+
var rag_executor_1 = require("./rag-executor");
|
|
5
|
+
Object.defineProperty(exports, "RagExecutor", { enumerable: true, get: function () { return rag_executor_1.RagExecutor; } });
|
|
6
|
+
var simple_chat_executor_1 = require("./simple-chat-executor");
|
|
7
|
+
Object.defineProperty(exports, "SimpleChatExecutor", { enumerable: true, get: function () { return simple_chat_executor_1.SimpleChatExecutor; } });
|
|
8
|
+
var tool_calling_executor_1 = require("./tool-calling-executor");
|
|
9
|
+
Object.defineProperty(exports, "ToolCallingExecutor", { enumerable: true, get: function () { return tool_calling_executor_1.ToolCallingExecutor; } });
|
|
10
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/executors/index.ts"],"names":[],"mappings":";;;AAMA,+CAA6C;AAApC,2GAAA,WAAW,OAAA;AAEpB,+DAA4D;AAAnD,0HAAA,kBAAkB,OAAA;AAC3B,iEAA8D;AAArD,4HAAA,mBAAmB,OAAA"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
import type { AnswerOptions, RagPipeline } from '../rag';
|
|
2
|
+
import type { AgentContext, AgentExecutor, AgentResult } from './base';
|
|
3
|
+
export interface RagExecutorOptions {
|
|
4
|
+
pipeline: RagPipeline;
|
|
5
|
+
answerOptions?: Omit<AnswerOptions, 'llm' | 'trace'>;
|
|
6
|
+
requireCitations?: boolean;
|
|
7
|
+
}
|
|
8
|
+
export declare class RagExecutor implements AgentExecutor {
|
|
9
|
+
private readonly pipeline;
|
|
10
|
+
private readonly answerOptions;
|
|
11
|
+
private readonly requireCitations;
|
|
12
|
+
constructor(options: RagExecutorOptions);
|
|
13
|
+
run(context: AgentContext): Promise<AgentResult>;
|
|
14
|
+
private buildAnswerOptions;
|
|
15
|
+
}
|
|
16
|
+
//# sourceMappingURL=rag-executor.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"rag-executor.d.ts","sourceRoot":"","sources":["../../src/executors/rag-executor.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACR,aAAa,EAGb,WAAW,EAEd,MAAM,QAAQ,CAAC;AAEhB,OAAO,KAAK,EACR,YAAY,EACZ,aAAa,EACb,WAAW,EACd,MAAM,QAAQ,CAAC;AAEhB,MAAM,WAAW,kBAAkB;IAC/B,QAAQ,EAAE,WAAW,CAAC;IACtB,aAAa,CAAC,EAAE,IAAI,CAAC,aAAa,EAAE,KAAK,GAAG,OAAO,CAAC,CAAC;IACrD,gBAAgB,CAAC,EAAE,OAAO,CAAC;CAC9B;AAED,qBAAa,WAAY,YAAW,aAAa;IAC7C,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAc;IACvC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAuC;IACrE,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAsB;gBAE3C,OAAO,EAAE,kBAAkB;IAMjC,GAAG,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC;IAmCtD,OAAO,CAAC,kBAAkB;CA8B7B"}
|
|
@@ -0,0 +1,120 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.RagExecutor = void 0;
|
|
4
|
+
const messages_1 = require("../messages");
|
|
5
|
+
class RagExecutor {
|
|
6
|
+
pipeline;
|
|
7
|
+
answerOptions;
|
|
8
|
+
requireCitations;
|
|
9
|
+
constructor(options) {
|
|
10
|
+
this.pipeline = options.pipeline;
|
|
11
|
+
this.answerOptions = options.answerOptions ?? {};
|
|
12
|
+
this.requireCitations = options.requireCitations;
|
|
13
|
+
}
|
|
14
|
+
async run(context) {
|
|
15
|
+
const answer = await this.pipeline.answer(context.input, this.buildAnswerOptions(context));
|
|
16
|
+
const userMessage = new messages_1.Message(context.input, 'user', context.options.metadata ? { metadata: context.options.metadata } : {});
|
|
17
|
+
const assistantMessage = new messages_1.Message(answer.answer, 'assistant', {
|
|
18
|
+
metadata: {
|
|
19
|
+
rag: {
|
|
20
|
+
confidence: answer.confidence,
|
|
21
|
+
citations: answer.citations.map(toCitationMetadata),
|
|
22
|
+
retrieved: answer.retrieved.map(toRetrievedMetadata),
|
|
23
|
+
},
|
|
24
|
+
},
|
|
25
|
+
});
|
|
26
|
+
await context.history.addMany([userMessage, assistantMessage]);
|
|
27
|
+
const result = {
|
|
28
|
+
output: answer.answer,
|
|
29
|
+
messages: await context.history.get(),
|
|
30
|
+
raw: answer,
|
|
31
|
+
};
|
|
32
|
+
if (answer.usage !== undefined) {
|
|
33
|
+
result.usage = answer.usage;
|
|
34
|
+
}
|
|
35
|
+
return result;
|
|
36
|
+
}
|
|
37
|
+
buildAnswerOptions(context) {
|
|
38
|
+
const metadata = context.options.metadata;
|
|
39
|
+
const options = {
|
|
40
|
+
...this.answerOptions,
|
|
41
|
+
llm: context.llm,
|
|
42
|
+
trace: {
|
|
43
|
+
runId: context.runId,
|
|
44
|
+
observer: context.observer,
|
|
45
|
+
agentName: context.agentName,
|
|
46
|
+
sessionId: context.sessionId,
|
|
47
|
+
},
|
|
48
|
+
};
|
|
49
|
+
assignIfDefined(options, 'metadata', metadata);
|
|
50
|
+
assignIfDefined(options, 'tenantId', readString(metadata, 'tenantId'));
|
|
51
|
+
assignIfDefined(options, 'knowledgeBaseId', readString(metadata, 'knowledgeBaseId'));
|
|
52
|
+
assignIfDefined(options, 'acl', readStringArray(metadata, 'acl'));
|
|
53
|
+
assignIfDefined(options, 'documentIds', readStringArray(metadata, 'documentIds'));
|
|
54
|
+
assignIfDefined(options, 'sources', readStringArray(metadata, 'sources'));
|
|
55
|
+
assignIfDefined(options, 'metadataFilter', readRecord(metadata, 'metadataFilter'));
|
|
56
|
+
assignIfDefined(options, 'topK', readNumber(metadata, 'topK'));
|
|
57
|
+
assignIfDefined(options, 'minScore', readNumber(metadata, 'minScore'));
|
|
58
|
+
assignIfDefined(options, 'maxContextTokens', readNumber(metadata, 'maxContextTokens'));
|
|
59
|
+
if (this.requireCitations !== undefined) {
|
|
60
|
+
options.requireCitations = this.requireCitations;
|
|
61
|
+
}
|
|
62
|
+
return options;
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
exports.RagExecutor = RagExecutor;
|
|
66
|
+
function toCitationMetadata(citation) {
|
|
67
|
+
const metadata = {
|
|
68
|
+
index: citation.index,
|
|
69
|
+
chunkId: citation.chunkId,
|
|
70
|
+
documentId: citation.documentId,
|
|
71
|
+
source: citation.source,
|
|
72
|
+
};
|
|
73
|
+
assignIfDefined(metadata, 'title', citation.title);
|
|
74
|
+
assignIfDefined(metadata, 'headingPath', citation.headingPath);
|
|
75
|
+
assignIfDefined(metadata, 'page', citation.page);
|
|
76
|
+
return metadata;
|
|
77
|
+
}
|
|
78
|
+
function toRetrievedMetadata(retrieved) {
|
|
79
|
+
const metadata = {
|
|
80
|
+
chunkId: retrieved.chunk.id,
|
|
81
|
+
documentId: retrieved.chunk.documentId,
|
|
82
|
+
source: retrieved.source,
|
|
83
|
+
score: retrieved.score,
|
|
84
|
+
documentSource: retrieved.chunk.metadata.source,
|
|
85
|
+
};
|
|
86
|
+
assignIfDefined(metadata, 'title', retrieved.chunk.metadata.title);
|
|
87
|
+
assignIfDefined(metadata, 'headingPath', retrieved.chunk.metadata.headingPath);
|
|
88
|
+
return metadata;
|
|
89
|
+
}
|
|
90
|
+
function readString(metadata, key) {
|
|
91
|
+
const value = metadata?.[key];
|
|
92
|
+
return typeof value === 'string' && value.trim().length > 0 ? value : undefined;
|
|
93
|
+
}
|
|
94
|
+
function readNumber(metadata, key) {
|
|
95
|
+
const value = metadata?.[key];
|
|
96
|
+
return typeof value === 'number' && Number.isFinite(value) ? value : undefined;
|
|
97
|
+
}
|
|
98
|
+
function readStringArray(metadata, key) {
|
|
99
|
+
const value = metadata?.[key];
|
|
100
|
+
if (!Array.isArray(value)) {
|
|
101
|
+
return undefined;
|
|
102
|
+
}
|
|
103
|
+
const strings = value
|
|
104
|
+
.filter((item) => typeof item === 'string')
|
|
105
|
+
.filter((item) => item.trim().length > 0);
|
|
106
|
+
return strings.length > 0 ? strings : undefined;
|
|
107
|
+
}
|
|
108
|
+
function readRecord(metadata, key) {
|
|
109
|
+
const value = metadata?.[key];
|
|
110
|
+
return isRecord(value) ? value : undefined;
|
|
111
|
+
}
|
|
112
|
+
function isRecord(value) {
|
|
113
|
+
return typeof value === 'object' && value !== null && !Array.isArray(value);
|
|
114
|
+
}
|
|
115
|
+
function assignIfDefined(target, key, value) {
|
|
116
|
+
if (value !== undefined) {
|
|
117
|
+
target[key] = value;
|
|
118
|
+
}
|
|
119
|
+
}
|
|
120
|
+
//# sourceMappingURL=rag-executor.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"rag-executor.js","sourceRoot":"","sources":["../../src/executors/rag-executor.ts"],"names":[],"mappings":";;;AAOA,0CAAsC;AAatC,MAAa,WAAW;IACH,QAAQ,CAAc;IACtB,aAAa,CAAuC;IACpD,gBAAgB,CAAsB;IAEvD,YAAY,OAA2B;QACnC,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QACjC,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,EAAE,CAAC;QACjD,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,OAAqB;QAC3B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CACrC,OAAO,CAAC,KAAK,EACb,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CACnC,CAAC;QACF,MAAM,WAAW,GAAG,IAAI,kBAAO,CAC3B,OAAO,CAAC,KAAK,EACb,MAAM,EACN,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CACzE,CAAC;QACF,MAAM,gBAAgB,GAAG,IAAI,kBAAO,CAAC,MAAM,CAAC,MAAM,EAAE,WAAW,EAAE;YAC7D,QAAQ,EAAE;gBACN,GAAG,EAAE;oBACD,UAAU,EAAE,MAAM,CAAC,UAAU;oBAC7B,SAAS,EAAE,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,kBAAkB,CAAC;oBACnD,SAAS,EAAE,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,mBAAmB,CAAC;iBACvD;aACJ;SACJ,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,gBAAgB,CAAC,CAAC,CAAC;QAE/D,MAAM,MAAM,GAAgB;YACxB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,QAAQ,EAAE,MAAM,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE;YACrC,GAAG,EAAE,MAAM;SACd,CAAC;QAEF,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAC7B,MAAM,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAChC,CAAC;QAED,OAAO,MAAM,CAAC;IAClB,CAAC;IAEO,kBAAkB,CAAC,OAAqB;QAC5C,MAAM,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC;QAC1C,MAAM,OAAO,GAAkB;YAC3B,GAAG,IAAI,CAAC,aAAa;YACrB,GAAG,EAAE,OAAO,CAAC,GAAG;YAChB,KAAK,EAAE;gBACH,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,SAAS,EAAE,OAAO,CAAC,SAAS;gBAC5B,SAAS,EAAE,OAAO,CAAC,SAAS;aAC/B;SACJ,CAAC;QAEF,eAAe,CAAC,OAAO,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC;QAC/C,eAAe,CAAC,OAAO,EAAE,UAAU,EAAE,UAAU,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC;QACvE,eAAe,CAAC,OAAO,EAAE,iBAAiB,EAAE,UAAU,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC,CAAC;QACrF,eAAe,CAAC,OAAO,EAAE,KAAK,EAAE,eAAe,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAC;QAClE,eAAe,CAAC,OAAO,EAAE,aAAa,EAAE,eAAe,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC;QAClF,eAAe,CAAC,OAAO,EAAE,SAAS,EAAE,eAAe,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC;QAC1E,eAAe,CAAC,OAAO,EAAE,gBAAgB,EAAE,UAAU,CAAC,QAAQ,EAAE,gBAAgB,CAAC,CAAC,CAAC;QACnF,eAAe,CAAC,OAAO,EAAE,MAAM,EAAE,UAAU,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC;QAC/D,eAAe,CAAC,OAAO,EAAE,UAAU,EAAE,UAAU,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC;QACvE,eAAe,CAAC,OAAO,EAAE,kBAAkB,EAAE,UAAU,CAAC,QAAQ,EAAE,kBAAkB,CAAC,CAAC,CAAC;QAEvF,IAAI,IAAI,CAAC,gBAAgB,KAAK,SAAS,EAAE,CAAC;YACtC,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC;QACrD,CAAC;QAED,OAAO,OAAO,CAAC;IACnB,CAAC;CACJ;AA5ED,kCA4EC;AAED,SAAS,kBAAkB,CAAC,QAAkB;IAC1C,MAAM,QAAQ,GAA4B;QACtC,KAAK,EAAE,QAAQ,CAAC,KAAK;QACrB,OAAO,EAAE,QAAQ,CAAC,OAAO;QACzB,UAAU,EAAE,QAAQ,CAAC,UAAU;QAC/B,MAAM,EAAE,QAAQ,CAAC,MAAM;KAC1B,CAAC;IAEF,eAAe,CAAC,QAAQ,EAAE,OAAO,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;IACnD,eAAe,CAAC,QAAQ,EAAE,aAAa,EAAE,QAAQ,CAAC,WAAW,CAAC,CAAC;IAC/D,eAAe,CAAC,QAAQ,EAAE,MAAM,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;IAEjD,OAAO,QAAQ,CAAC;AACpB,CAAC;AAED,SAAS,mBAAmB,CAAC,SAAyB;IAClD,MAAM,QAAQ,GAA4B;QACtC,OAAO,EAAE,SAAS,CAAC,KAAK,CAAC,EAAE;QAC3B,UAAU,EAAE,SAAS,CAAC,KAAK,CAAC,UAAU;QACtC,MAAM,EAAE,SAAS,CAAC,MAAM;QACxB,KAAK,EAAE,SAAS,CAAC,KAAK;QACtB,cAAc,EAAE,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM;KAClD,CAAC;IAEF,eAAe,CAAC,QAAQ,EAAE,OAAO,EAAE,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IACnE,eAAe,CAAC,QAAQ,EAAE,aAAa,EAAE,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;IAE/E,OAAO,QAAQ,CAAC;AACpB,CAAC;AAED,SAAS,UAAU,CAAC,QAA6C,EAAE,GAAW;IAC1E,MAAM,KAAK,GAAG,QAAQ,EAAE,CAAC,GAAG,CAAC,CAAC;IAE9B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;AACpF,CAAC;AAED,SAAS,UAAU,CAAC,QAA6C,EAAE,GAAW;IAC1E,MAAM,KAAK,GAAG,QAAQ,EAAE,CAAC,GAAG,CAAC,CAAC;IAE9B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;AACnF,CAAC;AAED,SAAS,eAAe,CACpB,QAA6C,EAC7C,GAAW;IAEX,MAAM,KAAK,GAAG,QAAQ,EAAE,CAAC,GAAG,CAAC,CAAC;IAC9B,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACxB,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,MAAM,OAAO,GAAG,KAAK;SAChB,MAAM,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC;SAC1D,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAE9C,OAAO,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;AACpD,CAAC;AAED,SAAS,UAAU,CACf,QAA6C,EAC7C,GAAW;IAEX,MAAM,KAAK,GAAG,QAAQ,EAAE,CAAC,GAAG,CAAC,CAAC;IAE9B,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;AAC/C,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC5B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAChF,CAAC;AAED,SAAS,eAAe,CACpB,MAAS,EACT,GAAM,EACN,KAAuB;IAEvB,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACtB,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;IACxB,CAAC;AACL,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"simple-chat-executor.d.ts","sourceRoot":"","sources":["../../src/executors/simple-chat-executor.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,QAAQ,CAAC;AAEvE,qBAAa,kBAAmB,YAAW,aAAa;IAC9C,GAAG,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC;CA6BzD"}
|
|
@@ -0,0 +1,77 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.SimpleChatExecutor = void 0;
|
|
4
|
+
const messages_1 = require("../messages");
|
|
5
|
+
class SimpleChatExecutor {
|
|
6
|
+
async run(context) {
|
|
7
|
+
const userMessage = new messages_1.Message(context.input, 'user', context.options.metadata ? { metadata: context.options.metadata } : {});
|
|
8
|
+
const messages = [
|
|
9
|
+
...(await context.history.get()),
|
|
10
|
+
userMessage,
|
|
11
|
+
];
|
|
12
|
+
const request = buildChatRequest(context, messages);
|
|
13
|
+
const response = await context.llm.chat(request);
|
|
14
|
+
const assistantMessage = new messages_1.Message(response.content, 'assistant');
|
|
15
|
+
await context.history.addMany([userMessage, assistantMessage]);
|
|
16
|
+
const result = {
|
|
17
|
+
output: response.content,
|
|
18
|
+
messages: await context.history.get(),
|
|
19
|
+
raw: response.raw,
|
|
20
|
+
};
|
|
21
|
+
if (response.usage !== undefined) {
|
|
22
|
+
result.usage = response.usage;
|
|
23
|
+
}
|
|
24
|
+
return result;
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
exports.SimpleChatExecutor = SimpleChatExecutor;
|
|
28
|
+
function buildChatRequest(context, history) {
|
|
29
|
+
const systemPrompt = context.options.systemPrompt ?? context.systemPrompt;
|
|
30
|
+
const messages = [
|
|
31
|
+
...(systemPrompt ? [new messages_1.Message(systemPrompt, 'system').toChatMessage()] : []),
|
|
32
|
+
...history.map((message) => message.toChatMessage()),
|
|
33
|
+
];
|
|
34
|
+
const request = {
|
|
35
|
+
messages,
|
|
36
|
+
trace: {
|
|
37
|
+
runId: context.runId,
|
|
38
|
+
observer: context.observer,
|
|
39
|
+
agentName: context.agentName,
|
|
40
|
+
sessionId: context.sessionId,
|
|
41
|
+
},
|
|
42
|
+
};
|
|
43
|
+
const model = context.options.model ?? context.config.defaultModel;
|
|
44
|
+
if (model) {
|
|
45
|
+
request.model = model;
|
|
46
|
+
}
|
|
47
|
+
request.temperature = context.options.temperature ?? context.config.temperature;
|
|
48
|
+
const maxTokens = context.options.maxTokens ?? context.config.maxTokens;
|
|
49
|
+
if (maxTokens !== undefined) {
|
|
50
|
+
request.maxTokens = maxTokens;
|
|
51
|
+
}
|
|
52
|
+
if (context.options.signal) {
|
|
53
|
+
request.signal = context.options.signal;
|
|
54
|
+
}
|
|
55
|
+
applyRuntimeOptions(context, request);
|
|
56
|
+
return request;
|
|
57
|
+
}
|
|
58
|
+
function applyRuntimeOptions(context, request) {
|
|
59
|
+
// 单次 run 的 options 优先级高于 Agent 的默认 config。
|
|
60
|
+
const timeoutMs = context.options.timeoutMs ?? context.config.requestTimeoutMs;
|
|
61
|
+
if (timeoutMs !== undefined) {
|
|
62
|
+
request.timeoutMs = timeoutMs;
|
|
63
|
+
}
|
|
64
|
+
const maxRetries = context.options.maxRetries ?? context.config.maxRetries;
|
|
65
|
+
if (maxRetries !== undefined) {
|
|
66
|
+
request.maxRetries = maxRetries;
|
|
67
|
+
}
|
|
68
|
+
const retryDelayMs = context.options.retryDelayMs ?? context.config.retryDelayMs;
|
|
69
|
+
if (retryDelayMs !== undefined) {
|
|
70
|
+
request.retryDelayMs = retryDelayMs;
|
|
71
|
+
}
|
|
72
|
+
const retryMaxDelayMs = context.options.retryMaxDelayMs ?? context.config.retryMaxDelayMs;
|
|
73
|
+
if (retryMaxDelayMs !== undefined) {
|
|
74
|
+
request.retryMaxDelayMs = retryMaxDelayMs;
|
|
75
|
+
}
|
|
76
|
+
}
|
|
77
|
+
//# sourceMappingURL=simple-chat-executor.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"simple-chat-executor.js","sourceRoot":"","sources":["../../src/executors/simple-chat-executor.ts"],"names":[],"mappings":";;;AACA,0CAAsC;AAGtC,MAAa,kBAAkB;IAC3B,KAAK,CAAC,GAAG,CAAC,OAAqB;QAC3B,MAAM,WAAW,GAAG,IAAI,kBAAO,CAC3B,OAAO,CAAC,KAAK,EACb,MAAM,EACN,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CACzE,CAAC;QAEF,MAAM,QAAQ,GAAG;YACb,GAAG,CAAC,MAAM,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;YAChC,WAAW;SACd,CAAC;QACF,MAAM,OAAO,GAAG,gBAAgB,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QACpD,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAEjD,MAAM,gBAAgB,GAAG,IAAI,kBAAO,CAAC,QAAQ,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QACpE,MAAM,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,gBAAgB,CAAC,CAAC,CAAC;QAE/D,MAAM,MAAM,GAAgB;YACxB,MAAM,EAAE,QAAQ,CAAC,OAAO;YACxB,QAAQ,EAAE,MAAM,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE;YACrC,GAAG,EAAE,QAAQ,CAAC,GAAG;SACpB,CAAC;QAEF,IAAI,QAAQ,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAC/B,MAAM,CAAC,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;QAClC,CAAC;QAED,OAAO,MAAM,CAAC;IAClB,CAAC;CACJ;AA9BD,gDA8BC;AAED,SAAS,gBAAgB,CAAC,OAAqB,EAAE,OAAkB;IAC/D,MAAM,YAAY,GAAG,OAAO,CAAC,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,YAAY,CAAC;IAC1E,MAAM,QAAQ,GAAG;QACb,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI,kBAAO,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC9E,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;KACvD,CAAC;IAEF,MAAM,OAAO,GAAgB;QACzB,QAAQ;QACR,KAAK,EAAE;YACH,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,SAAS,EAAE,OAAO,CAAC,SAAS;SAC/B;KACJ,CAAC;IACF,MAAM,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,MAAM,CAAC,YAAY,CAAC;IACnE,IAAI,KAAK,EAAE,CAAC;QACR,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC;IAC1B,CAAC;IAED,OAAO,CAAC,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC;IAEhF,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC;IACxE,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;QAC1B,OAAO,CAAC,SAAS,GAAG,SAAS,CAAC;IAClC,CAAC;IAED,IAAI,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;QACzB,OAAO,CAAC,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;IAC5C,CAAC;IAED,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAEtC,OAAO,OAAO,CAAC;AACnB,CAAC;AAED,SAAS,mBAAmB,CAAC,OAAqB,EAAE,OAAoB;IACpE,2CAA2C;IAC3C,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,MAAM,CAAC,gBAAgB,CAAC;IAC/E,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;QAC1B,OAAO,CAAC,SAAS,GAAG,SAAS,CAAC;IAClC,CAAC;IAED,MAAM,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC;IAC3E,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAC3B,OAAO,CAAC,UAAU,GAAG,UAAU,CAAC;IACpC,CAAC;IAED,MAAM,YAAY,GAAG,OAAO,CAAC,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,MAAM,CAAC,YAAY,CAAC;IACjF,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;QAC7B,OAAO,CAAC,YAAY,GAAG,YAAY,CAAC;IACxC,CAAC;IAED,MAAM,eAAe,GAAG,OAAO,CAAC,OAAO,CAAC,eAAe,IAAI,OAAO,CAAC,MAAM,CAAC,eAAe,CAAC;IAC1F,IAAI,eAAe,KAAK,SAAS,EAAE,CAAC;QAChC,OAAO,CAAC,eAAe,GAAG,eAAe,CAAC;IAC9C,CAAC;AACL,CAAC"}
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
import type { AgentContext, AgentExecutor, AgentResult } from './base';
|
|
2
|
+
export interface ToolCallingExecutorOptions {
|
|
3
|
+
maxIterations?: number;
|
|
4
|
+
}
|
|
5
|
+
export declare class ToolCallingExecutor implements AgentExecutor {
|
|
6
|
+
private readonly maxIterations;
|
|
7
|
+
constructor(options?: ToolCallingExecutorOptions);
|
|
8
|
+
run(context: AgentContext): Promise<AgentResult>;
|
|
9
|
+
}
|
|
10
|
+
//# sourceMappingURL=tool-calling-executor.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool-calling-executor.d.ts","sourceRoot":"","sources":["../../src/executors/tool-calling-executor.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,QAAQ,CAAC;AAEvE,MAAM,WAAW,0BAA0B;IACvC,aAAa,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED,qBAAa,mBAAoB,YAAW,aAAa;IACrD,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAS;gBAE3B,OAAO,GAAE,0BAA+B;IAI9C,GAAG,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC;CA+DzD"}
|
|
@@ -0,0 +1,151 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.ToolCallingExecutor = void 0;
|
|
4
|
+
const messages_1 = require("../messages");
|
|
5
|
+
const tools_1 = require("../tools");
|
|
6
|
+
class ToolCallingExecutor {
|
|
7
|
+
maxIterations;
|
|
8
|
+
constructor(options = {}) {
|
|
9
|
+
this.maxIterations = options.maxIterations ?? 5;
|
|
10
|
+
}
|
|
11
|
+
async run(context) {
|
|
12
|
+
const userMessage = new messages_1.Message(context.input, 'user', context.options.metadata ? { metadata: context.options.metadata } : {});
|
|
13
|
+
const runMessages = [userMessage];
|
|
14
|
+
const toolExecutor = new tools_1.ToolExecutor(context.tools, {
|
|
15
|
+
runId: context.runId,
|
|
16
|
+
observer: context.observer,
|
|
17
|
+
agentName: context.agentName,
|
|
18
|
+
sessionId: context.sessionId,
|
|
19
|
+
});
|
|
20
|
+
const messages = await buildInitialMessages(context, userMessage);
|
|
21
|
+
let usage;
|
|
22
|
+
let raw;
|
|
23
|
+
for (let iteration = 0; iteration < this.maxIterations; iteration += 1) {
|
|
24
|
+
const response = await context.llm.chat(buildChatRequest(context, messages));
|
|
25
|
+
usage = mergeTokenUsage(usage, response.usage);
|
|
26
|
+
raw = response.raw;
|
|
27
|
+
if (response.toolCalls.length === 0) {
|
|
28
|
+
const assistantMessage = new messages_1.Message(response.content, 'assistant');
|
|
29
|
+
runMessages.push(assistantMessage);
|
|
30
|
+
await context.history.addMany(runMessages);
|
|
31
|
+
const result = {
|
|
32
|
+
output: response.content,
|
|
33
|
+
messages: await context.history.get(),
|
|
34
|
+
raw,
|
|
35
|
+
};
|
|
36
|
+
if (usage !== undefined) {
|
|
37
|
+
result.usage = usage;
|
|
38
|
+
}
|
|
39
|
+
return result;
|
|
40
|
+
}
|
|
41
|
+
const assistantMessage = new messages_1.Message(response.content, 'assistant', {
|
|
42
|
+
metadata: { toolCalls: response.toolCalls },
|
|
43
|
+
});
|
|
44
|
+
runMessages.push(assistantMessage);
|
|
45
|
+
messages.push(assistantMessage.toChatMessage());
|
|
46
|
+
for (const toolCall of response.toolCalls) {
|
|
47
|
+
const result = await toolExecutor.execute(toolCall);
|
|
48
|
+
const toolMessage = new messages_1.Message(result, 'tool', {
|
|
49
|
+
metadata: {
|
|
50
|
+
toolCallId: toolCall.id,
|
|
51
|
+
toolName: toolCall.name,
|
|
52
|
+
},
|
|
53
|
+
});
|
|
54
|
+
runMessages.push(toolMessage);
|
|
55
|
+
messages.push(toolMessage.toChatMessage());
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
throw new Error(`Tool calling exceeded max iterations: ${this.maxIterations}`);
|
|
59
|
+
}
|
|
60
|
+
}
|
|
61
|
+
exports.ToolCallingExecutor = ToolCallingExecutor;
|
|
62
|
+
async function buildInitialMessages(context, userMessage) {
|
|
63
|
+
const systemPrompt = context.options.systemPrompt ?? context.systemPrompt;
|
|
64
|
+
const history = await context.history.get();
|
|
65
|
+
return [
|
|
66
|
+
...(systemPrompt ? [new messages_1.Message(systemPrompt, 'system').toChatMessage()] : []),
|
|
67
|
+
...history.map((message) => message.toChatMessage()),
|
|
68
|
+
userMessage.toChatMessage(),
|
|
69
|
+
];
|
|
70
|
+
}
|
|
71
|
+
function buildChatRequest(context, messages) {
|
|
72
|
+
const request = {
|
|
73
|
+
messages,
|
|
74
|
+
trace: {
|
|
75
|
+
runId: context.runId,
|
|
76
|
+
observer: context.observer,
|
|
77
|
+
agentName: context.agentName,
|
|
78
|
+
sessionId: context.sessionId,
|
|
79
|
+
},
|
|
80
|
+
};
|
|
81
|
+
const model = context.options.model ?? context.config.defaultModel;
|
|
82
|
+
if (model) {
|
|
83
|
+
request.model = model;
|
|
84
|
+
}
|
|
85
|
+
request.temperature = context.options.temperature ?? context.config.temperature;
|
|
86
|
+
const maxTokens = context.options.maxTokens ?? context.config.maxTokens;
|
|
87
|
+
if (maxTokens !== undefined) {
|
|
88
|
+
request.maxTokens = maxTokens;
|
|
89
|
+
}
|
|
90
|
+
const tools = context.tools.toOpenAISchemas();
|
|
91
|
+
if (tools.length > 0) {
|
|
92
|
+
request.tools = tools;
|
|
93
|
+
request.toolChoice = 'auto';
|
|
94
|
+
}
|
|
95
|
+
if (context.options.signal) {
|
|
96
|
+
request.signal = context.options.signal;
|
|
97
|
+
}
|
|
98
|
+
applyRuntimeOptions(context, request);
|
|
99
|
+
return request;
|
|
100
|
+
}
|
|
101
|
+
function applyRuntimeOptions(context, request) {
|
|
102
|
+
// 单次 run 的 options 优先级高于 Agent 的默认 config。
|
|
103
|
+
const timeoutMs = context.options.timeoutMs ?? context.config.requestTimeoutMs;
|
|
104
|
+
if (timeoutMs !== undefined) {
|
|
105
|
+
request.timeoutMs = timeoutMs;
|
|
106
|
+
}
|
|
107
|
+
const maxRetries = context.options.maxRetries ?? context.config.maxRetries;
|
|
108
|
+
if (maxRetries !== undefined) {
|
|
109
|
+
request.maxRetries = maxRetries;
|
|
110
|
+
}
|
|
111
|
+
const retryDelayMs = context.options.retryDelayMs ?? context.config.retryDelayMs;
|
|
112
|
+
if (retryDelayMs !== undefined) {
|
|
113
|
+
request.retryDelayMs = retryDelayMs;
|
|
114
|
+
}
|
|
115
|
+
const retryMaxDelayMs = context.options.retryMaxDelayMs ?? context.config.retryMaxDelayMs;
|
|
116
|
+
if (retryMaxDelayMs !== undefined) {
|
|
117
|
+
request.retryMaxDelayMs = retryMaxDelayMs;
|
|
118
|
+
}
|
|
119
|
+
}
|
|
120
|
+
function mergeTokenUsage(current, next) {
|
|
121
|
+
if (next === undefined) {
|
|
122
|
+
return current;
|
|
123
|
+
}
|
|
124
|
+
if (current === undefined) {
|
|
125
|
+
return { ...next };
|
|
126
|
+
}
|
|
127
|
+
const merged = {};
|
|
128
|
+
const promptTokens = addOptionalNumbers(current.promptTokens, next.promptTokens);
|
|
129
|
+
const completionTokens = addOptionalNumbers(current.completionTokens, next.completionTokens);
|
|
130
|
+
const totalTokens = addOptionalNumbers(current.totalTokens, next.totalTokens);
|
|
131
|
+
if (promptTokens !== undefined) {
|
|
132
|
+
merged.promptTokens = promptTokens;
|
|
133
|
+
}
|
|
134
|
+
if (completionTokens !== undefined) {
|
|
135
|
+
merged.completionTokens = completionTokens;
|
|
136
|
+
}
|
|
137
|
+
if (totalTokens !== undefined) {
|
|
138
|
+
merged.totalTokens = totalTokens;
|
|
139
|
+
}
|
|
140
|
+
return merged;
|
|
141
|
+
}
|
|
142
|
+
function addOptionalNumbers(left, right) {
|
|
143
|
+
if (left === undefined) {
|
|
144
|
+
return right;
|
|
145
|
+
}
|
|
146
|
+
if (right === undefined) {
|
|
147
|
+
return left;
|
|
148
|
+
}
|
|
149
|
+
return left + right;
|
|
150
|
+
}
|
|
151
|
+
//# sourceMappingURL=tool-calling-executor.js.map
|