Merge pull request #9 from aurelio-labs/james/remove-hf-refs

bug fix missing encoder
aurelio-labs · Nov 13, 2023 · 423c38a · 423c38a
2 parents 372c23f + f67e5d2
commit 423c38a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 12 deletions.
diff --git a/semantic_router/encoders/huggingface.py b/semantic_router/encoders/huggingface.py
diff --git a/semantic_router/schema.py b/semantic_router/schema.py
@@ -6,7 +6,6 @@
 from semantic_router.encoders import (
     BaseEncoder,
     CohereEncoder,
-    HuggingFaceEncoder,
     OpenAIEncoder,
 )
 
@@ -18,7 +17,6 @@ class Decision(BaseModel):
 
 
 class EncoderType(Enum):
-    HUGGINGFACE = "huggingface"
     OPENAI = "openai"
     COHERE = "cohere"
 
@@ -33,7 +31,7 @@ def __init__(self, type: str, name: str):
         self.type = EncoderType(type)
         self.name = name
         if self.type == EncoderType.HUGGINGFACE:
-            self.model = HuggingFaceEncoder(name)
+            raise NotImplementedError
         elif self.type == EncoderType.OPENAI:
             self.model = OpenAIEncoder(name)
         elif self.type == EncoderType.COHERE: