athena-intelligence 0.1.31__tar.gz → 0.1.33__tar.gz

{athena_intelligence-0.1.31 → athena_intelligence-0.1.33}/PKG-INFO

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

{athena_intelligence-0.1.31 → athena_intelligence-0.1.33}/pyproject.toml

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

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

{athena_intelligence-0.1.31 → athena_intelligence-0.1.33}/src/athena/message/client.py

@@ -61,8 +61,8 @@ class MessageClient:
         )
         client.message.submit(
             content="visit www.athenaintelligence.ai and summarize the website in one paragraph",
-            model=Model.GPT_3_5_TURBO,
-            tools=[Tools.SEARCH, Tools.BROWSE, Tools.SEARCH],
+            model=Model.GPT_35_TURBO,
+            tools=[Tools.SEARCH, Tools.BROWSE],
         )
         """
         _request: typing.Dict[str, typing.Any] = {"content": content}
@@ -193,8 +193,8 @@ class AsyncMessageClient:
         )
         await client.message.submit(
             content="visit www.athenaintelligence.ai and summarize the website in one paragraph",
-            model=Model.GPT_3_5_TURBO,
-            tools=[Tools.SEARCH, Tools.BROWSE, Tools.SEARCH],
+            model=Model.GPT_35_TURBO,
+            tools=[Tools.SEARCH, Tools.BROWSE],
         )
         """
         _request: typing.Dict[str, typing.Any] = {"content": content}

athena_intelligence-0.1.33/src/athena/types/model.py

@@ -0,0 +1,61 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class Model(str, enum.Enum):
+    """
+    An enumeration.
+    """
+
+    GPT_35_TURBO = "gpt-3.5-turbo"
+    GPT_4_TURBO_PREVIEW = "gpt-4-turbo-preview"
+    GPT_4 = "gpt-4"
+    MIXTRAL_SMALL_8_X_7_B_0211 = "mixtral-small-8x7b-0211"
+    MISTRAL_LARGE_0224 = "mistral-large-0224"
+    LLAMA_V_270_B_CHAT = "llama-v2-70b-chat"
+    CLAUDE_3_OPUS_20240229 = "claude-3-opus-20240229"
+    CLAUDE_3_SONNET_20240229 = "claude-3-sonnet-20240229"
+    CLAUDE_3_HAIKU_20240307 = "claude-3-haiku-20240307"
+    GOOGLE_GEMINI_10_PRO_LATEST = "google-gemini-1.0-pro-latest"
+    DATABRICKS_DBRX = "databricks-dbrx"
+
+    def visit(
+        self,
+        gpt_35_turbo: typing.Callable[[], T_Result],
+        gpt_4_turbo_preview: 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],
+        llama_v_270_b_chat: typing.Callable[[], T_Result],
+        claude_3_opus_20240229: typing.Callable[[], T_Result],
+        claude_3_sonnet_20240229: typing.Callable[[], T_Result],
+        claude_3_haiku_20240307: typing.Callable[[], T_Result],
+        google_gemini_10_pro_latest: typing.Callable[[], T_Result],
+        databricks_dbrx: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is Model.GPT_35_TURBO:
+            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.MIXTRAL_SMALL_8_X_7_B_0211:
+            return mixtral_small_8_x_7_b_0211()
+        if self is Model.MISTRAL_LARGE_0224:
+            return mistral_large_0224()
+        if self is Model.LLAMA_V_270_B_CHAT:
+            return llama_v_270_b_chat()
+        if self is Model.CLAUDE_3_OPUS_20240229:
+            return claude_3_opus_20240229()
+        if self is Model.CLAUDE_3_SONNET_20240229:
+            return claude_3_sonnet_20240229()
+        if self is Model.CLAUDE_3_HAIKU_20240307:
+            return claude_3_haiku_20240307()
+        if self is Model.GOOGLE_GEMINI_10_PRO_LATEST:
+            return google_gemini_10_pro_latest()
+        if self is Model.DATABRICKS_DBRX:
+            return databricks_dbrx()

athena_intelligence-0.1.31/src/athena/types/model.py

@@ -1,37 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import enum
-import typing
-
-T_Result = typing.TypeVar("T_Result")
-
-
-class Model(str, enum.Enum):
-    """
-    An enumeration.
-    """
-
-    GPT_35_TURBO = "gpt-3.5-turbo"
-    GPT_4_TURBO_PREVIEW = "gpt-4-turbo-preview"
-    GPT_4 = "gpt-4"
-    MIXTRAL_SMALL_8_X_7_B_0211 = "mixtral-small-8x7b-0211"
-    MISTRAL_LARGE_0224 = "mistral-large-0224"
-
-    def visit(
-        self,
-        gpt_35_turbo: typing.Callable[[], T_Result],
-        gpt_4_turbo_preview: 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],
-    ) -> T_Result:
-        if self is Model.GPT_35_TURBO:
-            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.MIXTRAL_SMALL_8_X_7_B_0211:
-            return mixtral_small_8_x_7_b_0211()
-        if self is Model.MISTRAL_LARGE_0224:
-            return mistral_large_0224()