Update optimum_encoder.py

optimum_encoder.py

@@ -8,9 +8,10 @@ from pydantic.v1 import PrivateAttr
 from fastembed.common.utils import normalize
 from semantic_router.encoders import BaseEncoder
 from semantic_router.utils.logger import logger
+from langchain_core.embeddings import Embeddings
 
 
-class OptimumEncoder(BaseEncoder):
+class OptimumEncoder(BaseEncoder, Embeddings):
     name: str = "mixedbread-ai/mxbai-embed-large-v1"
     type: str = "huggingface"
     score_threshold: float = 0.5
@@ -104,7 +105,7 @@ class OptimumEncoder(BaseEncoder):
 
         return tokenizer, ort_model
 
-    def __call__(
+    def embed_documents(
         self,
         docs: List[str],
         batch_size: int = 32,