From ff284779d9757c128acaf690ddf64a87a986e413 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Mon, 17 Jun 2024 15:17:07 +0800
Subject: refactor: make `set max_output_tokens` work for role/session/bot
 (#609)

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

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

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