Plugins: Switch AI agent from OpenAI to Gemini

Keys are easy to get at https://aistudio.google.com/app/api-keys

Author: reggna

plugins/shrimpgpt/__init__.py plugins/shrimpgemini/__init__.pyplugins/shrimpgpt/__init__.py renamed to plugins/shrimpgemini/__init__.py

@@ -1,6 +1,6 @@
-# This plugin will act as an interactive ChatGPT.  It has the option to save the last few messages,
-# but note that each message save will incur a higher token usage when communitcating with OpenAI
-# as each message is included in the query to OpenAI.  Also note that messages sent by other
+# This plugin will act as an interactive Gemini.  It has the option to save the last few messages,
+# but note that each message save will incur a higher token usage when communitcating with Gemini
+# as each message is included in the query to Gemini.  Also note that messages sent by other
 # plugins are currently not being saved.
 #
 # You can specify a "history_timeout" if you only want the plugin to only remember messages within a
@@ -9,12 +9,12 @@
 #
 # Sample setting:
 #
-# "shrimpgpt": {
-#   "key": "your-openai-app-key",
-#   "trigger": "chatgpt:", # Only respond to direct questions
-#   "channel": "#chatgpt", # Only monitor one specific channel
+# "shrimpgemini": {
+#   "key": "your-gemini-app-key",
+#   "trigger": "gemini:", # Only respond to direct questions
+#   "channel": "#gemini", # Only monitor one specific channel
 #   "prompt": "You are an IRC bot. Users will post questions that you answer.",
-#   "model": "gpt-4o-mini",
+#   "model": "gemini-2.5-flash",
 #   "max_tokens": 256,
 #   "temperature": 0.2,
 #   "max_history": 5, # Let the plugin "remember" the past 5 messages (including responses)
@@ -26,23 +26,23 @@
 # If "channel" is not specified, the plugin will respond in every channel.
 #
 # Commands:
-# * !gpt history reset
+# * !gemini history reset
 # Will manually trigger a reset of any previous messages saved.
 
 import logging
 import sys
 import json
 import plugin
 
-from utils import openai
+from utils import gemini
 
 DEFAULT_PROMPT = """You are an IRC bot. Users will post questions that you answer."""
-DEFAULT_MODEL = "gpt-4o-mini"
+DEFAULT_MODEL = "gemini-2.5-flash"
 
 
-class shrimpgpt(plugin.Plugin):
+class shrimpgemini(plugin.Plugin):
     def __init__(self):
-        plugin.Plugin.__init__(self, "shrimpgpt")
+        plugin.Plugin.__init__(self, "shrimpgemini")
 
     def started(self, settings):
         s = json.loads(settings)
@@ -52,15 +52,15 @@ def started(self, settings):
         prompt = s["prompt"] if "prompt" in s else DEFAULT_PROMPT
         self.system_message = {"role": "system", "content": prompt}
         self.model = s["model"] if "model" in s else DEFAULT_MODEL
-        self.max_tokens = int(s["max_tokens"]) if "max_tokens" in s else 256
+        self.max_tokens = int(s["max_tokens"]) if "max_tokens" in s else 512
         self.temperature = int(s["temperature"]) if "temperature" in s else 0.2
         self.history = []
         self.max_history = int(s["max_history"]) if "max_history" in s else 5
         # Max number of seconds of inactivity before cleaning out the message history
         self.history_timeout = int(s["history_timeout"]) if "history_timeout" in s else 120
         self.update_count = 0
         logging.info(
-            "ShrimpGPT: Trigger: %s, channel: %s, model: %s, max_tokens: %i, temp: %i",
+            "ShrimpGemini: Trigger: %s, channel: %s, model: %s, max_tokens: %i, temp: %i",
             self.trigger,
             self.channel,
             self.model,
@@ -74,12 +74,12 @@ def update(self):
         self.update_count += 1
         # Reset history if update_count exceeds "update_count"
         if self.update_count >= self.history_timeout:
-            logging.info("ShrimpGPT: Resetting history: []")
+            logging.info("ShrimpGemini: Resetting history: []")
             self.history = []
             self.update_count = 0
 
     def reset_history(self, server, channel):
-        logging.info("ShrimpGPT: Manual history reset")
+        logging.info("ShrimpGemini: Manual history reset")
         self.history = []
         self.safe_privmsg(server, channel, "Reset!")
 
@@ -96,17 +96,17 @@ def respond_to_message(self, query, server, channel):
         message = {"role": "user", "content": query}
         self.add_to_history(message)
         messages = [self.system_message] + self.history
-        result = openai.get_response(
+        result = gemini.get_response(
             self.key, messages, self.model, self.max_tokens, self.temperature
         )
-        self.add_to_history({"role": "assistant", "content": result})
+        self.add_to_history({"role": "model", "content": result})
         self.safe_privmsg(server, channel, result)
 
     def on_pubmsg(self, server, user, channel, message):
         if self.channel and channel != self.channel:
             # Message is not in the specified channel
             return
-        if message.startswith("!gpt history reset"):
+        if message.startswith("!gemini history reset"):
             self._thread(self.reset_history, server, channel)
             return
         username = user.split("!", 1)[0]
@@ -122,4 +122,4 @@ def on_pubmsg(self, server, user, channel, message):
 
 
 if __name__ == "__main__":
-    sys.exit(shrimpgpt.run())
+    sys.exit(shrimpgemini.run())

plugins/shrimpgpt/shrimpgpt.py plugins/shrimpgemini/shrimpgemini.pyplugins/shrimpgpt/shrimpgpt.py renamed to plugins/shrimpgemini/shrimpgemini.py

@@ -1,36 +1,36 @@
-# Plugin that summarizes given youtube videos using OpenAI.
+# Plugin that summarizes given youtube videos using Gemini.
 #
 # Sample setting:
 #
 # "youtubesummarizer": {
-#   "key": "your-openai-app-key",
+#   "key": "your-gemini-app-key",
 #   "yt-key": "your-youtube-api-key",
 #   "channel": "#youtube",
-#   "model": "gpt-4o-mini",
+#   "model": "gemini-2.5-flash",
 #   "max_tokens": 4096,
 #   "temperature": 0.2
 # }
 #
 # The only required parameter is "key".
 # If "yt-key" is not specified, you could get a bit worse summary as some information
-# will not be passed on to openai.
+# will not be passed on to gemini.
 #
 # Commands:
-# * !gpt yt-summary https://www.youtube.com/watch?v=jNQXAC9IVRw
+# * !gemini yt-summary https://www.youtube.com/watch?v=jNQXAC9IVRw
 
 import logging
 import sys
 import json
 import plugin
 
 from youtube_transcript_api import YouTubeTranscriptApi
-from utils import youtube, openai
+from utils import youtube, gemini
 
 YT_PROMPT = """
             Summarize the content of this YouTube video.  End by giving a highlight link to the
             most important part of the video in the form of https://youtu.be/[id]?t=[timestamp]
             """
-DEFAULT_MODEL = "gpt-4o-mini"
+DEFAULT_MODEL = "gemini-2.5-flash"
 
 
 class youtubesummarizer(plugin.Plugin):
@@ -68,7 +68,7 @@ def process_youtube(self, id, server, channel):
             {"role": "system", "content": YT_PROMPT},
             {"role": "user", "content": trans_str},
         ]
-        result = openai.get_response(
+        result = gemini.get_response(
             self.key, messages, self.model, self.max_tokens, self.temperature
         )
         logging.info(result)
@@ -77,7 +77,7 @@ def process_youtube(self, id, server, channel):
     def on_pubmsg(self, server, user, channel, message):
         if self.channel and channel != self.channel:
             return
-        if message.startswith("!gpt yt-summary"):
+        if message.startswith("!gemini yt-summary"):
             for id in youtube.YouTube.find_all_ids(message):
                 self._thread(self.process_youtube, id, server, channel)
 

plugins/youtubesummarizer/youtubesummarizer.py

@@ -31,9 +31,9 @@ requests = "^2.32.3"
 feedparser = "^6.0.11"
 irc = "^20.5.0"
 pyzmq = "^26.2.0"
-openai = "^1.72.0"
 youtube-transcript-api = "^1.0.3"
 httpx = {version = "^0.28.1", extras = ["http2"]}
+google-genai = "^1.65.0"
 
 [tool.poetry.dev-dependencies]
 requests-mock = "^1.12.1"

pyproject.toml

@@ -0,0 +1,41 @@
+import traceback
+import logging
+from google import genai
+from google.genai import types
+
+
+def get_response(key, messages, model, max_tokens, temperature):
+    client = genai.Client(api_key=key)
+
+    system_instruction = None
+    contents = []
+
+    for msg in messages:
+        role = msg["role"]
+        content = msg["content"]
+        if role == "system":
+            system_instruction = content
+        elif role == "user":
+            contents.append(types.Content(role="user", parts=[types.Part.from_text(text=content)]))
+        elif role == "assistant" or role == "model":
+            contents.append(types.Content(role="model", parts=[types.Part.from_text(text=content)]))
+
+    result = "No result ...."
+    try:
+        response = client.models.generate_content(
+            model=model,
+            contents=contents,
+            config=types.GenerateContentConfig(
+                system_instruction=system_instruction,
+                max_output_tokens=max_tokens,
+                temperature=temperature,
+            ),
+        )
+
+        if response.text:
+            result = response.text
+    except Exception as e:
+        logging.error(traceback.format_exc())
+
+    logging.info("Result: " + result)
+    return result

utils/gemini.py

@@ -0,0 +1,65 @@
+import unittest
+from unittest.mock import MagicMock, patch
+from utils import gemini
+from google.genai import types
+
+
+class TestGemini(unittest.TestCase):
+    @patch("google.genai.Client")
+    def test_get_response(self, mock_client_class):
+        mock_client = MagicMock()
+        mock_client_class.return_value = mock_client
+        mock_response = MagicMock()
+        mock_response.text = "Hello, I am Gemini!"
+        mock_client.models.generate_content.return_value = mock_response
+
+        messages = [
+            {"role": "system", "content": "You are a helpful assistant."},
+            {"role": "user", "content": "Hi there!"},
+        ]
+
+        result = gemini.get_response("fake-key", messages, "gemini-1.5-flash", 256, 0.2)
+
+        mock_client_class.assert_called_once_with(api_key="fake-key")
+        mock_client.models.generate_content.assert_called_once()
+        args, kwargs = mock_client.models.generate_content.call_args
+        self.assertEqual(kwargs["model"], "gemini-1.5-flash")
+        self.assertEqual(kwargs["config"].system_instruction, "You are a helpful assistant.")
+        self.assertEqual(kwargs["config"].max_output_tokens, 256)
+        self.assertEqual(kwargs["config"].temperature, 0.2)
+        self.assertEqual(result, "Hello, I am Gemini!")
+
+    @patch("google.genai.Client")
+    def test_get_response_with_history(self, mock_client_class):
+        mock_client = MagicMock()
+        mock_client_class.return_value = mock_client
+        mock_response = MagicMock()
+        mock_response.text = "I remember you!"
+        mock_client.models.generate_content.return_value = mock_response
+
+        messages = [
+            {"role": "system", "content": "You are a helpful assistant."},
+            {"role": "user", "content": "My name is Alice."},
+            {"role": "assistant", "content": "Hello Alice!"},
+            {"role": "user", "content": "What is my name?"},
+        ]
+
+        result = gemini.get_response("fake-key", messages, "gemini-1.5-flash", 256, 0.2)
+
+        mock_client.models.generate_content.assert_called_once()
+        args, kwargs = mock_client.models.generate_content.call_args
+
+        contents = kwargs["contents"]
+        self.assertEqual(len(contents), 3)
+        self.assertEqual(contents[0].role, "user")
+        self.assertEqual(contents[0].parts[0].text, "My name is Alice.")
+        self.assertEqual(contents[1].role, "model")
+        self.assertEqual(contents[1].parts[0].text, "Hello Alice!")
+        self.assertEqual(contents[2].role, "user")
+        self.assertEqual(contents[2].parts[0].text, "What is my name?")
+
+        self.assertEqual(result, "I remember you!")
+
+
+if __name__ == "__main__":
+    unittest.main()