niansa/discord_llama
1
0
Fork 0
mirror of https://gitlab.com/niansa/discord_llama.git synced 2025-03-06 20:48:25 +01:00
Code Releases Activity

Append unrelated messages to history

Browse source
This commit is contained in:
niansa 2023-04-17 00:31:11 +02:00
parent 1ee724afa3
commit 7f9a32b52a
1 changed files with 13 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
main.cpp
View file

@ -315,7 +315,7 @@ class Bot {
}
// Must run in llama thread
void attempt_reply(const dpp::message& msg, const std::function<void ()>& after_placeholder_creation = nullptr) {
bool attempt_reply(const dpp::message& msg, const std::function<void ()>& after_placeholder_creation = nullptr) {
ENSURE_LLM_THREAD();
// Decide randomly
/*if (rng.getBool(0.075f)) {
@ -323,14 +323,18 @@ class Bot {
}*/
// Reply if message contains username, mention or ID
if (msg.content.find(bot.me.username) != std::string::npos) {
return reply(after_placeholder_creation);
reply(after_placeholder_creation);
return true;
}
// Reply if message references user
for (const auto msg_id : my_messages) {
if (msg.message_reference.message_id == msg_id) {
return reply(after_placeholder_creation);
reply(after_placeholder_creation);
return true;
}
}
// Don't reply otherwise
return false;
}
void enqueue_reply() {
@ -403,10 +407,15 @@ public:
} else {
tPool.submit([=, this] () {
// Attempt to send a reply
attempt_reply(msg, [=, this] () {
bool replied = attempt_reply(msg, [=, this] () {
// Append message to history
prompt_add_msg(msg);
});
// If none was send, still append message to history.
if (!replied) {
// Append message to history
prompt_add_msg(msg);
}
});
}
} catch (const std::exception& e) {