@mozilla-ai/mcpd 0.0.4 → 0.1.0

package/dist/index.mjs

@@ -1,5 +1,7 @@
 import { LRUCache } from "lru-cache";
 import { z } from "zod";
+const PIPELINE_FLOW_REQUEST = "request";
+const PIPELINE_FLOW_RESPONSE = "response";
 class McpdError extends Error {
   constructor(message, cause) {
     super(message);
@@ -86,6 +88,19 @@ class TimeoutError extends McpdError {
     Error.captureStackTrace(this, this.constructor);
   }
 }
+class PipelineError extends McpdError {
+  serverName;
+  operation;
+  pipelineFlow;
+  constructor(message, serverName, operation, pipelineFlow, cause) {
+    super(message, cause);
+    this.name = "PipelineError";
+    this.serverName = serverName;
+    this.operation = operation;
+    this.pipelineFlow = pipelineFlow;
+    Error.captureStackTrace(this, this.constructor);
+  }
+}
 var HealthStatus = /* @__PURE__ */ ((HealthStatus2) => {
   HealthStatus2["OK"] = "ok";
   HealthStatus2["TIMEOUT"] = "timeout";
@@ -1042,6 +1057,11 @@ const REQUEST_TIMEOUT_SECONDS = 30;
 const SERVER_HEALTH_CACHE_TTL_SECONDS = 10;
 const SERVER_HEALTH_CACHE_MAXSIZE = 100;
 const TOOL_SEPARATOR = "__";
+const MCPD_ERROR_TYPE_HEADER = "Mcpd-Error-Type";
+const PIPELINE_ERROR_FLOWS = {
+  "request-pipeline-failure": PIPELINE_FLOW_REQUEST,
+  "response-pipeline-failure": PIPELINE_FLOW_RESPONSE
+};
 class McpdClient {
   #endpoint;
   #apiKey;
@@ -1128,6 +1148,21 @@ class McpdClient {
         } catch {
           errorModel = null;
         }
+        if (response.status === 500) {
+          const errorType = response.headers.get(MCPD_ERROR_TYPE_HEADER)?.toLowerCase();
+          const flow = errorType ? PIPELINE_ERROR_FLOWS[errorType] : void 0;
+          if (flow) {
+            const message = errorModel?.detail || body || "Pipeline failure";
+            throw new PipelineError(
+              message,
+              void 0,
+              // serverName - enriched by caller if available.
+              void 0,
+              // operation - enriched by caller if available.
+              flow
+            );
+          }
+        }
         if (errorModel && errorModel.detail) {
           const errorDetails = errorModel.errors?.map((e) => `${e.location}: ${e.message}`).join("; ");
           const fullMessage = errorDetails ? `${errorModel.detail} - ${errorDetails}` : errorModel.detail;
@@ -1560,6 +1595,15 @@ class McpdClient {
       }
       return response;
     } catch (error) {
+      if (error instanceof PipelineError) {
+        throw new PipelineError(
+          error.message,
+          serverName,
+          `${serverName}.${toolName}`,
+          error.pipelineFlow,
+          error.cause
+        );
+      }
       if (error instanceof McpdError) {
         throw error;
       }
@@ -1678,6 +1722,9 @@ export {
   HealthStatusHelpers,
   McpdClient,
   McpdError,
+  PIPELINE_FLOW_REQUEST,
+  PIPELINE_FLOW_RESPONSE,
+  PipelineError,
   ServerNotFoundError,
   ServerUnhealthyError,
   TimeoutError,