athena-intelligence 0.1.231__tar.gz → 0.1.242__tar.gz

{athena_intelligence-0.1.231 → athena_intelligence-0.1.242}/PKG-INFO

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

{athena_intelligence-0.1.231 → athena_intelligence-0.1.242}/pyproject.toml

@@ -3,7 +3,7 @@ name = "athena-intelligence"
 
 [tool.poetry]
 name = "athena-intelligence"
-version = "0.1.231"
+version = "0.1.242"
 description = "Athena Intelligence Python Library"
 readme = "README.md"
 authors = []

athena_intelligence-0.1.242/src/athena/__init__.py

@@ -0,0 +1,294 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+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",
+    "AopExecuteRequestIn",
+    "AopExecuteResponseOut",
+    "AssetContentRequestOut",
+    "AssetNode",
+    "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",
+    "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",
+    "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",
+    "assets",
+    "query",
+    "threads",
+    "tools",
+]

athena_intelligence-0.1.242/src/athena/agents/__init__.py

@@ -0,0 +1,39 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+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_intelligence-0.1.231 → athena_intelligence-0.1.242}/src/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_intelligence-0.1.231 → athena_intelligence-0.1.242}/src/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())

{athena_intelligence-0.1.231 → athena_intelligence-0.1.242}/src/athena/base_client.py

@@ -1,16 +1,20 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
+
 import typing
 
 import httpx
-from .agents.client import AgentsClient, AsyncAgentsClient
-from .aop.client import AopClient, AsyncAopClient
-from .assets.client import AssetsClient, AsyncAssetsClient
 from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from .environment import AthenaEnvironment
-from .query.client import AsyncQueryClient, QueryClient
-from .threads.client import AsyncThreadsClient, ThreadsClient
-from .tools.client import AsyncToolsClient, ToolsClient
+
+if typing.TYPE_CHECKING:
+    from .agents.client import AgentsClient, AsyncAgentsClient
+    from .aop.client import AopClient, AsyncAopClient
+    from .assets.client import AssetsClient, AsyncAssetsClient
+    from .query.client import AsyncQueryClient, QueryClient
+    from .threads.client import AsyncThreadsClient, ThreadsClient
+    from .tools.client import AsyncToolsClient, ToolsClient
 
 
 class BaseAthena:
@@ -78,12 +82,60 @@ class BaseAthena:
             else httpx.Client(timeout=_defaulted_timeout),
             timeout=_defaulted_timeout,
         )
-        self.agents = AgentsClient(client_wrapper=self._client_wrapper)
-        self.aop = AopClient(client_wrapper=self._client_wrapper)
-        self.assets = AssetsClient(client_wrapper=self._client_wrapper)
-        self.query = QueryClient(client_wrapper=self._client_wrapper)
-        self.threads = ThreadsClient(client_wrapper=self._client_wrapper)
-        self.tools = ToolsClient(client_wrapper=self._client_wrapper)
+        self._agents: typing.Optional[AgentsClient] = None
+        self._aop: typing.Optional[AopClient] = None
+        self._assets: typing.Optional[AssetsClient] = None
+        self._query: typing.Optional[QueryClient] = None
+        self._threads: typing.Optional[ThreadsClient] = None
+        self._tools: typing.Optional[ToolsClient] = None
+
+    @property
+    def agents(self):
+        if self._agents is None:
+            from .agents.client import AgentsClient  # noqa: E402
+
+            self._agents = AgentsClient(client_wrapper=self._client_wrapper)
+        return self._agents
+
+    @property
+    def aop(self):
+        if self._aop is None:
+            from .aop.client import AopClient  # noqa: E402
+
+            self._aop = AopClient(client_wrapper=self._client_wrapper)
+        return self._aop
+
+    @property
+    def assets(self):
+        if self._assets is None:
+            from .assets.client import AssetsClient  # noqa: E402
+
+            self._assets = AssetsClient(client_wrapper=self._client_wrapper)
+        return self._assets
+
+    @property
+    def query(self):
+        if self._query is None:
+            from .query.client import QueryClient  # noqa: E402
+
+            self._query = QueryClient(client_wrapper=self._client_wrapper)
+        return self._query
+
+    @property
+    def threads(self):
+        if self._threads is None:
+            from .threads.client import ThreadsClient  # noqa: E402
+
+            self._threads = ThreadsClient(client_wrapper=self._client_wrapper)
+        return self._threads
+
+    @property
+    def tools(self):
+        if self._tools is None:
+            from .tools.client import ToolsClient  # noqa: E402
+
+            self._tools = ToolsClient(client_wrapper=self._client_wrapper)
+        return self._tools
 
 
 class AsyncBaseAthena:
@@ -151,12 +203,60 @@ class AsyncBaseAthena:
             else httpx.AsyncClient(timeout=_defaulted_timeout),
             timeout=_defaulted_timeout,
         )
-        self.agents = AsyncAgentsClient(client_wrapper=self._client_wrapper)
-        self.aop = AsyncAopClient(client_wrapper=self._client_wrapper)
-        self.assets = AsyncAssetsClient(client_wrapper=self._client_wrapper)
-        self.query = AsyncQueryClient(client_wrapper=self._client_wrapper)
-        self.threads = AsyncThreadsClient(client_wrapper=self._client_wrapper)
-        self.tools = AsyncToolsClient(client_wrapper=self._client_wrapper)
+        self._agents: typing.Optional[AsyncAgentsClient] = None
+        self._aop: typing.Optional[AsyncAopClient] = None
+        self._assets: typing.Optional[AsyncAssetsClient] = None
+        self._query: typing.Optional[AsyncQueryClient] = None
+        self._threads: typing.Optional[AsyncThreadsClient] = None
+        self._tools: typing.Optional[AsyncToolsClient] = None
+
+    @property
+    def agents(self):
+        if self._agents is None:
+            from .agents.client import AsyncAgentsClient  # noqa: E402
+
+            self._agents = AsyncAgentsClient(client_wrapper=self._client_wrapper)
+        return self._agents
+
+    @property
+    def aop(self):
+        if self._aop is None:
+            from .aop.client import AsyncAopClient  # noqa: E402
+
+            self._aop = AsyncAopClient(client_wrapper=self._client_wrapper)
+        return self._aop
+
+    @property
+    def assets(self):
+        if self._assets is None:
+            from .assets.client import AsyncAssetsClient  # noqa: E402
+
+            self._assets = AsyncAssetsClient(client_wrapper=self._client_wrapper)
+        return self._assets
+
+    @property
+    def query(self):
+        if self._query is None:
+            from .query.client import AsyncQueryClient  # noqa: E402
+
+            self._query = AsyncQueryClient(client_wrapper=self._client_wrapper)
+        return self._query
+
+    @property
+    def threads(self):
+        if self._threads is None:
+            from .threads.client import AsyncThreadsClient  # noqa: E402
+
+            self._threads = AsyncThreadsClient(client_wrapper=self._client_wrapper)
+        return self._threads
+
+    @property
+    def tools(self):
+        if self._tools is None:
+            from .tools.client import AsyncToolsClient  # noqa: E402
+
+            self._tools = AsyncToolsClient(client_wrapper=self._client_wrapper)
+        return self._tools
 
 
 def _get_base_url(*, base_url: typing.Optional[str] = None, environment: AthenaEnvironment) -> str: