@cueai/omni-reader-mcp 1.0.0 → 1.0.2

package/README.md

@@ -1,4 +1,4 @@
-# @cue/omni-reader-mcp
+# @cueai/omni-reader-mcp
 
 Local stdio MCP Bridge for parsing files from an Agent workspace through Omni Reader. Source bytes upload directly from the local Bridge to IIIS; they do not pass through Cube-MCP or the model conversation before `parse_document` is called.
 
@@ -13,7 +13,7 @@ Set the existing general Cue API Key, then run the interactive user-scope setup:
 
 ```sh
 export CUE_API_KEY="YOUR_CUE_API_KEY"
-npx -y @cue/omni-reader-mcp setup
+npx -y @cueai/omni-reader-mcp setup
 ```
 
 Setup supports Cursor, Claude Desktop, and a generic MCP configuration for other Agents. The current working directory is allowed by default. To allow additional directories, provide them during setup or configure absolute paths in `OMNI_ALLOWED_ROOTS` (colon-separated on macOS/Linux and semicolon-separated on Windows).
@@ -41,9 +41,9 @@ Local parsing defaults to `no_store=true`. Files are restricted to configured ro
 ## Commands
 
 ```sh
-npx -y @cue/omni-reader-mcp setup
-npx -y @cue/omni-reader-mcp doctor
-npx -y @cue/omni-reader-mcp clean
+npx -y @cueai/omni-reader-mcp setup
+npx -y @cueai/omni-reader-mcp doctor
+npx -y @cueai/omni-reader-mcp clean
 ```
 
-Running `npx -y @cue/omni-reader-mcp` without a command starts the stdio MCP server.
+Running `npx -y @cueai/omni-reader-mcp` without a command starts the stdio MCP server.

package/dist/cli/agent-config.js

@@ -51,7 +51,7 @@ export function displayConfigPath(configPath, environment) {
 export function buildAgentEntry(extraRoots, platform) {
     const entry = {
         command: "npx",
-        args: ["-y", "@cue/omni-reader-mcp"],
+        args: ["-y", "@cueai/omni-reader-mcp"],
     };
     if (extraRoots.length > 0) {
         entry.env = {
@@ -277,7 +277,7 @@ function isExpectedOmniEntry(value) {
         || !Array.isArray(args)
         || args.length !== 2
         || args[0] !== "-y"
-        || args[1] !== "@cue/omni-reader-mcp")
+        || args[1] !== "@cueai/omni-reader-mcp")
         return false;
     if (value.env === undefined) {
         return keys.length === 2 && keys[0] === "args" && keys[1] === "command";

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 = "@cue/omni-reader-mcp";
-declare const BRIDGE_VERSION = "1.0.0";
+declare const BRIDGE_PACKAGE = "@cueai/omni-reader-mcp";
+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 = "@cue/omni-reader-mcp";
-const BRIDGE_VERSION = "1.0.0";
+const BRIDGE_PACKAGE = "@cueai/omni-reader-mcp";
+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/index.js

@@ -1,7 +1,7 @@
 #!/usr/bin/env node
 import { execFile as execFileCallback } from "node:child_process";
-import { readFile } from "node:fs/promises";
-import { pathToFileURL } from "node:url";
+import { readFile, realpath } from "node:fs/promises";
+import { fileURLToPath } from "node:url";
 import { createInterface } from "node:readline/promises";
 import { stdin, stdout as processStdout } from "node:process";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
@@ -165,8 +165,17 @@ async function main() {
     if (code !== 0)
         process.exitCode = code;
 }
-const invokedPath = process.argv[1];
-if (invokedPath !== undefined && import.meta.url === pathToFileURL(invokedPath).href) {
+async function isMainModule(invokedPath) {
+    if (invokedPath === undefined)
+        return false;
+    try {
+        return await realpath(invokedPath) === fileURLToPath(import.meta.url);
+    }
+    catch {
+        return false;
+    }
+}
+if (await isMainModule(process.argv[1])) {
     void main().catch(() => {
         process.stderr.write("Bridge startup failed.\n");
         process.exitCode = 1;

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.0";
+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.0",
+  "version": "1.0.2",
   "description": "Local stdio MCP bridge for direct Omni document parsing",
   "type": "module",
   "license": "UNLICENSED",