refactor: adjust command revision prompt

1 files changed, 4 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index 90e1136..4ab4cb8 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -198,7 +198,10 @@ fn execute(config: &GlobalConfig, mut input: Input) -> Result<()> {
                 }
                 "3" => {
                     let revision = Text::new("Enter your revision:").prompt()?;
-                    let text = format!("INPUT: {}\n{eval_str}\nINPUT: {revision}\n", input.text());
+                    let text = format!(
+                        "[INST] {} [/INST]\n{eval_str}\n[INST] {revision} [/INST]\n",
+                        input.text()
+                    );
                     input.set_text(text);
                     return execute(config, input);
                 }