From 7d23106fcfaf304045a3ce976b50d0783f807215 Mon Sep 17 00:00:00 2001
From: Chesars <cesarponce19544@gmail.com>
Date: Mon, 2 Mar 2026 19:15:37 -0300
Subject: [PATCH 1/2] fix(helicone): correct provider URL for Vertex AI Gemini
 models

Reorder elif branches so is_vertex_ai is checked before "gemini" in model.
Previously, Vertex AI Gemini models (e.g. vertex_ai/gemini-2.5-flash) matched
the "gemini" substring check first and were logged with the Google AI Studio
URL instead of the Vertex AI URL.
---
 litellm/integrations/helicone.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/litellm/integrations/helicone.py b/litellm/integrations/helicone.py
index c77a1b2564..51e6699c5f 100644
--- a/litellm/integrations/helicone.py
+++ b/litellm/integrations/helicone.py
@@ -167,12 +167,12 @@ class HeliconeLogger:
             if "claude" in model and not is_vertex_ai:
                 url = f"{self.api_base}/anthropic/v1/log"
                 provider_url = "https://api.anthropic.com/v1/messages"
-            elif "gemini" in model:
-                url = f"{self.api_base}/custom/v1/log"
-                provider_url = "https://generativelanguage.googleapis.com/v1beta"
             elif is_vertex_ai:
                 url = f"{self.api_base}/custom/v1/log"
                 provider_url = "https://aiplatform.googleapis.com/v1"
+            elif "gemini" in model:
+                url = f"{self.api_base}/custom/v1/log"
+                provider_url = "https://generativelanguage.googleapis.com/v1beta"
             headers = {
                 "Authorization": f"Bearer {self.key}",
                 "Content-Type": "application/json",

From 4a88d854462be6cc1b8d7316a9df26ac1e65a487 Mon Sep 17 00:00:00 2001
From: Chesars <cesarponce19544@gmail.com>
Date: Tue, 3 Mar 2026 15:20:51 -0300
Subject: [PATCH 2/2] test: add provider_url routing test for vertex_ai/gemini
 models

Verifies that vertex_ai gemini models route to
aiplatform.googleapis.com instead of
generativelanguage.googleapis.com, preventing
regressions if the branch ordering changes.
---
 .../helicone/test_helicone_gemini.py          | 71 +++++++++++++++++++
 1 file changed, 71 insertions(+)

diff --git a/tests/litellm/integrations/helicone/test_helicone_gemini.py b/tests/litellm/integrations/helicone/test_helicone_gemini.py
index f42a701613..67c4515c1e 100644
--- a/tests/litellm/integrations/helicone/test_helicone_gemini.py
+++ b/tests/litellm/integrations/helicone/test_helicone_gemini.py
@@ -62,3 +62,74 @@ def test_helicone_vertex_ai_via_custom_llm_provider():
     for model, custom_llm_provider in test_cases:
         is_vertex_ai = custom_llm_provider == "vertex_ai" or model.startswith("vertex_ai/")
         assert is_vertex_ai, f"{model} with custom_llm_provider={custom_llm_provider} should be recognized as vertex_ai"
+
+
+def test_helicone_vertex_gemini_gets_vertex_provider_url():
+    """
+    Test that vertex_ai/gemini-* models route to aiplatform.googleapis.com,
+    not generativelanguage.googleapis.com.
+
+    This verifies the branch ordering fix: is_vertex_ai must be checked
+    before "gemini" in model, otherwise vertex gemini models get the wrong
+    provider_url.
+    """
+    from unittest.mock import MagicMock, patch
+
+    from litellm.integrations.helicone import HeliconeLogger
+
+    logger = HeliconeLogger()
+
+    captured = {}
+
+    def mock_post(url, **kwargs):
+        captured["url"] = url
+        captured["data"] = kwargs.get("json", {})
+        mock_resp = MagicMock()
+        mock_resp.status_code = 200
+        return mock_resp
+
+    test_cases = [
+        # (model, custom_llm_provider, expected_provider_url)
+        (
+            "vertex_ai/gemini-1.5-pro",
+            "",
+            "https://aiplatform.googleapis.com/v1",
+        ),
+        (
+            "gemini-2.0-flash",
+            "vertex_ai",
+            "https://aiplatform.googleapis.com/v1",
+        ),
+        (
+            "gemini-1.5-flash",
+            "",
+            "https://generativelanguage.googleapis.com/v1beta",
+        ),
+    ]
+
+    for model, custom_llm_provider, expected_url in test_cases:
+        captured.clear()
+        mock_client = MagicMock()
+        mock_client.post = mock_post
+        with patch("litellm.module_level_client", mock_client):
+            logger.log_success(
+                model=model,
+                messages=[{"role": "user", "content": "test"}],
+                response_obj={"choices": [{"message": {"content": "hi"}}]},
+                start_time=MagicMock(),
+                end_time=MagicMock(),
+                print_verbose=lambda *args, **kwargs: None,
+                kwargs={
+                    "litellm_params": {
+                        "custom_llm_provider": custom_llm_provider,
+                        "metadata": {},
+                    },
+                },
+            )
+
+        assert "data" in captured, f"No request captured for {model}"
+        actual_url = captured["data"]["providerRequest"]["url"]
+        assert actual_url == expected_url, (
+            f"Model {model} (provider={custom_llm_provider!r}): "
+            f"expected provider_url={expected_url}, got {actual_url}"
+        )