@qvac/sdk 0.5.0 ā 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 +48 -11
- 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/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 -1
- package/dist/client/api/index.js +2 -1
- package/dist/client/api/load-model.js +6 -1
- 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 +1 -1
- package/dist/client/init-hooks.d.ts +4 -3
- package/dist/client/init-hooks.js +40 -31
- package/dist/client/rpc/bare-client.d.ts +1 -178
- package/dist/client/rpc/bare-client.js +42 -9
- package/dist/client/rpc/expo-rpc-client.js +57 -2
- package/dist/client/rpc/node-rpc-client.js +5 -1
- package/dist/client/rpc/rpc-client.d.ts +1 -1
- package/dist/client/rpc/rpc-client.js +11 -57
- 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/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 +4 -3
- package/dist/examples/embed-hyperdrive.js +4 -0
- package/dist/examples/kv-cache-custom-key.js +16 -1
- package/dist/examples/llamacpp-sharded.js +12 -3
- 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 +3 -2
- package/dist/index.js +3 -1
- package/dist/logging/addon.js +23 -7
- package/dist/logging/base-logger.js +2 -2
- package/dist/logging/client-logger.js +15 -4
- package/dist/logging/index.d.ts +1 -0
- package/dist/logging/index.js +1 -0
- package/dist/logging/logger.js +11 -3
- package/dist/logging/namespaces.d.ts +1 -0
- package/dist/logging/namespaces.js +1 -0
- package/dist/logging/registry.d.ts +5 -0
- package/dist/logging/registry.js +17 -5
- package/dist/logging/transport.js +2 -2
- package/dist/logging/utils.d.ts +2 -0
- package/dist/logging/utils.js +62 -0
- package/dist/models/hyperdrive/models.d.ts +1096 -178
- package/dist/models/hyperdrive/models.js +1476 -507
- 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 +582 -77
- package/dist/schemas/common.js +6 -1
- 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/get-model-info.d.ts +12 -10
- package/dist/schemas/get-model-info.js +9 -1
- package/dist/schemas/index.d.ts +6 -2
- package/dist/schemas/index.js +9 -2
- package/dist/schemas/llamacpp-config.d.ts +112 -8
- package/dist/schemas/llamacpp-config.js +40 -8
- package/dist/schemas/load-model.d.ts +2993 -525
- package/dist/schemas/load-model.js +88 -33
- package/dist/schemas/logging-stream.d.ts +2 -2
- package/dist/schemas/model-src-utils.d.ts +38 -16
- package/dist/schemas/model-src-utils.js +4 -8
- 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 +20 -8
- package/dist/schemas/reload-config.js +7 -5
- package/dist/schemas/runtime-context.d.ts +19 -0
- package/dist/schemas/runtime-context.js +7 -0
- package/dist/schemas/sdk-config.d.ts +697 -2
- package/dist/schemas/sdk-config.js +99 -1
- package/dist/schemas/sdk-errors-client.d.ts +1 -0
- package/dist/schemas/sdk-errors-client.js +5 -0
- package/dist/schemas/sdk-errors-server.d.ts +14 -0
- package/dist/schemas/sdk-errors-server.js +70 -0
- package/dist/schemas/shard.d.ts +21 -0
- package/dist/schemas/shard.js +17 -0
- package/dist/schemas/translate.d.ts +68 -23
- package/dist/schemas/translate.js +37 -11
- package/dist/schemas/translation-config.d.ts +272 -29
- package/dist/schemas/translation-config.js +63 -6
- 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 +3 -26
- 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 +43 -10
- 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/index.d.ts +3 -0
- package/dist/server/bare/addons/llamacpp-embedding/index.js +2 -0
- package/dist/server/bare/addons/load-model.js +28 -24
- 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/{whispercpp ā whispercpp-transcription}/create-model.js +2 -1
- package/dist/server/bare/delegate-rpc-client.js +19 -1
- 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.js +11 -0
- package/dist/server/bare/registry/logging-stream-registry.js +3 -3
- 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 -29
- 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/handle-request.js +91 -9
- package/dist/server/rpc/handlers/cancelHandler.js +10 -1
- package/dist/server/rpc/handlers/completion-stream.js +1 -1
- package/dist/server/rpc/handlers/delete-cache.js +1 -1
- package/dist/server/rpc/handlers/embed.js +1 -1
- 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 -1
- package/dist/server/rpc/handlers/index.js +2 -0
- package/dist/server/rpc/handlers/load-model/download-manager.d.ts +11 -11
- package/dist/server/rpc/handlers/load-model/download-manager.js +36 -26
- 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 +342 -33
- package/dist/server/rpc/handlers/load-model/hyperdrive.js +5 -3
- 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 +32 -3
- package/dist/server/rpc/handlers/load-model-delegated.js +6 -3
- 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/rag.d.ts +2 -2
- package/dist/server/rpc/handlers/rag.js +107 -43
- package/dist/server/rpc/handlers/text-to-speech.js +1 -1
- package/dist/server/rpc/handlers/transcribe-stream.js +1 -1
- package/dist/server/rpc/handlers/translate.js +6 -15
- package/dist/server/utils/archive.d.ts +3 -0
- package/dist/server/utils/archive.js +171 -0
- package/dist/server/utils/audio/decoder.js +2 -2
- package/dist/server/utils/cache.d.ts +5 -5
- package/dist/server/utils/cache.js +6 -6
- package/dist/server/utils/config-transform.d.ts +2 -2
- package/dist/server/utils/config-transform.js +8 -6
- 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 +3 -1
- package/dist/server/utils/index.js +3 -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 +8 -3
- package/dist/utils/errors-base.d.ts +1 -1
- package/dist/utils/errors-base.js +2 -2
- package/dist/utils/errors-client.d.ts +12 -0
- package/dist/utils/errors-client.js +20 -0
- package/dist/utils/errors-server.d.ts +33 -0
- package/dist/utils/errors-server.js +55 -0
- package/package.json +40 -15
- package/dist/client/rpc/bare-rpc-client.d.ts +0 -4
- package/dist/client/rpc/bare-rpc-client.js +0 -94
- package/dist/examples/config/config-path.d.ts +0 -8
- package/dist/examples/config/config-path.js +0 -13
- 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/server/bare/addons/llamacpp/index.d.ts +0 -4
- package/dist/server/bare/addons/llamacpp/index.js +0 -3
- package/dist/server/bare/rag-hyperdb/rag-manager.d.ts +0 -5
- package/dist/server/bare/rag-hyperdb/rag-manager.js +0 -66
- package/dist/server/rpc/handlers/load-model.js +0 -111
- package/dist/server/utils/hash.d.ts +0 -5
- package/dist/server/utils/hash.js +0 -11
- package/dist/utils/getProcessSafe.d.ts +0 -7
- package/dist/utils/getProcessSafe.js +0 -17
- package/dist/utils/runtime.d.ts +0 -30
- package/dist/utils/runtime.js +0 -48
- /package/dist/server/bare/addons/{llamacpp ā llamacpp-completion}/handlers/completion-stream.d.ts +0 -0
- /package/dist/server/bare/addons/{llamacpp ā llamacpp-embedding}/handlers/embed.d.ts +0 -0
- /package/dist/server/bare/addons/{llamacpp ā llamacpp-embedding}/handlers/embed.js +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}/handlers/transcribe-stream.js +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
|
@@ -1,7 +1,13 @@
|
|
|
1
|
-
import { completion, loadModel, unloadModel, VERBOSITY
|
|
1
|
+
import { completion, loadModel, unloadModel, VERBOSITY } from "@qvac/sdk";
|
|
2
|
+
// Sharded models can be loaded from:
|
|
3
|
+
// 1. HTTP archives: "https://example.com/model.tar.gz"
|
|
4
|
+
// 2. HTTP pattern: "https://example.com/model-00001-of-00005.gguf"
|
|
5
|
+
// 3. Hyperdrive: use any sharded model source/constant, eg: LLAMA_3_2_1B_INST_Q4_0_SHARD
|
|
6
|
+
// 4. Local filesystem: pass the path to the first shard file (Note: All shards must be in the same directory)
|
|
7
|
+
// 5. Local archive: pass the path to the archive file (.tar, .tar.gz, .tgz)
|
|
2
8
|
try {
|
|
3
9
|
const modelId = await loadModel({
|
|
4
|
-
modelSrc:
|
|
10
|
+
modelSrc: "https://huggingface.co/Qwen/Qwen2.5-Coder-7B-Instruct-GGUF/resolve/main/qwen2.5-coder-7b-instruct-q4_0-00001-of-00002.gguf",
|
|
5
11
|
modelType: "llm",
|
|
6
12
|
modelConfig: {
|
|
7
13
|
device: "gpu",
|
|
@@ -9,14 +15,17 @@ try {
|
|
|
9
15
|
verbosity: VERBOSITY.ERROR,
|
|
10
16
|
},
|
|
11
17
|
onProgress: (progress) => {
|
|
18
|
+
// For sharded models, progress.shardInfo contains detailed progress for both
|
|
19
|
+
// individual shards AND overall download progress across all shards
|
|
12
20
|
if (progress.shardInfo) {
|
|
21
|
+
// For pattern-based or Hyperdrive shards
|
|
13
22
|
const { shardInfo } = progress;
|
|
14
23
|
console.log(`š„ Downloading ${shardInfo.shardName} (${shardInfo.currentShard}/${shardInfo.totalShards})\n` +
|
|
15
24
|
` File: ${progress.percentage.toFixed(1)}% (${(progress.downloaded / 1024 / 1024).toFixed(2)}MB / ${(progress.total / 1024 / 1024).toFixed(2)}MB)\n` +
|
|
16
25
|
` Overall: ${shardInfo.overallPercentage.toFixed(1)}% (${(shardInfo.overallDownloaded / 1024 / 1024).toFixed(2)}MB / ${(shardInfo.overallTotal / 1024 / 1024).toFixed(2)}MB)`);
|
|
17
26
|
}
|
|
18
27
|
else {
|
|
19
|
-
//
|
|
28
|
+
// For archive-based shards
|
|
20
29
|
console.log(`š„ Progress: ${progress.percentage.toFixed(1)}% ` +
|
|
21
30
|
`(${(progress.downloaded / 1024 / 1024).toFixed(2)}MB / ${(progress.total / 1024 / 1024).toFixed(2)}MB)`);
|
|
22
31
|
}
|
|
@@ -0,0 +1,52 @@
|
|
|
1
|
+
import { close, loadModel, ocr, OCR_CRAFT_LATIN_RECOGNIZER_1, unloadModel, } from "@qvac/sdk";
|
|
2
|
+
import path from "path";
|
|
3
|
+
import { fileURLToPath } from "url";
|
|
4
|
+
const __dirname = path.dirname(fileURLToPath(import.meta.url));
|
|
5
|
+
const imagePath = process.argv[2] || path.join(__dirname, "image/basic_test.bmp");
|
|
6
|
+
try {
|
|
7
|
+
console.log("š Loading OCR model...");
|
|
8
|
+
// Only need to pass the recognizer - detector is auto-derived from same hyperdrive key
|
|
9
|
+
const modelId = await loadModel({
|
|
10
|
+
modelSrc: OCR_CRAFT_LATIN_RECOGNIZER_1,
|
|
11
|
+
modelType: "ocr",
|
|
12
|
+
modelConfig: {
|
|
13
|
+
langList: ["en"],
|
|
14
|
+
useGPU: true,
|
|
15
|
+
timeout: 30000,
|
|
16
|
+
magRatio: 1.5,
|
|
17
|
+
defaultRotationAngles: [90, 180, 270],
|
|
18
|
+
contrastRetry: false,
|
|
19
|
+
lowConfidenceThreshold: 0.5,
|
|
20
|
+
recognizerBatchSize: 1,
|
|
21
|
+
},
|
|
22
|
+
});
|
|
23
|
+
console.log(`ā
Model loaded successfully! Model ID: ${modelId}`);
|
|
24
|
+
console.log(`\nš Running OCR on: ${imagePath}`);
|
|
25
|
+
const { blocks } = ocr({
|
|
26
|
+
modelId,
|
|
27
|
+
image: imagePath,
|
|
28
|
+
options: {
|
|
29
|
+
paragraph: false,
|
|
30
|
+
},
|
|
31
|
+
});
|
|
32
|
+
const result = await blocks;
|
|
33
|
+
console.log("\nš OCR Results:");
|
|
34
|
+
console.log("================================");
|
|
35
|
+
for (const block of result) {
|
|
36
|
+
console.log(`\nš Text: ${block.text}`);
|
|
37
|
+
if (block.bbox) {
|
|
38
|
+
console.log(` š BBox: [${block.bbox.join(", ")}]`);
|
|
39
|
+
}
|
|
40
|
+
if (block.confidence !== undefined) {
|
|
41
|
+
console.log(` ā Confidence: ${block.confidence}`);
|
|
42
|
+
}
|
|
43
|
+
}
|
|
44
|
+
console.log("\n================================");
|
|
45
|
+
console.log("\nš Unloading model...");
|
|
46
|
+
await unloadModel({ modelId, clearStorage: false });
|
|
47
|
+
console.log("ā
Model unloaded successfully.");
|
|
48
|
+
}
|
|
49
|
+
catch (error) {
|
|
50
|
+
console.error("ā Error during OCR processing:", error);
|
|
51
|
+
close();
|
|
52
|
+
}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
import { loadModel, LLAMA_3_2_1B_INST_Q4_0, completion, unloadModel, } from "@qvac/sdk";
|
|
2
|
+
try {
|
|
3
|
+
// Load a model into memory
|
|
4
|
+
const modelId = await loadModel({
|
|
5
|
+
modelSrc: LLAMA_3_2_1B_INST_Q4_0,
|
|
6
|
+
modelType: "llm",
|
|
7
|
+
onProgress: (progress) => {
|
|
8
|
+
console.log(progress);
|
|
9
|
+
},
|
|
10
|
+
});
|
|
11
|
+
// You can use the loaded model multiple times
|
|
12
|
+
const history = [
|
|
13
|
+
{
|
|
14
|
+
role: "user",
|
|
15
|
+
content: "Explain quantum computing in one sentence",
|
|
16
|
+
},
|
|
17
|
+
];
|
|
18
|
+
const result = completion({ modelId, history, stream: true });
|
|
19
|
+
for await (const token of result.tokenStream) {
|
|
20
|
+
process.stdout.write(token);
|
|
21
|
+
}
|
|
22
|
+
// Unload model to free up system resources
|
|
23
|
+
await unloadModel({ modelId });
|
|
24
|
+
}
|
|
25
|
+
catch (error) {
|
|
26
|
+
console.error("ā Error:", error);
|
|
27
|
+
process.exit(1);
|
|
28
|
+
}
|
|
@@ -0,0 +1,75 @@
|
|
|
1
|
+
import { loadModel, unloadModel, GTE_LARGE_FP16, ragIngest, ragCloseWorkspace, cancel, } from "@qvac/sdk";
|
|
2
|
+
const WORKSPACE = "cancellation-demo";
|
|
3
|
+
// Generate sample documents
|
|
4
|
+
function generateDocuments(count) {
|
|
5
|
+
const topics = [
|
|
6
|
+
"Machine learning algorithms process data to identify patterns and make decisions.",
|
|
7
|
+
"Deep neural networks consist of multiple layers that extract higher-level features.",
|
|
8
|
+
"Natural language processing enables computers to understand human language.",
|
|
9
|
+
"Computer vision systems analyze visual information from images and videos.",
|
|
10
|
+
"Reinforcement learning trains agents through trial and error using rewards.",
|
|
11
|
+
];
|
|
12
|
+
return Array.from({ length: count }, (_, i) => {
|
|
13
|
+
const topic = topics[i % topics.length];
|
|
14
|
+
return `Document ${i + 1}: ${topic} Extended content for document ${i + 1}.`;
|
|
15
|
+
});
|
|
16
|
+
}
|
|
17
|
+
try {
|
|
18
|
+
console.log("RAG Cancellation Example\n");
|
|
19
|
+
// Load embedding model
|
|
20
|
+
console.log("Loading embedding model...");
|
|
21
|
+
const modelId = await loadModel({
|
|
22
|
+
modelSrc: GTE_LARGE_FP16,
|
|
23
|
+
modelType: "embeddings",
|
|
24
|
+
onProgress: (progress) => {
|
|
25
|
+
process.stdout.write(`\r ${progress.percentage.toFixed(1)}%`);
|
|
26
|
+
},
|
|
27
|
+
});
|
|
28
|
+
console.log("\nā
Model loaded\n");
|
|
29
|
+
// Generate documents
|
|
30
|
+
const documents = generateDocuments(200);
|
|
31
|
+
console.log(`š Processing ${documents.length} documents...\n`);
|
|
32
|
+
let progressCount = 0;
|
|
33
|
+
let cancelled = false;
|
|
34
|
+
try {
|
|
35
|
+
await ragIngest({
|
|
36
|
+
modelId,
|
|
37
|
+
workspace: WORKSPACE,
|
|
38
|
+
documents,
|
|
39
|
+
progressInterval: 50,
|
|
40
|
+
onProgress: (stage, current, total) => {
|
|
41
|
+
progressCount++;
|
|
42
|
+
const pct = total > 0 ? Math.round((current / total) * 100) : 0;
|
|
43
|
+
console.log(` [${stage}] ${current}/${total} (${pct}%)`);
|
|
44
|
+
// Cancel during embedding stage after a few updates
|
|
45
|
+
if (!cancelled && stage === "embedding" && current > 10) {
|
|
46
|
+
console.log("\nš Triggering cancellation...\n");
|
|
47
|
+
cancelled = true;
|
|
48
|
+
void cancel({ operation: "rag", workspace: WORKSPACE });
|
|
49
|
+
}
|
|
50
|
+
},
|
|
51
|
+
});
|
|
52
|
+
console.log("\nā ļø Ingest completed (cancellation didn't interrupt in time)");
|
|
53
|
+
}
|
|
54
|
+
catch (error) {
|
|
55
|
+
const msg = error instanceof Error ? error.message : String(error);
|
|
56
|
+
const wasCancelled = msg.toLowerCase().includes("cancel") ||
|
|
57
|
+
msg.toLowerCase().includes("abort");
|
|
58
|
+
if (wasCancelled) {
|
|
59
|
+
console.log("ā
Operation cancelled successfully!");
|
|
60
|
+
console.log(` Progress updates received: ${progressCount}`);
|
|
61
|
+
}
|
|
62
|
+
else {
|
|
63
|
+
throw error;
|
|
64
|
+
}
|
|
65
|
+
}
|
|
66
|
+
// Cleanup (workspace may not exist if cancelled early)
|
|
67
|
+
await ragCloseWorkspace({ workspace: WORKSPACE, deleteOnClose: true }).catch(() => { });
|
|
68
|
+
await unloadModel({ modelId });
|
|
69
|
+
console.log("ā
Done");
|
|
70
|
+
process.exit(0);
|
|
71
|
+
}
|
|
72
|
+
catch (error) {
|
|
73
|
+
console.error("ā Error:", error);
|
|
74
|
+
process.exit(1);
|
|
75
|
+
}
|
|
@@ -1,7 +1,8 @@
|
|
|
1
|
-
import { loadModel, unloadModel, GTE_LARGE_FP16,
|
|
1
|
+
import { loadModel, unloadModel, GTE_LARGE_FP16, ragIngest, ragSearch, ragCloseWorkspace, } from "@qvac/sdk";
|
|
2
2
|
try {
|
|
3
3
|
// Get query from command line or use default
|
|
4
4
|
const query = process.argv[2] || "machine learning algorithms";
|
|
5
|
+
const workspace = "ingest-example";
|
|
5
6
|
console.log(`š Query: "${query}"`);
|
|
6
7
|
const modelId = await loadModel({
|
|
7
8
|
modelSrc: GTE_LARGE_FP16,
|
|
@@ -20,16 +21,18 @@ try {
|
|
|
20
21
|
"Cloud computing delivers computing services over the internet, allowing users to access resources like storage, processing power, and applications on-demand from anywhere.",
|
|
21
22
|
"Cybersecurity protects digital systems, networks, and data from malicious attacks, unauthorized access, and various forms of cyber threats through multiple layers of defense.",
|
|
22
23
|
];
|
|
23
|
-
console.log("š
|
|
24
|
-
const
|
|
24
|
+
console.log("š Ingesting documents...");
|
|
25
|
+
const result = await ragIngest({
|
|
25
26
|
modelId,
|
|
27
|
+
workspace,
|
|
26
28
|
documents: samples,
|
|
27
29
|
chunk: false,
|
|
28
30
|
});
|
|
29
|
-
console.log(
|
|
31
|
+
console.log(`ā
Ingested ${result.processed.length} documents`);
|
|
30
32
|
console.log("š Searching for similar documents...");
|
|
31
33
|
const results = await ragSearch({
|
|
32
34
|
modelId,
|
|
35
|
+
workspace,
|
|
33
36
|
query,
|
|
34
37
|
topK: 3,
|
|
35
38
|
});
|
|
@@ -41,6 +44,9 @@ try {
|
|
|
41
44
|
console.log("=".repeat(100));
|
|
42
45
|
console.log();
|
|
43
46
|
});
|
|
47
|
+
// Cleanup: close and delete workspace
|
|
48
|
+
await ragCloseWorkspace({ workspace, deleteOnClose: true });
|
|
49
|
+
console.log(`\nšļø Deleted '${workspace}' workspace`);
|
|
44
50
|
await unloadModel({ modelId });
|
|
45
51
|
}
|
|
46
52
|
catch (error) {
|
|
@@ -0,0 +1,74 @@
|
|
|
1
|
+
import { loadModel, unloadModel, embed, GTE_LARGE_FP16, ragChunk, ragSaveEmbeddings, ragSearch, ragCloseWorkspace, } from "@qvac/sdk";
|
|
2
|
+
try {
|
|
3
|
+
// Get query from command line or use default
|
|
4
|
+
const query = process.argv[2] || "machine learning algorithms";
|
|
5
|
+
console.log("š§ RAG Pipeline Example (Segregated Flow)\n");
|
|
6
|
+
console.log(`š Query: "${query}"`);
|
|
7
|
+
const workspace = "pipeline-example";
|
|
8
|
+
const modelId = await loadModel({
|
|
9
|
+
modelSrc: GTE_LARGE_FP16,
|
|
10
|
+
modelType: "embeddings",
|
|
11
|
+
onProgress: (progress) => {
|
|
12
|
+
console.log(`Loading model... ${progress.percentage.toFixed(1)}%`);
|
|
13
|
+
},
|
|
14
|
+
});
|
|
15
|
+
const samples = [
|
|
16
|
+
"Machine learning is a subset of artificial intelligence that focuses on algorithms that can learn and make predictions from data without being explicitly programmed for every task.",
|
|
17
|
+
"Deep learning uses neural networks with multiple layers to process and learn from complex data patterns, enabling breakthroughs in image recognition and natural language processing.",
|
|
18
|
+
"Natural language processing combines computational linguistics with machine learning to help computers understand, interpret, and generate human language in a meaningful way.",
|
|
19
|
+
"Computer vision enables machines to interpret and understand visual information from the world, using techniques like image classification, object detection, and facial recognition.",
|
|
20
|
+
"Quantum computing leverages quantum mechanical phenomena to process information in fundamentally different ways than classical computers, potentially solving certain problems exponentially faster.",
|
|
21
|
+
"Blockchain technology creates decentralized, immutable ledgers that enable secure peer-to-peer transactions without requiring a central authority or intermediary.",
|
|
22
|
+
"Cloud computing delivers computing services over the internet, allowing users to access resources like storage, processing power, and applications on-demand from anywhere.",
|
|
23
|
+
"Cybersecurity protects digital systems, networks, and data from malicious attacks, unauthorized access, and various forms of cyber threats through multiple layers of defense.",
|
|
24
|
+
];
|
|
25
|
+
console.log("\nš Step 1: Chunking documents...");
|
|
26
|
+
const chunks = await ragChunk({
|
|
27
|
+
documents: samples,
|
|
28
|
+
chunkOpts: {
|
|
29
|
+
chunkSize: 128,
|
|
30
|
+
chunkOverlap: 20,
|
|
31
|
+
},
|
|
32
|
+
});
|
|
33
|
+
console.log(` Created ${chunks.length} chunks`);
|
|
34
|
+
console.log("\nš§ Step 2: Generating embeddings (batch)...");
|
|
35
|
+
const texts = chunks.map((chunk) => chunk.content);
|
|
36
|
+
const embeddings = await embed({ modelId, text: texts });
|
|
37
|
+
const embeddedDocs = chunks.map((chunk, i) => ({
|
|
38
|
+
id: chunk.id,
|
|
39
|
+
content: chunk.content,
|
|
40
|
+
embedding: embeddings[i],
|
|
41
|
+
embeddingModelId: modelId,
|
|
42
|
+
}));
|
|
43
|
+
console.log(` Generated ${embeddedDocs.length} embeddings`);
|
|
44
|
+
console.log("\nš¾ Step 3: Saving to vector database...");
|
|
45
|
+
const saveResult = await ragSaveEmbeddings({
|
|
46
|
+
workspace,
|
|
47
|
+
documents: embeddedDocs,
|
|
48
|
+
});
|
|
49
|
+
const saved = saveResult.filter((r) => r.status === "fulfilled").length;
|
|
50
|
+
console.log(` Saved ${saved}/${saveResult.length} documents`);
|
|
51
|
+
console.log("\nš Step 4: Searching...");
|
|
52
|
+
const results = await ragSearch({
|
|
53
|
+
workspace,
|
|
54
|
+
modelId,
|
|
55
|
+
query,
|
|
56
|
+
topK: 3,
|
|
57
|
+
});
|
|
58
|
+
console.log("\nš Top 3 most similar documents:");
|
|
59
|
+
results.forEach((result, index) => {
|
|
60
|
+
console.log("=".repeat(50) + " Top result:");
|
|
61
|
+
console.log(`\n${index + 1}. (Score: ${result.score})`);
|
|
62
|
+
console.log(` ${result.content}`);
|
|
63
|
+
console.log("=".repeat(100));
|
|
64
|
+
console.log();
|
|
65
|
+
});
|
|
66
|
+
// Cleanup: close and delete workspace
|
|
67
|
+
await ragCloseWorkspace({ workspace, deleteOnClose: true });
|
|
68
|
+
console.log(`šļø Deleted '${workspace}' workspace`);
|
|
69
|
+
await unloadModel({ modelId });
|
|
70
|
+
}
|
|
71
|
+
catch (error) {
|
|
72
|
+
console.error("ā Error:", error);
|
|
73
|
+
process.exit(1);
|
|
74
|
+
}
|
|
@@ -0,0 +1,178 @@
|
|
|
1
|
+
import { loadModel, unloadModel, GTE_LARGE_FP16, ragIngest, ragSearch, ragDeleteEmbeddings, ragListWorkspaces, ragCloseWorkspace, ragReindex, } from "@qvac/sdk";
|
|
2
|
+
try {
|
|
3
|
+
console.log("š RAG Workspaces with Chunking Example\n");
|
|
4
|
+
// Load model
|
|
5
|
+
const modelId = await loadModel({
|
|
6
|
+
modelSrc: GTE_LARGE_FP16,
|
|
7
|
+
modelType: "embeddings",
|
|
8
|
+
onProgress: (p) => process.stdout.write(`\rLoading: ${Math.round(p.percentage * 100)}%`),
|
|
9
|
+
});
|
|
10
|
+
console.log("\n");
|
|
11
|
+
// Medical articles (will be chunked)
|
|
12
|
+
const medicalArticles = [
|
|
13
|
+
`COVID-19: A Comprehensive Overview
|
|
14
|
+
|
|
15
|
+
COVID-19 is a respiratory illness caused by the SARS-CoV-2 virus, first identified in December 2019
|
|
16
|
+
in Wuhan, China. The virus spreads primarily through respiratory droplets when an infected person
|
|
17
|
+
coughs, sneezes, or talks.
|
|
18
|
+
|
|
19
|
+
Common symptoms include fever, cough, fatigue, and loss of taste or smell. Some patients may
|
|
20
|
+
experience shortness of breath, muscle aches, sore throat, and headaches. Symptoms typically
|
|
21
|
+
appear 2-14 days after exposure to the virus.
|
|
22
|
+
|
|
23
|
+
Vaccines have been developed using various technologies including mRNA, viral vector, and
|
|
24
|
+
inactivated virus approaches. These vaccines have proven effective in preventing severe illness,
|
|
25
|
+
hospitalization, and death. Booster doses are recommended to maintain immunity.
|
|
26
|
+
|
|
27
|
+
Treatment options vary based on severity. Mild cases often require only supportive care including
|
|
28
|
+
rest and hydration. Severe cases may require hospitalization, oxygen therapy, and antiviral
|
|
29
|
+
medications such as remdesivir or Paxlovid.`,
|
|
30
|
+
`Diabetes Management and Prevention
|
|
31
|
+
|
|
32
|
+
Diabetes is a chronic condition affecting how your body processes blood sugar. Type 1 diabetes
|
|
33
|
+
occurs when the immune system attacks insulin-producing cells. Type 2 diabetes develops when
|
|
34
|
+
the body becomes resistant to insulin or doesn't produce enough.
|
|
35
|
+
|
|
36
|
+
Prevention strategies for Type 2 diabetes include maintaining a healthy weight, regular physical
|
|
37
|
+
activity, and a balanced diet rich in whole grains, vegetables, and lean proteins. Regular
|
|
38
|
+
monitoring of blood glucose levels is essential for managing the condition.
|
|
39
|
+
|
|
40
|
+
Treatment may include lifestyle modifications, oral medications, and insulin therapy. Complications
|
|
41
|
+
can affect the heart, kidneys, eyes, and nervous system if blood sugar is not well controlled.`,
|
|
42
|
+
`Heart Disease: Risk Factors and Prevention
|
|
43
|
+
|
|
44
|
+
Cardiovascular disease remains the leading cause of death worldwide. Risk factors include high
|
|
45
|
+
blood pressure, high cholesterol, smoking, obesity, diabetes, and physical inactivity.
|
|
46
|
+
|
|
47
|
+
Prevention involves adopting a heart-healthy lifestyle: regular exercise, a diet low in saturated
|
|
48
|
+
fats and sodium, maintaining a healthy weight, and avoiding tobacco. Regular check-ups can help
|
|
49
|
+
detect problems early.
|
|
50
|
+
|
|
51
|
+
Treatment options range from lifestyle changes and medications to surgical interventions like
|
|
52
|
+
angioplasty, stent placement, or bypass surgery depending on severity.`,
|
|
53
|
+
`Mental Health and Wellness
|
|
54
|
+
|
|
55
|
+
Mental health encompasses emotional, psychological, and social well-being. Common conditions
|
|
56
|
+
include anxiety disorders, depression, bipolar disorder, and schizophrenia. These affect
|
|
57
|
+
thinking, mood, and behavior.
|
|
58
|
+
|
|
59
|
+
Treatment typically involves a combination of psychotherapy, medication, and lifestyle changes.
|
|
60
|
+
Cognitive behavioral therapy has proven effective for many conditions. Support systems and
|
|
61
|
+
stress management techniques are crucial components of recovery.
|
|
62
|
+
|
|
63
|
+
Early intervention improves outcomes significantly. Reducing stigma around mental health
|
|
64
|
+
encourages people to seek help when needed.`,
|
|
65
|
+
];
|
|
66
|
+
// Tech documents (shorter, won't be chunked)
|
|
67
|
+
const techDocs = [
|
|
68
|
+
"Artificial intelligence is transforming industries through automation and advanced data analysis capabilities.",
|
|
69
|
+
"Deep learning neural networks enable breakthroughs in computer vision and natural language processing.",
|
|
70
|
+
];
|
|
71
|
+
// Ingest medical articles with chunking
|
|
72
|
+
console.log("š Ingesting medical articles with chunking...");
|
|
73
|
+
const medicalResult = await ragIngest({
|
|
74
|
+
modelId,
|
|
75
|
+
workspace: "medical",
|
|
76
|
+
documents: medicalArticles,
|
|
77
|
+
chunk: true,
|
|
78
|
+
chunkOpts: {
|
|
79
|
+
chunkSize: 50,
|
|
80
|
+
chunkOverlap: 10,
|
|
81
|
+
chunkStrategy: "paragraph",
|
|
82
|
+
splitStrategy: "token",
|
|
83
|
+
},
|
|
84
|
+
});
|
|
85
|
+
console.log(`ā
Created ${medicalResult.processed.length} chunks from medical articles`);
|
|
86
|
+
// Ingest tech docs without chunking
|
|
87
|
+
console.log("\nš Ingesting tech documents...");
|
|
88
|
+
const techResult = await ragIngest({
|
|
89
|
+
modelId,
|
|
90
|
+
workspace: "technology",
|
|
91
|
+
documents: techDocs,
|
|
92
|
+
chunk: false,
|
|
93
|
+
});
|
|
94
|
+
console.log(`ā
Ingested ${techResult.processed.length} tech documents`);
|
|
95
|
+
// Test searches
|
|
96
|
+
const searches = [
|
|
97
|
+
{ workspace: "medical", query: "COVID symptoms fever", label: "Medical" },
|
|
98
|
+
{ workspace: "medical", query: "vaccine technology", label: "Medical" },
|
|
99
|
+
{ workspace: "technology", query: "neural networks", label: "Tech" },
|
|
100
|
+
{ workspace: "technology", query: "COVID", label: "Tech (isolation test)" },
|
|
101
|
+
];
|
|
102
|
+
console.log("\nš Running searches:");
|
|
103
|
+
for (const { workspace, query, label } of searches) {
|
|
104
|
+
const results = await ragSearch({ modelId, workspace, query, topK: 1 });
|
|
105
|
+
console.log(`\n${label}: "${query}"`);
|
|
106
|
+
if (results.length > 0 && results[0]) {
|
|
107
|
+
console.log(` ā Score: ${results[0].score.toFixed(3)}`);
|
|
108
|
+
console.log(` ā Match: ${results[0].content.substring(0, 80)}...`);
|
|
109
|
+
}
|
|
110
|
+
else {
|
|
111
|
+
console.log(` ā No results (workspace isolation working correctly)`);
|
|
112
|
+
}
|
|
113
|
+
}
|
|
114
|
+
// Test a third workspace
|
|
115
|
+
console.log("\nš Testing 'general' workspace...");
|
|
116
|
+
await ragIngest({
|
|
117
|
+
modelId,
|
|
118
|
+
workspace: "general",
|
|
119
|
+
documents: ["General workspace content for testing"],
|
|
120
|
+
chunk: false,
|
|
121
|
+
});
|
|
122
|
+
const generalSearch = await ragSearch({
|
|
123
|
+
modelId,
|
|
124
|
+
workspace: "general",
|
|
125
|
+
query: "general workspace",
|
|
126
|
+
});
|
|
127
|
+
console.log(`General workspace: ${generalSearch?.[0]?.content === "General workspace content for testing" ? "ā
Working" : "ā Failed"}`);
|
|
128
|
+
// Reindexing - regenerate embeddings for an existing workspace
|
|
129
|
+
console.log("\nš Reindexing medical workspace...");
|
|
130
|
+
const reindexResult = await ragReindex({
|
|
131
|
+
workspace: "medical",
|
|
132
|
+
modelId,
|
|
133
|
+
onProgress: (stage, current, total) => {
|
|
134
|
+
const pct = total > 0 ? Math.round((current / total) * 100) : 0;
|
|
135
|
+
console.log(` [${stage}] ${current}/${total} (${pct}%)`);
|
|
136
|
+
},
|
|
137
|
+
});
|
|
138
|
+
console.log(`${reindexResult.reindexed ? "ā
Reindex completed" : "ā ļø Reindex returned false"} for medical workspace`);
|
|
139
|
+
if (reindexResult.details) {
|
|
140
|
+
console.log(" Details:", reindexResult.details);
|
|
141
|
+
}
|
|
142
|
+
// Verify reindexed data still works
|
|
143
|
+
const reindexVerify = await ragSearch({
|
|
144
|
+
modelId,
|
|
145
|
+
workspace: "medical",
|
|
146
|
+
query: "COVID symptoms",
|
|
147
|
+
topK: 1,
|
|
148
|
+
});
|
|
149
|
+
console.log(` Verification search: ${reindexVerify.length > 0 ? "ā
Working" : "ā Failed"}`);
|
|
150
|
+
// Cleanup example
|
|
151
|
+
const firstChunk = medicalResult.processed.find((p) => p.status === "fulfilled" && p.id);
|
|
152
|
+
if (firstChunk?.id) {
|
|
153
|
+
await ragDeleteEmbeddings({
|
|
154
|
+
workspace: "medical",
|
|
155
|
+
ids: [firstChunk.id],
|
|
156
|
+
});
|
|
157
|
+
console.log("\nšļø Deleted one medical chunk");
|
|
158
|
+
}
|
|
159
|
+
// List all workspaces
|
|
160
|
+
console.log("\nš Listing workspaces...");
|
|
161
|
+
const workspaces = await ragListWorkspaces();
|
|
162
|
+
workspaces.forEach((ws) => {
|
|
163
|
+
console.log(` ⢠${ws.name} (${ws.open ? "open" : "closed"})`);
|
|
164
|
+
});
|
|
165
|
+
// Close and delete all workspaces created in this example
|
|
166
|
+
console.log("\nšļø Closing and deleting workspaces...");
|
|
167
|
+
await ragCloseWorkspace({ workspace: "medical", deleteOnClose: true });
|
|
168
|
+
console.log("Deleted 'medical' workspace");
|
|
169
|
+
await ragCloseWorkspace({ workspace: "technology", deleteOnClose: true });
|
|
170
|
+
console.log("Deleted 'technology' workspace");
|
|
171
|
+
await ragCloseWorkspace({ workspace: "general", deleteOnClose: true });
|
|
172
|
+
console.log("Deleted 'general' workspace");
|
|
173
|
+
await unloadModel({ modelId });
|
|
174
|
+
}
|
|
175
|
+
catch (error) {
|
|
176
|
+
console.error("ā Error:", error);
|
|
177
|
+
process.exit(1);
|
|
178
|
+
}
|
|
@@ -0,0 +1,49 @@
|
|
|
1
|
+
import { loadModel, translate, unloadModel, BERGAMOT_ENFR } from "@qvac/sdk";
|
|
2
|
+
try {
|
|
3
|
+
const modelId = await loadModel({
|
|
4
|
+
modelSrc: BERGAMOT_ENFR,
|
|
5
|
+
modelType: "nmt",
|
|
6
|
+
modelConfig: {
|
|
7
|
+
engine: "Bergamot",
|
|
8
|
+
from: "en",
|
|
9
|
+
to: "fr",
|
|
10
|
+
beamsize: 1,
|
|
11
|
+
normalize: 1,
|
|
12
|
+
temperature: 0.2,
|
|
13
|
+
norepeatngramsize: 3,
|
|
14
|
+
lengthpenalty: 1.2,
|
|
15
|
+
},
|
|
16
|
+
onProgress: (progress) => {
|
|
17
|
+
console.log(progress);
|
|
18
|
+
},
|
|
19
|
+
});
|
|
20
|
+
console.log(`ā
Bergamot model loaded: ${modelId}`);
|
|
21
|
+
// Test with array of texts for batch processing
|
|
22
|
+
const texts = [
|
|
23
|
+
"Hello world",
|
|
24
|
+
"How are you today?",
|
|
25
|
+
"This is a test of batch translation",
|
|
26
|
+
"The weather is nice",
|
|
27
|
+
];
|
|
28
|
+
console.log("\nš Translating batch of texts:");
|
|
29
|
+
texts.forEach((text, i) => console.log(` ${i + 1}. ${text}`));
|
|
30
|
+
const result = translate({
|
|
31
|
+
modelId,
|
|
32
|
+
text: texts, // Pass array for batch processing
|
|
33
|
+
modelType: "nmt",
|
|
34
|
+
stream: false,
|
|
35
|
+
});
|
|
36
|
+
const translatedText = await result.text;
|
|
37
|
+
const translations = translatedText.split("\n");
|
|
38
|
+
console.log("\nā
Translations:");
|
|
39
|
+
translations.forEach((translation, i) => {
|
|
40
|
+
if (i < texts.length) {
|
|
41
|
+
console.log(` ${i + 1}. ${texts[i]} -> "${translation}"`);
|
|
42
|
+
}
|
|
43
|
+
});
|
|
44
|
+
await unloadModel({ modelId });
|
|
45
|
+
}
|
|
46
|
+
catch (error) {
|
|
47
|
+
console.error("ā Error:", error);
|
|
48
|
+
process.exit(1);
|
|
49
|
+
}
|
|
@@ -0,0 +1,39 @@
|
|
|
1
|
+
import { loadModel, translate, unloadModel, BERGAMOT_ENFR } from "@qvac/sdk";
|
|
2
|
+
try {
|
|
3
|
+
// Bergamot models automatically derive vocabulary files from the model source
|
|
4
|
+
// when using pear:// URLs. You can still override them explicitly if needed:
|
|
5
|
+
// - srcVocabSrc: source vocabulary file (optional)
|
|
6
|
+
// - dstVocabSrc: target vocabulary file (optional)
|
|
7
|
+
const modelId = await loadModel({
|
|
8
|
+
modelSrc: BERGAMOT_ENFR,
|
|
9
|
+
modelType: "nmt",
|
|
10
|
+
modelConfig: {
|
|
11
|
+
engine: "Bergamot",
|
|
12
|
+
from: "en",
|
|
13
|
+
to: "fr",
|
|
14
|
+
beamsize: 1,
|
|
15
|
+
normalize: 1,
|
|
16
|
+
temperature: 0.2,
|
|
17
|
+
norepeatngramsize: 3,
|
|
18
|
+
lengthpenalty: 1.2,
|
|
19
|
+
},
|
|
20
|
+
onProgress: (progress) => {
|
|
21
|
+
console.log(progress);
|
|
22
|
+
},
|
|
23
|
+
});
|
|
24
|
+
console.log(`ā
Bergamot model loaded: ${modelId}`);
|
|
25
|
+
const text = "This is a test of the Bergamot translation model.";
|
|
26
|
+
const result = translate({
|
|
27
|
+
modelId,
|
|
28
|
+
text,
|
|
29
|
+
modelType: "nmt",
|
|
30
|
+
stream: false,
|
|
31
|
+
});
|
|
32
|
+
const translatedText = await result.text;
|
|
33
|
+
console.log(`Translated text EN -> FR: ${text} -> "${translatedText}"`);
|
|
34
|
+
await unloadModel({ modelId });
|
|
35
|
+
}
|
|
36
|
+
catch (error) {
|
|
37
|
+
console.error("ā Error:", error);
|
|
38
|
+
process.exit(1);
|
|
39
|
+
}
|