Skip to content

Commit 3ad237f

Browse files
authored
fix: google gemini insturmentation (#3055)
1 parent 622d1e4 commit 3ad237f

File tree

2 files changed

+8
-8
lines changed
  • packages
    • opentelemetry-instrumentation-google-generativeai/opentelemetry/instrumentation/google_generativeai
    • opentelemetry-instrumentation-vertexai/opentelemetry/instrumentation/vertexai

2 files changed

+8
-8
lines changed

packages/opentelemetry-instrumentation-google-generativeai/opentelemetry/instrumentation/google_generativeai/__init__.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -287,7 +287,7 @@ async def _awrap(tracer, to_wrap, wrapped, instance, args, kwargs):
287287
name,
288288
kind=SpanKind.CLIENT,
289289
attributes={
290-
SpanAttributes.LLM_SYSTEM: "Gemini",
290+
SpanAttributes.LLM_SYSTEM: "Google",
291291
SpanAttributes.LLM_REQUEST_TYPE: LLMRequestTypeValues.COMPLETION.value,
292292
},
293293
)
@@ -333,7 +333,7 @@ def _wrap(tracer, to_wrap, wrapped, instance, args, kwargs):
333333
name,
334334
kind=SpanKind.CLIENT,
335335
attributes={
336-
SpanAttributes.LLM_SYSTEM: "Gemini",
336+
SpanAttributes.LLM_SYSTEM: "Google",
337337
SpanAttributes.LLM_REQUEST_TYPE: LLMRequestTypeValues.COMPLETION.value,
338338
},
339339
)
@@ -362,17 +362,17 @@ def __init__(self, exception_logger=None):
362362
Config.exception_logger = exception_logger
363363

364364
def instrumentation_dependencies(self) -> Collection[str]:
365-
if is_package_installed("google-genai"):
365+
if is_package_installed("google.genai"):
366366
return ("google-genai >= 0.1.0",)
367-
elif is_package_installed("google-generativeai"):
367+
elif is_package_installed("google.generativeai"):
368368
return ["google-generativeai >= 0.5.0"]
369369
else:
370370
return []
371371

372372
def _wrapped_methods(self):
373-
if is_package_installed("google-genai"):
373+
if is_package_installed("google.genai"):
374374
return WRAPPED_METHODS
375-
elif is_package_installed("google-generativeai"):
375+
elif is_package_installed("google.generativeai"):
376376
return LEGACY_WRAPPED_METHODS
377377
else:
378378
return []

packages/opentelemetry-instrumentation-vertexai/opentelemetry/instrumentation/vertexai/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -283,7 +283,7 @@ async def _awrap(tracer, to_wrap, wrapped, instance, args, kwargs):
283283
name,
284284
kind=SpanKind.CLIENT,
285285
attributes={
286-
SpanAttributes.LLM_SYSTEM: "VertexAI",
286+
SpanAttributes.LLM_SYSTEM: "Google",
287287
SpanAttributes.LLM_REQUEST_TYPE: LLMRequestTypeValues.COMPLETION.value,
288288
},
289289
)
@@ -323,7 +323,7 @@ def _wrap(tracer, to_wrap, wrapped, instance, args, kwargs):
323323
name,
324324
kind=SpanKind.CLIENT,
325325
attributes={
326-
SpanAttributes.LLM_SYSTEM: "VertexAI",
326+
SpanAttributes.LLM_SYSTEM: "Google",
327327
SpanAttributes.LLM_REQUEST_TYPE: LLMRequestTypeValues.COMPLETION.value,
328328
},
329329
)

0 commit comments

Comments
 (0)