53 lines
3.2 KiB
TypeScript
53 lines
3.2 KiB
TypeScript
/** Host-only proxy catalog, provider transports, authentication, and quoted token prices. */
|
|
import { createModels } from '@earendil-works/pi-ai/models';
|
|
import { openaiProvider } from '@earendil-works/pi-ai/providers/openai';
|
|
import { anthropicProvider } from '@earendil-works/pi-ai/providers/anthropic';
|
|
|
|
export async function proxyModels(proxy: {base_url: string, api_key: string}, readProxy = async () => proxy) {
|
|
const models = createModels();
|
|
const base = proxy.base_url.replace(/\/$/, '');
|
|
let catalog: any[] = [];
|
|
try {
|
|
if (proxy.base_url && proxy.api_key) {
|
|
const response = await fetch(`${base}/v1/models`, {
|
|
headers: {Authorization: `Bearer ${proxy.api_key}`}, signal: AbortSignal.timeout(5000),
|
|
});
|
|
if (response.ok) catalog = (await response.json()).data ?? [];
|
|
}
|
|
} catch { /* An unavailable catalog must not prevent durable recovery. */ }
|
|
for (const [provider, api, path, fallback] of [
|
|
[openaiProvider(), 'openai-responses', '/v1/responses', 'gpt-6.1-sol'],
|
|
[anthropicProvider(), 'anthropic-messages', '/apis/anthropic/v1/messages', 'claude-opus-5-5'],
|
|
] as const) {
|
|
const rows = catalog.filter(row => row.creator === provider.id && row.endpoints?.some((endpoint: any) => endpoint.api === api));
|
|
const entries = (rows.length ? rows : [{id: fallback}]).map(row => {
|
|
const endpoint = row.endpoints?.find((endpoint: any) => endpoint.api === api)?.path ?? path;
|
|
// pi-ai appends /responses or /v1/messages, respectively.
|
|
const suffix = api === 'anthropic-messages' ? '/v1/messages' : '/responses';
|
|
const baseUrl = `${base}${endpoint.slice(0, -suffix.length)}`;
|
|
const price = row.providers?.[0]?.pricing;
|
|
const scale = price?.currency === 'USD' ? 1000000 / price.unit_tokens : 0;
|
|
return {
|
|
id: row.id, name: row.name ?? row.id, provider: provider.id, api, baseUrl,
|
|
reasoning: row.capabilities?.reasoning ?? true, input: ['text'],
|
|
thinkingLevelMap: {off: null, minimal: null,
|
|
...Object.fromEntries((row.capabilities?.reasoning_options?.find((option: any) => option.type === 'effort')?.values ?? []).map((level: string) => [level, level]))},
|
|
compat: api === 'anthropic-messages' ? {forceAdaptiveThinking: row.capabilities?.reasoning_options?.some((option: any) => option.type === 'effort') ?? true} : undefined,
|
|
cost: {input: (price?.input ?? 0) * scale, output: (price?.output ?? 0) * scale,
|
|
cacheRead: (price?.cached_input ?? 0) * scale, cacheWrite: (price?.cache_write ?? 0) * scale},
|
|
contextWindow: row.limits?.context_window ?? 200000, maxTokens: 4096,
|
|
};
|
|
});
|
|
models.setProvider({...provider, baseUrl: entries[0].baseUrl, getModels: () => entries as any,
|
|
auth: {apiKey: {name: 'Sausage proxy', resolve: async () => {
|
|
const current = await readProxy();
|
|
if (!current.base_url || !current.api_key) throw new Error('Set proxy.base_url and proxy.api_key in /etc/sausage/config.yaml before a model request');
|
|
return {auth:{apiKey:current.api_key,
|
|
baseUrl:current.base_url.replace(/\/$/,'') + entries[0].baseUrl.slice(base.length),
|
|
headers:{'Shopify-Usage-Tag':JSON.stringify(['pi','sausage'])}},source:'/etc/sausage/config.yaml'};
|
|
}}},
|
|
});
|
|
}
|
|
return models;
|
|
}
|