Skip to content

feat: Add Google GenAI support#473

Open
balalofernandez wants to merge 3 commits intoneo4j:mainfrom
balalofernandez:alvaro/google-genai-provider
Open

feat: Add Google GenAI support#473
balalofernandez wants to merge 3 commits intoneo4j:mainfrom
balalofernandez:alvaro/google-genai-provider

Conversation

@balalofernandez
Copy link

Description

  • Introduced GeminiEmbedder and GeminiLLM classes for embedding and LLM functionalities using Google's Gemini API.
  • Updated pyproject.toml to include google-genai dependency.
  • Added example scripts for using GeminiEmbedder and GeminiLLM.
  • Enhanced documentation to include new classes and their usage.
  • Updated tests for the new functionalities, ensuring proper error handling and response validation.

Type of Change

  • New feature
  • Bug fix
  • Breaking change
  • Documentation update
  • Project configuration change

Complexity

Low

Complexity:

How Has This Been Tested?

  • Unit tests
  • E2E tests
  • Manual tests

Checklist

The following requirements should have been met (depending on the changes in the branch):

  • Documentation has been updated
  • Unit tests have been updated
  • E2E tests have been updated
  • Examples have been updated
  • New files have copyright header
  • CLA (https://neo4j.com/developer/cla/) has been signed
  • CHANGELOG.md updated if appropriate

@balalofernandez balalofernandez requested a review from a team as a code owner February 9, 2026 12:17
@balalofernandez balalofernandez force-pushed the alvaro/google-genai-provider branch from 362d6a1 to 967abb3 Compare February 9, 2026 12:19
@serengil
Copy link
Contributor

Would you please add google gen ai into README's Optional Dependencies section also.

Copy link
Contributor

@serengil serengil left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Added a couple of comments. Happy to discuss.

- Introduced  and  classes for embedding and LLM functionalities using Google's Gemini API.
- Updated  to include  dependency.
- Added example scripts for using  and .
- Enhanced documentation to include new classes and their usage.
- Updated tests for the new functionalities, ensuring proper error handling and response validation.
- Added support for Google Gemini embeddings and LLMs in the README.
- Updated example scripts to assert the presence of the GOOGLE_API_KEY environment variable.
- Modified the default embedding model and dimension to use environment variables for better configurability.
- Improved error handling in the GeminiEmbedder class to ensure valid responses from the API.
@balalofernandez balalofernandez force-pushed the alvaro/google-genai-provider branch from 967abb3 to 9555678 Compare February 21, 2026 16:48
@balalofernandez
Copy link
Author

Thank you for your comments, I fixed them

@balalofernandez balalofernandez requested a review from serengil March 2, 2026 00:32
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.

2 participants