The Fine Print

from crewai import Agent, Task, Crew
from langchain_openai import ChatOpenAI
llm = ChatOpenAI(model="gpt-4-turbo", temperature=0)
hr_agent = Agent(
    role='HR Identity Architect',
    goal='Map the user current title and department.',
    backstory='HR systems expert who tracks employee lifecycles.',
    verbose=True,
    llm=llm
)
usage_agent = Agent(
    role='Access Telemetry Analyst',
    goal='Analyze audit logs to see if the user has used the assigned permission recently.',
    backstory='Data miner focused on application telemetry.',
    verbose=True,
    llm=llm
)
peer_agent = Agent(
    role='Peer Analytics Engine',
    goal='Determine if this permission is standard for the user peer group, and recommend Revoke or Keep.',
    backstory='Statistical analyst who identifies outliers in access entitlements.',
    verbose=True,
    llm=llm
)
user_data = """
Employee: Sarah Connor
Current Role: Marketing Manager (Transferred from IT Support 6 months ago)
Permission Under Review: AWS_Prod_DB_Admin
Usage: 0 logins in the last 120 days.
Peer Group: Marketing Managers (0% of peers have this access).
"""
hr_task = Task(
    description=f'Analyze the employee role history: {user_data}',
    expected_output='Summary of role changes and current department.',
    agent=hr_agent
)
usage_task = Task(
    description=f'Analyze the telemetry for the specific permission: {user_data}',
    expected_output='Usage frequency report.',
    agent=usage_agent
)
decision_task = Task(
    description='Compare the role, usage, and peer group data. Make a definitive recommendation to the manager to KEEP or REVOKE the AWS_Prod_DB_Admin permission.',
    expected_output='Final Revoke/Keep recommendation with justification.',
    agent=peer_agent,
    context=[hr_task, usage_task]
)
uar_swarm = Crew(
    agents=[hr_agent, usage_agent, peer_agent],
    tasks=[hr_task, usage_task, decision_task]
)
if __name__ == "__main__":
    print("Initiating UAR Swarm...")
    uar_recommendation = uar_swarm.kickoff()
    print("\n--- Access Review Recommendation ---\n")
    print(uar_recommendation)