1

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

@@ -1,4 +1,4 @@
-use std::{env, fs, net::SocketAddr, path::PathBuf};
+use std::{env, fs, net::SocketAddr, path::PathBuf, time::Duration};
 
 use platform_llm::{
     DEFAULT_ARK_BASE_URL, DEFAULT_MAX_RETRIES, DEFAULT_REQUEST_TIMEOUT_MS,
@@ -74,6 +74,7 @@ pub struct AppConfig {
     pub spacetime_database: String,
     pub spacetime_token: Option<String>,
     pub spacetime_pool_size: u32,
+    pub spacetime_procedure_timeout: Duration,
     pub llm_provider: LlmProvider,
     pub llm_base_url: String,
     pub llm_api_key: Option<String>,
@@ -165,6 +166,7 @@ impl Default for AppConfig {
             spacetime_database: "genarrative-dev".to_string(),
             spacetime_token: None,
             spacetime_pool_size: 4,
+            spacetime_procedure_timeout: Duration::from_secs(30),
             llm_provider: LlmProvider::Ark,
             llm_base_url: DEFAULT_ARK_BASE_URL.to_string(),
             llm_api_key: None,
@@ -436,6 +438,12 @@ impl AppConfig {
         {
             config.spacetime_pool_size = spacetime_pool_size;
         }
+        if let Some(spacetime_procedure_timeout_seconds) =
+            read_first_duration_seconds_env(&["GENARRATIVE_SPACETIME_PROCEDURE_TIMEOUT_SECONDS"])
+        {
+            config.spacetime_procedure_timeout =
+                Duration::from_secs(spacetime_procedure_timeout_seconds);
+        }
 
         if let Some(llm_provider) =
             read_first_llm_provider_env(&["GENARRATIVE_LLM_PROVIDER", "LLM_PROVIDER"])
@@ -840,6 +848,26 @@ mod tests {
         }
     }
 
+    #[test]
+    fn from_env_reads_spacetime_procedure_timeout() {
+        let _guard = ENV_LOCK
+            .get_or_init(|| Mutex::new(()))
+            .lock()
+            .expect("env lock should not poison");
+
+        unsafe {
+            std::env::remove_var("GENARRATIVE_SPACETIME_PROCEDURE_TIMEOUT_SECONDS");
+            std::env::set_var("GENARRATIVE_SPACETIME_PROCEDURE_TIMEOUT_SECONDS", "45");
+        }
+
+        let config = AppConfig::from_env();
+        assert_eq!(config.spacetime_procedure_timeout.as_secs(), 45);
+
+        unsafe {
+            std::env::remove_var("GENARRATIVE_SPACETIME_PROCEDURE_TIMEOUT_SECONDS");
+        }
+    }
+
     #[test]
     fn from_env_reads_rpg_llm_web_search_switch() {
         let _guard = ENV_LOCK