1

server-rs/crates/api-server/src/config.rs

@@ -81,6 +81,7 @@ pub struct AppConfig {
     pub llm_max_retries: u32,
     pub llm_retry_backoff_ms: u64,
     pub rpg_llm_web_search_enabled: bool,
+    pub creation_agent_llm_web_search_enabled: bool,
     pub dashscope_base_url: String,
     pub dashscope_api_key: Option<String>,
     pub dashscope_scene_image_model: String,
@@ -170,6 +171,7 @@ impl Default for AppConfig {
             llm_max_retries: DEFAULT_MAX_RETRIES,
             llm_retry_backoff_ms: DEFAULT_RETRY_BACKOFF_MS,
             rpg_llm_web_search_enabled: true,
+            creation_agent_llm_web_search_enabled: true,
             dashscope_base_url: "https://dashscope.aliyuncs.com/api/v1".to_string(),
             dashscope_api_key: None,
             dashscope_scene_image_model: "wan2.2-t2i-flash".to_string(),
@@ -475,6 +477,13 @@ impl AppConfig {
             config.rpg_llm_web_search_enabled = rpg_llm_web_search_enabled;
         }
 
+        if let Some(creation_agent_llm_web_search_enabled) = read_first_bool_env(&[
+            "GENARRATIVE_CREATION_AGENT_LLM_WEB_SEARCH_ENABLED",
+            "CREATION_AGENT_LLM_WEB_SEARCH_ENABLED",
+        ]) {
+            config.creation_agent_llm_web_search_enabled = creation_agent_llm_web_search_enabled;
+        }
+
         if let Some(dashscope_base_url) = read_first_non_empty_env(&["DASHSCOPE_BASE_URL"]) {
             config.dashscope_base_url = dashscope_base_url;
         }
@@ -843,4 +852,24 @@ mod tests {
             std::env::remove_var("GENARRATIVE_RPG_LLM_WEB_SEARCH_ENABLED");
         }
     }
+
+    #[test]
+    fn from_env_reads_creation_agent_llm_web_search_switch() {
+        let _guard = ENV_LOCK
+            .get_or_init(|| Mutex::new(()))
+            .lock()
+            .expect("env lock should not poison");
+
+        unsafe {
+            std::env::remove_var("GENARRATIVE_CREATION_AGENT_LLM_WEB_SEARCH_ENABLED");
+            std::env::set_var("GENARRATIVE_CREATION_AGENT_LLM_WEB_SEARCH_ENABLED", "false");
+        }
+
+        let config = AppConfig::from_env();
+        assert!(!config.creation_agent_llm_web_search_enabled);
+
+        unsafe {
+            std::env::remove_var("GENARRATIVE_CREATION_AGENT_LLM_WEB_SEARCH_ENABLED");
+        }
+    }
 }