plugins: unknown cmd/q now use gpt3 if possible
This commit is contained in:
parent
237e88cf16
commit
6eccd2974b
|
@ -189,10 +189,6 @@
|
||||||
"param_source": "from",
|
"param_source": "from",
|
||||||
"param_dest": "to"
|
"param_dest": "to"
|
||||||
},
|
},
|
||||||
"unknowncommand": {
|
|
||||||
"api_key": "",
|
|
||||||
"max_pods": 1
|
|
||||||
},
|
|
||||||
"unknownquestion": {
|
"unknownquestion": {
|
||||||
"pass_to_misc_rate": 50
|
"pass_to_misc_rate": 50
|
||||||
},
|
},
|
||||||
|
|
|
@ -1,11 +1,19 @@
|
||||||
from edmond.plugin import Plugin
|
from edmond.plugin import Plugin
|
||||||
|
from edmond.plugins.gpt3 import Gpt3Plugin
|
||||||
|
|
||||||
|
|
||||||
class UnknownCommandPlugin(Plugin):
|
class UnknownCommandPlugin(Plugin):
|
||||||
|
|
||||||
def __init__(self, bot):
|
def __init__(self, bot):
|
||||||
super().__init__(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):
|
def on_pubmsg(self, event):
|
||||||
message = self.should_read_message(event.arguments[0])
|
message = self.should_read_message(event.arguments[0])
|
||||||
|
@ -16,5 +24,5 @@ class UnknownCommandPlugin(Plugin):
|
||||||
return False
|
return False
|
||||||
|
|
||||||
query = " ".join(words[:-1])
|
query = " ".join(words[:-1])
|
||||||
self.bot.say(event.target, f"UnknownCommand: {query}")
|
self.gpt3_plugin.reply(query, event.target)
|
||||||
return True
|
return True
|
||||||
|
|
|
@ -7,26 +7,34 @@ from edmond.utils import proc
|
||||||
class UnknownQuestionPlugin(Plugin):
|
class UnknownQuestionPlugin(Plugin):
|
||||||
"""Handle unknown questions.
|
"""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):
|
def __init__(self, bot):
|
||||||
super().__init__(bot)
|
super().__init__(bot)
|
||||||
self.priority = -7
|
self.priority = -7
|
||||||
self.misc_plugin = None
|
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.misc_plugin = self.bot.get_plugin("miscreactions")
|
||||||
|
self.gpt3_plugin = self.bot.get_plugin("gpt3")
|
||||||
|
|
||||||
def on_pubmsg(self, event):
|
def on_pubmsg(self, event):
|
||||||
message = self.should_read_message(event.arguments[0])
|
message = self.should_read_message(event.arguments[0])
|
||||||
if message is None:
|
if message is None:
|
||||||
return False
|
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)
|
self.misc_plugin.react(event)
|
||||||
else:
|
else:
|
||||||
self.bot.say(event.target, "?" * random.randint(1, 3))
|
self.bot.say(event.target, "?" * random.randint(1, 3))
|
||||||
return True
|
|
||||||
|
|
Loading…
Reference in a new issue