athena-intelligence 0.1.66__tar.gz → 0.1.68__tar.gz

{athena_intelligence-0.1.66 → athena_intelligence-0.1.68}/PKG-INFO

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

{athena_intelligence-0.1.66 → athena_intelligence-0.1.68}/pyproject.toml

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

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

{athena_intelligence-0.1.66 → athena_intelligence-0.1.68}/src/athena/types/llm_model.py

@@ -28,9 +28,11 @@ class LlmModel(str, enum.Enum):
     GROQ_LLAMA_38_B_8192 = "groq-llama3-8b-8192"
     GROQ_LLAMA_370_B_8192 = "groq-llama3-70b-8192"
     GROQ_GEMMA_7_B_IT = "groq-gemma-7b-it"
-    GOOGLE_GEMINI_10_PRO_LATEST = "google-gemini-1.0-pro-latest"
     DATABRICKS_DBRX = "databricks-dbrx"
+    GOOGLE_GEMINI_10_PRO_LATEST = "google-gemini-1.0-pro-latest"
     GOOGLE_GEMINI_15_PRO_LATEST = "google-gemini-1.5-pro-latest"
+    GOOGLE_GEMINI_15_FLASH_001 = "google-gemini-1.5-flash-001"
+    GOOGLE_GEMINI_15_PRO_001 = "google-gemini-1.5-pro-001"
 
     def visit(
         self,
@@ -51,9 +53,11 @@ class LlmModel(str, enum.Enum):
         groq_llama_38_b_8192: typing.Callable[[], T_Result],
         groq_llama_370_b_8192: typing.Callable[[], T_Result],
         groq_gemma_7_b_it: typing.Callable[[], T_Result],
-        google_gemini_10_pro_latest: typing.Callable[[], T_Result],
         databricks_dbrx: typing.Callable[[], T_Result],
+        google_gemini_10_pro_latest: typing.Callable[[], T_Result],
         google_gemini_15_pro_latest: typing.Callable[[], T_Result],
+        google_gemini_15_flash_001: typing.Callable[[], T_Result],
+        google_gemini_15_pro_001: typing.Callable[[], T_Result],
     ) -> T_Result:
         if self is LlmModel.GPT_35_TURBO:
             return gpt_35_turbo()
@@ -89,9 +93,13 @@ class LlmModel(str, enum.Enum):
             return groq_llama_370_b_8192()
         if self is LlmModel.GROQ_GEMMA_7_B_IT:
             return groq_gemma_7_b_it()
-        if self is LlmModel.GOOGLE_GEMINI_10_PRO_LATEST:
-            return google_gemini_10_pro_latest()
         if self is LlmModel.DATABRICKS_DBRX:
             return databricks_dbrx()
+        if self is LlmModel.GOOGLE_GEMINI_10_PRO_LATEST:
+            return google_gemini_10_pro_latest()
         if self is LlmModel.GOOGLE_GEMINI_15_PRO_LATEST:
             return google_gemini_15_pro_latest()
+        if self is LlmModel.GOOGLE_GEMINI_15_FLASH_001:
+            return google_gemini_15_flash_001()
+        if self is LlmModel.GOOGLE_GEMINI_15_PRO_001:
+            return google_gemini_15_pro_001()