praisonai-code 0.0.1__py3-none-any.whl

praisonai_code/cli/features/templates.py

@@ -0,0 +1,1390 @@
+"""
+Templates CLI Feature Handler
+
+Provides CLI commands for template management:
+- list, search, info, install, cache clear
+- run templates directly
+- init projects from templates
+- Custom templates directory support with precedence
+"""
+
+import shutil
+from pathlib import Path
+from typing import Any, Dict, List
+
+
+class TemplatesHandler:
+    """
+    CLI handler for template operations.
+    
+    Commands:
+    - list: List available templates
+    - search: Search templates by query
+    - info: Show template details
+    - install: Install a template to cache
+    - cache: Cache management (clear, list)
+    - run: Run a template directly
+    - init: Initialize a project from template
+    """
+    
+    def __init__(self):
+        """Initialize the handler with lazy-loaded dependencies."""
+        self._loader = None
+        self._registry = None
+        self._cache = None
+    
+    @property
+    def loader(self):
+        """Lazy load template loader."""
+        if self._loader is None:
+            from praisonai.templates import TemplateLoader
+            self._loader = TemplateLoader()
+        return self._loader
+    
+    @property
+    def registry(self):
+        """Lazy load template registry."""
+        if self._registry is None:
+            from praisonai.templates import TemplateRegistry
+            self._registry = TemplateRegistry()
+        return self._registry
+    
+    @property
+    def cache(self):
+        """Lazy load template cache."""
+        if self._cache is None:
+            from praisonai.templates import TemplateCache
+            self._cache = TemplateCache()
+        return self._cache
+    
+    def handle(self, args: List[str]) -> int:
+        """
+        Handle templates subcommand.
+        
+        Args:
+            args: Command arguments
+            
+        Returns:
+            Exit code (0 for success)
+        """
+        if not args:
+            self._print_help()
+            return 0
+        
+        command = args[0]
+        remaining = args[1:]
+        
+        commands = {
+            "list": self.cmd_list,
+            "search": self.cmd_search,
+            "info": self.cmd_info,
+            "install": self.cmd_install,
+            "cache": self.cmd_cache,
+            "run": self.cmd_run,
+            "init": self.cmd_init,
+            "add": self.cmd_add,
+            "add-sources": self.cmd_add_sources,
+            "remove-sources": self.cmd_remove_sources,
+            "browse": self.cmd_browse,
+            "catalog": self.cmd_catalog,
+            "validate": self.cmd_validate,
+            "help": lambda _: self._print_help() or 0,
+        }
+        
+        if command in commands:
+            return commands[command](remaining)
+        else:
+            print(f"[red]Unknown command: {command}[/red]")
+            self._print_help()
+            return 1
+    
+    def _print_help(self):
+        """Print help message."""
+        help_text = """
+[bold cyan]PraisonAI Templates[/bold cyan]
+
+[bold]Usage:[/bold]
+  praisonai templates <command> [options]
+
+[bold]Commands:[/bold]
+  list              List available templates
+  search <query>    Search templates by name or tags
+  info <template>   Show template details
+  install <uri>     Install a template to cache
+  cache clear       Clear the template cache
+  cache list        List cached templates
+  run <template>    Run a template directly
+  init <name>       Initialize a project from template
+  add <source>      Add template from GitHub or local path
+  add-sources <src> Add a template source to persistent config
+  remove-sources    Remove a template source from config
+  browse            Open template catalog in browser
+  catalog build     Build catalog locally
+  catalog sync      Sync catalog sources
+  validate          Validate template YAML files
+
+[bold]Options:[/bold]
+  --offline         Use only cached templates (no network)
+  --source <src>    Filter by source (custom, package, all)
+  --custom-dir <path>  Add custom templates directory
+  --paths           Show template search paths
+
+[bold]Examples:[/bold]
+  praisonai templates list
+  praisonai templates search video
+  praisonai templates info transcript-generator
+  praisonai templates install github:MervinPraison/agent-recipes/transcript-generator
+  praisonai templates run transcript-generator ./audio.mp3
+  praisonai templates add github:user/repo/my-template
+  praisonai templates add-sources github:MervinPraison/Agent-Recipes
+  praisonai init my-project --template transcript-generator
+"""
+        try:
+            from rich import print as rprint
+            rprint(help_text)
+        except ImportError:
+            print(help_text.replace("[bold cyan]", "").replace("[/bold cyan]", "")
+                  .replace("[bold]", "").replace("[/bold]", "")
+                  .replace("[red]", "").replace("[/red]", ""))
+    
+    def cmd_list(self, args: List[str]) -> int:
+        """List available templates from all sources including custom directories."""
+        source_filter = None
+        custom_dirs = []
+        show_paths = "--paths" in args
+        
+        # Parse --source filter
+        if "--source" in args:
+            idx = args.index("--source")
+            if idx + 1 < len(args):
+                source_filter = args[idx + 1]
+        
+        # Parse --custom-dir (can be specified multiple times)
+        i = 0
+        while i < len(args):
+            if args[i] == "--custom-dir" and i + 1 < len(args):
+                custom_dirs.append(args[i + 1])
+                i += 2
+            else:
+                i += 1
+        
+        try:
+            from praisonai.templates.discovery import TemplateDiscovery
+            
+            discovery = TemplateDiscovery(
+                custom_dirs=custom_dirs if custom_dirs else None,
+                include_package=True,
+                include_defaults=True
+            )
+            
+            # Show search paths if requested
+            if show_paths:
+                print("\nTemplate Search Paths (in priority order):")
+                for path, source, exists in discovery.get_search_paths():
+                    status = "✓" if exists else "✗"
+                    print(f"  {status} [{source}] {path}")
+                print()
+            
+            # Discover templates
+            templates = discovery.list_templates(source_filter=source_filter)
+            
+            if not templates:
+                print("No templates found.")
+                if not show_paths:
+                    print("Use --paths to see search locations.")
+                return 0
+            
+            try:
+                from rich.console import Console
+                from rich.table import Table
+                
+                console = Console()
+                table = Table(title="Available Templates")
+                table.add_column("Name", style="cyan")
+                table.add_column("Version", style="green")
+                table.add_column("Description")
+                table.add_column("Source", style="dim")
+                
+                for t in sorted(templates, key=lambda x: (x.priority, x.name)):
+                    desc = t.description or ""
+                    table.add_row(
+                        t.name,
+                        t.version or "1.0.0",
+                        desc[:50] + "..." if len(desc) > 50 else desc,
+                        t.source
+                    )
+                
+                console.print(table)
+            except ImportError:
+                print(f"{'Name':<25} {'Version':<10} {'Source':<10} {'Description':<35}")
+                print("-" * 80)
+                for t in sorted(templates, key=lambda x: (x.priority, x.name)):
+                    desc = (t.description or "")[:35]
+                    if len(t.description or "") > 35:
+                        desc += "..."
+                    print(f"{t.name:<25} {t.version or '1.0.0':<10} {t.source:<10} {desc:<35}")
+            
+            return 0
+            
+        except Exception as e:
+            print(f"Error listing templates: {e}")
+            import traceback
+            traceback.print_exc()
+            return 1
+    
+    def cmd_search(self, args: List[str]) -> int:
+        """Search templates by query."""
+        if not args or args[0].startswith("--"):
+            print("Usage: praisonai templates search <query>")
+            return 1
+        
+        query = args[0]
+        offline = "--offline" in args
+        
+        try:
+            from praisonai.templates import search_templates
+            templates = search_templates(query, offline=offline)
+            
+            if not templates:
+                print(f"No templates found matching '{query}'")
+                return 0
+            
+            print(f"Found {len(templates)} template(s) matching '{query}':\n")
+            for t in templates:
+                print(f"  • {t.name} (v{t.version})")
+                if t.description:
+                    print(f"    {t.description}")
+                if t.tags:
+                    print(f"    Tags: {', '.join(t.tags)}")
+                print()
+            
+            return 0
+            
+        except Exception as e:
+            print(f"Error searching templates: {e}")
+            return 1
+    
+    def cmd_info(self, args: List[str]) -> int:
+        """Show template details with custom directory support."""
+        if not args or args[0].startswith("--"):
+            print("Usage: praisonai templates info <template>")
+            return 1
+        
+        name_or_uri = args[0]
+        offline = "--offline" in args
+        custom_dirs = []
+        
+        # Parse --custom-dir
+        i = 0
+        while i < len(args):
+            if args[i] == "--custom-dir" and i + 1 < len(args):
+                custom_dirs.append(args[i + 1])
+                i += 2
+            else:
+                i += 1
+        
+        try:
+            # First try to find in custom/local directories
+            from praisonai.templates.discovery import TemplateDiscovery
+            
+            discovery = TemplateDiscovery(
+                custom_dirs=custom_dirs if custom_dirs else None,
+                include_package=True,
+                include_defaults=True
+            )
+            
+            discovered = discovery.find_template(name_or_uri)
+            
+            if discovered:
+                # Load from discovered path
+                from praisonai.templates import load_template
+                template = load_template(str(discovered.path), offline=offline)
+            else:
+                # Fall back to URI resolution
+                from praisonai.templates import load_template
+                template = load_template(name_or_uri, offline=offline)
+            
+            # Check dependency availability
+            from praisonai.templates.dependency_checker import DependencyChecker
+            checker = DependencyChecker()
+            deps = checker.check_template_dependencies(template)
+            
+            # Build availability strings
+            def format_dep_list(items, key="available"):
+                result = []
+                for item in items:
+                    status = "✓" if item.get(key, False) else "✗"
+                    hint = ""
+                    if not item.get(key, False):
+                        if item.get("install_hint"):
+                            hint = f" ({item['install_hint']})"
+                    result.append(f"{status} {item['name']}{hint}")
+                return result
+            
+            tools_status = format_dep_list(deps["tools"])
+            pkgs_status = format_dep_list(deps["packages"])
+            env_status = format_dep_list(deps["env"])
+            
+            try:
+                from rich.console import Console
+                from rich.panel import Panel
+                from rich.markdown import Markdown
+                
+                console = Console()
+                
+                info = f"""
+**Name:** {template.name}
+**Version:** {template.version}
+**Author:** {template.author or 'Unknown'}
+**License:** {template.license or 'Not specified'}
+
+**Description:**
+{template.description}
+
+**Tags:** {', '.join(template.tags) if template.tags else 'None'}
+
+**Skills:** {', '.join(template.skills) if template.skills else 'None'}
+
+**Path:** {template.path}
+"""
+                console.print(Panel(Markdown(info), title=f"Template: {template.name}"))
+                
+                # Print dependency status
+                all_ok = "✓" if deps["all_satisfied"] else "✗"
+                console.print(f"\n[bold]Dependencies Status:[/bold] {all_ok}")
+                
+                if tools_status:
+                    console.print("\n[bold]Required Tools:[/bold]")
+                    for t in tools_status:
+                        color = "green" if t.startswith("✓") else "red"
+                        console.print(f"  [{color}]{t}[/{color}]")
+                
+                if pkgs_status:
+                    console.print("\n[bold]Required Packages:[/bold]")
+                    for p in pkgs_status:
+                        color = "green" if p.startswith("✓") else "red"
+                        console.print(f"  [{color}]{p}[/{color}]")
+                
+                if env_status:
+                    console.print("\n[bold]Required Environment:[/bold]")
+                    for e in env_status:
+                        color = "green" if e.startswith("✓") else "red"
+                        console.print(f"  [{color}]{e}[/{color}]")
+                
+                # Print install hints if any missing
+                if not deps["all_satisfied"]:
+                    hints = checker.get_install_hints(template)
+                    if hints:
+                        console.print("\n[bold yellow]To fix missing dependencies:[/bold yellow]")
+                        for hint in hints:
+                            console.print(f"  • {hint}")
+                
+            except ImportError:
+                print(f"Template: {template.name}")
+                print(f"Version: {template.version}")
+                print(f"Author: {template.author or 'Unknown'}")
+                print(f"Description: {template.description}")
+                print(f"Path: {template.path}")
+                print(f"\nDependencies: {'All satisfied' if deps['all_satisfied'] else 'Some missing'}")
+                if tools_status:
+                    print("\nRequired Tools:")
+                    for t in tools_status:
+                        print(f"  {t}")
+                if pkgs_status:
+                    print("\nRequired Packages:")
+                    for p in pkgs_status:
+                        print(f"  {p}")
+                if env_status:
+                    print("\nRequired Environment:")
+                    for e in env_status:
+                        print(f"  {e}")
+            
+            return 0
+            
+        except Exception as e:
+            print(f"Error loading template info: {e}")
+            return 1
+    
+    def cmd_install(self, args: List[str]) -> int:
+        """Install a template to cache."""
+        if not args or args[0].startswith("--"):
+            print("Usage: praisonai templates install <uri>")
+            return 1
+        
+        uri = args[0]
+        
+        try:
+            from praisonai.templates import install_template
+            
+            print(f"Installing template: {uri}")
+            cached = install_template(uri)
+            print(f"✓ Template installed to: {cached.path}")
+            
+            return 0
+            
+        except Exception as e:
+            print(f"Error installing template: {e}")
+            return 1
+    
+    def cmd_cache(self, args: List[str]) -> int:
+        """Cache management commands."""
+        if not args:
+            print("Usage: praisonai templates cache <clear|list|size>")
+            return 1
+        
+        subcmd = args[0]
+        
+        if subcmd == "clear":
+            source = None
+            if len(args) > 1:
+                source = args[1]
+            
+            try:
+                from praisonai.templates import clear_cache
+                count = clear_cache(source)
+                print(f"✓ Cleared {count} cached template(s)")
+                return 0
+            except Exception as e:
+                print(f"Error clearing cache: {e}")
+                return 1
+        
+        elif subcmd == "list":
+            try:
+                cached = self.cache.list_cached()
+                if not cached:
+                    print("No cached templates.")
+                    return 0
+                
+                print(f"Cached templates ({len(cached)}):\n")
+                for path, meta in cached:
+                    status = "pinned" if meta.is_pinned else "expires in " + str(int(meta.ttl_seconds - (import_time() - meta.fetched_at))) + "s"
+                    print(f"  • {path.name}")
+                    print(f"    Path: {path}")
+                    print(f"    Status: {status}")
+                    print()
+                
+                return 0
+            except Exception as e:
+                print(f"Error listing cache: {e}")
+                return 1
+        
+        elif subcmd == "size":
+            size = self.cache.get_cache_size()
+            print(f"Cache size: {size / 1024 / 1024:.2f} MB")
+            return 0
+        
+        else:
+            print(f"Unknown cache command: {subcmd}")
+            return 1
+    
+    def cmd_run(self, args: List[str]) -> int:
+        """Run a template directly."""
+        if not args or args[0].startswith("--"):
+            print("Usage: praisonai templates run <template> [args...] [--strict-tools] [--offline]")
+            return 1
+        
+        uri = args[0]
+        template_args = args[1:]
+        offline = "--offline" in args
+        strict_tools = "--strict-tools" in args
+        no_template_tools_py = "--no-template-tools-py" in args
+        
+        # Parse --tools, --tools-dir, --tools-source overrides
+        tools_files = []
+        tools_dirs = []
+        tools_sources_override = []
+        i = 0
+        while i < len(args):
+            if args[i] == "--tools" and i + 1 < len(args):
+                tools_files.append(args[i + 1])
+                i += 2
+            elif args[i] == "--tools-dir" and i + 1 < len(args):
+                tools_dirs.append(args[i + 1])
+                i += 2
+            elif args[i] == "--tools-source" and i + 1 < len(args):
+                tools_sources_override.append(args[i + 1])
+                i += 2
+            else:
+                i += 1
+        
+        try:
+            from praisonai.templates.loader import TemplateLoader
+            from praisonai.templates.discovery import TemplateDiscovery
+            
+            # First try to find in custom/local directories
+            discovery = TemplateDiscovery(
+                custom_dirs=tools_dirs if tools_dirs else None,
+                include_package=True,
+                include_defaults=True
+            )
+            
+            discovered = discovery.find_template(uri)
+            
+            loader = TemplateLoader(offline=offline)
+            
+            if discovered:
+                # Load from discovered path
+                template = loader.load(str(discovered.path), offline=offline)
+            else:
+                # Fall back to URI resolution
+                template = loader.load(uri, offline=offline)
+            
+            # Strict mode: fail-fast on missing dependencies
+            if strict_tools:
+                from praisonai.templates.dependency_checker import DependencyChecker, StrictModeError
+                checker = DependencyChecker()
+                try:
+                    checker.enforce_strict_mode(template)
+                    print("✓ All dependencies satisfied (strict mode)")
+                except StrictModeError as e:
+                    print(f"✗ Strict mode check failed:\n{e}")
+                    return 1
+            else:
+                # Non-strict: warn but continue
+                missing = loader.check_requirements(template)
+                if missing["missing_packages"]:
+                    print(f"Warning: Missing packages: {', '.join(missing['missing_packages'])}")
+                    print("Install with: pip install " + " ".join(missing["missing_packages"]))
+                if missing["missing_env"]:
+                    print(f"Warning: Missing environment variables: {', '.join(missing['missing_env'])}")
+            
+            # Load tool overrides if specified
+            tool_registry = None
+            from praisonai.templates.tool_override import create_tool_registry_with_overrides, resolve_tools
+            
+            # Get template directory for local tools.py autoload (unless disabled)
+            template_dir = None
+            if not no_template_tools_py:
+                template_dir = str(template.path) if template.path else None
+            
+            # Get tools_sources from template requires + CLI overrides
+            tools_sources = []
+            if template.requires and isinstance(template.requires, dict):
+                ts = template.requires.get("tools_sources", [])
+                if ts:
+                    tools_sources.extend(ts)
+            # Add CLI --tools-source overrides
+            if tools_sources_override:
+                tools_sources.extend(tools_sources_override)
+            
+            # Always build registry (includes defaults + template sources)
+            tool_registry = create_tool_registry_with_overrides(
+                override_files=tools_files if tools_files else None,
+                override_dirs=tools_dirs if tools_dirs else None,
+                include_defaults=True,
+                tools_sources=tools_sources if tools_sources else None,
+                template_dir=template_dir,
+            )
+            
+            if tools_files or tools_dirs:
+                print(f"✓ Loaded {len(tool_registry)} tools from overrides")
+            
+            # Load and run workflow
+            workflow_config = loader.load_workflow_config(template)
+            
+            # Parse template args into config
+            config = self._parse_template_args(template_args, template)
+            
+            # Merge config
+            if config:
+                workflow_config = {**workflow_config, **config}
+            
+            # Run workflow - determine which class to use based on config structure
+            if "agents" in workflow_config and "tasks" in workflow_config:
+                # Agents format (agents + tasks)
+                from praisonaiagents import Agent, Task, AgentTeam
+                
+                # Build agents
+                agents_config = workflow_config.get("agents", [])
+                agents = []
+                agent_map = {}
+                
+                for agent_cfg in agents_config:
+                    # Resolve tool names to callable tools
+                    agent_tool_names = agent_cfg.get("tools", [])
+                    resolved_agent_tools = resolve_tools(
+                        agent_tool_names,
+                        registry=tool_registry,
+                        template_dir=template_dir
+                    )
+                    
+                    agent = Agent(
+                        name=agent_cfg.get("name", "Agent"),
+                        role=agent_cfg.get("role", ""),
+                        goal=agent_cfg.get("goal", ""),
+                        backstory=agent_cfg.get("backstory", ""),
+                        tools=resolved_agent_tools,
+                        llm=agent_cfg.get("llm"),
+                        verbose=agent_cfg.get("verbose", True)
+                    )
+                    agents.append(agent)
+                    agent_map[agent_cfg.get("name", "Agent")] = agent
+                
+                # Build tasks
+                tasks_config = workflow_config.get("tasks", [])
+                tasks = []
+                
+                for task_cfg in tasks_config:
+                    agent_name = task_cfg.get("agent", "")
+                    agent = agent_map.get(agent_name, agents[0] if agents else None)
+                    
+                    task = Task(
+                        name=task_cfg.get("name", "Task"),
+                        description=task_cfg.get("description", ""),
+                        expected_output=task_cfg.get("expected_output", ""),
+                        agent=agent
+                    )
+                    tasks.append(task)
+                
+                # Run
+                praison_agents = AgentTeam(
+                    agents=agents,
+                    tasks=tasks,
+                    process=workflow_config.get("process", "sequential"),
+                    verbose=workflow_config.get("verbose", 1)
+                )
+                praison_agents.start()
+            elif "steps" in workflow_config:
+                # Workflow format (steps) - resolve tools in each step's agent
+                from praisonaiagents import Workflow
+                
+                # Resolve tools for each step's agent if specified
+                steps = workflow_config.get("steps", [])
+                for step in steps:
+                    if "agent" in step and isinstance(step["agent"], dict):
+                        agent_cfg = step["agent"]
+                        if "tools" in agent_cfg:
+                            agent_cfg["tools"] = resolve_tools(
+                                agent_cfg["tools"],
+                                registry=tool_registry,
+                                template_dir=template_dir
+                            )
+                
+                workflow = Workflow(**workflow_config)
+                workflow.run()
+            else:
+                raise ValueError("Invalid workflow config: must have 'agents'+'tasks' or 'steps'")
+            
+            print(f"\n✓ Template '{template.name}' completed successfully")
+            return 0
+            
+        except Exception as e:
+            print(f"Error running template: {e}")
+            import traceback
+            traceback.print_exc()
+            return 1
+    
+    def cmd_init(self, args: List[str]) -> int:
+        """Initialize a project from template."""
+        if not args or args[0].startswith("--"):
+            print("Usage: praisonai templates init <project-name> --template <template>")
+            return 1
+        
+        project_name = args[0]
+        template_uri = None
+        offline = "--offline" in args
+        
+        if "--template" in args:
+            idx = args.index("--template")
+            if idx + 1 < len(args):
+                template_uri = args[idx + 1]
+        
+        if not template_uri:
+            print("Error: --template is required")
+            return 1
+        
+        try:
+            from praisonai.templates import load_template
+            
+            template = load_template(template_uri, offline=offline)
+            
+            # Create project directory
+            project_dir = Path(project_name)
+            if project_dir.exists():
+                print(f"Error: Directory '{project_name}' already exists")
+                return 1
+            
+            project_dir.mkdir(parents=True)
+            
+            # Copy template files
+            for item in template.path.iterdir():
+                if item.name.startswith("."):
+                    continue
+                if item.is_file():
+                    shutil.copy2(item, project_dir / item.name)
+                elif item.is_dir():
+                    shutil.copytree(item, project_dir / item.name)
+            
+            print(f"✓ Created project '{project_name}' from template '{template.name}'")
+            print("Next steps:")
+            print(f"  cd {project_name}")
+            print("  praisonai run workflow.yaml")
+            
+            return 0
+            
+        except Exception as e:
+            print(f"Error initializing project: {e}")
+            return 1
+    
+    def _parse_template_args(
+        self,
+        args: List[str],
+        template
+    ) -> Dict[str, Any]:
+        """Parse template-specific arguments."""
+        config = {}
+        
+        # Handle positional args based on CLI config
+        cli_config = template.cli
+        if cli_config and "args" in cli_config:
+            positional_idx = 0
+            for arg_def in cli_config["args"]:
+                if arg_def.get("positional"):
+                    if positional_idx < len(args) and not args[positional_idx].startswith("--"):
+                        config[arg_def["name"]] = args[positional_idx]
+                        positional_idx += 1
+        
+        # Handle named args
+        i = 0
+        while i < len(args):
+            if args[i].startswith("--"):
+                key = args[i][2:].replace("-", "_")
+                if i + 1 < len(args) and not args[i + 1].startswith("--"):
+                    config[key] = args[i + 1]
+                    i += 2
+                else:
+                    config[key] = True
+                    i += 1
+            else:
+                i += 1
+        
+        return config
+    
+    def _get_templates_config_path(self):
+        """Get the path to the templates config file."""
+        config_dir = Path.home() / ".praison"
+        config_dir.mkdir(parents=True, exist_ok=True)
+        return config_dir / "templates_sources.yaml"
+    
+    def _load_templates_config(self) -> Dict[str, Any]:
+        """Load templates config from file."""
+        import yaml
+        config_path = self._get_templates_config_path()
+        if config_path.exists():
+            with open(config_path, 'r') as f:
+                return yaml.safe_load(f) or {}
+        return {"sources": []}
+    
+    def _save_templates_config(self, config: Dict[str, Any]):
+        """Save templates config to file."""
+        import yaml
+        config_path = self._get_templates_config_path()
+        with open(config_path, 'w') as f:
+            yaml.dump(config, f, default_flow_style=False)
+    
+    def cmd_add(self, args: List[str]) -> int:
+        """
+        Add template from GitHub or local path.
+        
+        Args:
+            args: [source] - github:user/repo/template or local path
+        """
+        if not args:
+            print("[red]Usage: praisonai templates add <source>[/red]")
+            print("  source: github:user/repo/template or local path")
+            return 1
+        
+        source = args[0]
+        
+        # Check if it's a local directory
+        if source.startswith("./") or source.startswith("/"):
+            path = Path(source).resolve()
+            if path.exists() and path.is_dir():
+                # Check for TEMPLATE.yaml
+                template_yaml = path / "TEMPLATE.yaml"
+                if not template_yaml.exists():
+                    print(f"[red]Not a valid template: {path} (missing TEMPLATE.yaml)[/red]")
+                    return 1
+                
+                # Copy to ~/.praison/templates/
+                templates_dir = Path.home() / ".praison" / "templates"
+                templates_dir.mkdir(parents=True, exist_ok=True)
+                dest = templates_dir / path.name
+                
+                if dest.exists():
+                    shutil.rmtree(dest)
+                shutil.copytree(path, dest)
+                
+                print(f"\n✅ Added template: {path.name}")
+                print(f"   Copied to: {dest}")
+                return 0
+            else:
+                print(f"[red]Directory not found: {source}[/red]")
+                return 1
+        
+        # Check if it's a GitHub reference
+        elif source.startswith("github:"):
+            github_path = source[7:]  # Remove "github:"
+            parts = github_path.split("/")
+            if len(parts) < 3:
+                print("[red]Invalid GitHub format. Use: github:user/repo/template-name[/red]")
+                return 1
+            
+            user, repo = parts[0], parts[1]
+            template_path = "/".join(parts[2:])
+            
+            # Download from GitHub
+            import urllib.request
+            import tempfile
+            import zipfile
+            
+            try:
+                # Download repo as zip
+                zip_url = f"https://github.com/{user}/{repo}/archive/refs/heads/main.zip"
+                print(f"Downloading from: {zip_url}")
+                
+                with tempfile.TemporaryDirectory() as tmpdir:
+                    zip_path = Path(tmpdir) / "repo.zip"
+                    urllib.request.urlretrieve(zip_url, zip_path)
+                    
+                    # Extract safely, preventing Zip Slip
+                    with zipfile.ZipFile(zip_path, 'r') as zip_ref:
+                        import os
+                        tmpdir_real = os.path.realpath(tmpdir)
+                        for zip_info in zip_ref.infolist():
+                            extracted_path = os.path.realpath(os.path.join(tmpdir_real, zip_info.filename))
+                            if not extracted_path.startswith(tmpdir_real):
+                                raise Exception(f"Attempted Zip Slip vulnerabilities detected: {zip_info.filename}")
+                        zip_ref.extractall(tmpdir)
+                    
+                    # Find the template
+                    extracted_dir = Path(tmpdir) / f"{repo}-main"
+                    template_src = extracted_dir / template_path
+                    
+                    if not template_src.exists():
+                        # Try common paths
+                        for alt_path in [
+                            extracted_dir / "agent_recipes" / "templates" / template_path,
+                            extracted_dir / "templates" / template_path,
+                        ]:
+                            if alt_path.exists():
+                                template_src = alt_path
+                                break
+                    
+                    if not template_src.exists():
+                        print(f"[red]Template not found: {template_path}[/red]")
+                        return 1
+                    
+                    # Copy to ~/.praison/templates/
+                    templates_dir = Path.home() / ".praison" / "templates"
+                    templates_dir.mkdir(parents=True, exist_ok=True)
+                    dest = templates_dir / template_src.name
+                    
+                    if dest.exists():
+                        shutil.rmtree(dest)
+                    shutil.copytree(template_src, dest)
+                    
+                    print(f"\n✅ Added template from GitHub: {user}/{repo}/{template_path}")
+                    print(f"   Saved to: {dest}")
+                    return 0
+                    
+            except Exception as e:
+                print(f"[red]Failed to download from GitHub: {e}[/red]")
+                return 1
+        
+        else:
+            print(f"[red]Unknown source format: {source}[/red]")
+            print("Use: github:user/repo/template or ./local-path")
+            return 1
+    
+    def cmd_add_sources(self, args: List[str]) -> int:
+        """
+        Add a template source to persistent config.
+        
+        Args:
+            args: [source] - github:user/repo or URL
+        """
+        if not args:
+            print("[red]Usage: praisonai templates add-sources <source>[/red]")
+            return 1
+        
+        source = args[0]
+        config = self._load_templates_config()
+        
+        if "sources" not in config:
+            config["sources"] = []
+        
+        if source in config["sources"]:
+            print(f"[yellow]Source '{source}' already in config[/yellow]")
+            return 0
+        
+        config["sources"].append(source)
+        self._save_templates_config(config)
+        
+        print(f"\n✅ Added template source: {source}")
+        print(f"   Config saved to: {self._get_templates_config_path()}")
+        return 0
+    
+    def cmd_remove_sources(self, args: List[str]) -> int:
+        """
+        Remove a template source from persistent config.
+        
+        Args:
+            args: [source] - source to remove
+        """
+        if not args:
+            print("[red]Usage: praisonai templates remove-sources <source>[/red]")
+            return 1
+        
+        source = args[0]
+        config = self._load_templates_config()
+        
+        if "sources" not in config or source not in config["sources"]:
+            print(f"[yellow]Source '{source}' not found in config[/yellow]")
+            return 1
+        
+        config["sources"].remove(source)
+        self._save_templates_config(config)
+        
+        print(f"\n✅ Removed template source: {source}")
+        return 0
+
+    def cmd_browse(self, args: List[str]) -> int:
+        """
+        Open template catalog in browser.
+        
+        Args:
+            args: [--local] [--url <url>] [--print]
+        """
+        import webbrowser
+        
+        # Default catalog URL
+        catalog_url = "https://mervinpraison.github.io/praisonai-template-catalog"
+        
+        # Parse arguments
+        print_only = "--print" in args
+        local_mode = "--local" in args
+        
+        # Custom URL
+        if "--url" in args:
+            idx = args.index("--url")
+            if idx + 1 < len(args):
+                catalog_url = args[idx + 1]
+        
+        if local_mode:
+            print("Local catalog server not implemented yet.")
+            print(f"Visit the online catalog at: {catalog_url}")
+            return 0
+        
+        if print_only:
+            print(catalog_url)
+            return 0
+        
+        print(f"Opening template catalog: {catalog_url}")
+        try:
+            webbrowser.open(catalog_url)
+            return 0
+        except Exception as e:
+            print(f"Failed to open browser: {e}")
+            print(f"Visit: {catalog_url}")
+            return 1
+
+    def cmd_catalog(self, args: List[str]) -> int:
+        """
+        Catalog management commands.
+        
+        Subcommands:
+            build   Build catalog locally
+            sync    Sync catalog sources
+        """
+        if not args:
+            print("Usage: praisonai templates catalog <build|sync> [options]")
+            print("\nSubcommands:")
+            print("  build     Build catalog locally")
+            print("  sync      Sync catalog sources from GitHub")
+            print("\nExamples:")
+            print("  praisonai templates catalog build --out ./dist")
+            print("  praisonai templates catalog sync --source agent-recipes")
+            return 0
+        
+        subcmd = args[0]
+        remaining = args[1:]
+        
+        if subcmd == "build":
+            return self._catalog_build(remaining)
+        elif subcmd == "sync":
+            return self._catalog_sync(remaining)
+        else:
+            print(f"Unknown catalog command: {subcmd}")
+            return 1
+
+    def _catalog_build(self, args: List[str]) -> int:
+        """Build catalog locally."""
+        import subprocess
+        import os
+        
+        # Parse arguments
+        out_dir = None
+        source_dir = None
+        minify = "--minify" in args
+        
+        if "--out" in args:
+            idx = args.index("--out")
+            if idx + 1 < len(args):
+                out_dir = args[idx + 1]
+        
+        if "--source" in args:
+            idx = args.index("--source")
+            if idx + 1 < len(args):
+                source_dir = args[idx + 1]
+        
+        # Check if catalog repo is available locally
+        catalog_repo = Path.home() / "praisonai-template-catalog"
+        if not catalog_repo.exists():
+            # Try to find it relative to this package
+            possible_paths = [
+                Path(__file__).parent.parent.parent.parent.parent.parent / "praisonai-template-catalog",
+                Path.cwd() / "praisonai-template-catalog",
+            ]
+            for p in possible_paths:
+                if p.exists():
+                    catalog_repo = p
+                    break
+        
+        if catalog_repo.exists() and (catalog_repo / "scripts" / "build-catalog.js").exists():
+            print(f"Building catalog using: {catalog_repo}")
+            cmd = ["node", "scripts/build-catalog.js"]
+            if out_dir:
+                cmd.extend(["--out", out_dir])
+            if source_dir:
+                cmd.extend(["--source", source_dir])
+            if minify:
+                cmd.append("--minify")
+            
+            try:
+                result = subprocess.run(cmd, cwd=str(catalog_repo), capture_output=True, text=True)
+                print(result.stdout)
+                if result.stderr:
+                    print(result.stderr)
+                return result.returncode
+            except FileNotFoundError:
+                print("Node.js not found. Please install Node.js to build the catalog.")
+                return 1
+        else:
+            # Fallback: Generate minimal catalog using Python
+            print("Catalog repo not found locally. Generating minimal catalog...")
+            return self._generate_minimal_catalog(out_dir, source_dir)
+
+    def _generate_minimal_catalog(self, out_dir: str = None, source_dir: str = None) -> int:
+        """Generate minimal catalog JSON using Python."""
+        import json
+        from datetime import datetime
+        
+        try:
+            import yaml
+        except ImportError:
+            print("PyYAML not installed. Install with: pip install pyyaml")
+            return 1
+        
+        # Find templates
+        if source_dir:
+            templates_dir = Path(source_dir)
+        else:
+            # Try Agent-Recipes
+            possible_paths = [
+                Path.home() / "Agent-Recipes" / "agent_recipes" / "templates",
+                Path.cwd() / "Agent-Recipes" / "agent_recipes" / "templates",
+            ]
+            templates_dir = None
+            for p in possible_paths:
+                if p.exists():
+                    templates_dir = p
+                    break
+            
+            if not templates_dir:
+                # Try package templates
+                try:
+                    import agent_recipes
+                    templates_dir = Path(agent_recipes.__file__).parent / "templates"
+                except ImportError:
+                    pass
+        
+        if not templates_dir or not templates_dir.exists():
+            print("No templates directory found.")
+            return 1
+        
+        print(f"Scanning templates in: {templates_dir}")
+        
+        templates = []
+        for entry in templates_dir.iterdir():
+            if not entry.is_dir():
+                continue
+            template_yaml = entry / "TEMPLATE.yaml"
+            if template_yaml.exists():
+                try:
+                    with open(template_yaml) as f:
+                        data = yaml.safe_load(f)
+                    if data:
+                        templates.append({
+                            "name": data.get("name", entry.name),
+                            "version": data.get("version", "1.0.0"),
+                            "description": data.get("description", ""),
+                            "author": data.get("author", "Unknown"),
+                            "license": data.get("license", "Apache-2.0"),
+                            "tags": data.get("tags", []),
+                            "requires": data.get("requires", {}),
+                        })
+                except Exception as e:
+                    print(f"  Warning: Failed to parse {template_yaml}: {e}")
+        
+        # Output
+        output = {
+            "version": "1.0.0",
+            "generated_at": datetime.now().isoformat(),
+            "count": len(templates),
+            "templates": templates
+        }
+        
+        out_path = Path(out_dir) if out_dir else Path.cwd() / "templates.json"
+        if out_path.is_dir():
+            out_path = out_path / "templates.json"
+        
+        out_path.parent.mkdir(parents=True, exist_ok=True)
+        with open(out_path, "w") as f:
+            json.dump(output, f, indent=2)
+        
+        print(f"Generated: {out_path} ({len(templates)} templates)")
+        return 0
+
+    def _catalog_sync(self, args: List[str]) -> int:
+        """Sync catalog sources."""
+        import subprocess
+        
+        # Parse arguments
+        config_path = None
+        source_name = None
+        cache_dir = None
+        
+        if "--config" in args:
+            idx = args.index("--config")
+            if idx + 1 < len(args):
+                config_path = args[idx + 1]
+        
+        if "--source" in args:
+            idx = args.index("--source")
+            if idx + 1 < len(args):
+                source_name = args[idx + 1]
+        
+        if "--cache-dir" in args:
+            idx = args.index("--cache-dir")
+            if idx + 1 < len(args):
+                cache_dir = args[idx + 1]
+        
+        # Check if catalog repo is available
+        catalog_repo = Path.home() / "praisonai-template-catalog"
+        if catalog_repo.exists() and (catalog_repo / "scripts" / "sync-sources.js").exists():
+            print(f"Syncing using: {catalog_repo}")
+            cmd = ["node", "scripts/sync-sources.js"]
+            if config_path:
+                cmd.extend(["--config", config_path])
+            if source_name:
+                cmd.extend(["--source", source_name])
+            if cache_dir:
+                cmd.extend(["--cache-dir", cache_dir])
+            
+            try:
+                result = subprocess.run(cmd, cwd=str(catalog_repo), capture_output=True, text=True)
+                print(result.stdout)
+                if result.stderr:
+                    print(result.stderr)
+                return result.returncode
+            except FileNotFoundError:
+                print("Node.js not found. Please install Node.js.")
+                return 1
+        else:
+            # Fallback: Clone Agent-Recipes
+            print("Catalog repo not found. Cloning Agent-Recipes directly...")
+            cache_path = Path(cache_dir) if cache_dir else Path.home() / ".praison" / "cache"
+            cache_path.mkdir(parents=True, exist_ok=True)
+            
+            target = cache_path / "Agent-Recipes"
+            if target.exists():
+                print(f"Updating: {target}")
+                try:
+                    subprocess.run(["git", "-C", str(target), "pull", "--depth=1"], check=True)
+                    print("✓ Updated successfully")
+                    return 0
+                except subprocess.CalledProcessError as e:
+                    print(f"Failed to update: {e}")
+                    return 1
+            else:
+                print(f"Cloning to: {target}")
+                try:
+                    subprocess.run([
+                        "git", "clone", "--depth=1",
+                        "https://github.com/MervinPraison/Agent-Recipes.git",
+                        str(target)
+                    ], check=True)
+                    print("✓ Cloned successfully")
+                    return 0
+                except subprocess.CalledProcessError as e:
+                    print(f"Failed to clone: {e}")
+                    return 1
+
+    def cmd_validate(self, args: List[str]) -> int:
+        """
+        Validate template YAML files.
+        
+        Args:
+            args: [--source <dir>] [--strict] [--json]
+        """
+        import subprocess
+        
+        # Parse arguments
+        source_dir = None
+        strict = "--strict" in args
+        json_output = "--json" in args
+        
+        if "--source" in args:
+            idx = args.index("--source")
+            if idx + 1 < len(args):
+                source_dir = args[idx + 1]
+        
+        # Check if catalog repo is available
+        catalog_repo = Path.home() / "praisonai-template-catalog"
+        if catalog_repo.exists() and (catalog_repo / "scripts" / "validate-templates.js").exists():
+            cmd = ["node", "scripts/validate-templates.js"]
+            if source_dir:
+                cmd.extend(["--source", source_dir])
+            if strict:
+                cmd.append("--strict")
+            if json_output:
+                cmd.append("--json")
+            
+            try:
+                result = subprocess.run(cmd, cwd=str(catalog_repo), capture_output=True, text=True)
+                print(result.stdout)
+                if result.stderr:
+                    print(result.stderr)
+                return result.returncode
+            except FileNotFoundError:
+                print("Node.js not found.")
+                return 1
+        else:
+            # Fallback: Basic Python validation
+            return self._validate_templates_python(source_dir, strict, json_output)
+
+    def _validate_templates_python(self, source_dir: str = None, strict: bool = False, json_output: bool = False) -> int:
+        """Validate templates using Python."""
+        import json as json_module
+        
+        try:
+            import yaml
+        except ImportError:
+            print("PyYAML not installed. Install with: pip install pyyaml")
+            return 1
+        
+        # Find templates directory
+        if source_dir:
+            templates_dir = Path(source_dir)
+        else:
+            possible_paths = [
+                Path.home() / "Agent-Recipes" / "agent_recipes" / "templates",
+                Path.cwd() / "Agent-Recipes" / "agent_recipes" / "templates",
+            ]
+            templates_dir = None
+            for p in possible_paths:
+                if p.exists():
+                    templates_dir = p
+                    break
+        
+        if not templates_dir or not templates_dir.exists():
+            print(f"Templates directory not found: {source_dir or 'default locations'}")
+            return 1
+        
+        print(f"Validating templates in: {templates_dir}\n")
+        
+        results = []
+        errors_count = 0
+        warnings_count = 0
+        
+        for entry in sorted(templates_dir.iterdir()):
+            if not entry.is_dir() or entry.name.startswith("."):
+                continue
+            
+            template_yaml = entry / "TEMPLATE.yaml"
+            if not template_yaml.exists():
+                continue
+            
+            result = {"name": entry.name, "valid": True, "errors": [], "warnings": []}
+            
+            try:
+                with open(template_yaml) as f:
+                    data = yaml.safe_load(f)
+                
+                # Check required fields
+                required = ["name", "version", "description"]
+                for field in required:
+                    if not data.get(field):
+                        result["errors"].append(f"Missing required field: {field}")
+                        result["valid"] = False
+                
+                # Check version format
+                version = data.get("version", "")
+                if version and not self._is_valid_version(version):
+                    result["warnings"].append(f"Invalid version format: {version}")
+                
+                # Check workflow file
+                workflow = data.get("workflow", "workflow.yaml")
+                if isinstance(workflow, str) and not (entry / workflow).exists():
+                    if strict:
+                        result["errors"].append(f"Workflow file not found: {workflow}")
+                        result["valid"] = False
+                    else:
+                        result["warnings"].append(f"Workflow file not found: {workflow}")
+                
+            except yaml.YAMLError as e:
+                result["errors"].append(f"Invalid YAML: {e}")
+                result["valid"] = False
+            except Exception as e:
+                result["errors"].append(f"Error: {e}")
+                result["valid"] = False
+            
+            results.append(result)
+            errors_count += len(result["errors"])
+            warnings_count += len(result["warnings"])
+        
+        # Output
+        if json_output:
+            print(json_module.dumps(results, indent=2))
+        else:
+            for r in results:
+                status = "✓" if r["valid"] else "✗"
+                color = "\033[32m" if r["valid"] else "\033[31m"
+                print(f"{color}{status}\033[0m {r['name']}")
+                for err in r["errors"]:
+                    print(f"  \033[31m✗ ERROR:\033[0m {err}")
+                for warn in r["warnings"]:
+                    print(f"  \033[33m⚠ WARNING:\033[0m {warn}")
+            
+            print(f"\n{len(results)} templates checked, {errors_count} errors, {warnings_count} warnings")
+        
+        return 1 if errors_count > 0 else 0
+
+    def _is_valid_version(self, version: str) -> bool:
+        """Check if version string is valid semver."""
+        import re
+        return bool(re.match(r'^\d+\.\d+\.\d+', version))
+
+
+def import_time():
+    """Get current time (helper for cache expiry display)."""
+    import time
+    return time.time()
+
+
+# Convenience function for CLI integration
+def handle_templates_command(args: List[str]) -> int:
+    """Handle templates command from main CLI."""
+    handler = TemplatesHandler()
+    return handler.handle(args)