From f1685c96bd8746ca596275cbf87f44ef484b8996 Mon Sep 17 00:00:00 2001
From: Dongri Jin <dongrify@gmail.com>
Date: Mon, 29 Jan 2024 05:03:13 +0900
Subject: [PATCH] Add dimensions parameter

---
 examples/embedding.rs | 4 +++-
 src/v1/embedding.rs   | 3 +++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/examples/embedding.rs b/examples/embedding.rs
index da7f201..f415763 100644
--- a/examples/embedding.rs
+++ b/examples/embedding.rs
@@ -6,7 +6,9 @@ use std::env;
 fn main() -> Result<(), Box<dyn std::error::Error>> {
     let client = Client::new(env::var("OPENAI_API_KEY").unwrap().to_string());
 
-    let req = EmbeddingRequest::new(TEXT_EMBEDDING_3_SMALL.to_string(), "story time".to_string());
+    let mut req =
+        EmbeddingRequest::new(TEXT_EMBEDDING_3_SMALL.to_string(), "story time".to_string());
+    req.dimensions = Some(10);
 
     let result = client.embedding(req)?;
     println!("{:?}", result.data);
diff --git a/src/v1/embedding.rs b/src/v1/embedding.rs
index ea897b3..ea11414 100644
--- a/src/v1/embedding.rs
+++ b/src/v1/embedding.rs
@@ -15,6 +15,8 @@ pub struct EmbeddingRequest {
     pub model: String,
     pub input: String,
     #[serde(skip_serializing_if = "Option::is_none")]
+    pub dimensions: Option<i32>,
+    #[serde(skip_serializing_if = "Option::is_none")]
     pub user: Option<String>,
 }
 
@@ -23,6 +25,7 @@ impl EmbeddingRequest {
         Self {
             model,
             input,
+            dimensions: None,
             user: None,
         }
     }