fix(client): handle keepalive events in streaming responses

Author: othelot

async-openai/src/client.rs

@@ -737,6 +737,11 @@ pub(crate) async fn stream<O>(
 where
     O: DeserializeOwned + std::marker::Send + 'static,
 {
+    #[derive(Debug, serde::Deserialize)]
+    struct BaseStreamEvent {
+        r#type: String,
+    }
+
     let (tx, rx) = tokio::sync::mpsc::unbounded_channel();
 
     tokio::spawn(async move {
@@ -763,6 +768,12 @@ where
                             break;
                         }
 
+                        if let Ok(event) = serde_json::from_str::<BaseStreamEvent>(&message.data) {
+                            if event.r#type == "keepalive" {
+                                continue;
+                            }
+                        }
+
                         let response = match serde_json::from_str::<O>(&message.data) {
                             Err(e) => Err(map_deserialization_error(e, message.data.as_bytes())),
                             Ok(output) => Ok(output),