UNPKG

@ubiquity-os/plugin-sdk

Version:

SDK for plugin support.

87 lines (83 loc) 2.64 kB
"use strict"; var __defProp = Object.defineProperty; var __getOwnPropDesc = Object.getOwnPropertyDescriptor; var __getOwnPropNames = Object.getOwnPropertyNames; var __hasOwnProp = Object.prototype.hasOwnProperty; var __export = (target, all) => { for (var name in all) __defProp(target, name, { get: all[name], enumerable: true }); }; var __copyProps = (to, from, except, desc) => { if (from && typeof from === "object" || typeof from === "function") { for (let key of __getOwnPropNames(from)) if (!__hasOwnProp.call(to, key) && key !== except) __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable }); } return to; }; var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod); // src/helpers/index.ts var helpers_exports = {}; __export(helpers_exports, { getOpenRouterModelTokenLimits: () => getOpenRouterModelTokenLimits, getOpenRouterModels: () => getOpenRouterModels, retry: () => retry }); module.exports = __toCommonJS(helpers_exports); // src/helpers/openrouter.ts async function getOpenRouterModels() { const response = await fetch("https://openrouter.ai/api/v1/models"); if (!response.ok) { throw new Error("Failed to fetch models"); } const result = await response.json(); return result.data; } async function getOpenRouterModelTokenLimits(modelId) { const models = await getOpenRouterModels(); const model = models.find((m) => m.id === modelId); if (!model) { return null; } return { contextLength: model.context_length, maxCompletionTokens: model.top_provider.max_completion_tokens || model.context_length }; } // src/helpers/retry.ts function sleep(ms) { return new Promise((resolve) => setTimeout(resolve, ms)); } async function retry(fn, options) { let delay = 1e3; let lastError = null; for (let i = 0; i < options.maxRetries + 1; i++) { try { return await fn(); } catch (err) { lastError = err; if (options.onError) { await options.onError(err); } let shouldRetry; if (options.isErrorRetryable) { shouldRetry = await options.isErrorRetryable(err); } if (shouldRetry === false) { throw lastError; } else if (typeof shouldRetry === "number" && Number.isFinite(shouldRetry)) { await sleep(shouldRetry); } else { await sleep(delay); delay *= 2; } } } throw lastError; } // Annotate the CommonJS export names for ESM import in node: 0 && (module.exports = { getOpenRouterModelTokenLimits, getOpenRouterModels, retry });