add claude-3.5-sonnet and set openai models to their generic versions…

…, which will always use the most up-to-date model in that family
datalab-org · Jul 3, 2024 · 71343a2 · 71343a2
1 parent f8d0ef6
commit 71343a2
Showing 1 changed file with 13 additions and 7 deletions.
diff --git a/pydatalab/pydatalab/apps/chat/blocks.py b/pydatalab/pydatalab/apps/chat/blocks.py
@@ -47,8 +47,14 @@ class ChatBlock(DataBlock):
         """,
         "temperature": 0.2,
         "error_message": None,
-        "model": "gpt-4o-2024-05-13",
+        "model": "gpt-4o",
         "available_models": {
+            "claude-3-5-sonnet-20240620": {
+                "name": "claude-3-5-sonnet-20240620",
+                "context_window": 200000,
+                "input_cost_usd_per_MTok": 3.00,
+                "output_cost_usd_per_MTok": 15.00,
+            },
             "claude-3-haiku-20240307": {
                 "name": "claude-3-haiku-20240307",
                 "context_window": 200000,
@@ -67,14 +73,14 @@ class ChatBlock(DataBlock):
                 "input_cost_usd_per_MTok": 3.00,
                 "output_cost_usd_per_MTok": 15.00,
             },
-            "gpt-4o-2024-05-13": {
-                "name": "gpt-4o-2024-05-13",
+            "gpt-4o": {
+                "name": "gpt-4o",
                 "context_window": 128000,
                 "input_cost_usd_per_MTok": 5.00,
                 "output_cost_usd_per_MTok": 15.00,
             },
-            "gpt-3.5-turbo-0125": {
-                "name": "gpt-3.5-turbo-0125",
+            "gpt-3.5-turbo": {
+                "name": "gpt-3.5-turbo",
                 "context_window": 16385,
                 "input_cost_usd_per_MTok": 1.50,
                 "output_cost_usd_per_MTok": 2.00,
@@ -85,8 +91,8 @@ class ChatBlock(DataBlock):
                 "input_cost_usd_per_MTok": 30.00,
                 "output_cost_usd_per_MTok": 60.00,
             },
-            "gpt-4-turbo-2024-04-09": {
-                "name": "gpt-4-turbo-2024-04-09",
+            "gpt-4-turbo": {
+                "name": "gpt-4-turbo",
                 "context_window": 128000,
                 "input_cost_usd_per_MTok": 10.00,
                 "output_cost_usd_per_MTok": 30.00,