We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Part.mime_type
1 parent 90efab2 commit 819a44eCopy full SHA for 819a44e
vertexai/generative_models/_generative_models.py
@@ -1926,7 +1926,12 @@ def text(self) -> str:
1926
1927
@property
1928
def mime_type(self) -> Optional[str]:
1929
- return self._raw_part.mime_type
+ part_type = self._raw_part._pb.WhichOneof("data")
1930
+ if part_type == "inline_data":
1931
+ return self._raw_part.inline_data.mime_type
1932
+ if part_type == "file_data":
1933
+ return self._raw_part.file_data.mime_type
1934
+ raise AttributeError(f"Part has no mime_type.\nPart:\n{self.to_dict()}")
1935
1936
1937
def inline_data(self) -> gapic_content_types.Blob:
0 commit comments