lollms-client 1.3.4__py3-none-any.whl → 1.3.6__py3-none-any.whl

lollms_client/lollms_llm_binding.py

@@ -34,13 +34,13 @@ def load_known_contexts():
         return known_contexts
     except FileNotFoundError:
         print(f"Error: File not found at {file_path}")
-        return None
+        return []
     except json.JSONDecodeError:
         print(f"Error: Could not decode JSON from {file_path}")
-        return None
+        return []
     except Exception as e:
         print(f"An unexpected error occurred: {e}")
-        return None
+        return []
 
 class LollmsLLMBinding(ABC):
     """Abstract base class for all LOLLMS LLM bindings"""

lollms_client/lollms_tts_binding.py

@@ -1,77 +1,50 @@
-# lollms_client/lollms_tts_binding.py
 from abc import ABC, abstractmethod
 import importlib
 from pathlib import Path
-from typing import Optional, List, Dict, Any, Union, Callable
+from typing import Optional, List, Dict, Any, Union
+import yaml
 from ascii_colors import trace_exception
 
 class LollmsTTSBinding(ABC):
-    """Abstract base class for all LOLLMS Text-to-Speech bindings."""
-
     def __init__(self,
-                 binding_name:str="unknown"):
-        """
-        Initialize the LollmsTTSBinding base class.
-
-        Args:
-            binding_name (Optional[str]): The binding name
-        """
+                 binding_name: str = "unknown",
+                 **kwargs):
         self.binding_name = binding_name
+        self.settings = kwargs
 
     @abstractmethod
-    def generate_audio(self, text: str, voice: Optional[str] = None, **kwargs) -> bytes:
-        """
-        Generates audio data from the provided text.
-
-        Args:
-            text (str): The text content to synthesize.
-            voice (Optional[str]): The specific voice or model to use (if supported by the binding).
-                                   If None, a default voice might be used.
-            **kwargs: Additional binding-specific parameters.
-
-        Returns:
-            bytes: The generated audio data (e.g., in WAV or MP3 format).
-
-        Raises:
-            Exception: If audio generation fails.
-        """
+    def generate_audio(self,
+                       text: str,
+                       voice: Optional[str] = None,
+                       **kwargs) -> bytes:
         pass
 
     @abstractmethod
     def list_voices(self, **kwargs) -> List[str]:
-        """
-        Lists the available voices or TTS models supported by the binding.
-
-        Args:
-            **kwargs: Additional binding-specific parameters.
-
-        Returns:
-            List[str]: A list of available voice/model identifiers.
-        """
         pass
 
-class LollmsTTSBindingManager:
-    """Manages TTS binding discovery and instantiation."""
+    @abstractmethod
+    def list_models(self, **kwargs) -> List[str]:
+        pass
+        
+    def get_settings(self, **kwargs) -> Dict[str, Any]:
+        return self.settings
 
-    def __init__(self, tts_bindings_dir: Union[str, Path] = Path(__file__).parent.parent / "tts_bindings"):
-        """
-        Initialize the LollmsTTSBindingManager.
+    def set_settings(self, settings: Dict[str, Any], **kwargs) -> bool:
+        self.settings.update(settings)
+        return True
 
-        Args:
-            tts_bindings_dir (Union[str, Path]): Directory containing TTS binding implementations.
-                                                 Defaults to the "tts_bindings" subdirectory.
-        """
+class LollmsTTSBindingManager:
+    def __init__(self, tts_bindings_dir: Union[str, Path] = Path(__file__).parent / "tts_bindings"):
         self.tts_bindings_dir = Path(tts_bindings_dir)
         self.available_bindings = {}
 
     def _load_binding(self, binding_name: str):
-        """Dynamically load a specific TTS binding implementation."""
         binding_dir = self.tts_bindings_dir / binding_name
         if binding_dir.is_dir() and (binding_dir / "__init__.py").exists():
             try:
-                # Adjust module path for dynamic loading
                 module = importlib.import_module(f"lollms_client.tts_bindings.{binding_name}")
-                binding_class = getattr(module, module.BindingName) # Assumes BindingName is defined
+                binding_class = getattr(module, module.BindingName)
                 self.available_bindings[binding_name] = binding_class
             except Exception as e:
                 trace_exception(e)
@@ -79,36 +52,76 @@ class LollmsTTSBindingManager:
 
     def create_binding(self,
                       binding_name: str,
-                      **kwargs) -> Optional[LollmsTTSBinding]:
-        """
-        Create an instance of a specific TTS binding.
-
-        Args:
-            binding_name (str): Name of the TTS binding to create.
-            **kwargs: Additional parameters specific to the binding's __init__.
-
-        Returns:
-            Optional[LollmsTTSBinding]: Binding instance or None if creation failed.
-        """
+                      config: Dict[str, Any] = None) -> Optional[LollmsTTSBinding]:
+        if config is None:
+            config = {}
+            
         if binding_name not in self.available_bindings:
             self._load_binding(binding_name)
 
         binding_class = self.available_bindings.get(binding_name)
         if binding_class:
             try:
-                return binding_class(**kwargs)
+                return binding_class(**config)
             except Exception as e:
                 trace_exception(e)
                 print(f"Failed to instantiate TTS binding {binding_name}: {str(e)}")
                 return None
         return None
 
-    def get_available_bindings(self) -> list[str]:
-        """
-        Return list of available TTS binding names based on subdirectories.
+    @staticmethod
+    def _get_fallback_description(binding_name: str) -> Dict:
+        return {
+            "binding_name": binding_name,
+            "title": binding_name.replace("_", " ").title(),
+            "author": "Unknown",
+            "version": "N/A",
+            "description": f"A binding for {binding_name}. No description.yaml file was found.",
+            "input_parameters": [
+                 {
+                    "name": "model_name",
+                    "type": "str",
+                    "description": "The model name or ID to be used.",
+                    "mandatory": False,
+                    "default": ""
+                }
+            ],
+            "generate_audio_parameters": []
+        }
+
+    @staticmethod
+    def get_bindings_list(tts_bindings_dir: Union[str, Path]) -> List[Dict]:
+        bindings_dir = Path(tts_bindings_dir)
+        if not bindings_dir.is_dir():
+            return []
+
+        bindings_list = []
+        for binding_folder in bindings_dir.iterdir():
+            if binding_folder.is_dir() and (binding_folder / "__init__.py").exists():
+                binding_name = binding_folder.name
+                description_file = binding_folder / "description.yaml"
+                
+                binding_info = {}
+                if description_file.exists():
+                    try:
+                        with open(description_file, 'r', encoding='utf-8') as f:
+                            binding_info = yaml.safe_load(f)
+                        binding_info['binding_name'] = binding_name
+                    except Exception as e:
+                        print(f"Error loading description.yaml for {binding_name}: {e}")
+                        binding_info = LollmsTTSBindingManager._get_fallback_description(binding_name)
+                else:
+                    binding_info = LollmsTTSBindingManager._get_fallback_description(binding_name)
+                
+                bindings_list.append(binding_info)
+
+        return sorted(bindings_list, key=lambda b: b.get('title', b['binding_name']))
 
-        Returns:
-            list[str]: List of binding names.
-        """
+    def get_available_bindings(self) -> list[str]:
         return [binding_dir.name for binding_dir in self.tts_bindings_dir.iterdir()
-                if binding_dir.is_dir() and (binding_dir / "__init__.py").exists()]
+                if binding_dir.is_dir() and (binding_dir / "__init__.py").exists()]
+
+def get_available_bindings(tts_bindings_dir: Union[str, Path] = None) -> List[Dict]:
+    if tts_bindings_dir is None:
+        tts_bindings_dir = Path(__file__).resolve().parent / "tts_bindings"
+    return LollmsTTSBindingManager.get_bindings_list(tts_bindings_dir)