fix: LLM - Fixed the InlineContext grounding source

PiperOrigin-RevId: 615887256
googleapis · Mar 14, 2024 · 6f59100 · 6f59100
1 parent f7f53e8
commit 6f59100
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vertexai/language_models/_language_models.py b/vertexai/language_models/_language_models.py
@@ -1036,9 +1036,9 @@ def _to_grounding_source_dict(self) -> Dict[str, Any]:
             "sources": [
                 {
                     "type": self._type,
+                    "inlineContext": self.inline_context,
                 }
             ],
-            "inlineContext": self.inline_context,
         }