New script ollama_bot.py: Adds non-blocking API calls using hook_process_hashtable()

Author: prietus

python/ollama.py

@@ -0,0 +1,123 @@
+import weechat
+import json
+
+"""
+Ollama Bot for WeeChat (Non-blocking version)
+
+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.
+- Uses hook_url for non-blocking HTTP requests.
+
+Usage:
+- To ask a question manually:
+  /ollama What is Python?
+
+- To enable or disable automatic responses in channels:
+  /set plugins.var.python.ollama.highlight_response on  # Enable responses in channels
+  /set plugins.var.python.ollama.highlight_response off # Disable responses in channels
+
+- To enable or disable automatic responses in private messages:
+  /set plugins.var.python.ollama.pm_response on  # Enable PM responses
+  /set plugins.var.python.ollama.pm_response off # Disable PM responses
+
+Dependencies:
+- Requires an Ollama server running locally at http://localhost:11434/api/generate
+"""
+
+# Script metadata
+SCRIPT_NAME = "ollama"
+SCRIPT_AUTHOR = "teraflops"
+SCRIPT_VERSION = "2.1"
+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"
+
+# Register the script
+weechat.register(SCRIPT_NAME, SCRIPT_AUTHOR, SCRIPT_VERSION, SCRIPT_LICENSE, SCRIPT_DESC, "", "")
+
+# Script configuration in Weechat
+def setup_config():
+    if not weechat.config_is_set_plugin("highlight_response"):
+        weechat.config_set_plugin("highlight_response", "on")  # Enable auto-responses by default
+    if not weechat.config_is_set_plugin("pm_response"):
+        weechat.config_set_plugin("pm_response", "off")  # Disable PM responses by default
+setup_config()
+
+def fetch_ollama(data, command, return_code, out, err):
+    """Callback function that handles the response from Ollama."""
+    if return_code == 0:
+        response = json.loads(out).get("response", "No response received from Ollama.")
+        weechat.prnt("", response)  # Print the response in the chat
+    else:
+        weechat.prnt("", f"Error fetching Ollama response: {err}")
+
+def ask_ollama(message):
+    """Send a query to Ollama using hook_url (non-blocking)."""
+    data = json.dumps({"model": "gemma", "prompt": message, "stream": False})
+    headers = "Content-Type: application/json"
+
+    weechat.hook_process_hashtable(
+        "url:fetch",
+        {"url": OLLAMA_API_URL, "type": "post", "data": data, "headers": headers},
+        60000,  # Timeout in milliseconds (60 seconds)
+        "fetch_ollama",
+        ""
+    )
+
+def command_ollama(data, buffer, args):
+    """Command /ollama to manually ask Ollama a question."""
+    if not args:
+        weechat.prnt(buffer, "Usage: /ollama <question>")
+        return weechat.WEECHAT_RC_OK
+    
+    ask_ollama(args)
+    return weechat.WEECHAT_RC_OK
+
+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
+
+    buffer_type = weechat.buffer_get_string(buffer, "localvar_type")
+    is_private = buffer_type == "private"
+    username = weechat.info_get("irc_nick", "")  # Get the current IRC username
+    is_mentioned = f"@{username.lower()}" in message.lower()  # Ensure @username is explicitly mentioned
+
+    # Ignore private messages if pm_response is off
+    if is_private and weechat.config_get_plugin("pm_response") == "off":
+        return weechat.WEECHAT_RC_OK
+
+    # Only respond in private messages if it's a direct question
+    if is_private and not message.strip().endswith("?"):
+        return weechat.WEECHAT_RC_OK
+
+    # Only respond in channels if explicitly mentioned or highlighted
+    if not is_private and not is_mentioned and not int(highlight):
+        return weechat.WEECHAT_RC_OK
+
+    ask_ollama(message)
+    return weechat.WEECHAT_RC_OK
+
+def config_callback(data, option, value):
+    """Callback for Weechat configuration changes."""
+    weechat.prnt("", f"Configuration changed: {option} = {value}")
+    return weechat.WEECHAT_RC_OK
+
+# Register configuration with /set
+weechat.config_set_desc_plugin("highlight_response", "Automatically respond to mentions in channels (on/off)")
+weechat.config_set_desc_plugin("pm_response", "Automatically respond to private messages (on/off)")
+weechat.hook_config("plugins.var.python.ollama.highlight_response", "config_callback", "")
+weechat.hook_config("plugins.var.python.ollama.pm_response", "config_callback", "")
+
+# 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", "")
+weechat.hook_print("", "notify_private", "", 1, "message_callback", "")
+