from langchain.agents import create_agent

from llm import gemini_model
from prompts import SUPERVISOR_AGENT_PROMPT
from agents import (
    general_query_tool,
    recommendation_tool,
    purchase_agent_tool,
    complain_handler_tool,
)

supervisor_agent = create_agent(
    gemini_model,
    tools=[
        general_query_tool,
        recommendation_tool,
        purchase_agent_tool,
        complain_handler_tool,
    ],
    system_prompt=SUPERVISOR_AGENT_PROMPT,
)