Add unified models UI and Rig Bedrock runtime

app/src/ai/bedrock/models.rs

@@ -129,6 +129,7 @@ pub fn get_effective_models(user_models: &[BedrockModelConfig]) -> Vec<BedrockMo
                 model_id: m.model_id.to_string(),
                 display_name: m.display_name.to_string(),
                 vision_supported: m.vision_supported,
+                use_rig: false,
             })
             .collect();
         for default in defaults {
@@ -145,10 +146,38 @@ pub fn get_effective_models(user_models: &[BedrockModelConfig]) -> Vec<BedrockMo
             model_id: m.model_id.to_string(),
             display_name: m.display_name.to_string(),
             vision_supported: m.vision_supported,
+            use_rig: false,
         })
         .collect()
 }
 
+pub fn configured_model_uses_rig(
+    selected_model_id: &str,
+    configured_models: &[BedrockModelConfig],
+    region: &str,
+    cross_region_inference: bool,
+) -> bool {
+    let selected_model_id = strip_context_marker(selected_model_id);
+    configured_models.iter().any(|model| {
+        if !model.use_rig {
+            return false;
+        }
+        let configured_model_id = strip_context_marker(&model.model_id);
+        if configured_model_id == selected_model_id {
+            return true;
+        }
+        galaxy_agent_rig::resolve_bedrock_model_id(&model.model_id, region, cross_region_inference)
+            .is_ok_and(|resolved| strip_context_marker(&resolved) == selected_model_id)
+    })
+}
+
+fn strip_context_marker(model_id: &str) -> &str {
+    model_id
+        .strip_suffix("[1m]")
+        .or_else(|| model_id.strip_suffix("[1M]"))
+        .unwrap_or(model_id)
+}
+
 pub fn apply_cross_region_prefix(model_id: &str, region: &str) -> String {
     if model_id.starts_with("arn:") {
         return model_id.to_string();