Coverage for mindsdb / integrations / handlers / sentence_transformers_handler / settings.py: 0%

1from pydantic import BaseModel, Extra 

2 

3DEFAULT_EMBEDDING_MODEL = "all-MiniLM-L6-v2" 

4 

5 

6class Parameters(BaseModel): 

7 embeddings_model_name: str = DEFAULT_EMBEDDING_MODEL 

8 text_columns: list = None 

9 use_gpu: bool = False 

10 

11 class Config: 

12 extra = Extra.forbid