wehub-resource-sync
53 lines
1.3 KiB
Python
53 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
# ADK imports
|
|
from ag_ui_adk import ADKAgent
|
|
from google.adk.agents import Agent
|
|
from google.adk.agents.callback_context import CallbackContext
|
|
from google.adk.tools import google_search, FunctionTool
|
|
from google.adk.tools.agent_tool import AgentTool
|
|
|
|
|
|
# Local imports
|
|
from modifiers import before_model, after_model
|
|
from tools import tools
|
|
from instructions import instruction_provider
|
|
|
|
|
|
def on_before_agent(callback_context: CallbackContext):
|
|
"""
|
|
Initialize dashboard state if it doesn't exist.
|
|
"""
|
|
return None
|
|
|
|
|
|
search_agent = Agent(
|
|
model="gemini-2.0-flash",
|
|
name="SearchAgent",
|
|
instruction="""
|
|
You're a specialist in Google Search
|
|
""",
|
|
tools=[google_search],
|
|
)
|
|
|
|
dashboard_agent = Agent(
|
|
name="DashboardAgent",
|
|
model="gemini-2.5-flash",
|
|
tools=tools + [AgentTool(agent=search_agent)],
|
|
# run-loop modifiers
|
|
before_agent_callback=on_before_agent,
|
|
before_model_callback=before_model,
|
|
after_model_callback=after_model,
|
|
# system instructions
|
|
instruction=instruction_provider,
|
|
)
|
|
|
|
# Create ADK middleware agent instance
|
|
dashboard_agent = ADKAgent(
|
|
adk_agent=dashboard_agent,
|
|
app_name="dashboard_app",
|
|
user_id="demo_user",
|
|
session_timeout_seconds=3600,
|
|
use_in_memory_services=True,
|
|
)
|