Skip to content

Commit 9f2efed

Browse files
committed
format
Change-Id: I4133647fbea3b6b995ed6edf56bd2a243c2131af
1 parent 4adbeeb commit 9f2efed

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

google/generativeai/client.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ def __init__(self, *args, **kwargs):
5151
self._discovery_api = None
5252
super().__init__(*args, **kwargs)
5353

54-
def _setup_discovery_api(self, metadata:dict|Sequence[tuple[str, str]]=()):
54+
def _setup_discovery_api(self, metadata: dict | Sequence[tuple[str, str]] = ()):
5555
api_key = self._client_options.api_key
5656
if api_key is None:
5757
raise ValueError(
@@ -62,7 +62,7 @@ def _setup_discovery_api(self, metadata:dict|Sequence[tuple[str, str]]=()):
6262
http=httplib2.Http(),
6363
postproc=lambda resp, content: (resp, content),
6464
uri=f"{GENAI_API_DISCOVERY_URL}?version=v1beta&key={api_key}",
65-
headers=dict(metadata)
65+
headers=dict(metadata),
6666
)
6767
response, content = request.execute()
6868
request.http.close()
@@ -80,7 +80,7 @@ def create_file(
8080
name: str | None = None,
8181
display_name: str | None = None,
8282
resumable: bool = True,
83-
metadata:Sequence[tuple[str, str]] = ()
83+
metadata: Sequence[tuple[str, str]] = (),
8484
) -> protos.File:
8585
if self._discovery_api is None:
8686
self._setup_discovery_api(metadata)
@@ -95,8 +95,8 @@ def create_file(
9595
filename=path, mimetype=mime_type, resumable=resumable
9696
)
9797
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
100100
result = request.execute()
101101

102102
return self.get_file({"name": result["file"]["name"]})
@@ -235,7 +235,7 @@ def keep(name, f):
235235
if not callable(f):
236236
return False
237237

238-
if 'metadata' not in inspect.signature(f).parameters.keys():
238+
if "metadata" not in inspect.signature(f).parameters.keys():
239239
return False
240240

241241
return True

0 commit comments

Comments
 (0)