Taiga integration

2026-06-09 12:47:13 +03:00
parent c8599b3d13
commit 1f83dcb574
30 changed files with 1543 additions and 115 deletions
@@ -4,6 +4,7 @@ from typing import Any
 from sqlalchemy.orm import Session

 from app.pomodoro.service import PomodoroService
+from app.projects.service import ProjectService

 TOOL_DEFINITIONS: list[dict[str, Any]] = [
    {
@@ -104,7 +105,7 @@ TOOL_DEFINITIONS: list[dict[str, Any]] = [
        "type": "function",
        "function": {
            "name": "get_pomodoro_history",
-            "description": "ОБЯЗАТЕЛЬНО при вопросах о задачах, истории работы или что пользователь делал.",
+            "description": "История помидоро-сессий (таймер), не Taiga-задачи.",
            "parameters": {
                "type": "object",
                "properties": {
@@ -114,44 +115,107 @@ 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": "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": "Список созданных work items (Taiga + Gitea связки).",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "status": {"type": "string", "description": "open или closed"},
+                    "limit": {"type": "integer"},
+                },
+                "required": [],
+            },
+        },
+    },
 ]


-def execute_tool(db: Session, name: str, arguments: dict[str, Any]) -> str:
-    service = PomodoroService(db)
+async def execute_tool(db: Session, name: str, arguments: dict[str, Any]) -> str:
+    pomodoro = PomodoroService(db)
+    projects = ProjectService(db)

    try:
        if name == "get_pomodoro_status":
-            result = service.get_status()
+            result = pomodoro.get_status()
        elif name == "start_pomodoro":
-            result = service.start_work(
+            result = pomodoro.start_work(
                duration_min=arguments.get("duration_min"),
                task_note=arguments.get("task_note", ""),
            )
        elif name == "start_short_break":
-            result = service.start_short_break(
-                duration_min=arguments.get("duration_min"),
-            )
+            result = pomodoro.start_short_break(duration_min=arguments.get("duration_min"))
        elif name == "start_long_break":
-            result = service.start_long_break(
-                duration_min=arguments.get("duration_min"),
-            )
+            result = pomodoro.start_long_break(duration_min=arguments.get("duration_min"))
        elif name == "stop_pomodoro":
-            result = service.stop(
+            result = pomodoro.stop(
                result=arguments.get("result", ""),
                completed=arguments.get("completed", False),
            )
        elif name == "skip_pomodoro_phase":
-            result = service.skip_phase()
+            result = pomodoro.skip_phase()
        elif name == "reset_pomodoro_cycle":
-            result = service.reset_cycle(
-                clear_task=arguments.get("clear_task", False),
-            )
+            result = pomodoro.reset_cycle(clear_task=arguments.get("clear_task", False))
        elif name == "get_pomodoro_history":
-            result = service.history(limit=arguments.get("limit", 10))
+            result = pomodoro.history(limit=arguments.get("limit", 10))
+        elif name == "sync_taiga_projects":
+            result = projects.sync_taiga_projects()
+        elif name == "list_taiga_projects":
+            result = projects.list_projects()
+        elif name == "create_work_item":
+            result = await projects.create_work_item(
+                arguments.get("text", ""),
+                project_slug=arguments.get("project_slug"),
+            )
+        elif name == "list_work_items":
+            result = projects.list_work_items(
+                limit=arguments.get("limit", 20),
+                status=arguments.get("status"),
+            )
        else:
            return json.dumps({"error": f"Unknown tool: {name}"}, ensure_ascii=False)

        return json.dumps(result, ensure_ascii=False)
    except ValueError as exc:
        return json.dumps({"error": str(exc)}, ensure_ascii=False)
+    except Exception as exc:
+        return json.dumps({"error": str(exc)}, ensure_ascii=False)