@@ -51,7 +51,7 @@ def __init__(self, *args, **kwargs):
51
51
self ._discovery_api = None
52
52
super ().__init__ (* args , ** kwargs )
53
53
54
- def _setup_discovery_api (self , metadata :dict | Sequence [tuple [str , str ]]= ()):
54
+ def _setup_discovery_api (self , metadata : dict | Sequence [tuple [str , str ]] = ()):
55
55
api_key = self ._client_options .api_key
56
56
if api_key is None :
57
57
raise ValueError (
@@ -62,7 +62,7 @@ def _setup_discovery_api(self, metadata:dict|Sequence[tuple[str, str]]=()):
62
62
http = httplib2 .Http (),
63
63
postproc = lambda resp , content : (resp , content ),
64
64
uri = f"{ GENAI_API_DISCOVERY_URL } ?version=v1beta&key={ api_key } " ,
65
- headers = dict (metadata )
65
+ headers = dict (metadata ),
66
66
)
67
67
response , content = request .execute ()
68
68
request .http .close ()
@@ -80,7 +80,7 @@ def create_file(
80
80
name : str | None = None ,
81
81
display_name : str | None = None ,
82
82
resumable : bool = True ,
83
- metadata :Sequence [tuple [str , str ]] = ()
83
+ metadata : Sequence [tuple [str , str ]] = (),
84
84
) -> protos .File :
85
85
if self ._discovery_api is None :
86
86
self ._setup_discovery_api (metadata )
@@ -95,8 +95,8 @@ def create_file(
95
95
filename = path , mimetype = mime_type , resumable = resumable
96
96
)
97
97
request = self ._discovery_api .media ().upload (body = {"file" : file }, media_body = media )
98
- for key ,value in metadata :
99
- request .headers [key ]= value
98
+ for key , value in metadata :
99
+ request .headers [key ] = value
100
100
result = request .execute ()
101
101
102
102
return self .get_file ({"name" : result ["file" ]["name" ]})
@@ -235,7 +235,7 @@ def keep(name, f):
235
235
if not callable (f ):
236
236
return False
237
237
238
- if ' metadata' not in inspect .signature (f ).parameters .keys ():
238
+ if " metadata" not in inspect .signature (f ).parameters .keys ():
239
239
return False
240
240
241
241
return True
0 commit comments