1 files changed, 2 insertions, 29 deletions
diff --git a/server/continuedev/core/models.py b/server/continuedev/core/models.py
index 21ebd8f6..c31177b9 100644
--- a/server/continuedev/core/models.py
+++ b/server/continuedev/core/models.py
@@ -2,18 +2,9 @@ from typing import List, Optional
 
 from pydantic import BaseModel
 
-from ..libs.llm.anthropic import AnthropicLLM
 from ..libs.llm.base import LLM
-from ..libs.llm.ggml import GGML
-from ..libs.llm.google_palm_api import GooglePaLMAPI
-from ..libs.llm.hf_inference_api import HuggingFaceInferenceAPI
-from ..libs.llm.hf_tgi import HuggingFaceTGI
 from ..libs.llm.llamacpp import LlamaCpp
 from ..libs.llm.ollama import Ollama
-from ..libs.llm.openai import OpenAI
-from ..libs.llm.openai_free_trial import OpenAIFreeTrial
-from ..libs.llm.replicate import ReplicateLLM
-from ..libs.llm.together import TogetherLLM
 
 
 class ContinueSDK(BaseModel):
@@ -30,32 +21,14 @@ ALL_MODEL_ROLES = [
 MODEL_CLASSES = {
     cls.__name__: cls
     for cls in [
-        OpenAI,
-        OpenAIFreeTrial,
-        GGML,
-        TogetherLLM,
-        AnthropicLLM,
-        ReplicateLLM,
         Ollama,
-        LlamaCpp,
-        HuggingFaceInferenceAPI,
-        HuggingFaceTGI,
-        GooglePaLMAPI,
+        LlamaCpp
     ]
 }
 
 MODEL_MODULE_NAMES = {
-    "OpenAI": "openai",
-    "OpenAIFreeTrial": "openai_free_trial",
-    "GGML": "ggml",
-    "TogetherLLM": "together",
-    "AnthropicLLM": "anthropic",
-    "ReplicateLLM": "replicate",
     "Ollama": "ollama",
-    "LlamaCpp": "llamacpp",
-    "HuggingFaceInferenceAPI": "hf_inference_api",
-    "HuggingFaceTGI": "hf_tgi",
-    "GooglePaLMAPI": "google_palm_api",
+    "LlamaCpp": "llamacpp"
 }