125 lines
4.3 KiB
Python
125 lines
4.3 KiB
Python
import weechat
|
|
import json
|
|
|
|
SCRIPT_NAME = "ollama_online"
|
|
SCRIPT_AUTHOR = "teraflops"
|
|
SCRIPT_VERSION = "2.1"
|
|
SCRIPT_LICENSE = "MIT"
|
|
SCRIPT_DESC = "Automatically responds via Ollama at https://ollama.priet.us with Basic Auth"
|
|
OLLAMA_API_URL = "https://ollama.priet.us/api/generate"
|
|
OLLAMA_USER = "nginx-user"
|
|
OLLAMA_PASSWORD = "password" # Ajusta a tus credenciales
|
|
|
|
weechat.register(SCRIPT_NAME, SCRIPT_AUTHOR, SCRIPT_VERSION, SCRIPT_LICENSE, SCRIPT_DESC, "", "")
|
|
|
|
def setup_config():
|
|
if not weechat.config_is_set_plugin("highlight_response"):
|
|
weechat.config_set_plugin("highlight_response", "on")
|
|
if not weechat.config_is_set_plugin("pm_response"):
|
|
weechat.config_set_plugin("pm_response", "off")
|
|
setup_config()
|
|
|
|
def ask_ollama_async(prompt, buffer, prefix=""):
|
|
payload = json.dumps({
|
|
"model": "gemma2:2b",
|
|
"prompt": prompt,
|
|
"stream": False
|
|
})
|
|
|
|
escaped_payload = payload.replace('"', '\\"')
|
|
curl_path = "/usr/bin/curl" # Ajusta ruta si no está en /usr/bin
|
|
cmd = (
|
|
f"{curl_path} -u {OLLAMA_USER}:{OLLAMA_PASSWORD} -s -X POST "
|
|
"-H 'Content-Type: application/json' "
|
|
f'--data \"{escaped_payload}\" {OLLAMA_API_URL}'
|
|
)
|
|
|
|
user_data = f"{buffer}||{prefix}"
|
|
weechat.hook_process_hashtable(
|
|
cmd,
|
|
{"timeout": "10000"},
|
|
10000,
|
|
"ollama_response_callback",
|
|
user_data
|
|
)
|
|
|
|
def ollama_response_callback(data, command, return_code, out, err):
|
|
buffer, prefix = data.split("||", 1)
|
|
|
|
# Manejamos errores genéricos (sin mostrar debug):
|
|
if return_code is None or return_code == weechat.WEECHAT_HOOK_PROCESS_ERROR:
|
|
response = "[Ollama] Error executing request."
|
|
elif out.strip() == "":
|
|
response = "[Ollama] Empty response from server."
|
|
else:
|
|
try:
|
|
parsed = json.loads(out)
|
|
response = parsed.get("response", "[Ollama] No 'response' field in reply.")
|
|
except Exception:
|
|
response = "[Ollama] Error parsing server response."
|
|
|
|
if prefix:
|
|
weechat.command(buffer, f"/msg {prefix} {response}")
|
|
else:
|
|
weechat.command(buffer, f"/say {response}")
|
|
|
|
return weechat.WEECHAT_RC_OK
|
|
|
|
def command_ollama(data, buffer, args):
|
|
if not args:
|
|
weechat.prnt(buffer, "[Ollama] Usage: /ollama <question>")
|
|
return weechat.WEECHAT_RC_OK
|
|
|
|
ask_ollama_async(args, buffer)
|
|
return weechat.WEECHAT_RC_OK
|
|
|
|
def message_callback(data, buffer, date, tags, displayed, highlight, prefix, message):
|
|
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", "")
|
|
is_mentioned = f"@{username.lower()}" in message.lower()
|
|
|
|
if is_private and weechat.config_get_plugin("pm_response") == "off":
|
|
return weechat.WEECHAT_RC_OK
|
|
|
|
if is_private and not message.strip().endswith("?"):
|
|
return weechat.WEECHAT_RC_OK
|
|
|
|
if not is_private and not is_mentioned and not int(highlight):
|
|
return weechat.WEECHAT_RC_OK
|
|
|
|
ask_ollama_async(message, buffer, prefix if is_private else "")
|
|
return weechat.WEECHAT_RC_OK
|
|
|
|
def config_callback(data, option, value):
|
|
weechat.prnt("", f"[Ollama] Configuration changed: {option} = {value}")
|
|
return weechat.WEECHAT_RC_OK
|
|
|
|
# Descripciones de las opciones
|
|
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)")
|
|
|
|
# Ganchos de configuración
|
|
weechat.hook_config("plugins.var.python.ollama_online.highlight_response", "config_callback", "")
|
|
weechat.hook_config("plugins.var.python.ollama_online.pm_response", "config_callback", "")
|
|
|
|
# Comando /ollama
|
|
weechat.hook_command(
|
|
"ollama",
|
|
"Ask something to Ollama",
|
|
"<question>",
|
|
"Example: /ollama 'Archivo: \"07.Diamantes para La Eternidad...\" ¿Título y artista?'",
|
|
"",
|
|
"command_ollama",
|
|
""
|
|
)
|
|
|
|
# Hooks de mensajes
|
|
weechat.hook_print("", "notify_highlight", "", 1, "message_callback", "")
|
|
weechat.hook_print("", "notify_message", "", 1, "message_callback", "")
|
|
weechat.hook_print("", "notify_private", "", 1, "message_callback", "")
|
|
|