from agno.agent import Agent
from agno.models.google import Gemini
from agno.tools.google_bigquery import GoogleBigQueryTools
# ---------------------------------------------------------------------------
# Create Agent
# ---------------------------------------------------------------------------
agent = Agent(
instructions=[
"You are an expert Big query Writer",
"Always prepend the table name with your_project_id.your_dataset_name when run_sql tool is invoked",
],
tools=[GoogleBigQueryTools(dataset="test_dataset")],
model=Gemini(id="gemini-3-flash-preview", vertexai=True),
)
# ---------------------------------------------------------------------------
# Run Agent
# ---------------------------------------------------------------------------
if __name__ == "__main__":
agent.print_response(
"List the tables in the dataset. Tell me about contents of one of the tables",
markdown=True,
)