(self, texts: List[str])
| 861 | self.a0_model_conf = model_config |
| 862 | |
| 863 | def embed_documents(self, texts: List[str]) -> List[List[float]]: |
| 864 | # Apply rate limiting if configured |
| 865 | apply_rate_limiter_sync(self.a0_model_conf, " ".join(texts)) |
| 866 | |
| 867 | embeddings = self.model.encode(texts, convert_to_tensor=False) # type: ignore |
| 868 | return embeddings.tolist() if hasattr(embeddings, "tolist") else embeddings # type: ignore |
| 869 | |
| 870 | def embed_query(self, text: str) -> List[float]: |
| 871 | # Apply rate limiting if configured |
nothing calls this directly
no test coverage detected