1 files changed, 4 insertions, 0 deletions
diff --git a/src/config/role.rs b/src/config/role.rs
index 8e4a20e..57da38c 100644
--- a/src/config/role.rs
+++ b/src/config/role.rs
@@ -89,6 +89,10 @@ For example if the prompt is "Hello world Python", you should return "print('Hel
         self.prompt.contains(INPUT_PLACEHOLDER)
     }
 
+    pub fn set_temperature(&mut self, value: Option<f64>) {
+        self.temperature = value;
+    }
+
     pub fn complete_prompt_args(&mut self, name: &str) {
         self.name = name.to_string();
         self.prompt = complete_prompt_args(&self.prompt, &self.name);