summaryrefslogtreecommitdiffstats
path: root/src/config
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2025-02-10 16:19:18 +0800
committerGitHub <noreply@github.com>2025-02-10 16:19:18 +0800
commit4fab4c54b2916931bf5705b5aa5e560e57f7f289 (patch)
tree1b35844aebc157bfcefad7f61e41c873a98fecdc /src/config
parent69974365ee786af5b643c335e2a12dd7817b84be (diff)
downloadaichat-4fab4c54b2916931bf5705b5aa5e560e57f7f289.tar.gz
feat: add model field `no_temperature` (#1164)
Diffstat (limited to 'src/config')
-rw-r--r--src/config/input.rs7
1 files changed, 5 insertions, 2 deletions
diff --git a/src/config/input.rs b/src/config/input.rs
index 1067b2c..a63e9a6 100644
--- a/src/config/input.rs
+++ b/src/config/input.rs
@@ -240,8 +240,11 @@ impl Input {
let mut messages = self.build_messages()?;
patch_messages(&mut messages, model);
model.guard_max_input_tokens(&messages)?;
- let temperature = self.role().temperature();
- let top_p = self.role().top_p();
+ let (temperature, top_p) = if model.no_temperature() {
+ (None, None)
+ } else {
+ (self.role().temperature(), self.role().top_p())
+ };
let functions = self.config.read().select_functions(self.role());
Ok(ChatCompletionsData {
messages,