52c42f3d06
- create_event(title, date, start_time?, end_time?, all_day?, owner)
owner обязателен (daniil | sveta). System prompt велит LLM уточнять
чей это календарь, если неясно.
- update_event(event_id, owner, ...fields) — меняет только переданные
поля. Сначала нужно вызвать get_today_events для получения event_id.
- delete_event(event_id, owner) — сначала get_today_events, найти
событие по названию, подтвердить если важное.
get_today_events теперь возвращает event_id и owner (daniil/sveta),
плюс принимает range=month. Description явно говорит LLM что это
первый tool для CRUD-сценариев.
System prompt (Cosmo и Люся) дополнен секцией 'Работа с календарём'
с правилами: даты YYYY-MM-DD, время HH:MM, «завтра» = +1 день,
вычислять от {today}.
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
393 lines
16 KiB
Python
393 lines
16 KiB
Python
"""
|
||
Tool definitions для Claude Haiku 4.5.
|
||
|
||
Каждый tool описан схемой (Anthropic format) + обёрткой-executor.
|
||
Большинство tools — тонкие прокси к /api/voice/tools/* на планшете.
|
||
Таймер — тоже прокси, но POST (создаёт таймер, тот появляется на дашборде).
|
||
|
||
Аутентификация: Bearer VOICE_API_KEY (тот же что для /api/voice/event).
|
||
"""
|
||
import os
|
||
from typing import Any
|
||
|
||
import requests
|
||
|
||
from .config import log
|
||
|
||
TABLET_URL = os.getenv("TABLET_URL", "").rstrip("/")
|
||
VOICE_API_KEY = os.getenv("VOICE_API_KEY", "")
|
||
|
||
_session = requests.Session()
|
||
|
||
|
||
def _headers() -> dict:
|
||
return {"Authorization": f"Bearer {VOICE_API_KEY}"}
|
||
|
||
|
||
def _tablet_get(path: str, params: dict | None = None) -> dict:
|
||
url = f"{TABLET_URL}{path}"
|
||
r = _session.get(url, headers=_headers(), params=params, timeout=8)
|
||
r.raise_for_status()
|
||
return r.json()
|
||
|
||
|
||
def _tablet_post(path: str, payload: dict) -> dict:
|
||
url = f"{TABLET_URL}{path}"
|
||
r = _session.post(url, headers=_headers(), json=payload, timeout=8)
|
||
r.raise_for_status()
|
||
return r.json()
|
||
|
||
|
||
# ─────────────────────────────────────────────────────────────
|
||
# Tool schemas (Anthropic format). Порядок = приоритет в Claude подсказках.
|
||
# ─────────────────────────────────────────────────────────────
|
||
|
||
TOOL_SCHEMAS: list[dict] = [
|
||
{
|
||
"name": "get_weather",
|
||
"description": (
|
||
"Получить текущую погоду и короткий прогноз для города. "
|
||
"Для вопросов вроде 'какая сегодня погода', 'холодно ли на улице', "
|
||
"'нужен ли зонт'. По умолчанию — Санкт-Петербург."
|
||
),
|
||
"input_schema": {
|
||
"type": "object",
|
||
"properties": {
|
||
"city": {
|
||
"type": "string",
|
||
"description": "Город на русском или шорткод (spb, msk, sochi, ekb, kzn, nsk, krd). По умолчанию Санкт-Петербург.",
|
||
},
|
||
},
|
||
},
|
||
},
|
||
{
|
||
"name": "get_transport",
|
||
"description": (
|
||
"Расписание ближайших трамваев на остановке Ул. Антонова-Овсеенко. "
|
||
"Для вопросов 'когда следующий 23-й', 'что ближайшее в центр', 'пора идти на остановку'."
|
||
),
|
||
"input_schema": {
|
||
"type": "object",
|
||
"properties": {
|
||
"direction": {
|
||
"type": "string",
|
||
"enum": ["to_center", "from_center", "all"],
|
||
"description": "to_center = в центр (к Новочеркасской), from_center = от центра (к Большевиков), all = оба направления",
|
||
},
|
||
"routes": {
|
||
"type": "string",
|
||
"description": "Фильтр маршрутов через запятую, например '23' или '23,27'. Пусто = все маршруты.",
|
||
},
|
||
},
|
||
},
|
||
},
|
||
{
|
||
"name": "get_today_events",
|
||
"description": (
|
||
"События из календаря (Даниил + Света). Вернёт id события, "
|
||
"title, start, end, owner ('daniil' или 'sveta'). "
|
||
"ВАЖНО: для update_event / delete_event сначала вызывай этот tool "
|
||
"чтобы получить event_id."
|
||
),
|
||
"input_schema": {
|
||
"type": "object",
|
||
"properties": {
|
||
"range": {
|
||
"type": "string",
|
||
"enum": ["today", "week", "month"],
|
||
"description": "today (по умолчанию), week (7 дней) или month (текущий месяц)",
|
||
},
|
||
},
|
||
},
|
||
},
|
||
{
|
||
"name": "create_event",
|
||
"description": (
|
||
"Создать событие в Google Calendar. "
|
||
"ВАЖНО: параметр owner обязателен. Если пользователь не сказал "
|
||
"чей это календарь — СПРОСИ у него ('в твой календарь или в Светин?') "
|
||
"и только потом вызывай tool. Не угадывай."
|
||
),
|
||
"input_schema": {
|
||
"type": "object",
|
||
"properties": {
|
||
"title": {"type": "string", "description": "Название события"},
|
||
"date": {"type": "string", "description": "Дата в формате YYYY-MM-DD"},
|
||
"start_time": {
|
||
"type": "string",
|
||
"description": "Время начала в формате HH:MM (24-часовой). Обязательно если all_day=false.",
|
||
},
|
||
"end_time": {
|
||
"type": "string",
|
||
"description": "Время окончания в формате HH:MM. По умолчанию start_time + 1 час.",
|
||
},
|
||
"all_day": {
|
||
"type": "boolean",
|
||
"description": "Событие на весь день без времени. По умолчанию false.",
|
||
},
|
||
"owner": {
|
||
"type": "string",
|
||
"enum": ["daniil", "sveta"],
|
||
"description": "Чей это календарь — Даниила или Светы",
|
||
},
|
||
},
|
||
"required": ["title", "date", "owner"],
|
||
},
|
||
},
|
||
{
|
||
"name": "update_event",
|
||
"description": (
|
||
"Изменить существующее событие. Сначала обязательно вызови get_today_events "
|
||
"чтобы получить event_id и owner нужного события. Передавай только те поля "
|
||
"которые меняешь."
|
||
),
|
||
"input_schema": {
|
||
"type": "object",
|
||
"properties": {
|
||
"event_id": {"type": "string"},
|
||
"owner": {
|
||
"type": "string",
|
||
"enum": ["daniil", "sveta"],
|
||
"description": "Чей календарь (из get_today_events)",
|
||
},
|
||
"title": {"type": "string"},
|
||
"date": {"type": "string", "description": "YYYY-MM-DD"},
|
||
"start_time": {"type": "string", "description": "HH:MM"},
|
||
"end_time": {"type": "string", "description": "HH:MM"},
|
||
"all_day": {"type": "boolean"},
|
||
},
|
||
"required": ["event_id", "owner"],
|
||
},
|
||
},
|
||
{
|
||
"name": "delete_event",
|
||
"description": (
|
||
"Удалить событие из календаря. Сначала вызови get_today_events чтобы "
|
||
"найти event_id и определить owner. Подтверди удаление с пользователем "
|
||
"если событие важное (встреча, врач, работа)."
|
||
),
|
||
"input_schema": {
|
||
"type": "object",
|
||
"properties": {
|
||
"event_id": {"type": "string"},
|
||
"owner": {
|
||
"type": "string",
|
||
"enum": ["daniil", "sveta"],
|
||
},
|
||
},
|
||
"required": ["event_id", "owner"],
|
||
},
|
||
},
|
||
{
|
||
"name": "get_notes",
|
||
"description": (
|
||
"Список заметок и списков покупок с планшета. "
|
||
"Для 'что мне купить', 'что в списке', 'какие записи'."
|
||
),
|
||
"input_schema": {"type": "object", "properties": {}},
|
||
},
|
||
{
|
||
"name": "set_timer",
|
||
"description": (
|
||
"Запустить таймер на планшете. Показывает обратный отсчёт с названием "
|
||
"и звенит по окончании. "
|
||
"Используй для 'поставь таймер на 10 минут', 'напомни через час', "
|
||
"'засеки 5 минут для чайника'."
|
||
),
|
||
"input_schema": {
|
||
"type": "object",
|
||
"properties": {
|
||
"seconds": {
|
||
"type": "integer",
|
||
"description": "Длительность в секундах (1..86400)",
|
||
"minimum": 1,
|
||
"maximum": 86400,
|
||
},
|
||
"label": {
|
||
"type": "string",
|
||
"description": "Короткое название таймера (например 'Чайник', 'Паста')",
|
||
},
|
||
},
|
||
"required": ["seconds", "label"],
|
||
},
|
||
},
|
||
{
|
||
"name": "cancel_timer",
|
||
"description": (
|
||
"Отменить активный таймер по его названию. "
|
||
"Для 'отмени таймер чайник', 'убери таймер пасты', 'останови отсчёт'."
|
||
),
|
||
"input_schema": {
|
||
"type": "object",
|
||
"properties": {
|
||
"label": {
|
||
"type": "string",
|
||
"description": "Название таймера (примерное совпадение — можно частично).",
|
||
},
|
||
},
|
||
"required": ["label"],
|
||
},
|
||
},
|
||
{
|
||
"name": "adjust_timer",
|
||
"description": (
|
||
"Изменить оставшееся время таймера. "
|
||
"Для 'добавь ещё 5 минут', 'убавь на минуту', 'накинь времени чайнику'. "
|
||
"Положительный delta_seconds = добавить, отрицательный = уменьшить."
|
||
),
|
||
"input_schema": {
|
||
"type": "object",
|
||
"properties": {
|
||
"label": {
|
||
"type": "string",
|
||
"description": "Название таймера для которого меняем время.",
|
||
},
|
||
"delta_seconds": {
|
||
"type": "integer",
|
||
"description": "Секунды (+ добавить, - уменьшить). Например 300 = +5 минут, -60 = -1 минута.",
|
||
},
|
||
},
|
||
"required": ["label", "delta_seconds"],
|
||
},
|
||
},
|
||
]
|
||
|
||
|
||
# ─────────────────────────────────────────────────────────────
|
||
# Executors — возвращают JSON-совместимый dict или строку-ошибку.
|
||
# ─────────────────────────────────────────────────────────────
|
||
|
||
def _exec_get_weather(params: dict, agent_id: str) -> Any:
|
||
city = params.get("city", "")
|
||
return _tablet_get("/api/voice/tools/weather", params={"city": city} if city else None)
|
||
|
||
|
||
def _exec_get_transport(params: dict, agent_id: str) -> Any:
|
||
q = {}
|
||
if "direction" in params:
|
||
q["direction"] = params["direction"]
|
||
if "routes" in params:
|
||
q["routes"] = params["routes"]
|
||
return _tablet_get("/api/voice/tools/transport", params=q)
|
||
|
||
|
||
def _exec_get_today_events(params: dict, agent_id: str) -> Any:
|
||
range_ = params.get("range", "today")
|
||
return _tablet_get("/api/voice/tools/events", params={"range": range_})
|
||
|
||
|
||
def _exec_create_event(params: dict, agent_id: str) -> Any:
|
||
payload = {
|
||
"title": params.get("title", "").strip(),
|
||
"date": params.get("date", "").strip(),
|
||
"owner": params.get("owner", "daniil"),
|
||
"all_day": bool(params.get("all_day", False)),
|
||
}
|
||
if not payload["title"] or not payload["date"]:
|
||
return {"error": "title and date required"}
|
||
if not payload["all_day"]:
|
||
payload["start_time"] = params.get("start_time", "")
|
||
if "end_time" in params:
|
||
payload["end_time"] = params.get("end_time", "")
|
||
return _tablet_post("/api/voice/tools/events", payload)
|
||
|
||
|
||
def _exec_update_event(params: dict, agent_id: str) -> Any:
|
||
event_id = params.get("event_id", "").strip()
|
||
owner = params.get("owner", "").strip()
|
||
if not event_id or not owner:
|
||
return {"error": "event_id and owner required"}
|
||
payload = {"event_id": event_id, "owner": owner}
|
||
for k in ("title", "date", "start_time", "end_time", "all_day"):
|
||
if k in params:
|
||
payload[k] = params[k]
|
||
url = f"{TABLET_URL}/api/voice/tools/events"
|
||
r = _session.put(url, headers={**_headers(), "Content-Type": "application/json"}, json=payload, timeout=8)
|
||
r.raise_for_status()
|
||
return r.json()
|
||
|
||
|
||
def _exec_delete_event(params: dict, agent_id: str) -> Any:
|
||
event_id = params.get("event_id", "").strip()
|
||
owner = params.get("owner", "daniil").strip()
|
||
if not event_id:
|
||
return {"error": "event_id required"}
|
||
url = f"{TABLET_URL}/api/voice/tools/events"
|
||
r = _session.delete(
|
||
url,
|
||
headers=_headers(),
|
||
params={"event_id": event_id, "owner": owner},
|
||
timeout=8,
|
||
)
|
||
r.raise_for_status()
|
||
return r.json()
|
||
|
||
|
||
def _exec_get_notes(params: dict, agent_id: str) -> Any:
|
||
return _tablet_get("/api/voice/tools/notes")
|
||
|
||
|
||
def _exec_set_timer(params: dict, agent_id: str) -> Any:
|
||
seconds = int(params.get("seconds", 0))
|
||
label = params.get("label", "Таймер")
|
||
if seconds < 1:
|
||
return {"error": "seconds must be positive"}
|
||
return _tablet_post(
|
||
"/api/voice/timer",
|
||
{"action": "start", "seconds": seconds, "label": label, "agent": agent_id},
|
||
)
|
||
|
||
|
||
def _exec_cancel_timer(params: dict, agent_id: str) -> Any:
|
||
label = params.get("label", "").strip()
|
||
if not label:
|
||
return {"error": "label required"}
|
||
return _tablet_post("/api/voice/timer", {"action": "cancel", "label": label})
|
||
|
||
|
||
def _exec_adjust_timer(params: dict, agent_id: str) -> Any:
|
||
label = params.get("label", "").strip()
|
||
delta = int(params.get("delta_seconds", 0))
|
||
if not label:
|
||
return {"error": "label required"}
|
||
if delta == 0:
|
||
return {"error": "delta_seconds must be non-zero"}
|
||
return _tablet_post(
|
||
"/api/voice/timer",
|
||
{"action": "adjust", "label": label, "delta_seconds": delta},
|
||
)
|
||
|
||
|
||
EXECUTORS = {
|
||
"get_weather": _exec_get_weather,
|
||
"get_transport": _exec_get_transport,
|
||
"get_today_events": _exec_get_today_events,
|
||
"create_event": _exec_create_event,
|
||
"update_event": _exec_update_event,
|
||
"delete_event": _exec_delete_event,
|
||
"get_notes": _exec_get_notes,
|
||
"set_timer": _exec_set_timer,
|
||
"cancel_timer": _exec_cancel_timer,
|
||
"adjust_timer": _exec_adjust_timer,
|
||
}
|
||
|
||
|
||
def execute_tool(name: str, params: dict, agent_id: str = "cosmo") -> Any:
|
||
"""Выполнить tool по имени. Возвращает результат (dict/list/str).
|
||
При ошибке возвращает {'error': '...'} — это отправляется в Claude как результат."""
|
||
fn = EXECUTORS.get(name)
|
||
if fn is None:
|
||
return {"error": f"unknown tool: {name}"}
|
||
try:
|
||
result = fn(params, agent_id)
|
||
return result
|
||
except requests.HTTPError as e:
|
||
log.warning(f"Tool {name} HTTP {e.response.status_code}: {e.response.text[:200]}")
|
||
return {"error": f"tool_http_{e.response.status_code}"}
|
||
except requests.RequestException as e:
|
||
log.warning(f"Tool {name} network error: {e}")
|
||
return {"error": "tool_network_error"}
|
||
except Exception as e:
|
||
log.exception(f"Tool {name} failed")
|
||
return {"error": f"tool_exception: {e}"}
|