(self, text: str)
| 91 | self._model = SentenceTransformer(model, **kwargs) |
| 92 | |
| 93 | def embed(self, text: str) -> List[float]: |
| 94 | vector = self._model.encode(text or " ", normalize_embeddings=True) |
| 95 | return _resize(list(vector.tolist() if hasattr(vector, "tolist") else vector), self.dimensions) |
| 96 | |
| 97 | def embed_batch(self, texts: Iterable[str]) -> List[List[float]]: |
| 98 | items = [text or " " for text in texts] |