athena-intelligence 0.1.90__py3-none-any.whl → 0.1.92__py3-none-any.whl

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,

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/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/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/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/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.dist-info → athena_intelligence-0.1.92.dist-info}/METADATA

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

@@ -1,11 +1,11 @@
 athena/__init__.py,sha256=vS1F7G3hdiHGM-q24LBoKf77EHpaDBrb3ZzM_sgXoGY,2801
 athena/base_client.py,sha256=yP2dIc8ja04pFizBx5xF4qhkeOPJ3V9DMfqYcW5HIGg,7119
 athena/chain/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
-athena/chain/client.py,sha256=5YJJsXnXLFbPN1y0Qe__QWoQ2QOMp6zK_is_obk7Gg8,11105
+athena/chain/client.py,sha256=lM0hdPC-JatI_o09NvGVBLi906V_XUwQ4Tb_FKrSAgk,11105
 athena/client.py,sha256=piHOo-a8KiuQnBCsJ509zSRahXd6DCaxkzaqqTbNWME,9680
 athena/core/__init__.py,sha256=UFXpYzcGxWQUucU1TkjOQ9mGWN3A5JohluOIWVYKU4I,973
 athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-athena/core/client_wrapper.py,sha256=ADV3_Uf4JxBw9Vczm9GQW32We6mpjtI9ahr081hxo6g,1805
+athena/core/client_wrapper.py,sha256=8NlNy6mCnsYvDhUh7mTSpgHKoIK7qdTu7q45J2Oyj2A,1805
 athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 athena/core/file.py,sha256=sy1RUGZ3aJYuw998bZytxxo6QdgKmlnlgBaMvwEKCGg,1480
 athena/core/http_client.py,sha256=Z4NuAsJD-51yqmoME17O5sxwx5orSp1wsnd6bPyKcgA,17768
@@ -48,7 +48,7 @@ athena/types/data_frame_request_out_data_item_item.py,sha256=KMTJRr-1bdKDNMbUeri
 athena/types/data_frame_request_out_index_item.py,sha256=bW7oe912trpkYKodj-I_AiTXXy61yWzliekcsUZkZE0,141
 athena/types/data_frame_unknown_format_error.py,sha256=g6Q-TJ9sbtYRuDhgy7fBv_vbQawtil5AIRX0IlU8U4E,1171
 athena/types/dataset.py,sha256=7uxfEhuHSk63SWyJO8pvxuFr-s6QR_tQvJdp5XUZP4Q,1257
-athena/types/document.py,sha256=XGeQXtpMwM3VUH9b5m_mormocKlIy8VAAXSi6OgQNbs,1324
+athena/types/document.py,sha256=HF_Cn04RvbKTD6gQws78AJyS1k1KUK98xvlpXujUewQ,1618
 athena/types/excecute_tool_first_workflow_out.py,sha256=F0tkk2uEXNZaVeKPIxA68Mdwy__VcUjfVQVs_8dv_dk,1138
 athena/types/file_data_response.py,sha256=3BNJCDo4VL6q9ksp2jDOkAP_Sp6e6eCsRjqyWGnqSa0,1434
 athena/types/file_fetch_error.py,sha256=iR7IXjj2C4lJFcWV3aS2gfhmT3CE-_hhBDSWFaCktHU,1162
@@ -61,11 +61,11 @@ athena/types/get_snippet_out.py,sha256=Dcy2rZhie8_X41r4lmE5W2R5Jd9L2GnDKJv39emKU
 athena/types/get_snippets_response.py,sha256=c7xxCp71lAxXjuKMavXAUw91UXAJGLSoWnGkqWfbuO0,1232
 athena/types/http_validation_error.py,sha256=ZuVHc-T6RFEo7_qBO8DWHa2jGJ6O--tUIZynmLFezsI,1216
 athena/types/langchain_documents_request_out.py,sha256=UUcByVZb6SVwhTtAScRey1FROpZitHEPsamAkdLbuhI,1155
-athena/types/llm_model.py,sha256=kGCC7WEAe-Lq7r6kiXhy5UCn6vdSdb7ydbq5gIjeye0,6666
+athena/types/llm_model.py,sha256=TnJnS1zT1xr0u2ht-uVlMyLLDDivpnwmEIY_stFAgbk,6850
 athena/types/map_reduce_chain_out.py,sha256=p9rN21wMZaki_rpd1Z_0DM4tvlDdYG8gHE2Xg4TEAvE,1213
 athena/types/message_out.py,sha256=Jli5sJBAtwOgkZjpx28kVHfBRoBLuINyNTm_9AXPuys,1108
 athena/types/message_out_dto.py,sha256=iELbR1DZBrlSshzIdfpZ3ZR2V1fzAzVw4aprxcfzg38,1294
-athena/types/model.py,sha256=2dpNftSHkrgeUl-00Zk3UZ1tSkeygOfimHFDu5mc3ek,3746
+athena/types/model.py,sha256=cK7bI1QdJrm6rO_nfQurDMojwucuYEOKpqSW8LWC15Y,3927
 athena/types/publish_formats.py,sha256=1j2fgJo0MMaHzEEYxd9uCFq_Y7bFoQox0JgKCfcPXLo,1148
 athena/types/query_model.py,sha256=F-HSvgcTys4hA3e7sYas_71APMZmAu_LOKqWVZ0-Jx4,1684
 athena/types/report.py,sha256=Yr_Ph7BS-2QgMJ3az5GP_fwlRMl6i4-1MGWYcLPSPhY,1189
@@ -76,7 +76,7 @@ athena/types/sql_results.py,sha256=ikxHio7BRnWqNIwdoZcoVOf3Un64Kn9qLAHiIjOVwiQ,1
 athena/types/status_enum.py,sha256=0UZbhdAx215GHC-U53RS98mYHtn1N3On4VBe4j02Qtc,672
 athena/types/structured_parse_result.py,sha256=BFjA9R20kTdiRaPpQd7tEmLuSCO9ux1SF7rDziq61nY,1148
 athena/types/time_dimension_model.py,sha256=Kp-s5Xd05JIyLzGG8X9eop5e0P3yi0S1bygW4BXApKI,1294
-athena/types/tools.py,sha256=W0ekZrKpwlf66HJC7kGLWYJE3C1agJRnmMbvfA4M93o,1577
+athena/types/tools.py,sha256=OrWzaeeBEVk2qsttuRNaStPq7XhRuitydA1oJVqJqPE,1747
 athena/types/upload_documents_out.py,sha256=NObG9duFoIyCTPggNs_arCkNybEvlsmEFNkeGTcHH3s,1249
 athena/types/url_result.py,sha256=y_77BhRPQM7a1ugJpDHNDbNv1e2T79Evm9y-0Y4wrr8,1136
 athena/types/validation_error.py,sha256=zub9I6bl192KHCE8RA0pEkdaEODR3ktTFnZ7E1B7Eco,1235
@@ -87,6 +87,6 @@ athena/upload/client.py,sha256=Z44mBN3Vg_HG4uOqrzBkkNhCGETjsAlbU1RA50KDxkQ,3897
 athena/version.py,sha256=8aYAOJtVLaJLpRp6mTiEIhnl8gXA7yE0aDtZ-3mKQ4k,87
 athena/workflow/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 athena/workflow/client.py,sha256=1WGH4MCnwae5Wgb7pGfiyZRr--HscVpdHxeouNSyPe0,4023
-athena_intelligence-0.1.90.dist-info/METADATA,sha256=0GdM4W4QOMBcDGBM97xrlEDacHScr7tDV2WgXEcwVFc,5273
-athena_intelligence-0.1.90.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.90.dist-info/RECORD,,
+athena_intelligence-0.1.92.dist-info/METADATA,sha256=VEY1PlfFI-gaJHq5Jriomvzm1YgPmmJbuRSSKwhLtOw,5273
+athena_intelligence-0.1.92.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.92.dist-info/RECORD,,