Skip to content

Commit cf8f890

Browse files
author
Dongri Jin
committed
Add parameter
1 parent 4ff7e51 commit cf8f890

File tree

4 files changed

+62
-0
lines changed

4 files changed

+62
-0
lines changed

README.md

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,18 @@ let req = ChatCompletionRequest {
3434
role: chat_completion::MessageRole::user,
3535
content: String::from("Hello OpenAI!"),
3636
}],
37+
functions: None,
38+
function_call: None,
39+
temperature: None,
40+
top_p: None,
41+
n: None,
42+
stream: None,
43+
stop: None,
44+
max_tokens: None,
45+
presence_penalty: None,
46+
frequency_penalty: None,
47+
logit_bias: None,
48+
user: None,
3749
};
3850
```
3951

@@ -62,6 +74,16 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
6274
}],
6375
functions: None,
6476
function_call: None,
77+
temperature: None,
78+
top_p: None,
79+
n: None,
80+
stream: None,
81+
stop: None,
82+
max_tokens: None,
83+
presence_penalty: None,
84+
frequency_penalty: None,
85+
logit_bias: None,
86+
user: None,
6587
};
6688
let result = client.chat_completion(req).await?;
6789
println!("{:?}", result.choices[0].message.content);

examples/chat_completion.rs

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,16 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
1515
}],
1616
functions: None,
1717
function_call: None,
18+
temperature: None,
19+
top_p: None,
20+
n: None,
21+
stream: None,
22+
stop: None,
23+
max_tokens: None,
24+
presence_penalty: None,
25+
frequency_penalty: None,
26+
logit_bias: None,
27+
user: None,
1828
};
1929
let result = client.chat_completion(req).await?;
2030
println!("{:?}", result.choices[0].message.content);

examples/function_call.rs

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,16 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
4848
}),
4949
}]),
5050
function_call: Some("auto".to_string()),
51+
temperature: None,
52+
top_p: None,
53+
n: None,
54+
stream: None,
55+
stop: None,
56+
max_tokens: None,
57+
presence_penalty: None,
58+
frequency_penalty: None,
59+
logit_bias: None,
60+
user: None,
5161
};
5262

5363
let result = client.chat_completion(req).await?;

src/v1/chat_completion.rs

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,26 @@ pub struct ChatCompletionRequest {
2121
pub functions: Option<Vec<Function>>,
2222
#[serde(skip_serializing_if = "Option::is_none")]
2323
pub function_call: Option<String>,
24+
#[serde(skip_serializing_if = "Option::is_none")]
25+
pub temperature: Option<f64>,
26+
#[serde(skip_serializing_if = "Option::is_none")]
27+
pub top_p: Option<f64>,
28+
#[serde(skip_serializing_if = "Option::is_none")]
29+
pub n: Option<i64>,
30+
#[serde(skip_serializing_if = "Option::is_none")]
31+
pub stream: Option<bool>,
32+
#[serde(skip_serializing_if = "Option::is_none")]
33+
pub stop: Option<Vec<String>>,
34+
#[serde(skip_serializing_if = "Option::is_none")]
35+
pub max_tokens: Option<i64>,
36+
#[serde(skip_serializing_if = "Option::is_none")]
37+
pub presence_penalty: Option<f64>,
38+
#[serde(skip_serializing_if = "Option::is_none")]
39+
pub frequency_penalty: Option<f64>,
40+
#[serde(skip_serializing_if = "Option::is_none")]
41+
pub logit_bias: Option<HashMap<String, i32>>,
42+
#[serde(skip_serializing_if = "Option::is_none")]
43+
pub user: Option<String>,
2444
}
2545

2646
#[derive(Debug, Serialize, Deserialize)]

0 commit comments

Comments
 (0)