(self, text: str)
| 815 | ] |
| 816 | |
| 817 | def embed_query(self, text: str) -> List[float]: |
| 818 | configure_litellm() |
| 819 | # Apply rate limiting if configured |
| 820 | apply_rate_limiter_sync(self.a0_model_conf, text) |
| 821 | |
| 822 | resp = embedding( |
| 823 | model=self.model_name, |
| 824 | input=[text], |
| 825 | **_merge_litellm_call_kwargs(self.kwargs), |
| 826 | ) |
| 827 | item = resp.data[0] # type: ignore |
| 828 | return item.get("embedding") if isinstance(item, dict) else item.embedding # type: ignore |
| 829 | |
| 830 | |
| 831 | class LocalSentenceTransformerWrapper(Embeddings): |
no test coverage detected