From 5bfe95d31110e75e84626598f033805b0ae4326c Mon Sep 17 00:00:00 2001 From: sigoden Date: Tue, 14 Nov 2023 11:58:56 +0800 Subject: refactor: trim trailing spaces from the role prompt (#246) --- src/config/role.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/config') 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); } -- cgit v1.2.3