Assistant - support new 4.1 models

Author: x4714

assistant/assistant.py

@@ -54,7 +54,7 @@ class Assistant(
     """
 
     __author__ = "[vertyco](https://github.com/vertyco/vrt-cogs)"
-    __version__ = "6.12.21"
+    __version__ = "6.12.22"
 
     def format_help_for_context(self, ctx):
         helpcmd = super().format_help_for_context(ctx)

assistant/common/api.py

@@ -170,6 +170,9 @@ async def count_function_tokens(self, functions: List[dict], model: str = "gpt-4
             "gpt-4o-2024-11-20",
             "gpt-4o-mini",
             "gpt-4o-mini-2024-07-18",
+            "gpt-4.1",
+            "gpt-4.1-mini",
+            "gpt-4.1-nano",
             "o1-preview",
             "o1-preview-2024-09-12",
             "o1",

assistant/common/constants.py

@@ -21,6 +21,9 @@
     "gpt-4o-2024-08-06": 128000,
     "gpt-4o-2024-11-20": 128000,
     "chatgpt-4o-latest": 128000,
+    "gpt-4.1": 1047576,
+    "gpt-4.1-mini": 1047576,
+    "gpt-4.1-nano": 1047576,
     "o1": 128000,
     "o1-2024-12-17": 200000,
     "o1-preview": 128000,
@@ -60,6 +63,9 @@
     "chatgpt-4o-latest": [0.0025, 0.01],
     "gpt-4o-mini": [0.00015, 0.0006],
     "gpt-4o-mini-2024-07-18": [0.00015, 0.0006],
+    "gpt-4.1": [0.002, 0.008],
+    "gpt-4.1-mini": [0.0004, 0.0016],
+    "gpt-4.1-nano": [0.0001, 0.0004],
     "o1": [0.015, 0.06],
     "o1-2024-12-17": [0.015, 0.06],
     "o1-preview": [0.015, 0.06],
@@ -105,6 +111,9 @@
     "gpt-4o-mini-2024-07-18",
     "gpt-4o-2024-08-06",
     "chatgpt-4o-latest",
+    "gpt-4.1",
+    "gpt-4.1-mini",
+    "gpt-4.1-nano",
 ]
 NO_DEVELOPER_ROLE = [  # Also doesnt support system messages
     "o1-mini",
@@ -125,6 +134,9 @@
     "gpt-4o-mini-2024-07-18",
     "gpt-4o-2024-08-06",
     "chatgpt-4o-latest",
+    "gpt-4.1",
+    "gpt-4.1-mini",
+    "gpt-4.1-nano",
     "o1",
     "o1-2024-12-17",
     # "o3-mini",
@@ -146,6 +158,9 @@
     "gpt-4o-mini-2024-07-18",
     "gpt-4o-2024-08-06",
     "chatgpt-4o-latest",
+    "gpt-4.1",
+    "gpt-4.1-mini",
+    "gpt-4.1-nano",
     "o1",
     "o1-2024-12-17",
     "o3-mini",