athena-intelligence 0.1.202__tar.gz → 0.1.338__tar.gz

{athena_intelligence-0.1.202 → athena_intelligence-0.1.338}/PKG-INFO

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

{athena_intelligence-0.1.202 → athena_intelligence-0.1.338}/pyproject.toml

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

athena_intelligence-0.1.338/src/athena/__init__.py

@@ -0,0 +1,297 @@
+# 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,
+        BorderModel,
+        BorderStyle,
+        BordersModel,
+        CellFormat,
+        CellFormatHorizontalAlignment,
+        CellFormatVerticalAlignment,
+        Chunk,
+        ChunkContentItem,
+        ChunkContentItem_ImageUrl,
+        ChunkContentItem_Text,
+        ChunkResult,
+        ChunkResultChunkId,
+        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,
+        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",
+    "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",
+    "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",
+    "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",
+    "BadRequestError",
+    "BorderModel",
+    "BorderStyle",
+    "BordersModel",
+    "CellFormat",
+    "CellFormatHorizontalAlignment",
+    "CellFormatVerticalAlignment",
+    "Chunk",
+    "ChunkContentItem",
+    "ChunkContentItem_ImageUrl",
+    "ChunkContentItem_Text",
+    "ChunkResult",
+    "ChunkResultChunkId",
+    "Content",
+    "ContentTooLargeError",
+    "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",
+    "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.338/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.202 → athena_intelligence-0.1.338}/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.338/src/athena/aop/client.py

@@ -0,0 +1,201 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.request_options import RequestOptions
+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 .raw_client import AsyncRawAopClient, RawAopClient
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class AopClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._raw_client = RawAopClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawAopClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawAopClient
+        """
+        return self._raw_client
+
+    def execute(
+        self, *, request: AopExecuteRequestIn, request_options: typing.Optional[RequestOptions] = None
+    ) -> AopExecuteResponseOut:
+        """
+        ⚠️ 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
+        ----------
+        request : AopExecuteRequestIn
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AopExecuteResponseOut
+            AOP execution started successfully
+
+        Examples
+        --------
+        from athena import AopExecuteRequestIn, Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.aop.execute(
+            request=AopExecuteRequestIn(
+                asset_id="asset_9249292-d118-42d3-95b4-00eccfe0754f",
+                user_inputs={"company": "Acme Corp", "quarter": "Q1 2024"},
+            ),
+        )
+        """
+        _response = self._raw_client.execute(request=request, request_options=request_options)
+        return _response.data
+
+    def execute_async(
+        self, *, request: AopExecuteRequestIn, request_options: typing.Optional[RequestOptions] = None
+    ) -> AopAsyncExecuteResponseOut:
+        """
+        Start execution of an Agent Operating Procedure (AOP) asset asynchronously. Returns immediately with a thread_id for tracking execution progress without waiting for completion.
+
+        Parameters
+        ----------
+        request : AopExecuteRequestIn
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AopAsyncExecuteResponseOut
+            AOP execution started successfully
+
+        Examples
+        --------
+        from athena import AopExecuteRequestIn, Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.aop.execute_async(
+            request=AopExecuteRequestIn(
+                asset_id="asset_9249292-d118-42d3-95b4-00eccfe0754f",
+                user_inputs={"company": "Acme Corp", "quarter": "Q1 2024"},
+            ),
+        )
+        """
+        _response = self._raw_client.execute_async(request=request, request_options=request_options)
+        return _response.data
+
+
+class AsyncAopClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawAopClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawAopClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawAopClient
+        """
+        return self._raw_client
+
+    async def execute(
+        self, *, request: AopExecuteRequestIn, request_options: typing.Optional[RequestOptions] = None
+    ) -> AopExecuteResponseOut:
+        """
+        ⚠️ 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
+        ----------
+        request : AopExecuteRequestIn
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AopExecuteResponseOut
+            AOP execution started successfully
+
+        Examples
+        --------
+        import asyncio
+
+        from athena import AopExecuteRequestIn, AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.aop.execute(
+                request=AopExecuteRequestIn(
+                    asset_id="asset_9249292-d118-42d3-95b4-00eccfe0754f",
+                    user_inputs={"company": "Acme Corp", "quarter": "Q1 2024"},
+                ),
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.execute(request=request, request_options=request_options)
+        return _response.data
+
+    async def execute_async(
+        self, *, request: AopExecuteRequestIn, request_options: typing.Optional[RequestOptions] = None
+    ) -> AopAsyncExecuteResponseOut:
+        """
+        Start execution of an Agent Operating Procedure (AOP) asset asynchronously. Returns immediately with a thread_id for tracking execution progress without waiting for completion.
+
+        Parameters
+        ----------
+        request : AopExecuteRequestIn
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AopAsyncExecuteResponseOut
+            AOP execution started successfully
+
+        Examples
+        --------
+        import asyncio
+
+        from athena import AopExecuteRequestIn, AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.aop.execute_async(
+                request=AopExecuteRequestIn(
+                    asset_id="asset_9249292-d118-42d3-95b4-00eccfe0754f",
+                    user_inputs={"company": "Acme Corp", "quarter": "Q1 2024"},
+                ),
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.execute_async(request=request, request_options=request_options)
+        return _response.data