fix: add log warning back to steam

flowaicom · Oct 25, 2024 · b5cf856 · b5cf856
1 parent afbe27f
commit b5cf856
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/flow_judge/models/adapters/baseten/adapter.py b/flow_judge/models/adapters/baseten/adapter.py
@@ -334,6 +334,7 @@ async def _fetch_stream(self, request_id: str) -> str:
                         signature = split_chunks[2].replace("\n\n", "").split("signature=")[1]
 
                     except (json.JSONDecodeError, KeyError, IndexError) as e:
+                        logger.warning(f"Failed to parse chunk: {e}")
                         raise BasetenResponseError(f"Invalid JSON response: {str(e)}") from e
 
                     if "data: eot" in decoded_chunk: