Adds Vertex AI embeddings class (#129)

* Added Vertex AI embeddings class * Updated poetry lock * Updated CHANGELOG
neo4j · Sep 17, 2024 · 66231f8 · 66231f8
1 parent b90003b
commit 66231f8
Show file tree

Hide file tree

Showing 6 changed files with 554 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,6 +2,11 @@
 
 ## Next
 
+### Added
+- Introduced a new Vertex AI embeddings class for generating text embeddings using Vertex AI.
+- Updated documentation to include OpenAI and Vertex AI embeddings classes.
+- Added google-cloud-aiplatform as an optional dependency for Vertex AI embeddings.
+
 ## 0.6.2
 
 ## 0.6.1

diff --git a/docs/source/api.rst b/docs/source/api.rst
@@ -144,6 +144,18 @@ SentenceTransformerEmbeddings
 .. autoclass:: neo4j_graphrag.embeddings.sentence_transformers.SentenceTransformerEmbeddings
     :members:
 
+OpenAIEmbeddings
+================
+
+.. autoclass:: neo4j_genai.embeddings.openai.OpenAIEmbeddings
+    :members:
+
+VertexAIEmbeddings
+==================
+
+.. autoclass:: neo4j_genai.embeddings.vertexai.VertexAIEmbeddings
+    :members:
+
 **********
 Generation
 **********