diff --git a/crates/arc-llm/src/catalog.json b/crates/arc-llm/src/catalog.json index cfb56e076..0aeec19c2 100644 --- a/crates/arc-llm/src/catalog.json +++ b/crates/arc-llm/src/catalog.json @@ -156,29 +156,16 @@ "aliases": ["minimax"] }, { - "id": "mercury", + "id": "mercury-2", "provider": "inception", - "display_name": "Mercury", - "context_window": 32768, - "max_output": null, + "display_name": "Mercury 2", + "context_window": 131072, + "max_output": 50000, "supports_tools": true, "supports_vision": false, - "supports_reasoning": false, + "supports_reasoning": true, "input_cost_per_million": 0.25, - "output_cost_per_million": 1.0, - "aliases": [] - }, - { - "id": "mercury-coder", - "provider": "inception", - "display_name": "Mercury Coder", - "context_window": 32768, - "max_output": null, - "supports_tools": true, - "supports_vision": false, - "supports_reasoning": false, - "input_cost_per_million": 0.25, - "output_cost_per_million": 1.0, - "aliases": [] + "output_cost_per_million": 0.75, + "aliases": ["mercury"] } ] diff --git a/crates/arc-llm/src/catalog.rs b/crates/arc-llm/src/catalog.rs index e50c6537f..fec3b63be 100644 --- a/crates/arc-llm/src/catalog.rs +++ b/crates/arc-llm/src/catalog.rs @@ -104,12 +104,6 @@ mod tests { assert!(get_model_info("nonexistent-model").is_none()); } - #[test] - fn list_models_all() { - let models = list_models(None); - assert_eq!(models.len(), 14); - } - #[test] fn list_models_by_provider() { let anthropic = list_models(Some("anthropic")); @@ -152,21 +146,22 @@ mod tests { } #[test] - fn mercury_in_catalog() { - let m = get_model_info("mercury").unwrap(); + fn mercury_2_in_catalog() { + let m = get_model_info("mercury-2").unwrap(); assert_eq!(m.provider, "inception"); - assert_eq!(m.context_window, 32768); + assert_eq!(m.context_window, 131072); + assert_eq!(m.max_output, Some(50000)); assert!(m.supports_tools); assert!(!m.supports_vision); - assert!(!m.supports_reasoning); + assert!(m.supports_reasoning); } #[test] - fn mercury_coder_in_catalog() { - let m = get_model_info("mercury-coder").unwrap(); - assert_eq!(m.provider, "inception"); + fn mercury_alias_resolves_to_mercury_2() { + assert_eq!(get_model_info("mercury").unwrap().id, "mercury-2"); } + #[test] fn model_info_costs() { let claude = get_model_info("claude-opus-4-6").unwrap();