fixes for tests

Author: Gaudy Blanco

.vscode/launch.json

@@ -50,5 +50,17 @@
 			"console": "integratedTerminal",
     		"justMyCode": false 
 		},
+		{
+			"name": "Debug Verbs Pytest",
+			"type": "debugpy",
+			"request": "launch",
+			"module": "pytest",
+			"args": [
+				"./tests/verbs",
+				"-k", "test_generate_text_embeddings"
+			],
+			"console": "integratedTerminal",
+    		"justMyCode": false 
+		},
 	]
 }

graphrag/vector_stores/lancedb.py

@@ -39,13 +39,16 @@ def load_documents(
         self, documents: list[VectorStoreDocument], overwrite: bool = True
     ) -> None:
         """Load documents into vector storage."""
-        # TODO GAUDY Step 1: Prepare data columns manually
+        # Step 1: Prepare data columns manually
         ids = []
         texts = []
         vectors = []
         attributes = []
 
         for document in documents:
+            self.vector_size = (
+                len(document.vector) if document.vector else self.vector_size
+            )
             if document.vector is not None and len(document.vector) == self.vector_size:
                 ids.append(document.id)
                 texts.append(document.text)