fix: prioritize predefined model context_limit over canonical registry (#7888)

Signed-off-by: Extra Small <littleshuai.bot@gmail.com>
Signed-off-by: Douwe Osinga <douwe@squareup.com>
Co-authored-by: Douwe Osinga <douwe@squareup.com>
This commit is contained in:
Extra Small
2026-03-25 06:26:33 -07:00
committed by GitHub
parent 70003c2e22
commit 09879ed67c
+6 -7
View File
@@ -111,6 +111,12 @@ impl ModelConfig {
}
pub fn with_canonical_limits(mut self, provider_name: &str) -> Self {
if let Some(pm) = find_predefined_model(&self.model_name) {
if self.context_limit.is_none() {
self.context_limit = pm.context_limit;
}
}
if let Some(canonical) =
crate::providers::canonical::maybe_get_canonical_model(provider_name, &self.model_name)
{
@@ -125,13 +131,6 @@ impl ModelConfig {
}
}
// Try filling remaining gaps from predefined models
if self.context_limit.is_none() {
if let Some(pm) = find_predefined_model(&self.model_name) {
self.context_limit = pm.context_limit;
}
}
self
}