Add GPT-4o to configure modal options

seratch · May 14, 2024 · f482cf8 · f482cf8
1 parent aa0f9ce
commit f482cf8
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/main_prod.py b/main_prod.py
@@ -1,4 +1,6 @@
 # Unzip the dependencies managed by serverless-python-requirements
+from app.openai_constants import GPT_3_5_TURBO_MODEL, GPT_4_MODEL, GPT_4_32K_MODEL, GPT_4O_MODEL
+
 try:
     import unzip_requirements  # type:ignore
 except ImportError:
@@ -237,23 +239,27 @@ def handle_some_action(ack, body: dict, client: WebClient, context: BoltContext)
                                         "type": "plain_text",
                                         "text": "GPT-3.5 Turbo",
                                     },
-                                    "value": "gpt-3.5-turbo",
+                                    "value": GPT_3_5_TURBO_MODEL,
                                 },
                                 {
                                     "text": {"type": "plain_text", "text": "GPT-4 8K"},
-                                    "value": "gpt-4",
+                                    "value": GPT_4_MODEL,
                                 },
                                 {
                                     "text": {"type": "plain_text", "text": "GPT-4 32K"},
-                                    "value": "gpt-4-32k",
+                                    "value": GPT_4_32K_MODEL,
+                                },
+                                {
+                                    "text": {"type": "plain_text", "text": "GPT-4o"},
+                                    "value": GPT_4O_MODEL,
                                 },
                             ],
                             "initial_option": {
                                 "text": {
                                     "type": "plain_text",
                                     "text": "GPT-3.5 Turbo",
                                 },
-                                "value": "gpt-3.5-turbo",
+                                "value": GPT_3_5_TURBO_MODEL,
                             },
                         },
                     },