lollms-client 1.5.6__py3-none-any.whl → 1.7.13__py3-none-any.whl

lollms_client/llm_bindings/vllm/__init__.py

@@ -451,7 +451,7 @@ class VLLMBinding(LollmsLLMBinding):
             }
         return info
 
-    def listModels(self) -> List[Dict[str, Any]]:
+    def list_models(self) -> List[Dict[str, Any]]:
         local_models = []
         if not self.models_folder.exists(): return []
         for item_path in self.models_folder.rglob('*'):

lollms_client/lollms_agentic.py

@@ -65,19 +65,21 @@ class TaskPlanner:
     def __init__(self, llm_client):
         self.llm_client = llm_client
         
-    def decompose_task(self, user_request: str, context: str = "") -> ExecutionPlan:
+    def decompose_task(self, user_request: str, context: str = "", all_visible_tools:str="") -> ExecutionPlan:
         """Break down complex requests into manageable subtasks"""
         decomposition_prompt = f"""
 Analyze this user request and break it down into specific, actionable subtasks:
 
 USER REQUEST: "{user_request}"
 CONTEXT: {context}
+AVAILABLE TOOLS: {all_visible_tools}
 
 Create a JSON plan with subtasks that are:
 1. Specific and actionable
 2. Have clear success criteria  
 3. Include estimated complexity (1-5 scale)
 4. List required tool types
+5. The tasks should either be thought based, or use one of the available tools. Do not plan tasks we can not do.
 
 Output format:
 {{
@@ -358,4 +360,4 @@ class UncertaintyManager:
         else:
             level = ConfidenceLevel.LOW
             
-        return confidence, level
+        return confidence, level

lollms_client/lollms_base_binding.py

@@ -0,0 +1,61 @@
+from abc import ABC, abstractmethod
+from typing import Optional, Dict, List, Any
+from pathlib import Path
+import yaml
+import inspect
+from ascii_colors import ASCIIColors
+
+class LollmsBaseBinding(ABC):
+    """
+    Base class for all LOLLMS bindings (LLM, TTI, TTS, STT, TTM, TTV, MCP).
+    Enforces a unified initialization and common methods.
+    """
+    def __init__(self, binding_name: str, **kwargs):
+        """
+        Initialize the binding.
+        
+        Args:
+            binding_name (str): The name of the binding.
+            **kwargs: Configuration parameters passed from the manager/app.
+        """
+        self.binding_name = binding_name
+        self.config = kwargs
+        self.binding_dir = self._get_binding_dir()
+        self.description = self._load_description()
+        
+    def _get_binding_dir(self) -> Path:
+        """
+        Locates the directory of the concrete binding class.
+        """
+        try:
+            return Path(inspect.getfile(self.__class__)).parent
+        except Exception:
+            return Path(".")
+
+    def _load_description(self) -> Dict:
+        """
+        Loads the description.yaml file from the binding directory.
+        """
+        desc_file = self.binding_dir / "description.yaml"
+        if desc_file.exists():
+            try:
+                with open(desc_file, 'r', encoding='utf-8') as f:
+                    return yaml.safe_load(f)
+            except Exception as e:
+                ASCIIColors.error(f"Failed to load description.yaml for {self.binding_name}: {e}")
+        return {}
+
+    @abstractmethod
+    def list_models(self) -> List[Any]:
+        """
+        List available models or resources provided by this binding.
+        Must be implemented by all bindings.
+        """
+        pass
+
+    def ps(self) -> List[Any]:
+        """
+        Verify resources or processes associated with this binding.
+        Returns a list of status information.
+        """
+        return []