FEAT: gpt-4 price added to constants

drudilorenzo · Apr 4, 2024 · 1a1eadc · 1a1eadc
1 parent aab1b27
commit 1a1eadc
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/constants.py b/src/constants.py
@@ -8,6 +8,8 @@ class Models(Enum):
     TURBO_3_5 = "gpt-3.5-turbo"
     TURBO_3_5_INSTRUCT = "gpt-3.5-turbo-instruct"
     AZURE_3_5_TURBO = "gpt-35-turbo-0125"
+    AZURE_4_TURBO = "gpt-4-0125-Preview"
+    AZURE_4 = "gpt-4-0613"
 
 """The costs of the models above (per million tokens)."""
 MODELS_COST = {
@@ -22,5 +24,13 @@ class Models(Enum):
     "gpt-3.5-turbo-instruct": {
         "input": 1.5,
         "output": 2
-    }
+    },
+    "gpt-4-0125-Preview": {
+        "input": 10,
+        "output": 30
+    },
+    "gpt-4-0613": {
+        "input": 30,
+        "output": 60
+    },
 }