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 123

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 17:01:11 +05:30 • committed by GitHub
parent d8f07b75f8
commit fa3d788a29
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -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"