remove not needed names (#14641)

Author: Sameerlite

tests/proxy_unit_tests/test_google_gemini_proxy_request.py

@@ -359,7 +359,7 @@ async def test_gemini_custom_api_base_proxy_integration():
     vertex_base = VertexBase()
 
     # Test case 1: Custom API base for Gemini
-    custom_api_base = "https://proxy.zapier.com/generativelanguage.googleapis.com/v1beta"
+    custom_api_base = "https://proxy.example.com/generativelanguage.googleapis.com/v1beta"
     model = "gemini-2.5-flash-lite"
     endpoint = "generateContent"
 
@@ -433,7 +433,7 @@ async def test_gemini_proxy_config_with_custom_api_base():
         "litellm_params": {
             "model": "gemini/*",
             "api_key": "dummy-key-for-testing",
-            "api_base": "https://proxy.zapier.com/generativelanguage.googleapis.com/v1beta"
+            "api_base": "https://proxy.example.com/generativelanguage.googleapis.com/v1beta"
         }
     }
 

tests/test_litellm/llms/vertex_ai/test_vertex_llm_base.py

@@ -710,7 +710,7 @@ def test_get_api_base(self, api_base, vertex_location, expected):
         [
             # Test case 1: Gemini with custom API base
             (
-                "https://proxy.zapier.com/generativelanguage.googleapis.com/v1beta",
+                "https://proxy.example.com/generativelanguage.googleapis.com/v1beta",
                 "gemini",
                 "test-api-key",
                 "generateContent",
@@ -719,11 +719,11 @@ def test_get_api_base(self, api_base, vertex_location, expected):
                 "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash-lite:generateContent",
                 "gemini-2.5-flash-lite",
                 "test-api-key",
-                "https://proxy.zapier.com/generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash-lite:generateContent"
+                "https://proxy.example.com/generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash-lite:generateContent"
             ),
             # Test case 2: Gemini with custom API base and streaming
             (
-                "https://proxy.zapier.com/generativelanguage.googleapis.com/v1beta",
+                "https://proxy.example.com/generativelanguage.googleapis.com/v1beta",
                 "gemini",
                 "test-api-key",
                 "generateContent",
@@ -732,7 +732,7 @@ def test_get_api_base(self, api_base, vertex_location, expected):
                 "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash-lite:generateContent",
                 "gemini-2.5-flash-lite",
                 "test-api-key",
-                "https://proxy.zapier.com/generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash-lite:generateContent?alt=sse"
+                "https://proxy.example.com/generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash-lite:generateContent?alt=sse"
             ),
             # Test case 3: Non-Gemini provider with custom API base
             (
@@ -762,7 +762,7 @@ def test_get_api_base(self, api_base, vertex_location, expected):
             ),
             # Test case 5: Gemini without API key (should raise ValueError)
             (
-                "https://proxy.zapier.com/generativelanguage.googleapis.com/v1beta",
+                "https://proxy.example.com/generativelanguage.googleapis.com/v1beta",
                 "gemini",
                 None,
                 "generateContent",
@@ -826,14 +826,14 @@ def test_check_custom_proxy_gemini_url_construction(self):
 
         # Test various Gemini models with custom API base
         test_cases = [
-            ("gemini-2.5-flash-lite", "generateContent", "https://proxy.zapier.com/generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash-lite:generateContent"),
-            ("gemini-2.5-pro", "generateContent", "https://proxy.zapier.com/generativelanguage.googleapis.com/v1beta/models/gemini-2.5-pro:generateContent"),
-            ("gemini-1.5-flash", "streamGenerateContent", "https://proxy.zapier.com/generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:streamGenerateContent"),
+            ("gemini-2.5-flash-lite", "generateContent", "https://proxy.example.com/generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash-lite:generateContent"),
+            ("gemini-2.5-pro", "generateContent", "https://proxy.example.com/generativelanguage.googleapis.com/v1beta/models/gemini-2.5-pro:generateContent"),
+            ("gemini-1.5-flash", "streamGenerateContent", "https://proxy.example.com/generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:streamGenerateContent"),
         ]
 
         for model, endpoint, expected_url in test_cases:
             _, result_url = vertex_base._check_custom_proxy(
-                api_base="https://proxy.zapier.com/generativelanguage.googleapis.com/v1beta",
+                api_base="https://proxy.example.com/generativelanguage.googleapis.com/v1beta",
                 custom_llm_provider="gemini",
                 gemini_api_key="test-api-key",
                 endpoint=endpoint,
@@ -851,7 +851,7 @@ def test_check_custom_proxy_streaming_parameter(self):
 
         # Test with streaming enabled
         _, result_url_streaming = vertex_base._check_custom_proxy(
-            api_base="https://proxy.zapier.com/generativelanguage.googleapis.com/v1beta",
+            api_base="https://proxy.example.com/generativelanguage.googleapis.com/v1beta",
             custom_llm_provider="gemini",
             gemini_api_key="test-api-key",
             endpoint="generateContent",
@@ -861,12 +861,12 @@ def test_check_custom_proxy_streaming_parameter(self):
             model="gemini-2.5-flash-lite",
         )
 
-        expected_streaming_url = "https://proxy.zapier.com/generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash-lite:generateContent?alt=sse"
+        expected_streaming_url = "https://proxy.example.com/generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash-lite:generateContent?alt=sse"
         assert result_url_streaming == expected_streaming_url, f"Expected {expected_streaming_url}, got {result_url_streaming}"
 
         # Test with streaming disabled
         _, result_url_no_streaming = vertex_base._check_custom_proxy(
-            api_base="https://proxy.zapier.com/generativelanguage.googleapis.com/v1beta",
+            api_base="https://proxy.example.com/generativelanguage.googleapis.com/v1beta",
             custom_llm_provider="gemini",
             gemini_api_key="test-api-key",
             endpoint="generateContent",
@@ -876,5 +876,5 @@ def test_check_custom_proxy_streaming_parameter(self):
             model="gemini-2.5-flash-lite",
         )
 
-        expected_no_streaming_url = "https://proxy.zapier.com/generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash-lite:generateContent"
+        expected_no_streaming_url = "https://proxy.example.com/generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash-lite:generateContent"
         assert result_url_no_streaming == expected_no_streaming_url, f"Expected {expected_no_streaming_url}, got {result_url_no_streaming}"