athena-intelligence 0.1.98__tar.gz → 0.1.99__tar.gz

{athena_intelligence-0.1.98 → athena_intelligence-0.1.99}/PKG-INFO

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

{athena_intelligence-0.1.98 → athena_intelligence-0.1.99}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "athena-intelligence"
-version = "0.1.98"
+version = "0.1.99"
 description = "Athena Intelligence Python Library"
 readme = "README.md"
 authors = []

{athena_intelligence-0.1.98 → athena_intelligence-0.1.99}/src/athena/client.py

@@ -34,11 +34,11 @@ def _inherit_signature_and_doc(
 
 class WrappedToolsClient(ToolsClient):
 
-    def get_file(self, document_id: str) -> io.BytesIO:
+    def get_file(self, asset_id: str) -> io.BytesIO:
         """
         Parameters
         ----------
-        document_id : str
+        asset_id : str
 
         Returns
         -------
@@ -50,28 +50,28 @@ class WrappedToolsClient(ToolsClient):
         from athena.client import Athena
 
         client = Athena(api_key="YOUR_API_KEY")
-        file_io = client.tools.get_file(document_id="document_id")
+        file_io = client.tools.get_file(asset_id="asset_id")
         pl.read_csv(file_io)
         """
-        file_bytes = b''.join(self.raw_data(document_id=document_id))
+        file_bytes = b''.join(self.raw_data(asset_id=asset_id))
         bytes_io = io.BytesIO(file_bytes)
         return bytes_io
 
     @_inherit_signature_and_doc(ToolsClient.data_frame, {'DataFrameRequestOut': 'pd.DataFrame'})
-    def data_frame(self, *, document_id: str, **kwargs) -> 'pd.DataFrame':
+    def data_frame(self, *, asset_id: str, **kwargs) -> 'pd.DataFrame':
         _check_pandas_installed()
-        model = super().data_frame(document_id=document_id, **kwargs)
+        model = super().data_frame(asset_id=asset_id, **kwargs)
         return _read_json_frame(model)
 
-    def read_data_frame(self, document_id: str, *args, **kwargs) -> 'pd.DataFrame':
+    def read_data_frame(self, asset_id: str, *args, **kwargs) -> 'pd.DataFrame':
         """
         Parameters
         ----------
-        document_id : str
+        asset_id : str
 
         **kwargs : dict
           keyword arguments passed to pandas `read_csv` or `read_excel` function,
-          depending on the file type of the document identified by `document_id`.
+          depending on the file type of the document identified by `asset_id`.
 
         Returns
         -------
@@ -82,10 +82,10 @@ class WrappedToolsClient(ToolsClient):
         from athena.client import Athena
 
         client = Athena(api_key="YOUR_API_KEY")
-        client.tools.read_data_frame(document_id="document_id")
+        client.tools.read_data_frame(asset_id="asset_id")
         """
         _check_pandas_installed()
-        file_bytes = self.get_file(document_id)
+        file_bytes = self.get_file(asset_id)
         return _to_pandas_df(file_bytes, *args, **kwargs)
 
 
@@ -105,21 +105,21 @@ def _add_docs_for_async_variant(obj):
 class WrappedAsyncToolsClient(AsyncToolsClient):
 
     @_add_docs_for_async_variant(WrappedToolsClient.get_file)
-    async def get_file(self, document_id: str) -> io.BytesIO:
-        file_bytes = b''.join([gen async for gen in self.raw_data(document_id=document_id)])
+    async def get_file(self, asset_id: str) -> io.BytesIO:
+        file_bytes = b''.join([gen async for gen in self.raw_data(asset_id=asset_id)])
         bytes_io = io.BytesIO(file_bytes)
         return bytes_io
 
     @_inherit_signature_and_doc(AsyncToolsClient.data_frame, {'DataFrameRequestOut': 'pd.DataFrame'})
-    async def data_frame(self, *, document_id: str, **kwargs) -> 'pd.DataFrame':
+    async def data_frame(self, *, asset_id: str, **kwargs) -> 'pd.DataFrame':
         _check_pandas_installed()
-        model = await super().data_frame(document_id=document_id, **kwargs)
+        model = await super().data_frame(asset_id=asset_id, **kwargs)
         return _read_json_frame(model)
 
     @_add_docs_for_async_variant(WrappedToolsClient.read_data_frame)
-    async def read_data_frame(self, document_id: str, *args, **kwargs) -> 'pd.DataFrame':
+    async def read_data_frame(self, asset_id: str, *args, **kwargs) -> 'pd.DataFrame':
         _check_pandas_installed()
-        file_bytes = await self.get_file(document_id)
+        file_bytes = await self.get_file(asset_id)
         return _to_pandas_df(file_bytes, *args, **kwargs)
 
 

{athena_intelligence-0.1.98 → athena_intelligence-0.1.99}/src/athena/core/client_wrapper.py

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

{athena_intelligence-0.1.98 → athena_intelligence-0.1.99}/src/athena/tools/client.py

@@ -26,7 +26,7 @@ class ToolsClient:
     def data_frame(
         self,
         *,
-        document_id: str,
+        asset_id: str,
         row_limit: typing.Optional[int] = None,
         index_column: typing.Optional[int] = None,
         columns: typing.Optional[
@@ -39,7 +39,7 @@ class ToolsClient:
         """
         Parameters
         ----------
-        document_id : str
+        asset_id : str
 
         row_limit : typing.Optional[int]
 
@@ -70,14 +70,14 @@ class ToolsClient:
             api_key="YOUR_API_KEY",
         )
         client.tools.data_frame(
-            document_id="document_id",
+            asset_id="asset_id",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
             "api/v0/tools/file/data-frame",
             method="GET",
             params={
-                "document_id": document_id,
+                "asset_id": asset_id,
                 "row_limit": row_limit,
                 "index_column": index_column,
                 "columns": columns,
@@ -107,14 +107,14 @@ class ToolsClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     def raw_data(
-        self, *, document_id: str, request_options: typing.Optional[RequestOptions] = None
+        self, *, asset_id: str, request_options: typing.Optional[RequestOptions] = None
     ) -> typing.Iterator[bytes]:
         """
         Get the raw file data for given asset.
 
         Parameters
         ----------
-        document_id : str
+        asset_id : str
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -132,14 +132,11 @@ class ToolsClient:
             api_key="YOUR_API_KEY",
         )
         client.tools.raw_data(
-            document_id="string",
+            asset_id="string",
         )
         """
         with self._client_wrapper.httpx_client.stream(
-            "api/v0/tools/file/raw-data",
-            method="GET",
-            params={"document_id": document_id},
-            request_options=request_options,
+            "api/v0/tools/file/raw-data", method="GET", params={"asset_id": asset_id}, request_options=request_options
         ) as _response:
             if 200 <= _response.status_code < 300:
                 for _chunk in _response.iter_bytes():
@@ -166,7 +163,7 @@ class AsyncToolsClient:
     async def data_frame(
         self,
         *,
-        document_id: str,
+        asset_id: str,
         row_limit: typing.Optional[int] = None,
         index_column: typing.Optional[int] = None,
         columns: typing.Optional[
@@ -179,7 +176,7 @@ class AsyncToolsClient:
         """
         Parameters
         ----------
-        document_id : str
+        asset_id : str
 
         row_limit : typing.Optional[int]
 
@@ -210,14 +207,14 @@ class AsyncToolsClient:
             api_key="YOUR_API_KEY",
         )
         await client.tools.data_frame(
-            document_id="document_id",
+            asset_id="asset_id",
         )
         """
         _response = await self._client_wrapper.httpx_client.request(
             "api/v0/tools/file/data-frame",
             method="GET",
             params={
-                "document_id": document_id,
+                "asset_id": asset_id,
                 "row_limit": row_limit,
                 "index_column": index_column,
                 "columns": columns,
@@ -247,14 +244,14 @@ class AsyncToolsClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     async def raw_data(
-        self, *, document_id: str, request_options: typing.Optional[RequestOptions] = None
+        self, *, asset_id: str, request_options: typing.Optional[RequestOptions] = None
     ) -> typing.AsyncIterator[bytes]:
         """
         Get the raw file data for given asset.
 
         Parameters
         ----------
-        document_id : str
+        asset_id : str
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -272,14 +269,11 @@ class AsyncToolsClient:
             api_key="YOUR_API_KEY",
         )
         await client.tools.raw_data(
-            document_id="string",
+            asset_id="string",
         )
         """
         async with self._client_wrapper.httpx_client.stream(
-            "api/v0/tools/file/raw-data",
-            method="GET",
-            params={"document_id": document_id},
-            request_options=request_options,
+            "api/v0/tools/file/raw-data", method="GET", params={"asset_id": asset_id}, request_options=request_options
         ) as _response:
             if 200 <= _response.status_code < 300:
                 async for _chunk in _response.aiter_bytes():

{athena_intelligence-0.1.98 → athena_intelligence-0.1.99}/src/athena/types/data_frame_parsing_error.py

@@ -8,7 +8,7 @@ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
 
 class DataFrameParsingError(pydantic_v1.BaseModel):
-    document_id: str
+    asset_id: str
     message: str
 
     def json(self, **kwargs: typing.Any) -> str:

{athena_intelligence-0.1.98 → athena_intelligence-0.1.99}/src/athena/types/data_frame_unknown_format_error.py

@@ -8,7 +8,7 @@ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
 
 class DataFrameUnknownFormatError(pydantic_v1.BaseModel):
-    document_id: str
+    asset_id: str
     media_type: str
     message: str