diff options
| author | sigoden <sigoden@gmail.com> | 2024-06-17 15:17:07 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-06-17 15:17:07 +0800 |
| commit | ff284779d9757c128acaf690ddf64a87a986e413 (patch) | |
| tree | 36d687eef35d04b3a57014b97a820b7112986a01 /src/config/session.rs | |
| parent | ba884c9fc65d820f8603859acafd055b98897b56 (diff) | |
| download | aichat-ff284779d9757c128acaf690ddf64a87a986e413.tar.gz | |
refactor: make `set max_output_tokens` work for role/session/bot (#609)
Diffstat (limited to 'src/config/session.rs')
| -rw-r--r-- | src/config/session.rs | 4 |
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 } |