From bb86619605fe39111beb88edb90e1dd121e8f5fc Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Wed, 25 Jun 2025 22:43:07 -0700
Subject: [PATCH] Revert "Revert "add google generate content to call types""

This reverts commit b2d4462ab5ee40d38c9003e41b3cbf74439a5f12.
---
 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",
 ]