CrewAI Integration

from crewai import Agent, Crew, Task
from stackone_ai import StackOneToolSet

def create_hr_crew(account_id: str):
    """
    Create multi-agent crew with tools for a specific account.

    In production, account_id comes from:
    - User/tenant context
    - Authentication middleware
    - Database lookup
    """
    # Initialize toolset
    toolset = StackOneToolSet()

    # Fetch tools dynamically for this account
    tools = toolset.fetch_tools(
        account_ids=[account_id]
    ).to_langchain()

    # Optional: Filter by operation type for security
    # read_only_tools = toolset.fetch_tools(
    #     account_ids=[account_id],
    #     actions=["*_list_*", "*_get_*"]  # Only list and get operations
    # ).to_langchain()

    # Create specialized agents
    hr_manager = Agent(
        role="HR Manager",
        goal="Manage employee data and handle HR requests",
        backstory="Expert in human resources with experience in employee management",
        tools=tools,
        verbose=True
    )

    recruiter = Agent(
        role="Recruiter",
        goal="Find and evaluate candidates for open positions",
        backstory="Experienced recruiter with talent acquisition expertise",
        tools=tools,
        verbose=True
    )

    # Create collaborative tasks
    employee_analysis_task = Task(
        description="Analyze current employee distribution by department",
        agent=hr_manager,
        expected_output="Employee distribution report with staffing recommendations"
    )

    # Execute multi-agent crew
    crew = Crew(
        agents=[hr_manager, recruiter],
        tasks=[employee_analysis_task],
        verbose=True
    )

    return crew

# Usage: Get account from user context
account_id = get_current_user_account()  # Your function
crew = create_hr_crew(account_id)
result = crew.kickoff()
print(result)