diff --git a/pkg/config/config.go b/pkg/config/config.go index 7165246e5..db9855eb2 100644 --- a/pkg/config/config.go +++ b/pkg/config/config.go @@ -1207,7 +1207,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 { @@ -1215,6 +1215,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 }