mirror of
https://github.com/tiennm99/litellm.git
synced 2026-07-05 15:08:18 +00:00
Ishaan Jaff
14d05fe422
* 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 * add google gemini methods to litellm router * [Feat] Add initial endpoints for using Gemini SDK (gemini-cli) with LiteLLM (#12040) * init with google endpoints * add Depends * feat - add gemini endpoints * google_generate_content * fix init * fixes import * fixes for streaming * fixes for sync/async * working streaming with google gemini cli * add google endpoints to llm api routes * add VertexAIGoogleGenAIConfig * use aiter_bytes * use common request for streaming data * re-use logic for anthropic streaming * add GoogleAIStudioDataGenerator
10 lines
251 B
Python
10 lines
251 B
Python
from base_google_test import BaseGoogleGenAITest
|
|
|
|
class TestVertexAIGenerateContent(BaseGoogleGenAITest):
|
|
"""Test Vertex AI"""
|
|
|
|
@property
|
|
def model_config(self):
|
|
return {
|
|
"model": "vertex_ai/gemini-1.5-flash",
|
|
} |