tkmind/tkmind_go
1
0
Fork 0
Code Issues Pull Requests Actions 4 Packages Projects Releases Wiki Activity

chore: update canonical model registry (#9331)

Browse Source 
Signed-off-by: Bradley Axen <baxen@squareup.com>
This commit is contained in:
Bradley Axen
2026-05-19 23:39:57 -07:00
committed by GitHub
parent 3631e3cbf1
commit 7a53017f44
4 changed files with 18444 additions and 7793 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/goose/src/model.rs
+1 -1
View File
@@ -502,7 +502,7 @@ mod tests {
("GOOSE_CONTEXT_LIMIT", None::<&str>),
]);
let config =
ModelConfig::new_or_fail("moonshotai/kimi-k2.5").with_canonical_limits("nvidia");
ModelConfig::new_or_fail("moonshotai/kimi-k2.6").with_canonical_limits("nvidia");
assert_eq!(config.context_limit, Some(262_144));
assert_eq!(config.max_tokens, None);
crates/goose/src/providers/canonical/data/canonical_models.json
+17561 -7065
View File
File diff suppressed because it is too large Load Diff
crates/goose/src/providers/canonical/data/provider_metadata.json
+868 -713
View File
File diff suppressed because it is too large Load Diff
crates/goose/src/providers/canonical/name_builder.rs
+14 -14
View File
@@ -448,16 +448,16 @@ mod tests {
// === Grok (X.AI) ===
assert_eq!(
map_to_canonical_model("databricks", "grok-3", r),
Some("x-ai/grok-3".to_string())
map_to_canonical_model("databricks", "grok-4.20", r),
Some("x-ai/grok-4.20".to_string())
);
assert_eq!(
map_to_canonical_model("databricks", "databricks-grok-4-fast", r),
Some("x-ai/grok-4-fast".to_string())
map_to_canonical_model("databricks", "databricks-grok-4.20", r),
Some("x-ai/grok-4.20".to_string())
);
assert_eq!(
map_to_canonical_model("databricks", "kgoose-grok-4-fast", r),
Some("x-ai/grok-4-fast".to_string())
map_to_canonical_model("databricks", "kgoose-grok-4.20", r),
Some("x-ai/grok-4.20".to_string())
);
// === Cohere Command ===
@@ -493,8 +493,8 @@ mod tests {
Some("deepseek/deepseek-chat".to_string())
);
assert_eq!(
map_to_canonical_model("databricks", "x-ai-grok-3", r),
Some("x-ai/grok-3".to_string())
map_to_canonical_model("databricks", "x-ai-grok-4.20", r),
Some("x-ai/grok-4.20".to_string())
);
// === Zhipu AI ===
@@ -518,27 +518,27 @@ mod tests {
);
assert_eq!(
map_to_canonical_model("gcp_vertex_ai", "claude-3-5-sonnet", r),
Some("anthropic/claude-3.5-sonnet".to_string())
Some("google-vertex/claude-3.5-sonnet".to_string())
);
assert_eq!(
map_to_canonical_model("gcp_vertex_ai", "claude-sonnet-4@20250514", r),
Some("anthropic/claude-sonnet-4".to_string())
Some("google-vertex/claude-sonnet-4".to_string())
);
assert_eq!(
map_to_canonical_model("gcp_vertex_ai", "claude-3-5-haiku@20241022", r),
Some("anthropic/claude-3.5-haiku".to_string())
Some("google-vertex/claude-3.5-haiku".to_string())
);
assert_eq!(
map_to_canonical_model("gcp_vertex_ai", "claude-sonnet-4-5@20250929", r),
Some("anthropic/claude-sonnet-4.5".to_string())
Some("google-vertex/claude-sonnet-4.5".to_string())
);
assert_eq!(
map_to_canonical_model("gcp_vertex_ai", "claude-opus-4-5@20251101", r),
Some("anthropic/claude-opus-4.5".to_string())
Some("google-vertex/claude-opus-4.5".to_string())
);
assert_eq!(
map_to_canonical_model("gcp_vertex_ai", "claude-haiku-4-5@20251001", r),
Some("anthropic/claude-haiku-4.5".to_string())
Some("google-vertex/claude-haiku-4.5".to_string())
);
}
}