feat: add `.prompt` repl command (#420)

2 files changed, 14 insertions, 0 deletions

diff --git a/src/config/mod.rs b/src/config/mod.rs
index 1fe97af..2a3a866 100644
--- a/src/config/mod.rs
+++ b/src/config/mod.rs
@@ -301,6 +301,11 @@ impl Config {
         Ok(path)
     }
 
+    pub fn set_prompt(&mut self, prompt: &str) -> Result<()> {
+        let role = Role::temp(prompt);
+        self.set_role_obj(role)
+    }
+
     pub fn set_role(&mut self, name: &str) -> Result<()> {
         let role = self.retrieve_role(name)?;
         self.set_role_obj(role)
diff --git a/src/config/role.rs b/src/config/role.rs
index 5a6ff3a..50d5b5e 100644
--- a/src/config/role.rs
+++ b/src/config/role.rs
@@ -7,6 +7,7 @@ use crate::{
 use anyhow::{Context, Result};
 use serde::{Deserialize, Serialize};
 
+pub const TEMP_ROLE: &str = "%%";
 pub const SHELL_ROLE: &str = "%shell%";
 pub const EXPLAIN_ROLE: &str = "%explain%";
 pub const CODE_ROLE: &str = "%code%";
@@ -24,6 +25,14 @@ pub struct Role {
 }
 
 impl Role {
+    pub fn temp(prompt: &str) -> Self {
+        Self {
+            name: TEMP_ROLE.into(),
+            prompt: prompt.into(),
+            temperature: None,
+        }
+    }
+
     pub fn find_system_role(name: &str) -> Option<Self> {
         match name {
             SHELL_ROLE => Some(Self::shell()),