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