Merge pull request #327 from gngpp/fix

fix(toapi): Fix duplicate reply content
0x676e67 · Nov 22, 2023 · 61c05ef · 61c05ef
2 parents ee779d8 + 89c9c10
commit 61c05ef
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/openai/src/chatgpt/model/resp.rs b/openai/src/chatgpt/model/resp.rs
@@ -211,6 +211,10 @@ impl ConvoResponse {
             .collect()
     }
 
+    pub fn raw_messages(&self) -> &[String] {
+        self.message.content.parts.as_slice()
+    }
+
     pub fn message_id(&self) -> &str {
         &self.message.id
     }

diff --git a/openai/src/serve/proxy/toapi/mod.rs b/openai/src/serve/proxy/toapi/mod.rs
@@ -311,7 +311,7 @@ async fn event_convert_handler(
     set_role: &mut bool,
     convo: ConvoResponse,
 ) -> anyhow::Result<Event> {
-    let messages = convo.messages();
+    let messages = convo.raw_messages();
     let message = messages
         .first()
         .ok_or(anyhow::anyhow!("message is empty"))?;