agent_core/client/providers/openai/
mod.rs

1mod types;
2
3use crate::client::error::LlmError;
4use crate::client::http::HttpClient;
5use crate::client::models::{Message, MessageOptions};
6use crate::client::traits::LlmProvider;
7use std::future::Future;
8use std::pin::Pin;
9
10/// OpenAI API provider.
11pub struct OpenAIProvider {
12    /// OpenAI API key.
13    pub api_key: String,
14    /// Model identifier (e.g., "gpt-4").
15    pub model: String,
16}
17
18impl OpenAIProvider {
19    /// Create a new OpenAI provider with API key and model.
20    pub fn new(api_key: String, model: String) -> Self {
21        Self { api_key, model }
22    }
23}
24
25impl LlmProvider for OpenAIProvider {
26    fn send_msg(
27        &self,
28        client: &HttpClient,
29        messages: &[Message],
30        options: &MessageOptions,
31    ) -> Pin<Box<dyn Future<Output = Result<Message, LlmError>> + Send>> {
32        // Clone data for the async block
33        let client = client.clone();
34        let api_key = self.api_key.clone();
35        let model = self.model.clone();
36        let messages = messages.to_vec();
37        let options = options.clone();
38
39        Box::pin(async move {
40            // Build request body
41            let body = types::build_request_body(&messages, &options, &model)?;
42
43            // Get headers
44            let headers = types::get_request_headers(&api_key);
45            let headers_ref: Vec<(&str, &str)> = headers
46                .iter()
47                .map(|(k, v)| (*k, v.as_str()))
48                .collect();
49
50            // Make the API call
51            let response = client
52                .post(types::get_api_url(), &headers_ref, &body)
53                .await?;
54
55            // Parse and return the response
56            types::parse_response(&response)
57        })
58    }
59}
agent_core/client/providers/openai/mod.rs

agent_core/client/providers/openai/
mod.rs
