01_basic_agent.py

# /// script
# requires-python = ">=3.11"
# dependencies = ["band-sdk[gemini]"]
#
# [tool.uv.sources]
# band-sdk = { git = "https://github.com/band-ai/band-sdk-python.git" }
# ///
"""
Basic Gemini agent example.

This is the simplest way to create a Band agent with the Gemini SDK.
The adapter handles tool registration and function-calling loops automatically.

Requires:
    - agent_config.yaml with gemini_agent credentials
    - Authentication via one of:
      - GOOGLE_API_KEY or GEMINI_API_KEY environment variable (Gemini Developer API)
      - gcloud CLI with Application Default Credentials (Vertex AI):
          gcloud auth application-default login
          export GOOGLE_GENAI_USE_VERTEXAI=true
          export GOOGLE_CLOUD_PROJECT=your-project-id

Run with:
    uv run examples/gemini/01_basic_agent.py
"""

from __future__ import annotations

import asyncio
import logging
import os
import sys


sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))

from setup_logging import setup_logging

from band import Agent
from band.adapters import GeminiAdapter

setup_logging()
logger = logging.getLogger(__name__)


async def main() -> None:
    # Create adapter with Gemini settings
    # Requires GEMINI_API_KEY environment variable or pass provider_key explicitly
    adapter = GeminiAdapter(
        model="gemini-2.5-flash",
        prompt="You are a helpful assistant. Be concise and friendly.",
    )

    # Create and start agent
    agent = Agent.from_config(
        "gemini_agent",
        adapter=adapter,
    )

    logger.info("Starting Gemini agent...")
    await agent.run()


if __name__ == "__main__":
    asyncio.run(main())