configurable per request path

Author: 64bit

async-openai/src/client.rs

@@ -204,9 +204,14 @@ impl<C: Config> Client<C> {
         path: &str,
         request_options: &RequestOptions,
     ) -> reqwest::RequestBuilder {
-        let mut request_builder = self
-            .http_client
-            .request(method, self.config.url(path))
+        let mut request_builder = if let Some(path) = request_options.path() {
+            self.http_client
+                .request(method, self.config.url(path.as_str()))
+        } else {
+            self.http_client.request(method, self.config.url(path))
+        };
+
+        request_builder = request_builder
             .query(&self.config.query())
             .headers(self.config.headers());
 

async-openai/src/request_options.rs

@@ -8,16 +8,28 @@ use crate::{config::OPENAI_API_BASE, error::OpenAIError};
 pub struct RequestOptions {
     query: Option<Vec<(String, String)>>,
     headers: Option<HeaderMap>,
+    path: Option<String>,
 }
 
 impl RequestOptions {
     pub(crate) fn new() -> Self {
         Self {
             query: None,
             headers: None,
+            path: None,
         }
     }
 
+    pub(crate) fn with_path(&mut self, path: &str) -> Result<(), OpenAIError> {
+        if path.is_empty() {
+            return Err(OpenAIError::InvalidArgument(
+                "Path cannot be empty".to_string(),
+            ));
+        }
+        self.path = Some(path.to_string());
+        Ok(())
+    }
+
     pub(crate) fn with_headers(&mut self, headers: HeaderMap) {
         // merge with existing headers or update with new headers
         if let Some(existing_headers) = &mut self.headers {
@@ -81,4 +93,8 @@ impl RequestOptions {
     pub(crate) fn headers(&self) -> Option<&HeaderMap> {
         self.headers.as_ref()
     }
+
+    pub(crate) fn path(&self) -> Option<&String> {
+        self.path.as_ref()
+    }
 }

async-openai/src/traits.rs

@@ -53,4 +53,10 @@ pub trait RequestOptionsBuilder: Sized {
         self.options_mut().with_query(query)?;
         Ok(self)
     }
+
+    /// Add a path to RequestOptions
+    fn path<P: Into<String>>(mut self, path: P) -> Result<Self, OpenAIError> {
+        self.options_mut().with_path(path.into().as_str())?;
+        Ok(self)
+    }
 }