Remove unused cols from final entities (#1226)

* Remove unused cols from final entities

* Move verbs test to integ

* Move verbs test to integ

* Move to smoke tests

.github/workflows/python-smoke-tests.yml

@@ -92,6 +92,10 @@ jobs:
         run: |
           poetry build
 
+      - name: Verb Test
+        run: |
+          poetry run poe test_verbs
+
       - name: Install Azurite
         id: azuright
         uses: potatoqualitee/[email protected]

.semversioner/next-release/patch-20240927171840021536.json

@@ -0,0 +1,4 @@
+{
+  "type": "patch",
+  "description": "Remove unused column from final entities"
+}

graphrag/index/workflows/v1/subflows/create_final_entities.py

@@ -91,6 +91,7 @@ async def create_final_entities(
         # Drop rows with NaN 'description_embedding' if not using vector store
         if not description_text_embed.get("strategy", {}).get("vector_store"):
             nodes = nodes.loc[nodes["description_embedding"].notna()]
+        nodes.drop(columns="name_description", inplace=True)
 
     # Return final result
     return create_verb_result(cast(Table, nodes))