ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: move helicone gemini test to tests/litellm/

Browse Source 
Move test to correct directory per PR template requirements.
This commit is contained in:
Chesars 2026-01-17 18:07:40 -03:00
parent 542abc0429
commit cc39c71ac6
2 changed files with 35 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
tests/litellm/integrations/helicone/test_helicone_gemini.py Normal file
View File

@ -0,0 +1,35 @@
"""
Test HeliconeLogger Gemini/Vertex AI support.
Fixes: https://github.com/BerriAI/litellm/issues/19093
"""
import pytest
def test_helicone_gemini_model_in_list():
"""
Test that Gemini models are in the helicone_model_list.
"""
from litellm.integrations.helicone import HeliconeLogger
logger = HeliconeLogger()
# Test that "gemini" is in the model list
assert "gemini" in logger.helicone_model_list, "gemini should be in helicone_model_list"
def test_helicone_gemini_models_recognized():
"""
Test that Gemini models are recognized and not replaced with gpt-3.5-turbo.
"""
from litellm.integrations.helicone import HeliconeLogger
logger = HeliconeLogger()
test_models = ["gemini-1.5-pro", "gemini-2.0-flash", "vertex_ai/gemini-1.5-flash"]
for model in test_models:
is_recognized = any(
accepted_model in model
for accepted_model in logger.helicone_model_list
)
assert is_recognized, f"{model} should be recognized by helicone_model_list"

24
tests/local_testing/test_helicone_integration.py
View File

@ -162,27 +162,3 @@ def test_helicone_removes_otel_span_from_metadata():
assert result_metadata["other_metadata"] == "some_value"
print("✅ Test passed: litellm_parent_otel_span was successfully removed from metadata")
def test_helicone_gemini_model_in_list():
"""
Test that Gemini models are in the helicone_model_list and use the correct endpoint.
Fixes: https://github.com/BerriAI/litellm/issues/19093
"""
from litellm.integrations.helicone import HeliconeLogger
logger = HeliconeLogger()
# Test that "gemini" is in the model list
assert "gemini" in logger.helicone_model_list, "gemini should be in helicone_model_list"
# Test that gemini models are recognized (not replaced with gpt-3.5-turbo)
test_models = ["gemini-1.5-pro", "gemini-2.0-flash", "vertex_ai/gemini-1.5-flash"]
for model in test_models:
is_recognized = any(
accepted_model in model
for accepted_model in logger.helicone_model_list
)
assert is_recognized, f"{model} should be recognized by helicone_model_list"
print("✅ Test passed: Gemini models are properly supported in HeliconeLogger")