File size: 1,298 Bytes
6f3e11f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
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"
)