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

athena/assets/raw_client.py

@@ -9,12 +9,20 @@ from ..core.http_response import AsyncHttpResponse, HttpResponse
 from ..core.jsonable_encoder import jsonable_encoder
 from ..core.pydantic_utilities import parse_obj_as
 from ..core.request_options import RequestOptions
+from ..errors.bad_request_error import BadRequestError
+from ..errors.internal_server_error import InternalServerError
 from ..errors.not_found_error import NotFoundError
 from ..errors.unauthorized_error import UnauthorizedError
 from ..errors.unprocessable_entity_error import UnprocessableEntityError
+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
 
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
 
 class RawAssetsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
@@ -91,6 +99,210 @@ class RawAssetsClient:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
+    def create(
+        self,
+        *,
+        asset_type: CreatableAssetType,
+        parent_folder_id: typing.Optional[str] = OMIT,
+        title: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[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
+        -------
+        HttpResponse[CreateAssetResponseOut]
+            Asset created successfully
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/assets/create",
+            method="POST",
+            json={
+                "asset_type": asset_type,
+                "parent_folder_id": parent_folder_id,
+                "title": title,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    CreateAssetResponseOut,
+                    parse_obj_as(
+                        type_=CreateAssetResponseOut,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=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(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    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,
+    ) -> HttpResponse[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
+        -------
+        HttpResponse[CreateProjectResponseOut]
+            Project created successfully
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/assets/create_project",
+            method="POST",
+            json={
+                "custom_metadata": custom_metadata,
+                "description": description,
+                "parent_folder_id": parent_folder_id,
+                "project_type": project_type,
+                "share_with_emails": share_with_emails,
+                "tags": tags,
+                "title": title,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    CreateProjectResponseOut,
+                    parse_obj_as(
+                        type_=CreateProjectResponseOut,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=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(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
     def get(
         self, asset_id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> HttpResponse[PublicAssetOut]:
@@ -238,6 +450,210 @@ class AsyncRawAssetsClient:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
+    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,
+    ) -> AsyncHttpResponse[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
+        -------
+        AsyncHttpResponse[CreateAssetResponseOut]
+            Asset created successfully
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/assets/create",
+            method="POST",
+            json={
+                "asset_type": asset_type,
+                "parent_folder_id": parent_folder_id,
+                "title": title,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    CreateAssetResponseOut,
+                    parse_obj_as(
+                        type_=CreateAssetResponseOut,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=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(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    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,
+    ) -> AsyncHttpResponse[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
+        -------
+        AsyncHttpResponse[CreateProjectResponseOut]
+            Project created successfully
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/assets/create_project",
+            method="POST",
+            json={
+                "custom_metadata": custom_metadata,
+                "description": description,
+                "parent_folder_id": parent_folder_id,
+                "project_type": project_type,
+                "share_with_emails": share_with_emails,
+                "tags": tags,
+                "title": title,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    CreateProjectResponseOut,
+                    parse_obj_as(
+                        type_=CreateProjectResponseOut,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=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(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
     async def get(
         self, asset_id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> AsyncHttpResponse[PublicAssetOut]:

athena/client.py

@@ -413,7 +413,7 @@ class AthenaModel(RemoteRunnable):
         self.api_key = api_key
         self.timeout = timeout
         super().__init__(
-            base_url + "/api/models/general",
+            base_url + "/api/v0/agents/general",
             headers={"X-API-KEY": api_key},
             timeout=timeout,
         )
@@ -467,14 +467,12 @@ class Athena(BaseAthena):
                     " (ATHENA_API_KEY environment variable not found)"
                 )
         if environment == SpecialEnvironments.AUTODETECT_ENVIRONMENT:
-            if "ATHENA_API_URL" in os.environ:
+            current_url = os.environ.get("ATHENA_API_URL", "https://api.athenaintel.com")
 
-                class AutodetectedEnvironments(enum.Enum):
-                    CURRENT = os.environ["ATHENA_API_URL"]
+            class _CurrentEnv(enum.Enum):
+                CURRENT = current_url
 
-                environment = cast(AthenaEnvironment, AutodetectedEnvironments.CURRENT)
-            else:
-                environment = AthenaEnvironment.PRODUCTION
+            environment = cast(AthenaEnvironment, _CurrentEnv.CURRENT)
         super().__init__(
             base_url=base_url,
             environment=environment,  # type: ignore[arg-type]
@@ -543,14 +541,12 @@ class AsyncAthena(AsyncBaseAthena):
                     " (ATHENA_API_KEY environment variable not found)"
                 )
         if environment == SpecialEnvironments.AUTODETECT_ENVIRONMENT:
-            if "ATHENA_API_URL" in os.environ:
+            current_url = os.environ.get("ATHENA_API_URL", "https://api.athenaintel.com")
 
-                class AutodetectedEnvironments(enum.Enum):
-                    CURRENT = os.environ["ATHENA_API_URL"]
+            class _CurrentEnv(enum.Enum):
+                CURRENT = current_url
 
-                environment = cast(AthenaEnvironment, AutodetectedEnvironments.CURRENT)
-            else:
-                environment = AthenaEnvironment.PRODUCTION
+            environment = cast(AthenaEnvironment, _CurrentEnv.CURRENT)
         self._tools: typing.Optional[WrappedAsyncToolsClient] = None
         super().__init__(
             base_url=base_url,

athena/core/client_wrapper.py

@@ -22,10 +22,10 @@ class BaseClientWrapper:
 
     def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
-            "User-Agent": "athena-intelligence/0.1.260",
+            "User-Agent": "athena-intelligence/0.1.375",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "athena-intelligence",
-            "X-Fern-SDK-Version": "0.1.260",
+            "X-Fern-SDK-Version": "0.1.375",
             **(self.get_custom_headers() or {}),
         }
         headers["X-API-KEY"] = self.api_key

athena/environment.py

@@ -5,5 +5,3 @@ import enum
 
 class AthenaEnvironment(enum.Enum):
     PRODUCTION = "https://api.athenaintel.com"
-    STAGING = "https://staging-api.athenaintel.com"
-    CANARY = "https://api-canary.athena-olympus.com"

athena/tools/client.py

@@ -10,7 +10,6 @@ from ..core.request_options import RequestOptions
 from ..types.asset_content_request_out import AssetContentRequestOut
 from ..types.asset_screenshot_response_out import AssetScreenshotResponseOut
 from ..types.data_frame_request_out import DataFrameRequestOut
-from ..types.file_chunk_request_out import FileChunkRequestOut
 from ..types.folder_response import FolderResponse
 from ..types.save_asset_request_out import SaveAssetRequestOut
 from .raw_client import AsyncRawToolsClient, RawToolsClient
@@ -49,7 +48,7 @@ class ToolsClient:
 
     def get_asset_chunks(
         self, *, asset_ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
-    ) -> FileChunkRequestOut:
+    ) -> typing.Optional[typing.Any]:
         """
         Get the chunks of a file.
 
@@ -63,7 +62,7 @@ class ToolsClient:
 
         Returns
         -------
-        FileChunkRequestOut
+        typing.Optional[typing.Any]
             Successful Response
 
         Examples
@@ -160,7 +159,6 @@ class ToolsClient:
         self,
         *,
         asset_id: typing.Optional[str] = None,
-        folder_id: typing.Optional[str] = None,
         include_asset_details: typing.Optional[bool] = None,
         include_system_files: typing.Optional[bool] = None,
         request_options: typing.Optional[RequestOptions] = None,
@@ -172,8 +170,6 @@ class ToolsClient:
         ----------
         asset_id : typing.Optional[str]
 
-        folder_id : typing.Optional[str]
-
         include_asset_details : typing.Optional[bool]
 
         include_system_files : typing.Optional[bool]
@@ -195,14 +191,12 @@ class ToolsClient:
         )
         client.tools.list_contents(
             asset_id="asset_id",
-            folder_id="folder_id",
             include_asset_details=True,
             include_system_files=True,
         )
         """
         _response = self._raw_client.list_contents(
             asset_id=asset_id,
-            folder_id=folder_id,
             include_asset_details=include_asset_details,
             include_system_files=include_system_files,
             request_options=request_options,
@@ -410,7 +404,7 @@ class AsyncToolsClient:
 
     async def get_asset_chunks(
         self, *, asset_ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
-    ) -> FileChunkRequestOut:
+    ) -> typing.Optional[typing.Any]:
         """
         Get the chunks of a file.
 
@@ -424,7 +418,7 @@ class AsyncToolsClient:
 
         Returns
         -------
-        FileChunkRequestOut
+        typing.Optional[typing.Any]
             Successful Response
 
         Examples
@@ -545,7 +539,6 @@ class AsyncToolsClient:
         self,
         *,
         asset_id: typing.Optional[str] = None,
-        folder_id: typing.Optional[str] = None,
         include_asset_details: typing.Optional[bool] = None,
         include_system_files: typing.Optional[bool] = None,
         request_options: typing.Optional[RequestOptions] = None,
@@ -557,8 +550,6 @@ class AsyncToolsClient:
         ----------
         asset_id : typing.Optional[str]
 
-        folder_id : typing.Optional[str]
-
         include_asset_details : typing.Optional[bool]
 
         include_system_files : typing.Optional[bool]
@@ -585,7 +576,6 @@ class AsyncToolsClient:
         async def main() -> None:
             await client.tools.list_contents(
                 asset_id="asset_id",
-                folder_id="folder_id",
                 include_asset_details=True,
                 include_system_files=True,
             )
@@ -595,7 +585,6 @@ class AsyncToolsClient:
         """
         _response = await self._raw_client.list_contents(
             asset_id=asset_id,
-            folder_id=folder_id,
             include_asset_details=include_asset_details,
             include_system_files=include_system_files,
             request_options=request_options,