From 232f092f27095fc37d552646f1b110dbe8f3cc63 Mon Sep 17 00:00:00 2001 From: hobokenchicken Date: Mon, 2 Mar 2026 14:14:56 -0500 Subject: [PATCH] fix(server): map provider names to registry keys for /v1/models The model registry from models.dev uses 'google' and 'xai' as provider IDs, but internal providers use 'gemini' and 'grok'. Added mapping so all provider models appear in the listing. --- src/server/mod.rs | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/src/server/mod.rs b/src/server/mod.rs index f2d01a0a..69666bf1 100644 --- a/src/server/mod.rs +++ b/src/server/mod.rs @@ -47,8 +47,15 @@ async fn list_models( for provider in &providers { let provider_name = provider.name(); + // Map internal provider names to registry provider IDs + let registry_key = match provider_name { + "gemini" => "google", + "grok" => "xai", + _ => provider_name, + }; + // Find this provider's models in the registry - if let Some(provider_info) = registry.providers.get(provider_name) { + if let Some(provider_info) = registry.providers.get(registry_key) { for (model_id, meta) in &provider_info.models { // Skip disabled models via the config cache if let Some(cfg) = state.model_config_cache.get(model_id).await {