1b2b935832
- Platform-based architecture (Windows/Linux/Mac) - Agent instance registry (agents.yaml) - Management dashboard with cross-platform monitoring - xmpp_bot with HTTP bridge + health endpoints - wechat_agent with WeChat-Hermes bridging - Platform services: ProcessGuardian, HealthProbe, APIRouter, ChannelBridge - Deployment: systemd (Linux) + PowerShell (Windows) - Monitoring: SSH+ejabberdctl for cross-platform presence
99 lines
4.2 KiB
Python
99 lines
4.2 KiB
Python
#!/usr/bin/env python3
|
|
"""XMPP Bot zhiwei@yoin.fun - Hermes API 版(稳定重连版)"""
|
|
import asyncio, logging, ssl, json, urllib.request, os, subprocess, time
|
|
from xml.sax.saxutils import escape
|
|
|
|
logging.basicConfig(level=logging.INFO, format='%(asctime)s %(message)s')
|
|
GATEWAY = "http://localhost:8643/v1/chat/completions"
|
|
API_KEY = "hermes123"
|
|
_opener = urllib.request.build_opener(urllib.request.ProxyHandler({}))
|
|
|
|
def send(from_jid, to_jid, body):
|
|
safe = escape(body)
|
|
subprocess.run(["docker","exec","ejabberd","ejabberdctl","send_stanza",
|
|
from_jid, to_jid,
|
|
f"<message from='{from_jid}' to='{to_jid}' type='chat' xml:lang='en'><body>{safe}</body></message>"
|
|
], capture_output=True, timeout=10)
|
|
|
|
class ZhiweiBot:
|
|
def __init__(self):
|
|
import slixmpp
|
|
self.xmpp = slixmpp.ClientXMPP('zhiwei@yoin.fun', 'hermes123')
|
|
self.xmpp.add_event_handler('session_bind', self.on_bind)
|
|
self.xmpp.add_event_handler('message', self.on_msg)
|
|
self.xmpp.add_event_handler('disconnected', self.on_disconnect)
|
|
self.xmpp.add_event_handler('connected', self.on_connected)
|
|
# 启用slixmpp内置自动重连(已禁用—与手动重连冲突)
|
|
# self.xmpp.auto_reconnect = True
|
|
ctx = ssl.create_default_context(); ctx.check_hostname = False; ctx.verify_mode = ssl.CERT_NONE
|
|
self.xmpp.ssl_context = ctx
|
|
self.ready = asyncio.Event()
|
|
self._call_seq = 0
|
|
|
|
async def on_connected(self, event):
|
|
logging.info("🔗 知微TCP连接已建立")
|
|
|
|
async def on_bind(self, event):
|
|
self.xmpp.send_presence(); self.xmpp.get_roster(); self.ready.set()
|
|
logging.info("✅ 知微上线")
|
|
|
|
async def on_disconnect(self, event):
|
|
self.ready.clear()
|
|
logging.warning("⚠️ 知微断线")
|
|
# 不要在这里调用 self.xmpp.disconnect(),让 auto_reconnect 处理
|
|
|
|
async def on_msg(self, msg):
|
|
body = msg['body']; sender = str(msg['from'])
|
|
if not body or msg['type'] != 'chat': return
|
|
if 'hmo@yoin.fun' in sender:
|
|
self._call_seq += 1
|
|
logging.info(f"📩 老爸(#{self._call_seq}): {body}")
|
|
try:
|
|
payload = json.dumps({
|
|
"model":"hermes-agent",
|
|
"messages":[{"role":"user","content":f"[zhiwei] {body}"}]
|
|
}).encode()
|
|
req = urllib.request.Request(GATEWAY, data=payload, method="POST")
|
|
req.add_header("Content-Type","application/json")
|
|
req.add_header("Authorization",f"Bearer {API_KEY}")
|
|
req.add_header("X-Hermes-Session-Id","xmpp-zhiwei")
|
|
loop = asyncio.get_event_loop()
|
|
result = await loop.run_in_executor(None, lambda: _opener.open(req, timeout=600))
|
|
data = json.loads(result.read())
|
|
reply = data.get("choices",[{}])[0].get("message",{}).get("content","")
|
|
finish = data.get("choices",[{}])[0].get("finish_reason","")
|
|
if reply.strip() and finish != "silent":
|
|
send("zhiwei@yoin.fun", sender, reply)
|
|
logging.info(f"✅ 知微回复: {reply[:80]}")
|
|
except Exception as e:
|
|
logging.error(f"❌ 知微错误: {e}")
|
|
|
|
async def main():
|
|
retry_delay = 1
|
|
max_delay = 60
|
|
while True:
|
|
try:
|
|
z = ZhiweiBot()
|
|
z.xmpp.register_plugin('xep_0030'); z.xmpp.register_plugin('xep_0199')
|
|
z.xmpp.connect(host='127.0.0.1', port=5222)
|
|
await asyncio.wait_for(z.ready.wait(), timeout=30); logging.info("知微就绪")
|
|
retry_delay = 1
|
|
|
|
# 保持运行,断线时自动重连
|
|
while True:
|
|
await asyncio.sleep(15)
|
|
if not z.xmpp.is_connected():
|
|
logging.warning("知微连接丢失,准备重连...")
|
|
break
|
|
|
|
except Exception as e:
|
|
logging.error(f"知微main错误: {e}")
|
|
|
|
# 指数退避重连
|
|
logging.info(f"⏳ 知微等待 {retry_delay} 秒后重连...")
|
|
await asyncio.sleep(retry_delay)
|
|
retry_delay = min(retry_delay * 2, max_delay)
|
|
|
|
if __name__ == '__main__':
|
|
asyncio.run(main())
|