@qvac/sdk 0.5.0 → 0.6.1

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.
Files changed (259) hide show
  1. package/README.md +48 -11
  2. package/bare-imports.json +35 -0
  3. package/dist/client/api/cancel.d.ts +12 -3
  4. package/dist/client/api/cancel.js +12 -3
  5. package/dist/client/api/get-model-info.d.ts +2 -2
  6. package/dist/client/api/get-model-info.js +2 -2
  7. package/dist/client/api/index.d.ts +2 -1
  8. package/dist/client/api/index.js +2 -1
  9. package/dist/client/api/load-model.js +6 -1
  10. package/dist/client/api/ocr.d.ts +31 -0
  11. package/dist/client/api/ocr.js +88 -0
  12. package/dist/client/api/rag.d.ts +215 -20
  13. package/dist/client/api/rag.js +382 -23
  14. package/dist/client/api/translate.js +10 -11
  15. package/dist/client/api/unload-model.js +1 -1
  16. package/dist/client/init-hooks.d.ts +4 -3
  17. package/dist/client/init-hooks.js +40 -31
  18. package/dist/client/rpc/bare-client.d.ts +1 -178
  19. package/dist/client/rpc/bare-client.js +42 -9
  20. package/dist/client/rpc/expo-rpc-client.js +57 -2
  21. package/dist/client/rpc/node-rpc-client.js +5 -1
  22. package/dist/client/rpc/rpc-client.d.ts +1 -1
  23. package/dist/client/rpc/rpc-client.js +11 -57
  24. package/dist/examples/config/blind-relay/blind-relay.config.d.ts +5 -0
  25. package/dist/examples/config/blind-relay/blind-relay.config.js +7 -0
  26. package/dist/examples/config/default/default.config.d.ts +8 -0
  27. package/dist/examples/config/default/default.config.js +6 -0
  28. package/dist/examples/default-config-usage.d.ts +17 -0
  29. package/dist/examples/default-config-usage.js +47 -0
  30. package/dist/examples/delegated-inference/consumer.js +1 -0
  31. package/dist/examples/download-with-blind-relays.js +4 -3
  32. package/dist/examples/embed-hyperdrive.js +4 -0
  33. package/dist/examples/kv-cache-custom-key.js +16 -1
  34. package/dist/examples/llamacpp-sharded.js +12 -3
  35. package/dist/examples/ocr-fasttext.d.ts +2 -0
  36. package/dist/examples/ocr-fasttext.js +52 -0
  37. package/dist/examples/quickstart.d.ts +2 -0
  38. package/dist/examples/quickstart.js +28 -0
  39. package/dist/examples/rag/rag-hyperdb/cancellation.d.ts +2 -0
  40. package/dist/examples/rag/rag-hyperdb/cancellation.js +75 -0
  41. package/dist/examples/rag/rag-hyperdb/ingest.d.ts +2 -0
  42. package/dist/examples/rag/{rag-hyperdb.js → rag-hyperdb/ingest.js} +10 -4
  43. package/dist/examples/rag/rag-hyperdb/pipeline.d.ts +2 -0
  44. package/dist/examples/rag/rag-hyperdb/pipeline.js +74 -0
  45. package/dist/examples/rag/rag-hyperdb/workspaces.d.ts +2 -0
  46. package/dist/examples/rag/rag-hyperdb/workspaces.js +178 -0
  47. package/dist/examples/translation/translation-bergamot-batch.d.ts +2 -0
  48. package/dist/examples/translation/translation-bergamot-batch.js +49 -0
  49. package/dist/examples/translation/translation-bergamot.d.ts +2 -0
  50. package/dist/examples/translation/translation-bergamot.js +39 -0
  51. package/dist/examples/translation/translation-indic.js +1 -0
  52. package/dist/examples/translation/translation-opus.js +1 -0
  53. package/dist/examples/translation/translation-stream.js +1 -0
  54. package/dist/expo/plugins/withMobileBundle.js +29 -10
  55. package/dist/index.d.ts +3 -2
  56. package/dist/index.js +3 -1
  57. package/dist/logging/addon.js +23 -7
  58. package/dist/logging/base-logger.js +2 -2
  59. package/dist/logging/client-logger.js +15 -4
  60. package/dist/logging/index.d.ts +1 -0
  61. package/dist/logging/index.js +1 -0
  62. package/dist/logging/logger.js +11 -3
  63. package/dist/logging/namespaces.d.ts +1 -0
  64. package/dist/logging/namespaces.js +1 -0
  65. package/dist/logging/registry.d.ts +5 -0
  66. package/dist/logging/registry.js +17 -5
  67. package/dist/logging/transport.js +2 -2
  68. package/dist/logging/utils.d.ts +2 -0
  69. package/dist/logging/utils.js +62 -0
  70. package/dist/models/hyperdrive/models.d.ts +1096 -178
  71. package/dist/models/hyperdrive/models.js +1476 -507
  72. package/dist/schemas/archive.d.ts +9 -0
  73. package/dist/schemas/archive.js +17 -0
  74. package/dist/schemas/cancel.d.ts +6 -0
  75. package/dist/schemas/cancel.js +5 -0
  76. package/dist/schemas/common.d.ts +582 -77
  77. package/dist/schemas/common.js +6 -1
  78. package/dist/schemas/delegate.d.ts +1 -0
  79. package/dist/schemas/delegate.js +1 -0
  80. package/dist/schemas/download-asset.d.ts +21 -9
  81. package/dist/schemas/get-model-info.d.ts +12 -10
  82. package/dist/schemas/get-model-info.js +9 -1
  83. package/dist/schemas/index.d.ts +6 -2
  84. package/dist/schemas/index.js +9 -2
  85. package/dist/schemas/llamacpp-config.d.ts +112 -8
  86. package/dist/schemas/llamacpp-config.js +40 -8
  87. package/dist/schemas/load-model.d.ts +2993 -525
  88. package/dist/schemas/load-model.js +88 -33
  89. package/dist/schemas/logging-stream.d.ts +2 -2
  90. package/dist/schemas/model-src-utils.d.ts +38 -16
  91. package/dist/schemas/model-src-utils.js +4 -8
  92. package/dist/schemas/model-types.d.ts +154 -0
  93. package/dist/schemas/model-types.js +140 -0
  94. package/dist/schemas/ocr.d.ts +88 -0
  95. package/dist/schemas/ocr.js +52 -0
  96. package/dist/schemas/rag.d.ts +183 -52
  97. package/dist/schemas/rag.js +148 -23
  98. package/dist/schemas/reload-config.d.ts +20 -8
  99. package/dist/schemas/reload-config.js +7 -5
  100. package/dist/schemas/runtime-context.d.ts +19 -0
  101. package/dist/schemas/runtime-context.js +7 -0
  102. package/dist/schemas/sdk-config.d.ts +697 -2
  103. package/dist/schemas/sdk-config.js +99 -1
  104. package/dist/schemas/sdk-errors-client.d.ts +1 -0
  105. package/dist/schemas/sdk-errors-client.js +5 -0
  106. package/dist/schemas/sdk-errors-server.d.ts +14 -0
  107. package/dist/schemas/sdk-errors-server.js +70 -0
  108. package/dist/schemas/shard.d.ts +21 -0
  109. package/dist/schemas/shard.js +17 -0
  110. package/dist/schemas/translate.d.ts +68 -23
  111. package/dist/schemas/translate.js +37 -11
  112. package/dist/schemas/translation-config.d.ts +272 -29
  113. package/dist/schemas/translation-config.js +63 -6
  114. package/dist/scripts/run-unit-tests.d.ts +2 -0
  115. package/dist/scripts/run-unit-tests.js +17 -0
  116. package/dist/server/bare/addons/llamacpp-completion/cache-logger.d.ts +6 -0
  117. package/dist/server/bare/addons/llamacpp-completion/cache-logger.js +34 -0
  118. package/dist/server/bare/addons/{llamacpp → llamacpp-completion}/create-model.d.ts +1 -5
  119. package/dist/server/bare/addons/{llamacpp → llamacpp-completion}/create-model.js +3 -26
  120. package/dist/server/bare/addons/{llamacpp → llamacpp-completion}/handlers/completion-stream.js +26 -7
  121. package/dist/server/bare/addons/llamacpp-completion/index.d.ts +4 -0
  122. package/dist/server/bare/addons/llamacpp-completion/index.js +3 -0
  123. package/dist/server/bare/addons/{llamacpp → llamacpp-completion}/kv-cache-utils.d.ts +4 -1
  124. package/dist/server/bare/addons/{llamacpp → llamacpp-completion}/kv-cache-utils.js +43 -10
  125. package/dist/server/bare/addons/llamacpp-embedding/create-model.d.ts +8 -0
  126. package/dist/server/bare/addons/llamacpp-embedding/create-model.js +38 -0
  127. package/dist/server/bare/addons/llamacpp-embedding/index.d.ts +3 -0
  128. package/dist/server/bare/addons/llamacpp-embedding/index.js +2 -0
  129. package/dist/server/bare/addons/load-model.js +28 -24
  130. package/dist/server/bare/addons/{translation → nmtcpp-translation}/create-model.js +23 -8
  131. package/dist/server/bare/addons/{translation → nmtcpp-translation}/handlers/translate.js +35 -11
  132. package/dist/server/bare/addons/onnx-ocr/create-model.d.ts +10 -0
  133. package/dist/server/bare/addons/onnx-ocr/create-model.js +39 -0
  134. package/dist/server/bare/addons/onnx-ocr/handlers/ocr-stream.d.ts +10 -0
  135. package/dist/server/bare/addons/onnx-ocr/handlers/ocr-stream.js +109 -0
  136. package/dist/server/bare/addons/onnx-ocr/index.d.ts +3 -0
  137. package/dist/server/bare/addons/onnx-ocr/index.js +2 -0
  138. package/dist/server/bare/addons/{whispercpp → whispercpp-transcription}/create-model.js +2 -1
  139. package/dist/server/bare/delegate-rpc-client.js +19 -1
  140. package/dist/server/bare/rag-hyperdb/handlers/chunk.d.ts +3 -0
  141. package/dist/server/bare/rag-hyperdb/handlers/chunk.js +7 -0
  142. package/dist/server/bare/rag-hyperdb/handlers/close-workspace.d.ts +3 -0
  143. package/dist/server/bare/rag-hyperdb/handlers/close-workspace.js +9 -0
  144. package/dist/server/bare/rag-hyperdb/handlers/delete-embeddings.d.ts +1 -1
  145. package/dist/server/bare/rag-hyperdb/handlers/delete-embeddings.js +9 -10
  146. package/dist/server/bare/rag-hyperdb/handlers/delete-workspace.d.ts +3 -0
  147. package/dist/server/bare/rag-hyperdb/handlers/delete-workspace.js +14 -0
  148. package/dist/server/bare/rag-hyperdb/handlers/ingest.d.ts +9 -0
  149. package/dist/server/bare/rag-hyperdb/handlers/ingest.js +18 -0
  150. package/dist/server/bare/rag-hyperdb/handlers/list-workspaces.d.ts +3 -0
  151. package/dist/server/bare/rag-hyperdb/handlers/list-workspaces.js +4 -0
  152. package/dist/server/bare/rag-hyperdb/handlers/reindex.d.ts +9 -0
  153. package/dist/server/bare/rag-hyperdb/handlers/reindex.js +16 -0
  154. package/dist/server/bare/rag-hyperdb/handlers/save-embeddings.d.ts +7 -4
  155. package/dist/server/bare/rag-hyperdb/handlers/save-embeddings.js +13 -14
  156. package/dist/server/bare/rag-hyperdb/handlers/search.js +10 -10
  157. package/dist/server/bare/rag-hyperdb/index.d.ts +8 -1
  158. package/dist/server/bare/rag-hyperdb/index.js +8 -1
  159. package/dist/server/bare/rag-hyperdb/rag-operation-manager.d.ts +7 -0
  160. package/dist/server/bare/rag-hyperdb/rag-operation-manager.js +39 -0
  161. package/dist/server/bare/rag-hyperdb/rag-workspace-manager.d.ts +15 -0
  162. package/dist/server/bare/rag-hyperdb/rag-workspace-manager.js +135 -0
  163. package/dist/server/bare/registry/config-registry.js +11 -0
  164. package/dist/server/bare/registry/logging-stream-registry.js +3 -3
  165. package/dist/server/bare/registry/model-config-registry.d.ts +4 -0
  166. package/dist/server/bare/registry/model-config-registry.js +15 -0
  167. package/dist/server/bare/registry/model-config-utils.d.ts +15 -0
  168. package/dist/server/bare/registry/model-config-utils.js +110 -0
  169. package/dist/server/bare/registry/model-registry.d.ts +6 -29
  170. package/dist/server/bare/registry/runtime-context-registry.d.ts +7 -0
  171. package/dist/server/bare/registry/runtime-context-registry.js +26 -0
  172. package/dist/server/bare/utils/loader-adapter.d.ts +7 -0
  173. package/dist/server/bare/utils/loader-adapter.js +9 -0
  174. package/dist/server/error-handlers.js +4 -2
  175. package/dist/server/rpc/handle-request.js +91 -9
  176. package/dist/server/rpc/handlers/cancelHandler.js +10 -1
  177. package/dist/server/rpc/handlers/completion-stream.js +1 -1
  178. package/dist/server/rpc/handlers/delete-cache.js +1 -1
  179. package/dist/server/rpc/handlers/embed.js +1 -1
  180. package/dist/server/rpc/handlers/get-model-info.d.ts +1 -1
  181. package/dist/server/rpc/handlers/get-model-info.js +22 -10
  182. package/dist/server/rpc/handlers/index.d.ts +4 -1
  183. package/dist/server/rpc/handlers/index.js +2 -0
  184. package/dist/server/rpc/handlers/load-model/download-manager.d.ts +11 -11
  185. package/dist/server/rpc/handlers/load-model/download-manager.js +36 -26
  186. package/dist/server/rpc/handlers/{load-model.d.ts → load-model/handler.d.ts} +2 -2
  187. package/dist/server/rpc/handlers/load-model/handler.js +179 -0
  188. package/dist/server/rpc/handlers/load-model/http.js +342 -33
  189. package/dist/server/rpc/handlers/load-model/hyperdrive.js +5 -3
  190. package/dist/server/rpc/handlers/load-model/index.d.ts +2 -2
  191. package/dist/server/rpc/handlers/load-model/index.js +2 -2
  192. package/dist/server/rpc/handlers/load-model/resolve.js +32 -3
  193. package/dist/server/rpc/handlers/load-model-delegated.js +6 -3
  194. package/dist/server/rpc/handlers/ocr-stream.d.ts +3 -0
  195. package/dist/server/rpc/handlers/ocr-stream.js +39 -0
  196. package/dist/server/rpc/handlers/rag.d.ts +2 -2
  197. package/dist/server/rpc/handlers/rag.js +107 -43
  198. package/dist/server/rpc/handlers/text-to-speech.js +1 -1
  199. package/dist/server/rpc/handlers/transcribe-stream.js +1 -1
  200. package/dist/server/rpc/handlers/translate.js +6 -15
  201. package/dist/server/utils/archive.d.ts +3 -0
  202. package/dist/server/utils/archive.js +171 -0
  203. package/dist/server/utils/audio/decoder.js +2 -2
  204. package/dist/server/utils/cache.d.ts +5 -5
  205. package/dist/server/utils/cache.js +6 -6
  206. package/dist/server/utils/config-transform.d.ts +2 -2
  207. package/dist/server/utils/config-transform.js +8 -6
  208. package/dist/server/utils/formatting.d.ts +13 -0
  209. package/dist/server/utils/formatting.js +26 -0
  210. package/dist/server/utils/gguf-tensor-extractor.d.ts +15 -0
  211. package/dist/server/utils/gguf-tensor-extractor.js +314 -0
  212. package/dist/server/utils/index.d.ts +3 -1
  213. package/dist/server/utils/index.js +3 -1
  214. package/dist/server/utils/shard-utils.d.ts +42 -16
  215. package/dist/server/utils/shard-utils.js +95 -7
  216. package/dist/server/worker.js +8 -3
  217. package/dist/utils/errors-base.d.ts +1 -1
  218. package/dist/utils/errors-base.js +2 -2
  219. package/dist/utils/errors-client.d.ts +12 -0
  220. package/dist/utils/errors-client.js +20 -0
  221. package/dist/utils/errors-server.d.ts +33 -0
  222. package/dist/utils/errors-server.js +55 -0
  223. package/package.json +42 -17
  224. package/dist/client/rpc/bare-rpc-client.d.ts +0 -4
  225. package/dist/client/rpc/bare-rpc-client.js +0 -94
  226. package/dist/examples/config/config-path.d.ts +0 -8
  227. package/dist/examples/config/config-path.js +0 -13
  228. package/dist/examples/rag/rag-hyperdb-workspaces.d.ts +0 -2
  229. package/dist/examples/rag/rag-hyperdb-workspaces.js +0 -106
  230. package/dist/examples/rag/rag-hyperdb.d.ts +0 -2
  231. package/dist/server/bare/addons/llamacpp/index.d.ts +0 -4
  232. package/dist/server/bare/addons/llamacpp/index.js +0 -3
  233. package/dist/server/bare/rag-hyperdb/rag-manager.d.ts +0 -5
  234. package/dist/server/bare/rag-hyperdb/rag-manager.js +0 -66
  235. package/dist/server/rpc/handlers/load-model.js +0 -111
  236. package/dist/server/utils/hash.d.ts +0 -5
  237. package/dist/server/utils/hash.js +0 -11
  238. package/dist/utils/getProcessSafe.d.ts +0 -7
  239. package/dist/utils/getProcessSafe.js +0 -17
  240. package/dist/utils/runtime.d.ts +0 -30
  241. package/dist/utils/runtime.js +0 -48
  242. /package/dist/server/bare/addons/{llamacpp → llamacpp-completion}/handlers/completion-stream.d.ts +0 -0
  243. /package/dist/server/bare/addons/{llamacpp → llamacpp-embedding}/handlers/embed.d.ts +0 -0
  244. /package/dist/server/bare/addons/{llamacpp → llamacpp-embedding}/handlers/embed.js +0 -0
  245. /package/dist/server/bare/addons/{translation → nmtcpp-translation}/create-model.d.ts +0 -0
  246. /package/dist/server/bare/addons/{translation → nmtcpp-translation}/handlers/translate.d.ts +0 -0
  247. /package/dist/server/bare/addons/{translation → nmtcpp-translation}/index.d.ts +0 -0
  248. /package/dist/server/bare/addons/{translation → nmtcpp-translation}/index.js +0 -0
  249. /package/dist/server/bare/addons/{tts → onnx-tts}/create-model.d.ts +0 -0
  250. /package/dist/server/bare/addons/{tts → onnx-tts}/create-model.js +0 -0
  251. /package/dist/server/bare/addons/{tts → onnx-tts}/handlers/text-to-speech.d.ts +0 -0
  252. /package/dist/server/bare/addons/{tts → onnx-tts}/handlers/text-to-speech.js +0 -0
  253. /package/dist/server/bare/addons/{tts → onnx-tts}/index.d.ts +0 -0
  254. /package/dist/server/bare/addons/{tts → onnx-tts}/index.js +0 -0
  255. /package/dist/server/bare/addons/{whispercpp → whispercpp-transcription}/create-model.d.ts +0 -0
  256. /package/dist/server/bare/addons/{whispercpp → whispercpp-transcription}/handlers/transcribe-stream.d.ts +0 -0
  257. /package/dist/server/bare/addons/{whispercpp → whispercpp-transcription}/handlers/transcribe-stream.js +0 -0
  258. /package/dist/server/bare/addons/{whispercpp → whispercpp-transcription}/index.d.ts +0 -0
  259. /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
+ });
@@ -1,28 +1,16 @@
1
1
  import { z } from "zod";
2
- declare const ragSaveEmbeddingsResultSchema: z.ZodObject<{
3
- status: z.ZodEnum<{
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 ragSaveEmbeddingsOperationSchema: z.ZodObject<{
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 ragSearchParamsSchema: z.ZodObject<{
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
- query: z.ZodString;
63
- topK: z.ZodOptional<z.ZodNumber>;
64
- n: z.ZodOptional<z.ZodNumber>;
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 ragSearchOperationSchema: z.ZodObject<{
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 deleteEmbeddingsParamsSchema: z.ZodObject<{
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 ragDeleteEmbeddingsOperationSchema: z.ZodObject<{
81
- modelId: z.ZodString;
77
+ export declare const ragCloseWorkspaceParamsSchema: z.ZodObject<{
82
78
  workspace: z.ZodOptional<z.ZodString>;
83
- ids: z.ZodArray<z.ZodString>;
84
- type: z.ZodLiteral<"rag">;
85
- operation: z.ZodLiteral<"deleteEmbeddings">;
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
- modelId: z.ZodString;
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<"saveEmbeddings">;
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 RagSaveEmbeddingsResult = z.infer<typeof ragSaveEmbeddingsResultSchema>;
157
- export type RagSearchResult = z.infer<typeof searchResultSchema>;
158
- export type RagSaveEmbeddingsParams = z.infer<typeof ragSaveEmbeddingsParamsSchema>;
159
- export type RagDeleteEmbeddingsParams = z.infer<typeof deleteEmbeddingsParamsSchema>;
160
- export type RagSearchParams = z.infer<typeof ragSearchParamsSchema>;
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