@ubiquity-os/plugin-sdk 3.6.3 → 3.7.0

package/dist/llm.js

@@ -31,6 +31,14 @@ function normalizeBaseUrl(baseUrl) {
   }
   return normalized;
 }
+var MAX_LLM_RETRIES = 2;
+var RETRY_BACKOFF_MS = [250, 750];
+function getRetryDelayMs(attempt) {
+  return RETRY_BACKOFF_MS[Math.min(attempt, RETRY_BACKOFF_MS.length - 1)] ?? 750;
+}
+function sleep(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
 function getEnvString(name) {
   if (typeof process === "undefined" || !process?.env) return EMPTY_STRING;
   return String(process.env[name] ?? EMPTY_STRING).trim();
@@ -45,7 +53,11 @@ function getAiBaseUrl(options) {
 }
 async function callLlm(options, input) {
   const authToken = String(input.authToken ?? EMPTY_STRING).trim();
-  if (!authToken) throw new Error("Missing authToken in input");
+  if (!authToken) {
+    const err = new Error("Missing authToken in input");
+    err.status = 401;
+    throw err;
+  }
   const kernelToken = "ubiquityKernelToken" in input ? input.ubiquityKernelToken : void 0;
   const payload = getPayload(input);
   const { owner, repo, installationId } = getRepoMetadata(payload);
@@ -65,13 +77,7 @@ async function callLlm(options, input) {
     installationId,
     ubiquityKernelToken: kernelToken
   });
-  const response = await fetch(url, { method: "POST", headers, body });
-  if (!response.ok) {
-    const err = await response.text();
-    const error = new Error(`LLM API error: ${response.status} - ${err}`);
-    error.status = response.status;
-    throw error;
-  }
+  const response = await fetchWithRetry(url, { method: "POST", headers, body }, MAX_LLM_RETRIES);
   if (isStream) {
     if (!response.body) {
       throw new Error("LLM API error: missing response body for streaming request");
@@ -83,17 +89,46 @@ async function callLlm(options, input) {
 function ensureKernelToken(authToken, kernelToken) {
   const isKernelTokenRequired = authToken.startsWith("gh");
   if (isKernelTokenRequired && !kernelToken) {
-    throw new Error("Missing ubiquityKernelToken in input (kernel attestation is required for GitHub auth)");
+    const err = new Error("Missing ubiquityKernelToken in input (kernel attestation is required for GitHub auth)");
+    err.status = 401;
+    throw err;
   }
 }
 function ensureMessages(messages) {
   if (!Array.isArray(messages) || messages.length === 0) {
-    throw new Error("messages must be a non-empty array");
+    const err = new Error("messages must be a non-empty array");
+    err.status = 400;
+    throw err;
   }
 }
 function buildAiUrl(options, baseUrl) {
   return `${getAiBaseUrl({ ...options, baseUrl })}/v1/chat/completions`;
 }
+async function fetchWithRetry(url, options, maxRetries) {
+  let attempt = 0;
+  let lastError;
+  while (attempt <= maxRetries) {
+    try {
+      const response = await fetch(url, options);
+      if (response.ok) return response;
+      const errText = await response.text();
+      if (response.status >= 500 && attempt < maxRetries) {
+        await sleep(getRetryDelayMs(attempt));
+        attempt += 1;
+        continue;
+      }
+      const error = new Error(`LLM API error: ${response.status} - ${errText}`);
+      error.status = response.status;
+      throw error;
+    } catch (error) {
+      lastError = error;
+      if (attempt >= maxRetries) throw error;
+      await sleep(getRetryDelayMs(attempt));
+      attempt += 1;
+    }
+  }
+  throw lastError ?? new Error("LLM API error: request failed after retries");
+}
 function getPayload(input) {
   if ("payload" in input) {
     return input.payload;
@@ -155,7 +190,7 @@ function getEventData(event) {
   if (!event.trim()) return null;
   const dataLines = event.split("\n").filter((line) => line.startsWith("data:"));
   if (!dataLines.length) return null;
-  const data = dataLines.map((line) => line.startsWith("data: ") ? line.slice(6) : line.slice(5).replace(/^ /, "")).join("\n");
+  const data = dataLines.map((line) => line.startsWith("data: ") ? line.slice(6) : line.slice(5).replace(/^ /, EMPTY_STRING)).join("\n");
   return data || null;
 }
 function parseEventData(data) {

package/dist/llm.mjs

@@ -7,6 +7,14 @@ function normalizeBaseUrl(baseUrl) {
   }
   return normalized;
 }
+var MAX_LLM_RETRIES = 2;
+var RETRY_BACKOFF_MS = [250, 750];
+function getRetryDelayMs(attempt) {
+  return RETRY_BACKOFF_MS[Math.min(attempt, RETRY_BACKOFF_MS.length - 1)] ?? 750;
+}
+function sleep(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
 function getEnvString(name) {
   if (typeof process === "undefined" || !process?.env) return EMPTY_STRING;
   return String(process.env[name] ?? EMPTY_STRING).trim();
@@ -21,7 +29,11 @@ function getAiBaseUrl(options) {
 }
 async function callLlm(options, input) {
   const authToken = String(input.authToken ?? EMPTY_STRING).trim();
-  if (!authToken) throw new Error("Missing authToken in input");
+  if (!authToken) {
+    const err = new Error("Missing authToken in input");
+    err.status = 401;
+    throw err;
+  }
   const kernelToken = "ubiquityKernelToken" in input ? input.ubiquityKernelToken : void 0;
   const payload = getPayload(input);
   const { owner, repo, installationId } = getRepoMetadata(payload);
@@ -41,13 +53,7 @@ async function callLlm(options, input) {
     installationId,
     ubiquityKernelToken: kernelToken
   });
-  const response = await fetch(url, { method: "POST", headers, body });
-  if (!response.ok) {
-    const err = await response.text();
-    const error = new Error(`LLM API error: ${response.status} - ${err}`);
-    error.status = response.status;
-    throw error;
-  }
+  const response = await fetchWithRetry(url, { method: "POST", headers, body }, MAX_LLM_RETRIES);
   if (isStream) {
     if (!response.body) {
       throw new Error("LLM API error: missing response body for streaming request");
@@ -59,17 +65,46 @@ async function callLlm(options, input) {
 function ensureKernelToken(authToken, kernelToken) {
   const isKernelTokenRequired = authToken.startsWith("gh");
   if (isKernelTokenRequired && !kernelToken) {
-    throw new Error("Missing ubiquityKernelToken in input (kernel attestation is required for GitHub auth)");
+    const err = new Error("Missing ubiquityKernelToken in input (kernel attestation is required for GitHub auth)");
+    err.status = 401;
+    throw err;
   }
 }
 function ensureMessages(messages) {
   if (!Array.isArray(messages) || messages.length === 0) {
-    throw new Error("messages must be a non-empty array");
+    const err = new Error("messages must be a non-empty array");
+    err.status = 400;
+    throw err;
   }
 }
 function buildAiUrl(options, baseUrl) {
   return `${getAiBaseUrl({ ...options, baseUrl })}/v1/chat/completions`;
 }
+async function fetchWithRetry(url, options, maxRetries) {
+  let attempt = 0;
+  let lastError;
+  while (attempt <= maxRetries) {
+    try {
+      const response = await fetch(url, options);
+      if (response.ok) return response;
+      const errText = await response.text();
+      if (response.status >= 500 && attempt < maxRetries) {
+        await sleep(getRetryDelayMs(attempt));
+        attempt += 1;
+        continue;
+      }
+      const error = new Error(`LLM API error: ${response.status} - ${errText}`);
+      error.status = response.status;
+      throw error;
+    } catch (error) {
+      lastError = error;
+      if (attempt >= maxRetries) throw error;
+      await sleep(getRetryDelayMs(attempt));
+      attempt += 1;
+    }
+  }
+  throw lastError ?? new Error("LLM API error: request failed after retries");
+}
 function getPayload(input) {
   if ("payload" in input) {
     return input.payload;
@@ -131,7 +166,7 @@ function getEventData(event) {
   if (!event.trim()) return null;
   const dataLines = event.split("\n").filter((line) => line.startsWith("data:"));
   if (!dataLines.length) return null;
-  const data = dataLines.map((line) => line.startsWith("data: ") ? line.slice(6) : line.slice(5).replace(/^ /, "")).join("\n");
+  const data = dataLines.map((line) => line.startsWith("data: ") ? line.slice(6) : line.slice(5).replace(/^ /, EMPTY_STRING)).join("\n");
   return data || null;
 }
 function parseEventData(data) {

package/dist/signature.js

@@ -83,7 +83,7 @@ async function verifySignature(publicKeyPem, inputs, signature) {
       ref: inputs.ref,
       command: inputs.command
     };
-    const pemContents = publicKeyPem.replace("-----BEGIN PUBLIC KEY-----", "").replace("-----END PUBLIC KEY-----", "").trim();
+    const pemContents = publicKeyPem.replace("-----BEGIN PUBLIC KEY-----", "").replace("-----END PUBLIC KEY-----", "").replace(/\s+/g, "");
     const binaryDer = Uint8Array.from(atob(pemContents), (c) => c.charCodeAt(0));
     const publicKey = await crypto.subtle.importKey(
       "spki",
@@ -104,7 +104,7 @@ async function verifySignature(publicKeyPem, inputs, signature) {
   }
 }
 async function importRsaPrivateKey(pem) {
-  const pemContents = pem.replace("-----BEGIN PRIVATE KEY-----", "").replace("-----END PRIVATE KEY-----", "").trim();
+  const pemContents = pem.replace("-----BEGIN PRIVATE KEY-----", "").replace("-----END PRIVATE KEY-----", "").replace(/\s+/g, "");
   const binaryDer = Uint8Array.from(atob(pemContents), (c) => c.charCodeAt(0));
   return await crypto.subtle.importKey(
     "pkcs8",

package/dist/signature.mjs

@@ -55,7 +55,7 @@ async function verifySignature(publicKeyPem, inputs, signature) {
       ref: inputs.ref,
       command: inputs.command
     };
-    const pemContents = publicKeyPem.replace("-----BEGIN PUBLIC KEY-----", "").replace("-----END PUBLIC KEY-----", "").trim();
+    const pemContents = publicKeyPem.replace("-----BEGIN PUBLIC KEY-----", "").replace("-----END PUBLIC KEY-----", "").replace(/\s+/g, "");
     const binaryDer = Uint8Array.from(atob(pemContents), (c) => c.charCodeAt(0));
     const publicKey = await crypto.subtle.importKey(
       "spki",
@@ -76,7 +76,7 @@ async function verifySignature(publicKeyPem, inputs, signature) {
   }
 }
 async function importRsaPrivateKey(pem) {
-  const pemContents = pem.replace("-----BEGIN PRIVATE KEY-----", "").replace("-----END PRIVATE KEY-----", "").trim();
+  const pemContents = pem.replace("-----BEGIN PRIVATE KEY-----", "").replace("-----END PRIVATE KEY-----", "").replace(/\s+/g, "");
   const binaryDer = Uint8Array.from(atob(pemContents), (c) => c.charCodeAt(0));
   return await crypto.subtle.importKey(
     "pkcs8",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ubiquity-os/plugin-sdk",
-  "version": "3.6.3",
+  "version": "3.7.0",
   "description": "SDK for plugin support.",
   "author": "Ubiquity DAO",
   "license": "MIT",