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

athena/__init__.py

@@ -31,7 +31,10 @@ if typing.TYPE_CHECKING:
         ConversationAssetInfo,
         ConversationMessage,
         ConversationResult,
+        CreatableAssetType,
+        CreateAssetResponseOut,
         CreateNewSheetTabResponse,
+        CreateProjectResponseOut,
         CustomAgentResponse,
         DataFrameRequestOut,
         DataFrameRequestOutColumnsItem,
@@ -39,9 +42,7 @@ if typing.TYPE_CHECKING:
         DataFrameRequestOutIndexItem,
         DataFrameUnknownFormatError,
         DimensionProperties,
-        DocumentChunk,
         DriveAgentResponse,
-        FileChunkRequestOut,
         FileTooLargeError,
         FolderResponse,
         GeneralAgentConfig,
@@ -124,7 +125,10 @@ _dynamic_imports: typing.Dict[str, str] = {
     "ConversationAssetInfo": ".types",
     "ConversationMessage": ".types",
     "ConversationResult": ".types",
+    "CreatableAssetType": ".types",
+    "CreateAssetResponseOut": ".types",
     "CreateNewSheetTabResponse": ".types",
+    "CreateProjectResponseOut": ".types",
     "CustomAgentResponse": ".types",
     "DataFrameRequestOut": ".types",
     "DataFrameRequestOutColumnsItem": ".types",
@@ -132,9 +136,7 @@ _dynamic_imports: typing.Dict[str, str] = {
     "DataFrameRequestOutIndexItem": ".types",
     "DataFrameUnknownFormatError": ".types",
     "DimensionProperties": ".types",
-    "DocumentChunk": ".types",
     "DriveAgentResponse": ".types",
-    "FileChunkRequestOut": ".types",
     "FileTooLargeError": ".types",
     "FolderResponse": ".types",
     "GeneralAgentConfig": ".types",
@@ -239,7 +241,10 @@ __all__ = [
     "ConversationAssetInfo",
     "ConversationMessage",
     "ConversationResult",
+    "CreatableAssetType",
+    "CreateAssetResponseOut",
     "CreateNewSheetTabResponse",
+    "CreateProjectResponseOut",
     "CustomAgentResponse",
     "DataFrameRequestOut",
     "DataFrameRequestOutColumnsItem",
@@ -247,9 +252,7 @@ __all__ = [
     "DataFrameRequestOutIndexItem",
     "DataFrameUnknownFormatError",
     "DimensionProperties",
-    "DocumentChunk",
     "DriveAgentResponse",
-    "FileChunkRequestOut",
     "FileTooLargeError",
     "FolderResponse",
     "GeneralAgentConfig",

athena/assets/client.py

@@ -4,10 +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):
@@ -77,6 +83,134 @@ class AssetsClient:
         )
         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.
@@ -184,6 +318,150 @@ class AsyncAssetsClient:
         )
         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.