added memmory

backend/app/tools/registry.py

@@ -3,6 +3,7 @@ from typing import Any
 
 from sqlalchemy.orm import Session
 
+from app.memory.service import MemoryService
 from app.pomodoro.service import PomodoroService
 from app.projects.service import ProjectService
 
@@ -173,6 +174,99 @@ TOOL_DEFINITIONS: list[dict[str, Any]] = [
             },
         },
     },
+    {
+        "type": "function",
+        "function": {
+            "name": "remember_fact",
+            "description": (
+                "Сохранить факт в долгосрочную память. "
+                "Когда пользователь просит «запомни», или сообщает устойчивое предпочтение/факт."
+            ),
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "content": {"type": "string", "description": "Что запомнить"},
+                    "category": {
+                        "type": "string",
+                        "description": "preference, person, habit, project, fact",
+                    },
+                    "importance": {"type": "integer", "description": "1-5, по умолчанию 3"},
+                },
+                "required": ["content"],
+            },
+        },
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "recall_memories",
+            "description": (
+                "Поиск в долгосрочной памяти. "
+                "Когда спрашивают «что ты помнишь», «что я говорил про X»."
+            ),
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "query": {"type": "string", "description": "Подстрока для поиска"},
+                    "category": {"type": "string"},
+                    "limit": {"type": "integer"},
+                },
+                "required": [],
+            },
+        },
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "forget_memory",
+            "description": "Удалить (деактивировать) факт по id из recall_memories или снимка памяти.",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "memory_id": {"type": "integer"},
+                },
+                "required": ["memory_id"],
+            },
+        },
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "update_profile",
+            "description": (
+                "Обновить профиль пользователя: name, timezone, language, notes. "
+                "Передавай только изменившиеся поля."
+            ),
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "name": {"type": "string"},
+                    "timezone": {"type": "string"},
+                    "language": {"type": "string"},
+                    "notes": {"type": "string"},
+                },
+                "required": [],
+            },
+        },
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "update_session_summary",
+            "description": (
+                "Сохранить краткую сводку темы текущего чата "
+                "(когда диалог длинный или пользователь просит «сожми контекст»)."
+            ),
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "summary": {"type": "string", "description": "2-5 предложений о теме чата"},
+                    "session_id": {"type": "integer"},
+                },
+                "required": ["summary", "session_id"],
+            },
+        },
+    },
     {
         "type": "function",
         "function": {
@@ -194,9 +288,16 @@ TOOL_DEFINITIONS: list[dict[str, Any]] = [
 ]
 
 
-async def execute_tool(db: Session, name: str, arguments: dict[str, Any]) -> str:
+async def execute_tool(
+    db: Session,
+    name: str,
+    arguments: dict[str, Any],
+    *,
+    session_id: int | None = None,
+) -> str:
     pomodoro = PomodoroService(db)
     projects = ProjectService(db)
+    memory = MemoryService(db)
 
     try:
         if name == "get_pomodoro_status":
@@ -240,6 +341,34 @@ async def execute_tool(db: Session, name: str, arguments: dict[str, Any]) -> str
                 limit=arguments.get("limit", 20),
                 status=arguments.get("status"),
             )
+        elif name == "remember_fact":
+            result = memory.remember_fact(
+                arguments.get("content", ""),
+                category=arguments.get("category", "fact"),
+                importance=arguments.get("importance", 3),
+                session_id=session_id,
+                source="tool",
+            )
+        elif name == "recall_memories":
+            result = memory.recall_memories(
+                query=arguments.get("query"),
+                category=arguments.get("category"),
+                limit=arguments.get("limit", 20),
+            )
+        elif name == "forget_memory":
+            result = memory.forget_memory(int(arguments["memory_id"]))
+        elif name == "update_profile":
+            updates = {
+                k: arguments[k]
+                for k in ("name", "timezone", "language", "notes")
+                if k in arguments and arguments[k] is not None
+            }
+            result = memory.update_profile(updates)
+        elif name == "update_session_summary":
+            result = memory.update_session_summary(
+                int(arguments["session_id"]),
+                arguments.get("summary", ""),
+            )
         else:
             return json.dumps({"error": f"Unknown tool: {name}"}, ensure_ascii=False)