UNPKG

@ubiquity-os/plugin-sdk

Version:

SDK for plugin support.

58 lines (56 loc) 1.49 kB
// 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; } export { getOpenRouterModelTokenLimits, getOpenRouterModels, retry };