@qvac/sdk 0.4.1 → 0.6.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 +365 -200
- package/bare-imports.json +35 -0
- package/dist/client/api/cancel.d.ts +12 -3
- package/dist/client/api/cancel.js +12 -3
- package/dist/client/api/completion-stream.d.ts +15 -5
- package/dist/client/api/completion-stream.js +50 -14
- package/dist/client/api/embed.d.ts +17 -3
- package/dist/client/api/embed.js +0 -8
- package/dist/client/api/get-model-info.d.ts +2 -2
- package/dist/client/api/get-model-info.js +2 -2
- package/dist/client/api/index.d.ts +2 -2
- package/dist/client/api/index.js +2 -2
- package/dist/client/api/load-model.d.ts +37 -11
- package/dist/client/api/load-model.js +34 -117
- package/dist/client/api/logging-stream.d.ts +6 -3
- package/dist/client/api/logging-stream.js +6 -3
- package/dist/client/api/ocr.d.ts +31 -0
- package/dist/client/api/ocr.js +88 -0
- package/dist/client/api/rag.d.ts +215 -20
- package/dist/client/api/rag.js +382 -23
- package/dist/client/api/translate.js +10 -11
- package/dist/client/api/unload-model.js +6 -3
- package/dist/client/config-loader/config-utils.d.ts +5 -0
- package/dist/client/config-loader/config-utils.js +20 -0
- package/dist/client/config-loader/resolve-config.bare.d.ts +9 -0
- package/dist/client/config-loader/resolve-config.bare.js +87 -0
- package/dist/client/config-loader/resolve-config.expo.d.ts +11 -0
- package/dist/client/config-loader/resolve-config.expo.js +62 -0
- package/dist/client/config-loader/resolve-config.node.d.ts +9 -0
- package/dist/client/config-loader/resolve-config.node.js +118 -0
- package/dist/client/init-hooks.d.ts +18 -3
- package/dist/client/init-hooks.js +57 -29
- package/dist/client/logging-stream-registry.d.ts +1 -1
- package/dist/client/logging-stream-registry.js +14 -13
- package/dist/client/rpc/bare-client.d.ts +3 -184
- package/dist/client/rpc/bare-client.js +67 -8
- package/dist/client/rpc/expo-rpc-client.d.ts +2 -2
- package/dist/client/rpc/expo-rpc-client.js +70 -12
- package/dist/client/rpc/node-rpc-client.d.ts +1 -1
- package/dist/client/rpc/node-rpc-client.js +14 -7
- package/dist/client/rpc/rpc-client.d.ts +1 -1
- package/dist/client/rpc/rpc-client.js +13 -58
- package/dist/examples/config/blind-relay/blind-relay.config.d.ts +5 -0
- package/dist/examples/config/blind-relay/blind-relay.config.js +7 -0
- package/dist/examples/config/default/default.config.d.ts +8 -0
- package/dist/examples/config/default/default.config.js +6 -0
- package/dist/examples/config-reload.d.ts +2 -0
- package/dist/examples/config-reload.js +49 -0
- package/dist/examples/default-config-usage.d.ts +17 -0
- package/dist/examples/default-config-usage.js +47 -0
- package/dist/examples/delegated-inference/consumer.js +1 -0
- package/dist/examples/download-with-blind-relays.js +22 -21
- package/dist/examples/embed-hyperdrive.js +39 -3
- package/dist/examples/kv-cache-cleanup.js +1 -1
- package/dist/examples/kv-cache-custom-key.js +16 -1
- package/dist/examples/kv-cache-example.js +1 -1
- package/dist/examples/llamacpp-hyperdrive.js +1 -1
- package/dist/examples/llamacpp-sharded.js +13 -4
- package/dist/examples/logging-file-transport.d.ts +2 -0
- package/dist/examples/{llamacpp-file-logging.js → logging-file-transport.js} +3 -7
- package/dist/examples/logging-streaming.d.ts +2 -0
- package/dist/examples/logging-streaming.js +75 -0
- package/dist/examples/mcp-websearch.d.ts +13 -0
- package/dist/examples/mcp-websearch.js +176 -0
- package/dist/examples/ocr-fasttext.d.ts +2 -0
- package/dist/examples/ocr-fasttext.js +52 -0
- package/dist/examples/quickstart.d.ts +2 -0
- package/dist/examples/quickstart.js +28 -0
- package/dist/examples/rag/rag-hyperdb/cancellation.d.ts +2 -0
- package/dist/examples/rag/rag-hyperdb/cancellation.js +75 -0
- package/dist/examples/rag/rag-hyperdb/ingest.d.ts +2 -0
- package/dist/examples/rag/{rag-hyperdb.js → rag-hyperdb/ingest.js} +10 -4
- package/dist/examples/rag/rag-hyperdb/pipeline.d.ts +2 -0
- package/dist/examples/rag/rag-hyperdb/pipeline.js +74 -0
- package/dist/examples/rag/rag-hyperdb/workspaces.d.ts +2 -0
- package/dist/examples/rag/rag-hyperdb/workspaces.js +178 -0
- package/dist/examples/translation/translation-bergamot-batch.d.ts +2 -0
- package/dist/examples/translation/translation-bergamot-batch.js +49 -0
- package/dist/examples/translation/translation-bergamot.d.ts +2 -0
- package/dist/examples/translation/translation-bergamot.js +39 -0
- package/dist/examples/translation/translation-indic.js +1 -0
- package/dist/examples/translation/translation-opus.js +1 -0
- package/dist/examples/translation/translation-stream.js +1 -0
- package/dist/expo/plugins/withMobileBundle.js +29 -10
- package/dist/index.d.ts +5 -4
- package/dist/index.js +4 -2
- package/dist/logging/addon.d.ts +6 -0
- package/dist/logging/addon.js +75 -0
- package/dist/logging/base-logger.js +26 -16
- package/dist/logging/client-logger.d.ts +3 -0
- package/dist/logging/client-logger.js +24 -0
- package/dist/logging/index.d.ts +7 -1
- package/dist/logging/index.js +7 -1
- package/dist/logging/logger.d.ts +0 -2
- package/dist/logging/logger.js +11 -6
- package/dist/logging/namespaces.d.ts +12 -0
- package/dist/logging/namespaces.js +12 -0
- package/dist/logging/registry.d.ts +31 -0
- package/dist/logging/registry.js +48 -0
- package/dist/logging/server-logger.d.ts +3 -0
- package/dist/logging/server-logger.js +16 -0
- package/dist/logging/stream-logger.d.ts +1 -1
- package/dist/logging/stream-logger.js +6 -3
- package/dist/logging/transport.js +2 -2
- package/dist/logging/types.d.ts +2 -0
- package/dist/logging/utils.d.ts +2 -0
- package/dist/logging/utils.js +62 -0
- package/dist/models/hyperdrive/models.d.ts +1886 -230
- package/dist/models/hyperdrive/models.js +2191 -443
- package/dist/schemas/archive.d.ts +9 -0
- package/dist/schemas/archive.js +17 -0
- package/dist/schemas/cancel.d.ts +6 -0
- package/dist/schemas/cancel.js +5 -0
- package/dist/schemas/common.d.ts +660 -92
- package/dist/schemas/common.js +6 -4
- package/dist/schemas/delegate.d.ts +1 -0
- package/dist/schemas/delegate.js +1 -0
- package/dist/schemas/download-asset.d.ts +21 -9
- package/dist/schemas/embed.d.ts +3 -3
- package/dist/schemas/embed.js +9 -2
- package/dist/schemas/get-model-info.d.ts +12 -10
- package/dist/schemas/get-model-info.js +9 -1
- package/dist/schemas/index.d.ts +9 -1
- package/dist/schemas/index.js +12 -1
- package/dist/schemas/json-schema.d.ts +16 -0
- package/dist/schemas/json-schema.js +2 -0
- package/dist/schemas/llamacpp-config.d.ts +117 -12
- package/dist/schemas/llamacpp-config.js +46 -12
- package/dist/schemas/load-model.d.ts +3669 -481
- package/dist/schemas/load-model.js +95 -34
- package/dist/schemas/logging-stream.d.ts +11 -4
- package/dist/schemas/logging-stream.js +4 -3
- package/dist/schemas/mcp-adapter.d.ts +47 -0
- package/dist/schemas/mcp-adapter.js +1 -0
- package/dist/schemas/model-src-utils.d.ts +38 -14
- package/dist/schemas/model-src-utils.js +4 -3
- package/dist/schemas/model-types.d.ts +154 -0
- package/dist/schemas/model-types.js +140 -0
- package/dist/schemas/ocr.d.ts +88 -0
- package/dist/schemas/ocr.js +52 -0
- package/dist/schemas/rag.d.ts +183 -52
- package/dist/schemas/rag.js +148 -23
- package/dist/schemas/reload-config.d.ts +356 -0
- package/dist/schemas/reload-config.js +44 -0
- package/dist/schemas/runtime-context.d.ts +19 -0
- package/dist/schemas/runtime-context.js +7 -0
- package/dist/schemas/sdk-config.d.ts +717 -0
- package/dist/schemas/sdk-config.js +131 -0
- package/dist/schemas/sdk-errors-client.d.ts +5 -0
- package/dist/schemas/sdk-errors-client.js +25 -0
- package/dist/schemas/sdk-errors-server.d.ts +17 -0
- package/dist/schemas/sdk-errors-server.js +85 -0
- package/dist/schemas/shard.d.ts +21 -0
- package/dist/schemas/shard.js +17 -0
- package/dist/schemas/tools.d.ts +3 -0
- package/dist/schemas/translate.d.ts +68 -23
- package/dist/schemas/translate.js +37 -11
- package/dist/schemas/translation-config.d.ts +337 -13
- package/dist/schemas/translation-config.js +65 -9
- package/dist/schemas/whispercpp-config.d.ts +9 -1
- package/dist/schemas/whispercpp-config.js +2 -1
- package/dist/scripts/run-unit-tests.d.ts +2 -0
- package/dist/scripts/run-unit-tests.js +17 -0
- package/dist/server/bare/addons/llamacpp-completion/cache-logger.d.ts +6 -0
- package/dist/server/bare/addons/llamacpp-completion/cache-logger.js +34 -0
- package/dist/server/bare/addons/{llamacpp → llamacpp-completion}/create-model.d.ts +1 -5
- package/dist/server/bare/addons/{llamacpp → llamacpp-completion}/create-model.js +6 -20
- package/dist/server/bare/addons/{llamacpp → llamacpp-completion}/handlers/completion-stream.js +26 -7
- package/dist/server/bare/addons/llamacpp-completion/index.d.ts +4 -0
- package/dist/server/bare/addons/llamacpp-completion/index.js +3 -0
- package/dist/server/bare/addons/{llamacpp → llamacpp-completion}/kv-cache-utils.d.ts +4 -1
- package/dist/server/bare/addons/{llamacpp → llamacpp-completion}/kv-cache-utils.js +49 -14
- package/dist/server/bare/addons/llamacpp-embedding/create-model.d.ts +8 -0
- package/dist/server/bare/addons/llamacpp-embedding/create-model.js +38 -0
- package/dist/server/bare/addons/llamacpp-embedding/handlers/embed.d.ts +12 -0
- package/dist/server/bare/addons/{llamacpp → llamacpp-embedding}/handlers/embed.js +18 -4
- package/dist/server/bare/addons/llamacpp-embedding/index.d.ts +3 -0
- package/dist/server/bare/addons/llamacpp-embedding/index.js +2 -0
- package/dist/server/bare/addons/load-model.js +38 -28
- package/dist/server/bare/addons/{translation → nmtcpp-translation}/create-model.js +23 -8
- package/dist/server/bare/addons/{translation → nmtcpp-translation}/handlers/translate.js +35 -11
- package/dist/server/bare/addons/onnx-ocr/create-model.d.ts +10 -0
- package/dist/server/bare/addons/onnx-ocr/create-model.js +39 -0
- package/dist/server/bare/addons/onnx-ocr/handlers/ocr-stream.d.ts +10 -0
- package/dist/server/bare/addons/onnx-ocr/handlers/ocr-stream.js +109 -0
- package/dist/server/bare/addons/onnx-ocr/index.d.ts +3 -0
- package/dist/server/bare/addons/onnx-ocr/index.js +2 -0
- package/dist/server/bare/addons/unload-model.js +6 -3
- package/dist/server/bare/addons/{whispercpp → whispercpp-transcription}/create-model.js +2 -1
- package/dist/server/bare/addons/whispercpp-transcription/handlers/transcribe-stream.js +82 -0
- package/dist/server/bare/delegate-rpc-client.js +31 -11
- package/dist/server/bare/hyperswarm.js +7 -5
- package/dist/server/bare/rag-hyperdb/handlers/chunk.d.ts +3 -0
- package/dist/server/bare/rag-hyperdb/handlers/chunk.js +7 -0
- package/dist/server/bare/rag-hyperdb/handlers/close-workspace.d.ts +3 -0
- package/dist/server/bare/rag-hyperdb/handlers/close-workspace.js +9 -0
- package/dist/server/bare/rag-hyperdb/handlers/delete-embeddings.d.ts +1 -1
- package/dist/server/bare/rag-hyperdb/handlers/delete-embeddings.js +9 -10
- package/dist/server/bare/rag-hyperdb/handlers/delete-workspace.d.ts +3 -0
- package/dist/server/bare/rag-hyperdb/handlers/delete-workspace.js +14 -0
- package/dist/server/bare/rag-hyperdb/handlers/ingest.d.ts +9 -0
- package/dist/server/bare/rag-hyperdb/handlers/ingest.js +18 -0
- package/dist/server/bare/rag-hyperdb/handlers/list-workspaces.d.ts +3 -0
- package/dist/server/bare/rag-hyperdb/handlers/list-workspaces.js +4 -0
- package/dist/server/bare/rag-hyperdb/handlers/reindex.d.ts +9 -0
- package/dist/server/bare/rag-hyperdb/handlers/reindex.js +16 -0
- package/dist/server/bare/rag-hyperdb/handlers/save-embeddings.d.ts +7 -4
- package/dist/server/bare/rag-hyperdb/handlers/save-embeddings.js +13 -14
- package/dist/server/bare/rag-hyperdb/handlers/search.js +10 -10
- package/dist/server/bare/rag-hyperdb/index.d.ts +8 -1
- package/dist/server/bare/rag-hyperdb/index.js +8 -1
- package/dist/server/bare/rag-hyperdb/rag-operation-manager.d.ts +7 -0
- package/dist/server/bare/rag-hyperdb/rag-operation-manager.js +39 -0
- package/dist/server/bare/rag-hyperdb/rag-workspace-manager.d.ts +15 -0
- package/dist/server/bare/rag-hyperdb/rag-workspace-manager.js +135 -0
- package/dist/server/bare/registry/config-registry.d.ts +13 -4
- package/dist/server/bare/registry/config-registry.js +53 -10
- package/dist/server/bare/registry/logging-stream-registry.d.ts +24 -7
- package/dist/server/bare/registry/logging-stream-registry.js +99 -20
- package/dist/server/bare/registry/model-config-registry.d.ts +4 -0
- package/dist/server/bare/registry/model-config-registry.js +15 -0
- package/dist/server/bare/registry/model-config-utils.d.ts +15 -0
- package/dist/server/bare/registry/model-config-utils.js +121 -0
- package/dist/server/bare/registry/model-registry.d.ts +6 -26
- package/dist/server/bare/registry/model-registry.js +9 -6
- package/dist/server/bare/registry/runtime-context-registry.d.ts +7 -0
- package/dist/server/bare/registry/runtime-context-registry.js +26 -0
- package/dist/server/bare/utils/loader-adapter.d.ts +7 -0
- package/dist/server/bare/utils/loader-adapter.js +9 -0
- package/dist/server/error-handlers.js +4 -2
- package/dist/server/rpc/create-server.js +5 -3
- package/dist/server/rpc/handle-request.js +115 -19
- package/dist/server/rpc/handlers/cancelHandler.js +13 -2
- package/dist/server/rpc/handlers/completion-stream-delegated.js +4 -2
- package/dist/server/rpc/handlers/completion-stream.js +1 -1
- package/dist/server/rpc/handlers/delete-cache.js +4 -2
- package/dist/server/rpc/handlers/download-asset.js +3 -1
- package/dist/server/rpc/handlers/embed.js +4 -2
- package/dist/server/rpc/handlers/get-model-info.d.ts +1 -1
- package/dist/server/rpc/handlers/get-model-info.js +22 -10
- package/dist/server/rpc/handlers/index.d.ts +4 -2
- package/dist/server/rpc/handlers/index.js +2 -2
- package/dist/server/rpc/handlers/load-model/download-manager.d.ts +11 -11
- package/dist/server/rpc/handlers/load-model/download-manager.js +40 -28
- package/dist/server/rpc/handlers/{load-model.d.ts → load-model/handler.d.ts} +2 -2
- package/dist/server/rpc/handlers/load-model/handler.js +179 -0
- package/dist/server/rpc/handlers/load-model/http.js +361 -50
- package/dist/server/rpc/handlers/load-model/hyperdrive.js +26 -22
- package/dist/server/rpc/handlers/load-model/index.d.ts +2 -2
- package/dist/server/rpc/handlers/load-model/index.js +2 -2
- package/dist/server/rpc/handlers/load-model/resolve.js +39 -8
- package/dist/server/rpc/handlers/load-model-delegated.d.ts +2 -2
- package/dist/server/rpc/handlers/load-model-delegated.js +15 -10
- package/dist/server/rpc/handlers/logging-stream.js +4 -4
- package/dist/server/rpc/handlers/ocr-stream.d.ts +3 -0
- package/dist/server/rpc/handlers/ocr-stream.js +39 -0
- package/dist/server/rpc/handlers/provideHandler/connection.js +12 -10
- package/dist/server/rpc/handlers/provideHandler/handler.js +18 -16
- package/dist/server/rpc/handlers/provideHandler/proxy.js +8 -6
- package/dist/server/rpc/handlers/rag.d.ts +2 -2
- package/dist/server/rpc/handlers/rag.js +107 -41
- package/dist/server/rpc/handlers/stopProvideHandler.js +3 -1
- package/dist/server/rpc/handlers/text-to-speech.js +1 -1
- package/dist/server/rpc/handlers/transcribe-stream.js +4 -2
- package/dist/server/rpc/handlers/translate.js +9 -16
- package/dist/server/rpc/handlers/unload-model.js +4 -2
- package/dist/server/utils/archive.d.ts +3 -0
- package/dist/server/utils/archive.js +171 -0
- package/dist/server/utils/audio/decoder.d.ts +2 -1
- package/dist/server/utils/audio/decoder.js +48 -11
- package/dist/server/utils/cache.d.ts +5 -5
- package/dist/server/utils/cache.js +13 -11
- package/dist/server/utils/config-transform.d.ts +3 -0
- package/dist/server/utils/config-transform.js +23 -0
- package/dist/server/utils/formatting.d.ts +13 -0
- package/dist/server/utils/formatting.js +26 -0
- package/dist/server/utils/gguf-tensor-extractor.d.ts +15 -0
- package/dist/server/utils/gguf-tensor-extractor.js +314 -0
- package/dist/server/utils/index.d.ts +4 -1
- package/dist/server/utils/index.js +4 -1
- package/dist/server/utils/shard-utils.d.ts +42 -16
- package/dist/server/utils/shard-utils.js +95 -7
- package/dist/server/worker.js +49 -19
- package/dist/utils/errors-base.d.ts +1 -1
- package/dist/utils/errors-base.js +2 -2
- package/dist/utils/errors-client.d.ts +24 -0
- package/dist/utils/errors-client.js +40 -0
- package/dist/utils/errors-server.d.ts +42 -0
- package/dist/utils/errors-server.js +71 -0
- package/dist/utils/mcp-adapter.d.ts +12 -0
- package/dist/utils/mcp-adapter.js +103 -0
- package/dist/utils/object.d.ts +2 -0
- package/dist/utils/object.js +3 -0
- package/dist/utils/tool-helpers.d.ts +10 -2
- package/dist/utils/tool-helpers.js +24 -11
- package/package.json +74 -36
- package/dist/client/api/set-config.d.ts +0 -7
- package/dist/client/api/set-config.js +0 -20
- package/dist/client/config-manager.d.ts +0 -4
- package/dist/client/config-manager.js +0 -8
- package/dist/client/rpc/bare-rpc-client.d.ts +0 -4
- package/dist/client/rpc/bare-rpc-client.js +0 -91
- package/dist/examples/llamacpp-file-logging.d.ts +0 -2
- package/dist/examples/rag/rag-hyperdb-workspaces.d.ts +0 -2
- package/dist/examples/rag/rag-hyperdb-workspaces.js +0 -106
- package/dist/examples/rag/rag-hyperdb.d.ts +0 -2
- package/dist/schemas/set-config.d.ts +0 -21
- package/dist/schemas/set-config.js +0 -14
- package/dist/server/bare/addons/llamacpp/handlers/embed.d.ts +0 -4
- package/dist/server/bare/addons/llamacpp/index.d.ts +0 -4
- package/dist/server/bare/addons/llamacpp/index.js +0 -3
- package/dist/server/bare/addons/whispercpp/handlers/transcribe-stream.js +0 -63
- package/dist/server/bare/rag-hyperdb/rag-manager.d.ts +0 -5
- package/dist/server/bare/rag-hyperdb/rag-manager.js +0 -64
- package/dist/server/rpc/handlers/load-model.js +0 -59
- package/dist/server/rpc/handlers/set-config.d.ts +0 -3
- package/dist/server/rpc/handlers/set-config.js +0 -18
- package/dist/server/utils/hash.d.ts +0 -5
- package/dist/server/utils/hash.js +0 -11
- /package/dist/server/bare/addons/{llamacpp → llamacpp-completion}/handlers/completion-stream.d.ts +0 -0
- /package/dist/server/bare/addons/{translation → nmtcpp-translation}/create-model.d.ts +0 -0
- /package/dist/server/bare/addons/{translation → nmtcpp-translation}/handlers/translate.d.ts +0 -0
- /package/dist/server/bare/addons/{translation → nmtcpp-translation}/index.d.ts +0 -0
- /package/dist/server/bare/addons/{translation → nmtcpp-translation}/index.js +0 -0
- /package/dist/server/bare/addons/{tts → onnx-tts}/create-model.d.ts +0 -0
- /package/dist/server/bare/addons/{tts → onnx-tts}/create-model.js +0 -0
- /package/dist/server/bare/addons/{tts → onnx-tts}/handlers/text-to-speech.d.ts +0 -0
- /package/dist/server/bare/addons/{tts → onnx-tts}/handlers/text-to-speech.js +0 -0
- /package/dist/server/bare/addons/{tts → onnx-tts}/index.d.ts +0 -0
- /package/dist/server/bare/addons/{tts → onnx-tts}/index.js +0 -0
- /package/dist/server/bare/addons/{whispercpp → whispercpp-transcription}/create-model.d.ts +0 -0
- /package/dist/server/bare/addons/{whispercpp → whispercpp-transcription}/handlers/transcribe-stream.d.ts +0 -0
- /package/dist/server/bare/addons/{whispercpp → whispercpp-transcription}/index.d.ts +0 -0
- /package/dist/server/bare/addons/{whispercpp → whispercpp-transcription}/index.js +0 -0
|
@@ -0,0 +1,140 @@
|
|
|
1
|
+
import { z } from "zod";
|
|
2
|
+
// === INTERNAL: Canonical types (used in SDK code) ===
|
|
3
|
+
/**
|
|
4
|
+
* Canonical model type values for internal SDK use.
|
|
5
|
+
* Format: `engine-usecase` (e.g., "llamacpp-completion")
|
|
6
|
+
* Use dot-accessor syntax: `ModelType.llamacppCompletion`
|
|
7
|
+
*/
|
|
8
|
+
export const ModelType = {
|
|
9
|
+
llamacppCompletion: "llamacpp-completion",
|
|
10
|
+
whispercppTranscription: "whispercpp-transcription",
|
|
11
|
+
llamacppEmbedding: "llamacpp-embedding",
|
|
12
|
+
nmtcppTranslation: "nmtcpp-translation",
|
|
13
|
+
onnxTts: "onnx-tts",
|
|
14
|
+
onnxOcr: "onnx-ocr",
|
|
15
|
+
};
|
|
16
|
+
// === INTERNAL: Alias keys (backward compat names) ===
|
|
17
|
+
const AliasKeys = {
|
|
18
|
+
llm: "llm",
|
|
19
|
+
whisper: "whisper",
|
|
20
|
+
embeddings: "embeddings",
|
|
21
|
+
nmt: "nmt",
|
|
22
|
+
tts: "tts",
|
|
23
|
+
ocr: "ocr",
|
|
24
|
+
};
|
|
25
|
+
// === INTERNAL: Aliases (backward compat mapping) ===
|
|
26
|
+
/**
|
|
27
|
+
* Alias mappings for backward compatibility.
|
|
28
|
+
* Maps old names to canonical values.
|
|
29
|
+
*/
|
|
30
|
+
export const ModelTypeAliases = {
|
|
31
|
+
[AliasKeys.llm]: ModelType.llamacppCompletion,
|
|
32
|
+
[AliasKeys.whisper]: ModelType.whispercppTranscription,
|
|
33
|
+
[AliasKeys.embeddings]: ModelType.llamacppEmbedding,
|
|
34
|
+
[AliasKeys.nmt]: ModelType.nmtcppTranslation,
|
|
35
|
+
[AliasKeys.tts]: ModelType.onnxTts,
|
|
36
|
+
[AliasKeys.ocr]: ModelType.onnxOcr,
|
|
37
|
+
};
|
|
38
|
+
// Set of canonical values for quick lookup
|
|
39
|
+
const canonicalValuesSet = new Set(Object.values(ModelType));
|
|
40
|
+
// === PUBLIC: Combined (exported via index.ts as MODEL_TYPES) ===
|
|
41
|
+
/**
|
|
42
|
+
* All valid model types: canonical names + aliases.
|
|
43
|
+
*
|
|
44
|
+
* Canonical names follow `engine-usecase` format (e.g., "llamacpp-completion").
|
|
45
|
+
* Aliases resolve to canonical names for backward compatibility.
|
|
46
|
+
*
|
|
47
|
+
* @example
|
|
48
|
+
* ```typescript
|
|
49
|
+
* // Using alias (backward compatible)
|
|
50
|
+
* loadModel({ modelSrc: "...", modelType: MODEL_TYPES.nmt });
|
|
51
|
+
* // MODEL_TYPES.nmt resolves to "nmtcpp-translation"
|
|
52
|
+
*
|
|
53
|
+
* // Using canonical name directly
|
|
54
|
+
* loadModel({ modelSrc: "...", modelType: MODEL_TYPES.nmtcppTranslation });
|
|
55
|
+
* ```
|
|
56
|
+
*/
|
|
57
|
+
export const PUBLIC_MODEL_TYPES = {
|
|
58
|
+
...ModelType,
|
|
59
|
+
...ModelTypeAliases,
|
|
60
|
+
};
|
|
61
|
+
// === ZOD SCHEMAS ===
|
|
62
|
+
// Derive input values from objects - no string repetition
|
|
63
|
+
const inputValues = [
|
|
64
|
+
...Object.values(ModelType),
|
|
65
|
+
...Object.keys(ModelTypeAliases),
|
|
66
|
+
];
|
|
67
|
+
/** Schema accepting both canonical and alias model type inputs */
|
|
68
|
+
export const modelTypeInputSchema = z.enum(inputValues);
|
|
69
|
+
/** Schema that transforms input to canonical model type */
|
|
70
|
+
export const modelTypeSchema = modelTypeInputSchema.transform((val) => {
|
|
71
|
+
// If already canonical, return as-is
|
|
72
|
+
if (canonicalValuesSet.has(val)) {
|
|
73
|
+
return val;
|
|
74
|
+
}
|
|
75
|
+
// Otherwise, look up in aliases
|
|
76
|
+
return ModelTypeAliases[val];
|
|
77
|
+
});
|
|
78
|
+
/** Normalize model type input to canonical form */
|
|
79
|
+
export function normalizeModelType(input) {
|
|
80
|
+
// If already canonical, return as-is
|
|
81
|
+
if (canonicalValuesSet.has(input)) {
|
|
82
|
+
return input;
|
|
83
|
+
}
|
|
84
|
+
// Otherwise, look up in aliases
|
|
85
|
+
return ModelTypeAliases[input];
|
|
86
|
+
}
|
|
87
|
+
/** Check if input is an alias (not canonical) */
|
|
88
|
+
export function isModelTypeAlias(input) {
|
|
89
|
+
return input in ModelTypeAliases;
|
|
90
|
+
}
|
|
91
|
+
// === PER-MODEL-TYPE SCHEMAS (for discriminated unions) ===
|
|
92
|
+
// Extracted from root enum - no value duplication
|
|
93
|
+
/**
|
|
94
|
+
* LLM/Completion model type schema.
|
|
95
|
+
* - Alias: `"llm"` → resolves to `"llamacpp-completion"`
|
|
96
|
+
* - Canonical: `"llamacpp-completion"`
|
|
97
|
+
*/
|
|
98
|
+
export const llmModelTypeSchema = modelTypeInputSchema
|
|
99
|
+
.extract([AliasKeys.llm, ModelType.llamacppCompletion])
|
|
100
|
+
.describe('LLM model type: "llm" (alias) or "llamacpp-completion" (canonical)');
|
|
101
|
+
/**
|
|
102
|
+
* Whisper/Transcription model type schema.
|
|
103
|
+
* - Alias: `"whisper"` → resolves to `"whispercpp-transcription"`
|
|
104
|
+
* - Canonical: `"whispercpp-transcription"`
|
|
105
|
+
*/
|
|
106
|
+
export const whisperModelTypeSchema = modelTypeInputSchema
|
|
107
|
+
.extract([AliasKeys.whisper, ModelType.whispercppTranscription])
|
|
108
|
+
.describe('Whisper model type: "whisper" (alias) or "whispercpp-transcription" (canonical)');
|
|
109
|
+
/**
|
|
110
|
+
* Embeddings model type schema.
|
|
111
|
+
* - Alias: `"embeddings"` → resolves to `"llamacpp-embedding"`
|
|
112
|
+
* - Canonical: `"llamacpp-embedding"`
|
|
113
|
+
*/
|
|
114
|
+
export const embeddingsModelTypeSchema = modelTypeInputSchema
|
|
115
|
+
.extract([AliasKeys.embeddings, ModelType.llamacppEmbedding])
|
|
116
|
+
.describe('Embeddings model type: "embeddings" (alias) or "llamacpp-embedding" (canonical)');
|
|
117
|
+
/**
|
|
118
|
+
* NMT/Translation model type schema.
|
|
119
|
+
* - Alias: `"nmt"` → resolves to `"nmtcpp-translation"`
|
|
120
|
+
* - Canonical: `"nmtcpp-translation"`
|
|
121
|
+
*/
|
|
122
|
+
export const nmtModelTypeSchema = modelTypeInputSchema
|
|
123
|
+
.extract([AliasKeys.nmt, ModelType.nmtcppTranslation])
|
|
124
|
+
.describe('NMT model type: "nmt" (alias) or "nmtcpp-translation" (canonical)');
|
|
125
|
+
/**
|
|
126
|
+
* TTS model type schema.
|
|
127
|
+
* - Alias: `"tts"` → resolves to `"onnx-tts"`
|
|
128
|
+
* - Canonical: `"onnx-tts"`
|
|
129
|
+
*/
|
|
130
|
+
export const ttsModelTypeSchema = modelTypeInputSchema
|
|
131
|
+
.extract([AliasKeys.tts, ModelType.onnxTts])
|
|
132
|
+
.describe('TTS model type: "tts" (alias) or "onnx-tts" (canonical)');
|
|
133
|
+
/**
|
|
134
|
+
* OCR model type schema.
|
|
135
|
+
* - Alias: `"ocr"` → resolves to `"onnx-ocr"`
|
|
136
|
+
* - Canonical: `"onnx-ocr"`
|
|
137
|
+
*/
|
|
138
|
+
export const ocrModelTypeSchema = modelTypeInputSchema
|
|
139
|
+
.extract([AliasKeys.ocr, ModelType.onnxOcr])
|
|
140
|
+
.describe('OCR model type: "ocr" (alias) or "onnx-ocr" (canonical)');
|
|
@@ -0,0 +1,88 @@
|
|
|
1
|
+
import { z } from "zod";
|
|
2
|
+
export declare const ocrConfigSchema: z.ZodObject<{
|
|
3
|
+
langList: z.ZodOptional<z.ZodArray<z.ZodString>>;
|
|
4
|
+
useGPU: z.ZodOptional<z.ZodBoolean>;
|
|
5
|
+
timeout: z.ZodOptional<z.ZodNumber>;
|
|
6
|
+
magRatio: z.ZodOptional<z.ZodNumber>;
|
|
7
|
+
defaultRotationAngles: z.ZodOptional<z.ZodArray<z.ZodNumber>>;
|
|
8
|
+
contrastRetry: z.ZodOptional<z.ZodBoolean>;
|
|
9
|
+
lowConfidenceThreshold: z.ZodOptional<z.ZodNumber>;
|
|
10
|
+
recognizerBatchSize: z.ZodOptional<z.ZodNumber>;
|
|
11
|
+
}, z.core.$strip>;
|
|
12
|
+
export declare const imageInputSchema: z.ZodDiscriminatedUnion<[z.ZodObject<{
|
|
13
|
+
type: z.ZodLiteral<"base64">;
|
|
14
|
+
value: z.ZodString;
|
|
15
|
+
}, z.core.$strip>, z.ZodObject<{
|
|
16
|
+
type: z.ZodLiteral<"filePath">;
|
|
17
|
+
value: z.ZodString;
|
|
18
|
+
}, z.core.$strip>], "type">;
|
|
19
|
+
export declare const ocrOptionsSchema: z.ZodObject<{
|
|
20
|
+
paragraph: z.ZodOptional<z.ZodBoolean>;
|
|
21
|
+
}, z.core.$strip>;
|
|
22
|
+
export declare const ocrParamsSchema: z.ZodObject<{
|
|
23
|
+
modelId: z.ZodString;
|
|
24
|
+
image: z.ZodDiscriminatedUnion<[z.ZodObject<{
|
|
25
|
+
type: z.ZodLiteral<"base64">;
|
|
26
|
+
value: z.ZodString;
|
|
27
|
+
}, z.core.$strip>, z.ZodObject<{
|
|
28
|
+
type: z.ZodLiteral<"filePath">;
|
|
29
|
+
value: z.ZodString;
|
|
30
|
+
}, z.core.$strip>], "type">;
|
|
31
|
+
options: z.ZodOptional<z.ZodObject<{
|
|
32
|
+
paragraph: z.ZodOptional<z.ZodBoolean>;
|
|
33
|
+
}, z.core.$strip>>;
|
|
34
|
+
}, z.core.$strip>;
|
|
35
|
+
export declare const ocrStreamRequestSchema: z.ZodObject<{
|
|
36
|
+
modelId: z.ZodString;
|
|
37
|
+
image: z.ZodDiscriminatedUnion<[z.ZodObject<{
|
|
38
|
+
type: z.ZodLiteral<"base64">;
|
|
39
|
+
value: z.ZodString;
|
|
40
|
+
}, z.core.$strip>, z.ZodObject<{
|
|
41
|
+
type: z.ZodLiteral<"filePath">;
|
|
42
|
+
value: z.ZodString;
|
|
43
|
+
}, z.core.$strip>], "type">;
|
|
44
|
+
options: z.ZodOptional<z.ZodObject<{
|
|
45
|
+
paragraph: z.ZodOptional<z.ZodBoolean>;
|
|
46
|
+
}, z.core.$strip>>;
|
|
47
|
+
type: z.ZodLiteral<"ocrStream">;
|
|
48
|
+
}, z.core.$strip>;
|
|
49
|
+
export declare const ocrTextBlockSchema: z.ZodObject<{
|
|
50
|
+
text: z.ZodString;
|
|
51
|
+
bbox: z.ZodOptional<z.ZodTuple<[z.ZodNumber, z.ZodNumber, z.ZodNumber, z.ZodNumber], null>>;
|
|
52
|
+
confidence: z.ZodOptional<z.ZodNumber>;
|
|
53
|
+
}, z.core.$strip>;
|
|
54
|
+
export declare const ocrStatsSchema: z.ZodObject<{
|
|
55
|
+
detectionTime: z.ZodOptional<z.ZodNumber>;
|
|
56
|
+
recognitionTime: z.ZodOptional<z.ZodNumber>;
|
|
57
|
+
totalTime: z.ZodOptional<z.ZodNumber>;
|
|
58
|
+
}, z.core.$strip>;
|
|
59
|
+
export declare const ocrStreamResponseSchema: z.ZodObject<{
|
|
60
|
+
type: z.ZodLiteral<"ocrStream">;
|
|
61
|
+
blocks: z.ZodOptional<z.ZodArray<z.ZodObject<{
|
|
62
|
+
text: z.ZodString;
|
|
63
|
+
bbox: z.ZodOptional<z.ZodTuple<[z.ZodNumber, z.ZodNumber, z.ZodNumber, z.ZodNumber], null>>;
|
|
64
|
+
confidence: z.ZodOptional<z.ZodNumber>;
|
|
65
|
+
}, z.core.$strip>>>;
|
|
66
|
+
done: z.ZodOptional<z.ZodBoolean>;
|
|
67
|
+
error: z.ZodOptional<z.ZodString>;
|
|
68
|
+
stats: z.ZodOptional<z.ZodObject<{
|
|
69
|
+
detectionTime: z.ZodOptional<z.ZodNumber>;
|
|
70
|
+
recognitionTime: z.ZodOptional<z.ZodNumber>;
|
|
71
|
+
totalTime: z.ZodOptional<z.ZodNumber>;
|
|
72
|
+
}, z.core.$strip>>;
|
|
73
|
+
}, z.core.$strip>;
|
|
74
|
+
export type OCRConfig = z.infer<typeof ocrConfigSchema>;
|
|
75
|
+
export type ImageInput = z.infer<typeof imageInputSchema>;
|
|
76
|
+
export type OCROptions = z.infer<typeof ocrOptionsSchema>;
|
|
77
|
+
export type OCRParams = z.infer<typeof ocrParamsSchema>;
|
|
78
|
+
export type OCRClientParams = {
|
|
79
|
+
modelId: string;
|
|
80
|
+
image: string | Buffer;
|
|
81
|
+
options?: OCROptions;
|
|
82
|
+
stream?: boolean;
|
|
83
|
+
};
|
|
84
|
+
export type OCRStreamRequest = z.infer<typeof ocrStreamRequestSchema>;
|
|
85
|
+
export type OCRStreamResponse = z.infer<typeof ocrStreamResponseSchema>;
|
|
86
|
+
export type OCRTextBlock = z.infer<typeof ocrTextBlockSchema>;
|
|
87
|
+
export type OCRStats = z.infer<typeof ocrStatsSchema>;
|
|
88
|
+
//# sourceMappingURL=ocr.d.ts.map
|
|
@@ -0,0 +1,52 @@
|
|
|
1
|
+
import { z } from "zod";
|
|
2
|
+
// Model config
|
|
3
|
+
export const ocrConfigSchema = z.object({
|
|
4
|
+
langList: z.array(z.string()).optional(),
|
|
5
|
+
useGPU: z.boolean().optional(),
|
|
6
|
+
timeout: z.number().optional(),
|
|
7
|
+
magRatio: z.number().optional(),
|
|
8
|
+
defaultRotationAngles: z.array(z.number()).optional(),
|
|
9
|
+
contrastRetry: z.boolean().optional(),
|
|
10
|
+
lowConfidenceThreshold: z.number().optional(),
|
|
11
|
+
recognizerBatchSize: z.number().optional(),
|
|
12
|
+
});
|
|
13
|
+
// Image input types
|
|
14
|
+
export const imageInputSchema = z.discriminatedUnion("type", [
|
|
15
|
+
z.object({
|
|
16
|
+
type: z.literal("base64"),
|
|
17
|
+
value: z.string(),
|
|
18
|
+
}),
|
|
19
|
+
z.object({
|
|
20
|
+
type: z.literal("filePath"),
|
|
21
|
+
value: z.string(),
|
|
22
|
+
}),
|
|
23
|
+
]);
|
|
24
|
+
// OCR options
|
|
25
|
+
export const ocrOptionsSchema = z.object({
|
|
26
|
+
paragraph: z.boolean().optional(),
|
|
27
|
+
});
|
|
28
|
+
export const ocrParamsSchema = z.object({
|
|
29
|
+
modelId: z.string(),
|
|
30
|
+
image: imageInputSchema,
|
|
31
|
+
options: ocrOptionsSchema.optional(),
|
|
32
|
+
});
|
|
33
|
+
export const ocrStreamRequestSchema = ocrParamsSchema.extend({
|
|
34
|
+
type: z.literal("ocrStream"),
|
|
35
|
+
});
|
|
36
|
+
export const ocrTextBlockSchema = z.object({
|
|
37
|
+
text: z.string(),
|
|
38
|
+
bbox: z.tuple([z.number(), z.number(), z.number(), z.number()]).optional(),
|
|
39
|
+
confidence: z.number().optional(),
|
|
40
|
+
});
|
|
41
|
+
export const ocrStatsSchema = z.object({
|
|
42
|
+
detectionTime: z.number().optional(),
|
|
43
|
+
recognitionTime: z.number().optional(),
|
|
44
|
+
totalTime: z.number().optional(),
|
|
45
|
+
});
|
|
46
|
+
export const ocrStreamResponseSchema = z.object({
|
|
47
|
+
type: z.literal("ocrStream"),
|
|
48
|
+
blocks: z.array(ocrTextBlockSchema).optional(),
|
|
49
|
+
done: z.boolean().optional(),
|
|
50
|
+
error: z.string().optional(),
|
|
51
|
+
stats: ocrStatsSchema.optional(),
|
|
52
|
+
});
|
package/dist/schemas/rag.d.ts
CHANGED
|
@@ -1,28 +1,16 @@
|
|
|
1
1
|
import { z } from "zod";
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
fulfilled: "fulfilled";
|
|
5
|
-
rejected: "rejected";
|
|
6
|
-
}>;
|
|
7
|
-
id: z.ZodOptional<z.ZodString>;
|
|
8
|
-
error: z.ZodOptional<z.ZodString>;
|
|
9
|
-
}, z.core.$strip>;
|
|
10
|
-
declare const searchResultSchema: z.ZodObject<{
|
|
11
|
-
id: z.ZodString;
|
|
12
|
-
content: z.ZodString;
|
|
13
|
-
score: z.ZodNumber;
|
|
14
|
-
}, z.core.$strip>;
|
|
15
|
-
declare const ragSaveEmbeddingsParamsSchema: z.ZodObject<{
|
|
16
|
-
modelId: z.ZodString;
|
|
2
|
+
import type { Doc, EmbeddedDoc, SaveEmbeddingsResult, SearchResult, ReindexResult, IngestResult, IngestStage, ReindexStage, SaveStage } from "@qvac/rag";
|
|
3
|
+
export declare const ragIngestParamsSchema: z.ZodObject<{
|
|
17
4
|
workspace: z.ZodOptional<z.ZodString>;
|
|
5
|
+
modelId: z.ZodString;
|
|
18
6
|
documents: z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodString>]>;
|
|
19
7
|
chunk: z.ZodDefault<z.ZodBoolean>;
|
|
20
8
|
chunkOpts: z.ZodOptional<z.ZodObject<{
|
|
21
9
|
chunkSize: z.ZodOptional<z.ZodNumber>;
|
|
22
10
|
chunkOverlap: z.ZodOptional<z.ZodNumber>;
|
|
23
11
|
chunkStrategy: z.ZodOptional<z.ZodEnum<{
|
|
24
|
-
character: "character";
|
|
25
12
|
paragraph: "paragraph";
|
|
13
|
+
character: "character";
|
|
26
14
|
}>>;
|
|
27
15
|
splitStrategy: z.ZodOptional<z.ZodEnum<{
|
|
28
16
|
token: "token";
|
|
@@ -32,18 +20,18 @@ declare const ragSaveEmbeddingsParamsSchema: z.ZodObject<{
|
|
|
32
20
|
line: "line";
|
|
33
21
|
}>>;
|
|
34
22
|
}, z.core.$strip>>;
|
|
23
|
+
progressInterval: z.ZodOptional<z.ZodNumber>;
|
|
24
|
+
onProgress: z.ZodOptional<z.ZodUnknown>;
|
|
25
|
+
withProgress: z.ZodOptional<z.ZodBoolean>;
|
|
35
26
|
}, z.core.$strip>;
|
|
36
|
-
export declare const
|
|
37
|
-
modelId: z.ZodString;
|
|
38
|
-
workspace: z.ZodOptional<z.ZodString>;
|
|
27
|
+
export declare const ragChunkParamsSchema: z.ZodObject<{
|
|
39
28
|
documents: z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodString>]>;
|
|
40
|
-
chunk: z.ZodDefault<z.ZodBoolean>;
|
|
41
29
|
chunkOpts: z.ZodOptional<z.ZodObject<{
|
|
42
30
|
chunkSize: z.ZodOptional<z.ZodNumber>;
|
|
43
31
|
chunkOverlap: z.ZodOptional<z.ZodNumber>;
|
|
44
32
|
chunkStrategy: z.ZodOptional<z.ZodEnum<{
|
|
45
|
-
character: "character";
|
|
46
33
|
paragraph: "paragraph";
|
|
34
|
+
character: "character";
|
|
47
35
|
}>>;
|
|
48
36
|
splitStrategy: z.ZodOptional<z.ZodEnum<{
|
|
49
37
|
token: "token";
|
|
@@ -53,48 +41,76 @@ export declare const ragSaveEmbeddingsOperationSchema: z.ZodObject<{
|
|
|
53
41
|
line: "line";
|
|
54
42
|
}>>;
|
|
55
43
|
}, z.core.$strip>>;
|
|
56
|
-
type: z.ZodLiteral<"rag">;
|
|
57
|
-
operation: z.ZodLiteral<"saveEmbeddings">;
|
|
58
44
|
}, z.core.$strip>;
|
|
59
|
-
declare const
|
|
60
|
-
modelId: z.ZodString
|
|
45
|
+
export declare const ragSaveEmbeddingsParamsSchema: z.ZodObject<{
|
|
46
|
+
modelId: z.ZodOptional<z.ZodString>;
|
|
61
47
|
workspace: z.ZodOptional<z.ZodString>;
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
|
|
48
|
+
documents: z.ZodArray<z.ZodObject<{
|
|
49
|
+
id: z.ZodString;
|
|
50
|
+
content: z.ZodString;
|
|
51
|
+
embedding: z.ZodArray<z.ZodNumber>;
|
|
52
|
+
embeddingModelId: z.ZodString;
|
|
53
|
+
metadata: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
|
|
54
|
+
}, z.core.$strip>>;
|
|
55
|
+
progressInterval: z.ZodOptional<z.ZodNumber>;
|
|
56
|
+
onProgress: z.ZodOptional<z.ZodUnknown>;
|
|
57
|
+
withProgress: z.ZodOptional<z.ZodBoolean>;
|
|
65
58
|
}, z.core.$strip>;
|
|
66
|
-
export declare const
|
|
67
|
-
modelId: z.ZodString
|
|
59
|
+
export declare const ragReindexParamsSchema: z.ZodObject<{
|
|
60
|
+
modelId: z.ZodOptional<z.ZodString>;
|
|
61
|
+
workspace: z.ZodOptional<z.ZodString>;
|
|
62
|
+
onProgress: z.ZodOptional<z.ZodUnknown>;
|
|
63
|
+
withProgress: z.ZodOptional<z.ZodBoolean>;
|
|
64
|
+
}, z.core.$strip>;
|
|
65
|
+
export declare const ragSearchParamsSchema: z.ZodObject<{
|
|
68
66
|
workspace: z.ZodOptional<z.ZodString>;
|
|
67
|
+
modelId: z.ZodString;
|
|
69
68
|
query: z.ZodString;
|
|
70
|
-
type: z.ZodLiteral<"rag">;
|
|
71
|
-
operation: z.ZodLiteral<"search">;
|
|
72
69
|
topK: z.ZodDefault<z.ZodNumber>;
|
|
73
70
|
n: z.ZodDefault<z.ZodNumber>;
|
|
74
71
|
}, z.core.$strip>;
|
|
75
|
-
declare const
|
|
76
|
-
modelId: z.ZodString
|
|
72
|
+
export declare const ragDeleteEmbeddingsParamsSchema: z.ZodObject<{
|
|
73
|
+
modelId: z.ZodOptional<z.ZodString>;
|
|
77
74
|
workspace: z.ZodOptional<z.ZodString>;
|
|
78
75
|
ids: z.ZodArray<z.ZodString>;
|
|
79
76
|
}, z.core.$strip>;
|
|
80
|
-
export declare const
|
|
81
|
-
modelId: z.ZodString;
|
|
77
|
+
export declare const ragCloseWorkspaceParamsSchema: z.ZodObject<{
|
|
82
78
|
workspace: z.ZodOptional<z.ZodString>;
|
|
83
|
-
|
|
84
|
-
|
|
85
|
-
|
|
79
|
+
deleteOnClose: z.ZodOptional<z.ZodBoolean>;
|
|
80
|
+
}, z.core.$strip>;
|
|
81
|
+
export declare const ragDeleteWorkspaceParamsSchema: z.ZodObject<{
|
|
82
|
+
workspace: z.ZodString;
|
|
86
83
|
}, z.core.$strip>;
|
|
87
84
|
export declare const ragRequestSchema: z.ZodDiscriminatedUnion<[z.ZodObject<{
|
|
88
|
-
|
|
85
|
+
documents: z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodString>]>;
|
|
86
|
+
chunkOpts: z.ZodOptional<z.ZodObject<{
|
|
87
|
+
chunkSize: z.ZodOptional<z.ZodNumber>;
|
|
88
|
+
chunkOverlap: z.ZodOptional<z.ZodNumber>;
|
|
89
|
+
chunkStrategy: z.ZodOptional<z.ZodEnum<{
|
|
90
|
+
paragraph: "paragraph";
|
|
91
|
+
character: "character";
|
|
92
|
+
}>>;
|
|
93
|
+
splitStrategy: z.ZodOptional<z.ZodEnum<{
|
|
94
|
+
token: "token";
|
|
95
|
+
character: "character";
|
|
96
|
+
word: "word";
|
|
97
|
+
sentence: "sentence";
|
|
98
|
+
line: "line";
|
|
99
|
+
}>>;
|
|
100
|
+
}, z.core.$strip>>;
|
|
101
|
+
type: z.ZodLiteral<"rag">;
|
|
102
|
+
operation: z.ZodLiteral<"chunk">;
|
|
103
|
+
}, z.core.$strip>, z.ZodObject<{
|
|
89
104
|
workspace: z.ZodOptional<z.ZodString>;
|
|
105
|
+
modelId: z.ZodString;
|
|
90
106
|
documents: z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodString>]>;
|
|
91
107
|
chunk: z.ZodDefault<z.ZodBoolean>;
|
|
92
108
|
chunkOpts: z.ZodOptional<z.ZodObject<{
|
|
93
109
|
chunkSize: z.ZodOptional<z.ZodNumber>;
|
|
94
110
|
chunkOverlap: z.ZodOptional<z.ZodNumber>;
|
|
95
111
|
chunkStrategy: z.ZodOptional<z.ZodEnum<{
|
|
96
|
-
character: "character";
|
|
97
112
|
paragraph: "paragraph";
|
|
113
|
+
character: "character";
|
|
98
114
|
}>>;
|
|
99
115
|
splitStrategy: z.ZodOptional<z.ZodEnum<{
|
|
100
116
|
token: "token";
|
|
@@ -104,28 +120,78 @@ export declare const ragRequestSchema: z.ZodDiscriminatedUnion<[z.ZodObject<{
|
|
|
104
120
|
line: "line";
|
|
105
121
|
}>>;
|
|
106
122
|
}, z.core.$strip>>;
|
|
123
|
+
progressInterval: z.ZodOptional<z.ZodNumber>;
|
|
124
|
+
onProgress: z.ZodOptional<z.ZodUnknown>;
|
|
125
|
+
withProgress: z.ZodOptional<z.ZodBoolean>;
|
|
126
|
+
type: z.ZodLiteral<"rag">;
|
|
127
|
+
operation: z.ZodLiteral<"ingest">;
|
|
128
|
+
}, z.core.$strip>, z.ZodObject<{
|
|
129
|
+
modelId: z.ZodOptional<z.ZodString>;
|
|
130
|
+
workspace: z.ZodOptional<z.ZodString>;
|
|
131
|
+
documents: z.ZodArray<z.ZodObject<{
|
|
132
|
+
id: z.ZodString;
|
|
133
|
+
content: z.ZodString;
|
|
134
|
+
embedding: z.ZodArray<z.ZodNumber>;
|
|
135
|
+
embeddingModelId: z.ZodString;
|
|
136
|
+
metadata: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
|
|
137
|
+
}, z.core.$strip>>;
|
|
138
|
+
progressInterval: z.ZodOptional<z.ZodNumber>;
|
|
139
|
+
onProgress: z.ZodOptional<z.ZodUnknown>;
|
|
140
|
+
withProgress: z.ZodOptional<z.ZodBoolean>;
|
|
107
141
|
type: z.ZodLiteral<"rag">;
|
|
108
142
|
operation: z.ZodLiteral<"saveEmbeddings">;
|
|
109
143
|
}, z.core.$strip>, z.ZodObject<{
|
|
110
|
-
modelId: z.ZodString;
|
|
111
144
|
workspace: z.ZodOptional<z.ZodString>;
|
|
145
|
+
modelId: z.ZodString;
|
|
112
146
|
query: z.ZodString;
|
|
113
|
-
type: z.ZodLiteral<"rag">;
|
|
114
|
-
operation: z.ZodLiteral<"search">;
|
|
115
147
|
topK: z.ZodDefault<z.ZodNumber>;
|
|
116
148
|
n: z.ZodDefault<z.ZodNumber>;
|
|
149
|
+
type: z.ZodLiteral<"rag">;
|
|
150
|
+
operation: z.ZodLiteral<"search">;
|
|
117
151
|
}, z.core.$strip>, z.ZodObject<{
|
|
118
|
-
modelId: z.ZodString
|
|
152
|
+
modelId: z.ZodOptional<z.ZodString>;
|
|
119
153
|
workspace: z.ZodOptional<z.ZodString>;
|
|
120
154
|
ids: z.ZodArray<z.ZodString>;
|
|
121
155
|
type: z.ZodLiteral<"rag">;
|
|
122
156
|
operation: z.ZodLiteral<"deleteEmbeddings">;
|
|
157
|
+
}, z.core.$strip>, z.ZodObject<{
|
|
158
|
+
modelId: z.ZodOptional<z.ZodString>;
|
|
159
|
+
workspace: z.ZodOptional<z.ZodString>;
|
|
160
|
+
onProgress: z.ZodOptional<z.ZodUnknown>;
|
|
161
|
+
withProgress: z.ZodOptional<z.ZodBoolean>;
|
|
162
|
+
type: z.ZodLiteral<"rag">;
|
|
163
|
+
operation: z.ZodLiteral<"reindex">;
|
|
164
|
+
}, z.core.$strip>, z.ZodObject<{
|
|
165
|
+
type: z.ZodLiteral<"rag">;
|
|
166
|
+
operation: z.ZodLiteral<"listWorkspaces">;
|
|
167
|
+
}, z.core.$strip>, z.ZodObject<{
|
|
168
|
+
workspace: z.ZodOptional<z.ZodString>;
|
|
169
|
+
deleteOnClose: z.ZodOptional<z.ZodBoolean>;
|
|
170
|
+
type: z.ZodLiteral<"rag">;
|
|
171
|
+
operation: z.ZodLiteral<"closeWorkspace">;
|
|
172
|
+
}, z.core.$strip>, z.ZodObject<{
|
|
173
|
+
workspace: z.ZodString;
|
|
174
|
+
type: z.ZodLiteral<"rag">;
|
|
175
|
+
operation: z.ZodLiteral<"deleteWorkspace">;
|
|
123
176
|
}, z.core.$strip>], "operation">;
|
|
177
|
+
declare const ragWorkspaceInfoSchema: z.ZodObject<{
|
|
178
|
+
name: z.ZodString;
|
|
179
|
+
open: z.ZodBoolean;
|
|
180
|
+
}, z.core.$strip>;
|
|
124
181
|
export declare const ragResponseSchema: z.ZodDiscriminatedUnion<[z.ZodObject<{
|
|
125
182
|
type: z.ZodLiteral<"rag">;
|
|
126
183
|
success: z.ZodBoolean;
|
|
127
184
|
error: z.ZodOptional<z.ZodString>;
|
|
128
|
-
operation: z.ZodLiteral<"
|
|
185
|
+
operation: z.ZodLiteral<"chunk">;
|
|
186
|
+
chunks: z.ZodArray<z.ZodObject<{
|
|
187
|
+
id: z.ZodString;
|
|
188
|
+
content: z.ZodString;
|
|
189
|
+
}, z.core.$strip>>;
|
|
190
|
+
}, z.core.$strip>, z.ZodObject<{
|
|
191
|
+
type: z.ZodLiteral<"rag">;
|
|
192
|
+
success: z.ZodBoolean;
|
|
193
|
+
error: z.ZodOptional<z.ZodString>;
|
|
194
|
+
operation: z.ZodLiteral<"ingest">;
|
|
129
195
|
processed: z.ZodArray<z.ZodObject<{
|
|
130
196
|
status: z.ZodEnum<{
|
|
131
197
|
fulfilled: "fulfilled";
|
|
@@ -135,6 +201,19 @@ export declare const ragResponseSchema: z.ZodDiscriminatedUnion<[z.ZodObject<{
|
|
|
135
201
|
error: z.ZodOptional<z.ZodString>;
|
|
136
202
|
}, z.core.$strip>>;
|
|
137
203
|
droppedIndices: z.ZodArray<z.ZodNumber>;
|
|
204
|
+
}, z.core.$strip>, z.ZodObject<{
|
|
205
|
+
type: z.ZodLiteral<"rag">;
|
|
206
|
+
success: z.ZodBoolean;
|
|
207
|
+
error: z.ZodOptional<z.ZodString>;
|
|
208
|
+
operation: z.ZodLiteral<"saveEmbeddings">;
|
|
209
|
+
processed: z.ZodArray<z.ZodObject<{
|
|
210
|
+
status: z.ZodEnum<{
|
|
211
|
+
fulfilled: "fulfilled";
|
|
212
|
+
rejected: "rejected";
|
|
213
|
+
}>;
|
|
214
|
+
id: z.ZodOptional<z.ZodString>;
|
|
215
|
+
error: z.ZodOptional<z.ZodString>;
|
|
216
|
+
}, z.core.$strip>>;
|
|
138
217
|
}, z.core.$strip>, z.ZodObject<{
|
|
139
218
|
type: z.ZodLiteral<"rag">;
|
|
140
219
|
success: z.ZodBoolean;
|
|
@@ -150,13 +229,65 @@ export declare const ragResponseSchema: z.ZodDiscriminatedUnion<[z.ZodObject<{
|
|
|
150
229
|
success: z.ZodBoolean;
|
|
151
230
|
error: z.ZodOptional<z.ZodString>;
|
|
152
231
|
operation: z.ZodLiteral<"deleteEmbeddings">;
|
|
232
|
+
}, z.core.$strip>, z.ZodObject<{
|
|
233
|
+
type: z.ZodLiteral<"rag">;
|
|
234
|
+
success: z.ZodBoolean;
|
|
235
|
+
error: z.ZodOptional<z.ZodString>;
|
|
236
|
+
operation: z.ZodLiteral<"reindex">;
|
|
237
|
+
result: z.ZodObject<{
|
|
238
|
+
reindexed: z.ZodBoolean;
|
|
239
|
+
details: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
|
|
240
|
+
}, z.core.$strip>;
|
|
241
|
+
}, z.core.$strip>, z.ZodObject<{
|
|
242
|
+
type: z.ZodLiteral<"rag">;
|
|
243
|
+
success: z.ZodBoolean;
|
|
244
|
+
error: z.ZodOptional<z.ZodString>;
|
|
245
|
+
operation: z.ZodLiteral<"listWorkspaces">;
|
|
246
|
+
workspaces: z.ZodArray<z.ZodObject<{
|
|
247
|
+
name: z.ZodString;
|
|
248
|
+
open: z.ZodBoolean;
|
|
249
|
+
}, z.core.$strip>>;
|
|
250
|
+
}, z.core.$strip>, z.ZodObject<{
|
|
251
|
+
type: z.ZodLiteral<"rag">;
|
|
252
|
+
success: z.ZodBoolean;
|
|
253
|
+
error: z.ZodOptional<z.ZodString>;
|
|
254
|
+
operation: z.ZodLiteral<"closeWorkspace">;
|
|
255
|
+
}, z.core.$strip>, z.ZodObject<{
|
|
256
|
+
type: z.ZodLiteral<"rag">;
|
|
257
|
+
success: z.ZodBoolean;
|
|
258
|
+
error: z.ZodOptional<z.ZodString>;
|
|
259
|
+
operation: z.ZodLiteral<"deleteWorkspace">;
|
|
153
260
|
}, z.core.$strip>], "operation">;
|
|
261
|
+
export declare const ragProgressUpdateSchema: z.ZodObject<{
|
|
262
|
+
type: z.ZodLiteral<"rag:progress">;
|
|
263
|
+
operation: z.ZodEnum<{
|
|
264
|
+
ingest: "ingest";
|
|
265
|
+
saveEmbeddings: "saveEmbeddings";
|
|
266
|
+
reindex: "reindex";
|
|
267
|
+
}>;
|
|
268
|
+
workspace: z.ZodString;
|
|
269
|
+
stage: z.ZodString;
|
|
270
|
+
current: z.ZodNumber;
|
|
271
|
+
total: z.ZodNumber;
|
|
272
|
+
timestamp: z.ZodNumber;
|
|
273
|
+
}, z.core.$strip>;
|
|
154
274
|
export type RagRequest = z.infer<typeof ragRequestSchema>;
|
|
155
275
|
export type RagResponse = z.infer<typeof ragResponseSchema>;
|
|
156
|
-
export type
|
|
157
|
-
export type
|
|
158
|
-
export type
|
|
159
|
-
|
|
160
|
-
|
|
161
|
-
export {
|
|
276
|
+
export type RagProgressUpdate = z.infer<typeof ragProgressUpdateSchema>;
|
|
277
|
+
export type RagChunkParams = z.infer<typeof ragChunkParamsSchema>;
|
|
278
|
+
export type RagIngestParams = z.input<typeof ragIngestParamsSchema> & {
|
|
279
|
+
onProgress?: (stage: IngestStage, current: number, total: number) => void;
|
|
280
|
+
};
|
|
281
|
+
export type RagSaveEmbeddingsParams = z.infer<typeof ragSaveEmbeddingsParamsSchema> & {
|
|
282
|
+
onProgress?: (stage: SaveStage, current: number, total: number) => void;
|
|
283
|
+
};
|
|
284
|
+
export type RagSearchParams = z.input<typeof ragSearchParamsSchema>;
|
|
285
|
+
export type RagDeleteEmbeddingsParams = z.infer<typeof ragDeleteEmbeddingsParamsSchema>;
|
|
286
|
+
export type RagReindexParams = z.input<typeof ragReindexParamsSchema> & {
|
|
287
|
+
onProgress?: (stage: ReindexStage, current: number, total: number) => void;
|
|
288
|
+
};
|
|
289
|
+
export type RagCloseWorkspaceParams = z.infer<typeof ragCloseWorkspaceParamsSchema>;
|
|
290
|
+
export type RagDeleteWorkspaceParams = z.infer<typeof ragDeleteWorkspaceParamsSchema>;
|
|
291
|
+
export type RagWorkspaceInfo = z.infer<typeof ragWorkspaceInfoSchema>;
|
|
292
|
+
export type { Doc as RagDoc, EmbeddedDoc as RagEmbeddedDoc, SaveEmbeddingsResult as RagSaveEmbeddingsResult, SearchResult as RagSearchResult, ReindexResult as RagReindexResult, IngestStage as RagIngestStage, ReindexStage as RagReindexStage, SaveStage as RagSaveStage, IngestResult as RagIngestResult, };
|
|
162
293
|
//# sourceMappingURL=rag.d.ts.map
|