From 75807241a6a33423a7916c30bef8d3d36cd44790 Mon Sep 17 00:00:00 2001
From: vbarda <vadym@langchain.dev>
Date: Fri, 26 Jul 2024 16:18:29 -0400
Subject: [PATCH] limit to claude and oai for now

---
 backend/app/config_schema.json | 4 +---
 backend/app/llms.py            | 3 ---
 2 files changed, 1 insertion(+), 6 deletions(-)

diff --git a/backend/app/config_schema.json b/backend/app/config_schema.json
index 434b4fbc..41077452 100644
--- a/backend/app/config_schema.json
+++ b/backend/app/config_schema.json
@@ -245,9 +245,7 @@
       "enum": [
         "GPT 4o Mini",
         "GPT 4o",
-        "Claude 3.5 Sonnet",
-        "GEMINI",
-        "Mixtral"
+        "Claude 3.5 Sonnet"
       ],
       "type": "string"
     },
diff --git a/backend/app/llms.py b/backend/app/llms.py
index f0d665ff..3db478aa 100644
--- a/backend/app/llms.py
+++ b/backend/app/llms.py
@@ -19,9 +19,6 @@ class LLMType(str, Enum):
     GPT_4O_MINI = "GPT 4o Mini"
     GPT_4O = "GPT 4o"
     CLAUDE_3_5_SONNET = "Claude 3.5 Sonnet"
-    GEMINI = "GEMINI"
-    MIXTRAL = "Mixtral"
-    OLLAMA = "Ollama"
 
 
 @lru_cache(maxsize=4)