fix(config): allow GetModelConfig to match by model id when model name fails (fixes #2286)
This commit is contained in:
parent
84edc462d6
commit
b374d8a7df
1 changed files with 9 additions and 1 deletions
|
|
@ -1207,7 +1207,7 @@ func (c *Config) GetModelConfig(modelName string) (*ModelConfig, error) {
|
||||||
return matches[idx], nil
|
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 {
|
func (c *Config) findMatches(modelName string) []*ModelConfig {
|
||||||
var matches []*ModelConfig
|
var matches []*ModelConfig
|
||||||
for i := range c.ModelList {
|
for i := range c.ModelList {
|
||||||
|
|
@ -1215,6 +1215,14 @@ func (c *Config) findMatches(modelName string) []*ModelConfig {
|
||||||
matches = append(matches, c.ModelList[i])
|
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
|
return matches
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue