athena-intelligence 0.1.210__py3-none-any.whl → 0.1.303__py3-none-any.whl

athena/assets/client.py

@@ -4,9 +4,16 @@ import typing
 
 from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
+from ..types.creatable_asset_type import CreatableAssetType
+from ..types.create_asset_response_out import CreateAssetResponseOut
+from ..types.create_project_response_out import CreateProjectResponseOut
 from ..types.paginated_assets_out import PaginatedAssetsOut
+from ..types.public_asset_out import PublicAssetOut
 from .raw_client import AsyncRawAssetsClient, RawAssetsClient
 
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
 
 class AssetsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
@@ -64,13 +71,176 @@ class AssetsClient:
         client = Athena(
             api_key="YOUR_API_KEY",
         )
-        client.assets.list()
+        client.assets.list(
+            limit=1,
+            offset=1,
+            filters="filters",
+            sort="sort",
+        )
         """
         _response = self._raw_client.list(
             limit=limit, offset=offset, filters=filters, sort=sort, request_options=request_options
         )
         return _response.data
 
+    def create(
+        self,
+        *,
+        asset_type: CreatableAssetType,
+        parent_folder_id: typing.Optional[str] = OMIT,
+        title: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> CreateAssetResponseOut:
+        """
+        Create a new asset such as a spreadsheet, document, or folder in your workspace. This endpoint uses internal GraphQL mutations to create assets with proper permissions and workspace integration.
+
+        Parameters
+        ----------
+        asset_type : CreatableAssetType
+            Type of asset to create. Supported types: 'spreadsheet' (or 'sheet'), 'document' (or 'doc'), 'folder'
+
+        parent_folder_id : typing.Optional[str]
+            ID of the parent folder to create the asset in
+
+        title : typing.Optional[str]
+            Title for the new asset
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        CreateAssetResponseOut
+            Asset created successfully
+
+        Examples
+        --------
+        from athena import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.assets.create(
+            asset_type="spreadsheet",
+            parent_folder_id="asset_folder_12345",
+            title="My New Spreadsheet",
+        )
+        """
+        _response = self._raw_client.create(
+            asset_type=asset_type, parent_folder_id=parent_folder_id, title=title, request_options=request_options
+        )
+        return _response.data
+
+    def create_project(
+        self,
+        *,
+        title: str,
+        custom_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        parent_folder_id: typing.Optional[str] = OMIT,
+        project_type: typing.Optional[str] = OMIT,
+        share_with_emails: typing.Optional[typing.Sequence[str]] = OMIT,
+        tags: typing.Optional[typing.Sequence[str]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> CreateProjectResponseOut:
+        """
+        Create a new project with custom metadata. Projects can be typed (e.g., 'candidate', 'user', 'company') and include flexible custom metadata for storing additional information.
+
+        Parameters
+        ----------
+        title : str
+            The project title
+
+        custom_metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+            A flexible dictionary for storing custom metadata
+
+        description : typing.Optional[str]
+            Optional project description
+
+        parent_folder_id : typing.Optional[str]
+            Optional parent folder ID
+
+        project_type : typing.Optional[str]
+            User-defined project type (e.g., 'candidate', 'user', 'company')
+
+        share_with_emails : typing.Optional[typing.Sequence[str]]
+            Optional list of email addresses to share the project with (VIEW permission)
+
+        tags : typing.Optional[typing.Sequence[str]]
+            Optional list of tags for categorizing the project
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        CreateProjectResponseOut
+            Project created successfully
+
+        Examples
+        --------
+        from athena import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.assets.create_project(
+            custom_metadata={
+                "email": "john.doe@example.com",
+                "phone": "+1-555-0123",
+                "source": "linkedin",
+                "status": "active",
+            },
+            description="Candidate profile for senior software engineer position",
+            parent_folder_id="asset_folder_123",
+            project_type="candidate",
+            share_with_emails=["colleague@example.com", "manager@example.com"],
+            tags=["engineering", "senior", "active"],
+            title="John Doe - Software Engineer",
+        )
+        """
+        _response = self._raw_client.create_project(
+            title=title,
+            custom_metadata=custom_metadata,
+            description=description,
+            parent_folder_id=parent_folder_id,
+            project_type=project_type,
+            share_with_emails=share_with_emails,
+            tags=tags,
+            request_options=request_options,
+        )
+        return _response.data
+
+    def get(self, asset_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> PublicAssetOut:
+        """
+        Retrieve a single asset by its ID. Returns comprehensive metadata including creation info, tags, timestamps, media type, and AI-generated summary.
+
+        Parameters
+        ----------
+        asset_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        PublicAssetOut
+            Successfully retrieved asset
+
+        Examples
+        --------
+        from athena import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.assets.get(
+            asset_id="asset_id",
+        )
+        """
+        _response = self._raw_client.get(asset_id, request_options=request_options)
+        return _response.data
+
 
 class AsyncAssetsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -133,7 +303,12 @@ class AsyncAssetsClient:
 
 
         async def main() -> None:
-            await client.assets.list()
+            await client.assets.list(
+                limit=1,
+                offset=1,
+                filters="filters",
+                sort="sort",
+            )
 
 
         asyncio.run(main())
@@ -142,3 +317,185 @@ class AsyncAssetsClient:
             limit=limit, offset=offset, filters=filters, sort=sort, request_options=request_options
         )
         return _response.data
+
+    async def create(
+        self,
+        *,
+        asset_type: CreatableAssetType,
+        parent_folder_id: typing.Optional[str] = OMIT,
+        title: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> CreateAssetResponseOut:
+        """
+        Create a new asset such as a spreadsheet, document, or folder in your workspace. This endpoint uses internal GraphQL mutations to create assets with proper permissions and workspace integration.
+
+        Parameters
+        ----------
+        asset_type : CreatableAssetType
+            Type of asset to create. Supported types: 'spreadsheet' (or 'sheet'), 'document' (or 'doc'), 'folder'
+
+        parent_folder_id : typing.Optional[str]
+            ID of the parent folder to create the asset in
+
+        title : typing.Optional[str]
+            Title for the new asset
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        CreateAssetResponseOut
+            Asset created successfully
+
+        Examples
+        --------
+        import asyncio
+
+        from athena import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.assets.create(
+                asset_type="spreadsheet",
+                parent_folder_id="asset_folder_12345",
+                title="My New Spreadsheet",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.create(
+            asset_type=asset_type, parent_folder_id=parent_folder_id, title=title, request_options=request_options
+        )
+        return _response.data
+
+    async def create_project(
+        self,
+        *,
+        title: str,
+        custom_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        parent_folder_id: typing.Optional[str] = OMIT,
+        project_type: typing.Optional[str] = OMIT,
+        share_with_emails: typing.Optional[typing.Sequence[str]] = OMIT,
+        tags: typing.Optional[typing.Sequence[str]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> CreateProjectResponseOut:
+        """
+        Create a new project with custom metadata. Projects can be typed (e.g., 'candidate', 'user', 'company') and include flexible custom metadata for storing additional information.
+
+        Parameters
+        ----------
+        title : str
+            The project title
+
+        custom_metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+            A flexible dictionary for storing custom metadata
+
+        description : typing.Optional[str]
+            Optional project description
+
+        parent_folder_id : typing.Optional[str]
+            Optional parent folder ID
+
+        project_type : typing.Optional[str]
+            User-defined project type (e.g., 'candidate', 'user', 'company')
+
+        share_with_emails : typing.Optional[typing.Sequence[str]]
+            Optional list of email addresses to share the project with (VIEW permission)
+
+        tags : typing.Optional[typing.Sequence[str]]
+            Optional list of tags for categorizing the project
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        CreateProjectResponseOut
+            Project created successfully
+
+        Examples
+        --------
+        import asyncio
+
+        from athena import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.assets.create_project(
+                custom_metadata={
+                    "email": "john.doe@example.com",
+                    "phone": "+1-555-0123",
+                    "source": "linkedin",
+                    "status": "active",
+                },
+                description="Candidate profile for senior software engineer position",
+                parent_folder_id="asset_folder_123",
+                project_type="candidate",
+                share_with_emails=["colleague@example.com", "manager@example.com"],
+                tags=["engineering", "senior", "active"],
+                title="John Doe - Software Engineer",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.create_project(
+            title=title,
+            custom_metadata=custom_metadata,
+            description=description,
+            parent_folder_id=parent_folder_id,
+            project_type=project_type,
+            share_with_emails=share_with_emails,
+            tags=tags,
+            request_options=request_options,
+        )
+        return _response.data
+
+    async def get(self, asset_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> PublicAssetOut:
+        """
+        Retrieve a single asset by its ID. Returns comprehensive metadata including creation info, tags, timestamps, media type, and AI-generated summary.
+
+        Parameters
+        ----------
+        asset_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        PublicAssetOut
+            Successfully retrieved asset
+
+        Examples
+        --------
+        import asyncio
+
+        from athena import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.assets.get(
+                asset_id="asset_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.get(asset_id, request_options=request_options)
+        return _response.data