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

refactor: Add proper pip install command for optional extras

Browse source
This commit is contained in:
Igor Ilic 2025-09-25 13:55:01 +02:00
parent ca2e63bd84
commit d1724c710b
6 changed files with 14 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/api/client.py
View file

@ -51,7 +51,7 @@ if os.getenv("ENV", "prod") == "prod":
) )
except ImportError: except ImportError:
logger.info( logger.info(
"Sentry SDK not available. Install with 'pip install cognee[monitoring]' to enable error monitoring." "Sentry SDK not available. Install with 'pip install cognee\"[monitoring]\"' to enable error monitoring."
) )

2
cognee/eval_framework/modal_eval_dashboard.py
View file

@ -82,7 +82,7 @@ def main():
import pandas as pd import pandas as pd
except ImportError: except ImportError:
st.error( st.error(
"Pandas is required for the evaluation dashboard. Install with 'pip install cognee[evals]' to use this feature." "Pandas is required for the evaluation dashboard. Install with 'pip install cognee\"[evals]\"' to use this feature."
) )
return return

2
cognee/infrastructure/databases/vector/create_vector_engine.py
View file

@ -70,7 +70,7 @@ def create_vector_engine(
from .pgvector.PGVectorAdapter import PGVectorAdapter from .pgvector.PGVectorAdapter import PGVectorAdapter
except ImportError: except ImportError:
raise ImportError( raise ImportError(
"PostgreSQL dependencies are not installed. Please install with 'pip install cognee[postgres]' or 'pip install cognee[postgres-binary]' to use PGVector functionality." "PostgreSQL dependencies are not installed. Please install with 'pip install cognee\"[postgres]\"' or 'pip install cognee\"[postgres-binary]\"' to use PGVector functionality."
) )
return PGVectorAdapter( return PGVectorAdapter(

2
cognee/infrastructure/files/storage/S3FileStorage.py
View file

@ -25,7 +25,7 @@ class S3FileStorage(Storage):
import s3fs import s3fs
except ImportError: except ImportError:
raise ImportError( raise ImportError(
"s3fs is required for S3FileStorage. Install it with: pip install cognee[aws]" 's3fs is required for S3FileStorage. Install it with: pip install cognee"[aws]"'
) )
self.storage_path = storage_path self.storage_path = storage_path

2
cognee/infrastructure/files/utils/open_data_file.py
View file

@ -26,7 +26,7 @@ async def open_data_file(file_path: str, mode: str = "rb", encoding: str = None,
from cognee.infrastructure.files.storage.S3FileStorage import S3FileStorage from cognee.infrastructure.files.storage.S3FileStorage import S3FileStorage
except ImportError: except ImportError:
raise ImportError( raise ImportError(
"S3 dependencies are not installed. Please install with 'pip install cognee[aws]' to use S3 functionality." "S3 dependencies are not installed. Please install with 'pip install cognee\"[aws]\"' to use S3 functionality."
) )
normalized_url = get_data_file_path(file_path) normalized_url = get_data_file_path(file_path)

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

@ -8,12 +8,7 @@ logger = get_logger()
async def cognee_network_visualization(graph_data, destination_file_path: str = None): async def cognee_network_visualization(graph_data, destination_file_path: str = None):
try: import networkx
import networkx
except ImportError:
raise ImportError(
"NetworkX is not installed. Please install with 'pip install cognee[visualization]' to use graph visualization features."
)
nodes_data, edges_data = graph_data nodes_data, edges_data = graph_data
@ -110,7 +105,7 @@ async def cognee_network_visualization(graph_data, destination_file_path: str =
.nodes circle { stroke: white; stroke-width: 0.5px; filter: drop-shadow(0 0 5px rgba(255,255,255,0.3)); } .nodes circle { stroke: white; stroke-width: 0.5px; filter: drop-shadow(0 0 5px rgba(255,255,255,0.3)); }
.node-label { font-size: 5px; font-weight: bold; fill: white; text-anchor: middle; dominant-baseline: middle; font-family: 'Inter', sans-serif; pointer-events: none; } .node-label { font-size: 5px; font-weight: bold; fill: white; text-anchor: middle; dominant-baseline: middle; font-family: 'Inter', sans-serif; pointer-events: none; }
.edge-label { font-size: 3px; fill: rgba(255, 255, 255, 0.7); text-anchor: middle; dominant-baseline: middle; font-family: 'Inter', sans-serif; pointer-events: none; } .edge-label { font-size: 3px; fill: rgba(255, 255, 255, 0.7); text-anchor: middle; dominant-baseline: middle; font-family: 'Inter', sans-serif; pointer-events: none; }
.tooltip { .tooltip {
position: absolute; position: absolute;
text-align: left; text-align: left;
@ -172,7 +167,7 @@ async def cognee_network_visualization(graph_data, destination_file_path: str =
// Create tooltip content for edge // Create tooltip content for edge
var content = "<strong>Edge Information</strong><br/>"; var content = "<strong>Edge Information</strong><br/>";
content += "Relationship: " + d.relation + "<br/>"; content += "Relationship: " + d.relation + "<br/>";
// Show all weights // Show all weights
if (d.all_weights && Object.keys(d.all_weights).length > 0) { if (d.all_weights && Object.keys(d.all_weights).length > 0) {
content += "<strong>Weights:</strong><br/>"; content += "<strong>Weights:</strong><br/>";
@ -182,23 +177,23 @@ async def cognee_network_visualization(graph_data, destination_file_path: str =
} else if (d.weight !== null && d.weight !== undefined) { } else if (d.weight !== null && d.weight !== undefined) {
content += "Weight: " + d.weight + "<br/>"; content += "Weight: " + d.weight + "<br/>";
} }
if (d.relationship_type) { if (d.relationship_type) {
content += "Type: " + d.relationship_type + "<br/>"; content += "Type: " + d.relationship_type + "<br/>";
} }
// Add other edge properties // Add other edge properties
if (d.edge_info) { if (d.edge_info) {
Object.keys(d.edge_info).forEach(function(key) { Object.keys(d.edge_info).forEach(function(key) {
if (key !== 'weight' && key !== 'weights' && key !== 'relationship_type' && if (key !== 'weight' && key !== 'weights' && key !== 'relationship_type' &&
key !== 'source_node_id' && key !== 'target_node_id' && key !== 'source_node_id' && key !== 'target_node_id' &&
key !== 'relationship_name' && key !== 'updated_at' && key !== 'relationship_name' && key !== 'updated_at' &&
!key.startsWith('weight_')) { !key.startsWith('weight_')) {
content += key + ": " + d.edge_info[key] + "<br/>"; content += key + ": " + d.edge_info[key] + "<br/>";
} }
}); });
} }
tooltip.html(content) tooltip.html(content)
.style("left", (d3.event.pageX + 10) + "px") .style("left", (d3.event.pageX + 10) + "px")
.style("top", (d3.event.pageY - 10) + "px") .style("top", (d3.event.pageY - 10) + "px")