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

athena/__init__.py

@@ -2,74 +2,214 @@
 
 # isort: skip_file
 
-from .types import (
-    AopAsyncExecuteResponseOut,
-    AopExecuteRequestIn,
-    AopExecuteResponseOut,
-    AssetContentRequestOut,
-    AssetNode,
-    AssetNotFoundError,
-    AssetScreenshotResponseOut,
-    Chunk,
-    ChunkContentItem,
-    ChunkContentItem_ImageUrl,
-    ChunkContentItem_Text,
-    ChunkResult,
-    ChunkResultChunkId,
-    Content,
-    ConversationAssetInfo,
-    CreateNewSheetTabResponse,
-    CustomAgentResponse,
-    DataFrameRequestOut,
-    DataFrameRequestOutColumnsItem,
-    DataFrameRequestOutDataItemItem,
-    DataFrameRequestOutIndexItem,
-    DataFrameUnknownFormatError,
-    DocumentChunk,
-    DriveAgentResponse,
-    FileChunkRequestOut,
-    FileTooLargeError,
-    FolderResponse,
-    GeneralAgentConfig,
-    GeneralAgentConfigEnabledToolsItem,
-    GeneralAgentRequest,
-    GeneralAgentResponse,
-    GeneralAgentResponseMessage,
-    GeneralAgentResponseMessageKwargs,
-    GetTableResponse,
-    Id,
-    ImageUrlContent,
-    InputMessage,
-    InputMessageContentItem,
-    InputMessageContentItem_ImageUrl,
-    InputMessageContentItem_Text,
-    PaginatedAssetsOut,
-    PromptMessage,
-    PublicAssetOut,
-    ResearchAgentResponse,
-    SaveAssetRequestOut,
-    SheetOperationResponse,
-    SqlAgentResponse,
-    StructuredDataExtractorResponse,
-    TextContent,
-    ThreadStatusResponseOut,
-    Type,
-)
-from .errors import (
-    BadRequestError,
-    ContentTooLargeError,
-    InternalServerError,
-    NotFoundError,
-    UnauthorizedError,
-    UnprocessableEntityError,
-    UnsupportedMediaTypeError,
-)
-from . import agents, aop, assets, query, threads, tools
-from .client import AsyncAthena, Athena
-from .environment import AthenaEnvironment
-from .query import QueryExecuteRequestDatabaseAssetIds
-from .tools import ToolsDataFrameRequestColumnsItem
-from .version import __version__
+import typing
+from importlib import import_module
+
+if typing.TYPE_CHECKING:
+    from .types import (
+        AopAsyncExecuteResponseOut,
+        AopExecuteRequestIn,
+        AopExecuteResponseOut,
+        AssetContentRequestOut,
+        AssetNode,
+        AssetScreenshotResponseOut,
+        Backgroundcolor,
+        BorderModel,
+        BorderStyle,
+        BordersModel,
+        CellFormat,
+        CellFormatHorizontalAlignment,
+        CellFormatVerticalAlignment,
+        Chunk,
+        ChunkContentItem,
+        ChunkContentItem_ImageUrl,
+        ChunkContentItem_Text,
+        ChunkResult,
+        ChunkResultChunkId,
+        Color,
+        Content,
+        ConversationAssetInfo,
+        ConversationMessage,
+        ConversationResult,
+        CreatableAssetType,
+        CreateAssetResponseOut,
+        CreateNewSheetTabResponse,
+        CreateProjectResponseOut,
+        CustomAgentResponse,
+        DataFrameRequestOut,
+        DataFrameRequestOutColumnsItem,
+        DataFrameRequestOutDataItemItem,
+        DataFrameRequestOutIndexItem,
+        DataFrameUnknownFormatError,
+        DimensionProperties,
+        DriveAgentResponse,
+        FileTooLargeError,
+        FolderResponse,
+        GeneralAgentConfig,
+        GeneralAgentConfigEnabledToolsItem,
+        GeneralAgentRequest,
+        GeneralAgentResponse,
+        GeneralAgentResponseMessage,
+        GeneralAgentResponseMessageKwargs,
+        GetTableResponse,
+        GridRange,
+        Id,
+        ImageUrlContent,
+        InputMessage,
+        InputMessageContentItem,
+        InputMessageContentItem_ImageUrl,
+        InputMessageContentItem_Text,
+        NumberFormatModel,
+        NumberFormatType,
+        PaginatedAssetsOut,
+        PromptMessage,
+        PublicAssetOut,
+        ResearchAgentResponse,
+        SaveAssetRequestOut,
+        Sheet,
+        SheetOperationResponse,
+        SqlAgentResponse,
+        StructuredDataExtractorResponse,
+        Tabcolor,
+        TableRowData,
+        TextContent,
+        TextFormatModel,
+        Textrotation,
+        ThemeColor,
+        ThreadStatusResponseOut,
+        Type,
+        WrapStrategy,
+    )
+    from .errors import (
+        BadRequestError,
+        ContentTooLargeError,
+        InternalServerError,
+        NotFoundError,
+        UnauthorizedError,
+        UnprocessableEntityError,
+        UnsupportedMediaTypeError,
+    )
+    from . import agents, aop, assets, query, threads, tools
+    from .client import AsyncAthena, Athena
+    from .environment import AthenaEnvironment
+    from .query import QueryExecuteRequestDatabaseAssetIds
+    from .tools import ToolsDataFrameRequestColumnsItem
+    from .version import __version__
+_dynamic_imports: typing.Dict[str, str] = {
+    "AopAsyncExecuteResponseOut": ".types",
+    "AopExecuteRequestIn": ".types",
+    "AopExecuteResponseOut": ".types",
+    "AssetContentRequestOut": ".types",
+    "AssetNode": ".types",
+    "AssetScreenshotResponseOut": ".types",
+    "AsyncAthena": ".client",
+    "Athena": ".client",
+    "AthenaEnvironment": ".environment",
+    "Backgroundcolor": ".types",
+    "BadRequestError": ".errors",
+    "BorderModel": ".types",
+    "BorderStyle": ".types",
+    "BordersModel": ".types",
+    "CellFormat": ".types",
+    "CellFormatHorizontalAlignment": ".types",
+    "CellFormatVerticalAlignment": ".types",
+    "Chunk": ".types",
+    "ChunkContentItem": ".types",
+    "ChunkContentItem_ImageUrl": ".types",
+    "ChunkContentItem_Text": ".types",
+    "ChunkResult": ".types",
+    "ChunkResultChunkId": ".types",
+    "Color": ".types",
+    "Content": ".types",
+    "ContentTooLargeError": ".errors",
+    "ConversationAssetInfo": ".types",
+    "ConversationMessage": ".types",
+    "ConversationResult": ".types",
+    "CreatableAssetType": ".types",
+    "CreateAssetResponseOut": ".types",
+    "CreateNewSheetTabResponse": ".types",
+    "CreateProjectResponseOut": ".types",
+    "CustomAgentResponse": ".types",
+    "DataFrameRequestOut": ".types",
+    "DataFrameRequestOutColumnsItem": ".types",
+    "DataFrameRequestOutDataItemItem": ".types",
+    "DataFrameRequestOutIndexItem": ".types",
+    "DataFrameUnknownFormatError": ".types",
+    "DimensionProperties": ".types",
+    "DriveAgentResponse": ".types",
+    "FileTooLargeError": ".types",
+    "FolderResponse": ".types",
+    "GeneralAgentConfig": ".types",
+    "GeneralAgentConfigEnabledToolsItem": ".types",
+    "GeneralAgentRequest": ".types",
+    "GeneralAgentResponse": ".types",
+    "GeneralAgentResponseMessage": ".types",
+    "GeneralAgentResponseMessageKwargs": ".types",
+    "GetTableResponse": ".types",
+    "GridRange": ".types",
+    "Id": ".types",
+    "ImageUrlContent": ".types",
+    "InputMessage": ".types",
+    "InputMessageContentItem": ".types",
+    "InputMessageContentItem_ImageUrl": ".types",
+    "InputMessageContentItem_Text": ".types",
+    "InternalServerError": ".errors",
+    "NotFoundError": ".errors",
+    "NumberFormatModel": ".types",
+    "NumberFormatType": ".types",
+    "PaginatedAssetsOut": ".types",
+    "PromptMessage": ".types",
+    "PublicAssetOut": ".types",
+    "QueryExecuteRequestDatabaseAssetIds": ".query",
+    "ResearchAgentResponse": ".types",
+    "SaveAssetRequestOut": ".types",
+    "Sheet": ".types",
+    "SheetOperationResponse": ".types",
+    "SqlAgentResponse": ".types",
+    "StructuredDataExtractorResponse": ".types",
+    "Tabcolor": ".types",
+    "TableRowData": ".types",
+    "TextContent": ".types",
+    "TextFormatModel": ".types",
+    "Textrotation": ".types",
+    "ThemeColor": ".types",
+    "ThreadStatusResponseOut": ".types",
+    "ToolsDataFrameRequestColumnsItem": ".tools",
+    "Type": ".types",
+    "UnauthorizedError": ".errors",
+    "UnprocessableEntityError": ".errors",
+    "UnsupportedMediaTypeError": ".errors",
+    "WrapStrategy": ".types",
+    "__version__": ".version",
+    "agents": ".agents",
+    "aop": ".aop",
+    "assets": ".assets",
+    "query": ".query",
+    "threads": ".threads",
+    "tools": ".tools",
+}
+
+
+def __getattr__(attr_name: str) -> typing.Any:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(f"No {attr_name} found in _dynamic_imports for module name -> {__name__}")
+    try:
+        module = import_module(module_name, __package__)
+        if module_name == f".{attr_name}":
+            return module
+        else:
+            return getattr(module, attr_name)
+    except ImportError as e:
+        raise ImportError(f"Failed to import {attr_name} from {module_name}: {e}") from e
+    except AttributeError as e:
+        raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)
+
 
 __all__ = [
     "AopAsyncExecuteResponseOut",
@@ -77,31 +217,42 @@ __all__ = [
     "AopExecuteResponseOut",
     "AssetContentRequestOut",
     "AssetNode",
-    "AssetNotFoundError",
     "AssetScreenshotResponseOut",
     "AsyncAthena",
     "Athena",
     "AthenaEnvironment",
+    "Backgroundcolor",
     "BadRequestError",
+    "BorderModel",
+    "BorderStyle",
+    "BordersModel",
+    "CellFormat",
+    "CellFormatHorizontalAlignment",
+    "CellFormatVerticalAlignment",
     "Chunk",
     "ChunkContentItem",
     "ChunkContentItem_ImageUrl",
     "ChunkContentItem_Text",
     "ChunkResult",
     "ChunkResultChunkId",
+    "Color",
     "Content",
     "ContentTooLargeError",
     "ConversationAssetInfo",
+    "ConversationMessage",
+    "ConversationResult",
+    "CreatableAssetType",
+    "CreateAssetResponseOut",
     "CreateNewSheetTabResponse",
+    "CreateProjectResponseOut",
     "CustomAgentResponse",
     "DataFrameRequestOut",
     "DataFrameRequestOutColumnsItem",
     "DataFrameRequestOutDataItemItem",
     "DataFrameRequestOutIndexItem",
     "DataFrameUnknownFormatError",
-    "DocumentChunk",
+    "DimensionProperties",
     "DriveAgentResponse",
-    "FileChunkRequestOut",
     "FileTooLargeError",
     "FolderResponse",
     "GeneralAgentConfig",
@@ -111,6 +262,7 @@ __all__ = [
     "GeneralAgentResponseMessage",
     "GeneralAgentResponseMessageKwargs",
     "GetTableResponse",
+    "GridRange",
     "Id",
     "ImageUrlContent",
     "InputMessage",
@@ -119,22 +271,31 @@ __all__ = [
     "InputMessageContentItem_Text",
     "InternalServerError",
     "NotFoundError",
+    "NumberFormatModel",
+    "NumberFormatType",
     "PaginatedAssetsOut",
     "PromptMessage",
     "PublicAssetOut",
     "QueryExecuteRequestDatabaseAssetIds",
     "ResearchAgentResponse",
     "SaveAssetRequestOut",
+    "Sheet",
     "SheetOperationResponse",
     "SqlAgentResponse",
     "StructuredDataExtractorResponse",
+    "Tabcolor",
+    "TableRowData",
     "TextContent",
+    "TextFormatModel",
+    "Textrotation",
+    "ThemeColor",
     "ThreadStatusResponseOut",
     "ToolsDataFrameRequestColumnsItem",
     "Type",
     "UnauthorizedError",
     "UnprocessableEntityError",
     "UnsupportedMediaTypeError",
+    "WrapStrategy",
     "__version__",
     "agents",
     "aop",

athena/agents/__init__.py

@@ -2,6 +2,38 @@
 
 # isort: skip_file
 
-from . import drive, general, research, sql
+import typing
+from importlib import import_module
+
+if typing.TYPE_CHECKING:
+    from . import drive, general, research, sql
+_dynamic_imports: typing.Dict[str, str] = {
+    "drive": ".drive",
+    "general": ".general",
+    "research": ".research",
+    "sql": ".sql",
+}
+
+
+def __getattr__(attr_name: str) -> typing.Any:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(f"No {attr_name} found in _dynamic_imports for module name -> {__name__}")
+    try:
+        module = import_module(module_name, __package__)
+        if module_name == f".{attr_name}":
+            return module
+        else:
+            return getattr(module, attr_name)
+    except ImportError as e:
+        raise ImportError(f"Failed to import {attr_name} from {module_name}: {e}") from e
+    except AttributeError as e:
+        raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)
+
 
 __all__ = ["drive", "general", "research", "sql"]

athena/agents/client.py

@@ -1,16 +1,19 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
+
 import typing
 
 from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
 from ..types.custom_agent_response import CustomAgentResponse
-from .drive.client import AsyncDriveClient, DriveClient
-from .general.client import AsyncGeneralClient, GeneralClient
 from .raw_client import AsyncRawAgentsClient, RawAgentsClient
-from .research.client import AsyncResearchClient, ResearchClient
-from .sql.client import AsyncSqlClient, SqlClient
 
+if typing.TYPE_CHECKING:
+    from .drive.client import AsyncDriveClient, DriveClient
+    from .general.client import AsyncGeneralClient, GeneralClient
+    from .research.client import AsyncResearchClient, ResearchClient
+    from .sql.client import AsyncSqlClient, SqlClient
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
 
@@ -18,13 +21,11 @@ OMIT = typing.cast(typing.Any, ...)
 class AgentsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._raw_client = RawAgentsClient(client_wrapper=client_wrapper)
-        self.drive = DriveClient(client_wrapper=client_wrapper)
-
-        self.general = GeneralClient(client_wrapper=client_wrapper)
-
-        self.research = ResearchClient(client_wrapper=client_wrapper)
-
-        self.sql = SqlClient(client_wrapper=client_wrapper)
+        self._client_wrapper = client_wrapper
+        self._drive: typing.Optional[DriveClient] = None
+        self._general: typing.Optional[GeneralClient] = None
+        self._research: typing.Optional[ResearchClient] = None
+        self._sql: typing.Optional[SqlClient] = None
 
     @property
     def with_raw_response(self) -> RawAgentsClient:
@@ -91,17 +92,47 @@ class AgentsClient:
         )
         return _response.data
 
+    @property
+    def drive(self):
+        if self._drive is None:
+            from .drive.client import DriveClient  # noqa: E402
 
-class AsyncAgentsClient:
-    def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._raw_client = AsyncRawAgentsClient(client_wrapper=client_wrapper)
-        self.drive = AsyncDriveClient(client_wrapper=client_wrapper)
+            self._drive = DriveClient(client_wrapper=self._client_wrapper)
+        return self._drive
 
-        self.general = AsyncGeneralClient(client_wrapper=client_wrapper)
+    @property
+    def general(self):
+        if self._general is None:
+            from .general.client import GeneralClient  # noqa: E402
+
+            self._general = GeneralClient(client_wrapper=self._client_wrapper)
+        return self._general
 
-        self.research = AsyncResearchClient(client_wrapper=client_wrapper)
+    @property
+    def research(self):
+        if self._research is None:
+            from .research.client import ResearchClient  # noqa: E402
 
-        self.sql = AsyncSqlClient(client_wrapper=client_wrapper)
+            self._research = ResearchClient(client_wrapper=self._client_wrapper)
+        return self._research
+
+    @property
+    def sql(self):
+        if self._sql is None:
+            from .sql.client import SqlClient  # noqa: E402
+
+            self._sql = SqlClient(client_wrapper=self._client_wrapper)
+        return self._sql
+
+
+class AsyncAgentsClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawAgentsClient(client_wrapper=client_wrapper)
+        self._client_wrapper = client_wrapper
+        self._drive: typing.Optional[AsyncDriveClient] = None
+        self._general: typing.Optional[AsyncGeneralClient] = None
+        self._research: typing.Optional[AsyncResearchClient] = None
+        self._sql: typing.Optional[AsyncSqlClient] = None
 
     @property
     def with_raw_response(self) -> AsyncRawAgentsClient:
@@ -175,3 +206,35 @@ class AsyncAgentsClient:
             agent_id, config=config, messages=messages, request_options=request_options
         )
         return _response.data
+
+    @property
+    def drive(self):
+        if self._drive is None:
+            from .drive.client import AsyncDriveClient  # noqa: E402
+
+            self._drive = AsyncDriveClient(client_wrapper=self._client_wrapper)
+        return self._drive
+
+    @property
+    def general(self):
+        if self._general is None:
+            from .general.client import AsyncGeneralClient  # noqa: E402
+
+            self._general = AsyncGeneralClient(client_wrapper=self._client_wrapper)
+        return self._general
+
+    @property
+    def research(self):
+        if self._research is None:
+            from .research.client import AsyncResearchClient  # noqa: E402
+
+            self._research = AsyncResearchClient(client_wrapper=self._client_wrapper)
+        return self._research
+
+    @property
+    def sql(self):
+        if self._sql is None:
+            from .sql.client import AsyncSqlClient  # noqa: E402
+
+            self._sql = AsyncSqlClient(client_wrapper=self._client_wrapper)
+        return self._sql

athena/aop/client.py

@@ -32,7 +32,7 @@ class AopClient:
         self, *, request: AopExecuteRequestIn, request_options: typing.Optional[RequestOptions] = None
     ) -> AopExecuteResponseOut:
         """
-        Execute an existing Agent Operating Procedure (AOP) asset with optional user inputs. AOPs are pre-configured AI workflows that can perform complex tasks like research, analysis, and content generation.
+        ⚠️ DEPRECATED: This endpoint is deprecated. Please use /aop/execute-async instead for better performance and reliability. Execute an existing Agent Operating Procedure (AOP) asset with optional user inputs. AOPs are pre-configured AI workflows that can perform complex tasks like research, analysis, and content generation.
 
         Parameters
         ----------
@@ -118,7 +118,7 @@ class AsyncAopClient:
         self, *, request: AopExecuteRequestIn, request_options: typing.Optional[RequestOptions] = None
     ) -> AopExecuteResponseOut:
         """
-        Execute an existing Agent Operating Procedure (AOP) asset with optional user inputs. AOPs are pre-configured AI workflows that can perform complex tasks like research, analysis, and content generation.
+        ⚠️ DEPRECATED: This endpoint is deprecated. Please use /aop/execute-async instead for better performance and reliability. Execute an existing Agent Operating Procedure (AOP) asset with optional user inputs. AOPs are pre-configured AI workflows that can perform complex tasks like research, analysis, and content generation.
 
         Parameters
         ----------

athena/aop/raw_client.py

@@ -17,7 +17,6 @@ from ..errors.unprocessable_entity_error import UnprocessableEntityError
 from ..types.aop_async_execute_response_out import AopAsyncExecuteResponseOut
 from ..types.aop_execute_request_in import AopExecuteRequestIn
 from ..types.aop_execute_response_out import AopExecuteResponseOut
-from ..types.asset_not_found_error import AssetNotFoundError
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -31,7 +30,7 @@ class RawAopClient:
         self, *, request: AopExecuteRequestIn, request_options: typing.Optional[RequestOptions] = None
     ) -> HttpResponse[AopExecuteResponseOut]:
         """
-        Execute an existing Agent Operating Procedure (AOP) asset with optional user inputs. AOPs are pre-configured AI workflows that can perform complex tasks like research, analysis, and content generation.
+        ⚠️ DEPRECATED: This endpoint is deprecated. Please use /aop/execute-async instead for better performance and reliability. Execute an existing Agent Operating Procedure (AOP) asset with optional user inputs. AOPs are pre-configured AI workflows that can perform complex tasks like research, analysis, and content generation.
 
         Parameters
         ----------
@@ -93,9 +92,9 @@ class RawAopClient:
                 raise NotFoundError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        AssetNotFoundError,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=AssetNotFoundError,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -193,9 +192,9 @@ class RawAopClient:
                 raise NotFoundError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        AssetNotFoundError,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=AssetNotFoundError,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -236,7 +235,7 @@ class AsyncRawAopClient:
         self, *, request: AopExecuteRequestIn, request_options: typing.Optional[RequestOptions] = None
     ) -> AsyncHttpResponse[AopExecuteResponseOut]:
         """
-        Execute an existing Agent Operating Procedure (AOP) asset with optional user inputs. AOPs are pre-configured AI workflows that can perform complex tasks like research, analysis, and content generation.
+        ⚠️ DEPRECATED: This endpoint is deprecated. Please use /aop/execute-async instead for better performance and reliability. Execute an existing Agent Operating Procedure (AOP) asset with optional user inputs. AOPs are pre-configured AI workflows that can perform complex tasks like research, analysis, and content generation.
 
         Parameters
         ----------
@@ -298,9 +297,9 @@ class AsyncRawAopClient:
                 raise NotFoundError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        AssetNotFoundError,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=AssetNotFoundError,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -398,9 +397,9 @@ class AsyncRawAopClient:
                 raise NotFoundError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        AssetNotFoundError,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=AssetNotFoundError,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),