1

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

@@ -90,6 +90,9 @@ pub struct AppConfig {
     pub dashscope_reference_image_model: String,
     pub dashscope_cover_image_model: String,
     pub dashscope_image_request_timeout_ms: u64,
+    pub apimart_base_url: String,
+    pub apimart_api_key: Option<String>,
+    pub apimart_image_request_timeout_ms: u64,
     pub draft_asset_generation_max_concurrent_requests: usize,
     pub ark_character_video_base_url: String,
     pub ark_character_video_api_key: Option<String>,
@@ -182,6 +185,9 @@ impl Default for AppConfig {
             dashscope_reference_image_model: "qwen-image-2.0".to_string(),
             dashscope_cover_image_model: "wan2.2-t2i-flash".to_string(),
             dashscope_image_request_timeout_ms: 150_000,
+            apimart_base_url: "https://api.apimart.ai/v1".to_string(),
+            apimart_api_key: None,
+            apimart_image_request_timeout_ms: 180_000,
             draft_asset_generation_max_concurrent_requests: 4,
             ark_character_video_base_url: DEFAULT_ARK_BASE_URL.to_string(),
             ark_character_video_api_key: None,
@@ -530,6 +536,18 @@ impl AppConfig {
             config.dashscope_image_request_timeout_ms = dashscope_image_request_timeout_ms;
         }
 
+        if let Some(apimart_base_url) = read_first_non_empty_env(&["APIMART_BASE_URL"]) {
+            config.apimart_base_url = apimart_base_url;
+        }
+
+        config.apimart_api_key = read_first_non_empty_env(&["APIMART_API_KEY"]);
+
+        if let Some(apimart_image_request_timeout_ms) =
+            read_first_positive_u64_env(&["APIMART_IMAGE_REQUEST_TIMEOUT_MS"])
+        {
+            config.apimart_image_request_timeout_ms = apimart_image_request_timeout_ms;
+        }
+
         if let Some(max_concurrent_requests) = read_first_usize_env(&[
             "GENARRATIVE_DRAFT_ASSET_GENERATION_MAX_CONCURRENT_REQUESTS",
             "DRAFT_ASSET_GENERATION_MAX_CONCURRENT_REQUESTS",