fix: add ollama and lmstudio to MODELS_BY_PROVIDER (#8511)

Co-authored-by: Roo Code <roomote@roocode.com>
This commit is contained in:
Matt Rubens 2025-10-05 10:16:09 -04:00 committed by GitHub
parent 97f968673f
commit 960958a6da
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -610,7 +610,7 @@ export const getApiProtocol = (provider: ProviderName | undefined, modelId?: str
*/
export const MODELS_BY_PROVIDER: Record<
Exclude<ProviderName, "fake-ai" | "human-relay" | "gemini-cli" | "lmstudio" | "openai" | "ollama">,
Exclude<ProviderName, "fake-ai" | "human-relay" | "gemini-cli" | "openai">,
{ id: ProviderName; label: string; models: string[] }
> = {
anthropic: {
@ -696,7 +696,7 @@ export const MODELS_BY_PROVIDER: Record<
xai: { id: "xai", label: "xAI (Grok)", models: Object.keys(xaiModels) },
zai: { id: "zai", label: "Zai", models: Object.keys(internationalZAiModels) },
// Dynamic providers; models pulled from the respective APIs.
// Dynamic providers; models pulled from remote APIs.
glama: { id: "glama", label: "Glama", models: [] },
huggingface: { id: "huggingface", label: "Hugging Face", models: [] },
litellm: { id: "litellm", label: "LiteLLM", models: [] },
@ -705,4 +705,8 @@ export const MODELS_BY_PROVIDER: Record<
unbound: { id: "unbound", label: "Unbound", models: [] },
deepinfra: { id: "deepinfra", label: "DeepInfra", models: [] },
"vercel-ai-gateway": { id: "vercel-ai-gateway", label: "Vercel AI Gateway", models: [] },
// Local providers; models discovered from localhost endpoints.
lmstudio: { id: "lmstudio", label: "LM Studio", models: [] },
ollama: { id: "ollama", label: "Ollama", models: [] },
}