import os
from google.adk.agents import Agent
from google.adk.models.lite_llm import LiteLlm # For multi-model support
import litellm # Import for proxy configuration
os.environ["LITELLM_PROXY_API_KEY"] = ""
os.environ["LITELLM_PROXY_API_BASE"] = ""
litellm.use_litellm_proxy = True
root_agent = Agent(
model=LiteLlm(model="gemini-2.5-pro"),
name="root_agent",
description="You are a helpful assistant that tells the current time in cities.",
instruction="You are a helpful assistant that tells the current time in cities.",
# tools=[get_current_time],
)