Add option to disable gpt-4 for public deployment (#124)

langchain-ai · Mar 29, 2024 · c72c87f · c72c87f
1 parent df1907a
commit c72c87f
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/backend/server/models.py b/backend/server/models.py
@@ -14,10 +14,11 @@ def get_supported_models():
             "chat_model": ChatOpenAI(model="gpt-3.5-turbo", temperature=0),
             "description": "GPT-3.5 Turbo",
         }
-        models["gpt-4-0125-preview"] = {
-            "chat_model": ChatOpenAI(model="gpt-4-0125-preview", temperature=0),
-            "description": "GPT-4 0125 Preview",
-        }
+        if os.environ.get("DISABLE_GPT4", "").lower() != "true":
+            models["gpt-4-0125-preview"] = {
+                "chat_model": ChatOpenAI(model="gpt-4-0125-preview", temperature=0),
+                "description": "GPT-4 0125 Preview",
+            }
     if "FIREWORKS_API_KEY" in os.environ:
         models["fireworks"] = {
             "chat_model": ChatFireworks(