from google.adk.agents import LlmAgent,Agent
from google.adk.tools import google_search
from constants import INSTRUCTIONS_ABOUT_COLLEGE, INSTRUCTIONS_COLLEGE_PREDICTOR, INSTRUCTIONS_COLLEGE_SENIOR, INSTRUCTIONS_COORDINATOR
from tools import db_tool, tavily_tool, predictor_tool, mentor_tool

about_college = LlmAgent(
    name="about_college_agent",
    description="An agent to provide information about the college.",
    instruction = INSTRUCTIONS_ABOUT_COLLEGE,
    tools=[db_tool, tavily_tool],
    model="gemini-2.0-flash"
)

college_predictor = LlmAgent(
    name="college_predictor_agent",
    description="An agent to predict college outcomes.",
    instruction = INSTRUCTIONS_COLLEGE_PREDICTOR,
    tools=[predictor_tool],
    model="gemini-2.0-flash"
)

college_senior = LlmAgent(
    name="college_senior_agent",
    description="An agent to assist college seniors.",
    instruction = INSTRUCTIONS_COLLEGE_SENIOR,
    tools=[mentor_tool],
    model="gemini-2.0-flash"
)


coordinator = Agent(
    name="coordinator_agent",
    description="An agent to coordinate with students in college counselling.",
    instruction=INSTRUCTIONS_COORDINATOR,
    sub_agents=[about_college, college_predictor, college_senior],
    model="gemini-2.0-flash"
)