diff options
| author | sigoden <sigoden@gmail.com> | 2023-11-14 11:58:56 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-11-14 11:58:56 +0800 |
| commit | 5bfe95d31110e75e84626598f033805b0ae4326c (patch) | |
| tree | 4503d4a730dc4fa8a102af4d2070bdcb27cd2b4e /src/config | |
| parent | 8f1d8dec5fceda13574ca7e6fbc4093d6ee10f8e (diff) | |
| download | aichat-5bfe95d31110e75e84626598f033805b0ae4326c.tar.gz | |
refactor: trim trailing spaces from the role prompt (#246)
Diffstat (limited to 'src/config')
| -rw-r--r-- | src/config/role.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/config/role.rs b/src/config/role.rs index eaf805f..2b8fea1 100644 --- a/src/config/role.rs +++ b/src/config/role.rs @@ -45,7 +45,7 @@ impl Role { if self.embedded() { merge_prompt_content(&self.prompt, content) } else { - format!("{}\n{content}", self.prompt) + format!("{}\n\n{content}", self.prompt) } } @@ -76,7 +76,7 @@ fn merge_prompt_content(prompt: &str, content: &str) -> String { } fn complete_prompt_args(prompt: &str, name: &str) -> String { - let mut prompt = prompt.to_string(); + let mut prompt = prompt.trim().to_string(); for (i, arg) in name.split(':').skip(1).enumerate() { prompt = prompt.replace(&format!("__ARG{}__", i + 1), arg); } |
