@@ -80,8 +80,16 @@ def file(
8080 cls , file : types .FileWithUri | types .FileWithBytes
8181 ) -> a2a_pb2 .FilePart :
8282 if isinstance (file , types .FileWithUri ):
83- return a2a_pb2 .FilePart (file_with_uri = file .uri )
84- return a2a_pb2 .FilePart (file_with_bytes = file .bytes .encode ('utf-8' ))
83+ return a2a_pb2 .FilePart (
84+ file_with_uri = file .uri ,
85+ mime_type = file .mime_type ,
86+ name = file .name
87+ )
88+ return a2a_pb2 .FilePart (
89+ file_with_bytes = file .bytes .encode ('utf-8' ),
90+ mime_type = file .mime_type ,
91+ name = file .name ,
92+ )
8593
8694 @classmethod
8795 def task (cls , task : types .Task ) -> a2a_pb2 .Task :
@@ -501,8 +509,16 @@ def file(
501509 cls , file : a2a_pb2 .FilePart
502510 ) -> types .FileWithUri | types .FileWithBytes :
503511 if file .HasField ('file_with_uri' ):
504- return types .FileWithUri (uri = file .file_with_uri )
505- return types .FileWithBytes (bytes = file .file_with_bytes .decode ('utf-8' ))
512+ return types .FileWithUri (
513+ uri = file .file_with_uri ,
514+ mime_type = file .mime_type ,
515+ name = file .name ,
516+ )
517+ return types .FileWithBytes (
518+ bytes = file .file_with_bytes .decode ('utf-8' ),
519+ mime_type = file .mime_type ,
520+ name = file .name ,
521+ )
506522
507523 @classmethod
508524 def task_or_message (
0 commit comments