Skip to content

Commit 7bf8083

Browse files
committed
Format updated generation_types and test using black.
1 parent acae3a2 commit 7bf8083

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

google/generativeai/types/generation_types.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -544,7 +544,7 @@ def prompt_feedback(self):
544544
@property
545545
def usage_metadata(self):
546546
return self._result.usage_metadata
547-
547+
548548
@property
549549
def model_version(self):
550550
return self._result.model_version

tests/test_generation.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -493,7 +493,7 @@ def test_join_chunks(self):
493493
prompt_token_count=5
494494
)
495495

496-
chunks[-1].model_version = 'gemini-1.5-flash-002'
496+
chunks[-1].model_version = "gemini-1.5-flash-002"
497497

498498
result = generation_types._join_chunks(chunks)
499499

@@ -511,7 +511,7 @@ def test_join_chunks(self):
511511
],
512512
},
513513
"usage_metadata": {"prompt_token_count": 5},
514-
"model_version": 'gemini-1.5-flash-002'
514+
"model_version": "gemini-1.5-flash-002",
515515
},
516516
)
517517

0 commit comments

Comments
 (0)