fix: normalize provider names in model pricing table
Some checks failed
CI / Lint (push) Has been cancelled
CI / Test (push) Has been cancelled
CI / Build (push) Has been cancelled

Mapped registry provider IDs (google, xai) to proxy-internal names (gemini, grok) for better dashboard consistency.
This commit is contained in:
2026-03-19 13:06:52 -04:00
parent f23796f0cc
commit f02fd6c249

View File

@@ -954,6 +954,12 @@ func (s *Server) handleTestProvider(c *gin.Context) {
func (s *Server) handleGetModels(c *gin.Context) {
usedOnly := c.Query("used_only") == "true"
// Provider name mapping (Registry -> Proxy)
providerMap := map[string]string{
"google": "gemini",
"xai": "grok",
}
// Merge registry models with DB overrides
var dbModels []db.ModelConfig
_ = s.database.Select(&dbModels, "SELECT * FROM model_configs")
@@ -981,6 +987,12 @@ func (s *Server) handleGetModels(c *gin.Context) {
var result []gin.H
if s.registry != nil {
for pID, pInfo := range s.registry.Providers {
// Normalize provider name for UI
uiProvider := pID
if mapped, ok := providerMap[pID]; ok {
uiProvider = mapped
}
for mID, mMeta := range pInfo.Models {
// Filter if used_only requested
if usedOnly && !usedModels[mID] {
@@ -1026,7 +1038,7 @@ func (s *Server) handleGetModels(c *gin.Context) {
result = append(result, gin.H{
"id": mID,
"name": mMeta.Name,
"provider": pID,
"provider": uiProvider,
"enabled": enabled,
"prompt_cost": promptCost,
"completion_cost": completionCost,