New script ollama.py: AI auto-responder using Ollama

Author: prietus

python/ollama.py

@@ -0,0 +1,71 @@
+import weechat
+import requests
+import json
+
+# Configuración inicial
+SCRIPT_NAME = "ollama_bot"
+SCRIPT_AUTHOR = "teraflops"
+SCRIPT_VERSION = "1.5"
+SCRIPT_LICENSE = "MIT"
+SCRIPT_DESC = "Responde automáticamente a menciones con Ollama y permite consultas manuales, incluyendo PMs"
+OLLAMA_API_URL = "http://localhost:11434/api/generate"
+
+# Registrar el script
+weechat.register(SCRIPT_NAME, SCRIPT_AUTHOR, SCRIPT_VERSION, SCRIPT_LICENSE, SCRIPT_DESC, "", "")
+
+# Configuración del script en Weechat
+def setup_config():
+    if not weechat.config_is_set_plugin("highlight_response"):
+        weechat.config_set_plugin("highlight_response", "on")  # Responder a menciones y PMs por defecto
+setup_config()
+
+def ask_ollama(message):
+    """Consulta a Ollama enviando un mensaje y devuelve la respuesta completa."""
+    try:
+        data = {"model": "gemma", "prompt": message, "stream": False}
+        response = requests.post(OLLAMA_API_URL, json=data)
+        response_json = response.json()
+        return response_json.get("response", "No se recibió respuesta de Ollama.")
+    except Exception as e:
+        return f"Error al conectar con Ollama: {str(e)}"
+
+def command_ollama(data, buffer, args):
+    """Comando /ollama para preguntar manualmente a Ollama."""
+    if not args:
+        weechat.prnt(buffer, "[Ollama] Uso: /ollama <pregunta>")
+        return weechat.WEECHAT_RC_OK
+    
+    response = ask_ollama(args)
+    weechat.prnt(buffer, f"[Ollama] {response}")
+    return weechat.WEECHAT_RC_OK
+
+def message_callback(data, buffer, date, tags, displayed, highlight, prefix, message):
+    """Detecta menciones en canales o mensajes privados y responde automáticamente con Ollama."""
+    if weechat.config_get_plugin("highlight_response") == "off":
+        return weechat.WEECHAT_RC_OK  # No responde si la opción está deshabilitada
+    
+    buffer_type = weechat.buffer_get_string(buffer, "localvar_type")
+    is_private = buffer_type == "private"
+    username = weechat.info_get("irc_nick", "")  # Obtiene tu nombre de usuario en IRC
+    is_mentioned = username.lower() in message.lower()
+    
+    if int(highlight) or is_private or is_mentioned:
+        response = ask_ollama(message)
+        weechat.command(buffer, f"/msg {prefix} {response}")
+    return weechat.WEECHAT_RC_OK
+
+def config_callback(data, option, value):
+    """Callback para cambios en la configuración de Weechat."""
+    weechat.prnt("", f"[Ollama] Configuración cambiada: {option} = {value}")
+    return weechat.WEECHAT_RC_OK
+
+# Registrar configuración con /set
+weechat.config_set_desc_plugin("highlight_response", "Responde automáticamente a menciones y PMs (on/off)")
+weechat.hook_config("plugins.var.python.ollama_bot.highlight_response", "config_callback", "")
+
+# Registrar comandos y hooks
+weechat.hook_command("ollama", "Pregunta algo a Ollama", "<pregunta>", "Ejemplo: /ollama ¿Qué es Python?", "", "command_ollama", "")
+weechat.hook_print("", "notify_highlight", "", 1, "message_callback", "")
+weechat.hook_print("", "notify_message", "", 1, "message_callback", "")  # Captura mensajes normales
+weechat.hook_print("", "notify_private", "", 1, "message_callback", "")
+