athena-intelligence 0.1.123__py3-none-any.whl → 0.1.124__py3-none-any.whl

athena/__init__.py

@@ -1,6 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .types import (
+    AssetNode,
     AssetNotFoundError,
     Chunk,
     ChunkContentItem,
@@ -18,6 +19,7 @@ from .types import (
     DriveAgentResponse,
     FileChunkRequestOut,
     FileTooLargeError,
+    FolderResponse,
     GeneralAgentConfig,
     GeneralAgentConfigEnabledToolsItem,
     GeneralAgentRequest,
@@ -49,6 +51,7 @@ from .tools import ToolsDataFrameRequestColumnsItem
 from .version import __version__
 
 __all__ = [
+    "AssetNode",
     "AssetNotFoundError",
     "AthenaEnvironment",
     "BadRequestError",
@@ -69,6 +72,7 @@ __all__ = [
     "DriveAgentResponse",
     "FileChunkRequestOut",
     "FileTooLargeError",
+    "FolderResponse",
     "GeneralAgentConfig",
     "GeneralAgentConfigEnabledToolsItem",
     "GeneralAgentRequest",

athena/core/client_wrapper.py

@@ -17,7 +17,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "athena-intelligence",
-            "X-Fern-SDK-Version": "0.1.123",
+            "X-Fern-SDK-Version": "0.1.124",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

athena/tools/client.py

@@ -20,6 +20,7 @@ from ..types.data_frame_request_out import DataFrameRequestOut
 from ..types.data_frame_unknown_format_error import DataFrameUnknownFormatError
 from ..types.file_chunk_request_out import FileChunkRequestOut
 from ..types.file_too_large_error import FileTooLargeError
+from ..types.folder_response import FolderResponse
 from ..types.parent_folder_error import ParentFolderError
 from ..types.save_asset_request_out import SaveAssetRequestOut
 from .calendar.client import AsyncCalendarClient, CalendarClient
@@ -94,6 +95,68 @@ class ToolsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def list_contents(
+        self,
+        *,
+        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
+    ) -> FolderResponse:
+        """
+        List contents of a folder or entire workspace in a tree structure.
+
+        Parameters
+        ----------
+        folder_id : typing.Optional[str]
+
+        include_asset_details : typing.Optional[bool]
+
+        include_system_files : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        FolderResponse
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.tools.list_contents()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/tools/contents",
+            method="GET",
+            params={
+                "folder_id": folder_id,
+                "include_asset_details": include_asset_details,
+                "include_system_files": include_system_files,
+            },
+            request_options=request_options,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(FolderResponse, _response.json())  # type: ignore
+        if _response.status_code == 400:
+            raise BadRequestError(pydantic_v1.parse_obj_as(ParentFolderError, _response.json()))  # type: ignore
+        if _response.status_code == 401:
+            raise UnauthorizedError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+        if _response.status_code == 404:
+            raise NotFoundError(pydantic_v1.parse_obj_as(AssetNotFoundError, _response.json()))  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _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 data_frame(
         self,
         *,
@@ -347,6 +410,68 @@ class AsyncToolsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def list_contents(
+        self,
+        *,
+        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
+    ) -> FolderResponse:
+        """
+        List contents of a folder or entire workspace in a tree structure.
+
+        Parameters
+        ----------
+        folder_id : typing.Optional[str]
+
+        include_asset_details : typing.Optional[bool]
+
+        include_system_files : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        FolderResponse
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.tools.list_contents()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/tools/contents",
+            method="GET",
+            params={
+                "folder_id": folder_id,
+                "include_asset_details": include_asset_details,
+                "include_system_files": include_system_files,
+            },
+            request_options=request_options,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(FolderResponse, _response.json())  # type: ignore
+        if _response.status_code == 400:
+            raise BadRequestError(pydantic_v1.parse_obj_as(ParentFolderError, _response.json()))  # type: ignore
+        if _response.status_code == 401:
+            raise UnauthorizedError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+        if _response.status_code == 404:
+            raise NotFoundError(pydantic_v1.parse_obj_as(AssetNotFoundError, _response.json()))  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _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 data_frame(
         self,
         *,

athena/tools/structured_data_extractor/client.py

@@ -36,7 +36,7 @@ class StructuredDataExtractorClient:
         tl;dr:
 
         - pass a valid JSON schema in `json_schema`
-        - pass the page chunks as a list of `Chunk` objects, by default: {"type": "text", "content": "..."}
+        - pass the page chunks as a list of `Chunk` objects, by default: `{"type": "text", "content": "..."}`
         - leave all other fields as default
 
         Detailed configuration (only relevant for complex use cases):
@@ -45,9 +45,12 @@ class StructuredDataExtractorClient:
         where the asset is divided into chunks, the schema is extracted from each chunk,
         and the chunks are then reduced to a single structured data object.
 
-        In some applications, you may not want to: - map (if your input asset is small enough) - reduce (if your output object is large enough that it will overflow the output length;
-        if you're extracting a long list of entities; if youre )
-        to extract all instances of the schema).
+        In some applications, you may not want to:
+
+        - map (if your input asset is small enough)
+        - reduce (if your output object is large enough that it will overflow the output length;
+          if you're extracting a long list of entities; if youre )
+          to extract all instances of the schema).
 
         You can configure these behaviors with the `map` and `reduce` fields.
 
@@ -160,7 +163,7 @@ class AsyncStructuredDataExtractorClient:
         tl;dr:
 
         - pass a valid JSON schema in `json_schema`
-        - pass the page chunks as a list of `Chunk` objects, by default: {"type": "text", "content": "..."}
+        - pass the page chunks as a list of `Chunk` objects, by default: `{"type": "text", "content": "..."}`
         - leave all other fields as default
 
         Detailed configuration (only relevant for complex use cases):
@@ -169,9 +172,12 @@ class AsyncStructuredDataExtractorClient:
         where the asset is divided into chunks, the schema is extracted from each chunk,
         and the chunks are then reduced to a single structured data object.
 
-        In some applications, you may not want to: - map (if your input asset is small enough) - reduce (if your output object is large enough that it will overflow the output length;
-        if you're extracting a long list of entities; if youre )
-        to extract all instances of the schema).
+        In some applications, you may not want to:
+
+        - map (if your input asset is small enough)
+        - reduce (if your output object is large enough that it will overflow the output length;
+          if you're extracting a long list of entities; if youre )
+          to extract all instances of the schema).
 
         You can configure these behaviors with the `map` and `reduce` fields.
 

athena/types/__init__.py

@@ -1,5 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from .asset_node import AssetNode
 from .asset_not_found_error import AssetNotFoundError
 from .chunk import Chunk
 from .chunk_content_item import ChunkContentItem, ChunkContentItem_ImageUrl, ChunkContentItem_Text
@@ -15,6 +16,7 @@ from .document_chunk import DocumentChunk
 from .drive_agent_response import DriveAgentResponse
 from .file_chunk_request_out import FileChunkRequestOut
 from .file_too_large_error import FileTooLargeError
+from .folder_response import FolderResponse
 from .general_agent_config import GeneralAgentConfig
 from .general_agent_config_enabled_tools_item import GeneralAgentConfigEnabledToolsItem
 from .general_agent_request import GeneralAgentRequest
@@ -31,6 +33,7 @@ from .tool import Tool
 from .type import Type
 
 __all__ = [
+    "AssetNode",
     "AssetNotFoundError",
     "Chunk",
     "ChunkContentItem",
@@ -48,6 +51,7 @@ __all__ = [
     "DriveAgentResponse",
     "FileChunkRequestOut",
     "FileTooLargeError",
+    "FolderResponse",
     "GeneralAgentConfig",
     "GeneralAgentConfigEnabledToolsItem",
     "GeneralAgentRequest",

athena/types/asset_node.py

@@ -0,0 +1,42 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+
+
+class AssetNode(pydantic_v1.BaseModel):
+    """
+    Model representing a node in the folder tree.
+    """
+
+    children: typing.Optional[typing.Dict[str, typing.Optional[AssetNode]]] = None
+    id: str
+    media_type: str
+    name: str
+    type: str
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}
+
+
+AssetNode.update_forward_refs()

athena/types/folder_response.py

@@ -0,0 +1,35 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+from .asset_node import AssetNode
+
+
+class FolderResponse(pydantic_v1.BaseModel):
+    """
+    Combined response with tree data and visualization.
+    """
+
+    structure_tree_ascii: str
+    tree_data: typing.Dict[str, AssetNode]
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

{athena_intelligence-0.1.123.dist-info → athena_intelligence-0.1.124.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: athena-intelligence
-Version: 0.1.123
+Version: 0.1.124
 Summary: Athena Intelligence Python Library
 Requires-Python: >=3.8,<4.0
 Classifier: Intended Audience :: Developers

{athena_intelligence-0.1.123.dist-info → athena_intelligence-0.1.124.dist-info}/RECORD

@@ -1,4 +1,4 @@
-athena/__init__.py,sha256=-QY5Njk1mv2eVt9CMdJSaU_nMqoXzZ8d5kHjxKCs7jU,2464
+athena/__init__.py,sha256=Q5F_OwlFj6Y4zWtw4ncDPXp16HWBRrQn7Q4fJYBa5Pc,2538
 athena/agents/__init__.py,sha256=I6MO2O_hb6KLa8oDHbGNSAhcPE-dsrX6LMcAEhsg3PQ,160
 athena/agents/client.py,sha256=aI8rNhXBSVJ-hvjnIoCK9sKvHB0e95Zkn-3YpXOKFrY,6721
 athena/agents/drive/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
@@ -13,7 +13,7 @@ athena/base_client.py,sha256=-kVdOlIibBz48lxWratdQAzT7fTvZsORvOMF3KoPDPw,5647
 athena/client.py,sha256=4PUPrBPCMTFpHR1yuKVR5eC1AYBl_25SMf6ZH82JHB0,19039
 athena/core/__init__.py,sha256=UFXpYzcGxWQUucU1TkjOQ9mGWN3A5JohluOIWVYKU4I,973
 athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-athena/core/client_wrapper.py,sha256=QpWDM_UvxCir-D4qH9GDRZGMoDC2fW-UpSPKEHT3S0o,1806
+athena/core/client_wrapper.py,sha256=PsnlYk4Sr5_x1d3Z8hUp0g3i5pD1dqfmKXH-mHvU04Q,1806
 athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 athena/core/file.py,sha256=sy1RUGZ3aJYuw998bZytxxo6QdgKmlnlgBaMvwEKCGg,1480
 athena/core/http_client.py,sha256=Z4NuAsJD-51yqmoME17O5sxwx5orSp1wsnd6bPyKcgA,17768
@@ -39,16 +39,17 @@ athena/query/types/query_execute_request_database_asset_ids.py,sha256=aoVl5Xb34Q
 athena/tools/__init__.py,sha256=DREW2sa5Z-Rj8v1LQ-tNhflI-EO_oDc6NCmF5v0LWeU,288
 athena/tools/calendar/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 athena/tools/calendar/client.py,sha256=hKWzWyl1GwFG69oX3tektwNfy2sV5Lt6PRy9vTyPLOo,5283
-athena/tools/client.py,sha256=9ec2gnf3z_vhr3EqT_-ZksevTDtFP1jftPY4os0Ty3Q,21166
+athena/tools/client.py,sha256=-hjk99ABo74-qT8OMMqASl0c7oOkdhHSMFluBa9l5qA,25965
 athena/tools/email/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 athena/tools/email/client.py,sha256=epUkV5af3eilVgRR81SFZAf29JuhEWKMkdMuN6qDLUM,7593
 athena/tools/structured_data_extractor/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
-athena/tools/structured_data_extractor/client.py,sha256=0rEq5bftoWQJwCF9fcP3r8nUSY5hZLy_pOAucena9Go,11235
+athena/tools/structured_data_extractor/client.py,sha256=Wr2r_kU8WTRiKA2qfCfqvmllddj3MjT1Vqr27La8gO8,11281
 athena/tools/tasks/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 athena/tools/tasks/client.py,sha256=5kT6ulh2YDIbNYiv-knBjtF-ST7p0dUvZyrd7t5O61s,2975
 athena/tools/types/__init__.py,sha256=cA-ZQm6veQAP3_vKu9KkZpISsQqgTBN_Z--FGY1c2iA,197
 athena/tools/types/tools_data_frame_request_columns_item.py,sha256=GA1FUlTV_CfSc-KToTAwFf4Exl0rr4fsweVZupztjw0,138
-athena/types/__init__.py,sha256=-VA36LN5Q1SLk5IUgw159TBfTtTmvbgAOfaL-3TRY7A,2522
+athena/types/__init__.py,sha256=IbCXHjNzSdiLn2HrPsxj90DLzYDp63WJk4FsCVoymK0,2639
+athena/types/asset_node.py,sha256=CiqYxuYCXhOs9XAvBrUaVVMZpf8gdocVTAUBmUS5l1g,1375
 athena/types/asset_not_found_error.py,sha256=ZcgqRuzvO4Z8vVVxwtDB-QtKhpVIVV3hqQuJeUoOoJE,1121
 athena/types/chunk.py,sha256=M4O7Sj3EMvkXioQneuKbptr1n5XNGCU9fVxYR12XG9o,1340
 athena/types/chunk_content_item.py,sha256=2B1mTc0a4h7jyKRiYwfC573fM4xijhNEgfd_FI-myj4,2251
@@ -64,6 +65,7 @@ athena/types/document_chunk.py,sha256=deXiiMA_U5EabUh1Fg2AB4ElYuc5OsTnrU7JwLApJm
 athena/types/drive_agent_response.py,sha256=UMKF43e5WScH0a9ITuxjwGWzAzvXAl1OsfRVeXSyfjk,1218
 athena/types/file_chunk_request_out.py,sha256=Ju0I_UpSitjQ-XqSIRvvg2XA6QtfCLZClPq5PUqmPNg,1258
 athena/types/file_too_large_error.py,sha256=AinkrcgR7lcTILAD8RX0x48P3GlSoAh1OihxMvSvRuo,1120
+athena/types/folder_response.py,sha256=qq0hLRvfJFeXUz7Cc8oeyCabG73ac2H4lM_j0QW38YY,1280
 athena/types/general_agent_config.py,sha256=FaswWVsDTsL5Fs9Tlx4zSK1S8OrsFnzruEt7l72XlGA,1457
 athena/types/general_agent_config_enabled_tools_item.py,sha256=6gYaU7uIDJbgygtBKLdYL-VbPxxbEcxwRsT8VaW5vN8,165
 athena/types/general_agent_request.py,sha256=NnUVtz8U1VoA1SJapbp163Wf_inEQVeFCYWJvM4P-qI,1449
@@ -79,6 +81,6 @@ athena/types/text_content.py,sha256=uG2poNIkM6o7tFgf-eKzZk9kZHYImY3JdI-NkYiqWgU,
 athena/types/tool.py,sha256=6H2BFZiBgQOtYUAwSYBeGZKhwev17IEwnIjgmno6dZw,436
 athena/types/type.py,sha256=JaUIt4ogmO4XxCQ9c56fqKN5qANKkrnpuZGmdqOCIow,581
 athena/version.py,sha256=8aYAOJtVLaJLpRp6mTiEIhnl8gXA7yE0aDtZ-3mKQ4k,87
-athena_intelligence-0.1.123.dist-info/METADATA,sha256=Au7QjNP7MXi0igYTXg3ypVp_OHoNiFm-Q8c-E53dEGU,5274
-athena_intelligence-0.1.123.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.123.dist-info/RECORD,,
+athena_intelligence-0.1.124.dist-info/METADATA,sha256=lwtK__HdkgNj6iBeRlDN4JVY5z7XoZPv1ABXIwaBRT0,5274
+athena_intelligence-0.1.124.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.124.dist-info/RECORD,,