Update ollamabot.py: translate comments to English

Author: prietus

python/ollama.py

@@ -2,10 +2,37 @@
 import requests
 import json
 
+"""
+Ollama Bot for WeeChat
+
+This script automatically responds to mentions in channels and private messages using an Ollama LLM running locally.
+
+Features:
+- Responds to mentions in channels.
+- Can respond to private messages if enabled.
+- Allows manual queries using the /ollama command.
+- Configurable via WeeChat /set commands.
+
+Usage:
+- To ask a question manually:
+  /ollama What is Python?
+
+- To enable or disable automatic responses in channels:
+  /set plugins.var.python.ollama_bot.highlight_response on  # Enable responses in channels
+  /set plugins.var.python.ollama_bot.highlight_response off # Disable responses in channels
+
+- To enable or disable automatic responses in private messages:
+  /set plugins.var.python.ollama_bot.pm_response on  # Enable PM responses
+  /set plugins.var.python.ollama_bot.pm_response off # Disable PM responses
+
+Dependencies:
+- Requires an Ollama server running locally at http://localhost:11434/api/generate
+"""
+
 # Script metadata
 SCRIPT_NAME = "ollama_bot"
 SCRIPT_AUTHOR = "teraflops"
-SCRIPT_VERSION = "1.6"
+SCRIPT_VERSION = "1.9"
 SCRIPT_LICENSE = "MIT"
 SCRIPT_DESC = "Automatically responds to mentions using Ollama and allows manual queries, including PMs"
 OLLAMA_API_URL = "http://localhost:11434/api/generate"
@@ -24,40 +51,55 @@ def setup_config():
 def ask_ollama(message):
     """Send a query to Ollama and return the complete response."""
     try:
-        data = {"model": "gemma", "prompt": message, "stream": False}
-        response = requests.post(OLLAMA_API_URL, json=data)
+        data = {"model": "mistral", "prompt": message, "stream": False}
+        headers = {"Content-Type": "application/json"}
+
+        response = requests.post(
+            OLLAMA_API_URL,
+            json=data,
+            headers=headers,
+            verify=False  # Change to True if you use a valid certificate
+        )
+
+        if response.status_code != 200:
+            return f"Error {response.status_code}: {response.text}"
+
         response_json = response.json()
         return response_json.get("response", "No response received from Ollama.")
-    except Exception as e:
-        return f"Error connecting to Ollama: {str(e)}"
-
-def command_ollama(data, buffer, args):
-    """Command /ollama to manually ask Ollama a question."""
-    if not args:
-        weechat.prnt(buffer, "[Ollama] Usage: /ollama <question>")
-        return weechat.WEECHAT_RC_OK
 
-    response = ask_ollama(args)
-    weechat.prnt(buffer, f"[Ollama] {response}")
-    return weechat.WEECHAT_RC_OK
+    except requests.exceptions.RequestException as e:
+        return f"Error connecting to Ollama: {str(e)}"
 
 def message_callback(data, buffer, date, tags, displayed, highlight, prefix, message):
     """Detect mentions in channels or private messages and respond automatically with Ollama."""
-    if weechat.config_get_plugin("highlight_response") == "off":
-        return weechat.WEECHAT_RC_OK  # Do not respond if auto-response is disabled
 
+    if weechat.config_get_plugin("highlight_response") == "off":
+        return weechat.WEECHAT_RC_OK
+
     buffer_type = weechat.buffer_get_string(buffer, "localvar_type")
     is_private = buffer_type == "private"
-    username = weechat.info_get("irc_nick", "")  # Get current IRC username
+    username = weechat.info_get("irc_nick", "")  # Get the current IRC username
     is_mentioned = username.lower() in message.lower()
-    
-    # Check if PM responses are disabled
+
+    # Ignore private messages if pm_response is off
     if is_private and weechat.config_get_plugin("pm_response") == "off":
         return weechat.WEECHAT_RC_OK
+
+    # Avoid responding to every PM automatically
+    if is_private and not message.strip().endswith("?"):
+        return weechat.WEECHAT_RC_OK
+
+    # Only respond in channels if mentioned
+    if not is_private and not is_mentioned and not int(highlight):
+        return weechat.WEECHAT_RC_OK
+
+    response = ask_ollama(message)
 
-    if int(highlight) or is_mentioned or is_private:
-        response = ask_ollama(message)
-        weechat.command(buffer, f"/msg {prefix} {response}")
+    if is_private:
+        weechat.command(buffer, f"/msg {prefix} {response}")  # Reply to private message
+    else:
+        weechat.command(buffer, f"/say {response}")  # Reply in the channel
+
     return weechat.WEECHAT_RC_OK
 
 def config_callback(data, option, value):
@@ -74,6 +116,5 @@ def config_callback(data, option, value):
 # Register commands and hooks
 weechat.hook_command("ollama", "Ask something to Ollama", "<question>", "Example: /ollama What is Python?", "", "command_ollama", "")
 weechat.hook_print("", "notify_highlight", "", 1, "message_callback", "")
-weechat.hook_print("", "notify_message", "", 1, "message_callback", "")  # Capture normal messages
+weechat.hook_print("", "notify_message", "", 1, "message_callback", "")
 weechat.hook_print("", "notify_private", "", 1, "message_callback", "")
-