Release v0.2.36

superagent-ai · May 3, 2024 · 2acd903 · 2acd903
1 parent e774051
commit 2acd903
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "superagent-py"
-version = "v0.2.35"
+version = "v0.2.36"
 description = ""
 readme = "README.md"
 authors = []

diff --git a/src/superagent/core/client_wrapper.py b/src/superagent/core/client_wrapper.py
@@ -16,7 +16,7 @@ def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "superagent-py",
-            "X-Fern-SDK-Version": "v0.2.35",
+            "X-Fern-SDK-Version": "v0.2.36",
         }
         token = self._get_token()
         if token is not None:

diff --git a/src/superagent/types/llm_model.py b/src/superagent/types/llm_model.py
@@ -11,30 +11,44 @@ class LlmModel(str, enum.Enum):
     An enumeration.
     """
 
+    GPT_3_5_TURBO = "GPT_3_5_TURBO"
     GPT_3_5_TURBO_16_K_0613 = "GPT_3_5_TURBO_16K_0613"
     GPT_3_5_TURBO_0613 = "GPT_3_5_TURBO_0613"
     GPT_3_5_TURBO_1106 = "GPT_3_5_TURBO_1106"
     GPT_3_5_TURBO_0125 = "GPT_3_5_TURBO_0125"
+    GPT_4 = "GPT_4"
     GPT_4_0613 = "GPT_4_0613"
+    GPT_4_32_K = "GPT_4_32K"
     GPT_4_32_K_0613 = "GPT_4_32K_0613"
     GPT_4_1106_PREVIEW = "GPT_4_1106_PREVIEW"
+    GPT_4_0125_PREVIEW = "GPT_4_0125_PREVIEW"
+    GPT_4_TURBO = "GPT_4_TURBO"
     GPT_4_TURBO_PREVIEW = "GPT_4_TURBO_PREVIEW"
+    GPT_4_TURBO_2024_04_09 = "GPT_4_TURBO_2024_04_09"
     MISTRAL_7_B_INSTRUCT_V_01 = "MISTRAL_7B_INSTRUCT_V01"
     MIXTRAL_8_X_7_B_INSTRUCT_V_01 = "MIXTRAL_8X7B_INSTRUCT_V01"
 
     def visit(
         self,
+        gpt_3_5_turbo: typing.Callable[[], T_Result],
         gpt_3_5_turbo_16_k_0613: typing.Callable[[], T_Result],
         gpt_3_5_turbo_0613: typing.Callable[[], T_Result],
         gpt_3_5_turbo_1106: typing.Callable[[], T_Result],
         gpt_3_5_turbo_0125: typing.Callable[[], T_Result],
+        gpt_4: typing.Callable[[], T_Result],
         gpt_4_0613: typing.Callable[[], T_Result],
+        gpt_4_32_k: typing.Callable[[], T_Result],
         gpt_4_32_k_0613: typing.Callable[[], T_Result],
         gpt_4_1106_preview: typing.Callable[[], T_Result],
+        gpt_4_0125_preview: typing.Callable[[], T_Result],
+        gpt_4_turbo: typing.Callable[[], T_Result],
         gpt_4_turbo_preview: typing.Callable[[], T_Result],
+        gpt_4_turbo_2024_04_09: typing.Callable[[], T_Result],
         mistral_7_b_instruct_v_01: typing.Callable[[], T_Result],
         mixtral_8_x_7_b_instruct_v_01: typing.Callable[[], T_Result],
     ) -> T_Result:
+        if self is LlmModel.GPT_3_5_TURBO:
+            return gpt_3_5_turbo()
         if self is LlmModel.GPT_3_5_TURBO_16_K_0613:
             return gpt_3_5_turbo_16_k_0613()
         if self is LlmModel.GPT_3_5_TURBO_0613:
@@ -43,14 +57,24 @@ def visit(
             return gpt_3_5_turbo_1106()
         if self is LlmModel.GPT_3_5_TURBO_0125:
             return gpt_3_5_turbo_0125()
+        if self is LlmModel.GPT_4:
+            return gpt_4()
         if self is LlmModel.GPT_4_0613:
             return gpt_4_0613()
+        if self is LlmModel.GPT_4_32_K:
+            return gpt_4_32_k()
         if self is LlmModel.GPT_4_32_K_0613:
             return gpt_4_32_k_0613()
         if self is LlmModel.GPT_4_1106_PREVIEW:
             return gpt_4_1106_preview()
+        if self is LlmModel.GPT_4_0125_PREVIEW:
+            return gpt_4_0125_preview()
+        if self is LlmModel.GPT_4_TURBO:
+            return gpt_4_turbo()
         if self is LlmModel.GPT_4_TURBO_PREVIEW:
             return gpt_4_turbo_preview()
+        if self is LlmModel.GPT_4_TURBO_2024_04_09:
+            return gpt_4_turbo_2024_04_09()
         if self is LlmModel.MISTRAL_7_B_INSTRUCT_V_01:
             return mistral_7_b_instruct_v_01()
         if self is LlmModel.MIXTRAL_8_X_7_B_INSTRUCT_V_01: