fix: LVM - Fixed VideoSegmentConfig in Multimodal Embeddings API call

PiperOrigin-RevId: 609072865
googleapis · Feb 21, 2024 · 665e78c · 665e78c
1 parent 81f6a25
commit 665e78c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/vertexai/vision_models/_vision_models.py b/vertexai/vision_models/_vision_models.py
@@ -918,7 +918,11 @@ def get_embeddings(
                 }  # pylint: disable=protected-access
 
             if video_segment_config:
-                instance["video"]["videoSegmentConfig"] = video_segment_config
+                instance["video"]["videoSegmentConfig"] = {
+                    "startOffsetSec": video_segment_config.start_offset_sec,
+                    "endOffsetSec": video_segment_config.end_offset_sec,
+                    "intervalSec": video_segment_config.interval_sec,
+                }
 
         if contextual_text:
             instance["text"] = contextual_text