Skip to content

Commit 960958a

Browse files
fix: add ollama and lmstudio to MODELS_BY_PROVIDER (#8511)
Co-authored-by: Roo Code <roomote@roocode.com>
1 parent 97f9686 commit 960958a

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

packages/types/src/provider-settings.ts

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -610,7 +610,7 @@ export const getApiProtocol = (provider: ProviderName | undefined, modelId?: str
610610
*/
611611

612612
export const MODELS_BY_PROVIDER: Record<
613-
Exclude<ProviderName, "fake-ai" | "human-relay" | "gemini-cli" | "lmstudio" | "openai" | "ollama">,
613+
Exclude<ProviderName, "fake-ai" | "human-relay" | "gemini-cli" | "openai">,
614614
{ id: ProviderName; label: string; models: string[] }
615615
> = {
616616
anthropic: {
@@ -696,7 +696,7 @@ export const MODELS_BY_PROVIDER: Record<
696696
xai: { id: "xai", label: "xAI (Grok)", models: Object.keys(xaiModels) },
697697
zai: { id: "zai", label: "Zai", models: Object.keys(internationalZAiModels) },
698698

699-
// Dynamic providers; models pulled from the respective APIs.
699+
// Dynamic providers; models pulled from remote APIs.
700700
glama: { id: "glama", label: "Glama", models: [] },
701701
huggingface: { id: "huggingface", label: "Hugging Face", models: [] },
702702
litellm: { id: "litellm", label: "LiteLLM", models: [] },
@@ -705,4 +705,8 @@ export const MODELS_BY_PROVIDER: Record<
705705
unbound: { id: "unbound", label: "Unbound", models: [] },
706706
deepinfra: { id: "deepinfra", label: "DeepInfra", models: [] },
707707
"vercel-ai-gateway": { id: "vercel-ai-gateway", label: "Vercel AI Gateway", models: [] },
708+
709+
// Local providers; models discovered from localhost endpoints.
710+
lmstudio: { id: "lmstudio", label: "LM Studio", models: [] },
711+
ollama: { id: "ollama", label: "Ollama", models: [] },
708712
}

0 commit comments

Comments
 (0)