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

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

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

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

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

{athena_intelligence-0.1.126 → 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,
@@ -52,8 +54,10 @@ from .tools import ToolsDataFrameRequestColumnsItem
 from .version import __version__
 
 __all__ = [
+    "AssetContentRequestOut",
     "AssetNode",
     "AssetNotFoundError",
+    "AssetScreenshotResponseOut",
     "AsyncAthena",
     "Athena",
     "AthenaEnvironment",

{athena_intelligence-0.1.126 → athena_intelligence-0.1.127}/src/athena/core/client_wrapper.py

@@ -16,7 +16,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "athena-intelligence",
-            "X-Fern-SDK-Version": "0.1.126",
+            "X-Fern-SDK-Version": "0.1.127",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

{athena_intelligence-0.1.126 → athena_intelligence-0.1.127}/src/athena/tools/client.py

@@ -15,6 +15,9 @@ from ..types.asset_not_found_error import AssetNotFoundError
 from ..errors.unprocessable_entity_error import UnprocessableEntityError
 from json.decoder import JSONDecodeError
 from ..core.api_error import ApiError
+from ..types.asset_content_request_out import AssetContentRequestOut
+from ..types.asset_screenshot_response_out import AssetScreenshotResponseOut
+from ..errors.internal_server_error import InternalServerError
 from ..types.folder_response import FolderResponse
 from ..errors.bad_request_error import BadRequestError
 from ..types.parent_folder_error import ParentFolderError
@@ -22,7 +25,6 @@ from .types.tools_data_frame_request_columns_item import ToolsDataFrameRequestCo
 from ..types.data_frame_request_out import DataFrameRequestOut
 from ..errors.unsupported_media_type_error import UnsupportedMediaTypeError
 from ..types.data_frame_unknown_format_error import DataFrameUnknownFormatError
-from ..errors.internal_server_error import InternalServerError
 from .. import core
 from ..types.save_asset_request_out import SaveAssetRequestOut
 from ..errors.content_too_large_error import ContentTooLargeError
@@ -134,6 +136,185 @@ class ToolsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def get_asset_content(
+        self, *, asset_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> AssetContentRequestOut:
+        """
+        Get the content of an asset.
+
+        Parameters
+        ----------
+        asset_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AssetContentRequestOut
+            Successful Response
+
+        Examples
+        --------
+        from athena import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.tools.get_asset_content(
+            asset_id="asset_id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/tools/asset/content",
+            method="GET",
+            params={
+                "asset_id": asset_id,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    AssetContentRequestOut,
+                    parse_obj_as(
+                        type_=AssetContentRequestOut,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    )
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    typing.cast(
+                        AssetNotFoundError,
+                        parse_obj_as(
+                            type_=AssetNotFoundError,  # 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)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def get_asset_screenshot(
+        self,
+        *,
+        asset_id: str,
+        page_number: typing.Optional[int] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AssetScreenshotResponseOut:
+        """
+        Get a screenshot of a specific page from an asset.
+
+        Parameters
+        ----------
+        asset_id : str
+
+        page_number : typing.Optional[int]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AssetScreenshotResponseOut
+            Successful Response
+
+        Examples
+        --------
+        from athena import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.tools.get_asset_screenshot(
+            asset_id="asset_id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/tools/asset/screenshot",
+            method="GET",
+            params={
+                "asset_id": asset_id,
+                "page_number": page_number,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    AssetScreenshotResponseOut,
+                    parse_obj_as(
+                        type_=AssetScreenshotResponseOut,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    )
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    typing.cast(
+                        AssetNotFoundError,
+                        parse_obj_as(
+                            type_=AssetNotFoundError,  # 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(),
+                        ),
+                    )
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    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)
+
     def list_contents(
         self,
         *,
@@ -243,7 +424,7 @@ class ToolsClient:
         columns: typing.Optional[
             typing.Union[ToolsDataFrameRequestColumnsItem, typing.Sequence[ToolsDataFrameRequestColumnsItem]]
         ] = None,
-        sheet_name: typing.Optional[int] = None,
+        sheet_name: typing.Optional[str] = None,
         separator: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DataFrameRequestOut:
@@ -259,7 +440,7 @@ class ToolsClient:
         columns : typing.Optional[typing.Union[ToolsDataFrameRequestColumnsItem, typing.Sequence[ToolsDataFrameRequestColumnsItem]]]
             should be a list of strings or a list of integers
 
-        sheet_name : typing.Optional[int]
+        sheet_name : typing.Optional[str]
             only for excel files
 
         separator : typing.Optional[str]
@@ -635,6 +816,201 @@ class AsyncToolsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def get_asset_content(
+        self, *, asset_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> AssetContentRequestOut:
+        """
+        Get the content of an asset.
+
+        Parameters
+        ----------
+        asset_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AssetContentRequestOut
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from athena import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.tools.get_asset_content(
+                asset_id="asset_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/tools/asset/content",
+            method="GET",
+            params={
+                "asset_id": asset_id,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    AssetContentRequestOut,
+                    parse_obj_as(
+                        type_=AssetContentRequestOut,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    )
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    typing.cast(
+                        AssetNotFoundError,
+                        parse_obj_as(
+                            type_=AssetNotFoundError,  # 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)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def get_asset_screenshot(
+        self,
+        *,
+        asset_id: str,
+        page_number: typing.Optional[int] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AssetScreenshotResponseOut:
+        """
+        Get a screenshot of a specific page from an asset.
+
+        Parameters
+        ----------
+        asset_id : str
+
+        page_number : typing.Optional[int]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AssetScreenshotResponseOut
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from athena import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.tools.get_asset_screenshot(
+                asset_id="asset_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/tools/asset/screenshot",
+            method="GET",
+            params={
+                "asset_id": asset_id,
+                "page_number": page_number,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    AssetScreenshotResponseOut,
+                    parse_obj_as(
+                        type_=AssetScreenshotResponseOut,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    )
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    typing.cast(
+                        AssetNotFoundError,
+                        parse_obj_as(
+                            type_=AssetNotFoundError,  # 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(),
+                        ),
+                    )
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    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)
+
     async def list_contents(
         self,
         *,
@@ -752,7 +1128,7 @@ class AsyncToolsClient:
         columns: typing.Optional[
             typing.Union[ToolsDataFrameRequestColumnsItem, typing.Sequence[ToolsDataFrameRequestColumnsItem]]
         ] = None,
-        sheet_name: typing.Optional[int] = None,
+        sheet_name: typing.Optional[str] = None,
         separator: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DataFrameRequestOut:
@@ -768,7 +1144,7 @@ class AsyncToolsClient:
         columns : typing.Optional[typing.Union[ToolsDataFrameRequestColumnsItem, typing.Sequence[ToolsDataFrameRequestColumnsItem]]]
             should be a list of strings or a list of integers
 
-        sheet_name : typing.Optional[int]
+        sheet_name : typing.Optional[str]
             only for excel files
 
         separator : typing.Optional[str]

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

@@ -1,7 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from .asset_content_request_out import AssetContentRequestOut
 from .asset_node import AssetNode
 from .asset_not_found_error import AssetNotFoundError
+from .asset_screenshot_response_out import AssetScreenshotResponseOut
 from .chunk import Chunk
 from .chunk_content_item import ChunkContentItem, ChunkContentItem_ImageUrl, ChunkContentItem_Text
 from .chunk_result import ChunkResult
@@ -33,8 +35,10 @@ from .tool import Tool
 from .type import Type
 
 __all__ = [
+    "AssetContentRequestOut",
     "AssetNode",
     "AssetNotFoundError",
+    "AssetScreenshotResponseOut",
     "Chunk",
     "ChunkContentItem",
     "ChunkContentItem_ImageUrl",

athena_intelligence-0.1.127/src/athena/types/asset_content_request_out.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+
+
+class AssetContentRequestOut(UniversalBaseModel):
+    """
+    Response model with asset content.
+    """
+
+    content: str = pydantic.Field()
+    """
+    The content of the asset
+    """
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

athena_intelligence-0.1.127/src/athena/types/asset_screenshot_response_out.py

@@ -0,0 +1,43 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing_extensions
+from ..core.serialization import FieldMetadata
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+
+
+class AssetScreenshotResponseOut(UniversalBaseModel):
+    """
+    Response model with screenshot data.
+    """
+
+    base_64_image: typing_extensions.Annotated[str, FieldMetadata(alias="base64_image")] = pydantic.Field()
+    """
+    Base64-encoded image data
+    """
+
+    message: str = pydantic.Field()
+    """
+    Status message
+    """
+
+    page_number: int = pydantic.Field()
+    """
+    The page number that was captured
+    """
+
+    total_pages: int = pydantic.Field()
+    """
+    Total number of pages in the document
+    """
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

{athena_intelligence-0.1.126 → athena_intelligence-0.1.127}/src/athena/types/general_agent_config.py

@@ -14,6 +14,7 @@ class GeneralAgentConfig(UniversalBaseModel):
 
     enabled_tools: typing.Optional[typing.List[GeneralAgentConfigEnabledToolsItem]] = None
     knowledge_base_asset_ids: typing.Optional[typing.List[str]] = None
+    model: typing.Optional[str] = None
     system_prompt: typing.Optional[str] = None
 
     if IS_PYDANTIC_V2: