1 files changed, 4 insertions, 0 deletions
diff --git a/src/config/session.rs b/src/config/session.rs
index df9b082..0b48cbd 100644
--- a/src/config/session.rs
+++ b/src/config/session.rs
@@ -426,6 +426,10 @@ impl RoleLike for Session {
         &self.model
     }
 
+    fn model_mut(&mut self) -> &mut Model {
+        &mut self.model
+    }
+
     fn temperature(&self) -> Option<f64> {
         self.temperature
     }