athena-intelligence 0.1.119__tar.gz → 0.1.272__tar.gz

{athena_intelligence-0.1.119 → athena_intelligence-0.1.272}/PKG-INFO

@@ -1,8 +1,8 @@
 Metadata-Version: 2.1
 Name: athena-intelligence
-Version: 0.1.119
+Version: 0.1.272
 Summary: Athena Intelligence Python Library
-Requires-Python: >=3.8,<4.0
+Requires-Python: >=3.9,<4.0
 Classifier: Intended Audience :: Developers
 Classifier: Operating System :: MacOS
 Classifier: Operating System :: Microsoft :: Windows
@@ -11,15 +11,19 @@ Classifier: Operating System :: POSIX
 Classifier: Operating System :: POSIX :: Linux
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.8
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Typing :: Typed
+Requires-Dist: fastapi (==0.115.10)
 Requires-Dist: httpx (>=0.21.2)
+Requires-Dist: langchain_core (>=0.3.40,<0.4.0)
+Requires-Dist: langserve (>=0.3.1,<0.4.0)
 Requires-Dist: pydantic (>=1.9.2)
+Requires-Dist: pydantic-core (>=2.18.2)
 Requires-Dist: python-magic (==0.4.27)
 Requires-Dist: typing_extensions (>=4.0.0)
 Description-Content-Type: text/markdown

{athena_intelligence-0.1.119 → athena_intelligence-0.1.272}/pyproject.toml

@@ -1,6 +1,9 @@
+[project]
+name = "athena-intelligence"
+
 [tool.poetry]
 name = "athena-intelligence"
-version = "0.1.119"
+version = "0.1.272"
 description = "Athena Intelligence Python Library"
 readme = "README.md"
 authors = []
@@ -28,25 +31,30 @@ packages = [
 ]
 
 [project.urls]
-Documentation = 'https://docs.athenaintelligence.ai/'
-Homepage = 'https://www.athenaintelligence.ai/'
+Documentation = 'https://docs.athenaintel.com/'
+Homepage = 'https://www.athenaintel.com/'
 Repository = 'https://github.com/Athena-Intel/athena-python'
 
 [tool.poetry.dependencies]
-python = "^3.8"
+python = ">=3.9,<4.0"
+fastapi = "0.115.10"
 httpx = ">=0.21.2"
+langchain_core = "^0.3.40"
+langserve = "^0.3.1"
 pydantic = ">= 1.9.2"
+pydantic-core = ">=2.18.2"
 python-magic = "0.4.27"
 typing_extensions = ">= 4.0.0"
 
-[tool.poetry.dev-dependencies]
-mypy = "1.0.1"
+[tool.poetry.group.dev.dependencies]
+mypy = "==1.13.0"
 pytest = "^7.4.0"
 pytest-asyncio = "^0.23.5"
 python-dateutil = "^2.9.0"
 types-python-dateutil = "^2.9.0.20240316"
 pandas = ">= 1.5.3"
 pandas-stubs = ">= 1.5.3"
+ruff = "==0.11.5"
 
 [tool.pytest.ini_options]
 testpaths = [ "tests" ]
@@ -55,6 +63,29 @@ asyncio_mode = "auto"
 [tool.mypy]
 plugins = ["pydantic.mypy"]
 
+[tool.ruff]
+line-length = 120
+
+[tool.ruff.lint]
+select = [
+    "E",  # pycodestyle errors
+    "F",  # pyflakes
+    "I",  # isort
+]
+ignore = [
+    "E402",  # Module level import not at top of file
+    "E501",  # Line too long
+    "E711",  # Comparison to `None` should be `cond is not None`
+    "E712",  # Avoid equality comparisons to `True`; use `if ...:` checks
+    "E721",  # Use `is` and `is not` for type comparisons, or `isinstance()` for insinstance checks
+    "E722",  # Do not use bare `except`
+    "E731",  # Do not assign a `lambda` expression, use a `def`
+    "F821",  # Undefined name
+    "F841"   # Local variable ... is assigned to but never used
+]
+
+[tool.ruff.lint.isort]
+section-order = ["future", "standard-library", "third-party", "first-party"]
 
 [build-system]
 requires = ["poetry-core"]

athena_intelligence-0.1.272/src/athena/__init__.py

@@ -0,0 +1,312 @@
+# 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,
+        CreatableAssetType,
+        CreateAssetResponseOut,
+        CreateNewSheetTabResponse,
+        CreateProjectResponseOut,
+        CustomAgentResponse,
+        DataFrameRequestOut,
+        DataFrameRequestOutColumnsItem,
+        DataFrameRequestOutDataItemItem,
+        DataFrameRequestOutIndexItem,
+        DataFrameUnknownFormatError,
+        DimensionProperties,
+        DocumentChunk,
+        DriveAgentResponse,
+        FileChunkRequestOut,
+        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",
+    "DocumentChunk": ".types",
+    "DriveAgentResponse": ".types",
+    "FileChunkRequestOut": ".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",
+    "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",
+    "CreatableAssetType",
+    "CreateAssetResponseOut",
+    "CreateNewSheetTabResponse",
+    "CreateProjectResponseOut",
+    "CustomAgentResponse",
+    "DataFrameRequestOut",
+    "DataFrameRequestOutColumnsItem",
+    "DataFrameRequestOutDataItemItem",
+    "DataFrameRequestOutIndexItem",
+    "DataFrameUnknownFormatError",
+    "DimensionProperties",
+    "DocumentChunk",
+    "DriveAgentResponse",
+    "FileChunkRequestOut",
+    "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",
+    "Tabcolor",
+    "TableRowData",
+    "TextContent",
+    "TextFormatModel",
+    "Textrotation",
+    "ThemeColor",
+    "ThreadStatusResponseOut",
+    "ToolsDataFrameRequestColumnsItem",
+    "Type",
+    "UnauthorizedError",
+    "UnprocessableEntityError",
+    "UnsupportedMediaTypeError",
+    "WrapStrategy",
+    "__version__",
+    "agents",
+    "aop",
+    "assets",
+    "query",
+    "threads",
+    "tools",
+]

athena_intelligence-0.1.272/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.272/src/athena/agents/client.py

@@ -0,0 +1,240 @@
+# 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 .raw_client import AsyncRawAgentsClient, RawAgentsClient
+
+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, ...)
+
+
+class AgentsClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._raw_client = RawAgentsClient(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:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawAgentsClient
+        """
+        return self._raw_client
+
+    def invoke_by_id(
+        self,
+        agent_id: str,
+        *,
+        config: typing.Dict[str, typing.Optional[typing.Any]],
+        messages: typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]],
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> CustomAgentResponse:
+        """
+        Coming soon!
+
+        Invoke a custom agent created in [spaces](https://resources.athenaintel.com/docs/agents/create-your-agent).
+
+        Custom agents can be created and configured in spaces to perform specialized tasks.
+        Refer to the specific agent's documentation for details on configuration options
+        and expected responses.
+
+        Parameters
+        ----------
+        agent_id : str
+            The ID of the custom agent to invoke. Create custom agents in [spaces](https://resources.athenaintel.com/docs/agents/create-your-agent).
+
+        config : typing.Dict[str, typing.Optional[typing.Any]]
+            Configuration for the custom agent. See the agent's documentation for specific configuration options.
+
+        messages : typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]]
+            The messages to send to the custom agent
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        CustomAgentResponse
+            Successful Response
+
+        Examples
+        --------
+        from athena import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.agents.invoke_by_id(
+            agent_id="agent_id",
+            config={"key": "value"},
+            messages=[{"key": "value"}],
+        )
+        """
+        _response = self._raw_client.invoke_by_id(
+            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 DriveClient  # noqa: E402
+
+            self._drive = DriveClient(client_wrapper=self._client_wrapper)
+        return self._drive
+
+    @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
+
+    @property
+    def research(self):
+        if self._research is None:
+            from .research.client import ResearchClient  # noqa: E402
+
+            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:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawAgentsClient
+        """
+        return self._raw_client
+
+    async def invoke_by_id(
+        self,
+        agent_id: str,
+        *,
+        config: typing.Dict[str, typing.Optional[typing.Any]],
+        messages: typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]],
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> CustomAgentResponse:
+        """
+        Coming soon!
+
+        Invoke a custom agent created in [spaces](https://resources.athenaintel.com/docs/agents/create-your-agent).
+
+        Custom agents can be created and configured in spaces to perform specialized tasks.
+        Refer to the specific agent's documentation for details on configuration options
+        and expected responses.
+
+        Parameters
+        ----------
+        agent_id : str
+            The ID of the custom agent to invoke. Create custom agents in [spaces](https://resources.athenaintel.com/docs/agents/create-your-agent).
+
+        config : typing.Dict[str, typing.Optional[typing.Any]]
+            Configuration for the custom agent. See the agent's documentation for specific configuration options.
+
+        messages : typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]]
+            The messages to send to the custom agent
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        CustomAgentResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from athena import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.agents.invoke_by_id(
+                agent_id="agent_id",
+                config={"key": "value"},
+                messages=[{"key": "value"}],
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.invoke_by_id(
+            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