lark-billing 0.0.6__py3-none-any.whl → 0.0.8__py3-none-any.whl

lark/core/http_sse/_api.py

@@ -0,0 +1,112 @@
+# 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)

lark/core/http_sse/_decoders.py

@@ -0,0 +1,61 @@
+# 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

lark/{types/credit_grant.py → core/http_sse/_exceptions.py}

@@ -1,5 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import typing
+import httpx
 
-CreditGrant = typing.Optional[typing.Any]
+
+class SSEError(httpx.TransportError):
+    pass

lark/core/http_sse/_models.py

@@ -0,0 +1,17 @@
+# 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)

lark/core/pydantic_utilities.py

@@ -220,7 +220,9 @@ def universal_root_validator(
 ) -> Callable[[AnyCallable], AnyCallable]:
     def decorator(func: AnyCallable) -> AnyCallable:
         if IS_PYDANTIC_V2:
-            return cast(AnyCallable, pydantic.model_validator(mode="before" if pre else "after")(func))  # type: ignore[attr-defined]
+            # In Pydantic v2, for RootModel we always use "before" mode
+            # The custom validators transform the input value before the model is created
+            return cast(AnyCallable, pydantic.model_validator(mode="before")(func))  # type: ignore[attr-defined]
         return cast(AnyCallable, pydantic.root_validator(pre=pre)(func))  # type: ignore[call-overload]
 
     return decorator

lark/{feature_access → customer_access}/client.py

@@ -4,37 +4,39 @@ import typing
 
 from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
-from ..types.check_feature_access_response import CheckFeatureAccessResponse
-from .raw_client import AsyncRawFeatureAccessClient, RawFeatureAccessClient
+from ..types.billing_state_response import BillingStateResponse
+from .raw_client import AsyncRawCustomerAccessClient, RawCustomerAccessClient
 
 
-class FeatureAccessClient:
+class CustomerAccessClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._raw_client = RawFeatureAccessClient(client_wrapper=client_wrapper)
+        self._raw_client = RawCustomerAccessClient(client_wrapper=client_wrapper)
 
     @property
-    def with_raw_response(self) -> RawFeatureAccessClient:
+    def with_raw_response(self) -> RawCustomerAccessClient:
         """
         Retrieves a raw implementation of this client that returns raw responses.
 
         Returns
         -------
-        RawFeatureAccessClient
+        RawCustomerAccessClient
         """
         return self._raw_client
 
-    def get_feature_access(
-        self, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> CheckFeatureAccessResponse:
+    def get_billing_state(
+        self, subject_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> BillingStateResponse:
         """
         Parameters
         ----------
+        subject_id : str
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        CheckFeatureAccessResponse
+        BillingStateResponse
             Successful Response
 
         Examples
@@ -43,41 +45,44 @@ class FeatureAccessClient:
 
         client = Lark(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
-        client.feature_access.get_feature_access()
+        client.customer_access.get_billing_state(
+            subject_id="subject_id",
+        )
         """
-        _response = self._raw_client.get_feature_access(request_options=request_options)
+        _response = self._raw_client.get_billing_state(subject_id, request_options=request_options)
         return _response.data
 
 
-class AsyncFeatureAccessClient:
+class AsyncCustomerAccessClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._raw_client = AsyncRawFeatureAccessClient(client_wrapper=client_wrapper)
+        self._raw_client = AsyncRawCustomerAccessClient(client_wrapper=client_wrapper)
 
     @property
-    def with_raw_response(self) -> AsyncRawFeatureAccessClient:
+    def with_raw_response(self) -> AsyncRawCustomerAccessClient:
         """
         Retrieves a raw implementation of this client that returns raw responses.
 
         Returns
         -------
-        AsyncRawFeatureAccessClient
+        AsyncRawCustomerAccessClient
         """
         return self._raw_client
 
-    async def get_feature_access(
-        self, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> CheckFeatureAccessResponse:
+    async def get_billing_state(
+        self, subject_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> BillingStateResponse:
         """
         Parameters
         ----------
+        subject_id : str
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        CheckFeatureAccessResponse
+        BillingStateResponse
             Successful Response
 
         Examples
@@ -88,15 +93,16 @@ class AsyncFeatureAccessClient:
 
         client = AsyncLark(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
 
 
         async def main() -> None:
-            await client.feature_access.get_feature_access()
+            await client.customer_access.get_billing_state(
+                subject_id="subject_id",
+            )
 
 
         asyncio.run(main())
         """
-        _response = await self._raw_client.get_feature_access(request_options=request_options)
+        _response = await self._raw_client.get_billing_state(subject_id, request_options=request_options)
         return _response.data

lark/{feature_access → customer_access}/raw_client.py

@@ -6,42 +6,45 @@ from json.decoder import JSONDecodeError
 from ..core.api_error import ApiError
 from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.http_response import AsyncHttpResponse, HttpResponse
+from ..core.jsonable_encoder import jsonable_encoder
 from ..core.pydantic_utilities import parse_obj_as
 from ..core.request_options import RequestOptions
 from ..errors.unprocessable_entity_error import UnprocessableEntityError
-from ..types.check_feature_access_response import CheckFeatureAccessResponse
+from ..types.billing_state_response import BillingStateResponse
 from ..types.http_validation_error import HttpValidationError
 
 
-class RawFeatureAccessClient:
+class RawCustomerAccessClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def get_feature_access(
-        self, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> HttpResponse[CheckFeatureAccessResponse]:
+    def get_billing_state(
+        self, subject_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> HttpResponse[BillingStateResponse]:
         """
         Parameters
         ----------
+        subject_id : str
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        HttpResponse[CheckFeatureAccessResponse]
+        HttpResponse[BillingStateResponse]
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
-            "feature-access",
+            f"customer-access/{jsonable_encoder(subject_id)}/billing-state",
             method="GET",
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    CheckFeatureAccessResponse,
+                    BillingStateResponse,
                     parse_obj_as(
-                        type_=CheckFeatureAccessResponse,  # type: ignore
+                        type_=BillingStateResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -63,35 +66,37 @@ class RawFeatureAccessClient:
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
 
-class AsyncRawFeatureAccessClient:
+class AsyncRawCustomerAccessClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def get_feature_access(
-        self, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> AsyncHttpResponse[CheckFeatureAccessResponse]:
+    async def get_billing_state(
+        self, subject_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> AsyncHttpResponse[BillingStateResponse]:
         """
         Parameters
         ----------
+        subject_id : str
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        AsyncHttpResponse[CheckFeatureAccessResponse]
+        AsyncHttpResponse[BillingStateResponse]
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "feature-access",
+            f"customer-access/{jsonable_encoder(subject_id)}/billing-state",
             method="GET",
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    CheckFeatureAccessResponse,
+                    BillingStateResponse,
                     parse_obj_as(
-                        type_=CheckFeatureAccessResponse,  # type: ignore
+                        type_=BillingStateResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )

lark/{types/rate_card.py → customer_portal/__init__.py}

@@ -1,5 +1,4 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import typing
+# isort: skip_file
 
-RateCard = typing.Optional[typing.Any]

lark/customer_portal/client.py

@@ -0,0 +1,115 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.request_options import RequestOptions
+from ..types.create_customer_portal_session_response import CreateCustomerPortalSessionResponse
+from .raw_client import AsyncRawCustomerPortalClient, RawCustomerPortalClient
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class CustomerPortalClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._raw_client = RawCustomerPortalClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawCustomerPortalClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawCustomerPortalClient
+        """
+        return self._raw_client
+
+    def create_customer_portal_session(
+        self, *, subject_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> CreateCustomerPortalSessionResponse:
+        """
+        Parameters
+        ----------
+        subject_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        CreateCustomerPortalSessionResponse
+            Successful Response
+
+        Examples
+        --------
+        from lark import Lark
+
+        client = Lark(
+            api_key="YOUR_API_KEY",
+        )
+        client.customer_portal.create_customer_portal_session(
+            subject_id="subject_id",
+        )
+        """
+        _response = self._raw_client.create_customer_portal_session(
+            subject_id=subject_id, request_options=request_options
+        )
+        return _response.data
+
+
+class AsyncCustomerPortalClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawCustomerPortalClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawCustomerPortalClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawCustomerPortalClient
+        """
+        return self._raw_client
+
+    async def create_customer_portal_session(
+        self, *, subject_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> CreateCustomerPortalSessionResponse:
+        """
+        Parameters
+        ----------
+        subject_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        CreateCustomerPortalSessionResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from lark import AsyncLark
+
+        client = AsyncLark(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.customer_portal.create_customer_portal_session(
+                subject_id="subject_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.create_customer_portal_session(
+            subject_id=subject_id, request_options=request_options
+        )
+        return _response.data