plugins: unknown cmd/q now use gpt3 if possible

master
dece 1 year ago
parent 237e88cf16
commit 6eccd2974b

@ -189,10 +189,6 @@
"param_source": "from",
"param_dest": "to"
},
"unknowncommand": {
"api_key": "",
"max_pods": 1
},
"unknownquestion": {
"pass_to_misc_rate": 50
},

@ -1,11 +1,19 @@
from edmond.plugin import Plugin
from edmond.plugins.gpt3 import Gpt3Plugin
class UnknownCommandPlugin(Plugin):
def __init__(self, bot):
super().__init__(bot)
self.priority = -6
self.priority: int = -6
self.gpt3_plugin: Gpt3Plugin
def on_welcome(self, _):
self.gpt3_plugin = self.bot.get_plugin("gpt3")
if self.gpt3_plugin is None or not self.gpt3_plugin:
self.bot.log_w("GPT-3 plugin is not available.")
self.is_ready = False
def on_pubmsg(self, event):
message = self.should_read_message(event.arguments[0])
@ -16,5 +24,5 @@ class UnknownCommandPlugin(Plugin):
return False
query = " ".join(words[:-1])
self.bot.say(event.target, f"UnknownCommand: {query}")
self.gpt3_plugin.reply(query, event.target)
return True

@ -7,26 +7,34 @@ from edmond.utils import proc
class UnknownQuestionPlugin(Plugin):
"""Handle unknown questions.
Reply with some question marks, but there is a configurable rate to pass
processing of the message to the misc_reactions if available.
"""
REQUIRED_CONFIGS = ["pass_to_misc_rate"]
def __init__(self, bot):
super().__init__(bot)
self.priority = -7
self.misc_plugin = None
self.gpt3_plugin = None
def on_welcome(self, event):
def on_welcome(self, _):
self.misc_plugin = self.bot.get_plugin("miscreactions")
self.gpt3_plugin = self.bot.get_plugin("gpt3")
def on_pubmsg(self, event):
message = self.should_read_message(event.arguments[0])
if message is None:
return False
if self.misc_plugin and proc(self.config["pass_to_misc_rate"]):
if self.gpt3_plugin:
return self.gpt3_plugin.reply(message, event.target)
else:
return self.classic_reply(event)
def classic_reply(self, event):
if (
self.misc_plugin
and "pass_to_misc_rate" in self.config
and proc(self.config["pass_to_misc_rate"])
):
self.misc_plugin.react(event)
else:
self.bot.say(event.target, "?" * random.randint(1, 3))
return True

Loading…
Cancel
Save