1

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

@@ -80,6 +80,7 @@ pub struct AppConfig {
     pub llm_request_timeout_ms: u64,
     pub llm_max_retries: u32,
     pub llm_retry_backoff_ms: u64,
+    pub rpg_llm_web_search_enabled: bool,
     pub dashscope_base_url: String,
     pub dashscope_api_key: Option<String>,
     pub dashscope_scene_image_model: String,
@@ -168,6 +169,7 @@ impl Default for AppConfig {
             llm_request_timeout_ms: DEFAULT_REQUEST_TIMEOUT_MS,
             llm_max_retries: DEFAULT_MAX_RETRIES,
             llm_retry_backoff_ms: DEFAULT_RETRY_BACKOFF_MS,
+            rpg_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(),
@@ -466,6 +468,13 @@ impl AppConfig {
             config.llm_retry_backoff_ms = llm_retry_backoff_ms;
         }
 
+        if let Some(rpg_llm_web_search_enabled) = read_first_bool_env(&[
+            "GENARRATIVE_RPG_LLM_WEB_SEARCH_ENABLED",
+            "RPG_LLM_WEB_SEARCH_ENABLED",
+        ]) {
+            config.rpg_llm_web_search_enabled = rpg_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;
         }
@@ -814,4 +823,24 @@ mod tests {
             std::env::remove_var("GENARRATIVE_SPACETIME_POOL_SIZE");
         }
     }
+
+    #[test]
+    fn from_env_reads_rpg_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_RPG_LLM_WEB_SEARCH_ENABLED");
+            std::env::set_var("GENARRATIVE_RPG_LLM_WEB_SEARCH_ENABLED", "false");
+        }
+
+        let config = AppConfig::from_env();
+        assert!(!config.rpg_llm_web_search_enabled);
+
+        unsafe {
+            std::env::remove_var("GENARRATIVE_RPG_LLM_WEB_SEARCH_ENABLED");
+        }
+    }
 }