fix: fixing removed entity

QuivrHQ · Dec 30, 2024 · 3a567d8 · 3a567d8
1 parent 5b8efcb
commit 3a567d8
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/core/quivr_core/rag/quivr_rag_langgraph.py b/core/quivr_core/rag/quivr_rag_langgraph.py
@@ -13,7 +13,6 @@
     TypedDict,
 )
 from uuid import UUID, uuid4
-
 import openai
 from langchain.retrievers import ContextualCompressionRetriever
 from langchain_cohere import CohereRerank
@@ -1009,8 +1008,6 @@ def _extract_node_name(self, event: dict) -> str:
                 if node.name == name:
                     if node.description:
                         return node.description
-                    elif node.label:
-                        return node.label
                     else:
                         return node.name
         return ""