fix: add VertexAI support for Gemini LLM provider (#3990)#7
Open
fix: add VertexAI support for Gemini LLM provider (#3990)#7
Conversation
Root cause: The GeminiLLM class only supported authentication via GOOGLE_API_KEY, with no way to use VertexAI which authenticates via Application Default Credentials (ADC). Fix: Check the GOOGLE_GENAI_USE_VERTEXAI environment variable (consistent with Google's own libraries). When set to true, initialize the genai.Client with vertexai=True, which uses ADC instead of an API key. Users can configure their project and region via GOOGLE_CLOUD_PROJECT and GOOGLE_CLOUD_LOCATION env vars.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Fixes mem0ai#3990.
Root cause:
GeminiLLMonly supportedGOOGLE_API_KEY, with no way to use VertexAI.Fix: Check
GOOGLE_GENAI_USE_VERTEXAIenv var. When true, initializegenai.Client(vertexai=True)for ADC-based auth.