Skip None chunks from DONE markers in streaming

Author: alice-cohere

src/cohere/base_client.py

@@ -1967,7 +1967,9 @@ async def main() -> None:
             request_options=request_options,
         ) as r:
             async for _chunk in r.data:
-                yield _chunk
+                # Skip None chunks (e.g., from [DONE] markers in SSE streams)
+                if _chunk is not None:
+                    yield _chunk
 
     async def chat(
         self,
@@ -2430,7 +2432,9 @@ async def main() -> None:
             request_options=request_options,
         ) as r:
             async for _chunk in r.data:
-                yield _chunk
+                # Skip None chunks (e.g., from [DONE] markers in SSE streams)
+                if _chunk is not None:
+                    yield _chunk
 
     async def generate(
         self,

src/cohere/v2/client.py

@@ -761,7 +761,9 @@ async def main() -> None:
             request_options=request_options,
         ) as r:
             async for _chunk in r.data:
-                yield _chunk
+                # Skip None chunks (e.g., from [DONE] markers in SSE streams)
+                if _chunk is not None:
+                    yield _chunk
 
     async def chat(
         self,