replace contents kw search with content

DataDog · Jan 22, 2025 · 7841124 · 7841124
1 parent 2f52abb
commit 7841124
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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": ""}]