summaryrefslogtreecommitdiffstats
path: root/src/config/role.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2023-11-14 11:58:56 +0800
committerGitHub <noreply@github.com>2023-11-14 11:58:56 +0800
commit5bfe95d31110e75e84626598f033805b0ae4326c (patch)
tree4503d4a730dc4fa8a102af4d2070bdcb27cd2b4e /src/config/role.rs
parent8f1d8dec5fceda13574ca7e6fbc4093d6ee10f8e (diff)
downloadaichat-5bfe95d31110e75e84626598f033805b0ae4326c.tar.gz
refactor: trim trailing spaces from the role prompt (#246)
Diffstat (limited to 'src/config/role.rs')
-rw-r--r--src/config/role.rs4
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);
}