Skip to content

Commit 448486a

Browse files
authored
Allow incomplete GenerationConfigDict
These are all non-required so mark the typed dict as such.
1 parent 948a5a8 commit 448486a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

google/generativeai/types/generation_types.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ class BrokenResponseError(Exception):
7373
pass
7474

7575

76-
class GenerationConfigDict(TypedDict):
76+
class GenerationConfigDict(TypedDict, total=False):
7777
# TODO(markdaoust): Python 3.11+ use `NotRequired`, ref: https://peps.python.org/pep-0655/
7878
candidate_count: int
7979
stop_sequences: Iterable[str]

0 commit comments

Comments
 (0)