PraisonAI 3.0.0__py3-none-any.whl

praisonai/cli/features/doctor/checks/serve_checks.py

@@ -0,0 +1,426 @@
+"""
+Doctor checks for serve and endpoints functionality.
+
+Checks for:
+- Serve module availability
+- Endpoints module availability
+- Server connectivity
+- Discovery endpoint
+- Provider types
+"""
+
+import os
+
+from ..models import CheckResult, CheckStatus, CheckCategory, CheckSeverity, DoctorConfig
+from ..registry import register_check
+
+
+@register_check(
+    id="serve_module",
+    title="Serve Module",
+    description="Check if serve module is available",
+    category=CheckCategory.ENVIRONMENT,
+    severity=CheckSeverity.MEDIUM,
+    tags=["serve", "deploy"],
+)
+def check_serve_module(config: DoctorConfig) -> CheckResult:
+    """Check if serve module is available."""
+    try:
+        from praisonai.cli.features.serve import ServeHandler
+        ServeHandler()  # Test instantiation
+        return CheckResult(
+            id="serve_module",
+            title="Serve Module",
+            category=CheckCategory.ENVIRONMENT,
+            status=CheckStatus.PASS,
+            message="Serve module available",
+            metadata={"handler": "ServeHandler"},
+        )
+    except ImportError as e:
+        return CheckResult(
+            id="serve_module",
+            title="Serve Module",
+            category=CheckCategory.ENVIRONMENT,
+            status=CheckStatus.WARN,
+            message=f"Serve module not available: {e}",
+            remediation="pip install praisonai[serve]",
+        )
+    except Exception as e:
+        return CheckResult(
+            id="serve_module",
+            title="Serve Module",
+            category=CheckCategory.ENVIRONMENT,
+            status=CheckStatus.ERROR,
+            message=f"Error checking serve module: {e}",
+        )
+
+
+@register_check(
+    id="endpoints_module",
+    title="Endpoints Module",
+    description="Check if endpoints module is available",
+    category=CheckCategory.ENVIRONMENT,
+    severity=CheckSeverity.MEDIUM,
+    tags=["endpoints", "deploy"],
+)
+def check_endpoints_module(config: DoctorConfig) -> CheckResult:
+    """Check if endpoints module is available."""
+    try:
+        from praisonai.endpoints import (
+            DiscoveryDocument,
+            EndpointInfo,
+            ProviderInfo,
+            list_provider_types,
+        )
+        types = list_provider_types()
+        return CheckResult(
+            id="endpoints_module",
+            title="Endpoints Module",
+            category=CheckCategory.ENVIRONMENT,
+            status=CheckStatus.PASS,
+            message=f"Endpoints module available ({len(types)} provider types)",
+            metadata={"provider_types": types},
+        )
+    except ImportError as e:
+        return CheckResult(
+            id="endpoints_module",
+            title="Endpoints Module",
+            category=CheckCategory.ENVIRONMENT,
+            status=CheckStatus.WARN,
+            message=f"Endpoints module not available: {e}",
+            remediation="pip install praisonai[serve]",
+        )
+    except Exception as e:
+        return CheckResult(
+            id="endpoints_module",
+            title="Endpoints Module",
+            category=CheckCategory.ENVIRONMENT,
+            status=CheckStatus.ERROR,
+            message=f"Error checking endpoints module: {e}",
+        )
+
+
+@register_check(
+    id="endpoints_cli",
+    title="Endpoints CLI",
+    description="Check if endpoints CLI handler is available",
+    category=CheckCategory.ENVIRONMENT,
+    severity=CheckSeverity.LOW,
+    tags=["endpoints", "cli"],
+)
+def check_endpoints_cli(config: DoctorConfig) -> CheckResult:
+    """Check if endpoints CLI handler is available."""
+    try:
+        from praisonai.cli.features.endpoints import EndpointsHandler
+        handler = EndpointsHandler()
+        return CheckResult(
+            id="endpoints_cli",
+            title="Endpoints CLI",
+            category=CheckCategory.ENVIRONMENT,
+            status=CheckStatus.PASS,
+            message="Endpoints CLI handler available",
+            metadata={"default_url": handler.DEFAULT_URL},
+        )
+    except ImportError as e:
+        return CheckResult(
+            id="endpoints_cli",
+            title="Endpoints CLI",
+            category=CheckCategory.ENVIRONMENT,
+            status=CheckStatus.WARN,
+            message=f"Endpoints CLI not available: {e}",
+            remediation="pip install praisonai",
+        )
+    except Exception as e:
+        return CheckResult(
+            id="endpoints_cli",
+            title="Endpoints CLI",
+            category=CheckCategory.ENVIRONMENT,
+            status=CheckStatus.ERROR,
+            message=f"Error checking endpoints CLI: {e}",
+        )
+
+
+@register_check(
+    id="server_connectivity",
+    title="Server Connectivity",
+    description="Check if a PraisonAI server is reachable",
+    category=CheckCategory.NETWORK,
+    severity=CheckSeverity.LOW,
+    requires_deep=True,
+    tags=["serve", "network"],
+)
+def check_server_connectivity(config: DoctorConfig) -> CheckResult:
+    """Check if a PraisonAI server is reachable."""
+    import urllib.request
+    import urllib.error
+    
+    url = os.environ.get("PRAISONAI_ENDPOINTS_URL", "http://localhost:8765")
+    health_url = f"{url}/health"
+    
+    try:
+        req = urllib.request.Request(health_url, method="GET")
+        with urllib.request.urlopen(req, timeout=5) as resp:
+            if resp.status == 200:
+                return CheckResult(
+                    id="server_connectivity",
+                    title="Server Connectivity",
+                    category=CheckCategory.NETWORK,
+                    status=CheckStatus.PASS,
+                    message=f"Server reachable at {url}",
+                    metadata={"url": url, "status": resp.status},
+                )
+            else:
+                return CheckResult(
+                    id="server_connectivity",
+                    title="Server Connectivity",
+                    category=CheckCategory.NETWORK,
+                    status=CheckStatus.WARN,
+                    message=f"Server returned status {resp.status}",
+                    metadata={"url": url, "status": resp.status},
+                )
+    except urllib.error.URLError as e:
+        return CheckResult(
+            id="server_connectivity",
+            title="Server Connectivity",
+            category=CheckCategory.NETWORK,
+            status=CheckStatus.SKIP,
+            message=f"No server running at {url}",
+            details=str(e.reason),
+            remediation="Start a server: praisonai serve unified --port 8765",
+        )
+    except Exception as e:
+        return CheckResult(
+            id="server_connectivity",
+            title="Server Connectivity",
+            category=CheckCategory.NETWORK,
+            status=CheckStatus.ERROR,
+            message=f"Error checking server: {e}",
+        )
+
+
+@register_check(
+    id="discovery_endpoint",
+    title="Discovery Endpoint",
+    description="Check if discovery endpoint is available",
+    category=CheckCategory.NETWORK,
+    severity=CheckSeverity.LOW,
+    requires_deep=True,
+    dependencies=["server_connectivity"],
+    tags=["serve", "discovery"],
+)
+def check_discovery_endpoint(config: DoctorConfig) -> CheckResult:
+    """Check if discovery endpoint is available."""
+    import urllib.request
+    import urllib.error
+    import json
+    
+    url = os.environ.get("PRAISONAI_ENDPOINTS_URL", "http://localhost:8765")
+    discovery_url = f"{url}/__praisonai__/discovery"
+    
+    try:
+        req = urllib.request.Request(discovery_url, method="GET")
+        with urllib.request.urlopen(req, timeout=5) as resp:
+            if resp.status == 200:
+                data = json.loads(resp.read().decode())
+                providers = data.get("providers", [])
+                endpoints = data.get("endpoints", [])
+                return CheckResult(
+                    id="discovery_endpoint",
+                    title="Discovery Endpoint",
+                    category=CheckCategory.NETWORK,
+                    status=CheckStatus.PASS,
+                    message=f"Discovery available: {len(providers)} providers, {len(endpoints)} endpoints",
+                    metadata={
+                        "url": discovery_url,
+                        "schema_version": data.get("schema_version"),
+                        "server_name": data.get("server_name"),
+                        "providers": [p.get("type") for p in providers],
+                    },
+                )
+            else:
+                return CheckResult(
+                    id="discovery_endpoint",
+                    title="Discovery Endpoint",
+                    category=CheckCategory.NETWORK,
+                    status=CheckStatus.WARN,
+                    message=f"Discovery returned status {resp.status}",
+                )
+    except urllib.error.URLError:
+        return CheckResult(
+            id="discovery_endpoint",
+            title="Discovery Endpoint",
+            category=CheckCategory.NETWORK,
+            status=CheckStatus.SKIP,
+            message="No server running (skipped)",
+        )
+    except Exception as e:
+        return CheckResult(
+            id="discovery_endpoint",
+            title="Discovery Endpoint",
+            category=CheckCategory.NETWORK,
+            status=CheckStatus.ERROR,
+            message=f"Error checking discovery: {e}",
+        )
+
+
+@register_check(
+    id="a2u_module",
+    title="A2U Module",
+    description="Check if A2U server module is available",
+    category=CheckCategory.ENVIRONMENT,
+    severity=CheckSeverity.LOW,
+    tags=["serve", "a2u"],
+)
+def check_a2u_module(config: DoctorConfig) -> CheckResult:
+    """Check if A2U server module is available."""
+    try:
+        from praisonai.endpoints.a2u_server import (
+            A2UEvent,
+            A2USubscription,
+            A2UEventBus,
+            create_a2u_routes,
+        )
+        return CheckResult(
+            id="a2u_module",
+            title="A2U Module",
+            category=CheckCategory.ENVIRONMENT,
+            status=CheckStatus.PASS,
+            message="A2U server module available",
+        )
+    except ImportError as e:
+        return CheckResult(
+            id="a2u_module",
+            title="A2U Module",
+            category=CheckCategory.ENVIRONMENT,
+            status=CheckStatus.WARN,
+            message=f"A2U module not available: {e}",
+            remediation="pip install praisonai[serve]",
+        )
+    except Exception as e:
+        return CheckResult(
+            id="a2u_module",
+            title="A2U Module",
+            category=CheckCategory.ENVIRONMENT,
+            status=CheckStatus.ERROR,
+            message=f"Error checking A2U module: {e}",
+        )
+
+
+@register_check(
+    id="provider_adapters",
+    title="Provider Adapters",
+    description="Check if all provider adapters are available",
+    category=CheckCategory.ENVIRONMENT,
+    severity=CheckSeverity.LOW,
+    tags=["serve", "providers"],
+)
+def check_provider_adapters(config: DoctorConfig) -> CheckResult:
+    """Check if all provider adapters are available."""
+    try:
+        from praisonai.endpoints.providers import (
+            BaseProvider,
+            RecipeProvider,
+            AgentsAPIProvider,
+            MCPProvider,
+            ToolsMCPProvider,
+            A2AProvider,
+            A2UProvider,
+        )
+        providers = [
+            "BaseProvider",
+            "RecipeProvider",
+            "AgentsAPIProvider",
+            "MCPProvider",
+            "ToolsMCPProvider",
+            "A2AProvider",
+            "A2UProvider",
+        ]
+        return CheckResult(
+            id="provider_adapters",
+            title="Provider Adapters",
+            category=CheckCategory.ENVIRONMENT,
+            status=CheckStatus.PASS,
+            message=f"All {len(providers)} provider adapters available",
+            metadata={"providers": providers},
+        )
+    except ImportError as e:
+        return CheckResult(
+            id="provider_adapters",
+            title="Provider Adapters",
+            category=CheckCategory.ENVIRONMENT,
+            status=CheckStatus.WARN,
+            message=f"Some provider adapters not available: {e}",
+            remediation="pip install praisonai[serve]",
+        )
+    except Exception as e:
+        return CheckResult(
+            id="provider_adapters",
+            title="Provider Adapters",
+            category=CheckCategory.ENVIRONMENT,
+            status=CheckStatus.ERROR,
+            message=f"Error checking provider adapters: {e}",
+        )
+
+
+@register_check(
+    id="fastapi_available",
+    title="FastAPI Available",
+    description="Check if FastAPI is available for serving",
+    category=CheckCategory.ENVIRONMENT,
+    severity=CheckSeverity.MEDIUM,
+    tags=["serve", "dependencies"],
+)
+def check_fastapi_available(config: DoctorConfig) -> CheckResult:
+    """Check if FastAPI is available."""
+    try:
+        import fastapi
+        version = getattr(fastapi, "__version__", "unknown")
+        return CheckResult(
+            id="fastapi_available",
+            title="FastAPI Available",
+            category=CheckCategory.ENVIRONMENT,
+            status=CheckStatus.PASS,
+            message=f"FastAPI {version} available",
+            metadata={"version": version},
+        )
+    except ImportError:
+        return CheckResult(
+            id="fastapi_available",
+            title="FastAPI Available",
+            category=CheckCategory.ENVIRONMENT,
+            status=CheckStatus.WARN,
+            message="FastAPI not installed",
+            remediation="pip install fastapi uvicorn",
+        )
+
+
+@register_check(
+    id="uvicorn_available",
+    title="Uvicorn Available",
+    description="Check if Uvicorn is available for serving",
+    category=CheckCategory.ENVIRONMENT,
+    severity=CheckSeverity.MEDIUM,
+    tags=["serve", "dependencies"],
+)
+def check_uvicorn_available(config: DoctorConfig) -> CheckResult:
+    """Check if Uvicorn is available."""
+    try:
+        import uvicorn
+        version = getattr(uvicorn, "__version__", "unknown")
+        return CheckResult(
+            id="uvicorn_available",
+            title="Uvicorn Available",
+            category=CheckCategory.ENVIRONMENT,
+            status=CheckStatus.PASS,
+            message=f"Uvicorn {version} available",
+            metadata={"version": version},
+        )
+    except ImportError:
+        return CheckResult(
+            id="uvicorn_available",
+            title="Uvicorn Available",
+            category=CheckCategory.ENVIRONMENT,
+            status=CheckStatus.WARN,
+            message="Uvicorn not installed",
+            remediation="pip install uvicorn",
+        )

praisonai/cli/features/doctor/checks/skills_checks.py

@@ -0,0 +1,231 @@
+"""
+Skills checks for the Doctor CLI module.
+
+Validates Agent Skills configurations and manifests.
+"""
+
+from pathlib import Path
+
+from ..models import (
+    CheckCategory,
+    CheckResult,
+    CheckStatus,
+    CheckSeverity,
+    DoctorConfig,
+)
+from ..registry import register_check
+
+
+def _find_skills_dirs() -> list:
+    """Find skills directories."""
+    locations = [
+        Path.cwd() / ".praison" / "skills",
+        Path.cwd() / ".claude" / "skills",
+        Path.home() / ".praison" / "skills",
+        Path.home() / ".config" / "praison" / "skills",
+    ]
+    
+    found = []
+    for loc in locations:
+        if loc.exists() and loc.is_dir():
+            found.append(str(loc))
+    
+    return found
+
+
+def _validate_skill_dir(skill_path: Path) -> dict:
+    """Validate a single skill directory."""
+    result = {"valid": False, "errors": [], "warnings": []}
+    
+    skill_md = skill_path / "SKILL.md"
+    if not skill_md.exists():
+        result["errors"].append("Missing SKILL.md")
+        return result
+    
+    try:
+        content = skill_md.read_text()
+        
+        # Check for YAML frontmatter
+        if not content.startswith("---"):
+            result["errors"].append("SKILL.md missing YAML frontmatter")
+            return result
+        
+        # Parse frontmatter
+        parts = content.split("---", 2)
+        if len(parts) < 3:
+            result["errors"].append("Invalid YAML frontmatter format")
+            return result
+        
+        try:
+            import yaml
+            frontmatter = yaml.safe_load(parts[1])
+            
+            # Check required fields
+            if not frontmatter.get("name"):
+                result["errors"].append("Missing 'name' in frontmatter")
+            if not frontmatter.get("description"):
+                result["errors"].append("Missing 'description' in frontmatter")
+            
+            # Check name format
+            name = frontmatter.get("name", "")
+            if name and not name.replace("-", "").replace("_", "").isalnum():
+                result["warnings"].append(f"Name '{name}' should be alphanumeric with hyphens")
+            
+            if not result["errors"]:
+                result["valid"] = True
+                result["name"] = name
+                result["description"] = frontmatter.get("description", "")[:100]
+            
+        except Exception as e:
+            result["errors"].append(f"YAML parse error: {e}")
+    
+    except Exception as e:
+        result["errors"].append(f"Cannot read SKILL.md: {e}")
+    
+    return result
+
+
+@register_check(
+    id="skills_dirs",
+    title="Skills Directories",
+    description="Check for skills directories",
+    category=CheckCategory.SKILLS,
+    severity=CheckSeverity.INFO,
+)
+def check_skills_dirs(config: DoctorConfig) -> CheckResult:
+    """Check for skills directories."""
+    dirs = _find_skills_dirs()
+    
+    if config.path:
+        # Check specific path
+        path = Path(config.path)
+        if path.exists():
+            dirs = [str(path)]
+        else:
+            return CheckResult(
+                id="skills_dirs",
+                title="Skills Directories",
+                category=CheckCategory.SKILLS,
+                status=CheckStatus.FAIL,
+                message=f"Specified path not found: {config.path}",
+            )
+    
+    if dirs:
+        return CheckResult(
+            id="skills_dirs",
+            title="Skills Directories",
+            category=CheckCategory.SKILLS,
+            status=CheckStatus.PASS,
+            message=f"Found {len(dirs)} skills directory(ies)",
+            details=", ".join(dirs),
+            metadata={"directories": dirs},
+        )
+    else:
+        return CheckResult(
+            id="skills_dirs",
+            title="Skills Directories",
+            category=CheckCategory.SKILLS,
+            status=CheckStatus.SKIP,
+            message="No skills directories found (optional)",
+            details="Create .praison/skills/ to add agent skills",
+        )
+
+
+@register_check(
+    id="skills_valid",
+    title="Skills Validation",
+    description="Validate installed skills",
+    category=CheckCategory.SKILLS,
+    severity=CheckSeverity.MEDIUM,
+    dependencies=["skills_dirs"],
+)
+def check_skills_valid(config: DoctorConfig) -> CheckResult:
+    """Validate installed skills."""
+    dirs = _find_skills_dirs()
+    
+    if config.path:
+        path = Path(config.path)
+        if path.exists():
+            dirs = [str(path)]
+    
+    if not dirs:
+        return CheckResult(
+            id="skills_valid",
+            title="Skills Validation",
+            category=CheckCategory.SKILLS,
+            status=CheckStatus.SKIP,
+            message="No skills to validate",
+        )
+    
+    valid_skills = []
+    invalid_skills = []
+    
+    for dir_path in dirs:
+        skills_dir = Path(dir_path)
+        for skill_path in skills_dir.iterdir():
+            if skill_path.is_dir() and not skill_path.name.startswith("."):
+                result = _validate_skill_dir(skill_path)
+                if result["valid"]:
+                    valid_skills.append(result.get("name", skill_path.name))
+                else:
+                    invalid_skills.append(f"{skill_path.name}: {'; '.join(result['errors'])}")
+    
+    total = len(valid_skills) + len(invalid_skills)
+    
+    if total == 0:
+        return CheckResult(
+            id="skills_valid",
+            title="Skills Validation",
+            category=CheckCategory.SKILLS,
+            status=CheckStatus.SKIP,
+            message="No skills found in directories",
+        )
+    
+    if invalid_skills:
+        return CheckResult(
+            id="skills_valid",
+            title="Skills Validation",
+            category=CheckCategory.SKILLS,
+            status=CheckStatus.WARN,
+            message=f"{len(valid_skills)}/{total} skills valid",
+            details="; ".join(invalid_skills[:3]) + ("..." if len(invalid_skills) > 3 else ""),
+            metadata={"valid": valid_skills, "invalid": invalid_skills},
+        )
+    else:
+        return CheckResult(
+            id="skills_valid",
+            title="Skills Validation",
+            category=CheckCategory.SKILLS,
+            status=CheckStatus.PASS,
+            message=f"All {total} skill(s) valid",
+            metadata={"valid": valid_skills},
+        )
+
+
+@register_check(
+    id="skills_praisonai_integration",
+    title="PraisonAI Skills Integration",
+    description="Check PraisonAI skills module",
+    category=CheckCategory.SKILLS,
+    severity=CheckSeverity.LOW,
+)
+def check_skills_praisonai_integration(config: DoctorConfig) -> CheckResult:
+    """Check PraisonAI skills module."""
+    try:
+        from praisonaiagents.skills import SkillManager
+        return CheckResult(
+            id="skills_praisonai_integration",
+            title="PraisonAI Skills Integration",
+            category=CheckCategory.SKILLS,
+            status=CheckStatus.PASS,
+            message="PraisonAI skills module available",
+        )
+    except ImportError as e:
+        return CheckResult(
+            id="skills_praisonai_integration",
+            title="PraisonAI Skills Integration",
+            category=CheckCategory.SKILLS,
+            status=CheckStatus.SKIP,
+            message="Skills module not available",
+            details=str(e),
+        )