athena-intelligence 0.1.32__tar.gz → 0.1.34__tar.gz

{athena_intelligence-0.1.32 → athena_intelligence-0.1.34}/PKG-INFO

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

{athena_intelligence-0.1.32 → athena_intelligence-0.1.34}/pyproject.toml

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

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

{athena_intelligence-0.1.32 → athena_intelligence-0.1.34}/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.32 → athena_intelligence-0.1.34}/src/athena/types/model.py

@@ -20,6 +20,8 @@ class Model(str, enum.Enum):
     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,
@@ -32,6 +34,8 @@ class Model(str, enum.Enum):
         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()
@@ -51,3 +55,7 @@ class Model(str, enum.Enum):
             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.32 → athena_intelligence-0.1.34}/src/athena/types/tools.py

@@ -13,6 +13,7 @@ class Tools(str, enum.Enum):
 
     SEARCH = "search"
     BROWSE = "browse"
+    WIKIPEDIA = "wikipedia"
     DATABASE_METADATA = "database_metadata"
     QUERY = "query"
     CHART = "chart"
@@ -23,6 +24,7 @@ class Tools(str, enum.Enum):
         self,
         search: typing.Callable[[], T_Result],
         browse: typing.Callable[[], T_Result],
+        wikipedia: typing.Callable[[], T_Result],
         database_metadata: typing.Callable[[], T_Result],
         query: typing.Callable[[], T_Result],
         chart: typing.Callable[[], T_Result],
@@ -33,6 +35,8 @@ class Tools(str, enum.Enum):
             return search()
         if self is Tools.BROWSE:
             return browse()
+        if self is Tools.WIKIPEDIA:
+            return wikipedia()
         if self is Tools.DATABASE_METADATA:
             return database_metadata()
         if self is Tools.QUERY: