Method generate_embeddings

core/src/embeddings.rs:239–330

(
        &self,
        request: EmbeddingRequest,
    )

#[async_trait]
impl EmbeddingProviderTrait for OpenAIEmbeddingProvider {
    async fn generate_embeddings(
        &self,
        request: EmbeddingRequest,
    ) -> GraphBitResult<EmbeddingResponse> {
        let url = format!("{}/embeddings", self.base_url());

        let input = match &request.input {
            EmbeddingInput::Single(text) => serde_json::Value::String(text.clone()),
            EmbeddingInput::Multiple(texts) => serde_json::Value::Array(
                texts
                    .iter()
                    .map(|t| serde_json::Value::String(t.clone()))
                    .collect(),
            ),
        };

        let mut body = serde_json::json!({
            "model": self.config.model,
            "input": input,
        });

        // Add user if provided
        if let Some(user) = &request.user {
            body["user"] = serde_json::Value::String(user.clone());
        }

        // Add extra parameters
        for (key, value) in &request.params {
            body[key] = value.clone();
        }

        let response = self
            .client
            .post(&url)
            .header("Authorization", format!("Bearer {}", self.config.api_key))
            .header("Content-Type", "application/json")
            .json(&body)
            .send()
            .await
            .map_err(|e| GraphBitError::llm(format!("Failed to send request to OpenAI: {e}")))?;

        if !response.status().is_success() {
            let error_text = response
                .text()
                .await
                .unwrap_or_else(|_| "Unknown error".to_string());
            return Err(GraphBitError::llm(format!(
                "OpenAI API error: {error_text}"
            )));
        }

        let response_json: serde_json::Value = response
            .json()
            .await
            .map_err(|e| GraphBitError::llm(format!("Failed to parse OpenAI response: {e}")))?;

        // Parse embeddings
        let embeddings_data = response_json["data"]