from dotenv import load_dotenv
from langgraph_supervisor import create_supervisor
from langchain.chat_models import init_chat_model
from langchain_core.messages import SystemMessage

from agents.shopper import shopper_agent
from agents.researcher import craft_research_agent
from agents.mentor import mentor_agent
from agents.prompts import supervisor_prompt

load_dotenv()


model = init_chat_model("gemini-2.0-flash", model_provider="google_genai")

supervisor = create_supervisor(
    model=model,
    agents=[shopper_agent, craft_research_agent, mentor_agent],

    prompt=SystemMessage(content=supervisor_prompt),
    add_handoff_messages=True,
    add_handoff_back_messages=True,
    output_mode="last_message",
).compile()