Cleaned and improved

2025-11-28 22:09:26 +01:00
parent 4d2cfb5db1
commit c07527c6c9
21 changed files with 1431 additions and 365 deletions
@@ -0,0 +1,93 @@
+"""Tool registry and definitions."""
+from dataclasses import dataclass
+from typing import Callable, Any, Dict
+from functools import partial
+
+from .memory import Memory
+from .tools.filesystem import set_path_for_folder, list_folder
+from .tools.api import find_media_imdb_id
+
+
+@dataclass
+class Tool:
+    """Represents a tool that can be used by the agent."""
+    name: str
+    description: str
+    func: Callable[..., Dict[str, Any]]
+    parameters: Dict[str, Any]  # JSON Schema des paramètres
+
+
+def make_tools(memory: Memory) -> Dict[str, Tool]:
+    """
+    Create all available tools with memory bound to them.
+    
+    Args:
+        memory: Memory instance to be used by the tools
+    
+    Returns:
+        Dictionary mapping tool names to Tool instances
+    """
+    # Create partial functions with memory pre-bound for filesystem tools
+    set_path_func = partial(set_path_for_folder, memory)
+    list_folder_func = partial(list_folder, memory)
+    
+    tools = [
+        Tool(
+            name="set_path_for_folder",
+            description="Sets a path in the configuration (download_folder, tvshow_folder, movie_folder, or torrent_folder).",
+            func=set_path_func,
+            parameters={
+                "type": "object",
+                "properties": {
+                    "folder_name": {
+                        "type": "string",
+                        "description": "Name of folder to set",
+                        "enum": ["download", "tvshow", "movie", "torrent"]
+                    },
+                    "path_value": {
+                        "type": "string",
+                        "description": "Absolute path to the folder (e.g., /home/user/downloads)"
+                    }
+                },
+                "required": ["folder_name", "path_value"]
+            }
+        ),
+        Tool(
+            name="list_folder",
+            description="Lists the contents of a specified folder (download, tvshow, movie, or torrent).",
+            func=list_folder_func,
+            parameters={
+                "type": "object",
+                "properties": {
+                    "folder_type": {
+                        "type": "string",
+                        "description": "Type of folder to list: 'download', 'tvshow', 'movie', or 'torrent'",
+                        "enum": ["download", "tvshow", "movie", "torrent"]
+                    },
+                    "path": {
+                        "type": "string",
+                        "description": "Relative path within the folder (default: '.' for root)",
+                        "default": "."
+                    }
+                },
+                "required": ["folder_type"]
+            }
+        ),
+        Tool(
+            name="find_media_imdb_id",
+            description="Finds the IMDb ID for a given media title using TMDB API.",
+            func=find_media_imdb_id,
+            parameters={
+                "type": "object",
+                "properties": {
+                    "media_title": {
+                        "type": "string",
+                        "description": "Title of the media to find the IMDb ID for"
+                    },
+                },
+                "required": ["media_title"]
+            }
+        ),
+    ]
+
+    return {t.name: t for t in tools}