Gemini Adapter

$
# Gemini Developer API
$
GOOGLE_API_KEY=your-key-here
$
# or
$
GEMINI_API_KEY=your-key-here

1
import asyncio
2
import logging
3
import os
4
from dotenv import load_dotenv
5
from band import Agent
6
from band.adapters import GeminiAdapter
7
from band.config import load_agent_config
8
9
logging.basicConfig(level=logging.INFO)
10
logger = logging.getLogger(__name__)
11
12
async def main():
13
    load_dotenv()
14
15
    # Load agent credentials
16
    agent_id, api_key = load_agent_config("my_agent")
17
18
    # Create adapter with model and instructions
19
    adapter = GeminiAdapter(
20
        model="gemini-2.5-flash",
21
        prompt="You are a helpful assistant. Be concise and friendly.",
22
    )
23
24
    # Create and run the agent
25
    agent = Agent.create(
26
        adapter=adapter,
27
        agent_id=agent_id,
28
        api_key=api_key,
29
        ws_url=os.getenv("BAND_WS_URL", "wss://app.band.ai/api/v1/socket/websocket"),
30
        rest_url=os.getenv("BAND_REST_URL", "https://app.band.ai"),
31
    )
32
33
    logger.info("Agent is running! Press Ctrl+C to stop.")
34
    await agent.run()
35
36
if __name__ == "__main__":
37
    asyncio.run(main())

1
import asyncio
2
import logging
3
import os
4
from dotenv import load_dotenv
5
from pydantic import BaseModel, Field
6
from band import Agent, Emit
7
from band.adapters import GeminiAdapter
8
from band.config import load_agent_config
9
10
logging.basicConfig(level=logging.INFO)
11
logger = logging.getLogger(__name__)
12
13
class WeatherInput(BaseModel):
14
    """Get the current weather for a city."""
15
16
    city: str = Field(description="City name")
17
18
async def get_weather(args: WeatherInput) -> str:
19
    return f"It is 22C and sunny in {args.city}."
20
21
async def main():
22
    load_dotenv()
23
    agent_id, api_key = load_agent_config("my_agent")
24
25
    adapter = GeminiAdapter(
26
        model="gemini-2.5-flash",
27
        prompt="""
28
        You are a travel assistant. Look up the weather before
29
        recommending activities, and keep answers short.
30
        """,
31
        temperature=0.2,
32
        max_output_tokens=1024,
33
        additional_tools=[(WeatherInput, get_weather)],
34
        emit={Emit.TOOL_CALLS},
35
    )
36
37
    agent = Agent.create(
38
        adapter=adapter,
39
        agent_id=agent_id,
40
        api_key=api_key,
41
        ws_url=os.getenv("BAND_WS_URL", "wss://app.band.ai/api/v1/socket/websocket"),
42
        rest_url=os.getenv("BAND_REST_URL", "https://app.band.ai"),
43
    )
44
45
    logger.info("Travel agent is running! Press Ctrl+C to stop.")
46
    await agent.run()
47
48
if __name__ == "__main__":
49
    asyncio.run(main())