50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
# tools/telegram_keyboard_tool.py
|
|
|
|
from .base_tool import BaseTool
|
|
|
|
class TelegramKeyboardTool(BaseTool):
|
|
|
|
def get_functions(self):
|
|
return [
|
|
{
|
|
"name": "create_keyboard",
|
|
"description": "Create a Telegram keyboard layout.",
|
|
"parameters": {
|
|
"type": "object",
|
|
"properties": {
|
|
"buttons": {
|
|
"type": "array",
|
|
"items": {
|
|
"type": "array",
|
|
"items": {
|
|
"type": "string",
|
|
"description": "Text on the button."
|
|
},
|
|
"description": "A row of buttons."
|
|
},
|
|
"description": "List of rows of buttons."
|
|
}
|
|
},
|
|
"required": ["buttons"]
|
|
}
|
|
}
|
|
]
|
|
|
|
def execute(self, function_name, **kwargs):
|
|
if function_name == "create_keyboard":
|
|
return self._create_keyboard(kwargs.get("buttons"))
|
|
else:
|
|
return f"Unknown function: {function_name}"
|
|
|
|
def _create_keyboard(self, buttons):
|
|
if not isinstance(buttons, list):
|
|
return "Invalid input: 'buttons' must be a list of lists."
|
|
|
|
keyboard = {
|
|
"keyboard": buttons,
|
|
"resize_keyboard": True,
|
|
"one_time_keyboard": True
|
|
}
|
|
|
|
return keyboard
|