Skip to content

Commit 66231f8

Browse files
authored
Adds Vertex AI embeddings class (#129)
* Added Vertex AI embeddings class * Updated poetry lock * Updated CHANGELOG
1 parent b90003b commit 66231f8

File tree

6 files changed

+554
-3
lines changed

6 files changed

+554
-3
lines changed

CHANGELOG.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,11 @@
22

33
## Next
44

5+
### Added
6+
- Introduced a new Vertex AI embeddings class for generating text embeddings using Vertex AI.
7+
- Updated documentation to include OpenAI and Vertex AI embeddings classes.
8+
- Added google-cloud-aiplatform as an optional dependency for Vertex AI embeddings.
9+
510
## 0.6.2
611

712
## 0.6.1

docs/source/api.rst

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,18 @@ SentenceTransformerEmbeddings
144144
.. autoclass:: neo4j_graphrag.embeddings.sentence_transformers.SentenceTransformerEmbeddings
145145
:members:
146146

147+
OpenAIEmbeddings
148+
================
149+
150+
.. autoclass:: neo4j_genai.embeddings.openai.OpenAIEmbeddings
151+
:members:
152+
153+
VertexAIEmbeddings
154+
==================
155+
156+
.. autoclass:: neo4j_genai.embeddings.vertexai.VertexAIEmbeddings
157+
:members:
158+
147159
**********
148160
Generation
149161
**********

0 commit comments

Comments
 (0)