From fb83cd4ff4cbf58b10f296e164880eaae9e7b37c Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Mon, 20 Feb 2023 21:43:02 -0800 Subject: [PATCH] catch networkx error (#1201) --- langchain/graphs/networkx_graph.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/langchain/graphs/networkx_graph.py b/langchain/graphs/networkx_graph.py index f48fe0f1..66dd5bb2 100644 --- a/langchain/graphs/networkx_graph.py +++ b/langchain/graphs/networkx_graph.py @@ -50,7 +50,13 @@ class NetworkxEntityGraph: def __init__(self) -> None: """Create a new graph.""" - import networkx as nx + try: + import networkx as nx + except ImportError: + raise ValueError( + "Could not import networkx python package. " + "Please it install it with `pip install networkx`." + ) self._graph = nx.DiGraph()