From fa3d788a2902991cf2e9764338cb037c3a269f15 Mon Sep 17 00:00:00 2001 From: Harshit Jain <48647625+Harshit28j@users.noreply.github.com> Date: Mon, 2 Mar 2026 17:01:11 +0530 Subject: [PATCH] Update tests/test_litellm/integrations/gcs_pubsub/test_pub_sub.py Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com> --- tests/test_litellm/integrations/gcs_pubsub/test_pub_sub.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/test_litellm/integrations/gcs_pubsub/test_pub_sub.py b/tests/test_litellm/integrations/gcs_pubsub/test_pub_sub.py index 7b54c56af55..950c9b469be 100644 --- a/tests/test_litellm/integrations/gcs_pubsub/test_pub_sub.py +++ b/tests/test_litellm/integrations/gcs_pubsub/test_pub_sub.py @@ -135,6 +135,9 @@ class TestGetVertexPublisherFromUrl: @pytest.mark.xfail(reason="Bug: _get_vertex_publisher_or_api_spec_from_url does not detect meta publisher") def test_meta_publisher(self): + url = "https://us-central1-aiplatform.googleapis.com/v1/projects/proj/locations/us-central1/publishers/meta/models/llama-3.1-70b-instruct-maas:rawPredict" + assert self._get_publisher(url) == "meta" + def test_google_publisher_returns_none(self): """Google publisher models go through generateContent, not rawPredict.""" url = "https://us-central1-aiplatform.googleapis.com/v1/projects/proj/locations/us-central1/publishers/google/models/gemini-2.5-flash:generateContent"