Plugins: Update the Gemini system prompt

Also making sure we're injecting the current time and date.

Author: reggna

plugins/shrimpai/shrimpai.py

@@ -32,11 +32,17 @@
 import logging
 import sys
 import json
+import datetime
 import plugin
 
 from utils import gemini
 
-DEFAULT_PROMPT = """You are an IRC bot. Users will post questions that you answer."""
+DEFAULT_PROMPT = (
+    "You are an IRC bot. Users will post questions that you answer. Keep your responses brief and "
+    "as precise as possible.  If you are not sure about an answer, make that clear but don't be "
+    "overly vague.  The user can ask questions in any language and, unless otherwise "
+    "instructed, you will respond in the same langage as the user."
+)
 DEFAULT_MODEL = "gemini-2.5-flash"
 
 
@@ -53,7 +59,7 @@ def started(self, settings):
         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 512
-        self.temperature = int(s["temperature"]) if "temperature" in s else 0.2
+        self.temperature = float(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
@@ -95,7 +101,13 @@ def add_to_history(self, message):
     def respond_to_message(self, query, server, channel):
         message = {"role": "user", "content": query}
         self.add_to_history(message)
-        messages = [self.system_message] + self.history
+
+        current_time = datetime.datetime.now().astimezone().strftime("%A, %B %d, %Y %H:%M:%S %Z")
+        prompt = self.system_message["content"]
+        full_prompt = f"{prompt}\n\nCurrent date and time: {current_time}"
+        system_message = {"role": "system", "content": full_prompt}
+
+        messages = [system_message] + self.history
         result = gemini.get_response(
             self.key, messages, self.model, self.max_tokens, self.temperature
         )

plugins/shrimpgemini/test/__init__.py

@@ -0,0 +1 @@
+__author__ = "reggna"

plugins/shrimpgemini/test/test_shrimpgemini.py

@@ -0,0 +1,56 @@
+import unittest
+from unittest.mock import MagicMock, patch
+import json
+import datetime
+from plugins.shrimpgemini.shrimpgemini import shrimpgemini
+
+
+class TestShrimpGemini(unittest.TestCase):
+    def setUp(self):
+        self.plugin = shrimpgemini()
+        # Mock settings for started()
+        self.settings = json.dumps(
+            {"key": "test-key", "trigger": "gemini:", "channel": "#test", "temperature": "0.5"}
+        )
+        self.plugin.started(self.settings)
+
+    def test_started_temperature_is_float(self):
+        self.assertEqual(self.plugin.temperature, 0.5)
+        self.assertIsInstance(self.plugin.temperature, float)
+
+    @patch("utils.gemini.get_response")
+    def test_respond_to_message_includes_date_time(self, mock_get_response):
+        mock_get_response.return_value = "Test response"
+        self.plugin.safe_privmsg = MagicMock()
+
+        # Test query
+        query = "What time is it?"
+        server = "test-server"
+        channel = "#test"
+
+        # Call the method
+        self.plugin.respond_to_message(query, server, channel)
+
+        # Check call arguments
+        # messages should be the second argument
+        args, kwargs = mock_get_response.call_args
+        messages = args[1]
+
+        # The system message should be the first in the list
+        system_msg = messages[0]
+        self.assertEqual(system_msg["role"], "system")
+
+        # Check if "Current date and time" is in the system message content
+        self.assertIn("Current date and time:", system_msg["content"])
+
+        # Verify the date format
+        now_str = datetime.datetime.now().astimezone().strftime("%A, %B %d, %Y")
+        self.assertIn(now_str, system_msg["content"])
+
+        # Verify timezone is present (e.g., UTC, CET, etc.)
+        tz_str = datetime.datetime.now().astimezone().strftime("%Z")
+        self.assertIn(tz_str, system_msg["content"])
+
+
+if __name__ == "__main__":
+    unittest.main()