devve1 commited on
Commit
310438e
1 Parent(s): 9396279

Update optimum_encoder.py

Browse files
Files changed (1) hide show
  1. optimum_encoder.py +2 -1
optimum_encoder.py CHANGED
@@ -6,11 +6,12 @@ from tqdm import tqdm
6
  from typing import Any, List, Optional, Dict
7
  from pydantic.v1 import PrivateAttr
8
  from fastembed.common.utils import normalize
 
9
  from semantic_router.utils.logger import logger
10
  from langchain_core.embeddings import Embeddings
11
 
12
 
13
- class OptimumEncoder(Embeddings):
14
  name: str = "mixedbread-ai/mxbai-embed-large-v1"
15
  type: str = "huggingface"
16
  score_threshold: float = 0.5
 
6
  from typing import Any, List, Optional, Dict
7
  from pydantic.v1 import PrivateAttr
8
  from fastembed.common.utils import normalize
9
+ from semantic_router.encoders import BaseEncoder
10
  from semantic_router.utils.logger import logger
11
  from langchain_core.embeddings import Embeddings
12
 
13
 
14
+ class OptimumEncoder(BaseEncoder, Embeddings):
15
  name: str = "mixedbread-ai/mxbai-embed-large-v1"
16
  type: str = "huggingface"
17
  score_threshold: float = 0.5