Unfucked gemini's mess

agent/tools/__init__.py

@@ -8,6 +8,7 @@ from .api import (
     get_torrent_by_index,
 )
 from .filesystem import list_folder, set_path_for_folder
+from .language import set_language
 
 __all__ = [
     "set_path_for_folder",
@@ -17,4 +18,5 @@ __all__ = [
     "get_torrent_by_index",
     "add_torrent_to_qbittorrent",
     "add_torrent_by_index",
+    "set_language",
 ]

agent/tools/language.py

@@ -0,0 +1,37 @@
+"""Language management tools for the agent."""
+import logging
+from typing import Dict, Any
+
+from infrastructure.persistence import get_memory
+
+logger = logging.getLogger(__name__)
+
+
+def set_language(language: str) -> Dict[str, Any]:
+    """
+    Set the conversation language.
+    
+    Args:
+        language: Language code (e.g., 'en', 'fr', 'es', 'de')
+        
+    Returns:
+        Status dictionary
+    """
+    try:
+        memory = get_memory()
+        memory.stm.set_language(language)
+        memory.save()
+        
+        logger.info(f"Language set to: {language}")
+        
+        return {
+            "status": "ok",
+            "message": f"Language set to {language}",
+            "language": language
+        }
+    except Exception as e:
+        logger.error(f"Failed to set language: {e}")
+        return {
+            "status": "error",
+            "error": str(e)
+        }