From 3180d1d48573a6e12bb955ea0f6a312f80020fe6 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Wed, 27 Mar 2024 08:11:00 +0800
Subject: refactor: seperate temperature from role/session (#378)

---
 src/config/role.rs | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'src/config/role.rs')

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);
-- 
cgit v1.2.3