Merge pull request #74 from dongri/add-dimensions

Add dimensions parameter

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);

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,
         }
     }