diff options
| author | sigoden <sigoden@gmail.com> | 2024-07-30 15:04:50 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-07-30 15:04:50 +0800 |
| commit | a0d421ffc83ba97a1f99f1ba9f85a1689941d171 (patch) | |
| tree | 15af76aab2baf1a2843fb98860b5a3c7099239a9 /src/config/mod.rs | |
| parent | cc74be5d2163fbb1e9d57f85d5546857c6602446 (diff) | |
| download | aichat-a0d421ffc83ba97a1f99f1ba9f85a1689941d171.tar.gz | |
feat: export agent variable as `LLM_AGENT_VAR_*` (#766)
Diffstat (limited to 'src/config/mod.rs')
| -rw-r--r-- | src/config/mod.rs | 8 |
1 files changed, 2 insertions, 6 deletions
diff --git a/src/config/mod.rs b/src/config/mod.rs index 8f719f4..81fa6f8 100644 --- a/src/config/mod.rs +++ b/src/config/mod.rs @@ -342,7 +342,7 @@ impl Config { } pub fn agent_config_dir(name: &str) -> Result<PathBuf> { - match env::var(format!("{}_CONFIG_DIR", convert_env_prefix(name))) { + match env::var(format!("{}_CONFIG_DIR", normalize_env_name(name))) { Ok(value) => Ok(PathBuf::from(value)), Err(_) => Ok(Self::agents_config_dir()?.join(name)), } @@ -365,7 +365,7 @@ impl Config { } pub fn agent_functions_dir(name: &str) -> Result<PathBuf> { - match env::var(format!("{}_FUNCTIONS_DIR", convert_env_prefix(name))) { + match env::var(format!("{}_FUNCTIONS_DIR", normalize_env_name(name))) { Ok(value) => Ok(PathBuf::from(value)), Err(_) => Ok(Self::agents_functions_dir()?.join(name)), } @@ -1904,7 +1904,3 @@ fn complete_option_bool(value: Option<bool>) -> Vec<String> { None => vec!["true".to_string(), "false".to_string()], } } - -fn convert_env_prefix(value: &str) -> String { - value.replace('-', "_").to_ascii_uppercase() -} |
