lint

langchain-ai · Oct 10, 2024 · 6dc61b8 · 6dc61b8
1 parent a14af29
commit 6dc61b8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/astradb/langchain_astradb/graph_vectorstores.py b/libs/astradb/langchain_astradb/graph_vectorstores.py
@@ -628,7 +628,7 @@ async def fetch_initial_candidates() -> None:
                     retrieved_docs[doc_id] = doc
 
                 if doc_id not in outgoing_links_map:
-                    node = self._doc_to_node(doc)
+                    node = _doc_to_node(doc)
                     outgoing_links_map[doc_id] = _outgoing_links(node=node)
                     candidates[doc_id] = embedding
             helper.add_candidates(candidates)
@@ -986,7 +986,7 @@ async def _get_outgoing_links(self, source_ids: Iterable[str]) -> set[Link]:
 
         for doc in docs:
             if doc is not None:
-                node = self._doc_to_node(doc=doc)
+                node = _doc_to_node(doc=doc)
                 links.update(_outgoing_links(node=node))
 
         return links