beta() apis

Author: 64bit

async-openai/src/beta.rs

@@ -0,0 +1,42 @@
+use std::cell::OnceCell;
+
+use crate::{
+    chatkit::Chatkit,
+    config::{Config, OpenAIConfig},
+    Client,
+};
+
+/// Beta API accessor for beta features like ChatKit.
+///
+/// Use `client.beta().chatkit()` to access ChatKit APIs with the required beta header.
+pub struct Beta<'c, C: Config> {
+    client: &'c Client<C>,
+    beta_client: OnceCell<Client<OpenAIConfig>>,
+}
+
+impl<'c> Beta<'c, OpenAIConfig> {
+    /// Access ChatKit APIs.
+    ///
+    /// This automatically includes the `OpenAI-Beta: chatkit_beta=v1` header in all requests.
+    pub fn chatkit(&self) -> Chatkit<'_, OpenAIConfig> {
+        let beta_client = self.beta_client.get_or_init(|| {
+            let mut beta_config = self.client.config().clone();
+            beta_config = beta_config
+                .with_header("OpenAI-Beta", "chatkit_beta=v1")
+                .unwrap();
+            Client::with_config(beta_config)
+                .with_http_client(self.client.http_client().clone())
+                .with_backoff(self.client.backoff().clone())
+        });
+        Chatkit::new(beta_client)
+    }
+}
+
+impl<'c, C: Config> Beta<'c, C> {
+    pub(crate) fn new_generic(client: &'c Client<C>) -> Self {
+        Self {
+            client,
+            beta_client: OnceCell::new(),
+        }
+    }
+}

async-openai/src/client.rs

@@ -7,7 +7,7 @@ use reqwest_eventsource::{Error as EventSourceError, Event, EventSource, Request
 use serde::{de::DeserializeOwned, Serialize};
 
 use crate::{
-    chatkit::Chatkit,
+    beta::Beta,
     config::{Config, OpenAIConfig},
     error::{map_deserialization_error, ApiError, OpenAIError, StreamError, WrappedError},
     file::Files,
@@ -189,15 +189,26 @@ impl<C: Config> Client<C> {
         Evals::new(self)
     }
 
-    /// To call [Chatkit] group related APIs using this client.
-    pub fn chatkit(&self) -> Chatkit<'_, C> {
-        Chatkit::new(self)
+    /// Access beta APIs like ChatKit.
+    ///
+    /// Beta APIs require special headers that are automatically added.
+    /// Use `client.beta().chatkit()` to access ChatKit APIs.
+    pub fn beta(&self) -> Beta<'_, C> {
+        Beta::new_generic(self)
     }
 
     pub fn config(&self) -> &C {
         &self.config
     }
 
+    pub(crate) fn http_client(&self) -> &reqwest::Client {
+        &self.http_client
+    }
+
+    pub(crate) fn backoff(&self) -> &backoff::ExponentialBackoff {
+        &self.backoff
+    }
+
     /// Make a GET request to {path} and deserialize the response body
     pub(crate) async fn get<O>(&self, path: &str) -> Result<O, OpenAIError>
     where

async-openai/src/lib.rs

@@ -144,6 +144,7 @@ mod assistants;
 mod audio;
 mod audit_logs;
 mod batches;
+mod beta;
 mod chat;
 mod chatkit;
 mod client;
@@ -193,6 +194,7 @@ pub use assistants::Assistants;
 pub use audio::Audio;
 pub use audit_logs::AuditLogs;
 pub use batches::Batches;
+pub use beta::Beta;
 pub use chat::Chat;
 pub use chatkit::Chatkit;
 pub use client::Client;