# This file was auto-generated by Fern from our API Definition. import typing import urllib.parse from json.decoder import JSONDecodeError from ...core.api_error import ApiError from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper from ...core.jsonable_encoder import jsonable_encoder from ...core.remove_none_from_dict import remove_none_from_dict from ...errors.unprocessable_entity_error import UnprocessableEntityError from ...types.http_validation_error import HttpValidationError from ...types.organization import Organization from ...types.organization_create import OrganizationCreate from ...types.project import Project from ...types.role import Role from ...types.usage_and_plan import UsageAndPlan from ...types.user_organization import UserOrganization from ...types.user_organization_create import UserOrganizationCreate from ...types.user_organization_delete import UserOrganizationDelete from ...types.user_organization_role import UserOrganizationRole try: import pydantic if pydantic.__version__.startswith("1."): raise ImportError import pydantic.v1 as pydantic # type: ignore except ImportError: import pydantic # type: ignore # this is used as the default value for optional parameters OMIT = typing.cast(typing.Any, ...) class OrganizationsClient: def __init__(self, *, client_wrapper: SyncClientWrapper): self._client_wrapper = client_wrapper def list_organizations(self) -> typing.List[Organization]: """ List organizations for a user. --- from llama_cloud.client import LlamaCloud client = LlamaCloud( token="YOUR_TOKEN", ) client.organizations.list_organizations() """ _response = self._client_wrapper.httpx_client.request( "GET", urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/organizations"), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(typing.List[Organization], _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) def create_organization(self, *, request: OrganizationCreate) -> Organization: """ Create a new organization. Parameters: - request: OrganizationCreate. --- from llama_cloud import OrganizationCreate from llama_cloud.client import LlamaCloud client = LlamaCloud( token="YOUR_TOKEN", ) client.organizations.create_organization( request=OrganizationCreate( name="string", ), ) """ _response = self._client_wrapper.httpx_client.request( "POST", urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/organizations"), json=jsonable_encoder(request), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) def upsert_organization(self, *, request: OrganizationCreate) -> Organization: """ Upsert a new organization. Parameters: - request: OrganizationCreate. --- from llama_cloud import OrganizationCreate from llama_cloud.client import LlamaCloud client = LlamaCloud( token="YOUR_TOKEN", ) client.organizations.upsert_organization( request=OrganizationCreate( name="string", ), ) """ _response = self._client_wrapper.httpx_client.request( "PUT", urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/organizations"), json=jsonable_encoder(request), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) def get_default_organization(self) -> Organization: """ Get the default organization for the user. --- from llama_cloud.client import LlamaCloud client = LlamaCloud( token="YOUR_TOKEN", ) client.organizations.get_default_organization() """ _response = self._client_wrapper.httpx_client.request( "GET", urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/organizations/default"), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) def set_default_organization(self, *, organization_id: str) -> Organization: """ Set the default organization for the user. Parameters: - organization_id: str. The organization's ID. --- from llama_cloud.client import LlamaCloud client = LlamaCloud( token="YOUR_TOKEN", ) client.organizations.set_default_organization( organization_id="string", ) """ _response = self._client_wrapper.httpx_client.request( "PUT", urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/organizations/default"), json=jsonable_encoder({"organization_id": organization_id}), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) def get_organization(self, organization_id: str) -> Organization: """ Get an organization by ID. Parameters: - organization_id: str. --- from llama_cloud.client import LlamaCloud client = LlamaCloud( token="YOUR_TOKEN", ) client.organizations.get_organization( organization_id="string", ) """ _response = self._client_wrapper.httpx_client.request( "GET", urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}"), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) def update_organization(self, organization_id: str, *, name: str) -> Organization: """ Update an existing organization. Parameters: - organization_id: str. - name: str. A name for the organization. --- from llama_cloud.client import LlamaCloud client = LlamaCloud( token="YOUR_TOKEN", ) client.organizations.update_organization( organization_id="string", name="string", ) """ _response = self._client_wrapper.httpx_client.request( "PUT", urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}"), json=jsonable_encoder({"name": name}), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) def delete_organization(self, organization_id: str) -> None: """ Delete an organization by ID. Parameters: - organization_id: str. --- from llama_cloud.client import LlamaCloud client = LlamaCloud( token="YOUR_TOKEN", ) client.organizations.delete_organization( organization_id="string", ) """ _response = self._client_wrapper.httpx_client.request( "DELETE", urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}"), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) def get_organization_usage( self, organization_id: typing.Optional[str], *, get_current_invoice_total: typing.Optional[bool] = None ) -> UsageAndPlan: """ Get usage for a project Parameters: - organization_id: typing.Optional[str]. - get_current_invoice_total: typing.Optional[bool]. --- from llama_cloud.client import LlamaCloud client = LlamaCloud( token="YOUR_TOKEN", ) client.organizations.get_organization_usage() """ _response = self._client_wrapper.httpx_client.request( "GET", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/usage" ), params=remove_none_from_dict({"get_current_invoice_total": get_current_invoice_total}), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(UsageAndPlan, _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) def list_organization_users(self, organization_id: str) -> typing.List[UserOrganization]: """ Get all users in an organization. Parameters: - organization_id: str. --- from llama_cloud.client import LlamaCloud client = LlamaCloud( token="YOUR_TOKEN", ) client.organizations.list_organization_users( organization_id="string", ) """ _response = self._client_wrapper.httpx_client.request( "GET", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users" ), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(typing.List[UserOrganization], _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) def add_users_to_organization( self, organization_id: str, *, request: typing.List[UserOrganizationCreate] ) -> typing.List[UserOrganization]: """ Add a user to an organization. Parameters: - organization_id: str. - request: typing.List[UserOrganizationCreate]. --- from llama_cloud.client import LlamaCloud client = LlamaCloud( token="YOUR_TOKEN", ) client.organizations.add_users_to_organization( organization_id="string", request=[], ) """ _response = self._client_wrapper.httpx_client.request( "PUT", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users" ), json=jsonable_encoder(request), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(typing.List[UserOrganization], _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) def remove_users_from_organization(self, organization_id: str, member_user_id: str) -> None: """ Remove users from an organization by email. Parameters: - organization_id: str. - member_user_id: str. --- from llama_cloud.client import LlamaCloud client = LlamaCloud( token="YOUR_TOKEN", ) client.organizations.remove_users_from_organization( organization_id="string", member_user_id="string", ) """ _response = self._client_wrapper.httpx_client.request( "DELETE", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/{member_user_id}", ), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) def batch_remove_users_from_organization( self, organization_id: str, *, request: typing.List[UserOrganizationDelete] ) -> None: """ Remove a batch of users from an organization. Parameters: - organization_id: str. - request: typing.List[UserOrganizationDelete]. --- from llama_cloud.client import LlamaCloud client = LlamaCloud( token="YOUR_TOKEN", ) client.organizations.batch_remove_users_from_organization( organization_id="string", request=[], ) """ _response = self._client_wrapper.httpx_client.request( "PUT", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/remove" ), json=jsonable_encoder(request), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) def list_roles(self, organization_id: str) -> typing.List[Role]: """ List all roles in an organization. Parameters: - organization_id: str. --- from llama_cloud.client import LlamaCloud client = LlamaCloud( token="YOUR_TOKEN", ) client.organizations.list_roles( organization_id="string", ) """ _response = self._client_wrapper.httpx_client.request( "GET", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/roles" ), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(typing.List[Role], _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) def get_user_role(self, organization_id: str) -> typing.Optional[UserOrganizationRole]: """ Get the role of a user in an organization. Parameters: - organization_id: str. --- from llama_cloud.client import LlamaCloud client = LlamaCloud( token="YOUR_TOKEN", ) client.organizations.get_user_role( organization_id="string", ) """ _response = self._client_wrapper.httpx_client.request( "GET", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/roles" ), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(typing.Optional[UserOrganizationRole], _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) def assign_role_to_user_in_organization( self, organization_id: str, *, user_id: str, user_organization_role_create_organization_id: str, role_id: str ) -> UserOrganizationRole: """ Assign a role to a user in an organization. Parameters: - organization_id: str. - user_id: str. The user's ID. - user_organization_role_create_organization_id: str. The organization's ID. - role_id: str. The role's ID. --- from llama_cloud.client import LlamaCloud client = LlamaCloud( token="YOUR_TOKEN", ) client.organizations.assign_role_to_user_in_organization( organization_id="string", user_id="string", user_organization_role_create_organization_id="string", role_id="string", ) """ _response = self._client_wrapper.httpx_client.request( "PUT", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/roles" ), json=jsonable_encoder( { "user_id": user_id, "organization_id": user_organization_role_create_organization_id, "role_id": role_id, } ), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(UserOrganizationRole, _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) def list_projects_by_user(self, organization_id: str, user_id: str) -> typing.List[Project]: """ List all projects for a user in an organization. Parameters: - organization_id: str. - user_id: str. --- from llama_cloud.client import LlamaCloud client = LlamaCloud( token="YOUR_TOKEN", ) client.organizations.list_projects_by_user( organization_id="string", user_id="string", ) """ _response = self._client_wrapper.httpx_client.request( "GET", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/{user_id}/projects", ), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(typing.List[Project], _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) def add_user_to_project( self, organization_id: typing.Optional[str], user_id: str, *, project_id: typing.Optional[str] = None ) -> typing.Any: """ Add a user to a project. Parameters: - organization_id: typing.Optional[str]. - user_id: str. - project_id: typing.Optional[str]. --- from llama_cloud.client import LlamaCloud client = LlamaCloud( token="YOUR_TOKEN", ) client.organizations.add_user_to_project( user_id="string", ) """ _response = self._client_wrapper.httpx_client.request( "PUT", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/{user_id}/projects", ), params=remove_none_from_dict({"project_id": project_id}), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(typing.Any, _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) def remove_user_from_project(self, organization_id: str, user_id: str, project_id: str) -> typing.Any: """ Remove a user from a project. Parameters: - organization_id: str. - user_id: str. - project_id: str. --- from llama_cloud.client import LlamaCloud client = LlamaCloud( token="YOUR_TOKEN", ) client.organizations.remove_user_from_project( organization_id="string", user_id="string", project_id="string", ) """ _response = self._client_wrapper.httpx_client.request( "DELETE", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/{user_id}/projects/{project_id}", ), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(typing.Any, _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) class AsyncOrganizationsClient: def __init__(self, *, client_wrapper: AsyncClientWrapper): self._client_wrapper = client_wrapper async def list_organizations(self) -> typing.List[Organization]: """ List organizations for a user. --- from llama_cloud.client import AsyncLlamaCloud client = AsyncLlamaCloud( token="YOUR_TOKEN", ) await client.organizations.list_organizations() """ _response = await self._client_wrapper.httpx_client.request( "GET", urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/organizations"), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(typing.List[Organization], _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) async def create_organization(self, *, request: OrganizationCreate) -> Organization: """ Create a new organization. Parameters: - request: OrganizationCreate. --- from llama_cloud import OrganizationCreate from llama_cloud.client import AsyncLlamaCloud client = AsyncLlamaCloud( token="YOUR_TOKEN", ) await client.organizations.create_organization( request=OrganizationCreate( name="string", ), ) """ _response = await self._client_wrapper.httpx_client.request( "POST", urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/organizations"), json=jsonable_encoder(request), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) async def upsert_organization(self, *, request: OrganizationCreate) -> Organization: """ Upsert a new organization. Parameters: - request: OrganizationCreate. --- from llama_cloud import OrganizationCreate from llama_cloud.client import AsyncLlamaCloud client = AsyncLlamaCloud( token="YOUR_TOKEN", ) await client.organizations.upsert_organization( request=OrganizationCreate( name="string", ), ) """ _response = await self._client_wrapper.httpx_client.request( "PUT", urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/organizations"), json=jsonable_encoder(request), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) async def get_default_organization(self) -> Organization: """ Get the default organization for the user. --- from llama_cloud.client import AsyncLlamaCloud client = AsyncLlamaCloud( token="YOUR_TOKEN", ) await client.organizations.get_default_organization() """ _response = await self._client_wrapper.httpx_client.request( "GET", urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/organizations/default"), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) async def set_default_organization(self, *, organization_id: str) -> Organization: """ Set the default organization for the user. Parameters: - organization_id: str. The organization's ID. --- from llama_cloud.client import AsyncLlamaCloud client = AsyncLlamaCloud( token="YOUR_TOKEN", ) await client.organizations.set_default_organization( organization_id="string", ) """ _response = await self._client_wrapper.httpx_client.request( "PUT", urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/organizations/default"), json=jsonable_encoder({"organization_id": organization_id}), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) async def get_organization(self, organization_id: str) -> Organization: """ Get an organization by ID. Parameters: - organization_id: str. --- from llama_cloud.client import AsyncLlamaCloud client = AsyncLlamaCloud( token="YOUR_TOKEN", ) await client.organizations.get_organization( organization_id="string", ) """ _response = await self._client_wrapper.httpx_client.request( "GET", urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}"), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) async def update_organization(self, organization_id: str, *, name: str) -> Organization: """ Update an existing organization. Parameters: - organization_id: str. - name: str. A name for the organization. --- from llama_cloud.client import AsyncLlamaCloud client = AsyncLlamaCloud( token="YOUR_TOKEN", ) await client.organizations.update_organization( organization_id="string", name="string", ) """ _response = await self._client_wrapper.httpx_client.request( "PUT", urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}"), json=jsonable_encoder({"name": name}), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) async def delete_organization(self, organization_id: str) -> None: """ Delete an organization by ID. Parameters: - organization_id: str. --- from llama_cloud.client import AsyncLlamaCloud client = AsyncLlamaCloud( token="YOUR_TOKEN", ) await client.organizations.delete_organization( organization_id="string", ) """ _response = await self._client_wrapper.httpx_client.request( "DELETE", urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}"), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) async def get_organization_usage( self, organization_id: typing.Optional[str], *, get_current_invoice_total: typing.Optional[bool] = None ) -> UsageAndPlan: """ Get usage for a project Parameters: - organization_id: typing.Optional[str]. - get_current_invoice_total: typing.Optional[bool]. --- from llama_cloud.client import AsyncLlamaCloud client = AsyncLlamaCloud( token="YOUR_TOKEN", ) await client.organizations.get_organization_usage() """ _response = await self._client_wrapper.httpx_client.request( "GET", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/usage" ), params=remove_none_from_dict({"get_current_invoice_total": get_current_invoice_total}), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(UsageAndPlan, _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) async def list_organization_users(self, organization_id: str) -> typing.List[UserOrganization]: """ Get all users in an organization. Parameters: - organization_id: str. --- from llama_cloud.client import AsyncLlamaCloud client = AsyncLlamaCloud( token="YOUR_TOKEN", ) await client.organizations.list_organization_users( organization_id="string", ) """ _response = await self._client_wrapper.httpx_client.request( "GET", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users" ), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(typing.List[UserOrganization], _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) async def add_users_to_organization( self, organization_id: str, *, request: typing.List[UserOrganizationCreate] ) -> typing.List[UserOrganization]: """ Add a user to an organization. Parameters: - organization_id: str. - request: typing.List[UserOrganizationCreate]. --- from llama_cloud.client import AsyncLlamaCloud client = AsyncLlamaCloud( token="YOUR_TOKEN", ) await client.organizations.add_users_to_organization( organization_id="string", request=[], ) """ _response = await self._client_wrapper.httpx_client.request( "PUT", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users" ), json=jsonable_encoder(request), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(typing.List[UserOrganization], _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) async def remove_users_from_organization(self, organization_id: str, member_user_id: str) -> None: """ Remove users from an organization by email. Parameters: - organization_id: str. - member_user_id: str. --- from llama_cloud.client import AsyncLlamaCloud client = AsyncLlamaCloud( token="YOUR_TOKEN", ) await client.organizations.remove_users_from_organization( organization_id="string", member_user_id="string", ) """ _response = await self._client_wrapper.httpx_client.request( "DELETE", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/{member_user_id}", ), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) async def batch_remove_users_from_organization( self, organization_id: str, *, request: typing.List[UserOrganizationDelete] ) -> None: """ Remove a batch of users from an organization. Parameters: - organization_id: str. - request: typing.List[UserOrganizationDelete]. --- from llama_cloud.client import AsyncLlamaCloud client = AsyncLlamaCloud( token="YOUR_TOKEN", ) await client.organizations.batch_remove_users_from_organization( organization_id="string", request=[], ) """ _response = await self._client_wrapper.httpx_client.request( "PUT", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/remove" ), json=jsonable_encoder(request), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) async def list_roles(self, organization_id: str) -> typing.List[Role]: """ List all roles in an organization. Parameters: - organization_id: str. --- from llama_cloud.client import AsyncLlamaCloud client = AsyncLlamaCloud( token="YOUR_TOKEN", ) await client.organizations.list_roles( organization_id="string", ) """ _response = await self._client_wrapper.httpx_client.request( "GET", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/roles" ), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(typing.List[Role], _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) async def get_user_role(self, organization_id: str) -> typing.Optional[UserOrganizationRole]: """ Get the role of a user in an organization. Parameters: - organization_id: str. --- from llama_cloud.client import AsyncLlamaCloud client = AsyncLlamaCloud( token="YOUR_TOKEN", ) await client.organizations.get_user_role( organization_id="string", ) """ _response = await self._client_wrapper.httpx_client.request( "GET", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/roles" ), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(typing.Optional[UserOrganizationRole], _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) async def assign_role_to_user_in_organization( self, organization_id: str, *, user_id: str, user_organization_role_create_organization_id: str, role_id: str ) -> UserOrganizationRole: """ Assign a role to a user in an organization. Parameters: - organization_id: str. - user_id: str. The user's ID. - user_organization_role_create_organization_id: str. The organization's ID. - role_id: str. The role's ID. --- from llama_cloud.client import AsyncLlamaCloud client = AsyncLlamaCloud( token="YOUR_TOKEN", ) await client.organizations.assign_role_to_user_in_organization( organization_id="string", user_id="string", user_organization_role_create_organization_id="string", role_id="string", ) """ _response = await self._client_wrapper.httpx_client.request( "PUT", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/roles" ), json=jsonable_encoder( { "user_id": user_id, "organization_id": user_organization_role_create_organization_id, "role_id": role_id, } ), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(UserOrganizationRole, _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) async def list_projects_by_user(self, organization_id: str, user_id: str) -> typing.List[Project]: """ List all projects for a user in an organization. Parameters: - organization_id: str. - user_id: str. --- from llama_cloud.client import AsyncLlamaCloud client = AsyncLlamaCloud( token="YOUR_TOKEN", ) await client.organizations.list_projects_by_user( organization_id="string", user_id="string", ) """ _response = await self._client_wrapper.httpx_client.request( "GET", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/{user_id}/projects", ), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(typing.List[Project], _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) async def add_user_to_project( self, organization_id: typing.Optional[str], user_id: str, *, project_id: typing.Optional[str] = None ) -> typing.Any: """ Add a user to a project. Parameters: - organization_id: typing.Optional[str]. - user_id: str. - project_id: typing.Optional[str]. --- from llama_cloud.client import AsyncLlamaCloud client = AsyncLlamaCloud( token="YOUR_TOKEN", ) await client.organizations.add_user_to_project( user_id="string", ) """ _response = await self._client_wrapper.httpx_client.request( "PUT", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/{user_id}/projects", ), params=remove_none_from_dict({"project_id": project_id}), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(typing.Any, _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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) async def remove_user_from_project(self, organization_id: str, user_id: str, project_id: str) -> typing.Any: """ Remove a user from a project. Parameters: - organization_id: str. - user_id: str. - project_id: str. --- from llama_cloud.client import AsyncLlamaCloud client = AsyncLlamaCloud( token="YOUR_TOKEN", ) await client.organizations.remove_user_from_project( organization_id="string", user_id="string", project_id="string", ) """ _response = await self._client_wrapper.httpx_client.request( "DELETE", urllib.parse.urljoin( f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/{user_id}/projects/{project_id}", ), headers=self._client_wrapper.get_headers(), timeout=60, ) if 200 <= _response.status_code < 300: return pydantic.parse_obj_as(typing.Any, _response.json()) # type: ignore if _response.status_code == 422: raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore try: _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)