From 7e29f642a99682ea7bb074eb4e7aa7d34d523515 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Sat, 19 Oct 2024 18:25:40 +0800
Subject: feat: support openai o1 models (#935)

---
 src/client/openai.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'src/client/openai.rs')

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| {
-- 
cgit v1.2.3