后端重写提交

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

@@ -1,5 +1,12 @@
 use std::{env, net::SocketAddr};
 
+use platform_llm::{
+    DEFAULT_ARK_BASE_URL, DEFAULT_MAX_RETRIES, DEFAULT_REQUEST_TIMEOUT_MS,
+    DEFAULT_RETRY_BACKOFF_MS, LlmProvider,
+};
+
+const DEFAULT_LLM_MODEL: &str = "doubao-1-5-pro-32k-character-250715";
+
 // 集中管理 api-server 的启动配置，避免入口层直接散落环境变量解析逻辑。
 #[derive(Clone, Debug)]
 pub struct AppConfig {
@@ -40,6 +47,13 @@ pub struct AppConfig {
     pub spacetime_server_url: String,
     pub spacetime_database: String,
     pub spacetime_token: Option<String>,
+    pub llm_provider: LlmProvider,
+    pub llm_base_url: String,
+    pub llm_api_key: Option<String>,
+    pub llm_model: String,
+    pub llm_request_timeout_ms: u64,
+    pub llm_max_retries: u32,
+    pub llm_retry_backoff_ms: u64,
 }
 
 impl Default for AppConfig {
@@ -83,6 +97,13 @@ impl Default for AppConfig {
             spacetime_server_url: "http://127.0.0.1:3000".to_string(),
             spacetime_database: "genarrative-dev".to_string(),
             spacetime_token: None,
+            llm_provider: LlmProvider::Ark,
+            llm_base_url: DEFAULT_ARK_BASE_URL.to_string(),
+            llm_api_key: None,
+            llm_model: DEFAULT_LLM_MODEL.to_string(),
+            llm_request_timeout_ms: DEFAULT_REQUEST_TIMEOUT_MS,
+            llm_max_retries: DEFAULT_MAX_RETRIES,
+            llm_retry_backoff_ms: DEFAULT_RETRY_BACKOFF_MS,
         }
     }
 }
@@ -244,6 +265,46 @@ impl AppConfig {
 
         config.spacetime_token = read_first_non_empty_env(&["GENARRATIVE_SPACETIME_TOKEN"]);
 
+        if let Some(llm_provider) =
+            read_first_llm_provider_env(&["GENARRATIVE_LLM_PROVIDER", "LLM_PROVIDER"])
+        {
+            config.llm_provider = llm_provider;
+        }
+
+        if let Some(llm_base_url) =
+            read_first_non_empty_env(&["GENARRATIVE_LLM_BASE_URL", "LLM_BASE_URL"])
+        {
+            config.llm_base_url = llm_base_url;
+        }
+
+        config.llm_api_key =
+            read_first_non_empty_env(&["GENARRATIVE_LLM_API_KEY", "LLM_API_KEY", "ARK_API_KEY"]);
+
+        if let Some(llm_model) =
+            read_first_non_empty_env(&["GENARRATIVE_LLM_MODEL", "LLM_MODEL", "VITE_LLM_MODEL"])
+        {
+            config.llm_model = llm_model;
+        }
+
+        if let Some(llm_request_timeout_ms) = read_first_positive_u64_env(&[
+            "GENARRATIVE_LLM_REQUEST_TIMEOUT_MS",
+            "LLM_REQUEST_TIMEOUT_MS",
+        ]) {
+            config.llm_request_timeout_ms = llm_request_timeout_ms;
+        }
+
+        if let Some(llm_max_retries) =
+            read_first_u32_env(&["GENARRATIVE_LLM_MAX_RETRIES", "LLM_MAX_RETRIES"])
+        {
+            config.llm_max_retries = llm_max_retries;
+        }
+
+        if let Some(llm_retry_backoff_ms) =
+            read_first_u64_env(&["GENARRATIVE_LLM_RETRY_BACKOFF_MS", "LLM_RETRY_BACKOFF_MS"])
+        {
+            config.llm_retry_backoff_ms = llm_retry_backoff_ms;
+        }
+
         config
     }
 
@@ -281,6 +342,14 @@ fn read_first_bool_env(keys: &[&str]) -> Option<bool> {
         .find_map(|key| env::var(key).ok().and_then(|value| parse_bool(&value)))
 }
 
+fn read_first_llm_provider_env(keys: &[&str]) -> Option<LlmProvider> {
+    keys.iter().find_map(|key| {
+        env::var(key)
+            .ok()
+            .and_then(|value| parse_llm_provider(&value))
+    })
+}
+
 fn read_first_positive_u32_env(keys: &[&str]) -> Option<u32> {
     keys.iter().find_map(|key| {
         env::var(key)
@@ -297,6 +366,16 @@ fn read_first_positive_u64_env(keys: &[&str]) -> Option<u64> {
     })
 }
 
+fn read_first_u32_env(keys: &[&str]) -> Option<u32> {
+    keys.iter()
+        .find_map(|key| env::var(key).ok().and_then(|value| parse_u32(&value)))
+}
+
+fn read_first_u64_env(keys: &[&str]) -> Option<u64> {
+    keys.iter()
+        .find_map(|key| env::var(key).ok().and_then(|value| parse_u64(&value)))
+}
+
 fn read_first_positive_u16_env(keys: &[&str]) -> Option<u16> {
     keys.iter().find_map(|key| {
         env::var(key)
@@ -338,6 +417,15 @@ fn parse_bool(raw: &str) -> Option<bool> {
     }
 }
 
+fn parse_llm_provider(raw: &str) -> Option<LlmProvider> {
+    match raw.trim().to_ascii_lowercase().as_str() {
+        "ark" => Some(LlmProvider::Ark),
+        "dash_scope" | "dashscope" => Some(LlmProvider::DashScope),
+        "openai_compatible" | "openai-compatible" | "openai" => Some(LlmProvider::OpenAiCompatible),
+        _ => None,
+    }
+}
+
 fn parse_positive_u32(raw: &str) -> Option<u32> {
     let value = raw.trim().parse::<u32>().ok()?;
     if value == 0 {
@@ -347,6 +435,10 @@ fn parse_positive_u32(raw: &str) -> Option<u32> {
     Some(value)
 }
 
+fn parse_u32(raw: &str) -> Option<u32> {
+    raw.trim().parse::<u32>().ok()
+}
+
 fn parse_positive_u64(raw: &str) -> Option<u64> {
     let value = raw.trim().parse::<u64>().ok()?;
     if value == 0 {
@@ -356,6 +448,10 @@ fn parse_positive_u64(raw: &str) -> Option<u64> {
     Some(value)
 }
 
+fn parse_u64(raw: &str) -> Option<u64> {
+    raw.trim().parse::<u64>().ok()
+}
+
 fn parse_positive_u16(raw: &str) -> Option<u16> {
     let value = raw.trim().parse::<u16>().ok()?;
     if value == 0 {