Updated docs

Author: alexthomas93

docs/source/index.rst

@@ -148,9 +148,9 @@ Note that the below example is not the only way you can upsert data into your Ne
 
 
 .. code:: python
-
     from neo4j import GraphDatabase
     from neo4j_graphrag.indexes import upsert_vectors
+    from neo4j_graphrag.types import EntityType
 
     URI = "neo4j://localhost:7687"
     AUTH = ("neo4j", "password")
@@ -165,7 +165,7 @@ Note that the below example is not the only way you can upsert data into your Ne
         ids=["1234"],
         embedding_property="vectorProperty",
         embeddings=[vector],
-        entity_type="NODE"
+        entity_type=EntityType.NODE,
     )
 
 

docs/source/user_guide_rag.rst

@@ -917,10 +917,11 @@ Populate a Vector Index
 ==========================
 
 .. code:: python
+    from random import random
 
     from neo4j import GraphDatabase
-    from random import random
     from neo4j_graphrag.indexes import upsert_vectors
+    from neo4j_graphrag.types import EntityType
 
     URI = "neo4j://localhost:7687"
     AUTH = ("neo4j", "password")
@@ -929,15 +930,15 @@ Populate a Vector Index
     driver = GraphDatabase.driver(URI, auth=AUTH)
 
     # Upsert the vector
+    DIMENSION = 1536
     vector = [random() for _ in range(DIMENSION)]
-        upsert_vectors(
+    upsert_vectors(
         driver,
         ids=["1234"],
         embedding_property="vectorProperty",
         embeddings=[vector],
-        entity_type="NODE"
+        entity_type=EntityType.NODE,
     )
-    upsert_vectors(driver, node_id="1234", embedding_property="embedding", vector=vector)
 
 This will update the node with `id(node)=1234` to add (or update) a `node.vectorProperty` property.
 This property will also be added to the vector index.