From 6f57dde293e50674c8fca0feac6e01c27d9e1c96 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 25 Jun 2025 19:31:02 -0700 Subject: [PATCH] add google generate content to call types --- litellm/types/utils.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) 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", ]