feat: implement cross-channel messaging functionality
- Enhanced the `handle_send_message_function_call` to support sending messages to multiple platforms (Telegram, Discord, Slack, WhatsApp) using their respective APIs. - Added error handling for missing parameters and platform configuration issues. - Introduced asynchronous message sending with helper functions for each platform, improving responsiveness and reliability. - Updated documentation within the function to clarify usage and requirements.
This commit is contained in:
145
model_tools.py
145
model_tools.py
@@ -1952,19 +1952,148 @@ def handle_tts_function_call(
|
|||||||
|
|
||||||
|
|
||||||
def handle_send_message_function_call(function_name, function_args):
|
def handle_send_message_function_call(function_name, function_args):
|
||||||
"""Handle cross-channel send_message tool calls."""
|
"""Handle cross-channel send_message tool calls.
|
||||||
|
|
||||||
|
Sends a message directly to the target platform using its API.
|
||||||
|
Works in both CLI and gateway contexts -- does not require the
|
||||||
|
gateway to be running. Loads credentials from the gateway config
|
||||||
|
(env vars / ~/.hermes/gateway.json).
|
||||||
|
"""
|
||||||
import json
|
import json
|
||||||
|
import asyncio
|
||||||
|
|
||||||
target = function_args.get("target", "")
|
target = function_args.get("target", "")
|
||||||
message = function_args.get("message", "")
|
message = function_args.get("message", "")
|
||||||
if not target or not message:
|
if not target or not message:
|
||||||
return json.dumps({"error": "Both 'target' and 'message' are required"})
|
return json.dumps({"error": "Both 'target' and 'message' are required"})
|
||||||
|
|
||||||
# Store the pending message for the gateway to deliver
|
# Parse target: "platform" or "platform:chat_id"
|
||||||
# The gateway runner checks this after the agent loop completes
|
parts = target.split(":", 1)
|
||||||
import os
|
platform_name = parts[0].strip().lower()
|
||||||
os.environ["_HERMES_PENDING_SEND_TARGET"] = target
|
chat_id = parts[1].strip() if len(parts) > 1 else None
|
||||||
os.environ["_HERMES_PENDING_SEND_MESSAGE"] = message
|
|
||||||
return json.dumps({"success": True, "delivered_to": target, "note": "Message queued for delivery"})
|
try:
|
||||||
|
from gateway.config import load_gateway_config, Platform
|
||||||
|
config = load_gateway_config()
|
||||||
|
except Exception as e:
|
||||||
|
return json.dumps({"error": f"Failed to load gateway config: {e}"})
|
||||||
|
|
||||||
|
platform_map = {
|
||||||
|
"telegram": Platform.TELEGRAM,
|
||||||
|
"discord": Platform.DISCORD,
|
||||||
|
"slack": Platform.SLACK,
|
||||||
|
"whatsapp": Platform.WHATSAPP,
|
||||||
|
}
|
||||||
|
platform = platform_map.get(platform_name)
|
||||||
|
if not platform:
|
||||||
|
avail = ", ".join(platform_map.keys())
|
||||||
|
return json.dumps({"error": f"Unknown platform: {platform_name}. Available: {avail}"})
|
||||||
|
|
||||||
|
pconfig = config.platforms.get(platform)
|
||||||
|
if not pconfig or not pconfig.enabled:
|
||||||
|
return json.dumps({"error": f"Platform '{platform_name}' is not configured. Set up credentials in ~/.hermes/gateway.json or environment variables."})
|
||||||
|
|
||||||
|
if not chat_id:
|
||||||
|
home = config.get_home_channel(platform)
|
||||||
|
if home:
|
||||||
|
chat_id = home.chat_id
|
||||||
|
else:
|
||||||
|
return json.dumps({"error": f"No chat_id specified and no home channel configured for {platform_name}. Use format 'platform:chat_id'."})
|
||||||
|
|
||||||
|
try:
|
||||||
|
result = _run_async(_send_to_platform(platform, pconfig, chat_id, message))
|
||||||
|
return json.dumps(result)
|
||||||
|
except Exception as e:
|
||||||
|
return json.dumps({"error": f"Send failed: {e}"})
|
||||||
|
|
||||||
|
|
||||||
|
def _run_async(coro):
|
||||||
|
"""Run an async coroutine from a sync context.
|
||||||
|
|
||||||
|
If the current thread already has a running event loop (e.g. inside
|
||||||
|
the gateway's async stack), we spin up a disposable thread so
|
||||||
|
asyncio.run() can create its own loop without conflicting.
|
||||||
|
"""
|
||||||
|
import asyncio
|
||||||
|
try:
|
||||||
|
loop = asyncio.get_running_loop()
|
||||||
|
except RuntimeError:
|
||||||
|
loop = None
|
||||||
|
|
||||||
|
if loop and loop.is_running():
|
||||||
|
import concurrent.futures
|
||||||
|
with concurrent.futures.ThreadPoolExecutor(max_workers=1) as pool:
|
||||||
|
future = pool.submit(asyncio.run, coro)
|
||||||
|
return future.result(timeout=30)
|
||||||
|
return asyncio.run(coro)
|
||||||
|
|
||||||
|
|
||||||
|
async def _send_to_platform(platform, pconfig, chat_id, message):
|
||||||
|
"""Route a message to the appropriate platform sender."""
|
||||||
|
from gateway.config import Platform
|
||||||
|
if platform == Platform.TELEGRAM:
|
||||||
|
return await _send_telegram(pconfig.token, chat_id, message)
|
||||||
|
elif platform == Platform.DISCORD:
|
||||||
|
return await _send_discord(pconfig.token, chat_id, message)
|
||||||
|
elif platform == Platform.SLACK:
|
||||||
|
return await _send_slack(pconfig.token, chat_id, message)
|
||||||
|
return {"error": f"Direct sending not yet implemented for {platform.value}"}
|
||||||
|
|
||||||
|
|
||||||
|
async def _send_telegram(token, chat_id, message):
|
||||||
|
"""Send via Telegram Bot API (one-shot, no polling needed)."""
|
||||||
|
try:
|
||||||
|
from telegram import Bot
|
||||||
|
bot = Bot(token=token)
|
||||||
|
msg = await bot.send_message(chat_id=int(chat_id), text=message)
|
||||||
|
return {"success": True, "platform": "telegram", "chat_id": chat_id, "message_id": str(msg.message_id)}
|
||||||
|
except ImportError:
|
||||||
|
return {"error": "python-telegram-bot not installed. Run: pip install python-telegram-bot"}
|
||||||
|
except Exception as e:
|
||||||
|
return {"error": f"Telegram send failed: {e}"}
|
||||||
|
|
||||||
|
|
||||||
|
async def _send_discord(token, chat_id, message):
|
||||||
|
"""Send via Discord REST API (no websocket client needed)."""
|
||||||
|
try:
|
||||||
|
import aiohttp
|
||||||
|
except ImportError:
|
||||||
|
return {"error": "aiohttp not installed. Run: pip install aiohttp"}
|
||||||
|
try:
|
||||||
|
url = f"https://discord.com/api/v10/channels/{chat_id}/messages"
|
||||||
|
headers = {"Authorization": f"Bot {token}", "Content-Type": "application/json"}
|
||||||
|
chunks = [message[i:i+2000] for i in range(0, len(message), 2000)]
|
||||||
|
message_ids = []
|
||||||
|
async with aiohttp.ClientSession() as session:
|
||||||
|
for chunk in chunks:
|
||||||
|
async with session.post(url, headers=headers, json={"content": chunk}) as resp:
|
||||||
|
if resp.status not in (200, 201):
|
||||||
|
body = await resp.text()
|
||||||
|
return {"error": f"Discord API error ({resp.status}): {body}"}
|
||||||
|
data = await resp.json()
|
||||||
|
message_ids.append(data.get("id"))
|
||||||
|
return {"success": True, "platform": "discord", "chat_id": chat_id, "message_ids": message_ids}
|
||||||
|
except Exception as e:
|
||||||
|
return {"error": f"Discord send failed: {e}"}
|
||||||
|
|
||||||
|
|
||||||
|
async def _send_slack(token, chat_id, message):
|
||||||
|
"""Send via Slack Web API."""
|
||||||
|
try:
|
||||||
|
import aiohttp
|
||||||
|
except ImportError:
|
||||||
|
return {"error": "aiohttp not installed. Run: pip install aiohttp"}
|
||||||
|
try:
|
||||||
|
url = "https://slack.com/api/chat.postMessage"
|
||||||
|
headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
|
||||||
|
async with aiohttp.ClientSession() as session:
|
||||||
|
async with session.post(url, headers=headers, json={"channel": chat_id, "text": message}) as resp:
|
||||||
|
data = await resp.json()
|
||||||
|
if data.get("ok"):
|
||||||
|
return {"success": True, "platform": "slack", "chat_id": chat_id, "message_id": data.get("ts")}
|
||||||
|
return {"error": f"Slack API error: {data.get('error', 'unknown')}"}
|
||||||
|
except Exception as e:
|
||||||
|
return {"error": f"Slack send failed: {e}"}
|
||||||
|
|
||||||
|
|
||||||
def handle_function_call(
|
def handle_function_call(
|
||||||
|
|||||||
Reference in New Issue
Block a user