diff --git a/google/generativeai/types/file_types.py b/google/generativeai/types/file_types.py index a2612d47c..d18404871 100644 --- a/google/generativeai/types/file_types.py +++ b/google/generativeai/types/file_types.py @@ -66,6 +66,10 @@ def sha256_hash(self) -> bytes: def uri(self) -> str: return self._proto.uri + @property + def state(self) -> glm.File.State: + return self._proto.state + def delete(self): client = get_default_file_client() client.delete_file(name=self.name)