Skip to content

Commit 76cc416

Browse files
fix(google-genai): Set agent name (#5038)
### Description Cherry-picked off of #5030 #### Issues Ref https://linear.app/getsentry/issue/TET-1293/make-sure-that-agent-name-is-set-on-all-of-its-gen-ai-children #### Reminders - Please add tests to validate your changes, and lint your code using `tox -e linters`. - Add GH Issue ID _&_ Linear ID (if applicable) - PR title should use [conventional commit](https://develop.sentry.dev/engineering-practices/commit-messages/#type) style (`feat:`, `fix:`, `ref:`, `meta:`) - For external contributors: [CONTRIBUTING.md](https://github.com/getsentry/sentry-python/blob/master/CONTRIBUTING.md), [Sentry SDK development docs](https://develop.sentry.dev/sdk/), [Discord community](https://discord.gg/Ww9hbqr) --------- Co-authored-by: Fabian Schindler <[email protected]>
1 parent b3b2eb6 commit 76cc416

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

sentry_sdk/integrations/google_genai/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,7 @@ def new_generate_content_stream(self, *args, **kwargs):
9292
chat_span.set_data(SPANDATA.GEN_AI_REQUEST_MODEL, model_name)
9393
set_span_data_for_request(chat_span, integration, model_name, contents, kwargs)
9494
chat_span.set_data(SPANDATA.GEN_AI_RESPONSE_STREAMING, True)
95+
chat_span.set_data(SPANDATA.GEN_AI_AGENT_NAME, model_name)
9596

9697
try:
9798
stream = f(self, *args, **kwargs)
@@ -165,6 +166,7 @@ async def new_async_generate_content_stream(self, *args, **kwargs):
165166
chat_span.set_data(SPANDATA.GEN_AI_REQUEST_MODEL, model_name)
166167
set_span_data_for_request(chat_span, integration, model_name, contents, kwargs)
167168
chat_span.set_data(SPANDATA.GEN_AI_RESPONSE_STREAMING, True)
169+
chat_span.set_data(SPANDATA.GEN_AI_AGENT_NAME, model_name)
168170

169171
try:
170172
stream = await f(self, *args, **kwargs)
@@ -233,6 +235,7 @@ def new_generate_content(self, *args, **kwargs):
233235
chat_span.set_data(SPANDATA.GEN_AI_OPERATION_NAME, "chat")
234236
chat_span.set_data(SPANDATA.GEN_AI_SYSTEM, GEN_AI_SYSTEM)
235237
chat_span.set_data(SPANDATA.GEN_AI_REQUEST_MODEL, model_name)
238+
chat_span.set_data(SPANDATA.GEN_AI_AGENT_NAME, model_name)
236239
set_span_data_for_request(
237240
chat_span, integration, model_name, contents, kwargs
238241
)

0 commit comments

Comments
 (0)