summaryrefslogtreecommitdiffstats
path: root/src/config/bot.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-06-17 15:17:07 +0800
committerGitHub <noreply@github.com>2024-06-17 15:17:07 +0800
commitff284779d9757c128acaf690ddf64a87a986e413 (patch)
tree36d687eef35d04b3a57014b97a820b7112986a01 /src/config/bot.rs
parentba884c9fc65d820f8603859acafd055b98897b56 (diff)
downloadaichat-ff284779d9757c128acaf690ddf64a87a986e413.tar.gz
refactor: make `set max_output_tokens` work for role/session/bot (#609)
Diffstat (limited to 'src/config/bot.rs')
-rw-r--r--src/config/bot.rs4
1 files changed, 4 insertions, 0 deletions
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
}