athena-intelligence 0.1.127__tar.gz → 0.1.185__tar.gz

{athena_intelligence-0.1.127 → athena_intelligence-0.1.185}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.1
 Name: athena-intelligence
-Version: 0.1.127
+Version: 0.1.185
 Summary: Athena Intelligence Python Library
 Requires-Python: >=3.9,<4.0
 Classifier: Intended Audience :: Developers
@@ -15,7 +15,6 @@ 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.13
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Typing :: Typed
@@ -24,12 +23,9 @@ 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,<3.0.0)
+Requires-Dist: pydantic-core (>=2.18.2)
 Requires-Dist: python-magic (==0.4.27)
 Requires-Dist: typing_extensions (>=4.0.0)
-Project-URL: Documentation, https://docs.athenaintel.com/
-Project-URL: Homepage, https://www.athenaintel.com/
-Project-URL: Repository, https://github.com/Athena-Intel/athena-python
 Description-Content-Type: text/markdown
 
 # Athena Intelligence Python Library

{athena_intelligence-0.1.127 → athena_intelligence-0.1.185}/pyproject.toml

@@ -3,7 +3,7 @@ name = "athena-intelligence"
 
 [tool.poetry]
 name = "athena-intelligence"
-version = "0.1.127"
+version = "0.1.185"
 description = "Athena Intelligence Python Library"
 readme = "README.md"
 authors = []
@@ -42,19 +42,19 @@ httpx = ">=0.21.2"
 langchain_core = "^0.3.40"
 langserve = "^0.3.1"
 pydantic = ">= 1.9.2"
-pydantic-core = "^2.18.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.5.6"
+ruff = "==0.11.5"
 
 [tool.pytest.ini_options]
 testpaths = [ "tests" ]
@@ -66,6 +66,26 @@ 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.127 → athena_intelligence-0.1.185}/src/athena/__init__.py

@@ -1,5 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+
 from .types import (
     AssetContentRequestOut,
     AssetNode,
@@ -11,6 +13,7 @@ from .types import (
     ChunkContentItem_Text,
     ChunkResult,
     ChunkResultChunkId,
+    Content,
     CustomAgentResponse,
     DataFrameRequestOut,
     DataFrameRequestOutColumnsItem,
@@ -26,15 +29,23 @@ from .types import (
     GeneralAgentConfigEnabledToolsItem,
     GeneralAgentRequest,
     GeneralAgentResponse,
+    GeneralAgentResponseMessage,
+    GeneralAgentResponseMessageKwargs,
+    Id,
     ImageUrlContent,
+    InputMessage,
+    InputMessageContentItem,
+    InputMessageContentItem_ImageUrl,
+    InputMessageContentItem_Text,
+    PaginatedAssetsOut,
     ParentFolderError,
     PromptMessage,
+    PublicAssetOut,
     ResearchAgentResponse,
     SaveAssetRequestOut,
     SqlAgentResponse,
     StructuredDataExtractorResponse,
     TextContent,
-    Tool,
     Type,
 )
 from .errors import (
@@ -46,7 +57,7 @@ from .errors import (
     UnprocessableEntityError,
     UnsupportedMediaTypeError,
 )
-from . import agents, query, tools
+from . import agents, assets, query, tools
 from .client import AsyncAthena, Athena
 from .environment import AthenaEnvironment
 from .query import QueryExecuteRequestDatabaseAssetIds
@@ -68,6 +79,7 @@ __all__ = [
     "ChunkContentItem_Text",
     "ChunkResult",
     "ChunkResultChunkId",
+    "Content",
     "ContentTooLargeError",
     "CustomAgentResponse",
     "DataFrameRequestOut",
@@ -84,18 +96,26 @@ __all__ = [
     "GeneralAgentConfigEnabledToolsItem",
     "GeneralAgentRequest",
     "GeneralAgentResponse",
+    "GeneralAgentResponseMessage",
+    "GeneralAgentResponseMessageKwargs",
+    "Id",
     "ImageUrlContent",
+    "InputMessage",
+    "InputMessageContentItem",
+    "InputMessageContentItem_ImageUrl",
+    "InputMessageContentItem_Text",
     "InternalServerError",
     "NotFoundError",
+    "PaginatedAssetsOut",
     "ParentFolderError",
     "PromptMessage",
+    "PublicAssetOut",
     "QueryExecuteRequestDatabaseAssetIds",
     "ResearchAgentResponse",
     "SaveAssetRequestOut",
     "SqlAgentResponse",
     "StructuredDataExtractorResponse",
     "TextContent",
-    "Tool",
     "ToolsDataFrameRequestColumnsItem",
     "Type",
     "UnauthorizedError",
@@ -103,6 +123,7 @@ __all__ = [
     "UnsupportedMediaTypeError",
     "__version__",
     "agents",
+    "assets",
     "query",
     "tools",
 ]

{athena_intelligence-0.1.127 → athena_intelligence-0.1.185}/src/athena/agents/__init__.py

@@ -1,5 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+
 from . import drive, general, research, sql
 
 __all__ = ["drive", "general", "research", "sql"]

athena_intelligence-0.1.185/src/athena/agents/client.py

@@ -0,0 +1,177 @@
+# 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.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
+
+# 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.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)
+
+    @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
+
+
+class AsyncAgentsClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawAgentsClient(client_wrapper=client_wrapper)
+        self.drive = AsyncDriveClient(client_wrapper=client_wrapper)
+
+        self.general = AsyncGeneralClient(client_wrapper=client_wrapper)
+
+        self.research = AsyncResearchClient(client_wrapper=client_wrapper)
+
+        self.sql = AsyncSqlClient(client_wrapper=client_wrapper)
+
+    @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

{athena_intelligence-0.1.127/src/athena/agents/sql → athena_intelligence-0.1.185/src/athena/agents/drive}/__init__.py

@@ -1,2 +1,4 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+

athena_intelligence-0.1.185/src/athena/agents/drive/client.py

@@ -0,0 +1,133 @@
+# 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.drive_agent_response import DriveAgentResponse
+from .raw_client import AsyncRawDriveClient, RawDriveClient
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class DriveClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._raw_client = RawDriveClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawDriveClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawDriveClient
+        """
+        return self._raw_client
+
+    def invoke(
+        self,
+        *,
+        config: typing.Dict[str, typing.Optional[typing.Any]],
+        messages: typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]],
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> DriveAgentResponse:
+        """
+        Coming soon! Manage folders and search for files in the internal drive.
+
+        Parameters
+        ----------
+        config : typing.Dict[str, typing.Optional[typing.Any]]
+            Configuration for the drive agent including folder paths and search parameters
+
+        messages : typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]]
+            The messages to send to the drive agent
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DriveAgentResponse
+            Successful Response
+
+        Examples
+        --------
+        from athena import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.agents.drive.invoke(
+            config={"key": "value"},
+            messages=[{"key": "value"}],
+        )
+        """
+        _response = self._raw_client.invoke(config=config, messages=messages, request_options=request_options)
+        return _response.data
+
+
+class AsyncDriveClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawDriveClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawDriveClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawDriveClient
+        """
+        return self._raw_client
+
+    async def invoke(
+        self,
+        *,
+        config: typing.Dict[str, typing.Optional[typing.Any]],
+        messages: typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]],
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> DriveAgentResponse:
+        """
+        Coming soon! Manage folders and search for files in the internal drive.
+
+        Parameters
+        ----------
+        config : typing.Dict[str, typing.Optional[typing.Any]]
+            Configuration for the drive agent including folder paths and search parameters
+
+        messages : typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]]
+            The messages to send to the drive agent
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DriveAgentResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from athena import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.agents.drive.invoke(
+                config={"key": "value"},
+                messages=[{"key": "value"}],
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.invoke(config=config, messages=messages, request_options=request_options)
+        return _response.data

athena_intelligence-0.1.127/src/athena/agents/drive/client.py → athena_intelligence-0.1.185/src/athena/agents/drive/raw_client.py

@@ -1,20 +1,21 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ...core.client_wrapper import SyncClientWrapper
-from ...core.request_options import RequestOptions
-from ...types.drive_agent_response import DriveAgentResponse
-from ...core.pydantic_utilities import parse_obj_as
-from ...errors.unprocessable_entity_error import UnprocessableEntityError
 from json.decoder import JSONDecodeError
+
 from ...core.api_error import ApiError
-from ...core.client_wrapper import AsyncClientWrapper
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.http_response import AsyncHttpResponse, HttpResponse
+from ...core.pydantic_utilities import parse_obj_as
+from ...core.request_options import RequestOptions
+from ...errors.unprocessable_entity_error import UnprocessableEntityError
+from ...types.drive_agent_response import DriveAgentResponse
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
 
 
-class DriveClient:
+class RawDriveClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
@@ -24,7 +25,7 @@ class DriveClient:
         config: typing.Dict[str, typing.Optional[typing.Any]],
         messages: typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]],
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> DriveAgentResponse:
+    ) -> HttpResponse[DriveAgentResponse]:
         """
         Coming soon! Manage folders and search for files in the internal drive.
 
@@ -41,20 +42,8 @@ class DriveClient:
 
         Returns
         -------
-        DriveAgentResponse
+        HttpResponse[DriveAgentResponse]
             Successful Response
-
-        Examples
-        --------
-        from athena import Athena
-
-        client = Athena(
-            api_key="YOUR_API_KEY",
-        )
-        client.agents.drive.invoke(
-            config={"key": "value"},
-            messages=[{"key": "value"}],
-        )
         """
         _response = self._client_wrapper.httpx_client.request(
             "api/v0/agents/drive/invoke",
@@ -71,30 +60,32 @@ class DriveClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return typing.cast(
+                _data = typing.cast(
                     DriveAgentResponse,
                     parse_obj_as(
                         type_=DriveAgentResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
+                return HttpResponse(response=_response, data=_data)
             if _response.status_code == 422:
                 raise UnprocessableEntityError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             _response_json = _response.json()
         except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
 
-class AsyncDriveClient:
+class AsyncRawDriveClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
@@ -104,7 +95,7 @@ class AsyncDriveClient:
         config: typing.Dict[str, typing.Optional[typing.Any]],
         messages: typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]],
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> DriveAgentResponse:
+    ) -> AsyncHttpResponse[DriveAgentResponse]:
         """
         Coming soon! Manage folders and search for files in the internal drive.
 
@@ -121,28 +112,8 @@ class AsyncDriveClient:
 
         Returns
         -------
-        DriveAgentResponse
+        AsyncHttpResponse[DriveAgentResponse]
             Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from athena import AsyncAthena
-
-        client = AsyncAthena(
-            api_key="YOUR_API_KEY",
-        )
-
-
-        async def main() -> None:
-            await client.agents.drive.invoke(
-                config={"key": "value"},
-                messages=[{"key": "value"}],
-            )
-
-
-        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
             "api/v0/agents/drive/invoke",
@@ -159,24 +130,26 @@ class AsyncDriveClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return typing.cast(
+                _data = typing.cast(
                     DriveAgentResponse,
                     parse_obj_as(
                         type_=DriveAgentResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
+                return AsyncHttpResponse(response=_response, data=_data)
             if _response.status_code == 422:
                 raise UnprocessableEntityError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             _response_json = _response.json()
         except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)

{athena_intelligence-0.1.127/src/athena/agents/drive → athena_intelligence-0.1.185/src/athena/agents/general}/__init__.py

@@ -1,2 +1,4 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+