Spaces:
Sleeping
Sleeping
| 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" | |
| ) | |