athena-intelligence 0.1.52__py3-none-any.whl → 0.1.53__py3-none-any.whl

athena/chain/client.py

@@ -16,7 +16,6 @@ from ..types.http_validation_error import HttpValidationError
 from ..types.llm_model import LlmModel
 from ..types.map_reduce_chain_out import MapReduceChainOut
 from ..types.structured_parse_result import StructuredParseResult
-from ..types.tool_models import ToolModels
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -98,7 +97,7 @@ class ChainClient:
         self,
         *,
         documents: typing.Sequence[Document],
-        model: ToolModels,
+        model: LlmModel,
         operator_prompt: str,
         reducer_prompt: str,
         input: str,
@@ -108,7 +107,7 @@ class ChainClient:
         Parameters:
             - documents: typing.Sequence[Document].
 
-            - model: ToolModels.
+            - model: LlmModel.
 
             - operator_prompt: str.
 
@@ -118,7 +117,7 @@ class ChainClient:
 
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
-        from athena import Document, ToolModels
+        from athena import Document, LlmModel
         from athena.client import Athena
 
         client = Athena(
@@ -130,7 +129,7 @@ class ChainClient:
                     page_content="Athena is an AI-native analytics platform and artificial employee built to accelerate analytics workflows by offering enterprise teams co-pilot and auto-pilot modes. Athena learns your workflow as a co-pilot, allowing you to hand over controls to her for autonomous execution with confidence.",
                 )
             ],
-            model=ToolModels.MISTRAL_LARGE_0224,
+            model=LlmModel.MISTRAL_LARGE_0224,
             operator_prompt="summarize the content",
             reducer_prompt="Combine these summaries",
             input="return a summary in a single sentence",
@@ -267,7 +266,7 @@ class AsyncChainClient:
         self,
         *,
         documents: typing.Sequence[Document],
-        model: ToolModels,
+        model: LlmModel,
         operator_prompt: str,
         reducer_prompt: str,
         input: str,
@@ -277,7 +276,7 @@ class AsyncChainClient:
         Parameters:
             - documents: typing.Sequence[Document].
 
-            - model: ToolModels.
+            - model: LlmModel.
 
             - operator_prompt: str.
 
@@ -287,7 +286,7 @@ class AsyncChainClient:
 
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
-        from athena import Document, ToolModels
+        from athena import Document, LlmModel
         from athena.client import AsyncAthena
 
         client = AsyncAthena(
@@ -299,7 +298,7 @@ class AsyncChainClient:
                     page_content="Athena is an AI-native analytics platform and artificial employee built to accelerate analytics workflows by offering enterprise teams co-pilot and auto-pilot modes. Athena learns your workflow as a co-pilot, allowing you to hand over controls to her for autonomous execution with confidence.",
                 )
             ],
-            model=ToolModels.MISTRAL_LARGE_0224,
+            model=LlmModel.MISTRAL_LARGE_0224,
             operator_prompt="summarize the content",
             reducer_prompt="Combine these summaries",
             input="return a summary in a single sentence",

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.52",
+            "X-Fern-SDK-Version": "0.1.53",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

{athena_intelligence-0.1.52.dist-info → athena_intelligence-0.1.53.dist-info}/METADATA

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

{athena_intelligence-0.1.52.dist-info → athena_intelligence-0.1.53.dist-info}/RECORD

@@ -1,11 +1,11 @@
 athena/__init__.py,sha256=PC9w42UGxMaPyahJJmpdqFI2bHiO_UCwKzb3qDp-eNA,1595
 athena/base_client.py,sha256=aniFuLr3wzh9PChzWZrkKV3ivSjOnS58p9g11dMXzgI,6860
 athena/chain/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
-athena/chain/client.py,sha256=sG2nlODFv5RNTjJMyGvXGKx0-Q5yD4S3mug28N7o_gk,16236
+athena/chain/client.py,sha256=2vSu7d4RvgbGc7jbWpKkCs5dU-ryCIJ1i0I1EsoCEdQ,16177
 athena/client.py,sha256=8QypiDlbZ0C1YsJh6GzhylLVCZXDQc1MCJTURo2_vvI,3576
 athena/core/__init__.py,sha256=1pNSKkwyQvMl_F0wohBqmoQAITptg3zlvCwsoSSzy7c,853
 athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-athena/core/client_wrapper.py,sha256=3MA97Atj6XCLxQXBax1jdhiwBaq5eY-EPfAEBmAE-jA,1495
+athena/core/client_wrapper.py,sha256=k1GLaDbWSvPxjvxJi4BAD9ik3hGxVf0o_vAI0F7AozM,1495
 athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 athena/core/file.py,sha256=sy1RUGZ3aJYuw998bZytxxo6QdgKmlnlgBaMvwEKCGg,1480
 athena/core/http_client.py,sha256=5ok6hqgZDJhg57EHvMnr0BBaHdG50QxFPKaCZ9aVWTc,5059
@@ -61,6 +61,6 @@ athena/types/url_result.py,sha256=lIgnQeyKy_UfFFPe7HMrrRzb-SK089RxcKcKN9Q3DNQ,87
 athena/types/validation_error.py,sha256=yqombbKLBSzTPFn6CJH_hbo7tpS68T3JvMdd7kBtO1g,972
 athena/types/validation_error_loc_item.py,sha256=LAtjCHIllWRBFXvAZ5QZpp7CPXjdtN9EB7HrLVo6EP0,128
 athena/version.py,sha256=8aYAOJtVLaJLpRp6mTiEIhnl8gXA7yE0aDtZ-3mKQ4k,87
-athena_intelligence-0.1.52.dist-info/METADATA,sha256=tgMC4l6VOKI1i8cpVNUZR6yhKE-dnyWBKdOjmUjETns,4738
-athena_intelligence-0.1.52.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.52.dist-info/RECORD,,
+athena_intelligence-0.1.53.dist-info/METADATA,sha256=OcBG4Yy1co-2Em2MrnPWmfG4JYhkehndzWyVmHTr-m4,4738
+athena_intelligence-0.1.53.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.53.dist-info/RECORD,,