Method complete

core/src/llm/ai21.rs:194–260

(&self, request: LlmRequest)

    }

    async fn complete(&self, request: LlmRequest) -> GraphBitResult<LlmResponse> {
        let url = format!("{}/chat/completions", self.base_url);

        let messages: Vec<Ai21Message> =
            request.messages.iter().map(Self::convert_message).collect();

        let tools: Option<Vec<Ai21Tool>> = if request.tools.is_empty() {
            None
        } else {
            Some(request.tools.iter().map(Self::convert_tool).collect())
        };

        let body = Ai21Request {
            model: self.model.clone(),
            messages,
            max_tokens: request.max_tokens,
            temperature: request.temperature,
            top_p: request.top_p,
            tools: tools.clone(),
            tool_choice: if tools.is_some() {
                Some("auto".to_string())
            } else {
                None
            },
        };

        // Merge extra_params into request JSON
        let mut req_json = serde_json::to_value(&body)?;
        if let serde_json::Value::Object(ref mut map) = req_json {
            for (k, v) in request.extra_params {
                map.insert(k, v);
            }
        }

        let mut builder = self
            .client
            .post(&url)
            .header("Authorization", format!("Bearer {}", self.api_key))
            .header("Content-Type", "application/json")
            .json(&req_json);

        if let Some(org) = &self.organization {
            builder = builder.header("Ai21-Organization", org);
        }

        let resp = builder
            .send()
            .await
            .map_err(|e| GraphBitError::llm_provider("ai21", format!("Request failed: {e}")))?;

        if !resp.status().is_success() {
            let text = resp
                .text()
                .await
                .unwrap_or_else(|_| "Unknown error".to_string());
            return Err(GraphBitError::llm_provider(
                "ai21",
                format!("API error: {text}"),