Remove huggingface encoder

superagent-ai · Feb 22, 2024 · b27e956 · b27e956
1 parent b49453a
commit b27e956
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 5 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -13,6 +13,9 @@ COPY pyproject.toml poetry.lock ./
 # Install the required packages of the application
 RUN poetry install --no-root && rm -rf $POETRY_CACHE_DIR
 
+# Install gunicorn
+RUN pip install gunicorn
+
 # Make port 80 available to the world outside this container
 ENV PORT="8080"
 

diff --git a/models/ingest.py b/models/ingest.py
@@ -10,8 +10,6 @@
 class EncoderEnum(str, Enum):
     cohere = "cohere"
     openai = "openai"
-    huggingface = "huggingface"
-    fastembed = "fastembed"
 
 
 class Encoder(BaseModel):

diff --git a/poetry.lock b/poetry.lock
diff --git a/service/embedding.py b/service/embedding.py
@@ -10,7 +10,6 @@
 from semantic_router.encoders import (
     BaseEncoder,
     CohereEncoder,
-    HuggingFaceEncoder,
     OpenAIEncoder,
 )
 from tqdm import tqdm
@@ -247,7 +246,6 @@ def get_encoder(*, encoder_config: Encoder) -> BaseEncoder:
     encoder_mapping = {
         EncoderEnum.cohere: CohereEncoder,
         EncoderEnum.openai: OpenAIEncoder,
-        EncoderEnum.huggingface: HuggingFaceEncoder,
     }
     encoder_provider = encoder_config.type
     encoder = encoder_config.name