Update google/generativeai/types/content_types.py

shilpakancharla · MarkDaoust · web-flow · commit 250fb41f75dc · 2024-09-16T14:27:05.000-07:00
Co-authored-by: Mark Daoust &lt;markdaoust@google.com&gt;
diff --git a/google/generativeai/types/content_types.py b/google/generativeai/types/content_types.py
@@ -73,7 +73,7 @@
 
 Mode = protos.DynamicRetrievalConfig.Mode
 
-ModeOptions = Union[str, str, Mode]
+ModeOptions = Union[int, str, Mode]
 
 _MODE: dict[ModeOptions, Mode] = {
     Mode.MODE_UNSPECIFIED: Mode.MODE_UNSPECIFIED,
