summaryrefslogtreecommitdiffstats
path: root/src/serve.rs
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/serve.rs
parent69974365ee786af5b643c335e2a12dd7817b84be (diff)
downloadaichat-4fab4c54b2916931bf5705b5aa5e560e57f7f289.tar.gz
feat: add model field `no_temperature` (#1164)
Diffstat (limited to 'src/serve.rs')
-rw-r--r--src/serve.rs11
1 files changed, 9 insertions, 2 deletions
diff --git a/src/serve.rs b/src/serve.rs
index f51e56f..4c1a741 100644
--- a/src/serve.rs
+++ b/src/serve.rs
@@ -270,8 +270,8 @@ impl Server {
let ChatCompletionsReqBody {
model,
messages,
- temperature,
- top_p,
+ mut temperature,
+ mut top_p,
max_tokens,
stream,
tools,
@@ -309,7 +309,14 @@ impl Server {
let completion_id = generate_completion_id();
let created = Utc::now().timestamp();
+
patch_messages(&mut messages, client.model());
+
+ if client.model().no_temperature() {
+ temperature = None;
+ top_p = None;
+ }
+
let data: ChatCompletionsData = ChatCompletionsData {
messages,
temperature,