fix: LLM - Fixed the InlineContext grounding source

Ark-kun · copybara-github · commit 6f591001e834 · 2024-03-14T13:35:34.000-07:00
PiperOrigin-RevId: 615887256
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,
         }
 
 

Original file line number	Diff line number	Diff line change
`@@ -1036,9 +1036,9 @@ def _to_grounding_source_dict(self) -> Dict[str, Any]:`
`1036`	`1036`	`"sources": [`
`1037`	`1037`	`{`
`1038`	`1038`	`"type": self._type,`
	`1039`	`+ "inlineContext": self.inline_context,`
`1039`	`1040`	`}`
`1040`	`1041`	`],`
`1041`		`- "inlineContext": self.inline_context,`
`1042`	`1042`	`}`
`1043`	`1043`
`1044`	`1044`