@ubiquity-os/plugin-sdk 3.6.2 → 3.7.0

package/dist/llm.d.mts

@@ -1,6 +1,6 @@
-import { C as Context } from './context-sqbr2o6i.mjs';
-import { PluginInput } from './signature.mjs';
 import { ChatCompletionMessageParam, ChatCompletionCreateParamsNonStreaming, ChatCompletion, ChatCompletionChunk } from 'openai/resources/chat/completions';
+import { C as Context } from './context-Ckj1HMjz.mjs';
+import { PluginInput } from './signature.mjs';
 import '@octokit/webhooks';
 import '@ubiquity-os/ubiquity-os-logger';
 import '@octokit/plugin-rest-endpoint-methods';

package/dist/llm.d.ts

@@ -1,6 +1,6 @@
-import { C as Context } from './context-BbEmsEct.js';
-import { PluginInput } from './signature.js';
 import { ChatCompletionMessageParam, ChatCompletionCreateParamsNonStreaming, ChatCompletion, ChatCompletionChunk } from 'openai/resources/chat/completions';
+import { C as Context } from './context-BE4WjJZf.js';
+import { PluginInput } from './signature.js';
 import '@octokit/webhooks';
 import '@ubiquity-os/ubiquity-os-logger';
 import '@octokit/plugin-rest-endpoint-methods';

package/dist/llm.js

@@ -23,86 +23,185 @@ __export(llm_exports, {
   callLlm: () => callLlm
 });
 module.exports = __toCommonJS(llm_exports);
+var EMPTY_STRING = "";
 function normalizeBaseUrl(baseUrl) {
-  return baseUrl.trim().replace(/\/+$/, "");
+  let normalized = baseUrl.trim();
+  while (normalized.endsWith("/")) {
+    normalized = normalized.slice(0, -1);
+  }
+  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 "";
-  return String(process.env[name] ?? "").trim();
+  if (typeof process === "undefined" || !process?.env) return EMPTY_STRING;
+  return String(process.env[name] ?? EMPTY_STRING).trim();
 }
 function getAiBaseUrl(options) {
   if (typeof options.baseUrl === "string" && options.baseUrl.trim()) {
     return normalizeBaseUrl(options.baseUrl);
   }
-  const envBaseUrl = getEnvString("UBQ_AI_BASE_URL") || getEnvString("UBQ_AI_URL");
+  const envBaseUrl = getEnvString("UOS_AI_URL") || getEnvString("UOS_AI_BASE_URL");
   if (envBaseUrl) return normalizeBaseUrl(envBaseUrl);
-  return "https://ai.ubq.fi";
+  return "https://ai-ubq-fi.deno.dev";
 }
 async function callLlm(options, input) {
-  const authToken = input?.authToken ?? "";
-  const ubiquityKernelToken = input?.ubiquityKernelToken ?? "";
-  const payload = input?.payload;
-  const owner = payload?.repository?.owner?.login ?? "";
-  const repo = payload?.repository?.name ?? "";
-  const installationId = payload?.installation?.id;
-  if (!authToken) throw new Error("Missing authToken in inputs");
-  const requiresKernelToken = authToken.trim().startsWith("gh");
-  if (requiresKernelToken && !ubiquityKernelToken) {
-    throw new Error("Missing ubiquityKernelToken in inputs (kernel attestation is required for GitHub auth)");
+  const authToken = String(input.authToken ?? EMPTY_STRING).trim();
+  if (!authToken) {
+    const err = new Error("Missing authToken in input");
+    err.status = 401;
+    throw err;
   }
-  const url = `${getAiBaseUrl(options)}/v1/chat/completions`;
-  const { baseUrl: _baseUrl, model, stream, messages, ...rest } = options;
+  const kernelToken = "ubiquityKernelToken" in input ? input.ubiquityKernelToken : void 0;
+  const payload = getPayload(input);
+  const { owner, repo, installationId } = getRepoMetadata(payload);
+  ensureKernelToken(authToken, kernelToken);
+  const { baseUrl, model, stream: isStream, messages, ...rest } = options;
+  ensureMessages(messages);
+  const url = buildAiUrl(options, baseUrl);
   const body = JSON.stringify({
     ...rest,
     ...model ? { model } : {},
     messages,
-    stream: stream ?? false
+    stream: isStream ?? false
+  });
+  const headers = buildHeaders(authToken, {
+    owner,
+    repo,
+    installationId,
+    ubiquityKernelToken: kernelToken
   });
+  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");
+    }
+    return parseSseStream(response.body);
+  }
+  return response.json();
+}
+function ensureKernelToken(authToken, kernelToken) {
+  const isKernelTokenRequired = authToken.startsWith("gh");
+  if (isKernelTokenRequired && !kernelToken) {
+    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) {
+    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;
+  }
+  return input.eventPayload;
+}
+function getRepoMetadata(payload) {
+  const repoPayload = payload;
+  return {
+    owner: repoPayload?.repository?.owner?.login ?? EMPTY_STRING,
+    repo: repoPayload?.repository?.name ?? EMPTY_STRING,
+    installationId: repoPayload?.installation?.id
+  };
+}
+function buildHeaders(authToken, options) {
   const headers = {
     Authorization: `Bearer ${authToken}`,
     "Content-Type": "application/json"
   };
-  if (owner) headers["X-GitHub-Owner"] = owner;
-  if (repo) headers["X-GitHub-Repo"] = repo;
-  if (typeof installationId === "number" && Number.isFinite(installationId)) {
-    headers["X-GitHub-Installation-Id"] = String(installationId);
+  if (options.owner) headers["X-GitHub-Owner"] = options.owner;
+  if (options.repo) headers["X-GitHub-Repo"] = options.repo;
+  if (typeof options.installationId === "number" && Number.isFinite(options.installationId)) {
+    headers["X-GitHub-Installation-Id"] = String(options.installationId);
   }
-  if (ubiquityKernelToken) {
-    headers["X-Ubiquity-Kernel-Token"] = ubiquityKernelToken;
+  if (options.ubiquityKernelToken) {
+    headers["X-Ubiquity-Kernel-Token"] = options.ubiquityKernelToken;
   }
-  const response = await fetch(url, { method: "POST", headers, body });
-  if (!response.ok) {
-    const err = await response.text();
-    throw new Error(`LLM API error: ${response.status} - ${err}`);
-  }
-  if (options.stream) {
-    return parseSseStream(response.body);
-  }
-  return response.json();
+  return headers;
 }
 async function* parseSseStream(body) {
   const reader = body.getReader();
   const decoder = new TextDecoder();
-  let buffer = "";
+  let buffer = EMPTY_STRING;
   try {
     while (true) {
-      const { value, done } = await reader.read();
-      if (done) break;
+      const { value, done: isDone } = await reader.read();
+      if (isDone) break;
       buffer += decoder.decode(value, { stream: true });
-      const events = buffer.split("\n\n");
-      buffer = events.pop() || "";
+      const { events, remainder } = splitSseEvents(buffer);
+      buffer = remainder;
       for (const event of events) {
-        if (event.startsWith("data: ")) {
-          const data = event.slice(6);
-          if (data === "[DONE]") return;
-          yield JSON.parse(data);
-        }
+        const data = getEventData(event);
+        if (!data) continue;
+        if (data.trim() === "[DONE]") return;
+        yield parseEventData(data);
       }
     }
   } finally {
     reader.releaseLock();
   }
 }
+function splitSseEvents(buffer) {
+  const normalized = buffer.replace(/\r\n/g, "\n").replace(/\r/g, "\n");
+  const parts = normalized.split("\n\n");
+  const remainder = parts.pop() ?? EMPTY_STRING;
+  return { events: parts, remainder };
+}
+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(/^ /, EMPTY_STRING)).join("\n");
+  return data || null;
+}
+function parseEventData(data) {
+  try {
+    return JSON.parse(data);
+  } catch (error) {
+    const message = error instanceof Error ? error.message : String(error);
+    const preview = data.length > 200 ? `${data.slice(0, 200)}...` : data;
+    throw new Error(`LLM stream parse error: ${message}. Data: ${preview}`);
+  }
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   callLlm

package/dist/llm.mjs

@@ -1,84 +1,183 @@
 // src/llm/index.ts
+var EMPTY_STRING = "";
 function normalizeBaseUrl(baseUrl) {
-  return baseUrl.trim().replace(/\/+$/, "");
+  let normalized = baseUrl.trim();
+  while (normalized.endsWith("/")) {
+    normalized = normalized.slice(0, -1);
+  }
+  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 "";
-  return String(process.env[name] ?? "").trim();
+  if (typeof process === "undefined" || !process?.env) return EMPTY_STRING;
+  return String(process.env[name] ?? EMPTY_STRING).trim();
 }
 function getAiBaseUrl(options) {
   if (typeof options.baseUrl === "string" && options.baseUrl.trim()) {
     return normalizeBaseUrl(options.baseUrl);
   }
-  const envBaseUrl = getEnvString("UBQ_AI_BASE_URL") || getEnvString("UBQ_AI_URL");
+  const envBaseUrl = getEnvString("UOS_AI_URL") || getEnvString("UOS_AI_BASE_URL");
   if (envBaseUrl) return normalizeBaseUrl(envBaseUrl);
-  return "https://ai.ubq.fi";
+  return "https://ai-ubq-fi.deno.dev";
 }
 async function callLlm(options, input) {
-  const authToken = input?.authToken ?? "";
-  const ubiquityKernelToken = input?.ubiquityKernelToken ?? "";
-  const payload = input?.payload;
-  const owner = payload?.repository?.owner?.login ?? "";
-  const repo = payload?.repository?.name ?? "";
-  const installationId = payload?.installation?.id;
-  if (!authToken) throw new Error("Missing authToken in inputs");
-  const requiresKernelToken = authToken.trim().startsWith("gh");
-  if (requiresKernelToken && !ubiquityKernelToken) {
-    throw new Error("Missing ubiquityKernelToken in inputs (kernel attestation is required for GitHub auth)");
+  const authToken = String(input.authToken ?? EMPTY_STRING).trim();
+  if (!authToken) {
+    const err = new Error("Missing authToken in input");
+    err.status = 401;
+    throw err;
   }
-  const url = `${getAiBaseUrl(options)}/v1/chat/completions`;
-  const { baseUrl: _baseUrl, model, stream, messages, ...rest } = options;
+  const kernelToken = "ubiquityKernelToken" in input ? input.ubiquityKernelToken : void 0;
+  const payload = getPayload(input);
+  const { owner, repo, installationId } = getRepoMetadata(payload);
+  ensureKernelToken(authToken, kernelToken);
+  const { baseUrl, model, stream: isStream, messages, ...rest } = options;
+  ensureMessages(messages);
+  const url = buildAiUrl(options, baseUrl);
   const body = JSON.stringify({
     ...rest,
     ...model ? { model } : {},
     messages,
-    stream: stream ?? false
+    stream: isStream ?? false
+  });
+  const headers = buildHeaders(authToken, {
+    owner,
+    repo,
+    installationId,
+    ubiquityKernelToken: kernelToken
   });
+  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");
+    }
+    return parseSseStream(response.body);
+  }
+  return response.json();
+}
+function ensureKernelToken(authToken, kernelToken) {
+  const isKernelTokenRequired = authToken.startsWith("gh");
+  if (isKernelTokenRequired && !kernelToken) {
+    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) {
+    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;
+  }
+  return input.eventPayload;
+}
+function getRepoMetadata(payload) {
+  const repoPayload = payload;
+  return {
+    owner: repoPayload?.repository?.owner?.login ?? EMPTY_STRING,
+    repo: repoPayload?.repository?.name ?? EMPTY_STRING,
+    installationId: repoPayload?.installation?.id
+  };
+}
+function buildHeaders(authToken, options) {
   const headers = {
     Authorization: `Bearer ${authToken}`,
     "Content-Type": "application/json"
   };
-  if (owner) headers["X-GitHub-Owner"] = owner;
-  if (repo) headers["X-GitHub-Repo"] = repo;
-  if (typeof installationId === "number" && Number.isFinite(installationId)) {
-    headers["X-GitHub-Installation-Id"] = String(installationId);
+  if (options.owner) headers["X-GitHub-Owner"] = options.owner;
+  if (options.repo) headers["X-GitHub-Repo"] = options.repo;
+  if (typeof options.installationId === "number" && Number.isFinite(options.installationId)) {
+    headers["X-GitHub-Installation-Id"] = String(options.installationId);
   }
-  if (ubiquityKernelToken) {
-    headers["X-Ubiquity-Kernel-Token"] = ubiquityKernelToken;
+  if (options.ubiquityKernelToken) {
+    headers["X-Ubiquity-Kernel-Token"] = options.ubiquityKernelToken;
   }
-  const response = await fetch(url, { method: "POST", headers, body });
-  if (!response.ok) {
-    const err = await response.text();
-    throw new Error(`LLM API error: ${response.status} - ${err}`);
-  }
-  if (options.stream) {
-    return parseSseStream(response.body);
-  }
-  return response.json();
+  return headers;
 }
 async function* parseSseStream(body) {
   const reader = body.getReader();
   const decoder = new TextDecoder();
-  let buffer = "";
+  let buffer = EMPTY_STRING;
   try {
     while (true) {
-      const { value, done } = await reader.read();
-      if (done) break;
+      const { value, done: isDone } = await reader.read();
+      if (isDone) break;
       buffer += decoder.decode(value, { stream: true });
-      const events = buffer.split("\n\n");
-      buffer = events.pop() || "";
+      const { events, remainder } = splitSseEvents(buffer);
+      buffer = remainder;
       for (const event of events) {
-        if (event.startsWith("data: ")) {
-          const data = event.slice(6);
-          if (data === "[DONE]") return;
-          yield JSON.parse(data);
-        }
+        const data = getEventData(event);
+        if (!data) continue;
+        if (data.trim() === "[DONE]") return;
+        yield parseEventData(data);
       }
     }
   } finally {
     reader.releaseLock();
   }
 }
+function splitSseEvents(buffer) {
+  const normalized = buffer.replace(/\r\n/g, "\n").replace(/\r/g, "\n");
+  const parts = normalized.split("\n\n");
+  const remainder = parts.pop() ?? EMPTY_STRING;
+  return { events: parts, remainder };
+}
+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(/^ /, EMPTY_STRING)).join("\n");
+  return data || null;
+}
+function parseEventData(data) {
+  try {
+    return JSON.parse(data);
+  } catch (error) {
+    const message = error instanceof Error ? error.message : String(error);
+    const preview = data.length > 200 ? `${data.slice(0, 200)}...` : data;
+    throw new Error(`LLM stream parse error: ${message}. Data: ${preview}`);
+  }
+}
 export {
   callLlm
 };

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.2",
+  "version": "3.7.0",
   "description": "SDK for plugin support.",
   "author": "Ubiquity DAO",
   "license": "MIT",