getsentry / getsentry/sentry-python

Instrument `generate_images` / `edit_image` / `upscale_image`

Open
#5,825 1 comment 0 reactions 0 assignees View on GitHub
Improvement Python Spans
Dominant language
Python
Stars
2.2k
Forks
669
Avg merge
1d 1h
Merged PRs (30d)
213

Description

## Problem

Image generation is a significant API surface on the google-genai library that is not instrumented:

* `Models.generate_images` — text-to-image (e.g. Imagen 3)
* `Models.edit_image` — edit with reference images (mask, style, subject, etc.)
* `Models.upscale_image` — upscale x2 or x4
* `Models.recontext_image` — product recontext / virtual try-on (Vertex AI only)
* `Models.segment_image` — image segmentation/masking (Vertex AI only)

No `gen_ai.generate_image` operation exists in OTel semconv yet, so a span operation will need to be defined.

## Files

* `sentry_sdk/integrations/google_genai/__init__.py` — `setup_once()` needs additional patches
* `sentry_sdk/consts.py` — may need new `OP` constant(s)

## Acceptance Criteria

* Image generation/editing calls produce spans with appropriate operation names
* Async variants are also instrumented
* Model name and prompt are captured (prompt respects PII settings)
* Tests verify span creation

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.