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) {
|
func (s *Server) handleGetModels(c *gin.Context) {
|
||||||
usedOnly := c.Query("used_only") == "true"
|
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
|
// Merge registry models with DB overrides
|
||||||
var dbModels []db.ModelConfig
|
var dbModels []db.ModelConfig
|
||||||
_ = s.database.Select(&dbModels, "SELECT * FROM model_configs")
|
_ = s.database.Select(&dbModels, "SELECT * FROM model_configs")
|
||||||
@@ -981,6 +987,12 @@ func (s *Server) handleGetModels(c *gin.Context) {
|
|||||||
var result []gin.H
|
var result []gin.H
|
||||||
if s.registry != nil {
|
if s.registry != nil {
|
||||||
for pID, pInfo := range s.registry.Providers {
|
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 {
|
for mID, mMeta := range pInfo.Models {
|
||||||
// Filter if used_only requested
|
// Filter if used_only requested
|
||||||
if usedOnly && !usedModels[mID] {
|
if usedOnly && !usedModels[mID] {
|
||||||
@@ -1026,7 +1038,7 @@ func (s *Server) handleGetModels(c *gin.Context) {
|
|||||||
result = append(result, gin.H{
|
result = append(result, gin.H{
|
||||||
"id": mID,
|
"id": mID,
|
||||||
"name": mMeta.Name,
|
"name": mMeta.Name,
|
||||||
"provider": pID,
|
"provider": uiProvider,
|
||||||
"enabled": enabled,
|
"enabled": enabled,
|
||||||
"prompt_cost": promptCost,
|
"prompt_cost": promptCost,
|
||||||
"completion_cost": completionCost,
|
"completion_cost": completionCost,
|
||||||
|
|||||||
Reference in New Issue
Block a user