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

test_get_supports_system_message

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-26 14:26:08 -07:00
parent 72f08bc6ea
commit c38b41f65b
1 changed files with 18 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
tests/litellm/llms/vertex_ai/test_vertex_ai_common_utils.py
View file

@ -41,3 +41,21 @@ async def test_get_vertex_location_from_url():
url = "https://invalid-url.com"
location = get_vertex_location_from_url(url)
assert location is None
@pytest.mark.asyncio
async def test_get_supports_system_message():
"""Test get_supports_system_message with different models"""
from litellm.llms.vertex_ai.common_utils import get_supports_system_message
# fine-tuned vertex gemini models will specifiy they are in the /gemini spec format
result = get_supports_system_message(
model="gemini/1234567890", custom_llm_provider="vertex_ai"
)
assert result == True
# non-fine-tuned vertex gemini models will not specifiy they are in the /gemini spec format
result = get_supports_system_message(
model="random-model-name", custom_llm_provider="vertex_ai"
)
assert result == False