mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-06 02:48:13 +00:00
99655b6f86
* test: move key-gated tests/test_litellm SDK tests into tests/llm_translation and drop empty folders * test: make token counter and health check unit tests run offline * ci: point unit shards, rust path filter, Makefile and docs at tests/unit * docs: fix stale test_litellm run paths in moved llm_translation tests * fix: correct databricks e2e sys.path depth and contributing example path --------- Co-authored-by: yuneng <yuneng@berri.ai> Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
23 lines
614 B
Python
23 lines
614 B
Python
"""
|
|
Tests for Gemini Interactions API.
|
|
|
|
Inherits from BaseInteractionsTest to run the same test suite against Gemini.
|
|
"""
|
|
|
|
import os
|
|
|
|
from tests.llm_translation.interactions.base_interactions_test import (
|
|
BaseInteractionsTest,
|
|
)
|
|
|
|
|
|
class TestGeminiInteractions(BaseInteractionsTest):
|
|
"""Test Gemini Interactions API using the base test suite."""
|
|
|
|
def get_model(self) -> str:
|
|
"""Return the Gemini model string."""
|
|
return "gemini/gemini-2.5-flash"
|
|
|
|
def get_api_key(self) -> str:
|
|
"""Return the Gemini API key from environment."""
|
|
return os.getenv("GEMINI_API_KEY", "")
|