From 7a15ad97f694091051836efa077ce30f366ceb64 Mon Sep 17 00:00:00 2001 From: Gordon Shumway Date: Wed, 14 Aug 2024 13:00:24 -0400 Subject: [PATCH] Gemini: Add 'enabled' config --- Gemini/config.py | 9 +++++++++ Gemini/plugin.py | 2 ++ 2 files changed, 11 insertions(+) diff --git a/Gemini/config.py b/Gemini/config.py index d69a16a..994b881 100644 --- a/Gemini/config.py +++ b/Gemini/config.py @@ -55,6 +55,15 @@ Gemini = conf.registerPlugin("Gemini") # conf.registerGlobalValue(Gemini, 'someConfigVariableName', # registry.Boolean(False, _("""Help for someConfigVariableName."""))) +conf.registerChannelValue( + Gemini, + "enabled", + registry.Boolean( + True, + _("""Set False to disable the plugin, True to enable."""), + ), +) + conf.registerChannelValue( Gemini, "api_key", diff --git a/Gemini/plugin.py b/Gemini/plugin.py index 451da56..80c3aec 100644 --- a/Gemini/plugin.py +++ b/Gemini/plugin.py @@ -53,6 +53,8 @@ class Gemini(callbacks.Plugin): channel = msg.channel if not irc.isChannel(channel): channel = msg.nick + if not self.registryValue("enabled", msg.channel): + return genai.configure(api_key=self.registryValue("api_key", msg.channel)) prompt = self.registryValue("prompt", msg.channel).replace("$botnick", irc.nick) max_tokens = self.registryValue("max_tokens", msg.channel)