scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 124

Update tests/test_litellm/integrations/gcs_pubsub/test_pub_sub.py

Browse source 
Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
This commit is contained in:
Harshit Jain 2026-03-02 16:43:38 +05:30 • committed by GitHub
parent 905307f9d9
commit 84fab4bf61
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
tests/test_litellm/integrations/gcs_pubsub/test_pub_sub.py
View file

@ -133,19 +133,8 @@ class TestGetVertexPublisherFromUrl:
url = "https://us-central1-aiplatform.googleapis.com/v1/projects/proj/locations/us-central1/publishers/mistralai/models/mistral-large:rawPredict"
assert self._get_publisher(url) == "mistralai"
@pytest.mark.xfail(reason="Bug: _get_vertex_publisher_or_api_spec_from_url does not detect meta publisher")
def test_meta_publisher(self):
"""
Bug: _get_vertex_publisher_or_api_spec_from_url does NOT detect 'meta' publisher.
Only mistralai, anthropic, ai21, and openapi are handled.
"""
url = "https://us-central1-aiplatform.googleapis.com/v1/projects/proj/locations/us-central1/publishers/meta/models/llama-3.1-70b-instruct-maas:rawPredict"
result = self._get_publisher(url)
assert result is not None, (
"Meta publisher not detected. _get_vertex_publisher_or_api_spec_from_url "
"does not handle '/publishers/meta/' URLs."
)
assert result == "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"