@ubiquity-os/plugin-sdk
Version:
SDK for plugin support.
58 lines (56 loc) • 1.49 kB
JavaScript
// 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
};