valter/dj-kitty-cat
1
0
Fork 0
Code Issues 7 Pull requests Projects 1 Releases Wiki Activity

Update openai crate

Browse source
This commit is contained in:
Alex Page 2023-03-06 03:05:38 -05:00
parent 2de52dfe9d
commit eb56c024cd
2 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
Cargo.toml
View file

@ -10,7 +10,7 @@ poise = "0.5.2"
tracing = "0.1.37" tracing = "0.1.37"
tracing-subscriber = "0.3.16" tracing-subscriber = "0.3.16"
tracing-futures = "0.2.5" tracing-futures = "0.2.5"
openai = "1.0.0-alpha.5" openai = "1.0.0-alpha.6"
rand = "0.8.5" rand = "0.8.5"
[dependencies.songbird] [dependencies.songbird]

10
src/personality.rs
View file

@ -1,6 +1,6 @@
use anyhow::{Context, Result}; use anyhow::{Context, Result};
use openai::{ use openai::{
chat::{ChatCompletion, ChatCompletionRequestMessage}, chat::{ChatCompletion, ChatCompletionMessage},
models::ModelID, models::ModelID,
}; };
use rand::seq::SliceRandom; use rand::seq::SliceRandom;
@ -55,24 +55,24 @@ pub async fn get_sassy_commentary(title: &str) -> Result<String> {
[ [
system system
.into_iter() .into_iter()
.map(|s| ChatCompletionRequestMessage { .map(|s| ChatCompletionMessage {
role: openai::chat::ChatCompletionMessageRole::System, role: openai::chat::ChatCompletionMessageRole::System,
content: String::from(s), content: String::from(s),
name: None, name: None,
}) })
.collect::<Vec<_>>(), .collect::<Vec<_>>(),
vec![ vec![
ChatCompletionRequestMessage { ChatCompletionMessage {
role: openai::chat::ChatCompletionMessageRole::User, role: openai::chat::ChatCompletionMessageRole::User,
content: String::from(example_prompt), content: String::from(example_prompt),
name: None, name: None,
}, },
ChatCompletionRequestMessage { ChatCompletionMessage {
role: openai::chat::ChatCompletionMessageRole::Assistant, role: openai::chat::ChatCompletionMessageRole::Assistant,
content: String::from(example_response), content: String::from(example_response),
name: None, name: None,
}, },
ChatCompletionRequestMessage { ChatCompletionMessage {
role: openai::chat::ChatCompletionMessageRole::User, role: openai::chat::ChatCompletionMessageRole::User,
content: prompt, content: prompt,
name: None, name: None,