athena-intelligence 0.1.28__py3-none-any.whl → 0.1.30__py3-none-any.whl

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/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/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/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/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/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/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:

{athena_intelligence-0.1.28.dist-info → athena_intelligence-0.1.30.dist-info}/METADATA

@@ -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.dist-info → athena_intelligence-0.1.30.dist-info}/RECORD

@@ -3,7 +3,7 @@ athena/base_client.py,sha256=0V7k8HeQrYl1dK2lni53I0EkrmMwFiTVW5Pdt-fScuE,4806
 athena/client.py,sha256=8QypiDlbZ0C1YsJh6GzhylLVCZXDQc1MCJTURo2_vvI,3576
 athena/core/__init__.py,sha256=RWfyDqkzWsf8e3VGc3NV60MovfJbg5XWzNFGB2DZ0hA,790
 athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-athena/core/client_wrapper.py,sha256=bQWP180vgBFbVegDjeeJSFJ5DlrAG_fNek0bSz7VY_Q,1198
+athena/core/client_wrapper.py,sha256=KZ7FY-ZxhFOGj0zURiA5iNdT6Wndof8UYC71q6wRYKY,1198
 athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 athena/core/file.py,sha256=sy1RUGZ3aJYuw998bZytxxo6QdgKmlnlgBaMvwEKCGg,1480
 athena/core/http_client.py,sha256=LI0yP3jUyE0Ue7oyBcI9nyo1pljOwh9Y5ycTeIpKwOg,4882
@@ -11,7 +11,7 @@ athena/core/jsonable_encoder.py,sha256=MTYkDov2EryHgee4QM46uZiBOuOXK9KTHlBdBwU-C
 athena/core/remove_none_from_dict.py,sha256=8m91FC3YuVem0Gm9_sXhJ2tGvP33owJJdrqCLEdowGw,330
 athena/core/request_options.py,sha256=-3QoOMMHI2exIyHH6Q2MD7rpo_6w-i6zMAy0nqWTN8c,1420
 athena/dataset/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
-athena/dataset/client.py,sha256=pohi3ugsXM6BKz4iN__cIKaqCLGlZDkzdaxDEgYnBLY,4472
+athena/dataset/client.py,sha256=ilxpF0K9AlnGAQjrumdqZ1V1Wu3ARoM-06iDE5gQiCQ,6087
 athena/environment.py,sha256=D_CljQlUahhEi9smvMslj_5Y8gMFO6D0fRCL0ydRLuM,165
 athena/errors/__init__.py,sha256=pbbVUFtB9LCocA1RMWMMF_RKjsy5YkOKX5BAuE49w6g,170
 athena/errors/unprocessable_entity_error.py,sha256=FvR7XPlV3Xx5nu8HNlmLhBRdk4so_gCHjYT5PyZe6sM,313
@@ -24,15 +24,15 @@ athena/query/client.py,sha256=UOx-Bq-xFFm-sTMTmJjWGrC6q_7vhVno3nYzmi81xwI,6243
 athena/report/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 athena/report/client.py,sha256=sGJDrgk_E1SPleRYNhvspmsz-G3FQwMW-3alFzZPquE,6528
 athena/snippet/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
-athena/snippet/client.py,sha256=O1Q7ytPNEI4E1cGgu8Dv2oAzrGMhAEsmMb0LXOvsSSw,4472
+athena/snippet/client.py,sha256=D0rSpm6ql9cnUj-mMe3z8OHRgRQQuk3bBW2CZSRnyp4,6087
 athena/types/__init__.py,sha256=kSz8xv8p-5znJI-sg3b6yD0E0bi9pkI1J8F60D5hu78,918
-athena/types/dataset.py,sha256=LqLZn_Br9jGg4toU_6wMoZQMIgAea6Hdn11xJpgi_M8,966
-athena/types/get_datasets_response.py,sha256=gikRLR9f9jQSfbMJedBh9JicH_9ccJ851nUmIZXfEvg,926
-athena/types/get_snippets_response.py,sha256=SJEWXhPHvZ0-72M8WckIogn-7w4v20oeSqYaXil85ic,926
+athena/types/dataset.py,sha256=70OJPxKBAYu7xthGEgrUolSdyLqiyh6X49INw1oN0sA,1014
+athena/types/get_datasets_response.py,sha256=BCdT8yTLfOsXeyFadlyoas4zzseFWGPAdGpkgkOuaD8,989
+athena/types/get_snippets_response.py,sha256=Lpn7bHJLpPQozN93unCV-8eByAAfz1MhQWR3G3Z1vl4,989
 athena/types/http_validation_error.py,sha256=Fcv_CTMMrLvCeTHjF0n5xf5tskMDgt-J6H9gp654eQw,973
 athena/types/message_out.py,sha256=uvZY_Podv2XccEk8CICug9I_S2hFJTSzCBwcHiauW7A,865
 athena/types/message_out_dto.py,sha256=qgRibRbDNOWVnVGP7Rribh9WdoCT2CSiPUXeIWECqq4,1051
-athena/types/model.py,sha256=yZyzOojHN4mRCd-ITHJNiKZDpQhy7LKZcM9_saJ0qy4,1020
+athena/types/model.py,sha256=hwksocGCn7CTMDUkxkc--sHs8qqm8WzC8urIbeGsX-A,1290
 athena/types/report.py,sha256=QVaqVfHMAV3s9_V2CqjIEMcRrbJhD8zmi82vrk2A8x0,946
 athena/types/snippet.py,sha256=POIVJNV9iQxiVegB_qwQx-PZPPSyoIPhyxTsueNVUGA,1126
 athena/types/sql_results.py,sha256=pNH32nyf1bzoYJs3FgHctLdLO02oOjyGgLkHACACB6k,900
@@ -40,6 +40,6 @@ athena/types/status_enum.py,sha256=0UZbhdAx215GHC-U53RS98mYHtn1N3On4VBe4j02Qtc,6
 athena/types/tools.py,sha256=Vfb-D9CYcPZH6MBATNODgfXjFyBpCs4qbkqpCMl7eBM,1277
 athena/types/validation_error.py,sha256=2JhGNJouo8QpfrMBoT_JCwYSn1nFN2Nnq0p9uPLDH-U,992
 athena/types/validation_error_loc_item.py,sha256=LAtjCHIllWRBFXvAZ5QZpp7CPXjdtN9EB7HrLVo6EP0,128
-athena_intelligence-0.1.28.dist-info/METADATA,sha256=24HLIs60fkaa29atNRu_yaQH_PC3RBYAgGYUHo9H_B8,4738
-athena_intelligence-0.1.28.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.28.dist-info/RECORD,,
+athena_intelligence-0.1.30.dist-info/METADATA,sha256=_0tdDHwC0b-A4kpjYlilzm-biSIGT7yBlKWeCq2ldYk,4738
+athena_intelligence-0.1.30.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.30.dist-info/RECORD,,