(self, text: str)
| 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 |
| 872 | apply_rate_limiter_sync(self.a0_model_conf, text) |
| 873 | |
| 874 | embedding = self.model.encode([text], convert_to_tensor=False) # type: ignore |
| 875 | result = ( |
| 876 | embedding[0].tolist() if hasattr(embedding[0], "tolist") else embedding[0] |
| 877 | ) |
| 878 | return result # type: ignore |
| 879 | |
| 880 | |
| 881 | def _get_litellm_chat( |
nothing calls this directly
no test coverage detected