feat!: add client.embeddings() method

BREAKING CHANGE: Models are now enforced by `Model` & `EmbedModel` enums.

src/v1/client.rs

@@ -5,7 +5,8 @@ use crate::v1::{
     chat_completion::{
         ChatCompletionMessage, ChatCompletionParams, ChatCompletionRequest, ChatCompletionResponse,
     },
-    constants::API_URL_BASE,
+    constants::{EmbedModel, Model, API_URL_BASE},
+    embedding::{EmbeddingRequest, EmbeddingRequestOptions, EmbeddingResponse},
     model_list::ModelListResponse,
 };
 
@@ -59,6 +60,8 @@ impl Client {
         let result = request.send();
         match result {
             Ok(response) => {
+                print!("{:?}", response.as_str().unwrap());
+
                 if (200..=299).contains(&response.status_code) {
                     Ok(response)
                 } else {
@@ -88,7 +91,7 @@ impl Client {
         let result = request.with_json(params).unwrap().send();
         match result {
             Ok(response) => {
-                // print!("{:?}", response.as_str().unwrap());
+                print!("{:?}", response.as_str().unwrap());
 
                 if (200..=299).contains(&response.status_code) {
                     Ok(response)
@@ -106,32 +109,9 @@ impl Client {
         }
     }
 
-    pub fn delete(&self, path: &str) -> Result<Response, APIError> {
-        let url = format!("{}{}", self.endpoint, path);
-        let request = self.build_request(minreq::post(url));
-
-        let result = request.send();
-        match result {
-            Ok(response) => {
-                if (200..=299).contains(&response.status_code) {
-                    Ok(response)
-                } else {
-                    Err(APIError {
-                        message: format!(
-                            "{}: {}",
-                            response.status_code,
-                            response.as_str().unwrap()
-                        ),
-                    })
-                }
-            }
-            Err(error) => Err(self.new_error(error)),
-        }
-    }
-
     pub fn chat(
         &self,
-        model: String,
+        model: Model,
         messages: Vec<ChatCompletionMessage>,
         options: Option<ChatCompletionParams>,
     ) -> Result<ChatCompletionResponse, APIError> {
@@ -145,6 +125,22 @@ impl Client {
         }
     }
 
+    pub fn embeddings(
+        &self,
+        model: EmbedModel,
+        input: Vec<String>,
+        options: Option<EmbeddingRequestOptions>,
+    ) -> Result<EmbeddingResponse, APIError> {
+        let request = EmbeddingRequest::new(model, input, options);
+
+        let response = self.post("/embeddings", &request)?;
+        let result = response.json::<EmbeddingResponse>();
+        match result {
+            Ok(response) => Ok(response),
+            Err(error) => Err(self.new_error(error)),
+        }
+    }
+
     pub fn list_models(&self) -> Result<ModelListResponse, APIError> {
         let response = self.get("/models")?;
         let result = response.json::<ModelListResponse>();