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 struct ChatMessage {
pub role: Role, pub role: Role,
pub content: String, pub content: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub name: Option<String>,
} }
impl ChatMessage { 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 { Self {
role, 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}"); println!("Generating completion for prompt: {PROMPT}");
let completion = ctx.create_chat_completion_sync( let completion = ctx.create_chat_completion_sync(
ChatHistoryBuilder::default() ChatHistoryBuilder::default()
.messages(vec![ChatMessage::new(Role::User, PROMPT)]) .messages(vec![ChatMessage::new(Role::User, PROMPT, None)])
.model("gpt-3.5-turbo") .model("gpt-3.5-turbo")
).await; ).await;
@ -89,7 +89,7 @@ mod tests {
println!("Generating streamed completion for prompt: {PROMPT}"); println!("Generating streamed completion for prompt: {PROMPT}");
let completion = ctx.create_chat_completion_streamed( let completion = ctx.create_chat_completion_streamed(
ChatHistoryBuilder::default() ChatHistoryBuilder::default()
.messages(vec![ChatMessage::new(Role::User, PROMPT)]) .messages(vec![ChatMessage::new(Role::User, PROMPT, None)])
.model("gpt-3.5-turbo") .model("gpt-3.5-turbo")
).await; ).await;