chore: filter /v1/models to allowed providers
Restricted model listing to OpenAI, Google (Gemini), DeepSeek, and xAI (Grok) to match available access.
This commit is contained in:
@@ -150,13 +150,23 @@ func (s *Server) handleListModels(c *gin.Context) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
var data []OpenAIModel
|
var data []OpenAIModel
|
||||||
|
allowedProviders := map[string]bool{
|
||||||
|
"openai": true,
|
||||||
|
"google": true, // Models from models.dev use 'google' ID for Gemini
|
||||||
|
"deepseek": true,
|
||||||
|
"xai": true, // Models from models.dev use 'xai' ID for Grok
|
||||||
|
}
|
||||||
|
|
||||||
if s.registry != nil {
|
if s.registry != nil {
|
||||||
for pID, pInfo := range s.registry.Providers {
|
for pID, pInfo := range s.registry.Providers {
|
||||||
|
if !allowedProviders[pID] {
|
||||||
|
continue
|
||||||
|
}
|
||||||
for mID := range pInfo.Models {
|
for mID := range pInfo.Models {
|
||||||
data = append(data, OpenAIModel{
|
data = append(data, OpenAIModel{
|
||||||
ID: mID,
|
ID: mID,
|
||||||
Object: "model",
|
Object: "model",
|
||||||
Created: 1700000000, // Static placeholder
|
Created: 1700000000,
|
||||||
OwnedBy: pID,
|
OwnedBy: pID,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user