From 7841124a277f369252b15034327d5073e866c4bb Mon Sep 17 00:00:00 2001 From: Nicole Cybul Date: Wed, 22 Jan 2025 14:34:01 -0500 Subject: [PATCH] replace contents kw search with content --- ddtrace/contrib/internal/vertexai/_utils.py | 2 +- ddtrace/llmobs/_integrations/vertexai.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ddtrace/contrib/internal/vertexai/_utils.py b/ddtrace/contrib/internal/vertexai/_utils.py index 129b97fd920..2c90e6d6022 100644 --- a/ddtrace/contrib/internal/vertexai/_utils.py +++ b/ddtrace/contrib/internal/vertexai/_utils.py @@ -183,7 +183,7 @@ def tag_request(span, integration, instance, args, kwargs): """ # instance is either a chat session or a model itself model_instance = instance if isinstance(instance, GenerativeModel) else instance._model - contents = get_argument_value(args, kwargs, 0, "contents") + contents = get_argument_value(args, kwargs, 0, "content") history = _get_attr(instance, "_history", []) if history: if isinstance(contents, list): diff --git a/ddtrace/llmobs/_integrations/vertexai.py b/ddtrace/llmobs/_integrations/vertexai.py index 4019268e0c4..eb31763ffcd 100644 --- a/ddtrace/llmobs/_integrations/vertexai.py +++ b/ddtrace/llmobs/_integrations/vertexai.py @@ -45,7 +45,7 @@ def _llmobs_set_tags( metadata = llmobs_get_metadata_google(kwargs, instance) system_instruction = get_system_instructions_from_google_model(instance) - input_contents = get_argument_value(args, kwargs, 0, "contents") + input_contents = get_argument_value(args, kwargs, 0, "content") input_messages = self._extract_input_message(input_contents, history, system_instruction) output_messages = [{"content": ""}]