124 lines
4.6 KiB
Python
124 lines
4.6 KiB
Python
from typing import Any
|
||
|
||
from app.projects.service import ProjectService
|
||
from app.tools._dispatch import NOT_HANDLED, ToolContext
|
||
|
||
TOOL_NAMES = frozenset({
|
||
"sync_taiga_projects",
|
||
"list_taiga_projects",
|
||
"list_taiga_tasks",
|
||
"create_work_item",
|
||
"list_work_items",
|
||
})
|
||
|
||
TOOL_DEFINITIONS: list[dict[str, Any]] = [
|
||
{
|
||
"type": "function",
|
||
"function": {
|
||
"name": "sync_taiga_projects",
|
||
"description": "Синхронизировать список проектов из Taiga API. Вызывай если проекты неизвестны.",
|
||
"parameters": {"type": "object", "properties": {}, "required": []},
|
||
},
|
||
},
|
||
{
|
||
"type": "function",
|
||
"function": {
|
||
"name": "list_taiga_projects",
|
||
"description": "Список проектов Taiga с привязкой Gitea.",
|
||
"parameters": {"type": "object", "properties": {}, "required": []},
|
||
},
|
||
},
|
||
{
|
||
"type": "function",
|
||
"function": {
|
||
"name": "list_taiga_tasks",
|
||
"description": (
|
||
"ОБЯЗАТЕЛЬНО при вопросах «какие задачи», «покажи задачи проекта», «что открыто в Taiga». "
|
||
"Живые user stories и tasks из Taiga API. НЕ путать с list_work_items."
|
||
),
|
||
"parameters": {
|
||
"type": "object",
|
||
"properties": {
|
||
"project_slug": {
|
||
"type": "string",
|
||
"description": "Slug проекта, например home_assistant. Пусто = все проекты.",
|
||
},
|
||
"limit": {"type": "integer", "description": "Макс. на проект, по умолчанию 20"},
|
||
},
|
||
"required": [],
|
||
},
|
||
},
|
||
},
|
||
{
|
||
"type": "function",
|
||
"function": {
|
||
"name": "create_work_item",
|
||
"description": (
|
||
"Создать фичу/баг из вольного текста: структурировать через LLM, "
|
||
"создать Taiga story + Gitea issue. Вызывай при «заведи баг», «оформи фичу», «добавь в таигу»."
|
||
),
|
||
"parameters": {
|
||
"type": "object",
|
||
"properties": {
|
||
"text": {"type": "string", "description": "Полное описание от пользователя"},
|
||
"project_slug": {
|
||
"type": "string",
|
||
"description": "Slug проекта Taiga, если известен",
|
||
},
|
||
},
|
||
"required": ["text"],
|
||
},
|
||
},
|
||
},
|
||
{
|
||
"type": "function",
|
||
"function": {
|
||
"name": "list_work_items",
|
||
"description": (
|
||
"Только задачи, созданные ЭТИМ ассистентом через create_work_item (локальная БД). "
|
||
"НЕ использовать для общего вопроса «какие задачи в Taiga» — для того list_taiga_tasks."
|
||
),
|
||
"parameters": {
|
||
"type": "object",
|
||
"properties": {
|
||
"status": {"type": "string", "description": "open или closed"},
|
||
"limit": {"type": "integer"},
|
||
},
|
||
"required": [],
|
||
},
|
||
},
|
||
},
|
||
]
|
||
|
||
|
||
async def execute(name: str, arguments: dict[str, Any], ctx: ToolContext) -> Any:
|
||
if name not in TOOL_NAMES:
|
||
return NOT_HANDLED
|
||
|
||
projects = ProjectService(ctx.db, ctx.user_id)
|
||
|
||
if name == "sync_taiga_projects":
|
||
from app.projects.context import invalidate_projects_snapshot_cache
|
||
|
||
result = projects.sync_taiga_projects()
|
||
invalidate_projects_snapshot_cache(ctx.user_id)
|
||
return result
|
||
if name == "list_taiga_projects":
|
||
return projects.list_projects()
|
||
if name == "list_taiga_tasks":
|
||
return projects.list_taiga_open_tasks(
|
||
project_slug=arguments.get("project_slug"),
|
||
limit=arguments.get("limit", 20),
|
||
)
|
||
if name == "create_work_item":
|
||
return await projects.create_work_item(
|
||
arguments.get("text", ""),
|
||
project_slug=arguments.get("project_slug"),
|
||
)
|
||
if name == "list_work_items":
|
||
return projects.list_work_items(
|
||
limit=arguments.get("limit", 20),
|
||
status=arguments.get("status"),
|
||
)
|
||
return NOT_HANDLED
|