agenta 0.48.9__py3-none-any.whl → 0.49.0__py3-none-any.whl

agenta/client/backend/organization/client.py

@@ -1,19 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ..core.client_wrapper import SyncClientWrapper
+
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
+from ..types.invite_request import InviteRequest
 from ..types.organization import Organization
-from ..core.pydantic_utilities import parse_obj_as
-from json.decoder import JSONDecodeError
-from ..core.api_error import ApiError
 from ..types.organization_details import OrganizationDetails
-from ..core.jsonable_encoder import jsonable_encoder
-from ..errors.unprocessable_entity_error import UnprocessableEntityError
-from ..types.http_validation_error import HttpValidationError
-from ..types.invite_request import InviteRequest
-from ..core.serialization import convert_and_respect_annotation_metadata
-from ..core.client_wrapper import AsyncClientWrapper
+from .raw_client import AsyncRawOrganizationClient, RawOrganizationClient
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -21,28 +15,35 @@ OMIT = typing.cast(typing.Any, ...)
 
 class OrganizationClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = RawOrganizationClient(client_wrapper=client_wrapper)
 
-    def list_organizations(
-        self, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.List[Organization]:
+    @property
+    def with_raw_response(self) -> RawOrganizationClient:
         """
-        Returns a list of organizations associated with the user's session.
+        Retrieves a raw implementation of this client that returns raw responses.
 
-        Returns:
-            list[Organization]: A list of organizations associated with the user's session.
+        Returns
+        -------
+        RawOrganizationClient
+        """
+        return self._raw_client
 
-        Raises:
-            HTTPException: If there is an error retrieving the organizations from the database.
+    def fetch_organization_details(
+        self, org_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> OrganizationDetails:
+        """
+        Return the details of the organization.
 
         Parameters
         ----------
+        org_id : str
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        typing.List[Organization]
+        OrganizationDetails
             Successful Response
 
         Examples
@@ -51,45 +52,36 @@ class OrganizationClient:
 
         client = AgentaApi(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
-        client.organization.list_organizations()
+        client.organization.fetch_organization_details(
+            org_id="org_id",
+        )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "organizations",
-            method="GET",
-            request_options=request_options,
+        _response = self._raw_client.fetch_organization_details(
+            org_id, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[Organization],
-                    parse_obj_as(
-                        type_=typing.List[Organization],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
-    def fetch_organization_details(
-        self, org_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> OrganizationDetails:
+    def list_organizations(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[Organization]:
         """
-        Return the details of the organization.
+        Returns a list of organizations associated with the user's session.
+
+        Returns:
+            list[Organization]: A list of organizations associated with the user's session.
+
+        Raises:
+            HTTPException: If there is an error retrieving the organizations from the database.
 
         Parameters
         ----------
-        org_id : str
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        OrganizationDetails
+        typing.List[Organization]
             Successful Response
 
         Examples
@@ -98,40 +90,11 @@ class OrganizationClient:
 
         client = AgentaApi(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
-        )
-        client.organization.fetch_organization_details(
-            org_id="org_id",
         )
+        client.organization.list_organizations()
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"organizations/{jsonable_encoder(org_id)}",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    OrganizationDetails,
-                    parse_obj_as(
-                        type_=OrganizationDetails,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        _response = self._raw_client.list_organizations(request_options=request_options)
+        return _response.data
 
     def invite_user_to_workspace(
         self,
@@ -178,7 +141,6 @@ class OrganizationClient:
 
         client = AgentaApi(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         client.organization.invite_user_to_workspace(
             org_id="org_id",
@@ -186,45 +148,14 @@ class OrganizationClient:
             request=[
                 InviteRequest(
                     email="email",
-                    roles=["roles"],
                 )
             ],
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"organizations/{jsonable_encoder(org_id)}/workspaces/{jsonable_encoder(workspace_id)}/invite",
-            method="POST",
-            json=convert_and_respect_annotation_metadata(
-                object_=request,
-                annotation=typing.Sequence[InviteRequest],
-                direction="write",
-            ),
-            request_options=request_options,
-            omit=OMIT,
+        _response = self._raw_client.invite_user_to_workspace(
+            org_id, workspace_id, request=request, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.Optional[typing.Any],
-                    parse_obj_as(
-                        type_=typing.Optional[typing.Any],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     def resend_invitation(
         self,
@@ -267,7 +198,6 @@ class OrganizationClient:
 
         client = AgentaApi(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         client.organization.resend_invitation(
             org_id="org_id",
@@ -275,41 +205,10 @@ class OrganizationClient:
             email="email",
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"organizations/{jsonable_encoder(org_id)}/workspaces/{jsonable_encoder(workspace_id)}/invite/resend",
-            method="POST",
-            json={
-                "email": email,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
+        _response = self._raw_client.resend_invitation(
+            org_id, workspace_id, email=email, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.Optional[typing.Any],
-                    parse_obj_as(
-                        type_=typing.Optional[typing.Any],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     def accept_invitation(
         self,
@@ -358,7 +257,6 @@ class OrganizationClient:
 
         client = AgentaApi(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         client.organization.accept_invitation(
             org_id="org_id",
@@ -368,71 +266,48 @@ class OrganizationClient:
             email="email",
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"organizations/{jsonable_encoder(org_id)}/workspaces/{jsonable_encoder(workspace_id)}/invite/accept",
-            method="POST",
-            params={
-                "project_id": project_id,
-            },
-            json={
-                "token": token,
-                "email": email,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        _response = self._raw_client.accept_invitation(
+            org_id,
+            workspace_id,
+            project_id=project_id,
+            token=token,
+            email=email,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.Optional[typing.Any],
-                    parse_obj_as(
-                        type_=typing.Optional[typing.Any],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
 
 class AsyncOrganizationClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = AsyncRawOrganizationClient(client_wrapper=client_wrapper)
 
-    async def list_organizations(
-        self, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.List[Organization]:
+    @property
+    def with_raw_response(self) -> AsyncRawOrganizationClient:
         """
-        Returns a list of organizations associated with the user's session.
+        Retrieves a raw implementation of this client that returns raw responses.
 
-        Returns:
-            list[Organization]: A list of organizations associated with the user's session.
+        Returns
+        -------
+        AsyncRawOrganizationClient
+        """
+        return self._raw_client
 
-        Raises:
-            HTTPException: If there is an error retrieving the organizations from the database.
+    async def fetch_organization_details(
+        self, org_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> OrganizationDetails:
+        """
+        Return the details of the organization.
 
         Parameters
         ----------
+        org_id : str
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        typing.List[Organization]
+        OrganizationDetails
             Successful Response
 
         Examples
@@ -443,51 +318,42 @@ class AsyncOrganizationClient:
 
         client = AsyncAgentaApi(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
 
 
         async def main() -> None:
-            await client.organization.list_organizations()
+            await client.organization.fetch_organization_details(
+                org_id="org_id",
+            )
 
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "organizations",
-            method="GET",
-            request_options=request_options,
+        _response = await self._raw_client.fetch_organization_details(
+            org_id, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[Organization],
-                    parse_obj_as(
-                        type_=typing.List[Organization],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
-    async def fetch_organization_details(
-        self, org_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> OrganizationDetails:
+    async def list_organizations(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[Organization]:
         """
-        Return the details of the organization.
+        Returns a list of organizations associated with the user's session.
+
+        Returns:
+            list[Organization]: A list of organizations associated with the user's session.
+
+        Raises:
+            HTTPException: If there is an error retrieving the organizations from the database.
 
         Parameters
         ----------
-        org_id : str
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        OrganizationDetails
+        typing.List[Organization]
             Successful Response
 
         Examples
@@ -498,46 +364,19 @@ class AsyncOrganizationClient:
 
         client = AsyncAgentaApi(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
 
 
         async def main() -> None:
-            await client.organization.fetch_organization_details(
-                org_id="org_id",
-            )
+            await client.organization.list_organizations()
 
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"organizations/{jsonable_encoder(org_id)}",
-            method="GET",
-            request_options=request_options,
+        _response = await self._raw_client.list_organizations(
+            request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    OrganizationDetails,
-                    parse_obj_as(
-                        type_=OrganizationDetails,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     async def invite_user_to_workspace(
         self,
@@ -586,7 +425,6 @@ class AsyncOrganizationClient:
 
         client = AsyncAgentaApi(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
 
 
@@ -597,7 +435,6 @@ class AsyncOrganizationClient:
                 request=[
                     InviteRequest(
                         email="email",
-                        roles=["roles"],
                     )
                 ],
             )
@@ -605,40 +442,10 @@ class AsyncOrganizationClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"organizations/{jsonable_encoder(org_id)}/workspaces/{jsonable_encoder(workspace_id)}/invite",
-            method="POST",
-            json=convert_and_respect_annotation_metadata(
-                object_=request,
-                annotation=typing.Sequence[InviteRequest],
-                direction="write",
-            ),
-            request_options=request_options,
-            omit=OMIT,
+        _response = await self._raw_client.invite_user_to_workspace(
+            org_id, workspace_id, request=request, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.Optional[typing.Any],
-                    parse_obj_as(
-                        type_=typing.Optional[typing.Any],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     async def resend_invitation(
         self,
@@ -683,7 +490,6 @@ class AsyncOrganizationClient:
 
         client = AsyncAgentaApi(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
 
 
@@ -697,41 +503,10 @@ class AsyncOrganizationClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"organizations/{jsonable_encoder(org_id)}/workspaces/{jsonable_encoder(workspace_id)}/invite/resend",
-            method="POST",
-            json={
-                "email": email,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
+        _response = await self._raw_client.resend_invitation(
+            org_id, workspace_id, email=email, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.Optional[typing.Any],
-                    parse_obj_as(
-                        type_=typing.Optional[typing.Any],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     async def accept_invitation(
         self,
@@ -782,7 +557,6 @@ class AsyncOrganizationClient:
 
         client = AsyncAgentaApi(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
 
 
@@ -798,42 +572,12 @@ class AsyncOrganizationClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"organizations/{jsonable_encoder(org_id)}/workspaces/{jsonable_encoder(workspace_id)}/invite/accept",
-            method="POST",
-            params={
-                "project_id": project_id,
-            },
-            json={
-                "token": token,
-                "email": email,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        _response = await self._raw_client.accept_invitation(
+            org_id,
+            workspace_id,
+            project_id=project_id,
+            token=token,
+            email=email,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.Optional[typing.Any],
-                    parse_obj_as(
-                        type_=typing.Optional[typing.Any],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data