feat(config): add environment variable overrides for model and provider (#2003)

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>

Author: tusharmath

crates/forge_app/src/orch_spec/orch_setup.rs

@@ -84,6 +84,8 @@ impl Default for TestContext {
                 custom_history_path: None,
                 max_conversations: 100,
                 max_image_size: 262144,
+                override_model: None,
+                override_provider: None,
             },
             title: Some("test-conversation".into()),
             agent: Agent::new(

crates/forge_domain/src/env.rs

@@ -6,7 +6,7 @@ use derive_setters::Setters;
 use serde::{Deserialize, Serialize};
 use url::Url;
 
-use crate::{HttpConfig, RetryConfig};
+use crate::{HttpConfig, ModelId, ProviderId, RetryConfig};
 
 const VERSION: &str = match option_env!("APP_VERSION") {
     Some(val) => val,
@@ -70,6 +70,15 @@ pub struct Environment {
     /// Maximum number of conversations to show in list.
     /// Controlled by FORGE_MAX_CONVERSATIONS environment variable.
     pub max_conversations: usize,
+    /// Override model for all providers from FORGE_OVERRIDE_MODEL environment
+    /// variable. If set, this model will be used instead of configured
+    /// models.
+    #[dummy(default)]
+    pub override_model: Option<ModelId>,
+    /// Override provider from FORGE_OVERRIDE_PROVIDER environment variable.
+    /// If set, this provider will be used as default.
+    #[dummy(default)]
+    pub override_provider: Option<ProviderId>,
 }
 
 impl Environment {
@@ -273,6 +282,8 @@ fn test_command_path() {
         custom_history_path: None,
         max_conversations: 100,
         max_image_size: 262144,
+        override_model: None,
+        override_provider: None,
     };
 
     let actual = fixture.command_path();
@@ -307,6 +318,8 @@ fn test_command_cwd_path() {
         custom_history_path: None,
         max_conversations: 100,
         max_image_size: 262144,
+        override_model: None,
+        override_provider: None,
     };
 
     let actual = fixture.command_cwd_path();
@@ -341,6 +354,8 @@ fn test_command_cwd_path_independent_from_command_path() {
         custom_history_path: None,
         max_conversations: 100,
         max_image_size: 262144,
+        override_model: None,
+        override_provider: None,
     };
 
     let command_path = fixture.command_path();

crates/forge_infra/src/env.rs

@@ -3,7 +3,7 @@ use std::path::{Path, PathBuf};
 use std::str::FromStr;
 
 use forge_app::EnvironmentInfra;
-use forge_domain::{Environment, RetryConfig, TlsBackend, TlsVersion};
+use forge_domain::{Environment, ModelId, ProviderId, RetryConfig, TlsBackend, TlsVersion};
 use reqwest::Url;
 
 #[derive(Clone)]
@@ -55,6 +55,10 @@ impl ForgeEnvironmentInfra {
         // Parse custom history file path from environment variable
         let custom_history_path = parse_env::<String>("FORGE_HISTORY_FILE").map(PathBuf::from);
 
+        let override_model = parse_env::<String>("FORGE_OVERRIDE_MODEL").map(ModelId::new);
+        let override_provider = parse_env::<String>("FORGE_OVERRIDE_PROVIDER")
+            .and_then(|s| ProviderId::from_str(&s).ok());
+
         Environment {
             os: std::env::consts::OS.to_string(),
             pid: std::process::id(),
@@ -82,6 +86,8 @@ impl ForgeEnvironmentInfra {
             forge_api_url,
             custom_history_path,
             max_conversations: parse_env::<usize>("FORGE_MAX_CONVERSATIONS").unwrap_or(100),
+            override_model,
+            override_provider,
         }
     }