athena-intelligence 0.1.51__tar.gz → 0.1.53__tar.gz

{athena_intelligence-0.1.51 → athena_intelligence-0.1.53}/PKG-INFO

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

{athena_intelligence-0.1.51 → athena_intelligence-0.1.53}/pyproject.toml

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

{athena_intelligence-0.1.51 → athena_intelligence-0.1.53}/src/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_intelligence-0.1.51 → athena_intelligence-0.1.53}/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.51",
+            "X-Fern-SDK-Version": "0.1.53",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

{athena_intelligence-0.1.51 → athena_intelligence-0.1.53}/src/athena/types/langchain_documents_request_out.py

@@ -5,11 +5,10 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from ..core.pydantic_utilities import pydantic_v1
-from .document import Document
 
 
 class LangchainDocumentsRequestOut(pydantic_v1.BaseModel):
-    documents: typing.List[Document]
+    output: typing.Dict[str, typing.Any]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}