fix(openai): handle done opcode

ulagbulag · Jul 18, 2024 · d976e9e · d976e9e
1 parent 551fd03
commit d976e9e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/crates/cassette-plugin-openai-chat/src/hooks.rs b/crates/cassette-plugin-openai-chat/src/hooks.rs
@@ -112,6 +112,11 @@ where
         }
         let data = &data[PATTERN.len()..];
 
+        // handle sub-opcodes
+        if data.starts_with("[DONE]".as_bytes()) {
+            return Ok(());
+        }
+
         let Response { mut choices } = ::serde_json::from_slice(data)?;
         if let Some(MessageChoice {
             index: _,