Follow python idiomatic standards better

Author: aditya-arolkar-swe

src/cohere/core/http_sse/_api.py

@@ -40,6 +40,8 @@ def iter_sse(self) -> Iterator[ServerSentEvent]:
                 line, buffer = buffer.split('\n', 1)
                 line = line.rstrip('\r')
                 sse = decoder.decode(line)
+                # when we reach a "\n\n" => line = '' 
+                # => decoder will attempt to return an SSE Event
                 if sse is not None:
                     yield sse
 

src/cohere/core/http_sse/_models.py

@@ -1,54 +1,15 @@
+from dataclasses import dataclass
 import json
 from typing import Any, Optional
 
 
+@dataclass
 class ServerSentEvent:
-    def __init__(
-        self,
-        event: Optional[str] = None,
-        data: Optional[str] = None,
-        id: Optional[str] = None,
-        retry: Optional[int] = None,
-    ) -> None:
-        if not event:
-            event = "message"
-
-        if data is None:
-            data = ""
-
-        if id is None:
-            id = ""
-
-        self._event = event
-        self._data = data
-        self._id = id
-        self._retry = retry
-
-    @property
-    def event(self) -> str:
-        return self._event
-
-    @property
-    def data(self) -> str:
-        return self._data
-
-    @property
-    def id(self) -> str:
-        return self._id
-
-    @property
-    def retry(self) -> Optional[int]:
-        return self._retry
+    event: str = "message"
+    data: str = ""
+    id: str = ""
+    retry: Optional[int] = None
 
     def json(self) -> Any:
+        """Parse the data field as JSON."""
         return json.loads(self.data)
-
-    def __repr__(self) -> str:
-        pieces = [f"event={self.event!r}"]
-        if self.data != "":
-            pieces.append(f"data={self.data!r}")
-        if self.id != "":
-            pieces.append(f"id={self.id!r}")
-        if self.retry is not None:
-            pieces.append(f"retry={self.retry!r}")
-        return f"ServerSentEvent({', '.join(pieces)})"

src/cohere/v2/raw_client.py

@@ -1,6 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import contextlib
+from dataclasses import asdict
 import json
 import typing
 from json.decoder import JSONDecodeError
@@ -227,21 +228,11 @@ def _iter():
                             _event_source = EventSource(_response)
                             for _sse in _event_source.iter_sse():
                                 try:
-                                    # Skip empty events
-                                    if not _sse.data or _sse.data.strip() == "":
-                                        continue
-                                        
-                                    # Handle [DONE] token from OpenAI-style APIs
-                                    if _sse.data.strip() == '[DONE]':
-                                        continue
-                                        
-                                    parsed_data = json.loads(_sse.data)
-
                                     yield typing.cast(
                                         V2ChatStreamResponse,
                                         construct_type(
                                             type_=V2ChatStreamResponse,  # type: ignore
-                                            object_=parsed_data,
+                                            object_=_sse.json(),
                                         ),
                                     )
                                 except json.JSONDecodeError as e: