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

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

@@ -1,8 +1,8 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: athena-intelligence
-Version: 0.1.125
+Version: 0.1.127
 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,17 +11,25 @@ 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.13
+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,<3.0.0)
 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.125 → athena_intelligence-0.1.127}/pyproject.toml

@@ -1,6 +1,9 @@
+[project]
+name = "athena-intelligence"
+
 [tool.poetry]
 name = "athena-intelligence"
-version = "0.1.125"
+version = "0.1.127"
 description = "Athena Intelligence Python Library"
 readme = "README.md"
 authors = []
@@ -28,14 +31,18 @@ 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"
 
@@ -47,6 +54,7 @@ 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"
 
 [tool.pytest.ini_options]
 testpaths = [ "tests" ]
@@ -55,6 +63,9 @@ asyncio_mode = "auto"
 [tool.mypy]
 plugins = ["pydantic.mypy"]
 
+[tool.ruff]
+line-length = 120
+
 
 [build-system]
 requires = ["poetry-core"]

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

@@ -1,8 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .types import (
+    AssetContentRequestOut,
     AssetNode,
     AssetNotFoundError,
+    AssetScreenshotResponseOut,
     Chunk,
     ChunkContentItem,
     ChunkContentItem_ImageUrl,
@@ -45,14 +47,19 @@ from .errors import (
     UnsupportedMediaTypeError,
 )
 from . import agents, query, tools
+from .client import AsyncAthena, Athena
 from .environment import AthenaEnvironment
 from .query import QueryExecuteRequestDatabaseAssetIds
 from .tools import ToolsDataFrameRequestColumnsItem
 from .version import __version__
 
 __all__ = [
+    "AssetContentRequestOut",
     "AssetNode",
     "AssetNotFoundError",
+    "AssetScreenshotResponseOut",
+    "AsyncAthena",
+    "Athena",
     "AthenaEnvironment",
     "BadRequestError",
     "Chunk",

{athena_intelligence-0.1.125 → athena_intelligence-0.1.127}/src/athena/agents/client.py

@@ -1,19 +1,23 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from json.decoder import JSONDecodeError
-
-from ..core.api_error import ApiError
-from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from ..core.jsonable_encoder import jsonable_encoder
-from ..core.pydantic_utilities import pydantic_v1
+from ..core.client_wrapper import SyncClientWrapper
+from .drive.client import DriveClient
+from .general.client import GeneralClient
+from .research.client import ResearchClient
+from .sql.client import SqlClient
 from ..core.request_options import RequestOptions
-from ..errors.unprocessable_entity_error import UnprocessableEntityError
 from ..types.custom_agent_response import CustomAgentResponse
-from .drive.client import AsyncDriveClient, DriveClient
-from .general.client import AsyncGeneralClient, GeneralClient
-from .research.client import AsyncResearchClient, ResearchClient
-from .sql.client import AsyncSqlClient, SqlClient
+from ..core.jsonable_encoder import jsonable_encoder
+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 .drive.client import AsyncDriveClient
+from .general.client import AsyncGeneralClient
+from .research.client import AsyncResearchClient
+from .sql.client import AsyncSqlClient
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -31,8 +35,8 @@ class AgentsClient:
         self,
         agent_id: str,
         *,
-        config: typing.Dict[str, typing.Any],
-        messages: typing.Sequence[typing.Dict[str, typing.Any]],
+        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:
         """
@@ -49,10 +53,10 @@ class AgentsClient:
         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.Any]
+        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.Any]]
+        messages : typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]]
             The messages to send to the custom agent
 
         request_options : typing.Optional[RequestOptions]
@@ -65,7 +69,7 @@ class AgentsClient:
 
         Examples
         --------
-        from athena.client import Athena
+        from athena import Athena
 
         client = Athena(
             api_key="YOUR_API_KEY",
@@ -79,15 +83,35 @@ class AgentsClient:
         _response = self._client_wrapper.httpx_client.request(
             f"api/v0/agents/{jsonable_encoder(agent_id)}/invoke",
             method="POST",
-            json={"config": config, "messages": messages},
+            json={
+                "config": config,
+                "messages": messages,
+            },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(CustomAgentResponse, _response.json())  # type: ignore
-        if _response.status_code == 422:
-            raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    CustomAgentResponse,
+                    parse_obj_as(
+                        type_=CustomAgentResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    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)
@@ -106,8 +130,8 @@ class AsyncAgentsClient:
         self,
         agent_id: str,
         *,
-        config: typing.Dict[str, typing.Any],
-        messages: typing.Sequence[typing.Dict[str, typing.Any]],
+        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:
         """
@@ -124,10 +148,10 @@ class AsyncAgentsClient:
         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.Any]
+        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.Any]]
+        messages : typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]]
             The messages to send to the custom agent
 
         request_options : typing.Optional[RequestOptions]
@@ -140,29 +164,57 @@ class AsyncAgentsClient:
 
         Examples
         --------
-        from athena.client import AsyncAthena
+        import asyncio
+
+        from athena import AsyncAthena
 
         client = AsyncAthena(
             api_key="YOUR_API_KEY",
         )
-        await client.agents.invoke_by_id(
-            agent_id="agent_id",
-            config={"key": "value"},
-            messages=[{"key": "value"}],
-        )
+
+
+        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._client_wrapper.httpx_client.request(
             f"api/v0/agents/{jsonable_encoder(agent_id)}/invoke",
             method="POST",
-            json={"config": config, "messages": messages},
+            json={
+                "config": config,
+                "messages": messages,
+            },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(CustomAgentResponse, _response.json())  # type: ignore
-        if _response.status_code == 422:
-            raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    CustomAgentResponse,
+                    parse_obj_as(
+                        type_=CustomAgentResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    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)

{athena_intelligence-0.1.125 → athena_intelligence-0.1.127}/src/athena/agents/drive/client.py

@@ -1,14 +1,14 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from json.decoder import JSONDecodeError
-
-from ...core.api_error import ApiError
-from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from ...core.pydantic_utilities import pydantic_v1
+from ...core.client_wrapper import SyncClientWrapper
 from ...core.request_options import RequestOptions
-from ...errors.unprocessable_entity_error import UnprocessableEntityError
 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
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -21,19 +21,19 @@ class DriveClient:
     def invoke(
         self,
         *,
-        config: typing.Dict[str, typing.Any],
-        messages: typing.Sequence[typing.Dict[str, typing.Any]],
-        request_options: typing.Optional[RequestOptions] = None
+        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.Any]
+        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.Any]]
+        messages : typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]]
             The messages to send to the drive agent
 
         request_options : typing.Optional[RequestOptions]
@@ -46,7 +46,7 @@ class DriveClient:
 
         Examples
         --------
-        from athena.client import Athena
+        from athena import Athena
 
         client = Athena(
             api_key="YOUR_API_KEY",
@@ -59,15 +59,35 @@ class DriveClient:
         _response = self._client_wrapper.httpx_client.request(
             "api/v0/agents/drive/invoke",
             method="POST",
-            json={"config": config, "messages": messages},
+            json={
+                "config": config,
+                "messages": messages,
+            },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(DriveAgentResponse, _response.json())  # type: ignore
-        if _response.status_code == 422:
-            raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    DriveAgentResponse,
+                    parse_obj_as(
+                        type_=DriveAgentResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    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)
@@ -81,19 +101,19 @@ class AsyncDriveClient:
     async def invoke(
         self,
         *,
-        config: typing.Dict[str, typing.Any],
-        messages: typing.Sequence[typing.Dict[str, typing.Any]],
-        request_options: typing.Optional[RequestOptions] = None
+        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.Any]
+        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.Any]]
+        messages : typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]]
             The messages to send to the drive agent
 
         request_options : typing.Optional[RequestOptions]
@@ -106,28 +126,56 @@ class AsyncDriveClient:
 
         Examples
         --------
-        from athena.client import AsyncAthena
+        import asyncio
+
+        from athena import AsyncAthena
 
         client = AsyncAthena(
             api_key="YOUR_API_KEY",
         )
-        await client.agents.drive.invoke(
-            config={"key": "value"},
-            messages=[{"key": "value"}],
-        )
+
+
+        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",
             method="POST",
-            json={"config": config, "messages": messages},
+            json={
+                "config": config,
+                "messages": messages,
+            },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(DriveAgentResponse, _response.json())  # type: ignore
-        if _response.status_code == 422:
-            raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    DriveAgentResponse,
+                    parse_obj_as(
+                        type_=DriveAgentResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    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)