quantalogic 0.35.0__py3-none-any.whl → 0.40.0__py3-none-any.whl

quantalogic/utils/git_ls.py

@@ -29,7 +29,7 @@ def git_ls(
 
     # Expand paths and get absolute path
     path = Path(os.path.expanduser(directory_path)).absolute()
-    
+
     # Verify access to base directory
     if not os.access(path, os.R_OK):
         return f"==== Error: No read access to directory {path} ====\n==== End of Block ===="
@@ -70,10 +70,10 @@ def generate_file_tree(
 ) -> Dict:
     """Generate file tree structure."""
     if current_depth > max_depth:
-        return {}
+        return None
 
     if ignore_spec.match_file(path) or path.name == ".git":
-        return {}
+        return None
 
     if path.is_file():
         try:
@@ -95,16 +95,18 @@ def generate_file_tree(
     except (PermissionError, OSError):
         tree["children"].append({"name": "no access", "type": "error"})
         return tree
+
     for child in children:
         if not ignore_spec.match_file(child):
             if child.is_file():
                 child_tree = generate_file_tree(child, ignore_spec, recursive, max_depth, current_depth)
-                tree["children"].append(child_tree)
+                if child_tree:  # Only append if not None
+                    tree["children"].append(child_tree)
             elif child.is_dir():
                 # Always include directories
                 child_tree = generate_file_tree(child, ignore_spec, recursive, max_depth, current_depth + 1)
                 if recursive:
-                    if child_tree:
+                    if child_tree:  # Only append if not None
                         tree["children"].append(child_tree)
                 else:
                     tree["children"].append({"name": child.name, "type": "directory", "children": []})
@@ -114,9 +116,16 @@ def generate_file_tree(
 
 def format_tree(tree: Dict, start: int, end: int) -> str:
     """Format tree structure into string output with line information."""
+    if not tree:  # Handle empty or None tree
+        return "==== No files to display ===="
+    
     lines = []
     _format_tree_recursive(tree, lines, 0)
     total_lines = len(lines)
+    
+    if total_lines == 0:  # Handle case with no valid lines
+        return "==== No files to display ===="
+    
     is_last_block = end >= total_lines
     output = "\n".join(lines[start - 1 : end])
 
@@ -129,13 +138,18 @@ def format_tree(tree: Dict, start: int, end: int) -> str:
 
 def _format_tree_recursive(node: Dict, lines: List[str], depth: int):
     """Recursively format tree nodes."""
+    if not node or "type" not in node:  # Skip invalid nodes
+        return
+        
     indent = "  " * depth
     if node["type"] == "file":
         lines.append(f"{indent}📄 {node['name']} ({node['size']})")
-    else:
+    elif node["type"] == "directory":
         lines.append(f"{indent}📁 {node['name']}/")
-        for child in node["children"]:
+        for child in node.get("children", []):
             _format_tree_recursive(child, lines, depth + 1)
+    elif node["type"] == "error":
+        lines.append(f"{indent}❌ {node['name']}")
 
 
 if __name__ == "__main__":

quantalogic/utils/lm_studio_model_info.py

@@ -9,14 +9,17 @@ class ModelType(str, Enum):
     EMBEDDINGS = "embeddings"
     VLM = "vlm"
 
+
 class CompatibilityType(str, Enum):
     MLX = "mlx"
     GGUF = "gguf"
 
+
 class ModelState(str, Enum):
     LOADED = "loaded"
     NOT_LOADED = "not-loaded"
 
+
 class ModelInfo(BaseModel):
     id: str = Field(..., description="Unique model identifier in LM Studio's namespace")
     object: Literal["model"] = Field("model", description="Always 'model' for model objects")
@@ -28,21 +31,20 @@ class ModelInfo(BaseModel):
     state: ModelState = Field(..., description="Current loading state in LM Studio")
     max_context_length: int = Field(..., alias="max_context_length", ge=0)
     loaded_context_length: Optional[int] = Field(
-        None, 
-        alias="loaded_context_length",
-        description="Currently allocated context length (only when loaded)",
-        ge=0
+        None, alias="loaded_context_length", description="Currently allocated context length (only when loaded)", ge=0
     )
 
+
 class ModelListResponse(BaseModel):
     data: List[ModelInfo] = Field(..., description="List of available models")
     object: Literal["list"] = Field("list", description="Always 'list' for list responses")
 
+
 def get_model_list() -> ModelListResponse:
     """Fetch and validate model information from LM Studio's API"""
     import requests
-    
+
     response = requests.get("http://localhost:1234/api/v0/models")
     response.raise_for_status()
-    
-    return ModelListResponse(**response.json())
+
+    return ModelListResponse(**response.json())