mirror of
https://github.com/Termix-SSH/Termix.git
synced 2026-10-29 14:45:24 +00:00
65 lines
1.9 KiB
TypeScript
65 lines
1.9 KiB
TypeScript
import { anthropicAdapter } from "./anthropic.js";
|
|
import { geminiAdapter } from "./gemini.js";
|
|
import { ollamaAdapter } from "./ollama.js";
|
|
import { openAiAdapter } from "./openai.js";
|
|
import {
|
|
AiProviderError,
|
|
type AiProviderType,
|
|
type ProviderAdapter,
|
|
} from "./types.js";
|
|
|
|
/**
|
|
* openai_compatible reuses the OpenAI adapter: OpenRouter, Groq, Mistral,
|
|
* DeepSeek, together.ai, LM Studio and vLLM all speak the same wire format,
|
|
* they differ only in base URL.
|
|
*/
|
|
const ADAPTERS: Record<AiProviderType, ProviderAdapter> = {
|
|
ollama: ollamaAdapter,
|
|
anthropic: anthropicAdapter,
|
|
openai: openAiAdapter,
|
|
gemini: geminiAdapter,
|
|
openai_compatible: openAiAdapter,
|
|
};
|
|
|
|
export function getAdapter(providerType: string): ProviderAdapter {
|
|
const adapter = ADAPTERS[providerType as AiProviderType];
|
|
if (!adapter) {
|
|
throw new AiProviderError(`Unknown provider type: ${providerType}`);
|
|
}
|
|
return adapter;
|
|
}
|
|
|
|
/** Provider types that cannot work without a base URL. */
|
|
export const REQUIRES_BASE_URL: AiProviderType[] = [
|
|
"ollama",
|
|
"openai_compatible",
|
|
];
|
|
|
|
/** Provider types that cannot work without an API key. */
|
|
export const REQUIRES_API_KEY: AiProviderType[] = [
|
|
"anthropic",
|
|
"openai",
|
|
"gemini",
|
|
];
|
|
|
|
/**
|
|
* Shown when a provider's model list cannot be fetched: no key entered yet,
|
|
* the endpoint is unreachable, or the vendor has no list endpoint. Users can
|
|
* always type a model id the list does not contain, so this is a starting
|
|
* point rather than a restriction.
|
|
*/
|
|
export const FALLBACK_MODELS: Record<AiProviderType, string[]> = {
|
|
ollama: [
|
|
"llama3.2",
|
|
"llama3.1",
|
|
"qwen2.5-coder",
|
|
"mistral",
|
|
"phi4",
|
|
"gemma2",
|
|
],
|
|
anthropic: ["claude-opus-5", "claude-sonnet-5", "claude-haiku-4-5"],
|
|
openai: ["gpt-5", "gpt-5-mini", "o4-mini", "gpt-4.1"],
|
|
gemini: ["gemini-2.5-pro", "gemini-2.5-flash", "gemini-2.0-flash"],
|
|
openai_compatible: [],
|
|
};
|