fix: normalize provider names in model pricing table
Mapped registry provider IDs (google, xai) to proxy-internal names (gemini, grok) for better dashboard consistency.
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user