59 lines
2.0 KiB
TypeScript
59 lines
2.0 KiB
TypeScript
|
import type { ProviderStyle } from "./types";
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Conventional `/models`-listing path for each wire-protocol family. These
|
||
|
|
* are the same paths the official Anthropic/OpenAI/Google clients use,
|
||
|
|
* so they line up with what compliant Anthropic-compat or OpenAI-compat
|
||
|
|
* third parties (DeepSeek, Moonshot, Qwen, Ollama, OpenRouter, ...)
|
||
|
|
* expose. Google is `undefined` because Generative AI's discovery isn't
|
||
|
|
* a standard REST listing.
|
||
|
|
*/
|
||
|
|
export const STYLE_DEFAULT_MODELS_ENDPOINT: Record<ProviderStyle, string | undefined> = {
|
||
|
|
openai: "/models",
|
||
|
|
anthropic: "/v1/models",
|
||
|
|
google: undefined,
|
||
|
|
};
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Pick the `/models` discovery path for a provider config. The resolved
|
||
|
|
* `style` wins — keeping it aligned with {@link buildModelDiscoveryHeaders}
|
||
|
|
* — falling back to the providerId-derived `presetEndpoint` only when the
|
||
|
|
* style has no convention of its own.
|
||
|
|
*/
|
||
|
|
export function resolveModelsDiscoveryEndpoint(
|
||
|
|
style: ProviderStyle,
|
||
|
|
presetEndpoint?: string,
|
||
|
|
): string | undefined {
|
||
|
|
return STYLE_DEFAULT_MODELS_ENDPOINT[style] ?? presetEndpoint;
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Pick auth headers for a provider's `/models` discovery endpoint.
|
||
|
|
*
|
||
|
|
* Each wire-protocol family uses its own auth dialect:
|
||
|
|
* - `anthropic`: `x-api-key` + `anthropic-version`
|
||
|
|
* - `google`: `x-goog-api-key` (Google Generative AI rejects Bearer)
|
||
|
|
* - `openai`: `Authorization: Bearer …` (also the OpenAI-compat default)
|
||
|
|
*
|
||
|
|
* Returning an empty object when the key is missing lets the caller still
|
||
|
|
* issue an unauthenticated probe (e.g. against local Ollama).
|
||
|
|
*/
|
||
|
|
export function buildModelDiscoveryHeaders(
|
||
|
|
style: ProviderStyle,
|
||
|
|
apiKey: string | undefined,
|
||
|
|
): Record<string, string> {
|
||
|
|
if (!apiKey) return {};
|
||
|
|
switch (style) {
|
||
|
|
case "anthropic":
|
||
|
|
return {
|
||
|
|
"x-api-key": apiKey,
|
||
|
|
"anthropic-version": "2023-06-01",
|
||
|
|
};
|
||
|
|
case "google":
|
||
|
|
return { "x-goog-api-key": apiKey };
|
||
|
|
case "openai":
|
||
|
|
default:
|
||
|
|
return { Authorization: `Bearer ${apiKey}` };
|
||
|
|
}
|
||
|
|
}
|