This commit is contained in:
Darren.Zeng 2026-04-17 22:56:05 +02:00 committed by GitHub
commit a399cce466
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -1360,7 +1360,7 @@ func (c *Config) GetModelConfig(modelName string) (*ModelConfig, error) {
return matches[idx], nil
}
// findMatches finds all ModelConfig entries with the given model_name.
// findMatches finds all ModelConfig entries with the given model_name or model id.
func (c *Config) findMatches(modelName string) []*ModelConfig {
var matches []*ModelConfig
for i := range c.ModelList {
@ -1368,6 +1368,14 @@ func (c *Config) findMatches(modelName string) []*ModelConfig {
matches = append(matches, c.ModelList[i])
}
}
// If no matches found, also try matching by model id (e.g., "openai/gpt-4o")
if len(matches) == 0 {
for i := range c.ModelList {
if c.ModelList[i].Model == modelName {
matches = append(matches, c.ModelList[i])
}
}
}
return matches
}