@cueai/omni-reader-mcp 1.5.5 → 1.7.0

package/dist/tools.js

@@ -2,12 +2,12 @@ import { createHash } from "node:crypto";
 import { CallToolRequestSchema, CancelTaskRequestSchema, ErrorCode, McpError, } from "@modelcontextprotocol/sdk/types.js";
 import { selectDirectProfile, selectUrlProfile, } from "./capabilities.js";
 import { createClientRequestId as newClientRequestId, } from "./cube-client.js";
-import { OmniBridgeError } from "./errors.js";
+import { OmniBridgeError, unsupportedDetailError } from "./errors.js";
 import { hydrateInlineUrlResultSafely } from "./operation-manager.js";
 import { openAllowedFile, } from "./path-security.js";
 import { NOOP_PROGRESS } from "./progress.js";
 import { MACHINE_INSTRUCTIONS, cancelParseSchema, discardResultSchema, getParseStatusSchema, normalizeParseArguments, normalizeRepresentation, parseSchema, parseSchemaShape, readOutlineSchema, readResultSchema, saveResultSchema, } from "./protocol.js";
-import { discardResultOutputSchema, discardResultToolOutputSchema, localResultToResultField, parseResultSchema, parseToolOutputSchema, readOutlineOutputSchema, readOutlineToolOutputSchema, readResultOutputSchema, readResultToolOutputSchema, saveResultOutputSchema, saveResultToolOutputSchema, structuredResult, } from "./result-contract.js";
+import { createFlatLocalResultCache, discardResultOutputSchema, discardResultToolOutputSchema, localResultToResultField, parseResultSchema, parseToolOutputSchema, presentRetainedResult, readOutlineOutputSchema, readOutlineToolOutputSchema, readResultOutputSchema, readResultToolOutputSchema, saveResultOutputSchema, saveResultToolOutputSchema, structuredResult, } from "./result-contract.js";
 import { classifySource } from "./source.js";
 import { TaskRuntime } from "./task-runtime.js";
 function bridgeError(code, message, facts = {}) {
@@ -58,9 +58,11 @@ function completedLocalResult(local) {
         operation_id: local.operationId,
         result,
         data_handling: dataHandling,
-        ...(result.kind === "artifact"
-            ? { local_result_cache: { expires_at: local.expiresAt, discard_action: "discard_result" } }
-            : {}),
+        local_result_cache: createFlatLocalResultCache({
+            resultId: local.resultId,
+            resultBytes: local.resultBytes,
+            expiresAt: local.expiresAt,
+        }),
     };
 }
 async function parseLocal(args, clientRequestId, signal, progress, dependencies) {
@@ -127,29 +129,35 @@ async function parseLocal(args, clientRequestId, signal, progress, dependencies)
 async function preflightNonTextRepresentation(detail, sourceKind, signal, dependencies) {
     if (sourceKind === "local") {
         const capabilities = await dependencies.cubeClient.getCapabilities(signal);
-        selectDirectProfile(capabilities, detail);
+        selectDirectProfile(capabilities, detail, "local");
     }
     else {
         if (dependencies.remoteClient === undefined) {
             throw bridgeError("REMOTE_PARSE_UNAVAILABLE", "Remote URL parsing is not available in this Bridge build.", { retryable: true });
         }
         const capabilities = await dependencies.remoteClient.initializeCapabilities(signal);
-        selectUrlProfile(capabilities, detail);
+        selectUrlProfile(capabilities, detail, "url");
     }
     if (dependencies.operationManager === undefined) {
-        throw new OmniBridgeError({
-            code: "UNSUPPORTED_DETAIL",
-            message: "This Bridge build cannot deliver the requested output detail.",
-            failureScope: "bridge",
-            userAction: "Use plain Markdown output for this source.",
-            operationCreated: false,
-            fileUploaded: false,
-            parserStarted: false,
-            billed: false,
-            contentReleased: false,
-            retryable: false,
-        });
+        throw unsupportedDetailError(sourceKind);
+    }
+}
+async function preflightArtifactDelivery(args, dependencies) {
+    if (args.resultDelivery !== "artifact")
+        return;
+    if (dependencies.artifactStore.preflightWrite === undefined
+        || dependencies.artifactStore.mintReadCursor === undefined) {
+        throw bridgeError("ARTIFACT_PREFLIGHT_FAILED", "The local result store cannot provide artifact delivery.");
     }
+    await dependencies.artifactStore.preflightWrite();
+}
+async function presentParseResult(result, args, dependencies) {
+    return presentRetainedResult(result, args.resultDelivery, async (resultId, byteOffset) => {
+        if (dependencies.artifactStore.mintReadCursor === undefined) {
+            throw bridgeError("LOCAL_RESULT_INTEGRITY_FAILED", "The retained result cursor capability is unavailable.");
+        }
+        return dependencies.artifactStore.mintReadCursor(resultId, byteOffset);
+    });
 }
 async function parseValue(args, signal, progress, dependencies) {
     try {
@@ -198,19 +206,23 @@ async function parseValue(args, signal, progress, dependencies) {
                 },
                 clientRequestId,
                 signal,
+                resultDelivery: args.resultDelivery,
                 context: source.kind === "local"
                     ? { source: source.source, progress }
                     : { source: source.source },
             });
         }
+        await preflightArtifactDelivery(args, dependencies);
         if (source.kind === "url") {
             if (dependencies.remoteClient === undefined) {
                 throw bridgeError("REMOTE_PARSE_UNAVAILABLE", "Remote URL parsing is not available in this Bridge build.", { retryable: true });
             }
             const remoteResult = await dependencies.remoteClient.parse(source.source, clientRequestId, signal);
-            return await hydrateInlineUrlResultSafely(remoteResult, dependencies.artifactStore);
+            const hydrated = await hydrateInlineUrlResultSafely(remoteResult, dependencies.artifactStore);
+            return presentParseResult(hydrated, args, dependencies);
         }
-        return await parseLocal(args, clientRequestId, signal, progress, dependencies);
+        const localResult = await parseLocal(args, clientRequestId, signal, progress, dependencies);
+        return presentParseResult(localResult, args, dependencies);
     }
     catch (error) {
         return failed(error);
@@ -325,10 +337,14 @@ async function callReadResult(resultId, cursor, maxBytes, dependencies) {
 }
 async function callDiscardResult(resultId, dependencies) {
     try {
+        const discarded = await dependencies.artifactStore.discard(resultId);
+        if (!discarded) {
+            throw bridgeError("RESULT_NOT_FOUND", "The requested local result was not found.");
+        }
         return structuredResult(discardResultOutputSchema, {
             status: "completed",
             result_id: resultId,
-            discarded: await dependencies.artifactStore.discard(resultId),
+            discarded,
         });
     }
     catch (error) {
@@ -361,13 +377,7 @@ async function callReadOutline(resultId, nodeId, dependencies) {
         }
         const node = outline.nodes.find((candidate) => candidate.id === nodeId);
         if (node === undefined) {
-            throw bridgeError("OUTLINE_NODE_NOT_FOUND", "The requested outline node does not exist.", {
-                operationCreated: true,
-                fileUploaded: true,
-                parserStarted: true,
-                billed: true,
-                contentReleased: true,
-            });
+            throw bridgeError("OUTLINE_NODE_NOT_FOUND", "The requested outline node does not exist.");
         }
         const cursor = await dependencies.artifactStore.mintOutlineCursor(resultId, node.byteOffset);
         return structuredResult(readOutlineOutputSchema, { status: "cursor", cursor });
@@ -455,7 +465,7 @@ async function dispatchTool(name, rawArguments, extra, dependencies) {
 }
 export function registerOmniTools(server, dependencies, taskStore) {
     const taskRuntime = new TaskRuntime({
-        parse: (source, signal) => parseValue({ source }, signal, NOOP_PROGRESS, dependencies),
+        parse: (args, signal) => parseValue(args, signal, NOOP_PROGRESS, dependencies),
         status: (operationId, waitMs, signal) => statusValue(operationId, waitMs, signal, dependencies),
         cancel: (operationId, signal) => cancelValue(operationId, signal, dependencies),
     });
@@ -466,7 +476,7 @@ export function registerOmniTools(server, dependencies, taskStore) {
             ttl: Math.min(600_000, Math.max(1_000, requestedTtl)),
             pollInterval: 1_000,
         });
-        void taskRuntime.start(task.taskId, normalized.source, extra.taskStore)
+        void taskRuntime.start(task.taskId, normalized, extra.taskStore)
             .catch(() => undefined);
         return { task };
     };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cueai/omni-reader-mcp",
-  "version": "1.5.5",
+  "version": "1.7.0",
   "description": "Local stdio MCP bridge for direct Omni document parsing",
   "keywords": [
     "mcp",