diff options
| author | sigoden <sigoden@gmail.com> | 2024-05-21 07:56:02 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-05-21 07:56:02 +0800 |
| commit | 2174dc055f660799ed18950fac6a9006f8471fbb (patch) | |
| tree | 999818e0df97b5a06fb1a4236f5659efcc61f1f4 /src/utils | |
| parent | d4de2d761bba9d1c0bd1972113c76cc64720392c (diff) | |
| download | aichat-2174dc055f660799ed18950fac6a9006f8471fbb.tar.gz | |
refactor: change prompt for executing tool functions (#529)
* refactor: change prompt for executing tool functions
* add AICHAT_FUNCTION_AUTO_EXECUTE to auto confirm
Diffstat (limited to 'src/utils')
| -rw-r--r-- | src/utils/mod.rs | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/src/utils/mod.rs b/src/utils/mod.rs index 3ce2a78..fa67c63 100644 --- a/src/utils/mod.rs +++ b/src/utils/mod.rs @@ -35,6 +35,14 @@ pub fn get_env_name(key: &str) -> String { ) } +pub fn get_env_bool(key: &str) -> bool { + if let Ok(value) = env::var(get_env_name(key)) { + value == "1" || value == "true" + } else { + false + } +} + pub fn tokenize(text: &str) -> Vec<&str> { if text.is_ascii() { text.split_inclusive(|c: char| c.is_ascii_whitespace()) |