GabrielTofvesson/OpenAI-Rust
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Implement optional author name for ChatMessage

Browse Source
This commit is contained in:
Gabriel Tofvesson 2023-06-12 16:03:07 +02:00
parent ff9f8f9339
commit 6310c2f8c8
No known key found for this signature in database
GPG Key ID: 6F1345DF28EDA13E
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/chat.rs
View File

@ -46,13 +46,17 @@ impl<'de> Deserialize<'de> for Role {
pub struct ChatMessage {
pub role: Role,
pub content: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub name: Option<String>,
}
impl ChatMessage {
pub fn new(role: Role, message: impl Into<String>) -> Self {
pub fn new(role: Role, message: impl Into<String>, name: Option<String>) -> Self {
Self {
role,
content: message.into()
content: message.into(),
name
}
}
}

4
src/lib.rs
View File

@ -76,7 +76,7 @@ mod tests {
println!("Generating completion for prompt: {PROMPT}");
let completion = ctx.create_chat_completion_sync(
ChatHistoryBuilder::default()
.messages(vec![ChatMessage::new(Role::User, PROMPT)])
.messages(vec![ChatMessage::new(Role::User, PROMPT, None)])
.model("gpt-3.5-turbo")
).await;
@ -89,7 +89,7 @@ mod tests {
println!("Generating streamed completion for prompt: {PROMPT}");
let completion = ctx.create_chat_completion_streamed(
ChatHistoryBuilder::default()
.messages(vec![ChatMessage::new(Role::User, PROMPT)])
.messages(vec![ChatMessage::new(Role::User, PROMPT, None)])
.model("gpt-3.5-turbo")
).await;