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/bot.rs | 4 ++++
 1 file changed, 4 insertions(+)

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

diff --git a/src/config/bot.rs b/src/config/bot.rs
index 098c8d7..7177f92 100644
--- a/src/config/bot.rs
+++ b/src/config/bot.rs
@@ -129,6 +129,10 @@ impl RoleLike for Bot {
         &self.model
     }
 
+    fn model_mut(&mut self) -> &mut Model {
+        &mut self.model
+    }
+
     fn temperature(&self) -> Option<f64> {
         self.config.temperature
     }
-- 
cgit v1.2.3