Merge branch 'main' of github.com:dongri/openai-api-rs

dongri · Jul 9, 2023 · 4521588 · 4521588
2 parents d87fa3e + 3b5a970
commit 4521588
Show file tree

Hide file tree

Showing 4 changed files with 62 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -34,6 +34,18 @@ let req = ChatCompletionRequest {
         role: chat_completion::MessageRole::user,
         content: String::from("Hello OpenAI!"),
     }],
+    functions: None,
+    function_call: None,
+    temperature: None,
+    top_p: None,
+    n: None,
+    stream: None,
+    stop: None,
+    max_tokens: None,
+    presence_penalty: None,
+    frequency_penalty: None,
+    logit_bias: None,
+    user: None,
 };
 ```
 
@@ -62,6 +74,16 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
         }],
         functions: None,
         function_call: None,
+        temperature: None,
+        top_p: None,
+        n: None,
+        stream: None,
+        stop: None,
+        max_tokens: None,
+        presence_penalty: None,
+        frequency_penalty: None,
+        logit_bias: None,
+        user: None,
     };
     let result = client.chat_completion(req).await?;
     println!("{:?}", result.choices[0].message.content);

diff --git a/examples/chat_completion.rs b/examples/chat_completion.rs
@@ -15,6 +15,16 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
         }],
         functions: None,
         function_call: None,
+        temperature: None,
+        top_p: None,
+        n: None,
+        stream: None,
+        stop: None,
+        max_tokens: None,
+        presence_penalty: None,
+        frequency_penalty: None,
+        logit_bias: None,
+        user: None,
     };
     let result = client.chat_completion(req).await?;
     println!("{:?}", result.choices[0].message.content);

diff --git a/examples/function_call.rs b/examples/function_call.rs
@@ -48,6 +48,16 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
             }),
         }]),
         function_call: Some("auto".to_string()),
+        temperature: None,
+        top_p: None,
+        n: None,
+        stream: None,
+        stop: None,
+        max_tokens: None,
+        presence_penalty: None,
+        frequency_penalty: None,
+        logit_bias: None,
+        user: None,
     };
 
     let result = client.chat_completion(req).await?;

diff --git a/src/v1/chat_completion.rs b/src/v1/chat_completion.rs
@@ -21,6 +21,26 @@ pub struct ChatCompletionRequest {
     pub functions: Option<Vec<Function>>,
     #[serde(skip_serializing_if = "Option::is_none")]
     pub function_call: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub temperature: Option<f64>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub top_p: Option<f64>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub n: Option<i64>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub stream: Option<bool>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub stop: Option<Vec<String>>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub max_tokens: Option<i64>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub presence_penalty: Option<f64>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub frequency_penalty: Option<f64>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub logit_bias: Option<HashMap<String, i32>>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub user: Option<String>,
 }
 
 #[derive(Debug, Serialize, Deserialize)]