parent
c7a0522ee1
commit
1bee375d29
GPG Key ID:
0104BC05F41B77B8
@ -1,40 +1,223 @@
|
|||||||
mod openai;
|
mod openai;
|
||||||
use openai::{ChatCompletion, Configuration, Message, OpenAiApiClient, Role};
|
use futures::StreamExt;
|
||||||
use std::env;
|
use serenity::{
|
||||||
|
async_trait,
|
||||||
|
model::{channel::Message, gateway::Ready, prelude::*},
|
||||||
|
prelude::*,
|
||||||
|
utils::Color,
|
||||||
|
};
|
||||||
|
use std::{env, time::Duration};
|
||||||
|
|
||||||
|
struct Handler {
|
||||||
|
channel: ChannelId,
|
||||||
|
client: openai::OpenAiApiClient,
|
||||||
|
reset_time: tokio::sync::RwLock<time::OffsetDateTime>,
|
||||||
|
model: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[async_trait]
|
||||||
|
impl EventHandler for Handler {
|
||||||
|
async fn message(&self, ctx: Context, msg: Message) {
|
||||||
|
if msg.content == "\\reset" {
|
||||||
|
let mut reset_time = self.reset_time.write().await;
|
||||||
|
*reset_time = time::OffsetDateTime::now_utc();
|
||||||
|
}
|
||||||
|
|
||||||
|
if msg.channel_id != self.channel
|
||||||
|
|| msg.webhook_id.is_some()
|
||||||
|
|| msg.author.bot
|
||||||
|
|| msg.content.starts_with('\\')
|
||||||
|
{
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if msg.content == "die" {
|
||||||
|
std::process::exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
let channel = msg.channel(&ctx).await.unwrap().guild().unwrap();
|
||||||
|
|
||||||
|
let typing = channel.clone().start_typing(&ctx.http).unwrap();
|
||||||
|
|
||||||
|
let after = (((self
|
||||||
|
.reset_time
|
||||||
|
.read()
|
||||||
|
.await
|
||||||
|
.max(time::OffsetDateTime::now_utc() - Duration::from_secs(5 * 60))
|
||||||
|
.unix_timestamp()
|
||||||
|
* 1000)
|
||||||
|
- 1420070400000)
|
||||||
|
<< 22) as u64;
|
||||||
|
|
||||||
|
let mut msgs = channel
|
||||||
|
.messages(&ctx, |messages| messages.after(after))
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
tokio::time::sleep(Duration::from_millis(250)).await;
|
||||||
|
|
||||||
|
let current_id = ctx.http.get_current_user().await.unwrap().id;
|
||||||
|
|
||||||
|
msgs.reverse();
|
||||||
|
|
||||||
|
for pk_msg in msgs.clone().into_iter().filter(|message| message.webhook_id.is_some()) {
|
||||||
|
if let Some(id) = msgs.iter().rposition(|message| message.id < pk_msg.id && message.content.contains(&pk_msg.content)) {
|
||||||
|
msgs.remove(id);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let ref_ctx = &ctx;
|
||||||
|
|
||||||
|
let msgs = msgs.into_iter().filter(|message| {
|
||||||
|
#[allow(clippy::if_same_then_else)]
|
||||||
|
#[allow(clippy::needless_bool)]
|
||||||
|
if message.webhook_id.is_some() && !message.content.trim().starts_with('\\') {
|
||||||
|
true
|
||||||
|
} else if message.content.starts_with('\\') {
|
||||||
|
false
|
||||||
|
} else if message.author.bot && message.author.id != current_id {
|
||||||
|
false
|
||||||
|
} else {
|
||||||
|
true
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
let last_msg = msgs.clone().last().unwrap();
|
||||||
|
|
||||||
|
let msgs = tokio_stream::iter(msgs)
|
||||||
|
.map(|message| async move {
|
||||||
|
if message.author.id == current_id {
|
||||||
|
openai::Message {
|
||||||
|
content: message.content.clone(),
|
||||||
|
role: openai::Role::Assistant,
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
openai::Message {
|
||||||
|
content: format!(
|
||||||
|
"{}: {}",
|
||||||
|
message
|
||||||
|
.author_nick(ref_ctx)
|
||||||
|
.await
|
||||||
|
.unwrap_or(message.author.name.clone()),
|
||||||
|
message.content
|
||||||
|
),
|
||||||
|
role: openai::Role::User,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.buffered(100)
|
||||||
|
.collect::<Vec<_>>()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
let response = self
|
||||||
|
.client
|
||||||
|
.create_chat_completion(openai::ChatCompletion {
|
||||||
|
model: self.model.clone(),
|
||||||
|
messages: msgs,
|
||||||
|
..Default::default()
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let response = response.choices[0].message.content.as_str();
|
||||||
|
|
||||||
|
let is_appropriate = !self
|
||||||
|
.client
|
||||||
|
.create_moderation(openai::ModerationRequest {
|
||||||
|
input: response.to_owned(),
|
||||||
|
model: None,
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap()
|
||||||
|
.results[0]
|
||||||
|
.flagged;
|
||||||
|
|
||||||
|
typing.stop().unwrap();
|
||||||
|
if is_appropriate {
|
||||||
|
last_msg.reply(&ctx, response).await.unwrap();
|
||||||
|
} else {
|
||||||
|
msg.channel(&ctx)
|
||||||
|
.await
|
||||||
|
.unwrap()
|
||||||
|
.guild()
|
||||||
|
.unwrap()
|
||||||
|
.send_message(&ctx, |message| {
|
||||||
|
message.embed(|embed| {
|
||||||
|
embed
|
||||||
|
.title("Response flagged!")
|
||||||
|
.description("The generated response may have been inappropriate.")
|
||||||
|
.color(Color::RED)
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn ready(&self, ctx: Context, ready: Ready) {
|
||||||
|
println!("{} is connected!", ready.user.name);
|
||||||
|
self.channel.say(&ctx, "\\Smolhaj reset").await.unwrap();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[tokio::main]
|
#[tokio::main]
|
||||||
async fn main() {
|
async fn main() {
|
||||||
let api_key =
|
let api_key =
|
||||||
env::var("OPENAI_API_KEY").expect("Error: OPENAI_API_KEY environment variable not found");
|
env::var("OPENAI_API_KEY").expect("Error: OPENAI_API_KEY environment variable not found");
|
||||||
|
|
||||||
|
let token = env::var("TOKEN").expect("Error: TOKEN environment variable not found");
|
||||||
|
|
||||||
|
let channel = env::var("CHANNEL")
|
||||||
|
.expect("Error: CHANNEL environment variable not found")
|
||||||
|
.parse()
|
||||||
|
.expect("Expected channel ID to be an integer");
|
||||||
|
|
||||||
|
let model = env::var("MODEL").expect("Error: MODEL environment variable not found");
|
||||||
|
|
||||||
// Replace with your organization ID (if applicable)
|
// Replace with your organization ID (if applicable)
|
||||||
let config = Configuration {
|
let config = openai::Configuration {
|
||||||
api_key,
|
api_key,
|
||||||
organization_id: None,
|
organization_id: None,
|
||||||
};
|
};
|
||||||
|
|
||||||
let client = OpenAiApiClient::new(config);
|
let openai_client = openai::OpenAiApiClient::new(config);
|
||||||
|
|
||||||
let messages = vec![Message {
|
let intents = GatewayIntents::GUILD_MESSAGES | GatewayIntents::MESSAGE_CONTENT;
|
||||||
role: Role::User,
|
|
||||||
content: String::from("Say this is a test!"),
|
|
||||||
}];
|
|
||||||
|
|
||||||
let chat_completion = ChatCompletion {
|
let mut client = Client::builder(&token, intents)
|
||||||
model: String::from("gpt-3.5-turbo"),
|
.event_handler(Handler {
|
||||||
messages,
|
channel: ChannelId(channel),
|
||||||
..Default::default()
|
client: openai_client,
|
||||||
};
|
reset_time: time::OffsetDateTime::now_utc().into(),
|
||||||
|
model
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.expect("Err creating client");
|
||||||
|
|
||||||
match client.create_chat_completion(chat_completion).await {
|
if let Err(why) = client.start().await {
|
||||||
Ok(response) => {
|
println!("Client error: {:?}", why);
|
||||||
println!(
|
|
||||||
"Chat completion response: {:?}",
|
|
||||||
response.choices.first().unwrap().message.content
|
|
||||||
);
|
|
||||||
}
|
|
||||||
Err(error) => {
|
|
||||||
eprintln!("Error: {}", error);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// let messages = vec![openai::Message {
|
||||||
|
// role: openai::Role::User,
|
||||||
|
// content: String::from("Say this is a test!"),
|
||||||
|
// }];
|
||||||
|
|
||||||
|
// let chat_completion = openai::ChatCompletion {
|
||||||
|
// model: String::from("gpt-3.5-turbo"),
|
||||||
|
// messages,
|
||||||
|
// ..Default::default()
|
||||||
|
// };
|
||||||
|
|
||||||
|
// match client.create_chat_completion(chat_completion).await {
|
||||||
|
// Ok(response) => {
|
||||||
|
// println!(
|
||||||
|
// "Chat completion response: {:?}",
|
||||||
|
// response.choices.first().unwrap().message.content
|
||||||
|
// );
|
||||||
|
// }
|
||||||
|
// Err(error) => {
|
||||||
|
// eprintln!("Error: {}", error);
|
||||||
|
// }
|
||||||
|
// }
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
Reference in new issue