diff --git a/litellm/types/utils.py b/litellm/types/utils.py index 68d861cbc11..733335ef548 100644 --- a/litellm/types/utils.py +++ b/litellm/types/utils.py @@ -277,6 +277,14 @@ class CallTypes(Enum): aresponses = "aresponses" alist_input_items = "alist_input_items" + ######################################################### + # Google GenAI Native Call Types + ######################################################### + generate_content = "generate_content" + agenerate_content = "agenerate_content" + generate_content_stream = "generate_content_stream" + agenerate_content_stream = "agenerate_content_stream" + CallTypesLiteral = Literal[ "embedding", @@ -304,6 +312,10 @@ CallTypesLiteral = Literal[ "anthropic_messages", "aretrieve_batch", "retrieve_batch", + "generate_content", + "agenerate_content", + "generate_content_stream", + "agenerate_content_stream", ]