Skip to content

feat(generate_image_gemini): add Google Gemini image generation tool #328

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
rjesh-git wants to merge 1 commit into
base: main
Choose a base branch
from
Open

feat(generate_image_gemini): add Google Gemini image generation tool #328

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
45 changes: 45 additions & 0 deletions README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -122,6 +122,7 @@ Below is a comprehensive table of all available tools, how to use them with an a
| memory | `agent.tool.memory(action="retrieve", query="product features")` | Store, retrieve, list, and manage documents in Amazon Bedrock Knowledge Bases with configurable parameters via environment variables |
| environment | `agent.tool.environment(action="list", prefix="AWS_")` | Managing environment variables, configuration management |
| generate_image_stability | `agent.tool.generate_image_stability(prompt="A tranquil pool")` | Creating images using Stability AI models |
| generate_image_gemini | `agent.tool.generate_image_gemini(prompt="A robot holding a red skateboard")` | Generate high-quality images using Google's Gemini Imagen models (Imagen 3 and Imagen 4) |
| generate_image | `agent.tool.generate_image(prompt="A sunset over mountains")` | Creating AI-generated images for various applications |
| image_reader | `agent.tool.image_reader(image_path="path/to/image.jpg")` | Processing and reading image files for AI analysis |
| journal | `agent.tool.journal(action="write", content="Today's progress notes")` | Creating structured logs, maintaining documentation |
Expand Down Expand Up @@ -561,6 +562,37 @@ result = agent.tool.batch(
)
```

### Image Generation with Google Gemini

```python
import os
from strands import Agent
from strands_tools import generate_image_gemini

# Set your API key as environment variable
os.environ['GOOGLE_API_KEY'] = 'your-api-key-here'

# Create agent with the tool
agent = Agent(tools=[generate_image_gemini])

# Basic usage with default parameters
agent.tool.generate_image_gemini(prompt="A robot holding a red skateboard")

# Advanced usage with custom parameters (Imagen 4 Preview - highest quality)
agent.tool.generate_image_gemini(
prompt="A futuristic city with flying cars at sunset",
model_id="gemini-3-pro-image-preview",
aspect_ratio="16:9"
)

# Using Imagen 3 Fast model for quick generation
agent.tool.generate_image_gemini(
prompt="A serene mountain landscape",
model_id="gemini-2.5-flash-image",
aspect_ratio="4:3"
)
```

### Video Tools

```python
Expand Down Expand Up @@ -1184,6 +1216,19 @@ The Mem0 Memory Tool supports three different backend configurations:
|----------------------|-------------|---------|
| RETRIEVE_ENABLE_METADATA_DEFAULT | Default setting for enabling metadata in retrieve tool responses | false |

#### Image Generation with Google Gemini

| Environment Variable | Description | Default |
|----------------------|-------------|---------|
| GOOGLE_API_KEY | Google Gemini API key (required for generate_image_gemini tool) | None |
| GEMINI_MODEL_ID | Default Gemini model to use for image generation | gemini-3-pro-image-preview |

**Supported Models:**
- `gemini-3-pro-image-preview` - Imagen 4 Preview (default, highest quality)
- `gemini-2.5-flash-image` - Imagen 3 Fast (quick generation)

**Note**: Visit https://ai.google.dev/gemini-api to create a free account and API key.

#### Video Tools

| Environment Variable | Description | Default |
Expand Down
7 changes: 5 additions & 2 deletions pyproject.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -117,9 +117,12 @@ elasticsearch_memory = [
mongodb_memory = [
"pymongo>=4.0.0,<5.0.0",
]
generate_image_gemini = [
"google-genai>=1.0.0,<2.0.0",
]

[tool.hatch.envs.hatch-static-analysis]
features = ["mem0_memory", "local_chromium_browser", "agent_core_browser", "agent_core_code_interpreter", "a2a_client", "diagram", "rss", "use_computer", "twelvelabs", "elasticsearch_memory", "mongodb_memory"]
features = ["mem0_memory", "local_chromium_browser", "agent_core_browser", "agent_core_code_interpreter", "a2a_client", "diagram", "rss", "use_computer", "twelvelabs", "elasticsearch_memory", "mongodb_memory", "generate_image_gemini"]
dependencies = [
"strands-agents>=1.0.0",
"mypy>=0.981,<1.0.0",
Expand All @@ -138,7 +141,7 @@ lint-check = [
lint-fix = ["ruff check --fix"]

[tool.hatch.envs.hatch-test]
features = ["mem0_memory", "local_chromium_browser", "agent_core_browser", "agent_core_code_interpreter", "a2a_client", "diagram", "rss", "use_computer", "twelvelabs", "elasticsearch_memory", "mongodb_memory"]
features = ["mem0_memory", "local_chromium_browser", "agent_core_browser", "agent_core_code_interpreter", "a2a_client", "diagram", "rss", "use_computer", "twelvelabs", "elasticsearch_memory", "mongodb_memory", "generate_image_gemini"]
extra-dependencies = [
"moto>=5.1.0,<6.0.0",
"pytest>=8.0.0,<9.0.0",
Expand Down
Loading
Loading