athena-intelligence 0.1.28__tar.gz → 0.1.30__tar.gz

{athena_intelligence-0.1.28 → athena_intelligence-0.1.30}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: athena-intelligence
-Version: 0.1.28
+Version: 0.1.30
 Summary: 
 Requires-Python: >=3.8,<4.0
 Classifier: Programming Language :: Python :: 3

{athena_intelligence-0.1.28 → athena_intelligence-0.1.30}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "athena-intelligence"
-version = "0.1.28"
+version = "0.1.30"
 description = ""
 readme = "README.md"
 authors = []

{athena_intelligence-0.1.28 → athena_intelligence-0.1.30}/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.28",
+            "X-Fern-SDK-Version": "0.1.30",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

{athena_intelligence-0.1.28 → athena_intelligence-0.1.30}/src/athena/dataset/client.py

@@ -9,7 +9,9 @@ from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.jsonable_encoder import jsonable_encoder
 from ..core.remove_none_from_dict import remove_none_from_dict
 from ..core.request_options import RequestOptions
+from ..errors.unprocessable_entity_error import UnprocessableEntityError
 from ..types.get_datasets_response import GetDatasetsResponse
+from ..types.http_validation_error import HttpValidationError
 
 try:
     import pydantic.v1 as pydantic  # type: ignore
@@ -21,9 +23,19 @@ class DatasetClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def get(self, *, request_options: typing.Optional[RequestOptions] = None) -> GetDatasetsResponse:
+    def get(
+        self,
+        *,
+        page: typing.Optional[int] = None,
+        page_size: typing.Optional[int] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> GetDatasetsResponse:
         """
         Parameters:
+            - page: typing.Optional[int]. Page number starting from 1
+
+            - page_size: typing.Optional[int]. Number of items per page
+
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
         from athena.client import Athena
@@ -37,7 +49,17 @@ class DatasetClient:
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/datasets"),
             params=jsonable_encoder(
-                request_options.get("additional_query_parameters") if request_options is not None else None
+                remove_none_from_dict(
+                    {
+                        "page": page,
+                        "page_size": page_size,
+                        **(
+                            request_options.get("additional_query_parameters", {})
+                            if request_options is not None
+                            else {}
+                        ),
+                    }
+                )
             ),
             headers=jsonable_encoder(
                 remove_none_from_dict(
@@ -55,6 +77,8 @@ class DatasetClient:
         )
         if 200 <= _response.status_code < 300:
             return pydantic.parse_obj_as(GetDatasetsResponse, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:
@@ -66,9 +90,19 @@ class AsyncDatasetClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def get(self, *, request_options: typing.Optional[RequestOptions] = None) -> GetDatasetsResponse:
+    async def get(
+        self,
+        *,
+        page: typing.Optional[int] = None,
+        page_size: typing.Optional[int] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> GetDatasetsResponse:
         """
         Parameters:
+            - page: typing.Optional[int]. Page number starting from 1
+
+            - page_size: typing.Optional[int]. Number of items per page
+
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
         from athena.client import AsyncAthena
@@ -82,7 +116,17 @@ class AsyncDatasetClient:
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/datasets"),
             params=jsonable_encoder(
-                request_options.get("additional_query_parameters") if request_options is not None else None
+                remove_none_from_dict(
+                    {
+                        "page": page,
+                        "page_size": page_size,
+                        **(
+                            request_options.get("additional_query_parameters", {})
+                            if request_options is not None
+                            else {}
+                        ),
+                    }
+                )
             ),
             headers=jsonable_encoder(
                 remove_none_from_dict(
@@ -100,6 +144,8 @@ class AsyncDatasetClient:
         )
         if 200 <= _response.status_code < 300:
             return pydantic.parse_obj_as(GetDatasetsResponse, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:

{athena_intelligence-0.1.28 → athena_intelligence-0.1.30}/src/athena/snippet/client.py

@@ -9,7 +9,9 @@ from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.jsonable_encoder import jsonable_encoder
 from ..core.remove_none_from_dict import remove_none_from_dict
 from ..core.request_options import RequestOptions
+from ..errors.unprocessable_entity_error import UnprocessableEntityError
 from ..types.get_snippets_response import GetSnippetsResponse
+from ..types.http_validation_error import HttpValidationError
 
 try:
     import pydantic.v1 as pydantic  # type: ignore
@@ -21,9 +23,19 @@ class SnippetClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def get(self, *, request_options: typing.Optional[RequestOptions] = None) -> GetSnippetsResponse:
+    def get(
+        self,
+        *,
+        page: typing.Optional[int] = None,
+        page_size: typing.Optional[int] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> GetSnippetsResponse:
         """
         Parameters:
+            - page: typing.Optional[int]. Page number starting from 1
+
+            - page_size: typing.Optional[int]. Number of items per page
+
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
         from athena.client import Athena
@@ -37,7 +49,17 @@ class SnippetClient:
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/snippets"),
             params=jsonable_encoder(
-                request_options.get("additional_query_parameters") if request_options is not None else None
+                remove_none_from_dict(
+                    {
+                        "page": page,
+                        "page_size": page_size,
+                        **(
+                            request_options.get("additional_query_parameters", {})
+                            if request_options is not None
+                            else {}
+                        ),
+                    }
+                )
             ),
             headers=jsonable_encoder(
                 remove_none_from_dict(
@@ -55,6 +77,8 @@ class SnippetClient:
         )
         if 200 <= _response.status_code < 300:
             return pydantic.parse_obj_as(GetSnippetsResponse, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:
@@ -66,9 +90,19 @@ class AsyncSnippetClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def get(self, *, request_options: typing.Optional[RequestOptions] = None) -> GetSnippetsResponse:
+    async def get(
+        self,
+        *,
+        page: typing.Optional[int] = None,
+        page_size: typing.Optional[int] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> GetSnippetsResponse:
         """
         Parameters:
+            - page: typing.Optional[int]. Page number starting from 1
+
+            - page_size: typing.Optional[int]. Number of items per page
+
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
         from athena.client import AsyncAthena
@@ -82,7 +116,17 @@ class AsyncSnippetClient:
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/snippets"),
             params=jsonable_encoder(
-                request_options.get("additional_query_parameters") if request_options is not None else None
+                remove_none_from_dict(
+                    {
+                        "page": page,
+                        "page_size": page_size,
+                        **(
+                            request_options.get("additional_query_parameters", {})
+                            if request_options is not None
+                            else {}
+                        ),
+                    }
+                )
             ),
             headers=jsonable_encoder(
                 remove_none_from_dict(
@@ -100,6 +144,8 @@ class AsyncSnippetClient:
         )
         if 200 <= _response.status_code < 300:
             return pydantic.parse_obj_as(GetSnippetsResponse, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:

{athena_intelligence-0.1.28 → athena_intelligence-0.1.30}/src/athena/types/dataset.py

@@ -16,6 +16,7 @@ class Dataset(pydantic.BaseModel):
     name: typing.Optional[str] = None
     description: typing.Optional[str] = None
     database_id: int
+    schema_details: typing.Optional[str] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

{athena_intelligence-0.1.28 → athena_intelligence-0.1.30}/src/athena/types/get_datasets_response.py

@@ -14,6 +14,10 @@ except ImportError:
 
 class GetDatasetsResponse(pydantic.BaseModel):
     datasets: typing.List[Dataset]
+    total: int
+    page: int
+    page_size: int
+    pages: int
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

{athena_intelligence-0.1.28 → athena_intelligence-0.1.30}/src/athena/types/get_snippets_response.py

@@ -14,6 +14,10 @@ except ImportError:
 
 class GetSnippetsResponse(pydantic.BaseModel):
     snippets: typing.List[Snippet]
+    total: int
+    page: int
+    page_size: int
+    pages: int
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

{athena_intelligence-0.1.28 → athena_intelligence-0.1.30}/src/athena/types/model.py

@@ -13,6 +13,8 @@ class Model(str, enum.Enum):
 
     GPT_35_TURBO = "gpt-3.5-turbo"
     GPT_4_TURBO_PREVIEW = "gpt-4-turbo-preview"
+    GPT_4 = "gpt-4"
+    GPT_432_K = "gpt-4-32k"
     MIXTRAL_SMALL_8_X_7_B_0211 = "mixtral-small-8x7b-0211"
     MISTRAL_LARGE_0224 = "mistral-large-0224"
 
@@ -20,6 +22,8 @@ class Model(str, enum.Enum):
         self,
         gpt_35_turbo: typing.Callable[[], T_Result],
         gpt_4_turbo_preview: typing.Callable[[], T_Result],
+        gpt_4: typing.Callable[[], T_Result],
+        gpt_432_k: typing.Callable[[], T_Result],
         mixtral_small_8_x_7_b_0211: typing.Callable[[], T_Result],
         mistral_large_0224: typing.Callable[[], T_Result],
     ) -> T_Result:
@@ -27,6 +31,10 @@ class Model(str, enum.Enum):
             return gpt_35_turbo()
         if self is Model.GPT_4_TURBO_PREVIEW:
             return gpt_4_turbo_preview()
+        if self is Model.GPT_4:
+            return gpt_4()
+        if self is Model.GPT_432_K:
+            return gpt_432_k()
         if self is Model.MIXTRAL_SMALL_8_X_7_B_0211:
             return mixtral_small_8_x_7_b_0211()
         if self is Model.MISTRAL_LARGE_0224: