summaryrefslogtreecommitdiffstats
path: root/src/client/openai.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-10-19 18:25:40 +0800
committerGitHub <noreply@github.com>2024-10-19 18:25:40 +0800
commit7e29f642a99682ea7bb074eb4e7aa7d34d523515 (patch)
tree248bbcb5d3af83981e451eb115f1fde23c8386de /src/client/openai.rs
parentc5e4421e0da6d647a636ec4cf23c110ff515e79a (diff)
downloadaichat-7e29f642a99682ea7bb074eb4e7aa7d34d523515.tar.gz
feat: support openai o1 models (#935)
Diffstat (limited to 'src/client/openai.rs')
-rw-r--r--src/client/openai.rs6
1 files changed, 5 insertions, 1 deletions
diff --git a/src/client/openai.rs b/src/client/openai.rs
index c4c2b0c..831b64f 100644
--- a/src/client/openai.rs
+++ b/src/client/openai.rs
@@ -193,13 +193,17 @@ struct EmbeddingsResBodyEmbedding {
pub fn openai_build_chat_completions_body(data: ChatCompletionsData, model: &Model) -> Value {
let ChatCompletionsData {
- messages,
+ mut messages,
temperature,
top_p,
functions,
stream,
} = data;
+ if model.no_system_message() {
+ patch_system_message(&mut messages);
+ }
+
let messages: Vec<Value> = messages
.into_iter()
.flat_map(|message| {