Generate embeddings for a single text
(&self, text: &str)
| 780 | |
| 781 | /// Generate embeddings for a single text |
| 782 | pub async fn embed_text(&self, text: &str) -> GraphBitResult<Vec<f32>> { |
| 783 | let request = EmbeddingRequest { |
| 784 | input: EmbeddingInput::Single(text.to_string()), |
| 785 | user: None, |
| 786 | params: HashMap::new(), |
| 787 | }; |
| 788 | |
| 789 | let response = self.provider.generate_embeddings(request).await?; |
| 790 | |
| 791 | response |
| 792 | .embeddings |
| 793 | .into_iter() |
| 794 | .next() |
| 795 | .ok_or_else(|| GraphBitError::llm("No embeddings returned".to_string())) |
| 796 | } |
| 797 | |
| 798 | /// Generate embeddings for multiple texts |
| 799 | pub async fn embed_texts(&self, texts: &[String]) -> GraphBitResult<Vec<Vec<f32>>> { |