diff options
| author | sigoden <sigoden@gmail.com> | 2025-02-10 16:19:18 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-02-10 16:19:18 +0800 |
| commit | 4fab4c54b2916931bf5705b5aa5e560e57f7f289 (patch) | |
| tree | 1b35844aebc157bfcefad7f61e41c873a98fecdc /src/serve.rs | |
| parent | 69974365ee786af5b643c335e2a12dd7817b84be (diff) | |
| download | aichat-4fab4c54b2916931bf5705b5aa5e560e57f7f289.tar.gz | |
feat: add model field `no_temperature` (#1164)
Diffstat (limited to 'src/serve.rs')
| -rw-r--r-- | src/serve.rs | 11 |
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, |
