feat: add Vertex AI provider for Google Cloud AI/ML integration#6264
Open
duckyduckycode wants to merge 1 commit intokeephq:mainfrom
Open
feat: add Vertex AI provider for Google Cloud AI/ML integration#6264duckyduckycode wants to merge 1 commit intokeephq:mainfrom
duckyduckycode wants to merge 1 commit intokeephq:mainfrom
Conversation
Adds a new Vertex AI provider that integrates Keep with Google Cloud's unified ML platform, resolving keephq#6087. Features: - Dual authentication: API key (express mode) and service account (full Vertex AI) - Support for all 11 Vertex AI regions - Structured output via JSON schema - Configurable model selection (gemini-2.0-flash default) - Temperature and max_tokens parameters - Automatic JSON response parsing - Workload identity support for GKE deployments - Comprehensive unit tests (15 test cases) Follows the same pattern as existing AI providers (Anthropic, Gemini, OpenAI) with the _query method for Keep workflow integration.
|
|
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
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.
Summary
Adds a new Vertex AI provider that integrates Keep with Google Cloud Vertex AI, resolving #6087.
Vertex AI is Google Cloud's unified ML platform providing access to foundation models (Gemini, etc.) for text generation, structured output, and AI-powered workflows within Keep.
What Changed
New Provider:
keep/providers/vertexai_provider/Dual Authentication Support:
Features:
_query()method following the same pattern as Anthropic/Gemini/OpenAI providersgemini-2.0-flash, supports all Gemini models)Files Added
keep/providers/vertexai_provider/__init__.pykeep/providers/vertexai_provider/vertexai_provider.pytests/providers/vertexai_provider/test_vertexai_provider.pydocs/providers/documentation/vertexai-provider.mdxUsage in Workflows
Testing
15 unit tests covering:
Checklist
*_provider)__init__.pyexports provider class and auth configvalidate_config()implementeddispose()implementedPROVIDER_DISPLAY_NAME,PROVIDER_CATEGORY,PROVIDER_TAGSsetResolves #6087