athena-intelligence 0.1.90__tar.gz → 0.1.92__tar.gz

{athena_intelligence-0.1.90 → athena_intelligence-0.1.92}/PKG-INFO

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

{athena_intelligence-0.1.90 → athena_intelligence-0.1.92}/pyproject.toml

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

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

{athena_intelligence-0.1.90 → athena_intelligence-0.1.92}/src/athena/types/document.py

@@ -10,10 +10,22 @@ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 class Document(pydantic_v1.BaseModel):
     """
     Class for storing a piece of text and associated metadata.
+
+    Example:
+
+        .. code-block:: python
+
+            from langchain_core.documents import Document
+
+            document = Document(
+                page_content="Hello, world!",
+                metadata={"source": "https://example.com"}
+            )
     """
 
-    page_content: str
+    id: typing.Optional[str] = None
     metadata: typing.Optional[typing.Dict[str, typing.Any]] = None
+    page_content: str
     type: typing.Optional[typing.Literal["Document"]] = None
 
     def json(self, **kwargs: typing.Any) -> str:

{athena_intelligence-0.1.90 → athena_intelligence-0.1.92}/src/athena/types/llm_model.py

@@ -16,6 +16,7 @@ class LlmModel(str, enum.Enum):
     GPT_4_TURBO_PREVIEW = "gpt-4-turbo-preview"
     GPT_4_O_MINI = "gpt-4o-mini"
     GPT_4_O = "gpt-4o"
+    GPT_4_O_20240806 = "gpt-4o-2024-08-06"
     GPT_4 = "gpt-4"
     MIXTRAL_SMALL_8_X_7_B_0211 = "mixtral-small-8x7b-0211"
     MISTRAL_LARGE_0224 = "mistral-large-0224"
@@ -49,6 +50,7 @@ class LlmModel(str, enum.Enum):
         gpt_4_turbo_preview: typing.Callable[[], T_Result],
         gpt_4_o_mini: typing.Callable[[], T_Result],
         gpt_4_o: typing.Callable[[], T_Result],
+        gpt_4_o_20240806: typing.Callable[[], T_Result],
         gpt_4: typing.Callable[[], T_Result],
         mixtral_small_8_x_7_b_0211: typing.Callable[[], T_Result],
         mistral_large_0224: typing.Callable[[], T_Result],
@@ -85,6 +87,8 @@ class LlmModel(str, enum.Enum):
             return gpt_4_o_mini()
         if self is LlmModel.GPT_4_O:
             return gpt_4_o()
+        if self is LlmModel.GPT_4_O_20240806:
+            return gpt_4_o_20240806()
         if self is LlmModel.GPT_4:
             return gpt_4()
         if self is LlmModel.MIXTRAL_SMALL_8_X_7_B_0211:

{athena_intelligence-0.1.90 → athena_intelligence-0.1.92}/src/athena/types/model.py

@@ -17,6 +17,7 @@ class Model(str, enum.Enum):
     GPT_4 = "gpt-4"
     GPT_4_O = "gpt-4o"
     GPT_4_O_MINI = "gpt-4o-mini"
+    GPT_4_O_20240806 = "gpt-4o-2024-08-06"
     MIXTRAL_SMALL_8_X_7_B_0211 = "mixtral-small-8x7b-0211"
     MISTRAL_LARGE_0224 = "mistral-large-0224"
     MIXTRAL_8_X_22_B_INSTRUCT = "mixtral-8x22b-instruct"
@@ -38,6 +39,7 @@ class Model(str, enum.Enum):
         gpt_4: typing.Callable[[], T_Result],
         gpt_4_o: typing.Callable[[], T_Result],
         gpt_4_o_mini: typing.Callable[[], T_Result],
+        gpt_4_o_20240806: typing.Callable[[], T_Result],
         mixtral_small_8_x_7_b_0211: typing.Callable[[], T_Result],
         mistral_large_0224: typing.Callable[[], T_Result],
         mixtral_8_x_22_b_instruct: typing.Callable[[], T_Result],
@@ -63,6 +65,8 @@ class Model(str, enum.Enum):
             return gpt_4_o()
         if self is Model.GPT_4_O_MINI:
             return gpt_4_o_mini()
+        if self is Model.GPT_4_O_20240806:
+            return gpt_4_o_20240806()
         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.90 → athena_intelligence-0.1.92}/src/athena/types/tools.py

@@ -20,6 +20,7 @@ class Tools(str, enum.Enum):
     ENRICH_PERSON = "enrich_person"
     ENRICH_COMPANY = "enrich_company"
     BROWSERBASE = "browserbase"
+    CODE_EXECUTION = "code_execution"
 
     def visit(
         self,
@@ -32,6 +33,7 @@ class Tools(str, enum.Enum):
         enrich_person: typing.Callable[[], T_Result],
         enrich_company: typing.Callable[[], T_Result],
         browserbase: typing.Callable[[], T_Result],
+        code_execution: typing.Callable[[], T_Result],
     ) -> T_Result:
         if self is Tools.SEARCH:
             return search()
@@ -51,3 +53,5 @@ class Tools(str, enum.Enum):
             return enrich_company()
         if self is Tools.BROWSERBASE:
             return browserbase()
+        if self is Tools.CODE_EXECUTION:
+            return code_execution()

{athena_intelligence-0.1.90 → athena_intelligence-0.1.92}/src/athena/chain/client.py

@@ -124,8 +124,8 @@ class ChainClient:
         client.chain.map_reduce_chain(
             documents=[
                 Document(
-                    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.",
                     metadata={"key": "value"},
+                    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=LlmModel.MISTRAL_LARGE_0224,
@@ -266,8 +266,8 @@ class AsyncChainClient:
         await client.chain.map_reduce_chain(
             documents=[
                 Document(
-                    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.",
                     metadata={"key": "value"},
+                    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=LlmModel.MISTRAL_LARGE_0224,