@qvac/sdk 0.10.2 → 0.11.0

package/dist/schemas/load-model.js

@@ -24,6 +24,7 @@ const loadModelRequestCommonFields = {
     onProgress: z.unknown().optional(),
     logger: z.unknown().optional(),
     withProgress: z.boolean().optional(),
+    requestId: z.string().min(1).optional(),
 };
 export const loadBuiltinModelOptionsBaseSchema = z.union([
     z
@@ -116,6 +117,7 @@ const loadModelOptionsToRequestBaseSchema = z.union([
         seed: data.seed ?? false,
         withProgress: data.withProgress ?? !!data.onProgress,
         delegate: data.delegate,
+        ...(data.requestId !== undefined && { requestId: data.requestId }),
     })),
     z
         .object({
@@ -133,6 +135,7 @@ const loadModelOptionsToRequestBaseSchema = z.union([
         seed: data.seed ?? false,
         withProgress: data.withProgress ?? !!data.onProgress,
         delegate: data.delegate,
+        ...(data.requestId !== undefined && { requestId: data.requestId }),
     })),
     z
         .object({
@@ -150,6 +153,7 @@ const loadModelOptionsToRequestBaseSchema = z.union([
         seed: data.seed ?? false,
         withProgress: data.withProgress ?? !!data.onProgress,
         delegate: data.delegate,
+        ...(data.requestId !== undefined && { requestId: data.requestId }),
     })),
     z
         .object({
@@ -167,6 +171,7 @@ const loadModelOptionsToRequestBaseSchema = z.union([
         seed: data.seed ?? false,
         withProgress: data.withProgress ?? !!data.onProgress,
         delegate: data.delegate,
+        ...(data.requestId !== undefined && { requestId: data.requestId }),
     })),
     z
         .object({
@@ -192,6 +197,7 @@ const loadModelOptionsToRequestBaseSchema = z.union([
         seed: data.seed ?? false,
         withProgress: data.withProgress ?? !!data.onProgress,
         delegate: data.delegate,
+        ...(data.requestId !== undefined && { requestId: data.requestId }),
     })),
     z
         .object({
@@ -209,6 +215,7 @@ const loadModelOptionsToRequestBaseSchema = z.union([
         seed: data.seed ?? false,
         withProgress: data.withProgress ?? !!data.onProgress,
         delegate: data.delegate,
+        ...(data.requestId !== undefined && { requestId: data.requestId }),
     })),
     z
         .object({
@@ -226,6 +233,7 @@ const loadModelOptionsToRequestBaseSchema = z.union([
         seed: data.seed ?? false,
         withProgress: data.withProgress ?? !!data.onProgress,
         delegate: data.delegate,
+        ...(data.requestId !== undefined && { requestId: data.requestId }),
     })),
     z
         .object({
@@ -243,6 +251,7 @@ const loadModelOptionsToRequestBaseSchema = z.union([
         seed: data.seed ?? false,
         withProgress: data.withProgress ?? !!data.onProgress,
         delegate: data.delegate,
+        ...(data.requestId !== undefined && { requestId: data.requestId }),
     })),
     z
         .object({
@@ -261,6 +270,7 @@ const loadModelOptionsToRequestBaseSchema = z.union([
         seed: data.seed ?? false,
         withProgress: data.withProgress ?? !!data.onProgress,
         delegate: data.delegate,
+        ...(data.requestId !== undefined && { requestId: data.requestId }),
     })),
 ]);
 export const loadModelOptionsToRequestSchema = loadModelOptionsToRequestBaseSchema;
@@ -271,6 +281,11 @@ const commonModelConfigSchema = z.object({
     withProgress: z.boolean().optional(),
     seed: z.boolean().optional(),
     delegate: delegateSchema,
+    requestId: z
+        .string()
+        .min(1)
+        .optional()
+        .describe("Stable identifier for this in-flight load, generated by the client at call time. Optional on the wire so legacy clients keep working — the server falls back to a server-generated id when the field is missing. Exposed on the client-side decorated promise so callers can target this load with `cancel({ requestId })`."),
 });
 // Request schemas for each model type (use canonical types since transforms normalize)
 // Use base schemas (no defaults) for client-side validation.

package/dist/schemas/model-src-utils.d.ts

@@ -12,6 +12,9 @@ export declare const modelDescriptorSchema: z.ZodObject<{
     sha256Checksum: z.ZodOptional<z.ZodString>;
     addon: z.ZodOptional<z.ZodUnion<readonly [z.ZodEnum<{
         embeddings: "embeddings";
+        diffusion: "diffusion";
+        tts: "tts";
+        ocr: "ocr";
         "llamacpp-completion": "llamacpp-completion";
         "whispercpp-transcription": "whispercpp-transcription";
         "llamacpp-embedding": "llamacpp-embedding";
@@ -24,9 +27,6 @@ export declare const modelDescriptorSchema: z.ZodObject<{
         whisper: "whisper";
         nmt: "nmt";
         parakeet: "parakeet";
-        tts: "tts";
-        ocr: "ocr";
-        diffusion: "diffusion";
     }>, z.ZodLiteral<"vad">]>>;
 }, z.core.$strip>;
 export declare const modelSrcInputSchema: z.ZodUnion<readonly [z.ZodString, z.ZodObject<{
@@ -42,6 +42,9 @@ export declare const modelSrcInputSchema: z.ZodUnion<readonly [z.ZodString, z.Zo
     sha256Checksum: z.ZodOptional<z.ZodString>;
     addon: z.ZodOptional<z.ZodUnion<readonly [z.ZodEnum<{
         embeddings: "embeddings";
+        diffusion: "diffusion";
+        tts: "tts";
+        ocr: "ocr";
         "llamacpp-completion": "llamacpp-completion";
         "whispercpp-transcription": "whispercpp-transcription";
         "llamacpp-embedding": "llamacpp-embedding";
@@ -54,9 +57,6 @@ export declare const modelSrcInputSchema: z.ZodUnion<readonly [z.ZodString, z.Zo
         whisper: "whisper";
         nmt: "nmt";
         parakeet: "parakeet";
-        tts: "tts";
-        ocr: "ocr";
-        diffusion: "diffusion";
     }>, z.ZodLiteral<"vad">]>>;
 }, z.core.$strip>]>;
 export type ModelDescriptor = z.infer<typeof modelDescriptorSchema>;
@@ -78,6 +78,9 @@ export declare const modelInputToSrcSchema: z.ZodPipe<z.ZodUnion<readonly [z.Zod
     sha256Checksum: z.ZodOptional<z.ZodString>;
     addon: z.ZodOptional<z.ZodUnion<readonly [z.ZodEnum<{
         embeddings: "embeddings";
+        diffusion: "diffusion";
+        tts: "tts";
+        ocr: "ocr";
         "llamacpp-completion": "llamacpp-completion";
         "whispercpp-transcription": "whispercpp-transcription";
         "llamacpp-embedding": "llamacpp-embedding";
@@ -90,9 +93,6 @@ export declare const modelInputToSrcSchema: z.ZodPipe<z.ZodUnion<readonly [z.Zod
         whisper: "whisper";
         nmt: "nmt";
         parakeet: "parakeet";
-        tts: "tts";
-        ocr: "ocr";
-        diffusion: "diffusion";
     }>, z.ZodLiteral<"vad">]>>;
 }, z.core.$strip>]>, z.ZodTransform<string, string | {
     src: string;
@@ -105,7 +105,7 @@ export declare const modelInputToSrcSchema: z.ZodPipe<z.ZodUnion<readonly [z.Zod
     engine?: string | undefined;
     expectedSize?: number | undefined;
     sha256Checksum?: string | undefined;
-    addon?: "embeddings" | "llamacpp-completion" | "whispercpp-transcription" | "llamacpp-embedding" | "nmtcpp-translation" | "onnx-tts" | "parakeet-transcription" | "onnx-ocr" | "sdcpp-generation" | "llm" | "whisper" | "nmt" | "parakeet" | "tts" | "ocr" | "diffusion" | "vad" | undefined;
+    addon?: "embeddings" | "diffusion" | "tts" | "ocr" | "llamacpp-completion" | "whispercpp-transcription" | "llamacpp-embedding" | "nmtcpp-translation" | "onnx-tts" | "parakeet-transcription" | "onnx-ocr" | "sdcpp-generation" | "llm" | "whisper" | "nmt" | "parakeet" | "vad" | undefined;
 }>>;
 /**
  * Schema that transforms ModelSrc to its optional name
@@ -124,6 +124,9 @@ export declare const modelInputToNameSchema: z.ZodPipe<z.ZodUnion<readonly [z.Zo
     sha256Checksum: z.ZodOptional<z.ZodString>;
     addon: z.ZodOptional<z.ZodUnion<readonly [z.ZodEnum<{
         embeddings: "embeddings";
+        diffusion: "diffusion";
+        tts: "tts";
+        ocr: "ocr";
         "llamacpp-completion": "llamacpp-completion";
         "whispercpp-transcription": "whispercpp-transcription";
         "llamacpp-embedding": "llamacpp-embedding";
@@ -136,9 +139,6 @@ export declare const modelInputToNameSchema: z.ZodPipe<z.ZodUnion<readonly [z.Zo
         whisper: "whisper";
         nmt: "nmt";
         parakeet: "parakeet";
-        tts: "tts";
-        ocr: "ocr";
-        diffusion: "diffusion";
     }>, z.ZodLiteral<"vad">]>>;
 }, z.core.$strip>]>, z.ZodTransform<string | undefined, string | {
     src: string;
@@ -151,7 +151,7 @@ export declare const modelInputToNameSchema: z.ZodPipe<z.ZodUnion<readonly [z.Zo
     engine?: string | undefined;
     expectedSize?: number | undefined;
     sha256Checksum?: string | undefined;
-    addon?: "embeddings" | "llamacpp-completion" | "whispercpp-transcription" | "llamacpp-embedding" | "nmtcpp-translation" | "onnx-tts" | "parakeet-transcription" | "onnx-ocr" | "sdcpp-generation" | "llm" | "whisper" | "nmt" | "parakeet" | "tts" | "ocr" | "diffusion" | "vad" | undefined;
+    addon?: "embeddings" | "diffusion" | "tts" | "ocr" | "llamacpp-completion" | "whispercpp-transcription" | "llamacpp-embedding" | "nmtcpp-translation" | "onnx-tts" | "parakeet-transcription" | "onnx-ocr" | "sdcpp-generation" | "llm" | "whisper" | "nmt" | "parakeet" | "vad" | undefined;
 }>>;
 export declare function inferModelTypeFromModelSrc(modelSrc: unknown): string | undefined;
 //# sourceMappingURL=model-src-utils.d.ts.map

package/dist/schemas/model-types.d.ts

@@ -68,6 +68,9 @@ export declare const PUBLIC_MODEL_TYPES: {
 /** Schema accepting both canonical and alias model type inputs */
 export declare const modelTypeInputSchema: z.ZodEnum<{
     embeddings: "embeddings";
+    diffusion: "diffusion";
+    tts: "tts";
+    ocr: "ocr";
     "llamacpp-completion": "llamacpp-completion";
     "whispercpp-transcription": "whispercpp-transcription";
     "llamacpp-embedding": "llamacpp-embedding";
@@ -80,13 +83,13 @@ export declare const modelTypeInputSchema: z.ZodEnum<{
     whisper: "whisper";
     nmt: "nmt";
     parakeet: "parakeet";
-    tts: "tts";
-    ocr: "ocr";
-    diffusion: "diffusion";
 }>;
 /** Schema that transforms input to canonical model type */
 export declare const modelTypeSchema: z.ZodPipe<z.ZodEnum<{
     embeddings: "embeddings";
+    diffusion: "diffusion";
+    tts: "tts";
+    ocr: "ocr";
     "llamacpp-completion": "llamacpp-completion";
     "whispercpp-transcription": "whispercpp-transcription";
     "llamacpp-embedding": "llamacpp-embedding";
@@ -99,10 +102,7 @@ export declare const modelTypeSchema: z.ZodPipe<z.ZodEnum<{
     whisper: "whisper";
     nmt: "nmt";
     parakeet: "parakeet";
-    tts: "tts";
-    ocr: "ocr";
-    diffusion: "diffusion";
-}>, z.ZodTransform<"llamacpp-completion" | "whispercpp-transcription" | "llamacpp-embedding" | "nmtcpp-translation" | "onnx-tts" | "parakeet-transcription" | "onnx-ocr" | "sdcpp-generation", "embeddings" | "llamacpp-completion" | "whispercpp-transcription" | "llamacpp-embedding" | "nmtcpp-translation" | "onnx-tts" | "parakeet-transcription" | "onnx-ocr" | "sdcpp-generation" | "llm" | "whisper" | "nmt" | "parakeet" | "tts" | "ocr" | "diffusion">>;
+}>, z.ZodTransform<"llamacpp-completion" | "whispercpp-transcription" | "llamacpp-embedding" | "nmtcpp-translation" | "onnx-tts" | "parakeet-transcription" | "onnx-ocr" | "sdcpp-generation", "embeddings" | "diffusion" | "tts" | "ocr" | "llamacpp-completion" | "whispercpp-transcription" | "llamacpp-embedding" | "nmtcpp-translation" | "onnx-tts" | "parakeet-transcription" | "onnx-ocr" | "sdcpp-generation" | "llm" | "whisper" | "nmt" | "parakeet">>;
 /**
  * Normalize model type input to canonical form.
  * Custom plugin types (not in built-ins) are returned as-is.
@@ -174,8 +174,8 @@ export type NmtModelTypeInput = z.infer<typeof nmtModelTypeSchema>;
  * - Canonical: `"onnx-tts"`
  */
 export declare const ttsModelTypeSchema: z.ZodEnum<{
-    "onnx-tts": "onnx-tts";
     tts: "tts";
+    "onnx-tts": "onnx-tts";
 }>;
 export type TtsModelTypeInput = z.infer<typeof ttsModelTypeSchema>;
 /**
@@ -184,8 +184,8 @@ export type TtsModelTypeInput = z.infer<typeof ttsModelTypeSchema>;
  * - Canonical: `"onnx-ocr"`
  */
 export declare const ocrModelTypeSchema: z.ZodEnum<{
-    "onnx-ocr": "onnx-ocr";
     ocr: "ocr";
+    "onnx-ocr": "onnx-ocr";
 }>;
 export type OcrModelTypeInput = z.infer<typeof ocrModelTypeSchema>;
 /**
@@ -194,8 +194,8 @@ export type OcrModelTypeInput = z.infer<typeof ocrModelTypeSchema>;
  * - Canonical: `"sdcpp-generation"`
  */
 export declare const diffusionModelTypeSchema: z.ZodEnum<{
-    "sdcpp-generation": "sdcpp-generation";
     diffusion: "diffusion";
+    "sdcpp-generation": "sdcpp-generation";
 }>;
 export type DiffusionModelTypeInput = z.infer<typeof diffusionModelTypeSchema>;
 //# sourceMappingURL=model-types.d.ts.map

package/dist/schemas/ocr.d.ts

@@ -30,6 +30,9 @@ export declare const ocrConfigSchema: z.ZodObject<{
         sha256Checksum: z.ZodOptional<z.ZodString>;
         addon: z.ZodOptional<z.ZodUnion<readonly [z.ZodEnum<{
             embeddings: "embeddings";
+            diffusion: "diffusion";
+            tts: "tts";
+            ocr: "ocr";
             "llamacpp-completion": "llamacpp-completion";
             "whispercpp-transcription": "whispercpp-transcription";
             "llamacpp-embedding": "llamacpp-embedding";
@@ -42,9 +45,6 @@ export declare const ocrConfigSchema: z.ZodObject<{
             whisper: "whisper";
             nmt: "nmt";
             parakeet: "parakeet";
-            tts: "tts";
-            ocr: "ocr";
-            diffusion: "diffusion";
         }>, z.ZodLiteral<"vad">]>>;
     }, z.core.$strip>]>>;
 }, z.core.$strip>;

package/dist/schemas/plugin.d.ts

@@ -1,5 +1,21 @@
 import { z } from "zod";
 import type { ModelSrcInput } from "./model-src-utils";
+/**
+ * Granularity at which the addon can cancel.
+ *  - `"request"` — addon cancels a specific in-flight `requestId`.
+ *  - `"model"` — addon cancels whatever is running on the model.
+ *  - `"none"` — no addon cancel surface; SDK falls back to soft-cancel
+ *    (stop yielding, drop result; the C++ work runs to completion).
+ */
+export type PluginHandlerCancelScope = "request" | "model" | "none";
+export interface PluginHandlerCancel {
+    scope: PluginHandlerCancelScope;
+    /**
+     * `true` — `addon.cancel()` interrupts compute; otherwise it's
+     * best-effort. Only meaningful for `scope: "model" | "request"`.
+     */
+    hard?: boolean;
+}
 /**
  * Definition for a plugin handler with explicit Zod schemas.
  * Each handler must define its request/response schemas for validation.
@@ -10,6 +26,11 @@ export interface PluginHandlerDefinition<TRequest extends z.ZodType = z.ZodType,
     streaming: boolean;
     duplex?: boolean;
     handler: TRequest extends z.ZodType<infer I> ? TResponse extends z.ZodType<infer O> ? (request: I, inputStream?: AsyncIterable<Buffer>) => Promise<O> | AsyncGenerator<O> : never : never;
+    /**
+     * Cancel surface this handler advertises. Omitting is equivalent
+     * to `{ scope: "none" }` (soft-cancel fallback).
+     */
+    cancel?: PluginHandlerCancel;
 }
 /**
  * Variant of `PluginHandlerDefinition` for duplex handlers where
@@ -21,6 +42,8 @@ export interface DuplexPluginHandlerDefinition<TRequest extends z.ZodType = z.Zo
     streaming: true;
     duplex: true;
     handler: TRequest extends z.ZodType<infer I> ? TResponse extends z.ZodType<infer O> ? (request: I, inputStream: AsyncIterable<Buffer>) => AsyncGenerator<O> : never : never;
+    /** See `PluginHandlerDefinition.cancel`. */
+    cancel?: PluginHandlerCancel;
 }
 /**
  * Parameters passed to createModel when loading a plugin model.
@@ -174,6 +197,14 @@ export declare const pluginHandlerDefinitionRuntimeSchema: z.ZodObject<{
     streaming: z.ZodBoolean;
     duplex: z.ZodOptional<z.ZodBoolean>;
     handler: z.ZodCustom<Function, Function>;
+    cancel: z.ZodOptional<z.ZodObject<{
+        scope: z.ZodEnum<{
+            request: "request";
+            none: "none";
+            model: "model";
+        }>;
+        hard: z.ZodOptional<z.ZodBoolean>;
+    }, z.core.$catchall<z.ZodUnknown>>>;
 }, z.core.$catchall<z.ZodUnknown>>;
 export declare const pluginDefinitionRuntimeSchema: z.ZodObject<{
     modelType: z.ZodString;
@@ -193,6 +224,14 @@ export declare const pluginDefinitionRuntimeSchema: z.ZodObject<{
         streaming: z.ZodBoolean;
         duplex: z.ZodOptional<z.ZodBoolean>;
         handler: z.ZodCustom<Function, Function>;
+        cancel: z.ZodOptional<z.ZodObject<{
+            scope: z.ZodEnum<{
+                request: "request";
+                none: "none";
+                model: "model";
+            }>;
+            hard: z.ZodOptional<z.ZodBoolean>;
+        }, z.core.$catchall<z.ZodUnknown>>>;
     }, z.core.$catchall<z.ZodUnknown>>>;
     logging: z.ZodOptional<z.ZodObject<{
         module: z.ZodOptional<z.ZodUnknown>;
@@ -238,7 +277,7 @@ export declare const PLUGIN_TTS: "@qvac/sdk/onnx-tts/plugin";
 export declare const PLUGIN_OCR: "@qvac/sdk/onnx-ocr/plugin";
 /**
  * Image generation plugin (stable-diffusion.cpp).
- * Provides: text-to-image generation.
+ * Provides: text-to-image generation and standalone ESRGAN image upscaling.
  */
 export declare const PLUGIN_DIFFUSION: "@qvac/sdk/sdcpp-generation/plugin";
 /**

package/dist/schemas/plugin.js

@@ -80,6 +80,14 @@ const zodSchemaLikeRuntimeSchema = z
     safeParse: functionRuntimeSchema,
 })
     .catchall(z.unknown());
+const pluginHandlerCancelRuntimeSchema = z
+    .object({
+    scope: z.enum(["request", "model", "none"], {
+        error: "cancel.scope must be 'request', 'model', or 'none'",
+    }),
+    hard: z.boolean().optional(),
+})
+    .catchall(z.unknown());
 export const pluginHandlerDefinitionRuntimeSchema = z
     .object({
     requestSchema: zodSchemaLikeRuntimeSchema,
@@ -87,6 +95,7 @@ export const pluginHandlerDefinitionRuntimeSchema = z
     streaming: z.boolean({ error: "streaming must be a boolean" }),
     duplex: z.boolean().optional(),
     handler: functionRuntimeSchema,
+    cancel: pluginHandlerCancelRuntimeSchema.optional(),
 })
     .catchall(z.unknown());
 export const pluginDefinitionRuntimeSchema = z
@@ -154,7 +163,7 @@ export const PLUGIN_TTS = "@qvac/sdk/onnx-tts/plugin";
 export const PLUGIN_OCR = "@qvac/sdk/onnx-ocr/plugin";
 /**
  * Image generation plugin (stable-diffusion.cpp).
- * Provides: text-to-image generation.
+ * Provides: text-to-image generation and standalone ESRGAN image upscaling.
  */
 export const PLUGIN_DIFFUSION = "@qvac/sdk/sdcpp-generation/plugin";
 /**

package/dist/schemas/rag.d.ts

@@ -100,6 +100,7 @@ export declare const ragRequestSchema: z.ZodDiscriminatedUnion<[z.ZodObject<{
     }, z.core.$strip>>;
     type: z.ZodLiteral<"rag">;
     operation: z.ZodLiteral<"chunk">;
+    requestId: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>, z.ZodObject<{
     workspace: z.ZodOptional<z.ZodString>;
     modelId: z.ZodString;
@@ -125,6 +126,7 @@ export declare const ragRequestSchema: z.ZodDiscriminatedUnion<[z.ZodObject<{
     withProgress: z.ZodOptional<z.ZodBoolean>;
     type: z.ZodLiteral<"rag">;
     operation: z.ZodLiteral<"ingest">;
+    requestId: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>, z.ZodObject<{
     modelId: z.ZodOptional<z.ZodString>;
     workspace: z.ZodOptional<z.ZodString>;
@@ -140,6 +142,7 @@ export declare const ragRequestSchema: z.ZodDiscriminatedUnion<[z.ZodObject<{
     withProgress: z.ZodOptional<z.ZodBoolean>;
     type: z.ZodLiteral<"rag">;
     operation: z.ZodLiteral<"saveEmbeddings">;
+    requestId: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>, z.ZodObject<{
     workspace: z.ZodOptional<z.ZodString>;
     modelId: z.ZodString;
@@ -148,12 +151,14 @@ export declare const ragRequestSchema: z.ZodDiscriminatedUnion<[z.ZodObject<{
     n: z.ZodDefault<z.ZodNumber>;
     type: z.ZodLiteral<"rag">;
     operation: z.ZodLiteral<"search">;
+    requestId: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>, z.ZodObject<{
     modelId: z.ZodOptional<z.ZodString>;
     workspace: z.ZodOptional<z.ZodString>;
     ids: z.ZodArray<z.ZodString>;
     type: z.ZodLiteral<"rag">;
     operation: z.ZodLiteral<"deleteEmbeddings">;
+    requestId: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>, z.ZodObject<{
     modelId: z.ZodOptional<z.ZodString>;
     workspace: z.ZodOptional<z.ZodString>;
@@ -161,18 +166,22 @@ export declare const ragRequestSchema: z.ZodDiscriminatedUnion<[z.ZodObject<{
     withProgress: z.ZodOptional<z.ZodBoolean>;
     type: z.ZodLiteral<"rag">;
     operation: z.ZodLiteral<"reindex">;
+    requestId: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>, z.ZodObject<{
     type: z.ZodLiteral<"rag">;
     operation: z.ZodLiteral<"listWorkspaces">;
+    requestId: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>, z.ZodObject<{
     workspace: z.ZodOptional<z.ZodString>;
     deleteOnClose: z.ZodOptional<z.ZodBoolean>;
     type: z.ZodLiteral<"rag">;
     operation: z.ZodLiteral<"closeWorkspace">;
+    requestId: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>, z.ZodObject<{
     workspace: z.ZodPipe<z.ZodString, z.ZodString>;
     type: z.ZodLiteral<"rag">;
     operation: z.ZodLiteral<"deleteWorkspace">;
+    requestId: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>], "operation">;
 declare const ragWorkspaceInfoSchema: z.ZodObject<{
     name: z.ZodString;

package/dist/schemas/rag.js

@@ -131,16 +131,24 @@ const ragDeleteWorkspaceOperationSchema = ragDeleteWorkspaceParamsSchema.extend(
     operation: z.literal("deleteWorkspace"),
 });
 // ============== Unified Request Schema ==============
+// `requestId` is threaded onto every RAG operation as an optional field so
+// the request registry can correlate the in-flight context with the
+// client-side decorated-promise (`op.requestId`). Optional on the wire
+// so legacy clients keep working — the server falls back to a
+// server-generated id when the field is missing.
+const ragRequestIdField = {
+    requestId: z.string().min(1).optional(),
+};
 export const ragRequestSchema = z.discriminatedUnion("operation", [
-    ragChunkOperationSchema,
-    ragIngestOperationSchema,
-    ragSaveEmbeddingsOperationSchema,
-    ragSearchOperationSchema,
-    ragDeleteEmbeddingsOperationSchema,
-    ragReindexOperationSchema,
-    ragListWorkspacesOperationSchema,
-    ragCloseWorkspaceOperationSchema,
-    ragDeleteWorkspaceOperationSchema,
+    ragChunkOperationSchema.extend(ragRequestIdField),
+    ragIngestOperationSchema.extend(ragRequestIdField),
+    ragSaveEmbeddingsOperationSchema.extend(ragRequestIdField),
+    ragSearchOperationSchema.extend(ragRequestIdField),
+    ragDeleteEmbeddingsOperationSchema.extend(ragRequestIdField),
+    ragReindexOperationSchema.extend(ragRequestIdField),
+    ragListWorkspacesOperationSchema.extend(ragRequestIdField),
+    ragCloseWorkspaceOperationSchema.extend(ragRequestIdField),
+    ragDeleteWorkspaceOperationSchema.extend(ragRequestIdField),
 ]);
 // ============== Response Schemas ==============
 const ragResponseBaseSchema = z.object({

package/dist/schemas/registry.d.ts

@@ -1,13 +1,13 @@
 import { z } from "zod";
 declare const modelRegistryEntryAddonSchema: z.ZodEnum<{
     embeddings: "embeddings";
+    diffusion: "diffusion";
+    tts: "tts";
+    ocr: "ocr";
     llm: "llm";
     whisper: "whisper";
     nmt: "nmt";
     parakeet: "parakeet";
-    tts: "tts";
-    ocr: "ocr";
-    diffusion: "diffusion";
     vad: "vad";
     other: "other";
 }>;
@@ -33,13 +33,13 @@ export declare const modelRegistryEntrySchema: z.ZodObject<{
     modelId: z.ZodString;
     addon: z.ZodEnum<{
         embeddings: "embeddings";
+        diffusion: "diffusion";
+        tts: "tts";
+        ocr: "ocr";
         llm: "llm";
         whisper: "whisper";
         nmt: "nmt";
         parakeet: "parakeet";
-        tts: "tts";
-        ocr: "ocr";
-        diffusion: "diffusion";
         vad: "vad";
         other: "other";
     }>;
@@ -79,13 +79,13 @@ export declare const modelRegistryListResponseSchema: z.ZodObject<{
         modelId: z.ZodString;
         addon: z.ZodEnum<{
             embeddings: "embeddings";
+            diffusion: "diffusion";
+            tts: "tts";
+            ocr: "ocr";
             llm: "llm";
             whisper: "whisper";
             nmt: "nmt";
             parakeet: "parakeet";
-            tts: "tts";
-            ocr: "ocr";
-            diffusion: "diffusion";
             vad: "vad";
             other: "other";
         }>;
@@ -116,13 +116,13 @@ export declare const modelRegistrySearchRequestSchema: z.ZodObject<{
     quantization: z.ZodOptional<z.ZodString>;
     addon: z.ZodOptional<z.ZodEnum<{
         embeddings: "embeddings";
+        diffusion: "diffusion";
+        tts: "tts";
+        ocr: "ocr";
         llm: "llm";
         whisper: "whisper";
         nmt: "nmt";
         parakeet: "parakeet";
-        tts: "tts";
-        ocr: "ocr";
-        diffusion: "diffusion";
         vad: "vad";
         other: "other";
     }>>;
@@ -141,13 +141,13 @@ export declare const modelRegistrySearchResponseSchema: z.ZodObject<{
         modelId: z.ZodString;
         addon: z.ZodEnum<{
             embeddings: "embeddings";
+            diffusion: "diffusion";
+            tts: "tts";
+            ocr: "ocr";
             llm: "llm";
             whisper: "whisper";
             nmt: "nmt";
             parakeet: "parakeet";
-            tts: "tts";
-            ocr: "ocr";
-            diffusion: "diffusion";
             vad: "vad";
             other: "other";
         }>;
@@ -190,13 +190,13 @@ export declare const modelRegistryGetModelResponseSchema: z.ZodObject<{
         modelId: z.ZodString;
         addon: z.ZodEnum<{
             embeddings: "embeddings";
+            diffusion: "diffusion";
+            tts: "tts";
+            ocr: "ocr";
             llm: "llm";
             whisper: "whisper";
             nmt: "nmt";
             parakeet: "parakeet";
-            tts: "tts";
-            ocr: "ocr";
-            diffusion: "diffusion";
             vad: "vad";
             other: "other";
         }>;
@@ -229,13 +229,13 @@ export declare const modelRegistryRequestSchema: z.ZodUnion<readonly [z.ZodObjec
     quantization: z.ZodOptional<z.ZodString>;
     addon: z.ZodOptional<z.ZodEnum<{
         embeddings: "embeddings";
+        diffusion: "diffusion";
+        tts: "tts";
+        ocr: "ocr";
         llm: "llm";
         whisper: "whisper";
         nmt: "nmt";
         parakeet: "parakeet";
-        tts: "tts";
-        ocr: "ocr";
-        diffusion: "diffusion";
         vad: "vad";
         other: "other";
     }>>;
@@ -258,13 +258,13 @@ export declare const modelRegistryResponseSchema: z.ZodDiscriminatedUnion<[z.Zod
         modelId: z.ZodString;
         addon: z.ZodEnum<{
             embeddings: "embeddings";
+            diffusion: "diffusion";
+            tts: "tts";
+            ocr: "ocr";
             llm: "llm";
             whisper: "whisper";
             nmt: "nmt";
             parakeet: "parakeet";
-            tts: "tts";
-            ocr: "ocr";
-            diffusion: "diffusion";
             vad: "vad";
             other: "other";
         }>;
@@ -299,13 +299,13 @@ export declare const modelRegistryResponseSchema: z.ZodDiscriminatedUnion<[z.Zod
         modelId: z.ZodString;
         addon: z.ZodEnum<{
             embeddings: "embeddings";
+            diffusion: "diffusion";
+            tts: "tts";
+            ocr: "ocr";
             llm: "llm";
             whisper: "whisper";
             nmt: "nmt";
             parakeet: "parakeet";
-            tts: "tts";
-            ocr: "ocr";
-            diffusion: "diffusion";
             vad: "vad";
             other: "other";
         }>;
@@ -340,13 +340,13 @@ export declare const modelRegistryResponseSchema: z.ZodDiscriminatedUnion<[z.Zod
         modelId: z.ZodString;
         addon: z.ZodEnum<{
             embeddings: "embeddings";
+            diffusion: "diffusion";
+            tts: "tts";
+            ocr: "ocr";
             llm: "llm";
             whisper: "whisper";
             nmt: "nmt";
             parakeet: "parakeet";
-            tts: "tts";
-            ocr: "ocr";
-            diffusion: "diffusion";
             vad: "vad";
             other: "other";
         }>;