Add chat workspace support and tests

Author: kumarUjjawal

src/chats.rs

@@ -1,7 +1,7 @@
 use std::collections::BTreeMap;
 
 use serde::{Deserialize, Serialize};
-use serde_json::Value;
+use serde_json::{to_vec, Value};
 
 use crate::{
     client::Client,
@@ -275,13 +275,34 @@ impl Client<crate::reqwest::ReqwestClient> {
         uid: impl AsRef<str>,
         body: &S,
     ) -> Result<reqwest::Response, Error> {
-        use reqwest::header::{HeaderValue, ACCEPT, CONTENT_TYPE};
+        use reqwest::{
+            header::{self, HeaderValue, ACCEPT, CONTENT_TYPE},
+            Client as HttpClient,
+        };
+
+        let payload = to_vec(body).map_err(Error::ParseError)?;
+
+        let mut headers = header::HeaderMap::new();
+        #[cfg(not(target_arch = "wasm32"))]
+        headers.insert(
+            header::USER_AGENT,
+            header::HeaderValue::from_str(&crate::reqwest::qualified_version()).unwrap(),
+        );
+        #[cfg(target_arch = "wasm32")]
+        headers.insert(
+            header::HeaderName::from_static("x-meilisearch-client"),
+            header::HeaderValue::from_str(&crate::reqwest::qualified_version()).unwrap(),
+        );
+        if let Some(api_key) = self.api_key.as_deref() {
+            headers.insert(
+                header::AUTHORIZATION,
+                header::HeaderValue::from_str(&format!("Bearer {api_key}")).unwrap(),
+            );
+        }
 
-        let payload = serde_json::to_vec(body).map_err(Error::ParseError)?;
+        let http = HttpClient::builder().default_headers(headers).build()?;
 
-        let response = self
-            .http_client
-            .inner()
+        let response = http
             .post(format!(
                 "{}/chats/{}/chat/completions",
                 self.host,
@@ -315,14 +336,22 @@ impl Client<crate::reqwest::ReqwestClient> {
 #[cfg(test)]
 mod tests {
     use super::*;
-    use crate::features::ExperimentalFeatures;
     use meilisearch_test_macro::meilisearch_test;
+    use serde_json::json;
 
     #[meilisearch_test]
     async fn chat_workspace_lifecycle(client: Client, name: String) -> Result<(), Error> {
-        let mut features = ExperimentalFeatures::new(&client);
-        features.set_chat_completions(true);
-        let _ = features.update().await?;
+        let _: serde_json::Value = client
+            .http_client
+            .request(
+                &format!("{}/experimental-features", client.host),
+                Method::Patch {
+                    query: (),
+                    body: &json!({ "chatCompletions": true }),
+                },
+                200,
+            )
+            .await?;
 
         let workspace = format!("{name}-workspace");
 

src/features.rs

@@ -14,8 +14,6 @@ pub struct ExperimentalFeaturesResult {
     pub contains_filter: bool,
     pub network: bool,
     pub edit_documents_by_function: bool,
-    #[serde(default)]
-    pub chat_completions: bool,
 }
 
 /// Struct representing the experimental features request.
@@ -47,8 +45,6 @@ pub struct ExperimentalFeatures<'a, Http: HttpClient> {
     pub network: Option<bool>,
     #[serde(skip_serializing_if = "Option::is_none")]
     pub edit_documents_by_function: Option<bool>,
-    #[serde(skip_serializing_if = "Option::is_none")]
-    pub chat_completions: Option<bool>,
 }
 
 impl<'a, Http: HttpClient> ExperimentalFeatures<'a, Http> {
@@ -61,7 +57,6 @@ impl<'a, Http: HttpClient> ExperimentalFeatures<'a, Http> {
             network: None,
             contains_filter: None,
             edit_documents_by_function: None,
-            chat_completions: None,
         }
     }
 
@@ -145,11 +140,6 @@ impl<'a, Http: HttpClient> ExperimentalFeatures<'a, Http> {
         self.network = Some(network);
         self
     }
-
-    pub fn set_chat_completions(&mut self, chat_completions: bool) -> &mut Self {
-        self.chat_completions = Some(chat_completions);
-        self
-    }
 }
 
 #[cfg(test)]
@@ -165,7 +155,6 @@ mod tests {
         features.set_contains_filter(true);
         features.set_network(true);
         features.set_edit_documents_by_function(true);
-        features.set_chat_completions(true);
         let _ = features.update().await.unwrap();
 
         let res = features.get().await.unwrap();
@@ -174,6 +163,5 @@ mod tests {
         assert!(res.contains_filter);
         assert!(res.network);
         assert!(res.edit_documents_by_function);
-        assert!(res.chat_completions);
     }
 }

src/reqwest.rs

@@ -49,10 +49,6 @@ impl ReqwestClient {
 
         Ok(ReqwestClient { client })
     }
-
-    pub(crate) fn inner(&self) -> &reqwest::Client {
-        &self.client
-    }
 }
 
 #[cfg_attr(feature = "futures-unsend", async_trait(?Send))]