@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 @@
|
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/rag/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAErD,MAAM,MAAM,SAAS,GAAG,MAAM,GAAG,OAAO,GAAG,MAAM,GAAG,YAAY,CAAC;AACjE,MAAM,MAAM,eAAe,GAAG,OAAO,GAAG,QAAQ,GAAG,QAAQ,GAAG,QAAQ,CAAC;AACvE,MAAM,MAAM,aAAa,GAAG,MAAM,GAAG,QAAQ,GAAG,KAAK,CAAC;AAEtD;;;GAGG;AACH,MAAM,WAAW,WAAW;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,MAAM,CAAC,EAAE,aAAa,EAAE,CAAC;CAC5B;AAED,MAAM,WAAW,gBAAgB;IAC7B,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,GAAG,CAAC,EAAE,MAAM,EAAE,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED;;;GAGG;AACH,MAAM,MAAM,aAAa,GACnB;IAAE,IAAI,EAAE,SAAS,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,GAChD;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GACnC;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAC;IAAC,IAAI,CAAC,EAAE,OAAO,EAAE,CAAA;CAAE,GACrD;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GACjD;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,GAAG,CAAC,EAAE,MAAM,CAAC;IAAC,IAAI,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC;AAErD;;;GAGG;AACH,MAAM,WAAW,KAAK;IAClB,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,SAAS,CAAC;IAChB,QAAQ,EAAE,aAAa,CAAC;CAC3B;AAED,MAAM,WAAW,aAAa;IAC1B,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,GAAG,CAAC,EAAE,MAAM,EAAE,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED,MAAM,WAAW,aAAa;IAC1B,KAAK,EAAE,KAAK,CAAC;IACb,SAAS,EAAE,MAAM,EAAE,CAAC;CACvB;AAED;;;GAGG;AACH,MAAM,WAAW,cAAc;IAC3B,KAAK,EAAE,KAAK,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,eAAe,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED,MAAM,WAAW,QAAQ;IACrB;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED,MAAM,WAAW,cAAc;IAC3B,QAAQ,EAAE,QAAQ,CAAC;IACnB,KAAK,EAAE,KAAK,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,eAAe,CAAC;IACxB,aAAa,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,UAAU;IACvB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,cAAc,EAAE,CAAC;IACxB,SAAS,EAAE,QAAQ,EAAE,CAAC;IACtB,eAAe,EAAE,MAAM,CAAC;CAC3B;AAED,MAAM,WAAW,SAAS;IACtB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,QAAQ,EAAE,CAAC;IACtB,UAAU,EAAE,aAAa,CAAC;IAC1B,SAAS,EAAE,cAAc,EAAE,CAAC;IAC5B,KAAK,CAAC,EAAE,UAAU,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED;;;GAGG;AACH,MAAM,WAAW,WAAW;IACxB,SAAS,CAAC,EAAE,WAAW,EAAE,CAAC;IAC1B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,UAAU,GAAG,WAAW,CAAC;IAChC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;IACrC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,GAAG,CAAC,EAAE,MAAM,EAAE,CAAC;IACf,eAAe,CAAC,EAAE,OAAO,CAAC;CAC7B;AAED,MAAM,WAAW,YAAY;IACzB,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,EAAE,CAAC;IACtB,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,kBAAkB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC9B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED,MAAM,WAAW,eAAe;IAC5B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,GAAG,CAAC,EAAE,MAAM,EAAE,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACzC,KAAK,CAAC,EAAE,YAAY,CAAC;CACxB;AAED,MAAM,WAAW,aAAc,SAAQ,eAAe;IAClD;;OAEG;IACH,GAAG,CAAC,EAAE,SAAS,CAAC;IAChB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/rag/types.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
import type { ChatToolSchema } from '../llm/types';
|
|
3
|
+
export type ToolInputSchema = z.ZodType;
|
|
4
|
+
export type ToolInput<TSchema extends ToolInputSchema> = z.infer<TSchema>;
|
|
5
|
+
export interface ToolOptions<TSchema extends ToolInputSchema> {
|
|
6
|
+
name: string;
|
|
7
|
+
description: string;
|
|
8
|
+
schema: TSchema;
|
|
9
|
+
}
|
|
10
|
+
export declare class ToolValidationError extends Error {
|
|
11
|
+
readonly toolName: string;
|
|
12
|
+
readonly issues: readonly z.core.$ZodIssue[];
|
|
13
|
+
constructor(toolName: string, error: z.ZodError);
|
|
14
|
+
}
|
|
15
|
+
export declare abstract class Tool<TSchema extends ToolInputSchema = ToolInputSchema, TResult = unknown> {
|
|
16
|
+
readonly name: string;
|
|
17
|
+
readonly description: string;
|
|
18
|
+
readonly schema: TSchema;
|
|
19
|
+
protected constructor(options: ToolOptions<TSchema>);
|
|
20
|
+
/**
|
|
21
|
+
* 模型返回的是未可信的 JSON 参数,进入具体工具前必须先通过 schema。
|
|
22
|
+
* 子类只实现 run(validatedInput),不用重复写参数校验样板代码。
|
|
23
|
+
*/
|
|
24
|
+
execute(parameters: unknown): Promise<TResult>;
|
|
25
|
+
toOpenAISchema(): ChatToolSchema;
|
|
26
|
+
protected abstract run(parameters: ToolInput<TSchema>): TResult | Promise<TResult>;
|
|
27
|
+
}
|
|
28
|
+
//# sourceMappingURL=base.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../src/tools/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAEnD,MAAM,MAAM,eAAe,GAAG,CAAC,CAAC,OAAO,CAAC;AACxC,MAAM,MAAM,SAAS,CAAC,OAAO,SAAS,eAAe,IAAI,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;AAE1E,MAAM,WAAW,WAAW,CAAC,OAAO,SAAS,eAAe;IACxD,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,OAAO,CAAC;CACnB;AAED,qBAAa,mBAAoB,SAAQ,KAAK;IAC1C,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBAEjC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,QAAQ;CAMlD;AAED,8BAAsB,IAAI,CACtB,OAAO,SAAS,eAAe,GAAG,eAAe,EACjD,OAAO,GAAG,OAAO;IAEjB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC;IAEzB,SAAS,aAAa,OAAO,EAAE,WAAW,CAAC,OAAO,CAAC;IAMnD;;;OAGG;IACG,OAAO,CAAC,UAAU,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;IASpD,cAAc,IAAI,cAAc;IAYhC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,EAAE,SAAS,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;CACrF"}
|
|
@@ -0,0 +1,68 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.Tool = exports.ToolValidationError = void 0;
|
|
4
|
+
const zod_1 = require("zod");
|
|
5
|
+
class ToolValidationError extends Error {
|
|
6
|
+
toolName;
|
|
7
|
+
issues;
|
|
8
|
+
constructor(toolName, error) {
|
|
9
|
+
super(`Tool "${toolName}" received invalid parameters.`);
|
|
10
|
+
this.name = 'ToolValidationError';
|
|
11
|
+
this.toolName = toolName;
|
|
12
|
+
this.issues = error.issues;
|
|
13
|
+
}
|
|
14
|
+
}
|
|
15
|
+
exports.ToolValidationError = ToolValidationError;
|
|
16
|
+
class Tool {
|
|
17
|
+
name;
|
|
18
|
+
description;
|
|
19
|
+
schema;
|
|
20
|
+
constructor(options) {
|
|
21
|
+
this.name = options.name;
|
|
22
|
+
this.description = options.description;
|
|
23
|
+
this.schema = options.schema;
|
|
24
|
+
}
|
|
25
|
+
/**
|
|
26
|
+
* 模型返回的是未可信的 JSON 参数,进入具体工具前必须先通过 schema。
|
|
27
|
+
* 子类只实现 run(validatedInput),不用重复写参数校验样板代码。
|
|
28
|
+
*/
|
|
29
|
+
async execute(parameters) {
|
|
30
|
+
const parsed = await this.schema.safeParseAsync(parameters);
|
|
31
|
+
if (!parsed.success) {
|
|
32
|
+
throw new ToolValidationError(this.name, parsed.error);
|
|
33
|
+
}
|
|
34
|
+
return await this.run(parsed.data);
|
|
35
|
+
}
|
|
36
|
+
toOpenAISchema() {
|
|
37
|
+
return {
|
|
38
|
+
type: 'function',
|
|
39
|
+
function: {
|
|
40
|
+
name: this.name,
|
|
41
|
+
description: this.description,
|
|
42
|
+
strict: true,
|
|
43
|
+
parameters: toStrictObjectSchema(this.schema),
|
|
44
|
+
},
|
|
45
|
+
};
|
|
46
|
+
}
|
|
47
|
+
}
|
|
48
|
+
exports.Tool = Tool;
|
|
49
|
+
function toStrictObjectSchema(schema) {
|
|
50
|
+
const jsonSchema = zod_1.z.toJSONSchema(schema);
|
|
51
|
+
const normalizedSchema = stripUnsupportedJsonSchemaFields(jsonSchema);
|
|
52
|
+
// OpenAI function calling 的参数根节点必须是 object。
|
|
53
|
+
normalizedSchema.type = 'object';
|
|
54
|
+
normalizedSchema.properties = isRecord(normalizedSchema.properties)
|
|
55
|
+
? normalizedSchema.properties
|
|
56
|
+
: {};
|
|
57
|
+
normalizedSchema.additionalProperties = false;
|
|
58
|
+
return normalizedSchema;
|
|
59
|
+
}
|
|
60
|
+
function stripUnsupportedJsonSchemaFields(schema) {
|
|
61
|
+
const result = { ...schema };
|
|
62
|
+
delete result.$schema;
|
|
63
|
+
return result;
|
|
64
|
+
}
|
|
65
|
+
function isRecord(value) {
|
|
66
|
+
return typeof value === 'object' && value !== null && !Array.isArray(value);
|
|
67
|
+
}
|
|
68
|
+
//# sourceMappingURL=base.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/tools/base.ts"],"names":[],"mappings":";;;AAAA,6BAAwB;AAYxB,MAAa,mBAAoB,SAAQ,KAAK;IACjC,QAAQ,CAAS;IACjB,MAAM,CAA8B;IAE7C,YAAY,QAAgB,EAAE,KAAiB;QAC3C,KAAK,CAAC,SAAS,QAAQ,gCAAgC,CAAC,CAAC;QACzD,IAAI,CAAC,IAAI,GAAG,qBAAqB,CAAC;QAClC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;IAC/B,CAAC;CACJ;AAVD,kDAUC;AAED,MAAsB,IAAI;IAIb,IAAI,CAAS;IACb,WAAW,CAAS;IACpB,MAAM,CAAU;IAEzB,YAAsB,OAA6B;QAC/C,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;QACzB,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;QACvC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IACjC,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,OAAO,CAAC,UAAmB;QAC7B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;QAC5D,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YAClB,MAAM,IAAI,mBAAmB,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QAC3D,CAAC;QAED,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACvC,CAAC;IAED,cAAc;QACV,OAAO;YACH,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE;gBACN,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,MAAM,EAAE,IAAI;gBACZ,UAAU,EAAE,oBAAoB,CAAC,IAAI,CAAC,MAAM,CAAC;aAChD;SACJ,CAAC;IACN,CAAC;CAGJ;AAxCD,oBAwCC;AAED,SAAS,oBAAoB,CAAC,MAAuB;IACjD,MAAM,UAAU,GAAG,OAAC,CAAC,YAAY,CAAC,MAAM,CAA4B,CAAC;IACrE,MAAM,gBAAgB,GAAG,gCAAgC,CAAC,UAAU,CAAC,CAAC;IAEtE,4CAA4C;IAC5C,gBAAgB,CAAC,IAAI,GAAG,QAAQ,CAAC;IACjC,gBAAgB,CAAC,UAAU,GAAG,QAAQ,CAAC,gBAAgB,CAAC,UAAU,CAAC;QAC/D,CAAC,CAAC,gBAAgB,CAAC,UAAU;QAC7B,CAAC,CAAC,EAAE,CAAC;IACT,gBAAgB,CAAC,oBAAoB,GAAG,KAAK,CAAC;IAE9C,OAAO,gBAAgB,CAAC;AAC5B,CAAC;AAED,SAAS,gCAAgC,CAAC,MAA+B;IACrE,MAAM,MAAM,GAAG,EAAE,GAAG,MAAM,EAAE,CAAC;IAC7B,OAAO,MAAM,CAAC,OAAO,CAAC;IAEtB,OAAO,MAAM,CAAC;AAClB,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"}
|
|
@@ -0,0 +1,24 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
import { Tool } from '../base';
|
|
3
|
+
declare const AdvancedSearchToolSchema: z.ZodObject<{
|
|
4
|
+
query: z.ZodString;
|
|
5
|
+
}, z.core.$strip>;
|
|
6
|
+
type AdvancedSearchToolInput = z.infer<typeof AdvancedSearchToolSchema>;
|
|
7
|
+
export interface AdvancedSearchToolLogger {
|
|
8
|
+
info?(message: string): void;
|
|
9
|
+
warn?(message: string): void;
|
|
10
|
+
}
|
|
11
|
+
export interface AdvancedSearchToolOptions {
|
|
12
|
+
logger?: AdvancedSearchToolLogger;
|
|
13
|
+
}
|
|
14
|
+
export declare class AdvancedSearchTool extends Tool<typeof AdvancedSearchToolSchema> {
|
|
15
|
+
private readonly searchSources;
|
|
16
|
+
private readonly logger;
|
|
17
|
+
constructor(options?: AdvancedSearchToolOptions);
|
|
18
|
+
private setupSearchSources;
|
|
19
|
+
protected run({ query }: AdvancedSearchToolInput): Promise<string>;
|
|
20
|
+
private searchWithTavily;
|
|
21
|
+
private searchWithSerpApi;
|
|
22
|
+
}
|
|
23
|
+
export {};
|
|
24
|
+
//# sourceMappingURL=advancedSearchTool.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"advancedSearchTool.d.ts","sourceRoot":"","sources":["../../../src/tools/builtin/advancedSearchTool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAE/B,QAAA,MAAM,wBAAwB;;iBAE5B,CAAC;AAEH,KAAK,uBAAuB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,wBAAwB,CAAC,CAAC;AAkBxE,MAAM,WAAW,wBAAwB;IACrC,IAAI,CAAC,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,IAAI,CAAC,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;CAChC;AAED,MAAM,WAAW,yBAAyB;IACtC,MAAM,CAAC,EAAE,wBAAwB,CAAC;CACrC;AAED,qBAAa,kBAAmB,SAAQ,IAAI,CAAC,OAAO,wBAAwB,CAAC;IACzE,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAsB;IACpD,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAuC;gBAElD,OAAO,GAAE,yBAA8B;IAWnD,OAAO,CAAC,kBAAkB;cAsBV,GAAG,CAAC,EAAE,KAAK,EAAE,EAAE,uBAAuB,GAAG,OAAO,CAAC,MAAM,CAAC;YA4B1D,gBAAgB;YAyChB,iBAAiB;CA4BlC"}
|
|
@@ -0,0 +1,134 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.AdvancedSearchTool = void 0;
|
|
4
|
+
const zod_1 = require("zod");
|
|
5
|
+
const base_1 = require("../base");
|
|
6
|
+
const AdvancedSearchToolSchema = zod_1.z.object({
|
|
7
|
+
query: zod_1.z.string().trim().min(1).describe('需要搜索的关键词或问题'),
|
|
8
|
+
});
|
|
9
|
+
class AdvancedSearchTool extends base_1.Tool {
|
|
10
|
+
searchSources = [];
|
|
11
|
+
logger;
|
|
12
|
+
constructor(options = {}) {
|
|
13
|
+
super({
|
|
14
|
+
name: 'advanced_search',
|
|
15
|
+
description: '高级搜索工具,整合了 Tavily 和 SerpAPI 多个搜索源,提供更全面的搜索结果。遇到特定事实、新闻或知识盲区时使用。',
|
|
16
|
+
schema: AdvancedSearchToolSchema,
|
|
17
|
+
});
|
|
18
|
+
this.logger = options.logger;
|
|
19
|
+
this.setupSearchSources();
|
|
20
|
+
}
|
|
21
|
+
setupSearchSources() {
|
|
22
|
+
if (process.env.TAVILY_API_KEY) {
|
|
23
|
+
this.searchSources.push('tavily');
|
|
24
|
+
this.logger?.info?.('Tavily 搜索源已启用');
|
|
25
|
+
}
|
|
26
|
+
else {
|
|
27
|
+
this.logger?.info?.('Tavily API Key 未配置');
|
|
28
|
+
}
|
|
29
|
+
if (process.env.SERPAPI_API_KEY) {
|
|
30
|
+
this.searchSources.push('serpapi');
|
|
31
|
+
this.logger?.info?.('SerpApi 搜索源已启用');
|
|
32
|
+
}
|
|
33
|
+
else {
|
|
34
|
+
this.logger?.info?.('SerpApi API Key 未配置');
|
|
35
|
+
}
|
|
36
|
+
if (this.searchSources.length > 0) {
|
|
37
|
+
this.logger?.info?.(`可用搜索源: ${this.searchSources.join(', ')}`);
|
|
38
|
+
}
|
|
39
|
+
else {
|
|
40
|
+
this.logger?.warn?.('没有可用的搜索源,请配置 API 密钥');
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
async run({ query }) {
|
|
44
|
+
if (this.searchSources.length === 0) {
|
|
45
|
+
return '没有可用的搜索源,请配置 TAVILY_API_KEY 或 SERPAPI_API_KEY 环境变量';
|
|
46
|
+
}
|
|
47
|
+
this.logger?.info?.(`开始智能搜索: "${query}"`);
|
|
48
|
+
for (const source of this.searchSources) {
|
|
49
|
+
try {
|
|
50
|
+
if (source === 'tavily') {
|
|
51
|
+
const result = await this.searchWithTavily(query);
|
|
52
|
+
if (result && !result.includes('未找到')) {
|
|
53
|
+
return `Tavily AI 搜索结果:\n\n${result}`;
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
else if (source === 'serpapi') {
|
|
57
|
+
const result = await this.searchWithSerpApi(query);
|
|
58
|
+
if (result && !result.includes('未找到')) {
|
|
59
|
+
return `SerpApi Google 搜索结果:\n\n${result}`;
|
|
60
|
+
}
|
|
61
|
+
}
|
|
62
|
+
}
|
|
63
|
+
catch (error) {
|
|
64
|
+
this.logger?.warn?.(`${source} 搜索失败: ${formatError(error)}`);
|
|
65
|
+
}
|
|
66
|
+
}
|
|
67
|
+
return '所有搜索源都失败了,请检查网络连接和 API 密钥配置';
|
|
68
|
+
}
|
|
69
|
+
async searchWithTavily(query) {
|
|
70
|
+
const apiKey = process.env.TAVILY_API_KEY;
|
|
71
|
+
if (!apiKey) {
|
|
72
|
+
throw new Error('TAVILY_API_KEY 未配置');
|
|
73
|
+
}
|
|
74
|
+
const response = await fetch('https://api.tavily.com/search', {
|
|
75
|
+
method: 'POST',
|
|
76
|
+
headers: { 'Content-Type': 'application/json' },
|
|
77
|
+
body: JSON.stringify({
|
|
78
|
+
api_key: apiKey,
|
|
79
|
+
query,
|
|
80
|
+
search_depth: 'basic',
|
|
81
|
+
max_results: 3,
|
|
82
|
+
include_answer: true,
|
|
83
|
+
}),
|
|
84
|
+
});
|
|
85
|
+
if (!response.ok) {
|
|
86
|
+
throw new Error(`Tavily 请求失败, 状态码: ${response.status}`);
|
|
87
|
+
}
|
|
88
|
+
const data = await response.json();
|
|
89
|
+
let result = '';
|
|
90
|
+
const answer = toText(data.answer);
|
|
91
|
+
if (answer) {
|
|
92
|
+
result += `AI 直接答案: ${answer}\n\n`;
|
|
93
|
+
}
|
|
94
|
+
result += '相关结果:\n';
|
|
95
|
+
const results = Array.isArray(data.results) ? data.results : [];
|
|
96
|
+
results.slice(0, 3).forEach((item, index) => {
|
|
97
|
+
const title = toText(item.title) || '未命名结果';
|
|
98
|
+
const content = toText(item.content);
|
|
99
|
+
result += `[${index + 1}] ${title}\n ${content.slice(0, 150)}...\n\n`;
|
|
100
|
+
});
|
|
101
|
+
return result.trim() || '未找到相关结果。';
|
|
102
|
+
}
|
|
103
|
+
async searchWithSerpApi(query) {
|
|
104
|
+
const apiKey = process.env.SERPAPI_API_KEY;
|
|
105
|
+
if (!apiKey) {
|
|
106
|
+
throw new Error('SERPAPI_API_KEY 未配置');
|
|
107
|
+
}
|
|
108
|
+
const url = new URL('https://serpapi.com/search.json');
|
|
109
|
+
url.searchParams.append('q', query);
|
|
110
|
+
url.searchParams.append('api_key', apiKey);
|
|
111
|
+
url.searchParams.append('num', '3');
|
|
112
|
+
const response = await fetch(url.toString());
|
|
113
|
+
if (!response.ok) {
|
|
114
|
+
throw new Error(`SerpApi 请求失败, 状态码: ${response.status}`);
|
|
115
|
+
}
|
|
116
|
+
const data = await response.json();
|
|
117
|
+
const organicResults = Array.isArray(data.organic_results) ? data.organic_results : [];
|
|
118
|
+
let result = 'Google 搜索结果:\n';
|
|
119
|
+
organicResults.slice(0, 3).forEach((item, index) => {
|
|
120
|
+
const title = toText(item.title) || '未命名结果';
|
|
121
|
+
const snippet = toText(item.snippet);
|
|
122
|
+
result += `[${index + 1}] ${title}\n ${snippet}\n\n`;
|
|
123
|
+
});
|
|
124
|
+
return organicResults.length > 0 ? result.trim() : '未找到相关结果。';
|
|
125
|
+
}
|
|
126
|
+
}
|
|
127
|
+
exports.AdvancedSearchTool = AdvancedSearchTool;
|
|
128
|
+
function toText(value) {
|
|
129
|
+
return typeof value === 'string' ? value : '';
|
|
130
|
+
}
|
|
131
|
+
function formatError(error) {
|
|
132
|
+
return error instanceof Error ? error.message : String(error);
|
|
133
|
+
}
|
|
134
|
+
//# sourceMappingURL=advancedSearchTool.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"advancedSearchTool.js","sourceRoot":"","sources":["../../../src/tools/builtin/advancedSearchTool.ts"],"names":[],"mappings":";;;AAAA,6BAAwB;AACxB,kCAA+B;AAE/B,MAAM,wBAAwB,GAAG,OAAC,CAAC,MAAM,CAAC;IACtC,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC;CAC1D,CAAC,CAAC;AA6BH,MAAa,kBAAmB,SAAQ,WAAqC;IACxD,aAAa,GAAmB,EAAE,CAAC;IACnC,MAAM,CAAuC;IAE9D,YAAY,UAAqC,EAAE;QAC/C,KAAK,CAAC;YACF,IAAI,EAAE,iBAAiB;YACvB,WAAW,EAAE,iEAAiE;YAC9E,MAAM,EAAE,wBAAwB;SACnC,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,kBAAkB,EAAE,CAAC;IAC9B,CAAC;IAEO,kBAAkB;QACtB,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC;YAC7B,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAClC,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,eAAe,CAAC,CAAC;QACzC,CAAC;aAAM,CAAC;YACJ,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,oBAAoB,CAAC,CAAC;QAC9C,CAAC;QAED,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,CAAC;YAC9B,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACnC,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,gBAAgB,CAAC,CAAC;QAC1C,CAAC;aAAM,CAAC;YACJ,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,qBAAqB,CAAC,CAAC;QAC/C,CAAC;QAED,IAAI,IAAI,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChC,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,UAAU,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACnE,CAAC;aAAM,CAAC;YACJ,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,qBAAqB,CAAC,CAAC;QAC/C,CAAC;IACL,CAAC;IAES,KAAK,CAAC,GAAG,CAAC,EAAE,KAAK,EAA2B;QAClD,IAAI,IAAI,CAAC,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAClC,OAAO,oDAAoD,CAAC;QAChE,CAAC;QAED,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,YAAY,KAAK,GAAG,CAAC,CAAC;QAE1C,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACtC,IAAI,CAAC;gBACD,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;oBACtB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;oBAClD,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;wBACpC,OAAO,sBAAsB,MAAM,EAAE,CAAC;oBAC1C,CAAC;gBACL,CAAC;qBAAM,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;oBAC9B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;oBACnD,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;wBACpC,OAAO,2BAA2B,MAAM,EAAE,CAAC;oBAC/C,CAAC;gBACL,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,GAAG,MAAM,UAAU,WAAW,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACjE,CAAC;QACL,CAAC;QAED,OAAO,6BAA6B,CAAC;IACzC,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,KAAa;QACxC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QAC1C,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,+BAA+B,EAAE;YAC1D,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACjB,OAAO,EAAE,MAAM;gBACf,KAAK;gBACL,YAAY,EAAE,OAAO;gBACrB,WAAW,EAAE,CAAC;gBACd,cAAc,EAAE,IAAI;aACvB,CAAC;SACL,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAA0B,CAAC;QAC3D,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnC,IAAI,MAAM,EAAE,CAAC;YACT,MAAM,IAAI,YAAY,MAAM,MAAM,CAAC;QACvC,CAAC;QAED,MAAM,IAAI,SAAS,CAAC;QACpB,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;QAChE,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;YACxC,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC;YAC5C,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrC,MAAM,IAAI,IAAI,KAAK,GAAG,CAAC,KAAK,KAAK,SAAS,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,SAAS,CAAC;QAC7E,CAAC,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC,IAAI,EAAE,IAAI,UAAU,CAAC;IACvC,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAAC,KAAa;QACzC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;QAC3C,IAAI,CAAC,MAAM,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;QAC3C,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,iCAAiC,CAAC,CAAC;QACvD,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QACpC,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QAC3C,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QAEpC,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC7C,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,sBAAsB,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAC7D,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAA2B,CAAC;QAC5D,MAAM,cAAc,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC;QACvF,IAAI,MAAM,GAAG,gBAAgB,CAAC;QAE9B,cAAc,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;YAC/C,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC;YAC5C,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrC,MAAM,IAAI,IAAI,KAAK,GAAG,CAAC,KAAK,KAAK,SAAS,OAAO,MAAM,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,OAAO,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC;IAClE,CAAC;CACJ;AAtID,gDAsIC;AAED,SAAS,MAAM,CAAC,KAAc;IAC1B,OAAO,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;AAClD,CAAC;AAED,SAAS,WAAW,CAAC,KAAc;IAC/B,OAAO,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAClE,CAAC"}
|
|
@@ -0,0 +1,29 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
import type { RagPipeline, RetrievedChunk, RetrieveOptions } from '../../rag';
|
|
3
|
+
import { Tool } from '../base';
|
|
4
|
+
export declare const RagSearchToolSchema: z.ZodObject<{
|
|
5
|
+
query: z.ZodString;
|
|
6
|
+
topK: z.ZodOptional<z.ZodNumber>;
|
|
7
|
+
minScore: z.ZodOptional<z.ZodNumber>;
|
|
8
|
+
tenantId: z.ZodOptional<z.ZodString>;
|
|
9
|
+
knowledgeBaseId: z.ZodOptional<z.ZodString>;
|
|
10
|
+
acl: z.ZodOptional<z.ZodArray<z.ZodString>>;
|
|
11
|
+
documentIds: z.ZodOptional<z.ZodArray<z.ZodString>>;
|
|
12
|
+
sources: z.ZodOptional<z.ZodArray<z.ZodString>>;
|
|
13
|
+
metadataFilter: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
|
|
14
|
+
}, z.core.$strip>;
|
|
15
|
+
export type RagSearchToolInput = z.infer<typeof RagSearchToolSchema>;
|
|
16
|
+
export interface RagSearchToolOptions {
|
|
17
|
+
pipeline: RagPipeline;
|
|
18
|
+
defaultRetrieveOptions?: Omit<RetrieveOptions, 'trace'>;
|
|
19
|
+
emptyMessage?: string;
|
|
20
|
+
}
|
|
21
|
+
export declare class RagSearchTool extends Tool<typeof RagSearchToolSchema, string> {
|
|
22
|
+
private readonly pipeline;
|
|
23
|
+
private readonly defaultRetrieveOptions;
|
|
24
|
+
private readonly emptyMessage;
|
|
25
|
+
constructor(options: RagSearchToolOptions);
|
|
26
|
+
protected run(input: RagSearchToolInput): Promise<string>;
|
|
27
|
+
}
|
|
28
|
+
export declare function formatRetrievedChunks(chunks: RetrievedChunk[], emptyMessage?: string): string;
|
|
29
|
+
//# sourceMappingURL=ragSearchTool.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ragSearchTool.d.ts","sourceRoot":"","sources":["../../../src/tools/builtin/ragSearchTool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EACR,WAAW,EACX,cAAc,EACd,eAAe,EAClB,MAAM,WAAW,CAAC;AACnB,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAE/B,eAAO,MAAM,mBAAmB;;;;;;;;;;iBAU9B,CAAC;AAEH,MAAM,MAAM,kBAAkB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,mBAAmB,CAAC,CAAC;AAErE,MAAM,WAAW,oBAAoB;IACjC,QAAQ,EAAE,WAAW,CAAC;IACtB,sBAAsB,CAAC,EAAE,IAAI,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC;IACxD,YAAY,CAAC,EAAE,MAAM,CAAC;CACzB;AAED,qBAAa,aAAc,SAAQ,IAAI,CAAC,OAAO,mBAAmB,EAAE,MAAM,CAAC;IACvE,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAc;IACvC,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAAiC;IACxE,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAS;gBAE1B,OAAO,EAAE,oBAAoB;cAYzB,GAAG,CAAC,KAAK,EAAE,kBAAkB,GAAG,OAAO,CAAC,MAAM,CAAC;CAQlE;AAED,wBAAgB,qBAAqB,CACjC,MAAM,EAAE,cAAc,EAAE,EACxB,YAAY,SAAgB,GAC7B,MAAM,CAQR"}
|
|
@@ -0,0 +1,91 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.RagSearchTool = exports.RagSearchToolSchema = void 0;
|
|
4
|
+
exports.formatRetrievedChunks = formatRetrievedChunks;
|
|
5
|
+
const zod_1 = require("zod");
|
|
6
|
+
const base_1 = require("../base");
|
|
7
|
+
exports.RagSearchToolSchema = zod_1.z.object({
|
|
8
|
+
query: zod_1.z.string().trim().min(1).describe('需要检索的知识库问题或关键词。'),
|
|
9
|
+
topK: zod_1.z.number().int().positive().optional().describe('最多返回的证据片段数量。'),
|
|
10
|
+
minScore: zod_1.z.number().optional().describe('最低相关性分数。'),
|
|
11
|
+
tenantId: zod_1.z.string().trim().min(1).optional().describe('租户 ID。'),
|
|
12
|
+
knowledgeBaseId: zod_1.z.string().trim().min(1).optional().describe('知识库 ID。'),
|
|
13
|
+
acl: zod_1.z.array(zod_1.z.string().trim().min(1)).optional().describe('调用者权限标签。'),
|
|
14
|
+
documentIds: zod_1.z.array(zod_1.z.string().trim().min(1)).optional().describe('限定检索的文档 ID。'),
|
|
15
|
+
sources: zod_1.z.array(zod_1.z.string().trim().min(1)).optional().describe('限定检索的来源路径或来源标识。'),
|
|
16
|
+
metadataFilter: zod_1.z.record(zod_1.z.string(), zod_1.z.unknown()).optional().describe('元数据精确过滤条件。'),
|
|
17
|
+
});
|
|
18
|
+
class RagSearchTool extends base_1.Tool {
|
|
19
|
+
pipeline;
|
|
20
|
+
defaultRetrieveOptions;
|
|
21
|
+
emptyMessage;
|
|
22
|
+
constructor(options) {
|
|
23
|
+
super({
|
|
24
|
+
name: 'rag_search',
|
|
25
|
+
description: '检索知识库证据,返回可引用的资料片段。只返回证据,不直接生成最终答案。',
|
|
26
|
+
schema: exports.RagSearchToolSchema,
|
|
27
|
+
});
|
|
28
|
+
this.pipeline = options.pipeline;
|
|
29
|
+
this.defaultRetrieveOptions = options.defaultRetrieveOptions ?? {};
|
|
30
|
+
this.emptyMessage = options.emptyMessage ?? '未找到相关知识库证据。';
|
|
31
|
+
}
|
|
32
|
+
async run(input) {
|
|
33
|
+
const retrieved = await this.pipeline.retrieve(input.query, buildRetrieveOptions(this.defaultRetrieveOptions, input));
|
|
34
|
+
return formatRetrievedChunks(retrieved, this.emptyMessage);
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
exports.RagSearchTool = RagSearchTool;
|
|
38
|
+
function formatRetrievedChunks(chunks, emptyMessage = '未找到相关知识库证据。') {
|
|
39
|
+
if (chunks.length === 0) {
|
|
40
|
+
return emptyMessage;
|
|
41
|
+
}
|
|
42
|
+
return chunks
|
|
43
|
+
.map((item, index) => formatRetrievedChunk(item, index + 1))
|
|
44
|
+
.join('\n\n');
|
|
45
|
+
}
|
|
46
|
+
function formatRetrievedChunk(item, index) {
|
|
47
|
+
const attributes = [
|
|
48
|
+
`source=${formatAttribute(item.chunk.metadata.source)}`,
|
|
49
|
+
`score=${formatScore(item.score)}`,
|
|
50
|
+
];
|
|
51
|
+
if (item.chunk.metadata.title !== undefined) {
|
|
52
|
+
attributes.push(`title=${formatAttribute(item.chunk.metadata.title)}`);
|
|
53
|
+
}
|
|
54
|
+
if (item.chunk.metadata.headingPath !== undefined
|
|
55
|
+
&& item.chunk.metadata.headingPath.length > 0) {
|
|
56
|
+
attributes.push(`section=${formatAttribute(item.chunk.metadata.headingPath.join(' > '))}`);
|
|
57
|
+
}
|
|
58
|
+
if (item.chunk.metadata.page !== undefined) {
|
|
59
|
+
attributes.push(`page=${item.chunk.metadata.page}`);
|
|
60
|
+
}
|
|
61
|
+
return `[${index}] ${attributes.join(' ')}\n${item.chunk.content.trim()}`;
|
|
62
|
+
}
|
|
63
|
+
function buildRetrieveOptions(defaults, input) {
|
|
64
|
+
const options = {
|
|
65
|
+
...defaults,
|
|
66
|
+
};
|
|
67
|
+
assignIfDefined(options, 'topK', input.topK);
|
|
68
|
+
assignIfDefined(options, 'minScore', input.minScore);
|
|
69
|
+
assignIfDefined(options, 'tenantId', input.tenantId);
|
|
70
|
+
assignIfDefined(options, 'knowledgeBaseId', input.knowledgeBaseId);
|
|
71
|
+
assignIfDefined(options, 'acl', input.acl);
|
|
72
|
+
assignIfDefined(options, 'documentIds', input.documentIds);
|
|
73
|
+
assignIfDefined(options, 'sources', input.sources);
|
|
74
|
+
assignIfDefined(options, 'metadataFilter', input.metadataFilter);
|
|
75
|
+
return options;
|
|
76
|
+
}
|
|
77
|
+
function formatAttribute(value) {
|
|
78
|
+
if (/^[A-Za-z0-9_./:-]+$/.test(value)) {
|
|
79
|
+
return value;
|
|
80
|
+
}
|
|
81
|
+
return JSON.stringify(value);
|
|
82
|
+
}
|
|
83
|
+
function formatScore(score) {
|
|
84
|
+
return Number.isFinite(score) ? score.toFixed(4) : String(score);
|
|
85
|
+
}
|
|
86
|
+
function assignIfDefined(target, key, value) {
|
|
87
|
+
if (value !== undefined) {
|
|
88
|
+
target[key] = value;
|
|
89
|
+
}
|
|
90
|
+
}
|
|
91
|
+
//# sourceMappingURL=ragSearchTool.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ragSearchTool.js","sourceRoot":"","sources":["../../../src/tools/builtin/ragSearchTool.ts"],"names":[],"mappings":";;;AAuDA,sDAWC;AAlED,6BAAwB;AAMxB,kCAA+B;AAElB,QAAA,mBAAmB,GAAG,OAAC,CAAC,MAAM,CAAC;IACxC,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,iBAAiB,CAAC;IAC3D,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC;IACrE,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC;IACpD,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC;IAChE,eAAe,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;IACxE,GAAG,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC;IACtE,WAAW,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC;IACjF,OAAO,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC;IACjF,cAAc,EAAE,OAAC,CAAC,MAAM,CAAC,OAAC,CAAC,MAAM,EAAE,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC;CACtF,CAAC,CAAC;AAUH,MAAa,aAAc,SAAQ,WAAwC;IACtD,QAAQ,CAAc;IACtB,sBAAsB,CAAiC;IACvD,YAAY,CAAS;IAEtC,YAAY,OAA6B;QACrC,KAAK,CAAC;YACF,IAAI,EAAE,YAAY;YAClB,WAAW,EAAE,qCAAqC;YAClD,MAAM,EAAE,2BAAmB;SAC9B,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QACjC,IAAI,CAAC,sBAAsB,GAAG,OAAO,CAAC,sBAAsB,IAAI,EAAE,CAAC;QACnE,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,IAAI,aAAa,CAAC;IAC9D,CAAC;IAES,KAAK,CAAC,GAAG,CAAC,KAAyB;QACzC,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAC1C,KAAK,CAAC,KAAK,EACX,oBAAoB,CAAC,IAAI,CAAC,sBAAsB,EAAE,KAAK,CAAC,CAC3D,CAAC;QAEF,OAAO,qBAAqB,CAAC,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;IAC/D,CAAC;CACJ;AAzBD,sCAyBC;AAED,SAAgB,qBAAqB,CACjC,MAAwB,EACxB,YAAY,GAAG,aAAa;IAE5B,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACtB,OAAO,YAAY,CAAC;IACxB,CAAC;IAED,OAAO,MAAM;SACR,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,oBAAoB,CAAC,IAAI,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;SAC3D,IAAI,CAAC,MAAM,CAAC,CAAC;AACtB,CAAC;AAED,SAAS,oBAAoB,CAAC,IAAoB,EAAE,KAAa;IAC7D,MAAM,UAAU,GAAG;QACf,UAAU,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE;QACvD,SAAS,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE;KACrC,CAAC;IAEF,IAAI,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QAC1C,UAAU,CAAC,IAAI,CAAC,SAAS,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAC3E,CAAC;IACD,IACI,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,WAAW,KAAK,SAAS;WAC1C,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAC/C,CAAC;QACC,UAAU,CAAC,IAAI,CAAC,WAAW,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;IAC/F,CAAC;IACD,IAAI,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;QACzC,UAAU,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;IACxD,CAAC;IAED,OAAO,IAAI,KAAK,KAAK,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;AAC9E,CAAC;AAED,SAAS,oBAAoB,CACzB,QAAwC,EACxC,KAAyB;IAEzB,MAAM,OAAO,GAAoB;QAC7B,GAAG,QAAQ;KACd,CAAC;IAEF,eAAe,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;IAC7C,eAAe,CAAC,OAAO,EAAE,UAAU,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;IACrD,eAAe,CAAC,OAAO,EAAE,UAAU,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;IACrD,eAAe,CAAC,OAAO,EAAE,iBAAiB,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC;IACnE,eAAe,CAAC,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;IAC3C,eAAe,CAAC,OAAO,EAAE,aAAa,EAAE,KAAK,CAAC,WAAW,CAAC,CAAC;IAC3D,eAAe,CAAC,OAAO,EAAE,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;IACnD,eAAe,CAAC,OAAO,EAAE,gBAAgB,EAAE,KAAK,CAAC,cAAc,CAAC,CAAC;IAEjE,OAAO,OAAO,CAAC;AACnB,CAAC;AAED,SAAS,eAAe,CAAC,KAAa;IAClC,IAAI,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QACpC,OAAO,KAAK,CAAC;IACjB,CAAC;IAED,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;AACjC,CAAC;AAED,SAAS,WAAW,CAAC,KAAa;IAC9B,OAAO,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AACrE,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,10 @@
|
|
|
1
|
+
import type { ChatToolCall } from '../llm/types';
|
|
2
|
+
import type { TraceContext } from '../observability';
|
|
3
|
+
import type { ToolRegistry } from './registry';
|
|
4
|
+
export declare class ToolExecutor {
|
|
5
|
+
private readonly registry;
|
|
6
|
+
private readonly trace?;
|
|
7
|
+
constructor(registry: ToolRegistry, trace?: TraceContext | undefined);
|
|
8
|
+
execute(toolCall: ChatToolCall): Promise<string>;
|
|
9
|
+
}
|
|
10
|
+
//# sourceMappingURL=executor.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"executor.d.ts","sourceRoot":"","sources":["../../src/tools/executor.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAKjD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAErD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE/C,qBAAa,YAAY;IAEjB,OAAO,CAAC,QAAQ,CAAC,QAAQ;IACzB,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC;gBADN,QAAQ,EAAE,YAAY,EACtB,KAAK,CAAC,EAAE,YAAY,YAAA;IAGnC,OAAO,CAAC,QAAQ,EAAE,YAAY,GAAG,OAAO,CAAC,MAAM,CAAC;CAuEzD"}
|
|
@@ -0,0 +1,86 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.ToolExecutor = void 0;
|
|
4
|
+
const observability_1 = require("../observability");
|
|
5
|
+
const base_1 = require("./base");
|
|
6
|
+
class ToolExecutor {
|
|
7
|
+
registry;
|
|
8
|
+
trace;
|
|
9
|
+
constructor(registry, trace) {
|
|
10
|
+
this.registry = registry;
|
|
11
|
+
this.trace = trace;
|
|
12
|
+
}
|
|
13
|
+
async execute(toolCall) {
|
|
14
|
+
const startedAt = Date.now();
|
|
15
|
+
await (0, observability_1.emitRuntimeEvent)(this.trace?.observer, {
|
|
16
|
+
type: 'tool:start',
|
|
17
|
+
runId: this.trace?.runId ?? 'standalone',
|
|
18
|
+
timestamp: new Date(),
|
|
19
|
+
agentName: this.trace?.agentName,
|
|
20
|
+
sessionId: this.trace?.sessionId,
|
|
21
|
+
toolName: toolCall.name,
|
|
22
|
+
toolCallId: toolCall.id,
|
|
23
|
+
});
|
|
24
|
+
const tool = this.registry.get(toolCall.name);
|
|
25
|
+
if (!tool) {
|
|
26
|
+
await (0, observability_1.emitRuntimeEvent)(this.trace?.observer, {
|
|
27
|
+
type: 'tool:error',
|
|
28
|
+
runId: this.trace?.runId ?? 'standalone',
|
|
29
|
+
timestamp: new Date(),
|
|
30
|
+
durationMs: Date.now() - startedAt,
|
|
31
|
+
agentName: this.trace?.agentName,
|
|
32
|
+
sessionId: this.trace?.sessionId,
|
|
33
|
+
toolName: toolCall.name,
|
|
34
|
+
toolCallId: toolCall.id,
|
|
35
|
+
error: {
|
|
36
|
+
name: 'ToolNotFoundError',
|
|
37
|
+
message: `Tool "${toolCall.name}" is not registered.`,
|
|
38
|
+
},
|
|
39
|
+
});
|
|
40
|
+
return `Tool "${toolCall.name}" is not registered.`;
|
|
41
|
+
}
|
|
42
|
+
try {
|
|
43
|
+
const result = await tool.execute(toolCall.arguments);
|
|
44
|
+
const output = String(result);
|
|
45
|
+
await (0, observability_1.emitRuntimeEvent)(this.trace?.observer, {
|
|
46
|
+
type: 'tool:end',
|
|
47
|
+
runId: this.trace?.runId ?? 'standalone',
|
|
48
|
+
timestamp: new Date(),
|
|
49
|
+
durationMs: Date.now() - startedAt,
|
|
50
|
+
agentName: this.trace?.agentName,
|
|
51
|
+
sessionId: this.trace?.sessionId,
|
|
52
|
+
toolName: toolCall.name,
|
|
53
|
+
toolCallId: toolCall.id,
|
|
54
|
+
outputLength: output.length,
|
|
55
|
+
});
|
|
56
|
+
return output;
|
|
57
|
+
}
|
|
58
|
+
catch (error) {
|
|
59
|
+
await (0, observability_1.emitRuntimeEvent)(this.trace?.observer, {
|
|
60
|
+
type: 'tool:error',
|
|
61
|
+
runId: this.trace?.runId ?? 'standalone',
|
|
62
|
+
timestamp: new Date(),
|
|
63
|
+
durationMs: Date.now() - startedAt,
|
|
64
|
+
agentName: this.trace?.agentName,
|
|
65
|
+
sessionId: this.trace?.sessionId,
|
|
66
|
+
toolName: toolCall.name,
|
|
67
|
+
toolCallId: toolCall.id,
|
|
68
|
+
error: (0, observability_1.toObservedError)(error),
|
|
69
|
+
});
|
|
70
|
+
if (error instanceof base_1.ToolValidationError) {
|
|
71
|
+
return formatValidationError(error);
|
|
72
|
+
}
|
|
73
|
+
const message = error instanceof Error ? error.message : String(error);
|
|
74
|
+
return `Tool "${toolCall.name}" failed: ${message}`;
|
|
75
|
+
}
|
|
76
|
+
}
|
|
77
|
+
}
|
|
78
|
+
exports.ToolExecutor = ToolExecutor;
|
|
79
|
+
function formatValidationError(error) {
|
|
80
|
+
const details = error.issues.map((issue) => {
|
|
81
|
+
const path = issue.path.length > 0 ? issue.path.join('.') : '<root>';
|
|
82
|
+
return `${path}: ${issue.message}`;
|
|
83
|
+
});
|
|
84
|
+
return `Tool "${error.toolName}" received invalid parameters: ${details.join('; ')}`;
|
|
85
|
+
}
|
|
86
|
+
//# sourceMappingURL=executor.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"executor.js","sourceRoot":"","sources":["../../src/tools/executor.ts"],"names":[],"mappings":";;;AACA,oDAG0B;AAE1B,iCAA6C;AAG7C,MAAa,YAAY;IAEA;IACA;IAFrB,YACqB,QAAsB,EACtB,KAAoB;QADpB,aAAQ,GAAR,QAAQ,CAAc;QACtB,UAAK,GAAL,KAAK,CAAe;IACtC,CAAC;IAEJ,KAAK,CAAC,OAAO,CAAC,QAAsB;QAChC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,MAAM,IAAA,gCAAgB,EAAC,IAAI,CAAC,KAAK,EAAE,QAAQ,EAAE;YACzC,IAAI,EAAE,YAAY;YAClB,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,KAAK,IAAI,YAAY;YACxC,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,SAAS,EAAE,IAAI,CAAC,KAAK,EAAE,SAAS;YAChC,SAAS,EAAE,IAAI,CAAC,KAAK,EAAE,SAAS;YAChC,QAAQ,EAAE,QAAQ,CAAC,IAAI;YACvB,UAAU,EAAE,QAAQ,CAAC,EAAE;SAC1B,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC9C,IAAI,CAAC,IAAI,EAAE,CAAC;YACR,MAAM,IAAA,gCAAgB,EAAC,IAAI,CAAC,KAAK,EAAE,QAAQ,EAAE;gBACzC,IAAI,EAAE,YAAY;gBAClB,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,KAAK,IAAI,YAAY;gBACxC,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;gBAClC,SAAS,EAAE,IAAI,CAAC,KAAK,EAAE,SAAS;gBAChC,SAAS,EAAE,IAAI,CAAC,KAAK,EAAE,SAAS;gBAChC,QAAQ,EAAE,QAAQ,CAAC,IAAI;gBACvB,UAAU,EAAE,QAAQ,CAAC,EAAE;gBACvB,KAAK,EAAE;oBACH,IAAI,EAAE,mBAAmB;oBACzB,OAAO,EAAE,SAAS,QAAQ,CAAC,IAAI,sBAAsB;iBACxD;aACJ,CAAC,CAAC;YAEH,OAAO,SAAS,QAAQ,CAAC,IAAI,sBAAsB,CAAC;QACxD,CAAC;QAED,IAAI,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;YACtD,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;YAE9B,MAAM,IAAA,gCAAgB,EAAC,IAAI,CAAC,KAAK,EAAE,QAAQ,EAAE;gBACzC,IAAI,EAAE,UAAU;gBAChB,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,KAAK,IAAI,YAAY;gBACxC,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;gBAClC,SAAS,EAAE,IAAI,CAAC,KAAK,EAAE,SAAS;gBAChC,SAAS,EAAE,IAAI,CAAC,KAAK,EAAE,SAAS;gBAChC,QAAQ,EAAE,QAAQ,CAAC,IAAI;gBACvB,UAAU,EAAE,QAAQ,CAAC,EAAE;gBACvB,YAAY,EAAE,MAAM,CAAC,MAAM;aAC9B,CAAC,CAAC;YAEH,OAAO,MAAM,CAAC;QAClB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,MAAM,IAAA,gCAAgB,EAAC,IAAI,CAAC,KAAK,EAAE,QAAQ,EAAE;gBACzC,IAAI,EAAE,YAAY;gBAClB,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,KAAK,IAAI,YAAY;gBACxC,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;gBAClC,SAAS,EAAE,IAAI,CAAC,KAAK,EAAE,SAAS;gBAChC,SAAS,EAAE,IAAI,CAAC,KAAK,EAAE,SAAS;gBAChC,QAAQ,EAAE,QAAQ,CAAC,IAAI;gBACvB,UAAU,EAAE,QAAQ,CAAC,EAAE;gBACvB,KAAK,EAAE,IAAA,+BAAe,EAAC,KAAK,CAAC;aAChC,CAAC,CAAC;YAEH,IAAI,KAAK,YAAY,0BAAmB,EAAE,CAAC;gBACvC,OAAO,qBAAqB,CAAC,KAAK,CAAC,CAAC;YACxC,CAAC;YAED,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACvE,OAAO,SAAS,QAAQ,CAAC,IAAI,aAAa,OAAO,EAAE,CAAC;QACxD,CAAC;IACL,CAAC;CACJ;AA7ED,oCA6EC;AAED,SAAS,qBAAqB,CAAC,KAA0B;IACrD,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;QACvC,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;QACrE,OAAO,GAAG,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,OAAO,SAAS,KAAK,CAAC,QAAQ,kCAAkC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;AACzF,CAAC"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
export { Tool, ToolValidationError } from './base';
|
|
2
|
+
export { ToolExecutor } from './executor';
|
|
3
|
+
export { ToolRegistry } from './registry';
|
|
4
|
+
export { AdvancedSearchTool } from './builtin/advancedSearchTool';
|
|
5
|
+
export { RagSearchTool, formatRetrievedChunks } from './builtin/ragSearchTool';
|
|
6
|
+
export type { AdvancedSearchToolLogger, AdvancedSearchToolOptions, } from './builtin/advancedSearchTool';
|
|
7
|
+
export type { RagSearchToolInput, RagSearchToolOptions, } from './builtin/ragSearchTool';
|
|
8
|
+
export type { ToolInput, ToolInputSchema, ToolOptions } from './base';
|
|
9
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,mBAAmB,EAAE,MAAM,QAAQ,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAC1C,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAC1C,OAAO,EAAE,kBAAkB,EAAE,MAAM,8BAA8B,CAAC;AAClE,OAAO,EAAE,aAAa,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAC/E,YAAY,EACR,wBAAwB,EACxB,yBAAyB,GAC5B,MAAM,8BAA8B,CAAC;AACtC,YAAY,EACR,kBAAkB,EAClB,oBAAoB,GACvB,MAAM,yBAAyB,CAAC;AACjC,YAAY,EAAE,SAAS,EAAE,eAAe,EAAE,WAAW,EAAE,MAAM,QAAQ,CAAC"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.formatRetrievedChunks = exports.RagSearchTool = exports.AdvancedSearchTool = exports.ToolRegistry = exports.ToolExecutor = exports.ToolValidationError = exports.Tool = void 0;
|
|
4
|
+
var base_1 = require("./base");
|
|
5
|
+
Object.defineProperty(exports, "Tool", { enumerable: true, get: function () { return base_1.Tool; } });
|
|
6
|
+
Object.defineProperty(exports, "ToolValidationError", { enumerable: true, get: function () { return base_1.ToolValidationError; } });
|
|
7
|
+
var executor_1 = require("./executor");
|
|
8
|
+
Object.defineProperty(exports, "ToolExecutor", { enumerable: true, get: function () { return executor_1.ToolExecutor; } });
|
|
9
|
+
var registry_1 = require("./registry");
|
|
10
|
+
Object.defineProperty(exports, "ToolRegistry", { enumerable: true, get: function () { return registry_1.ToolRegistry; } });
|
|
11
|
+
var advancedSearchTool_1 = require("./builtin/advancedSearchTool");
|
|
12
|
+
Object.defineProperty(exports, "AdvancedSearchTool", { enumerable: true, get: function () { return advancedSearchTool_1.AdvancedSearchTool; } });
|
|
13
|
+
var ragSearchTool_1 = require("./builtin/ragSearchTool");
|
|
14
|
+
Object.defineProperty(exports, "RagSearchTool", { enumerable: true, get: function () { return ragSearchTool_1.RagSearchTool; } });
|
|
15
|
+
Object.defineProperty(exports, "formatRetrievedChunks", { enumerable: true, get: function () { return ragSearchTool_1.formatRetrievedChunks; } });
|
|
16
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":";;;AAAA,+BAAmD;AAA1C,4FAAA,IAAI,OAAA;AAAE,2GAAA,mBAAmB,OAAA;AAClC,uCAA0C;AAAjC,wGAAA,YAAY,OAAA;AACrB,uCAA0C;AAAjC,wGAAA,YAAY,OAAA;AACrB,mEAAkE;AAAzD,wHAAA,kBAAkB,OAAA;AAC3B,yDAA+E;AAAtE,8GAAA,aAAa,OAAA;AAAE,sHAAA,qBAAqB,OAAA"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
import type { ChatToolSchema } from '../llm/types';
|
|
2
|
+
import { Tool } from './base';
|
|
3
|
+
export declare class ToolRegistry {
|
|
4
|
+
private readonly tools;
|
|
5
|
+
constructor(initialTools?: Tool[]);
|
|
6
|
+
register(tool: Tool): void;
|
|
7
|
+
get(name: string): Tool | undefined;
|
|
8
|
+
has(name: string): boolean;
|
|
9
|
+
list(): Tool[];
|
|
10
|
+
toOpenAISchemas(): ChatToolSchema[];
|
|
11
|
+
/**
|
|
12
|
+
* 这个描述主要用于非 function-calling 模型,或调试 prompt 时查看工具清单。
|
|
13
|
+
*/
|
|
14
|
+
getToolsDescription(): string;
|
|
15
|
+
}
|
|
16
|
+
//# sourceMappingURL=registry.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"registry.d.ts","sourceRoot":"","sources":["../../src/tools/registry.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AACnD,OAAO,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AAE9B,qBAAa,YAAY;IACrB,OAAO,CAAC,QAAQ,CAAC,KAAK,CAA2B;gBAErC,YAAY,GAAE,IAAI,EAAO;IAMrC,QAAQ,CAAC,IAAI,EAAE,IAAI,GAAG,IAAI;IAI1B,GAAG,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,GAAG,SAAS;IAInC,GAAG,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAI1B,IAAI,IAAI,IAAI,EAAE;IAId,eAAe,IAAI,cAAc,EAAE;IAInC;;OAEG;IACH,mBAAmB,IAAI,MAAM;CAOhC"}
|