@angri450/zvec-grep 0.2.2
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/LICENSE +201 -0
- package/README.md +319 -0
- package/README_CN.md +297 -0
- package/dist/authorization/index.d.ts +7 -0
- package/dist/authorization/index.js +7 -0
- package/dist/authorization/index.js.map +1 -0
- package/dist/authorization/manager.d.ts +8 -0
- package/dist/authorization/manager.js +19 -0
- package/dist/authorization/manager.js.map +1 -0
- package/dist/authorization/operation.d.ts +7 -0
- package/dist/authorization/operation.js +50 -0
- package/dist/authorization/operation.js.map +1 -0
- package/dist/authorization/planner.d.ts +21 -0
- package/dist/authorization/planner.js +97 -0
- package/dist/authorization/planner.js.map +1 -0
- package/dist/authorization/prompt.d.ts +12 -0
- package/dist/authorization/prompt.js +63 -0
- package/dist/authorization/prompt.js.map +1 -0
- package/dist/authorization/store.d.ts +20 -0
- package/dist/authorization/store.js +251 -0
- package/dist/authorization/store.js.map +1 -0
- package/dist/authorization/target.d.ts +15 -0
- package/dist/authorization/target.js +49 -0
- package/dist/authorization/target.js.map +1 -0
- package/dist/authorization/types.d.ts +61 -0
- package/dist/authorization/types.js +2 -0
- package/dist/authorization/types.js.map +1 -0
- package/dist/cli/args.d.ts +6 -0
- package/dist/cli/args.js +1255 -0
- package/dist/cli/args.js.map +1 -0
- package/dist/cli/auth.d.ts +21 -0
- package/dist/cli/auth.js +209 -0
- package/dist/cli/auth.js.map +1 -0
- package/dist/cli/commands.d.ts +4 -0
- package/dist/cli/commands.js +801 -0
- package/dist/cli/commands.js.map +1 -0
- package/dist/cli/errors.d.ts +9 -0
- package/dist/cli/errors.js +131 -0
- package/dist/cli/errors.js.map +1 -0
- package/dist/cli/format/context.d.ts +9 -0
- package/dist/cli/format/context.js +786 -0
- package/dist/cli/format/context.js.map +1 -0
- package/dist/cli/format/debug.d.ts +4 -0
- package/dist/cli/format/debug.js +60 -0
- package/dist/cli/format/debug.js.map +1 -0
- package/dist/cli/format/highlight.d.ts +3 -0
- package/dist/cli/format/highlight.js +106 -0
- package/dist/cli/format/highlight.js.map +1 -0
- package/dist/cli/format/progress.d.ts +13 -0
- package/dist/cli/format/progress.js +290 -0
- package/dist/cli/format/progress.js.map +1 -0
- package/dist/cli/format/range.d.ts +3 -0
- package/dist/cli/format/range.js +24 -0
- package/dist/cli/format/range.js.map +1 -0
- package/dist/cli/format/status.d.ts +83 -0
- package/dist/cli/format/status.js +530 -0
- package/dist/cli/format/status.js.map +1 -0
- package/dist/cli/help.d.ts +1 -0
- package/dist/cli/help.js +546 -0
- package/dist/cli/help.js.map +1 -0
- package/dist/cli/index.d.ts +2 -0
- package/dist/cli/index.js +40 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/cli/install.d.ts +4 -0
- package/dist/cli/install.js +1735 -0
- package/dist/cli/install.js.map +1 -0
- package/dist/cli/managed-rg.d.ts +10 -0
- package/dist/cli/managed-rg.js +337 -0
- package/dist/cli/managed-rg.js.map +1 -0
- package/dist/cli/server-search.d.ts +3 -0
- package/dist/cli/server-search.js +12 -0
- package/dist/cli/server-search.js.map +1 -0
- package/dist/cli/types.d.ts +87 -0
- package/dist/cli/types.js +10 -0
- package/dist/cli/types.js.map +1 -0
- package/dist/cli/version.d.ts +1 -0
- package/dist/cli/version.js +13 -0
- package/dist/cli/version.js.map +1 -0
- package/dist/client/daemon-client.d.ts +31 -0
- package/dist/client/daemon-client.js +203 -0
- package/dist/client/daemon-client.js.map +1 -0
- package/dist/client/mode-router.d.ts +9 -0
- package/dist/client/mode-router.js +28 -0
- package/dist/client/mode-router.js.map +1 -0
- package/dist/client/search-policy.d.ts +11 -0
- package/dist/client/search-policy.js +15 -0
- package/dist/client/search-policy.js.map +1 -0
- package/dist/daemon/backend.d.ts +78 -0
- package/dist/daemon/backend.js +1118 -0
- package/dist/daemon/backend.js.map +1 -0
- package/dist/daemon/change-set.d.ts +30 -0
- package/dist/daemon/change-set.js +163 -0
- package/dist/daemon/change-set.js.map +1 -0
- package/dist/daemon/config.d.ts +27 -0
- package/dist/daemon/config.js +107 -0
- package/dist/daemon/config.js.map +1 -0
- package/dist/daemon/errors.d.ts +5 -0
- package/dist/daemon/errors.js +11 -0
- package/dist/daemon/errors.js.map +1 -0
- package/dist/daemon/http-server.d.ts +30 -0
- package/dist/daemon/http-server.js +307 -0
- package/dist/daemon/http-server.js.map +1 -0
- package/dist/daemon/index-coordinator.d.ts +22 -0
- package/dist/daemon/index-coordinator.js +55 -0
- package/dist/daemon/index-coordinator.js.map +1 -0
- package/dist/daemon/job-scheduler.d.ts +85 -0
- package/dist/daemon/job-scheduler.js +492 -0
- package/dist/daemon/job-scheduler.js.map +1 -0
- package/dist/daemon/logger.d.ts +10 -0
- package/dist/daemon/logger.js +101 -0
- package/dist/daemon/logger.js.map +1 -0
- package/dist/daemon/model-pool.d.ts +48 -0
- package/dist/daemon/model-pool.js +190 -0
- package/dist/daemon/model-pool.js.map +1 -0
- package/dist/daemon/root-lease.d.ts +14 -0
- package/dist/daemon/root-lease.js +197 -0
- package/dist/daemon/root-lease.js.map +1 -0
- package/dist/daemon/root-runtime.d.ts +80 -0
- package/dist/daemon/root-runtime.js +340 -0
- package/dist/daemon/root-runtime.js.map +1 -0
- package/dist/daemon/runtime-manager.d.ts +47 -0
- package/dist/daemon/runtime-manager.js +252 -0
- package/dist/daemon/runtime-manager.js.map +1 -0
- package/dist/daemon/runtime.d.ts +12 -0
- package/dist/daemon/runtime.js +105 -0
- package/dist/daemon/runtime.js.map +1 -0
- package/dist/daemon/server-controller.d.ts +38 -0
- package/dist/daemon/server-controller.js +327 -0
- package/dist/daemon/server-controller.js.map +1 -0
- package/dist/daemon/watch-manager.d.ts +62 -0
- package/dist/daemon/watch-manager.js +415 -0
- package/dist/daemon/watch-manager.js.map +1 -0
- package/dist/daemon/workspace-read-session-cache.d.ts +31 -0
- package/dist/daemon/workspace-read-session-cache.js +115 -0
- package/dist/daemon/workspace-read-session-cache.js.map +1 -0
- package/dist/engine/code-formats.d.ts +3 -0
- package/dist/engine/code-formats.js +14 -0
- package/dist/engine/code-formats.js.map +1 -0
- package/dist/engine/config.d.ts +57 -0
- package/dist/engine/config.js +351 -0
- package/dist/engine/config.js.map +1 -0
- package/dist/engine/errors.d.ts +20 -0
- package/dist/engine/errors.js +51 -0
- package/dist/engine/errors.js.map +1 -0
- package/dist/engine/extraction/code/adapter.d.ts +19 -0
- package/dist/engine/extraction/code/adapter.js +24 -0
- package/dist/engine/extraction/code/adapter.js.map +1 -0
- package/dist/engine/extraction/code/extractor.d.ts +13 -0
- package/dist/engine/extraction/code/extractor.js +696 -0
- package/dist/engine/extraction/code/extractor.js.map +1 -0
- package/dist/engine/extraction/code/families/c-family.d.ts +2 -0
- package/dist/engine/extraction/code/families/c-family.js +122 -0
- package/dist/engine/extraction/code/families/c-family.js.map +1 -0
- package/dist/engine/extraction/code/families/js-ts.d.ts +12 -0
- package/dist/engine/extraction/code/families/js-ts.js +118 -0
- package/dist/engine/extraction/code/families/js-ts.js.map +1 -0
- package/dist/engine/extraction/code/families/metadata.d.ts +8 -0
- package/dist/engine/extraction/code/families/metadata.js +86 -0
- package/dist/engine/extraction/code/families/metadata.js.map +1 -0
- package/dist/engine/extraction/code/families/name-field.d.ts +3 -0
- package/dist/engine/extraction/code/families/name-field.js +22 -0
- package/dist/engine/extraction/code/families/name-field.js.map +1 -0
- package/dist/engine/extraction/code/languages/c.d.ts +1 -0
- package/dist/engine/extraction/code/languages/c.js +12 -0
- package/dist/engine/extraction/code/languages/c.js.map +1 -0
- package/dist/engine/extraction/code/languages/cpp.d.ts +1 -0
- package/dist/engine/extraction/code/languages/cpp.js +18 -0
- package/dist/engine/extraction/code/languages/cpp.js.map +1 -0
- package/dist/engine/extraction/code/languages/go.d.ts +2 -0
- package/dist/engine/extraction/code/languages/go.js +70 -0
- package/dist/engine/extraction/code/languages/go.js.map +1 -0
- package/dist/engine/extraction/code/languages/java.d.ts +1 -0
- package/dist/engine/extraction/code/languages/java.js +17 -0
- package/dist/engine/extraction/code/languages/java.js.map +1 -0
- package/dist/engine/extraction/code/languages/javascript.d.ts +2 -0
- package/dist/engine/extraction/code/languages/javascript.js +23 -0
- package/dist/engine/extraction/code/languages/javascript.js.map +1 -0
- package/dist/engine/extraction/code/languages/python.d.ts +2 -0
- package/dist/engine/extraction/code/languages/python.js +52 -0
- package/dist/engine/extraction/code/languages/python.js.map +1 -0
- package/dist/engine/extraction/code/languages/rust.d.ts +2 -0
- package/dist/engine/extraction/code/languages/rust.js +25 -0
- package/dist/engine/extraction/code/languages/rust.js.map +1 -0
- package/dist/engine/extraction/code/languages/typescript.d.ts +2 -0
- package/dist/engine/extraction/code/languages/typescript.js +37 -0
- package/dist/engine/extraction/code/languages/typescript.js.map +1 -0
- package/dist/engine/extraction/code/tree-sitter/grammar.d.ts +4 -0
- package/dist/engine/extraction/code/tree-sitter/grammar.js +49 -0
- package/dist/engine/extraction/code/tree-sitter/grammar.js.map +1 -0
- package/dist/engine/extraction/code/tree-sitter/nodes.d.ts +3 -0
- package/dist/engine/extraction/code/tree-sitter/nodes.js +27 -0
- package/dist/engine/extraction/code/tree-sitter/nodes.js.map +1 -0
- package/dist/engine/extraction/code/tree-sitter/parser.d.ts +2 -0
- package/dist/engine/extraction/code/tree-sitter/parser.js +31 -0
- package/dist/engine/extraction/code/tree-sitter/parser.js.map +1 -0
- package/dist/engine/extraction/ids.d.ts +1 -0
- package/dist/engine/extraction/ids.js +5 -0
- package/dist/engine/extraction/ids.js.map +1 -0
- package/dist/engine/extraction/image/extractor.d.ts +5 -0
- package/dist/engine/extraction/image/extractor.js +35 -0
- package/dist/engine/extraction/image/extractor.js.map +1 -0
- package/dist/engine/extraction/index.d.ts +4 -0
- package/dist/engine/extraction/index.js +3 -0
- package/dist/engine/extraction/index.js.map +1 -0
- package/dist/engine/extraction/markdown/extractor.d.ts +7 -0
- package/dist/engine/extraction/markdown/extractor.js +362 -0
- package/dist/engine/extraction/markdown/extractor.js.map +1 -0
- package/dist/engine/extraction/runtime.d.ts +9 -0
- package/dist/engine/extraction/runtime.js +37 -0
- package/dist/engine/extraction/runtime.js.map +1 -0
- package/dist/engine/extraction/source.d.ts +18 -0
- package/dist/engine/extraction/source.js +22 -0
- package/dist/engine/extraction/source.js.map +1 -0
- package/dist/engine/extraction/text/extractor.d.ts +7 -0
- package/dist/engine/extraction/text/extractor.js +168 -0
- package/dist/engine/extraction/text/extractor.js.map +1 -0
- package/dist/engine/extraction/types.d.ts +4 -0
- package/dist/engine/extraction/types.js +2 -0
- package/dist/engine/extraction/types.js.map +1 -0
- package/dist/engine/extraction/vector-content.d.ts +5 -0
- package/dist/engine/extraction/vector-content.js +83 -0
- package/dist/engine/extraction/vector-content.js.map +1 -0
- package/dist/engine/file-size-policy.d.ts +6 -0
- package/dist/engine/file-size-policy.js +20 -0
- package/dist/engine/file-size-policy.js.map +1 -0
- package/dist/engine/file-type.d.ts +15 -0
- package/dist/engine/file-type.js +152 -0
- package/dist/engine/file-type.js.map +1 -0
- package/dist/engine/index-status.d.ts +9 -0
- package/dist/engine/index-status.js +38 -0
- package/dist/engine/index-status.js.map +1 -0
- package/dist/engine/manifest.d.ts +13 -0
- package/dist/engine/manifest.js +124 -0
- package/dist/engine/manifest.js.map +1 -0
- package/dist/engine/models/artifact-cache-lock.d.ts +16 -0
- package/dist/engine/models/artifact-cache-lock.js +228 -0
- package/dist/engine/models/artifact-cache-lock.js.map +1 -0
- package/dist/engine/models/artifact-downloader.d.ts +77 -0
- package/dist/engine/models/artifact-downloader.js +766 -0
- package/dist/engine/models/artifact-downloader.js.map +1 -0
- package/dist/engine/models/backends/llama-cpp.d.ts +100 -0
- package/dist/engine/models/backends/llama-cpp.js +567 -0
- package/dist/engine/models/backends/llama-cpp.js.map +1 -0
- package/dist/engine/models/backends/model2vec-runtime.d.ts +63 -0
- package/dist/engine/models/backends/model2vec-runtime.js +112 -0
- package/dist/engine/models/backends/model2vec-runtime.js.map +1 -0
- package/dist/engine/models/backends/model2vec-tokenizer.d.ts +2 -0
- package/dist/engine/models/backends/model2vec-tokenizer.js +42 -0
- package/dist/engine/models/backends/model2vec-tokenizer.js.map +1 -0
- package/dist/engine/models/backends/model2vec-worker-pool.d.ts +25 -0
- package/dist/engine/models/backends/model2vec-worker-pool.js +264 -0
- package/dist/engine/models/backends/model2vec-worker-pool.js.map +1 -0
- package/dist/engine/models/backends/model2vec-worker.d.ts +1 -0
- package/dist/engine/models/backends/model2vec-worker.js +41 -0
- package/dist/engine/models/backends/model2vec-worker.js.map +1 -0
- package/dist/engine/models/backends/model2vec.d.ts +38 -0
- package/dist/engine/models/backends/model2vec.js +344 -0
- package/dist/engine/models/backends/model2vec.js.map +1 -0
- package/dist/engine/models/backends/qwen.d.ts +44 -0
- package/dist/engine/models/backends/qwen.js +411 -0
- package/dist/engine/models/backends/qwen.js.map +1 -0
- package/dist/engine/models/backends/transformers-js.d.ts +90 -0
- package/dist/engine/models/backends/transformers-js.js +411 -0
- package/dist/engine/models/backends/transformers-js.js.map +1 -0
- package/dist/engine/models/catalog.d.ts +490 -0
- package/dist/engine/models/catalog.js +522 -0
- package/dist/engine/models/catalog.js.map +1 -0
- package/dist/engine/models/download-progress.d.ts +16 -0
- package/dist/engine/models/download-progress.js +47 -0
- package/dist/engine/models/download-progress.js.map +1 -0
- package/dist/engine/models/embeddings.d.ts +73 -0
- package/dist/engine/models/embeddings.js +125 -0
- package/dist/engine/models/embeddings.js.map +1 -0
- package/dist/engine/models/factory.d.ts +2 -0
- package/dist/engine/models/factory.js +51 -0
- package/dist/engine/models/factory.js.map +1 -0
- package/dist/engine/models/index.d.ts +5 -0
- package/dist/engine/models/index.js +6 -0
- package/dist/engine/models/index.js.map +1 -0
- package/dist/engine/models/local-embedding-parallelism.d.ts +15 -0
- package/dist/engine/models/local-embedding-parallelism.js +63 -0
- package/dist/engine/models/local-embedding-parallelism.js.map +1 -0
- package/dist/engine/models/local-embedding-queue.d.ts +12 -0
- package/dist/engine/models/local-embedding-queue.js +65 -0
- package/dist/engine/models/local-embedding-queue.js.map +1 -0
- package/dist/engine/models/ranking.d.ts +19 -0
- package/dist/engine/models/ranking.js +2 -0
- package/dist/engine/models/ranking.js.map +1 -0
- package/dist/engine/models/resolution.d.ts +8 -0
- package/dist/engine/models/resolution.js +22 -0
- package/dist/engine/models/resolution.js.map +1 -0
- package/dist/engine/pipeline/indexing/index.d.ts +14 -0
- package/dist/engine/pipeline/indexing/index.js +1219 -0
- package/dist/engine/pipeline/indexing/index.js.map +1 -0
- package/dist/engine/pipeline/indexing/input-budget.d.ts +2 -0
- package/dist/engine/pipeline/indexing/input-budget.js +51 -0
- package/dist/engine/pipeline/indexing/input-budget.js.map +1 -0
- package/dist/engine/pipeline/indexing/root-paths.d.ts +7 -0
- package/dist/engine/pipeline/indexing/root-paths.js +148 -0
- package/dist/engine/pipeline/indexing/root-paths.js.map +1 -0
- package/dist/engine/pipeline/indexing/scanner/index.d.ts +14 -0
- package/dist/engine/pipeline/indexing/scanner/index.js +776 -0
- package/dist/engine/pipeline/indexing/scanner/index.js.map +1 -0
- package/dist/engine/pipeline/search/index.d.ts +12 -0
- package/dist/engine/pipeline/search/index.js +811 -0
- package/dist/engine/pipeline/search/index.js.map +1 -0
- package/dist/engine/service/index.d.ts +3 -0
- package/dist/engine/service/index.js +2 -0
- package/dist/engine/service/index.js.map +1 -0
- package/dist/engine/service/lexical.d.ts +28 -0
- package/dist/engine/service/lexical.js +446 -0
- package/dist/engine/service/lexical.js.map +1 -0
- package/dist/engine/service/root.d.ts +15 -0
- package/dist/engine/service/root.js +57 -0
- package/dist/engine/service/root.js.map +1 -0
- package/dist/engine/service/structure-enrichment.d.ts +8 -0
- package/dist/engine/service/structure-enrichment.js +178 -0
- package/dist/engine/service/structure-enrichment.js.map +1 -0
- package/dist/engine/service/types.d.ts +212 -0
- package/dist/engine/service/types.js +2 -0
- package/dist/engine/service/types.js.map +1 -0
- package/dist/engine/service/workspace-index.d.ts +26 -0
- package/dist/engine/service/workspace-index.js +157 -0
- package/dist/engine/service/workspace-index.js.map +1 -0
- package/dist/engine/service/zvec-grep.d.ts +15 -0
- package/dist/engine/service/zvec-grep.js +1520 -0
- package/dist/engine/service/zvec-grep.js.map +1 -0
- package/dist/engine/storage/index.d.ts +53 -0
- package/dist/engine/storage/index.js +3 -0
- package/dist/engine/storage/index.js.map +1 -0
- package/dist/engine/storage/layout.d.ts +10 -0
- package/dist/engine/storage/layout.js +29 -0
- package/dist/engine/storage/layout.js.map +1 -0
- package/dist/engine/storage/zvec.d.ts +4 -0
- package/dist/engine/storage/zvec.js +976 -0
- package/dist/engine/storage/zvec.js.map +1 -0
- package/dist/engine/types.d.ts +303 -0
- package/dist/engine/types.js +14 -0
- package/dist/engine/types.js.map +1 -0
- package/dist/engine/utils/atomic-file.d.ts +3 -0
- package/dist/engine/utils/atomic-file.js +39 -0
- package/dist/engine/utils/atomic-file.js.map +1 -0
- package/dist/engine/utils/daemon-lease.d.ts +17 -0
- package/dist/engine/utils/daemon-lease.js +175 -0
- package/dist/engine/utils/daemon-lease.js.map +1 -0
- package/dist/engine/utils/file-selection.d.ts +10 -0
- package/dist/engine/utils/file-selection.js +157 -0
- package/dist/engine/utils/file-selection.js.map +1 -0
- package/dist/engine/utils/glob.d.ts +9 -0
- package/dist/engine/utils/glob.js +208 -0
- package/dist/engine/utils/glob.js.map +1 -0
- package/dist/engine/utils/hash.d.ts +2 -0
- package/dist/engine/utils/hash.js +8 -0
- package/dist/engine/utils/hash.js.map +1 -0
- package/dist/engine/utils/json.d.ts +8 -0
- package/dist/engine/utils/json.js +83 -0
- package/dist/engine/utils/json.js.map +1 -0
- package/dist/engine/utils/lock.d.ts +19 -0
- package/dist/engine/utils/lock.js +254 -0
- package/dist/engine/utils/lock.js.map +1 -0
- package/dist/engine/utils/path.d.ts +4 -0
- package/dist/engine/utils/path.js +18 -0
- package/dist/engine/utils/path.js.map +1 -0
- package/dist/engine/utils/timing.d.ts +19 -0
- package/dist/engine/utils/timing.js +87 -0
- package/dist/engine/utils/timing.js.map +1 -0
- package/dist/index-progress.d.ts +10 -0
- package/dist/index-progress.js +159 -0
- package/dist/index-progress.js.map +1 -0
- package/dist/index.d.ts +6 -0
- package/dist/index.js +4 -0
- package/dist/index.js.map +1 -0
- package/dist/mcp/http-transport.d.ts +27 -0
- package/dist/mcp/http-transport.js +182 -0
- package/dist/mcp/http-transport.js.map +1 -0
- package/dist/mcp/input-normalization.d.ts +38 -0
- package/dist/mcp/input-normalization.js +104 -0
- package/dist/mcp/input-normalization.js.map +1 -0
- package/dist/mcp/progress-heartbeat.d.ts +17 -0
- package/dist/mcp/progress-heartbeat.js +46 -0
- package/dist/mcp/progress-heartbeat.js.map +1 -0
- package/dist/mcp/request-metadata.d.ts +2 -0
- package/dist/mcp/request-metadata.js +11 -0
- package/dist/mcp/request-metadata.js.map +1 -0
- package/dist/mcp/request-state.d.ts +40 -0
- package/dist/mcp/request-state.js +210 -0
- package/dist/mcp/request-state.js.map +1 -0
- package/dist/mcp/result-format.d.ts +13 -0
- package/dist/mcp/result-format.js +75 -0
- package/dist/mcp/result-format.js.map +1 -0
- package/dist/mcp/schemas.d.ts +668 -0
- package/dist/mcp/schemas.js +512 -0
- package/dist/mcp/schemas.js.map +1 -0
- package/dist/mcp/stdio-bridge.d.ts +15 -0
- package/dist/mcp/stdio-bridge.js +190 -0
- package/dist/mcp/stdio-bridge.js.map +1 -0
- package/dist/mcp/tools.d.ts +153 -0
- package/dist/mcp/tools.js +488 -0
- package/dist/mcp/tools.js.map +1 -0
- package/dist/mcp/toolset.d.ts +5 -0
- package/dist/mcp/toolset.js +12 -0
- package/dist/mcp/toolset.js.map +1 -0
- package/dist/observability/trace-context.d.ts +11 -0
- package/dist/observability/trace-context.js +138 -0
- package/dist/observability/trace-context.js.map +1 -0
- package/dist/prompts/zvec-grep-guidance.d.ts +2 -0
- package/dist/prompts/zvec-grep-guidance.js +9 -0
- package/dist/prompts/zvec-grep-guidance.js.map +1 -0
- package/docs/01-agents.md +217 -0
- package/docs/02-cli.md +259 -0
- package/docs/03-mcp.md +185 -0
- package/docs/04-pipeline.md +234 -0
- package/docs/05-architecture.md +86 -0
- package/docs/06-server.md +279 -0
- package/docs/07-embedding.md +227 -0
- package/docs/08-roadmap.md +85 -0
- package/docs/README.md +53 -0
- package/package.json +87 -0
|
@@ -0,0 +1,411 @@
|
|
|
1
|
+
import { globalConfigPath } from "../../config.js";
|
|
2
|
+
import { EngineError } from "../../errors.js";
|
|
3
|
+
import { BaseEmbeddingModel, } from "../embeddings.js";
|
|
4
|
+
import { traceHeaders } from "../../../observability/trace-context.js";
|
|
5
|
+
// -----------------------------------------------------------------------------
|
|
6
|
+
// Text embedding models (OpenAI-compatible API)
|
|
7
|
+
// -----------------------------------------------------------------------------
|
|
8
|
+
const DEFAULT_REMOTE_EMBEDDING_TIMEOUT_MS = 60_000;
|
|
9
|
+
const defaultDependencies = {
|
|
10
|
+
fetch: (...args) => globalThis.fetch(...args),
|
|
11
|
+
};
|
|
12
|
+
const QWEN_TEXT_EMBEDDING_V4_SPEC = {
|
|
13
|
+
displayName: "Qwen text-embedding-v4",
|
|
14
|
+
errorCodePrefix: "QWEN_TEXT_EMBEDDING_V4",
|
|
15
|
+
};
|
|
16
|
+
const QWEN37_TEXT_EMBEDDING_SPEC = {
|
|
17
|
+
displayName: "Qwen3.7 text embedding",
|
|
18
|
+
errorCodePrefix: "QWEN37_TEXT_EMBEDDING",
|
|
19
|
+
};
|
|
20
|
+
class QwenTextEmbeddingModel extends BaseEmbeddingModel {
|
|
21
|
+
info;
|
|
22
|
+
entry;
|
|
23
|
+
apiKey;
|
|
24
|
+
endpoint;
|
|
25
|
+
displayName;
|
|
26
|
+
errorCodePrefix;
|
|
27
|
+
dependencies;
|
|
28
|
+
constructor(entry, spec, options, dependencies) {
|
|
29
|
+
super();
|
|
30
|
+
this.entry = entry;
|
|
31
|
+
const endpoint = options.endpoint === undefined
|
|
32
|
+
? entry.defaultEndpoint
|
|
33
|
+
: options.endpoint.trim();
|
|
34
|
+
this.info = {
|
|
35
|
+
reference: entry.reference,
|
|
36
|
+
provider: entry.provider,
|
|
37
|
+
name: entry.model,
|
|
38
|
+
dimension: entry.dimension,
|
|
39
|
+
metric: entry.metric,
|
|
40
|
+
endpoint,
|
|
41
|
+
inputKinds: ["text"],
|
|
42
|
+
limits: {
|
|
43
|
+
maxBatchSize: entry.maxBatchSize,
|
|
44
|
+
maxInputTokens: entry.maxInputTokens,
|
|
45
|
+
},
|
|
46
|
+
};
|
|
47
|
+
this.displayName = spec.displayName;
|
|
48
|
+
this.errorCodePrefix = spec.errorCodePrefix;
|
|
49
|
+
this.dependencies = { ...defaultDependencies, ...dependencies };
|
|
50
|
+
const apiKey = options.apiKey?.trim() ?? "";
|
|
51
|
+
if (apiKey.length === 0) {
|
|
52
|
+
throw new EngineError(`${this.displayName} model requires an API key`, {
|
|
53
|
+
code: this.errorCode("MISSING_API_KEY"),
|
|
54
|
+
context: `model=${this.info.reference}\nhint=Pass --api-key, set ZVEC_GREP_API_KEY, or configure providers.qwen.apiKey in ${globalConfigPath()}.`,
|
|
55
|
+
});
|
|
56
|
+
}
|
|
57
|
+
if (endpoint.length === 0) {
|
|
58
|
+
throw new EngineError(`${this.displayName} model requires an endpoint`, {
|
|
59
|
+
code: this.errorCode("MISSING_ENDPOINT"),
|
|
60
|
+
context: `model=${this.info.reference}`,
|
|
61
|
+
});
|
|
62
|
+
}
|
|
63
|
+
this.apiKey = apiKey;
|
|
64
|
+
this.endpoint = endpoint;
|
|
65
|
+
}
|
|
66
|
+
async doEmbed(contents, options) {
|
|
67
|
+
const texts = contents.map((content) => content.text);
|
|
68
|
+
let response;
|
|
69
|
+
const signal = remoteEmbeddingSignal(options.signal);
|
|
70
|
+
try {
|
|
71
|
+
response = await this.dependencies.fetch(this.endpoint, {
|
|
72
|
+
method: "POST",
|
|
73
|
+
headers: {
|
|
74
|
+
...traceHeaders(),
|
|
75
|
+
Authorization: `Bearer ${this.apiKey}`,
|
|
76
|
+
"Content-Type": "application/json",
|
|
77
|
+
},
|
|
78
|
+
body: JSON.stringify({
|
|
79
|
+
model: this.entry.model,
|
|
80
|
+
input: texts,
|
|
81
|
+
dimensions: this.info.dimension,
|
|
82
|
+
encoding_format: "float",
|
|
83
|
+
}),
|
|
84
|
+
signal,
|
|
85
|
+
});
|
|
86
|
+
}
|
|
87
|
+
catch (cause) {
|
|
88
|
+
throwIfEmbeddingCancelled(options.signal);
|
|
89
|
+
throw new EngineError(`${this.displayName} request failed`, {
|
|
90
|
+
code: this.errorCode("REQUEST_FAILED"),
|
|
91
|
+
context: `model=${this.info.reference} endpoint=${this.endpoint} timeoutMs=${DEFAULT_REMOTE_EMBEDDING_TIMEOUT_MS}`,
|
|
92
|
+
cause,
|
|
93
|
+
});
|
|
94
|
+
}
|
|
95
|
+
let body;
|
|
96
|
+
try {
|
|
97
|
+
body = await response.json();
|
|
98
|
+
}
|
|
99
|
+
catch (cause) {
|
|
100
|
+
throw new EngineError(`${this.displayName} response was not valid JSON`, {
|
|
101
|
+
code: this.errorCode("INVALID_JSON"),
|
|
102
|
+
context: `model=${this.info.reference} status=${response.status}`,
|
|
103
|
+
cause,
|
|
104
|
+
});
|
|
105
|
+
}
|
|
106
|
+
if (!response.ok) {
|
|
107
|
+
const error = readProviderError(body);
|
|
108
|
+
throw new EngineError(`${this.displayName} request returned an error`, {
|
|
109
|
+
code: this.errorCode("API_ERROR"),
|
|
110
|
+
context: providerErrorContext(this.entry.model, response, error),
|
|
111
|
+
});
|
|
112
|
+
}
|
|
113
|
+
if (!isRecord(body) || !Array.isArray(body.data)) {
|
|
114
|
+
throw new EngineError(`${this.displayName} response did not include data`, {
|
|
115
|
+
code: this.errorCode("MISSING_DATA"),
|
|
116
|
+
context: `model=${this.info.reference}`,
|
|
117
|
+
});
|
|
118
|
+
}
|
|
119
|
+
const vectors = new Array(texts.length);
|
|
120
|
+
for (const item of body.data) {
|
|
121
|
+
if (!isRecord(item) ||
|
|
122
|
+
typeof item.index !== "number" ||
|
|
123
|
+
!Number.isInteger(item.index)) {
|
|
124
|
+
throw new EngineError(`${this.displayName} response included an invalid index`, {
|
|
125
|
+
code: this.errorCode("INVALID_INDEX"),
|
|
126
|
+
context: `model=${this.info.reference} index=${isRecord(item) ? String(item.index) : "unknown"}`,
|
|
127
|
+
});
|
|
128
|
+
}
|
|
129
|
+
if (item.index < 0 || item.index >= texts.length) {
|
|
130
|
+
throw new EngineError(`${this.displayName} response index was out of range`, {
|
|
131
|
+
code: this.errorCode("INDEX_OUT_OF_RANGE"),
|
|
132
|
+
context: `model=${this.info.reference} index=${item.index} inputCount=${texts.length}`,
|
|
133
|
+
});
|
|
134
|
+
}
|
|
135
|
+
if (!Array.isArray(item.embedding)) {
|
|
136
|
+
throw new EngineError(`${this.displayName} response included an invalid embedding`, {
|
|
137
|
+
code: this.errorCode("INVALID_VECTOR"),
|
|
138
|
+
context: `model=${this.info.reference} index=${item.index}`,
|
|
139
|
+
});
|
|
140
|
+
}
|
|
141
|
+
vectors[item.index] = item.embedding;
|
|
142
|
+
}
|
|
143
|
+
return { vectors, truncated: [] };
|
|
144
|
+
}
|
|
145
|
+
errorCode(suffix) {
|
|
146
|
+
return `ZVEC_GREP.ENGINE.MODELS.${this.errorCodePrefix}_${suffix}`;
|
|
147
|
+
}
|
|
148
|
+
}
|
|
149
|
+
export class QwenTextEmbeddingV4Model extends QwenTextEmbeddingModel {
|
|
150
|
+
constructor(entry, options, dependencies = {}) {
|
|
151
|
+
super(entry, QWEN_TEXT_EMBEDDING_V4_SPEC, options, dependencies);
|
|
152
|
+
}
|
|
153
|
+
}
|
|
154
|
+
export class Qwen37TextEmbeddingModel extends QwenTextEmbeddingModel {
|
|
155
|
+
constructor(entry, options, dependencies = {}) {
|
|
156
|
+
super(entry, QWEN37_TEXT_EMBEDDING_SPEC, options, dependencies);
|
|
157
|
+
}
|
|
158
|
+
}
|
|
159
|
+
// -----------------------------------------------------------------------------
|
|
160
|
+
// qwen3-vl-embedding
|
|
161
|
+
// -----------------------------------------------------------------------------
|
|
162
|
+
const QWEN3_VL_EMBEDDING_MAX_IMAGE_COUNT = 10;
|
|
163
|
+
const QWEN3_VL_EMBEDDING_SUPPORTED_IMAGE_FORMATS = [
|
|
164
|
+
"jpeg",
|
|
165
|
+
"png",
|
|
166
|
+
"webp",
|
|
167
|
+
];
|
|
168
|
+
export class Qwen3VlEmbeddingModel extends BaseEmbeddingModel {
|
|
169
|
+
info;
|
|
170
|
+
entry;
|
|
171
|
+
apiKey;
|
|
172
|
+
endpoint;
|
|
173
|
+
dependencies;
|
|
174
|
+
constructor(entry, options, dependencies = {}) {
|
|
175
|
+
super();
|
|
176
|
+
this.entry = entry;
|
|
177
|
+
this.dependencies = { ...defaultDependencies, ...dependencies };
|
|
178
|
+
const endpoint = options.endpoint === undefined
|
|
179
|
+
? entry.defaultEndpoint
|
|
180
|
+
: options.endpoint.trim();
|
|
181
|
+
this.info = {
|
|
182
|
+
reference: entry.reference,
|
|
183
|
+
provider: entry.provider,
|
|
184
|
+
name: entry.model,
|
|
185
|
+
dimension: entry.dimension,
|
|
186
|
+
metric: entry.metric,
|
|
187
|
+
endpoint,
|
|
188
|
+
inputKinds: ["text", "image"],
|
|
189
|
+
limits: {
|
|
190
|
+
maxBatchSize: entry.maxBatchSize,
|
|
191
|
+
maxInputTokens: entry.maxInputTokens,
|
|
192
|
+
maxImageBytes: entry.maxImageBytes,
|
|
193
|
+
},
|
|
194
|
+
};
|
|
195
|
+
const apiKey = options.apiKey?.trim() ?? "";
|
|
196
|
+
if (apiKey.length === 0) {
|
|
197
|
+
throw new EngineError("Qwen3 VL embedding model requires an API key", {
|
|
198
|
+
code: "ZVEC_GREP.ENGINE.MODELS.QWEN3_VL_EMBEDDING_MISSING_API_KEY",
|
|
199
|
+
context: `model=${this.info.reference}\nhint=Pass --api-key, set ZVEC_GREP_API_KEY, or configure providers.qwen.apiKey in ${globalConfigPath()}.`,
|
|
200
|
+
});
|
|
201
|
+
}
|
|
202
|
+
if (endpoint.length === 0) {
|
|
203
|
+
throw new EngineError("Qwen3 VL embedding model requires an endpoint", {
|
|
204
|
+
code: "ZVEC_GREP.ENGINE.MODELS.QWEN3_VL_EMBEDDING_MISSING_ENDPOINT",
|
|
205
|
+
context: `model=${this.info.reference}`,
|
|
206
|
+
});
|
|
207
|
+
}
|
|
208
|
+
this.apiKey = apiKey;
|
|
209
|
+
this.endpoint = endpoint;
|
|
210
|
+
}
|
|
211
|
+
async doEmbed(contents, options) {
|
|
212
|
+
validateQwen3VlContents(this.entry.model, contents);
|
|
213
|
+
const requestContents = contents.map((content) => {
|
|
214
|
+
if (content.kind === "text") {
|
|
215
|
+
return {
|
|
216
|
+
text: content.text,
|
|
217
|
+
};
|
|
218
|
+
}
|
|
219
|
+
return {
|
|
220
|
+
image: bytesToBase64(content.data),
|
|
221
|
+
};
|
|
222
|
+
});
|
|
223
|
+
let response;
|
|
224
|
+
const signal = remoteEmbeddingSignal(options.signal);
|
|
225
|
+
try {
|
|
226
|
+
response = await this.dependencies.fetch(this.endpoint, {
|
|
227
|
+
method: "POST",
|
|
228
|
+
headers: {
|
|
229
|
+
...traceHeaders(),
|
|
230
|
+
Authorization: `Bearer ${this.apiKey}`,
|
|
231
|
+
"Content-Type": "application/json",
|
|
232
|
+
},
|
|
233
|
+
body: JSON.stringify({
|
|
234
|
+
model: this.entry.model,
|
|
235
|
+
input: {
|
|
236
|
+
contents: requestContents,
|
|
237
|
+
},
|
|
238
|
+
parameters: {
|
|
239
|
+
dimension: this.info.dimension,
|
|
240
|
+
},
|
|
241
|
+
}),
|
|
242
|
+
signal,
|
|
243
|
+
});
|
|
244
|
+
}
|
|
245
|
+
catch (cause) {
|
|
246
|
+
throwIfEmbeddingCancelled(options.signal);
|
|
247
|
+
throw new EngineError("Qwen3 VL embedding request failed", {
|
|
248
|
+
code: "ZVEC_GREP.ENGINE.MODELS.QWEN3_VL_EMBEDDING_REQUEST_FAILED",
|
|
249
|
+
context: `model=${this.info.reference} endpoint=${this.endpoint} timeoutMs=${DEFAULT_REMOTE_EMBEDDING_TIMEOUT_MS}`,
|
|
250
|
+
cause,
|
|
251
|
+
});
|
|
252
|
+
}
|
|
253
|
+
let body;
|
|
254
|
+
try {
|
|
255
|
+
body = await response.json();
|
|
256
|
+
}
|
|
257
|
+
catch (cause) {
|
|
258
|
+
throw new EngineError("Qwen3 VL embedding response was not valid JSON", {
|
|
259
|
+
code: "ZVEC_GREP.ENGINE.MODELS.QWEN3_VL_EMBEDDING_INVALID_JSON",
|
|
260
|
+
context: `model=${this.info.reference} status=${response.status}`,
|
|
261
|
+
cause,
|
|
262
|
+
});
|
|
263
|
+
}
|
|
264
|
+
if (!response.ok) {
|
|
265
|
+
const error = readProviderError(body);
|
|
266
|
+
throw new EngineError("Qwen3 VL embedding request returned an error", {
|
|
267
|
+
code: "ZVEC_GREP.ENGINE.MODELS.QWEN3_VL_EMBEDDING_API_ERROR",
|
|
268
|
+
context: providerErrorContext(this.entry.model, response, error),
|
|
269
|
+
});
|
|
270
|
+
}
|
|
271
|
+
if (!isRecord(body) ||
|
|
272
|
+
!isRecord(body.output) ||
|
|
273
|
+
!Array.isArray(body.output.embeddings)) {
|
|
274
|
+
throw new EngineError("Qwen3 VL embedding response did not include embeddings", {
|
|
275
|
+
code: "ZVEC_GREP.ENGINE.MODELS.QWEN3_VL_EMBEDDING_MISSING_EMBEDDINGS",
|
|
276
|
+
context: `model=${this.info.reference}`,
|
|
277
|
+
});
|
|
278
|
+
}
|
|
279
|
+
const vectors = new Array(contents.length);
|
|
280
|
+
for (const [fallbackIndex, item] of body.output.embeddings.entries()) {
|
|
281
|
+
if (!isRecord(item)) {
|
|
282
|
+
throw new EngineError("Qwen3 VL embedding response included an invalid embedding item", {
|
|
283
|
+
code: "ZVEC_GREP.ENGINE.MODELS.QWEN3_VL_EMBEDDING_INVALID_ITEM",
|
|
284
|
+
context: `model=${this.info.reference} index=${fallbackIndex}`,
|
|
285
|
+
});
|
|
286
|
+
}
|
|
287
|
+
const index = readEmbeddingIndex(item, fallbackIndex);
|
|
288
|
+
if (index < 0 || index >= contents.length) {
|
|
289
|
+
throw new EngineError("Qwen3 VL embedding response index was out of range", {
|
|
290
|
+
code: "ZVEC_GREP.ENGINE.MODELS.QWEN3_VL_EMBEDDING_INDEX_OUT_OF_RANGE",
|
|
291
|
+
context: `model=${this.info.reference} index=${index} inputCount=${contents.length}`,
|
|
292
|
+
});
|
|
293
|
+
}
|
|
294
|
+
if (!Array.isArray(item.embedding)) {
|
|
295
|
+
throw new EngineError("Qwen3 VL embedding response included an invalid embedding", {
|
|
296
|
+
code: "ZVEC_GREP.ENGINE.MODELS.QWEN3_VL_EMBEDDING_INVALID_VECTOR",
|
|
297
|
+
context: `model=${this.info.reference} index=${index}`,
|
|
298
|
+
});
|
|
299
|
+
}
|
|
300
|
+
vectors[index] = item.embedding;
|
|
301
|
+
}
|
|
302
|
+
return { vectors, truncated: [] };
|
|
303
|
+
}
|
|
304
|
+
}
|
|
305
|
+
// -----------------------------------------------------------------------------
|
|
306
|
+
// Shared helpers
|
|
307
|
+
// -----------------------------------------------------------------------------
|
|
308
|
+
const BASE64_CHARS = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
|
|
309
|
+
function providerErrorContext(model, response, error) {
|
|
310
|
+
const retryAfter = retryAfterHeaderMs(response.headers.get("retry-after"));
|
|
311
|
+
const retryAfterDetail = typeof retryAfter === "number" ? ` retryAfterMs=${retryAfter}` : "";
|
|
312
|
+
return `model=${model} status=${response.status}${retryAfterDetail} providerCode=${error.code} providerType=${error.type} providerMessage=${error.message}`;
|
|
313
|
+
}
|
|
314
|
+
function retryAfterHeaderMs(value) {
|
|
315
|
+
if (!value) {
|
|
316
|
+
return undefined;
|
|
317
|
+
}
|
|
318
|
+
const seconds = Number(value);
|
|
319
|
+
if (Number.isFinite(seconds) && seconds >= 0) {
|
|
320
|
+
return Math.round(seconds * 1000);
|
|
321
|
+
}
|
|
322
|
+
const dateMs = Date.parse(value);
|
|
323
|
+
if (Number.isFinite(dateMs)) {
|
|
324
|
+
return Math.max(0, dateMs - Date.now());
|
|
325
|
+
}
|
|
326
|
+
return undefined;
|
|
327
|
+
}
|
|
328
|
+
function readProviderError(body) {
|
|
329
|
+
if (!isRecord(body) || !isRecord(body.error)) {
|
|
330
|
+
if (isRecord(body)) {
|
|
331
|
+
return {
|
|
332
|
+
code: typeof body.code === "string" ? body.code : "unknown",
|
|
333
|
+
type: "unknown",
|
|
334
|
+
message: typeof body.message === "string" ? body.message : "unknown",
|
|
335
|
+
};
|
|
336
|
+
}
|
|
337
|
+
return {
|
|
338
|
+
code: "unknown",
|
|
339
|
+
type: "unknown",
|
|
340
|
+
message: "unknown",
|
|
341
|
+
};
|
|
342
|
+
}
|
|
343
|
+
return {
|
|
344
|
+
code: typeof body.error.code === "string" ? body.error.code : "unknown",
|
|
345
|
+
type: typeof body.error.type === "string" ? body.error.type : "unknown",
|
|
346
|
+
message: typeof body.error.message === "string" ? body.error.message : "unknown",
|
|
347
|
+
};
|
|
348
|
+
}
|
|
349
|
+
function readEmbeddingIndex(item, fallbackIndex) {
|
|
350
|
+
if (typeof item.index === "number" && Number.isInteger(item.index)) {
|
|
351
|
+
return item.index;
|
|
352
|
+
}
|
|
353
|
+
if (typeof item.text_index === "number" &&
|
|
354
|
+
Number.isInteger(item.text_index)) {
|
|
355
|
+
return item.text_index;
|
|
356
|
+
}
|
|
357
|
+
return fallbackIndex;
|
|
358
|
+
}
|
|
359
|
+
function validateQwen3VlContents(model, contents) {
|
|
360
|
+
let imageCount = 0;
|
|
361
|
+
for (const [index, content] of contents.entries()) {
|
|
362
|
+
if (content.kind !== "image") {
|
|
363
|
+
continue;
|
|
364
|
+
}
|
|
365
|
+
imageCount += 1;
|
|
366
|
+
if (!QWEN3_VL_EMBEDDING_SUPPORTED_IMAGE_FORMATS.includes(content.format)) {
|
|
367
|
+
throw new EngineError("Qwen3 VL embedding model does not support image format", {
|
|
368
|
+
code: "ZVEC_GREP.ENGINE.MODELS.QWEN3_VL_EMBEDDING_UNSUPPORTED_IMAGE_FORMAT",
|
|
369
|
+
context: `model=${model} index=${index} format=${content.format}`,
|
|
370
|
+
});
|
|
371
|
+
}
|
|
372
|
+
}
|
|
373
|
+
if (imageCount > QWEN3_VL_EMBEDDING_MAX_IMAGE_COUNT) {
|
|
374
|
+
throw new EngineError("Qwen3 VL embedding image count exceeds model limit", {
|
|
375
|
+
code: "ZVEC_GREP.ENGINE.MODELS.QWEN3_VL_EMBEDDING_TOO_MANY_IMAGES",
|
|
376
|
+
context: `model=${model} imageCount=${imageCount} maxImageCount=${QWEN3_VL_EMBEDDING_MAX_IMAGE_COUNT}`,
|
|
377
|
+
});
|
|
378
|
+
}
|
|
379
|
+
}
|
|
380
|
+
function isRecord(value) {
|
|
381
|
+
return typeof value === "object" && value !== null;
|
|
382
|
+
}
|
|
383
|
+
function bytesToBase64(bytes) {
|
|
384
|
+
let output = "";
|
|
385
|
+
for (let index = 0; index < bytes.length; index += 3) {
|
|
386
|
+
const first = bytes[index] ?? 0;
|
|
387
|
+
const second = bytes[index + 1];
|
|
388
|
+
const third = bytes[index + 2];
|
|
389
|
+
output += BASE64_CHARS.charAt(first >> 2);
|
|
390
|
+
output += BASE64_CHARS.charAt(((first & 3) << 4) | ((second ?? 0) >> 4));
|
|
391
|
+
output +=
|
|
392
|
+
second === undefined
|
|
393
|
+
? "="
|
|
394
|
+
: BASE64_CHARS.charAt(((second & 15) << 2) | ((third ?? 0) >> 6));
|
|
395
|
+
output += third === undefined ? "=" : BASE64_CHARS.charAt(third & 63);
|
|
396
|
+
}
|
|
397
|
+
return output;
|
|
398
|
+
}
|
|
399
|
+
function remoteEmbeddingSignal(signal) {
|
|
400
|
+
const timeout = AbortSignal.timeout(DEFAULT_REMOTE_EMBEDDING_TIMEOUT_MS);
|
|
401
|
+
return signal ? AbortSignal.any([signal, timeout]) : timeout;
|
|
402
|
+
}
|
|
403
|
+
function throwIfEmbeddingCancelled(signal) {
|
|
404
|
+
if (!signal?.aborted) {
|
|
405
|
+
return;
|
|
406
|
+
}
|
|
407
|
+
throw signal.reason instanceof Error
|
|
408
|
+
? signal.reason
|
|
409
|
+
: new Error("Embedding request was cancelled.");
|
|
410
|
+
}
|
|
411
|
+
//# sourceMappingURL=qwen.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"qwen.js","sourceRoot":"","sources":["../../../../src/engine/models/backends/qwen.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AACnD,OAAO,EAAE,WAAW,EAAwB,MAAM,iBAAiB,CAAC;AAEpE,OAAO,EACL,kBAAkB,GAKnB,MAAM,kBAAkB,CAAC;AAK1B,OAAO,EAAE,YAAY,EAAE,MAAM,yCAAyC,CAAC;AAEvE,gFAAgF;AAChF,gDAAgD;AAChD,gFAAgF;AAEhF,MAAM,mCAAmC,GAAG,MAAM,CAAC;AAMnD,MAAM,mBAAmB,GAAqB;IAC5C,KAAK,EAAE,CAAC,GAAG,IAAI,EAAE,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC;CAC9C,CAAC;AAOF,MAAM,2BAA2B,GAAG;IAClC,WAAW,EAAE,wBAAwB;IACrC,eAAe,EAAE,wBAAwB;CACD,CAAC;AAE3C,MAAM,0BAA0B,GAAG;IACjC,WAAW,EAAE,wBAAwB;IACrC,eAAe,EAAE,uBAAuB;CACA,CAAC;AAY3C,MAAe,sBAAuB,SAAQ,kBAAkB;IACrD,IAAI,CAAqB;IAEjB,KAAK,CAAgC;IACrC,MAAM,CAAS;IACf,QAAQ,CAAS;IACjB,WAAW,CAAS;IACpB,eAAe,CAAS;IACxB,YAAY,CAAmB;IAEhD,YACE,KAAoC,EACpC,IAA2B,EAC3B,OAAoC,EACpC,YAAuC;QAEvC,KAAK,EAAE,CAAC;QAER,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,MAAM,QAAQ,GACZ,OAAO,CAAC,QAAQ,KAAK,SAAS;YAC5B,CAAC,CAAC,KAAK,CAAC,eAAe;YACvB,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC9B,IAAI,CAAC,IAAI,GAAG;YACV,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,IAAI,EAAE,KAAK,CAAC,KAAK;YACjB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,QAAQ;YACR,UAAU,EAAE,CAAC,MAAM,CAAC;YACpB,MAAM,EAAE;gBACN,YAAY,EAAE,KAAK,CAAC,YAAY;gBAChC,cAAc,EAAE,KAAK,CAAC,cAAc;aACrC;SACF,CAAC;QACF,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;QACpC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC;QAC5C,IAAI,CAAC,YAAY,GAAG,EAAE,GAAG,mBAAmB,EAAE,GAAG,YAAY,EAAE,CAAC;QAEhE,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QAC5C,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,MAAM,IAAI,WAAW,CAAC,GAAG,IAAI,CAAC,WAAW,4BAA4B,EAAE;gBACrE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC;gBACvC,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,uFAAuF,gBAAgB,EAAE,GAAG;aAClJ,CAAC,CAAC;QACL,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,WAAW,CAAC,GAAG,IAAI,CAAC,WAAW,6BAA6B,EAAE;gBACtE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC;gBACxC,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;aACxC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAES,KAAK,CAAC,OAAO,CACrB,QAA4B,EAC5B,OAAmC;QAEnC,MAAM,KAAK,GAAI,QAAmC,CAAC,GAAG,CACpD,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAC1B,CAAC;QAEF,IAAI,QAAkB,CAAC;QACvB,MAAM,MAAM,GAAG,qBAAqB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAErD,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE;gBACtD,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,GAAG,YAAY,EAAE;oBACjB,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;oBACtC,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK;oBACvB,KAAK,EAAE,KAAK;oBACZ,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS;oBAC/B,eAAe,EAAE,OAAO;iBACzB,CAAC;gBACF,MAAM;aACP,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,yBAAyB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAC1C,MAAM,IAAI,WAAW,CAAC,GAAG,IAAI,CAAC,WAAW,iBAAiB,EAAE;gBAC1D,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC;gBACtC,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,aAAa,IAAI,CAAC,QAAQ,cAAc,mCAAmC,EAAE;gBAClH,KAAK;aACN,CAAC,CAAC;QACL,CAAC;QAED,IAAI,IAAa,CAAC;QAElB,IAAI,CAAC;YACH,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,WAAW,CAAC,GAAG,IAAI,CAAC,WAAW,8BAA8B,EAAE;gBACvE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC;gBACpC,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,WAAW,QAAQ,CAAC,MAAM,EAAE;gBACjE,KAAK;aACN,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,KAAK,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAEtC,MAAM,IAAI,WAAW,CAAC,GAAG,IAAI,CAAC,WAAW,4BAA4B,EAAE;gBACrE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;gBACjC,OAAO,EAAE,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,QAAQ,EAAE,KAAK,CAAC;aACjE,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YACjD,MAAM,IAAI,WAAW,CACnB,GAAG,IAAI,CAAC,WAAW,gCAAgC,EACnD;gBACE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC;gBACpC,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;aACxC,CACF,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,KAAK,CAAW,KAAK,CAAC,MAAM,CAAC,CAAC;QAElD,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YAC7B,IACE,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACf,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ;gBAC9B,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,EAC7B,CAAC;gBACD,MAAM,IAAI,WAAW,CACnB,GAAG,IAAI,CAAC,WAAW,qCAAqC,EACxD;oBACE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC;oBACrC,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,UAAU,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE;iBACjG,CACF,CAAC;YACJ,CAAC;YAED,IAAI,IAAI,CAAC,KAAK,GAAG,CAAC,IAAI,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;gBACjD,MAAM,IAAI,WAAW,CACnB,GAAG,IAAI,CAAC,WAAW,kCAAkC,EACrD;oBACE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC;oBAC1C,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,UAAU,IAAI,CAAC,KAAK,eAAe,KAAK,CAAC,MAAM,EAAE;iBACvF,CACF,CAAC;YACJ,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;gBACnC,MAAM,IAAI,WAAW,CACnB,GAAG,IAAI,CAAC,WAAW,yCAAyC,EAC5D;oBACE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC;oBACtC,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,UAAU,IAAI,CAAC,KAAK,EAAE;iBAC5D,CACF,CAAC;YACJ,CAAC;YAED,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,SAAqB,CAAC;QACnD,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;IACpC,CAAC;IAEO,SAAS,CAAC,MAAc;QAC9B,OAAO,2BAA2B,IAAI,CAAC,eAAe,IAAI,MAAM,EAAE,CAAC;IACrE,CAAC;CACF;AAED,MAAM,OAAO,wBAAyB,SAAQ,sBAAsB;IAClE,YACE,KAAsC,EACtC,OAAoC,EACpC,eAA0C,EAAE;QAE5C,KAAK,CAAC,KAAK,EAAE,2BAA2B,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC;IACnE,CAAC;CACF;AAED,MAAM,OAAO,wBAAyB,SAAQ,sBAAsB;IAClE,YACE,KAAsC,EACtC,OAAoC,EACpC,eAA0C,EAAE;QAE5C,KAAK,CAAC,KAAK,EAAE,0BAA0B,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC;IAClE,CAAC;CACF;AAED,gFAAgF;AAChF,qBAAqB;AACrB,gFAAgF;AAEhF,MAAM,kCAAkC,GAAG,EAAE,CAAC;AAE9C,MAAM,0CAA0C,GAA2B;IACzE,MAAM;IACN,KAAK;IACL,MAAM;CACP,CAAC;AAEF,MAAM,OAAO,qBAAsB,SAAQ,kBAAkB;IAClD,IAAI,CAAqB;IAEjB,KAAK,CAAsC;IAC3C,MAAM,CAAS;IACf,QAAQ,CAAS;IACjB,YAAY,CAAmB;IAEhD,YACE,KAA0C,EAC1C,OAAoC,EACpC,eAA0C,EAAE;QAE5C,KAAK,EAAE,CAAC;QAER,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,YAAY,GAAG,EAAE,GAAG,mBAAmB,EAAE,GAAG,YAAY,EAAE,CAAC;QAChE,MAAM,QAAQ,GACZ,OAAO,CAAC,QAAQ,KAAK,SAAS;YAC5B,CAAC,CAAC,KAAK,CAAC,eAAe;YACvB,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC9B,IAAI,CAAC,IAAI,GAAG;YACV,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,IAAI,EAAE,KAAK,CAAC,KAAK;YACjB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,QAAQ;YACR,UAAU,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC;YAC7B,MAAM,EAAE;gBACN,YAAY,EAAE,KAAK,CAAC,YAAY;gBAChC,cAAc,EAAE,KAAK,CAAC,cAAc;gBACpC,aAAa,EAAE,KAAK,CAAC,aAAa;aACnC;SACF,CAAC;QAEF,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QAC5C,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,MAAM,IAAI,WAAW,CAAC,8CAA8C,EAAE;gBACpE,IAAI,EAAE,4DAA4D;gBAClE,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,uFAAuF,gBAAgB,EAAE,GAAG;aAClJ,CAAC,CAAC;QACL,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,WAAW,CAAC,+CAA+C,EAAE;gBACrE,IAAI,EAAE,6DAA6D;gBACnE,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;aACxC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAES,KAAK,CAAC,OAAO,CACrB,QAA4B,EAC5B,OAAmC;QAEnC,uBAAuB,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QAEpD,MAAM,eAAe,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;YAC/C,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC5B,OAAO;oBACL,IAAI,EAAE,OAAO,CAAC,IAAI;iBACnB,CAAC;YACJ,CAAC;YAED,OAAO;gBACL,KAAK,EAAE,aAAa,CAAC,OAAO,CAAC,IAAI,CAAC;aACnC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,QAAkB,CAAC;QACvB,MAAM,MAAM,GAAG,qBAAqB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAErD,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE;gBACtD,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,GAAG,YAAY,EAAE;oBACjB,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;oBACtC,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK;oBACvB,KAAK,EAAE;wBACL,QAAQ,EAAE,eAAe;qBAC1B;oBACD,UAAU,EAAE;wBACV,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS;qBAC/B;iBACF,CAAC;gBACF,MAAM;aACP,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,yBAAyB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAC1C,MAAM,IAAI,WAAW,CAAC,mCAAmC,EAAE;gBACzD,IAAI,EAAE,2DAA2D;gBACjE,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,aAAa,IAAI,CAAC,QAAQ,cAAc,mCAAmC,EAAE;gBAClH,KAAK;aACN,CAAC,CAAC;QACL,CAAC;QAED,IAAI,IAAa,CAAC;QAElB,IAAI,CAAC;YACH,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,WAAW,CAAC,gDAAgD,EAAE;gBACtE,IAAI,EAAE,yDAAyD;gBAC/D,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,WAAW,QAAQ,CAAC,MAAM,EAAE;gBACjE,KAAK;aACN,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,KAAK,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAEtC,MAAM,IAAI,WAAW,CAAC,8CAA8C,EAAE;gBACpE,IAAI,EAAE,sDAAsD;gBAC5D,OAAO,EAAE,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,QAAQ,EAAE,KAAK,CAAC;aACjE,CAAC,CAAC;QACL,CAAC;QAED,IACE,CAAC,QAAQ,CAAC,IAAI,CAAC;YACf,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC;YACtB,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,EACtC,CAAC;YACD,MAAM,IAAI,WAAW,CACnB,wDAAwD,EACxD;gBACE,IAAI,EAAE,+DAA+D;gBACrE,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;aACxC,CACF,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,KAAK,CAAW,QAAQ,CAAC,MAAM,CAAC,CAAC;QAErD,KAAK,MAAM,CAAC,aAAa,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC;YACrE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;gBACpB,MAAM,IAAI,WAAW,CACnB,gEAAgE,EAChE;oBACE,IAAI,EAAE,yDAAyD;oBAC/D,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,UAAU,aAAa,EAAE;iBAC/D,CACF,CAAC;YACJ,CAAC;YAED,MAAM,KAAK,GAAG,kBAAkB,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;YAEtD,IAAI,KAAK,GAAG,CAAC,IAAI,KAAK,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;gBAC1C,MAAM,IAAI,WAAW,CACnB,oDAAoD,EACpD;oBACE,IAAI,EAAE,+DAA+D;oBACrE,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,UAAU,KAAK,eAAe,QAAQ,CAAC,MAAM,EAAE;iBACrF,CACF,CAAC;YACJ,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;gBACnC,MAAM,IAAI,WAAW,CACnB,2DAA2D,EAC3D;oBACE,IAAI,EAAE,2DAA2D;oBACjE,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,UAAU,KAAK,EAAE;iBACvD,CACF,CAAC;YACJ,CAAC;YAED,OAAO,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,SAAqB,CAAC;QAC9C,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;IACpC,CAAC;CACF;AAED,gFAAgF;AAChF,iBAAiB;AACjB,gFAAgF;AAEhF,MAAM,YAAY,GAChB,kEAAkE,CAAC;AAErE,SAAS,oBAAoB,CAC3B,KAAa,EACb,QAAkB,EAClB,KAAsD;IAEtD,MAAM,UAAU,GAAG,kBAAkB,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC;IAC3E,MAAM,gBAAgB,GACpB,OAAO,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,iBAAiB,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAEtE,OAAO,SAAS,KAAK,WAAW,QAAQ,CAAC,MAAM,GAAG,gBAAgB,iBAAiB,KAAK,CAAC,IAAI,iBAAiB,KAAK,CAAC,IAAI,oBAAoB,KAAK,CAAC,OAAO,EAAE,CAAC;AAC9J,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAoB;IAC9C,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC9B,IAAI,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,OAAO,IAAI,CAAC,EAAE,CAAC;QAC7C,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,CAAC;IACpC,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACjC,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QAC5B,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,iBAAiB,CAAC,IAAa;IAKtC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QAC7C,IAAI,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YACnB,OAAO;gBACL,IAAI,EAAE,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;gBAC3D,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,OAAO,IAAI,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;aACrE,CAAC;QACJ,CAAC;QAED,OAAO;YACL,IAAI,EAAE,SAAS;YACf,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,SAAS;SACnB,CAAC;IACJ,CAAC;IAED,OAAO;QACL,IAAI,EAAE,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;QACvE,IAAI,EAAE,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;QACvE,OAAO,EACL,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;KAC1E,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CACzB,IAA6B,EAC7B,aAAqB;IAErB,IAAI,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QACnE,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED,IACE,OAAO,IAAI,CAAC,UAAU,KAAK,QAAQ;QACnC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,EACjC,CAAC;QACD,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,OAAO,aAAa,CAAC;AACvB,CAAC;AAED,SAAS,uBAAuB,CAC9B,KAAa,EACb,QAA4B;IAE5B,IAAI,UAAU,GAAG,CAAC,CAAC;IAEnB,KAAK,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,IAAI,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC;QAClD,IAAI,OAAO,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YAC7B,SAAS;QACX,CAAC;QAED,UAAU,IAAI,CAAC,CAAC;QAEhB,IAAI,CAAC,0CAA0C,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YACzE,MAAM,IAAI,WAAW,CACnB,wDAAwD,EACxD;gBACE,IAAI,EAAE,qEAAqE;gBAC3E,OAAO,EAAE,SAAS,KAAK,UAAU,KAAK,WAAW,OAAO,CAAC,MAAM,EAAE;aAClE,CACF,CAAC;QACJ,CAAC;IACH,CAAC;IAED,IAAI,UAAU,GAAG,kCAAkC,EAAE,CAAC;QACpD,MAAM,IAAI,WAAW,CACnB,oDAAoD,EACpD;YACE,IAAI,EAAE,4DAA4D;YAClE,OAAO,EAAE,SAAS,KAAK,eAAe,UAAU,kBAAkB,kCAAkC,EAAE;SACvG,CACF,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,CAAC;AACrD,CAAC;AAED,SAAS,aAAa,CAAC,KAAiB;IACtC,IAAI,MAAM,GAAG,EAAE,CAAC;IAEhB,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;QACrD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAChC,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;QAChC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;QAE/B,MAAM,IAAI,YAAY,CAAC,MAAM,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC;QAC1C,MAAM,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzE,MAAM;YACJ,MAAM,KAAK,SAAS;gBAClB,CAAC,CAAC,GAAG;gBACL,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACtE,MAAM,IAAI,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,GAAG,EAAE,CAAC,CAAC;IACxE,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,qBAAqB,CAAC,MAA+B;IAC5D,MAAM,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC,mCAAmC,CAAC,CAAC;IACzE,OAAO,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;AAC/D,CAAC;AAED,SAAS,yBAAyB,CAAC,MAA+B;IAChE,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;QACrB,OAAO;IACT,CAAC;IACD,MAAM,MAAM,CAAC,MAAM,YAAY,KAAK;QAClC,CAAC,CAAC,MAAM,CAAC,MAAM;QACf,CAAC,CAAC,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;AACpD,CAAC"}
|
|
@@ -0,0 +1,90 @@
|
|
|
1
|
+
import type { Content } from "../../types.js";
|
|
2
|
+
import { BaseEmbeddingModel, type CreateEmbeddingModelOptions, type EmbeddingModelInfo, type EmbeddingResult, type NormalizedEmbeddingOptions } from "../embeddings.js";
|
|
3
|
+
import type { TransformersJsEmbeddingCatalogEntry } from "../catalog.js";
|
|
4
|
+
import { resolveModelArtifacts } from "../artifact-downloader.js";
|
|
5
|
+
type TensorLike = {
|
|
6
|
+
data: ArrayLike<number>;
|
|
7
|
+
dims: readonly number[];
|
|
8
|
+
};
|
|
9
|
+
type TokenizerLike = {
|
|
10
|
+
(text: string | string[], options: {
|
|
11
|
+
truncation: true;
|
|
12
|
+
max_length: number;
|
|
13
|
+
padding: true;
|
|
14
|
+
}): {
|
|
15
|
+
input_ids: {
|
|
16
|
+
data: ArrayLike<number | bigint>;
|
|
17
|
+
dims: readonly number[];
|
|
18
|
+
};
|
|
19
|
+
attention_mask: {
|
|
20
|
+
data: ArrayLike<number | bigint>;
|
|
21
|
+
dims: readonly number[];
|
|
22
|
+
};
|
|
23
|
+
} | Promise<{
|
|
24
|
+
input_ids: {
|
|
25
|
+
data: ArrayLike<number | bigint>;
|
|
26
|
+
dims: readonly number[];
|
|
27
|
+
};
|
|
28
|
+
attention_mask: {
|
|
29
|
+
data: ArrayLike<number | bigint>;
|
|
30
|
+
dims: readonly number[];
|
|
31
|
+
};
|
|
32
|
+
}>;
|
|
33
|
+
model_max_length: number;
|
|
34
|
+
};
|
|
35
|
+
type FeatureExtractionPipeline = {
|
|
36
|
+
(texts: string[], options: {
|
|
37
|
+
pooling: "mean" | "cls";
|
|
38
|
+
normalize: boolean;
|
|
39
|
+
truncation: true;
|
|
40
|
+
max_length: number;
|
|
41
|
+
}): Promise<TensorLike>;
|
|
42
|
+
tokenizer: TokenizerLike;
|
|
43
|
+
dispose(): Promise<void>;
|
|
44
|
+
};
|
|
45
|
+
type TransformersJsModule = {
|
|
46
|
+
pipeline(task: "feature-extraction", repo: string, options: {
|
|
47
|
+
dtype: "fp32" | "q8" | "q4";
|
|
48
|
+
local_files_only: true;
|
|
49
|
+
session_options?: {
|
|
50
|
+
executionProviders: TransformersJsExecutionProvider[];
|
|
51
|
+
};
|
|
52
|
+
}): Promise<FeatureExtractionPipeline>;
|
|
53
|
+
};
|
|
54
|
+
type TransformersJsLoader = () => Promise<TransformersJsModule>;
|
|
55
|
+
type TransformersJsExecutionProvider = "cpu" | "webgpu" | "cuda" | "dml";
|
|
56
|
+
type ModelArtifactResolver = typeof resolveModelArtifacts;
|
|
57
|
+
type TransformersJsDependencies = {
|
|
58
|
+
loadRuntime: TransformersJsLoader;
|
|
59
|
+
resolveArtifacts: ModelArtifactResolver;
|
|
60
|
+
};
|
|
61
|
+
export declare class TransformersJsEmbeddingModel extends BaseEmbeddingModel {
|
|
62
|
+
private readonly entry;
|
|
63
|
+
readonly info: EmbeddingModelInfo;
|
|
64
|
+
private readonly modelCacheDir;
|
|
65
|
+
private readonly executionProvider;
|
|
66
|
+
private readonly dependencies;
|
|
67
|
+
private readonly embeddingQueue;
|
|
68
|
+
private pipeline;
|
|
69
|
+
private pipelineLoadPromise;
|
|
70
|
+
private pipelineLoadError;
|
|
71
|
+
private resolvedArtifacts;
|
|
72
|
+
private artifactResolutionPromise;
|
|
73
|
+
private sourceFallbackWarningReported;
|
|
74
|
+
private usingCpuFallback;
|
|
75
|
+
private disposed;
|
|
76
|
+
constructor(entry: TransformersJsEmbeddingCatalogEntry, options: CreateEmbeddingModelOptions, dependencies?: Partial<TransformersJsDependencies>);
|
|
77
|
+
protected doEmbed(contents: readonly Content[], options: NormalizedEmbeddingOptions): Promise<EmbeddingResult>;
|
|
78
|
+
private embedBatch;
|
|
79
|
+
private runEmbeddingAttempt;
|
|
80
|
+
dispose(): Promise<void>;
|
|
81
|
+
private ensurePipeline;
|
|
82
|
+
private loadPipeline;
|
|
83
|
+
private embedTexts;
|
|
84
|
+
private fallbackToCpu;
|
|
85
|
+
private createPipeline;
|
|
86
|
+
private ensureArtifacts;
|
|
87
|
+
private createArtifactSources;
|
|
88
|
+
private ensureNotDisposed;
|
|
89
|
+
}
|
|
90
|
+
export {};
|