fix: GenAI - Fixed the Part.mime_type property

PiperOrigin-RevId: 628436837
googleapis · Apr 26, 2024 · 819a44e · 819a44e
1 parent 90efab2
commit 819a44e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/vertexai/generative_models/_generative_models.py b/vertexai/generative_models/_generative_models.py
@@ -1926,7 +1926,12 @@ def text(self) -> str:
 
     @property
     def mime_type(self) -> Optional[str]:
-        return self._raw_part.mime_type
+        part_type = self._raw_part._pb.WhichOneof("data")
+        if part_type == "inline_data":
+            return self._raw_part.inline_data.mime_type
+        if part_type == "file_data":
+            return self._raw_part.file_data.mime_type
+        raise AttributeError(f"Part has no mime_type.\nPart:\n{self.to_dict()}")
 
     @property
     def inline_data(self) -> gapic_content_types.Blob: