@cueai/omni-reader-mcp 1.0.1 → 1.0.2

package/dist/cli/doctor.js

@@ -1,7 +1,7 @@
 import { constants as fsConstants } from "node:fs";
 import { lstat, open, readdir } from "node:fs/promises";
 import path from "node:path";
-import { DEFAULT_CUBE_BASE_URL, GRANTED_STREAM_PROTOCOL_VERSION, MAX_FILE_BYTES, PROTOCOL_VERSION, } from "../constants.js";
+import { DEFAULT_CUBE_BASE_URL, GRANTED_STREAM_PROTOCOL_VERSION, MAX_FILE_BYTES, CUBE_GRANT_PROTOCOL_VERSION, } from "../constants.js";
 import { agentConfigPath, inspectAgentConfig, } from "./agent-config.js";
 const CUBE_HEALTH_PATH = "/api/omni-reader/direct-upload/v1/health";
 const IIIS_HEALTH_URL = "https://cubefile.ai.iiis.co:9443/omni/granted/health";
@@ -28,7 +28,7 @@ async function fetchHealth(fetchImpl, url, authorization) {
 export async function checkCubeHealth(fetchImpl, apiKey) {
     const cubeUrl = new URL(CUBE_HEALTH_PATH, DEFAULT_CUBE_BASE_URL).toString();
     const cube = await fetchHealth(fetchImpl, cubeUrl, `Bearer ${apiKey}`);
-    if (!cube.enabled || cube.protocol_version !== PROTOCOL_VERSION) {
+    if (!cube.enabled || cube.protocol_version !== CUBE_GRANT_PROTOCOL_VERSION) {
         throw new Error("Cube direct upload is disabled or incompatible");
     }
     return cube.protocol_version;

package/dist/constants.d.ts

@@ -2,6 +2,6 @@ export declare const MAX_FILE_BYTES = 268435456;
 export declare const INLINE_RESULT_MAX_BYTES = 65536;
 export declare const ARTIFACT_TTL_MS: number;
 export declare const RESULT_CHUNK_MAX_BYTES = 65536;
-export declare const PROTOCOL_VERSION = "omni.direct_upload.v1";
+export declare const CUBE_GRANT_PROTOCOL_VERSION = "omni.parse_grant.v1";
 export declare const GRANTED_STREAM_PROTOCOL_VERSION = "omni.granted_parse_stream.v1";
 export declare const DEFAULT_CUBE_BASE_URL = "https://mcp.cuecue.cn";

package/dist/constants.js

@@ -2,6 +2,6 @@ export const MAX_FILE_BYTES = 268435456;
 export const INLINE_RESULT_MAX_BYTES = 65536;
 export const ARTIFACT_TTL_MS = 24 * 60 * 60 * 1000;
 export const RESULT_CHUNK_MAX_BYTES = 65536;
-export const PROTOCOL_VERSION = "omni.direct_upload.v1";
+export const CUBE_GRANT_PROTOCOL_VERSION = "omni.parse_grant.v1";
 export const GRANTED_STREAM_PROTOCOL_VERSION = "omni.granted_parse_stream.v1";
 export const DEFAULT_CUBE_BASE_URL = "https://mcp.cuecue.cn";

package/dist/cube-client.d.ts

@@ -1,6 +1,6 @@
 import { OperationJournal } from "./operation-journal.js";
 declare const BRIDGE_PACKAGE = "@cueai/omni-reader-mcp";
-declare const BRIDGE_VERSION = "1.0.1";
+declare const BRIDGE_VERSION = "1.0.2";
 export interface GrantRequestInput {
     readonly contentLength: number;
     readonly contentType: string;

package/dist/cube-client.js

@@ -1,10 +1,10 @@
 import { createHash, randomUUID } from "node:crypto";
 import { z } from "zod";
-import { DEFAULT_CUBE_BASE_URL, MAX_FILE_BYTES, PROTOCOL_VERSION, } from "./constants.js";
+import { DEFAULT_CUBE_BASE_URL, MAX_FILE_BYTES, CUBE_GRANT_PROTOCOL_VERSION, } from "./constants.js";
 import { OmniBridgeError } from "./errors.js";
 const GRANT_PATH = "/api/omni-reader/direct-upload/v1/parse-grants";
 const BRIDGE_PACKAGE = "@cueai/omni-reader-mcp";
-const BRIDGE_VERSION = "1.0.1";
+const BRIDGE_VERSION = "1.0.2";
 const grantResponseSchema = z
     .object({
     grant_id: z.string().min(1),
@@ -17,7 +17,7 @@ const grantResponseSchema = z
         .refine((value) => new URL(value).protocol === "https:"),
     expires_at: z.string().datetime({ offset: true }),
     max_bytes: z.literal(MAX_FILE_BYTES),
-    protocol_version: z.literal(PROTOCOL_VERSION),
+    protocol_version: z.literal(CUBE_GRANT_PROTOCOL_VERSION),
 })
     .strict();
 function bridgeError(code, message, retryable) {

package/dist/iiis-client.d.ts

@@ -38,6 +38,6 @@ export declare class IiisClient {
     #private;
     constructor(options?: IiisClientOptions);
     uploadAndWait(input: IiisOperationInput): Promise<ReleasedResult>;
-    downloadResult(input: IiisOperationInput): Promise<ReleasedResult>;
+    downloadResult(input: IiisOperationInput, progress?: ProgressSink): Promise<ReleasedResult>;
     ack(input: IiisOperationInput): Promise<void>;
 }

package/dist/iiis-client.js

@@ -110,6 +110,11 @@ function monotonicProgress(sink) {
         },
     };
 }
+function parserProgress(done, total) {
+    return total > 0
+        ? 40 + Math.floor(Math.min(done / total, 1) * 59)
+        : 40;
+}
 async function errorFromResponse(response, input) {
     try {
         const body = await response.json();
@@ -217,7 +222,7 @@ export class IiisClient {
             return this.#recover(input, true, progress);
         }
     }
-    async downloadResult(input) {
+    async downloadResult(input, progress = NOOP_PROGRESS) {
         throwIfCanceled(input);
         const confirmedFacts = recoveryFacts({
             fileUploaded: true,
@@ -316,6 +321,7 @@ export class IiisClient {
             };
             await input.retention.complete(metadata);
             throwIfCanceled(input);
+            await progress.report(100, 100, "Completed");
             completed = true;
             return metadata;
         }
@@ -415,10 +421,7 @@ export class IiisClient {
                 }
                 if (event.type === "progress") {
                     const parsed = progressValue(event);
-                    const scaled = parsed.total > 0
-                        ? 40 + Math.floor(Math.min(parsed.done / parsed.total, 1) * 60)
-                        : 40;
-                    await progress.report(scaled, 100, parsed.message);
+                    await progress.report(parserProgress(parsed.done, parsed.total), 100, parsed.message);
                     continue;
                 }
                 if (event.type === "error") {
@@ -519,8 +522,11 @@ export class IiisClient {
                 }
                 throw bridgeError("INVALID_SSE_EVENT", "IIIS returned an unsupported progress event.");
             }
-            if (completedMetadata !== undefined)
+            if (completedMetadata !== undefined) {
+                throwIfCanceled(input);
+                await progress.report(100, 100, "Completed");
                 return completedMetadata;
+            }
             throw new TransportFailure("SSE ended before completion");
         }
         catch (error) {
@@ -643,16 +649,13 @@ export class IiisClient {
                     }
                 }
                 else if (status.status === "RELEASED") {
-                    return await this.downloadResult(input);
+                    return await this.downloadResult(input, progress);
                 }
                 else if (ACTIVE_STATUSES.has(status.status)) {
                     const page = await this.#events(input, eventCursor);
                     for (const event of page.events) {
                         const parsed = progressValue(event);
-                        const scaled = parsed.total > 0
-                            ? 40 + Math.floor(Math.min(parsed.done / parsed.total, 1) * 60)
-                            : 40;
-                        await progress.report(scaled, 100, parsed.message);
+                        await progress.report(parserProgress(parsed.done, parsed.total), 100, parsed.message);
                     }
                     eventCursor = page.nextSequence;
                 }

package/dist/server.js

@@ -2,7 +2,7 @@ import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { SAFETY_INSTRUCTIONS } from "./protocol.js";
 import { registerOmniTools } from "./tools.js";
 const SERVER_NAME = "omni-reader-mcp";
-const SERVER_VERSION = "1.0.1";
+const SERVER_VERSION = "1.0.2";
 export function createOmniMcpServer(dependencies) {
     const server = new McpServer({ name: SERVER_NAME, version: SERVER_VERSION }, { instructions: SAFETY_INSTRUCTIONS });
     registerOmniTools(server, dependencies);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cueai/omni-reader-mcp",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "Local stdio MCP bridge for direct Omni document parsing",
   "type": "module",
   "license": "UNLICENSED",