Generate embedding for text. Args: text: Text to embed Returns: Pickled numpy array of embedding, or None on error
(self, text: str)
| 70 | self._loaded = False |
| 71 | |
| 72 | def embed(self, text: str) -> Optional[bytes]: |
| 73 | """ |
| 74 | Generate embedding for text. |
| 75 | |
| 76 | Args: |
| 77 | text: Text to embed |
| 78 | |
| 79 | Returns: |
| 80 | Pickled numpy array of embedding, or None on error |
| 81 | """ |
| 82 | self._load_model() |
| 83 | |
| 84 | if not self._loaded or self._model is None: |
| 85 | return None |
| 86 | |
| 87 | try: |
| 88 | import numpy as np |
| 89 | embedding = self._model.encode(text, convert_to_numpy=True) |
| 90 | return pickle.dumps(embedding) |
| 91 | except Exception as e: |
| 92 | error(f"Embedding error: {e}") |
| 93 | return None |
| 94 | |
| 95 | def embed_batch(self, texts: List[str]) -> Optional[List[bytes]]: |
| 96 | """ |
no test coverage detected