Skip to content

feat: add Gemini/Vertex AI as LLM provider#12

Merged
netsirius merged 1 commit intomainfrom
feature/gemini-provider
Apr 6, 2026
Merged

feat: add Gemini/Vertex AI as LLM provider#12
netsirius merged 1 commit intomainfrom
feature/gemini-provider

Conversation

@netsirius
Copy link
Copy Markdown
Owner

Adds Google Gemini support across all LLM components.

  • provider: gemini in pipeline YAML
  • Google AI Studio API (generativelanguage.googleapis.com)
  • Default model: gemini-2.0-flash
  • API key via GOOGLE_API_KEY or GEMINI_API_KEY env var
  • Vertex AI supported via custom baseUrl

Files changed:

  • LLMTransformPlugin: Gemini request/response handling
  • GraphExtractionPlugin: Gemini for entity extraction
  • LLMClient: weaver generate with Gemini
  • WeaverConfig: API key resolution
  • README.md: updated LLM provider table

Adds Google Gemini support across all LLM components:
- LLMTransformPlugin: provider "gemini" with generateContent API
- GraphExtractionPlugin: Gemini for entity extraction
- LLMClient: weaver generate with Gemini
- WeaverConfig: GOOGLE_API_KEY / GEMINI_API_KEY resolution

API: Google AI Studio (generativelanguage.googleapis.com)
Default model: gemini-2.0-flash
Auth: API key via URL parameter (?key=...)

Also supports Vertex AI via custom baseUrl for production GCP.
@netsirius netsirius merged commit bbe457e into main Apr 6, 2026
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant