Method embed_voyage

atomic-repository/src/ai/mod.rs:196–230

(&self, texts: &[String])

    }

    async fn embed_voyage(&self, texts: &[String]) -> Result<Vec<Vec<f32>>, AiError> {
        let api_key = self.api_key.as_ref().ok_or(AiError::NoApiKey)?;

        let body = serde_json::json!({
            "model": self.model,
            "input": texts,
            "input_type": "document"
        });

        let client = reqwest::Client::new();
        let resp = client
            .post("https://api.voyageai.com/v1/embeddings")
            .header("Authorization", format!("Bearer {}", api_key))
            .header("content-type", "application/json")
            .json(&body)
            .send()
            .await
            .map_err(|e| AiError::Http(e.to_string()))?;

        if !resp.status().is_success() {
            let status = resp.status();
            let text = resp.text().await.unwrap_or_default();
            return Err(AiError::ApiError {
                status: status.as_u16(),
                body: text,
            });
        }

        let json: serde_json::Value = resp
            .json()
            .await
            .map_err(|e| AiError::Http(e.to_string()))?;

        parse_embedding_response(&json)
    }

    async fn embed_openai(&self, texts: &[String]) -> Result<Vec<Vec<f32>>, AiError> {
        let api_key = self.api_key.as_ref().ok_or(AiError::NoApiKey)?;