mirror of
https://github.com/tiennm99/litellm.git
synced 2026-07-17 18:17:52 +00:00
24 lines
611 B
Python
24 lines
611 B
Python
"""
|
|
Tests for Gemini Interactions API.
|
|
|
|
Inherits from BaseInteractionsTest to run the same test suite against Gemini.
|
|
"""
|
|
|
|
import os
|
|
|
|
from tests.test_litellm.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", "")
|