athena-intelligence 0.1.230__py3-none-any.whl → 0.1.242__py3-none-any.whl

athena/__init__.py

@@ -2,74 +2,206 @@
 
 # isort: skip_file
 
-from .types import (
-    AopAsyncExecuteResponseOut,
-    AopExecuteRequestIn,
-    AopExecuteResponseOut,
-    AssetContentRequestOut,
-    AssetNode,
-    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,
-    TableRowData,
-    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,
+        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,
+        NumberFormatModel,
+        NumberFormatType,
+        PaginatedAssetsOut,
+        PromptMessage,
+        PublicAssetOut,
+        ResearchAgentResponse,
+        SaveAssetRequestOut,
+        Sheet,
+        SheetOperationResponse,
+        SqlAgentResponse,
+        StructuredDataExtractorResponse,
+        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",
+    "CreateNewSheetTabResponse": ".types",
+    "CustomAgentResponse": ".types",
+    "DataFrameRequestOut": ".types",
+    "DataFrameRequestOutColumnsItem": ".types",
+    "DataFrameRequestOutDataItemItem": ".types",
+    "DataFrameRequestOutIndexItem": ".types",
+    "DataFrameUnknownFormatError": ".types",
+    "DocumentChunk": ".types",
+    "DriveAgentResponse": ".types",
+    "FileChunkRequestOut": ".types",
+    "FileTooLargeError": ".types",
+    "FolderResponse": ".types",
+    "GeneralAgentConfig": ".types",
+    "GeneralAgentConfigEnabledToolsItem": ".types",
+    "GeneralAgentRequest": ".types",
+    "GeneralAgentResponse": ".types",
+    "GeneralAgentResponseMessage": ".types",
+    "GeneralAgentResponseMessageKwargs": ".types",
+    "GetTableResponse": ".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",
+    "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",
@@ -81,16 +213,26 @@ __all__ = [
     "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",
     "CreateNewSheetTabResponse",
     "CustomAgentResponse",
     "DataFrameRequestOut",
@@ -118,23 +260,30 @@ __all__ = [
     "InputMessageContentItem_Text",
     "InternalServerError",
     "NotFoundError",
+    "NumberFormatModel",
+    "NumberFormatType",
     "PaginatedAssetsOut",
     "PromptMessage",
     "PublicAssetOut",
     "QueryExecuteRequestDatabaseAssetIds",
     "ResearchAgentResponse",
     "SaveAssetRequestOut",
+    "Sheet",
     "SheetOperationResponse",
     "SqlAgentResponse",
     "StructuredDataExtractorResponse",
     "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/assets/client.py

@@ -64,7 +64,12 @@ 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
@@ -133,7 +138,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())