assistant / tools /send_message_to_arseniy.py
RichCake's picture
Update tools/send_message_to_arseniy.py
fc03c36 verified
raw
history blame contribute delete
863 Bytes
from smolagents import Tool
from typing import Any, Optional
import os
BOT_TOKEN = os.getenv("BOT_TOKEN")
class SimpleTool(Tool):
name = "send_message_to_arseniy"
description = "Sends any message to arseniy."
inputs = {'message': {'type': 'string', 'description': 'message to send. Do not need to be url safe. Returns True if ok, else False'}}
output_type = "boolean"
def forward(self, message: str) -> bool:
"""
Sends any message to arseniy.
Args:
message (str): message to send. Do not need to be url safe.
Returns True if ok, else False
"""
import requests
import urllib.parse
safe_message = urllib.parse.quote(message)
response = requests.get(f"https://api.telegram.org/bot{BOT_TOKEN}/sendMessage?chat_id=1064727140&text={safe_message}")
return response.json()["ok"]