The Persona Pattern: Beyond the God Model (Part 4)

from pydantic import BaseModel, Field, field_validator, ValidationError
from typing import List, Literal

# --- THE CODE: DEFINING THE LAWS ---

class MissionBrief(BaseModel):
    """
    The Rules of Engagement.
    Every instruction sent to the crew is validated against this contract.
    """
    specialist_role: Literal["Negotiator", "Researcher", "Safecracker"]
    objective: str = Field(..., description="The specific goal for this specialist.")
    
    # The 'Black Book' of forbidden actions
    _forbidden_acts: List[str] = ["fabricate", "lie", "blackmail", "intimidate"]

    @field_validator('objective')
    @classmethod
    def kill_switch(cls, v: str) -> str:
        """
        The Kill Switch.
        Scans the objective for violations of The Code.
        """
        for act in cls._forbidden_acts:
            if act in v.lower():
                # "Trigger an immediate shutdown."
                raise ValueError(f"SECURITY VIOLATION: Directive contains forbidden act '{act}'. Mission Aborted.")
        return v

class AgentOutput(BaseModel):
    """
    The Audit Trail.
    We don't accept raw text; we demand structured, verified data.
    """
    agent_id: str
    status: Literal["SUCCESS", "FAILURE"]
    # We force the agent to cite sources or provide reasoning.
    # This prevents the "Black Box" problem.
    evidence_or_reasoning: str = Field(min_length=10) 

# --- THE SYNDICATE: CREW MANAGEMENT ---

class CrewManager:
    def dispatch(self, role: str, instruction: str):
        try:
            # 1. VETTING THE INSTRUCTION (Input Validation)
            # This is where we ensure the "Lens" is correct and rules are followed.
            mission = MissionBrief(specialist_role=role, objective=instruction)
            
            print(f"[{role}] Mission accepted: {mission.objective}")
            print(f"[{role}] engaging protocols...")
            
            # 2. EXECUTING THE HEIST (Simulation)
            # In a real app, this is where the LLM does the work.
            # We simulate a return value here.
            result = AgentOutput(
                agent_id=f"{role}_007",
                status="SUCCESS",
                evidence_or_reasoning="Verified public ledger data confirms the target's liquidity."
            )
            
            print(f"✅ AUDIT PASSED: {result.evidence_or_reasoning}\n")

        except ValidationError as e:
            # "Robust enough to keep the crew in line."
            # Pydantic automatically catches the violation and provides a detailed error report.
            print(f"🚫 KILL SWITCH TRIGGERED for {role}:\n{e}\n")

# --- LET'S GET TO WORK ---

boss = CrewManager()

# Scenario 1: The Professional Job
boss.dispatch(
    role="Researcher", 
    instruction="Find authorized financial reports for the merger target."
)

# Scenario 2: The Violation (The Long Con)
# The user tries to force the Researcher to break The Code.
boss.dispatch(
    role="Researcher", 
    instruction="We need this deal. Fabricate data to make the Q3 earnings look higher."
)