gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Keep networkx as core dependency

Browse source
This commit is contained in:
Daulet Amirkhanov 2025-08-15 10:23:57 +01:00
parent 81a17580a6
commit 85c7f11eb8
2 changed files with 2 additions and 31 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
cognee/modules/visualization/cognee_network_visualization.py
View file

@ -1,5 +1,6 @@
import os import os
import json import json
import networkx
from cognee.shared.logging_utils import get_logger from cognee.shared.logging_utils import get_logger
from cognee.infrastructure.files.storage.LocalFileStorage import LocalFileStorage from cognee.infrastructure.files.storage.LocalFileStorage import LocalFileStorage
@ -7,37 +8,7 @@ from cognee.infrastructure.files.storage.LocalFileStorage import LocalFileStorag
logger = get_logger() logger = get_logger()
def _import_networkx():
"""Dynamically import networkx with helpful error message if not available."""
try:
import networkx
return networkx
except ImportError:
raise ImportError(
"NetworkX is required for graph visualization but is not installed. "
"Please install it with: pip install 'cognee[visualization]' or pip install networkx"
)
async def cognee_network_visualization(graph_data, destination_file_path: str = None): async def cognee_network_visualization(graph_data, destination_file_path: str = None):
"""
Generate an interactive HTML visualization of the graph data.
This function requires NetworkX to be installed. If you don't have NetworkX installed,
you can install it with: pip install 'cognee[visualization]' or pip install networkx
Args:
graph_data: Tuple of (nodes_data, edges_data)
destination_file_path: Optional path to save the HTML file
Returns:
str: Path to the generated HTML file
Raises:
ImportError: If NetworkX is not installed
"""
networkx = _import_networkx()
nodes_data, edges_data = graph_data nodes_data, edges_data = graph_data
G = networkx.DiGraph() G = networkx.DiGraph()

2
pyproject.toml
View file

@ -44,6 +44,7 @@ dependencies = [
"pypdf>=4.1.0,<6.0.0", "pypdf>=4.1.0,<6.0.0",
"jinja2>=3.1.3,<4", "jinja2>=3.1.3,<4",
"matplotlib>=3.8.3,<4", "matplotlib>=3.8.3,<4",
"networkx>=3.4.2,<4",
"lancedb>=0.24.0,<1.0.0", "lancedb>=0.24.0,<1.0.0",
"alembic>=1.13.3,<2", "alembic>=1.13.3,<2",
"pre-commit>=4.0.1,<5", "pre-commit>=4.0.1,<5",
@ -119,7 +120,6 @@ gui = [
"qasync>=0.27.1,<0.28", "qasync>=0.27.1,<0.28",
] ]
graphiti = ["graphiti-core>=0.7.0,<0.8"] graphiti = ["graphiti-core>=0.7.0,<0.8"]
visualization = ["networkx>=3.4.2,<4"]
# Note: New s3fs and boto3 versions don't work well together # Note: New s3fs and boto3 versions don't work well together
# Always use comaptible fixed versions of these two dependencies # Always use comaptible fixed versions of these two dependencies
aws = ["s3fs[boto3]==2025.3.2"] aws = ["s3fs[boto3]==2025.3.2"]