sarvamai 0.1.23a2__py3-none-any.whl → 0.1.23a4__py3-none-any.whl

sarvamai/core/http_sse/_api.py

@@ -1,112 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import re
-from contextlib import asynccontextmanager, contextmanager
-from typing import Any, AsyncGenerator, AsyncIterator, Iterator, cast
-
-import httpx
-from ._decoders import SSEDecoder
-from ._exceptions import SSEError
-from ._models import ServerSentEvent
-
-
-class EventSource:
-    def __init__(self, response: httpx.Response) -> None:
-        self._response = response
-
-    def _check_content_type(self) -> None:
-        content_type = self._response.headers.get("content-type", "").partition(";")[0]
-        if "text/event-stream" not in content_type:
-            raise SSEError(
-                f"Expected response header Content-Type to contain 'text/event-stream', got {content_type!r}"
-            )
-
-    def _get_charset(self) -> str:
-        """Extract charset from Content-Type header, fallback to UTF-8."""
-        content_type = self._response.headers.get("content-type", "")
-
-        # Parse charset parameter using regex
-        charset_match = re.search(r"charset=([^;\s]+)", content_type, re.IGNORECASE)
-        if charset_match:
-            charset = charset_match.group(1).strip("\"'")
-            # Validate that it's a known encoding
-            try:
-                # Test if the charset is valid by trying to encode/decode
-                "test".encode(charset).decode(charset)
-                return charset
-            except (LookupError, UnicodeError):
-                # If charset is invalid, fall back to UTF-8
-                pass
-
-        # Default to UTF-8 if no charset specified or invalid charset
-        return "utf-8"
-
-    @property
-    def response(self) -> httpx.Response:
-        return self._response
-
-    def iter_sse(self) -> Iterator[ServerSentEvent]:
-        self._check_content_type()
-        decoder = SSEDecoder()
-        charset = self._get_charset()
-
-        buffer = ""
-        for chunk in self._response.iter_bytes():
-            # Decode chunk using detected charset
-            text_chunk = chunk.decode(charset, errors="replace")
-            buffer += text_chunk
-
-            # Process complete lines
-            while "\n" in buffer:
-                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
-
-        # Process any remaining data in buffer
-        if buffer.strip():
-            line = buffer.rstrip("\r")
-            sse = decoder.decode(line)
-            if sse is not None:
-                yield sse
-
-    async def aiter_sse(self) -> AsyncGenerator[ServerSentEvent, None]:
-        self._check_content_type()
-        decoder = SSEDecoder()
-        lines = cast(AsyncGenerator[str, None], self._response.aiter_lines())
-        try:
-            async for line in lines:
-                line = line.rstrip("\n")
-                sse = decoder.decode(line)
-                if sse is not None:
-                    yield sse
-        finally:
-            await lines.aclose()
-
-
-@contextmanager
-def connect_sse(client: httpx.Client, method: str, url: str, **kwargs: Any) -> Iterator[EventSource]:
-    headers = kwargs.pop("headers", {})
-    headers["Accept"] = "text/event-stream"
-    headers["Cache-Control"] = "no-store"
-
-    with client.stream(method, url, headers=headers, **kwargs) as response:
-        yield EventSource(response)
-
-
-@asynccontextmanager
-async def aconnect_sse(
-    client: httpx.AsyncClient,
-    method: str,
-    url: str,
-    **kwargs: Any,
-) -> AsyncIterator[EventSource]:
-    headers = kwargs.pop("headers", {})
-    headers["Accept"] = "text/event-stream"
-    headers["Cache-Control"] = "no-store"
-
-    async with client.stream(method, url, headers=headers, **kwargs) as response:
-        yield EventSource(response)

sarvamai/core/http_sse/_decoders.py

@@ -1,61 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from typing import List, Optional
-
-from ._models import ServerSentEvent
-
-
-class SSEDecoder:
-    def __init__(self) -> None:
-        self._event = ""
-        self._data: List[str] = []
-        self._last_event_id = ""
-        self._retry: Optional[int] = None
-
-    def decode(self, line: str) -> Optional[ServerSentEvent]:
-        # See: https://html.spec.whatwg.org/multipage/server-sent-events.html#event-stream-interpretation  # noqa: E501
-
-        if not line:
-            if not self._event and not self._data and not self._last_event_id and self._retry is None:
-                return None
-
-            sse = ServerSentEvent(
-                event=self._event,
-                data="\n".join(self._data),
-                id=self._last_event_id,
-                retry=self._retry,
-            )
-
-            # NOTE: as per the SSE spec, do not reset last_event_id.
-            self._event = ""
-            self._data = []
-            self._retry = None
-
-            return sse
-
-        if line.startswith(":"):
-            return None
-
-        fieldname, _, value = line.partition(":")
-
-        if value.startswith(" "):
-            value = value[1:]
-
-        if fieldname == "event":
-            self._event = value
-        elif fieldname == "data":
-            self._data.append(value)
-        elif fieldname == "id":
-            if "\0" in value:
-                pass
-            else:
-                self._last_event_id = value
-        elif fieldname == "retry":
-            try:
-                self._retry = int(value)
-            except (TypeError, ValueError):
-                pass
-        else:
-            pass  # Field is ignored.
-
-        return None

sarvamai/core/http_sse/_exceptions.py

@@ -1,7 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import httpx
-
-
-class SSEError(httpx.TransportError):
-    pass

sarvamai/core/http_sse/_models.py

@@ -1,17 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import json
-from dataclasses import dataclass
-from typing import Any, Optional
-
-
-@dataclass(frozen=True)
-class ServerSentEvent:
-    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)