Spaces:
Running
on
T4
Running
on
T4
Update optimum_encoder.py
Browse files- 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
|