mirror of
https://github.com/tiennm99/litellm.git
synced 2026-06-25 05:07:03 +00:00
Ishaan Jaff
35e46784d3
* init litellm google gen ai methods * feat init structure of functions for generate content * add init * add BaseGoogleGenAIGenerateContentConfig * add generate_content_handler * add get_provider_google_genai_generate_content_config * fixes for generate content * add get_vertex_ai_project etc to base * use VertexBase * fixes for BaseGoogleGenAIGenerateContentConfig * working validate env for google gemini * feat - add transform google response * fixes for transform_generate_content_request * fix get_supported_generate_content_optional_params * add BaseGoogleGenAITest * working e2e test * fixes init config * use correct types * fix test for google gen ai * fix types * add sync_get_auth_token_and_url * fixes for transform * add llm http handler for google * working non-streaming google endpoints * add BaseGoogleGenAIGenerateContentStreamingIterator * add GoogleGenAIGenerateContentStreamingIterator * fix working sync stream * fixes for litellm logging obj * working async streaming * add google gen ai types * fix - required imports * fix readme * fix deps * fix deps * fix ruff code QA checks * fix linting * fixes TYPE_CHECKING * fixes for typing
10 lines
252 B
Python
10 lines
252 B
Python
from base_google_test import BaseGoogleGenAITest
|
|
|
|
class TestGoogleGenAIStudio(BaseGoogleGenAITest):
|
|
"""Test Google GenAI Studio"""
|
|
|
|
@property
|
|
def model_config(self):
|
|
return {
|
|
"model": "gemini/gemini-1.5-flash",
|
|
} |