Source code for pitchlense_mcp.analyzers.legal_risk

"""
Legal Risk Analyzer for PitchLense MCP Package.

Analyzes legal and regulatory risks including regulatory environment, compliance issues, and legal disputes.
"""

from typing import List, Dict, Any
from ..core.base import BaseRiskAnalyzer, BaseMCPTool
from ..models.risk_models import RiskLevel
from ..prompts import LEGAL_RISK_PROMPT



[docs]
class LegalRiskAnalyzer(BaseRiskAnalyzer):
    """Legal risk analyzer implementation."""
    
    def __init__(self, llm_client):
        """Initialize the legal risk analyzer."""
        super().__init__(llm_client, "Legal & Regulatory Risks")
        self.risk_indicators = self.get_risk_indicators()
    

[docs]
    def get_risk_indicators(self) -> List[str]:
        """Get the list of legal risk indicators."""
        return [
            "Regulatory Environment Risk",
            "Compliance Risk",
            "Legal Disputes Risk",
            "IP Protection Risk",
            "Regulatory Changes Risk"
        ]

    

[docs]
    def get_analysis_prompt(self) -> str:
        """Get the analysis prompt for legal risks."""
        return LEGAL_RISK_PROMPT





[docs]
class LegalRiskMCPTool(BaseMCPTool):
    """MCP tool for legal risk analysis."""
    
    def __init__(self):
        """Initialize the legal risk MCP tool."""
        super().__init__("Legal Risk Analyzer", "Analyze legal and regulatory risks for startups")
        self.analyzer = LegalRiskAnalyzer(None)  # Will be set when LLM client is available
    

[docs]
    def set_llm_client(self, llm_client):
        """Set the LLM client for the analyzer."""
        self.analyzer.llm_client = llm_client

    

[docs]
    def analyze_legal_risks(self, startup_data: str) -> dict:
        """
        Analyze legal risks for a startup.
        
        Args:
            startup_data: Dictionary containing startup information
        
        Returns:
            JSON response with legal risk analysis
        """
        if not self.validate_startup_data(startup_data):
            return self.create_error_response("Invalid startup data format")
        
        try:
            return self.analyzer.analyze(startup_data)
        except Exception as e:
            return self.create_error_response(f"Analysis failed: {str(e)}")

    

[docs]
    def register_tools(self):
        """Register MCP tools."""
        self.register_tool(self.analyze_legal_risks)