llama-cloud 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

llama_cloud/resources/files/types/__init__.py

@@ -1,5 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from .file_create_from_url_resource_info_value import FileCreateFromUrlResourceInfoValue
+from .file_create_permission_info_value import FileCreatePermissionInfoValue
 from .file_create_resource_info_value import FileCreateResourceInfoValue
 
-__all__ = ["FileCreateResourceInfoValue"]
+__all__ = ["FileCreateFromUrlResourceInfoValue", "FileCreatePermissionInfoValue", "FileCreateResourceInfoValue"]

llama_cloud/resources/files/types/file_create_from_url_resource_info_value.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+FileCreateFromUrlResourceInfoValue = typing.Union[
+    typing.Dict[str, typing.Any], typing.List[typing.Any], str, int, float, bool
+]

llama_cloud/resources/files/types/file_create_permission_info_value.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+FileCreatePermissionInfoValue = typing.Union[
+    typing.Dict[str, typing.Any], typing.List[typing.Any], str, int, float, bool
+]

llama_cloud/resources/organizations/client.py

@@ -10,6 +10,7 @@ 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.interval_usage_and_plan import IntervalUsageAndPlan
 from ...types.organization import Organization
 from ...types.organization_create import OrganizationCreate
 from ...types.project import Project
@@ -78,7 +79,7 @@ class OrganizationsClient:
         )
         client.organizations.create_organization(
             request=OrganizationCreate(
-                name="string",
+                name="name",
             ),
         )
         """
@@ -114,7 +115,7 @@ class OrganizationsClient:
         )
         client.organizations.upsert_organization(
             request=OrganizationCreate(
-                name="string",
+                name="name",
             ),
         )
         """
@@ -176,7 +177,7 @@ class OrganizationsClient:
             token="YOUR_TOKEN",
         )
         client.organizations.set_default_organization(
-            organization_id="string",
+            organization_id="organization_id",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -209,7 +210,7 @@ class OrganizationsClient:
             token="YOUR_TOKEN",
         )
         client.organizations.get_organization(
-            organization_id="string",
+            organization_id="organization_id",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -235,7 +236,7 @@ class OrganizationsClient:
         Parameters:
             - organization_id: str.
 
-            - name: typing.Optional[str].
+            - name: typing.Optional[str]. A name for the organization.
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -243,7 +244,7 @@ class OrganizationsClient:
             token="YOUR_TOKEN",
         )
         client.organizations.update_organization(
-            organization_id="string",
+            organization_id="organization_id",
         )
         """
         _request: typing.Dict[str, typing.Any] = {}
@@ -279,7 +280,7 @@ class OrganizationsClient:
             token="YOUR_TOKEN",
         )
         client.organizations.delete_organization(
-            organization_id="string",
+            organization_id="organization_id",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -298,6 +299,38 @@ class OrganizationsClient:
             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]) -> IntervalUsageAndPlan:
+        """
+        Get usage for a project
+
+        Parameters:
+            - organization_id: typing.Optional[str].
+        ---
+        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"
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(IntervalUsageAndPlan, _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.
@@ -311,7 +344,7 @@ class OrganizationsClient:
             token="YOUR_TOKEN",
         )
         client.organizations.list_organization_users(
-            organization_id="string",
+            organization_id="organization_id",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -343,14 +376,15 @@ class OrganizationsClient:
 
             - request: typing.List[UserOrganizationCreate].
         ---
+        from llama_cloud import UserOrganizationCreate
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
         client.organizations.add_users_to_organization(
-            organization_id="string",
-            request=[],
+            organization_id="organization_id",
+            request=[UserOrganizationCreate()],
         )
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -387,8 +421,8 @@ class OrganizationsClient:
             token="YOUR_TOKEN",
         )
         client.organizations.remove_users_from_organization(
-            organization_id="string",
-            member_user_id="string",
+            organization_id="organization_id",
+            member_user_id="member_user_id",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -421,14 +455,15 @@ class OrganizationsClient:
 
             - request: typing.List[UserOrganizationDelete].
         ---
+        from llama_cloud import UserOrganizationDelete
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
         client.organizations.batch_remove_users_from_organization(
-            organization_id="string",
-            request=[],
+            organization_id="organization_id",
+            request=[UserOrganizationDelete()],
         )
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -463,7 +498,7 @@ class OrganizationsClient:
             token="YOUR_TOKEN",
         )
         client.organizations.list_roles(
-            organization_id="string",
+            organization_id="organization_id",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -497,7 +532,7 @@ class OrganizationsClient:
             token="YOUR_TOKEN",
         )
         client.organizations.get_user_role(
-            organization_id="string",
+            organization_id="organization_id",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -539,10 +574,10 @@ class OrganizationsClient:
             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",
+            organization_id="organization_id",
+            user_id="user_id",
+            user_organization_role_create_organization_id="organization_id",
+            role_id="role_id",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -585,8 +620,8 @@ class OrganizationsClient:
             token="YOUR_TOKEN",
         )
         client.organizations.list_projects_by_user(
-            organization_id="string",
-            user_id="string",
+            organization_id="organization_id",
+            user_id="user_id",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -625,9 +660,9 @@ class OrganizationsClient:
             token="YOUR_TOKEN",
         )
         client.organizations.add_user_to_project(
-            organization_id="string",
-            user_id="string",
-            project_id="string",
+            organization_id="organization_id",
+            user_id="user_id",
+            project_id="project_id",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -667,9 +702,9 @@ class OrganizationsClient:
             token="YOUR_TOKEN",
         )
         client.organizations.remove_user_from_project(
-            organization_id="string",
-            user_id="string",
-            project_id="string",
+            organization_id="organization_id",
+            user_id="user_id",
+            project_id="project_id",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -739,7 +774,7 @@ class AsyncOrganizationsClient:
         )
         await client.organizations.create_organization(
             request=OrganizationCreate(
-                name="string",
+                name="name",
             ),
         )
         """
@@ -775,7 +810,7 @@ class AsyncOrganizationsClient:
         )
         await client.organizations.upsert_organization(
             request=OrganizationCreate(
-                name="string",
+                name="name",
             ),
         )
         """
@@ -837,7 +872,7 @@ class AsyncOrganizationsClient:
             token="YOUR_TOKEN",
         )
         await client.organizations.set_default_organization(
-            organization_id="string",
+            organization_id="organization_id",
         )
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -870,7 +905,7 @@ class AsyncOrganizationsClient:
             token="YOUR_TOKEN",
         )
         await client.organizations.get_organization(
-            organization_id="string",
+            organization_id="organization_id",
         )
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -896,7 +931,7 @@ class AsyncOrganizationsClient:
         Parameters:
             - organization_id: str.
 
-            - name: typing.Optional[str].
+            - name: typing.Optional[str]. A name for the organization.
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -904,7 +939,7 @@ class AsyncOrganizationsClient:
             token="YOUR_TOKEN",
         )
         await client.organizations.update_organization(
-            organization_id="string",
+            organization_id="organization_id",
         )
         """
         _request: typing.Dict[str, typing.Any] = {}
@@ -940,7 +975,7 @@ class AsyncOrganizationsClient:
             token="YOUR_TOKEN",
         )
         await client.organizations.delete_organization(
-            organization_id="string",
+            organization_id="organization_id",
         )
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -959,6 +994,38 @@ class AsyncOrganizationsClient:
             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]) -> IntervalUsageAndPlan:
+        """
+        Get usage for a project
+
+        Parameters:
+            - organization_id: typing.Optional[str].
+        ---
+        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"
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(IntervalUsageAndPlan, _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.
@@ -972,7 +1039,7 @@ class AsyncOrganizationsClient:
             token="YOUR_TOKEN",
         )
         await client.organizations.list_organization_users(
-            organization_id="string",
+            organization_id="organization_id",
         )
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -1004,14 +1071,15 @@ class AsyncOrganizationsClient:
 
             - request: typing.List[UserOrganizationCreate].
         ---
+        from llama_cloud import UserOrganizationCreate
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
         await client.organizations.add_users_to_organization(
-            organization_id="string",
-            request=[],
+            organization_id="organization_id",
+            request=[UserOrganizationCreate()],
         )
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -1048,8 +1116,8 @@ class AsyncOrganizationsClient:
             token="YOUR_TOKEN",
         )
         await client.organizations.remove_users_from_organization(
-            organization_id="string",
-            member_user_id="string",
+            organization_id="organization_id",
+            member_user_id="member_user_id",
         )
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -1082,14 +1150,15 @@ class AsyncOrganizationsClient:
 
             - request: typing.List[UserOrganizationDelete].
         ---
+        from llama_cloud import UserOrganizationDelete
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
         await client.organizations.batch_remove_users_from_organization(
-            organization_id="string",
-            request=[],
+            organization_id="organization_id",
+            request=[UserOrganizationDelete()],
         )
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -1124,7 +1193,7 @@ class AsyncOrganizationsClient:
             token="YOUR_TOKEN",
         )
         await client.organizations.list_roles(
-            organization_id="string",
+            organization_id="organization_id",
         )
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -1158,7 +1227,7 @@ class AsyncOrganizationsClient:
             token="YOUR_TOKEN",
         )
         await client.organizations.get_user_role(
-            organization_id="string",
+            organization_id="organization_id",
         )
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -1200,10 +1269,10 @@ class AsyncOrganizationsClient:
             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",
+            organization_id="organization_id",
+            user_id="user_id",
+            user_organization_role_create_organization_id="organization_id",
+            role_id="role_id",
         )
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -1246,8 +1315,8 @@ class AsyncOrganizationsClient:
             token="YOUR_TOKEN",
         )
         await client.organizations.list_projects_by_user(
-            organization_id="string",
-            user_id="string",
+            organization_id="organization_id",
+            user_id="user_id",
         )
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -1286,9 +1355,9 @@ class AsyncOrganizationsClient:
             token="YOUR_TOKEN",
         )
         await client.organizations.add_user_to_project(
-            organization_id="string",
-            user_id="string",
-            project_id="string",
+            organization_id="organization_id",
+            user_id="user_id",
+            project_id="project_id",
         )
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -1328,9 +1397,9 @@ class AsyncOrganizationsClient:
             token="YOUR_TOKEN",
         )
         await client.organizations.remove_user_from_project(
-            organization_id="string",
-            user_id="string",
-            project_id="string",
+            organization_id="organization_id",
+            user_id="user_id",
+            project_id="project_id",
         )
         """
         _response = await self._client_wrapper.httpx_client.request(