preBot / agents.py
wearevenom's picture
Upload 7 files
6f3e11f verified
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"
)