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

adds: ads default handling in expand_with_nodes_and_edges

Browse source
This commit is contained in:
hajdul88 2025-09-19 13:56:11 +02:00
parent 54cf0967cc
commit ea487d2ca3
1 changed files with 12 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
cognee/modules/graph/utils/expand_with_nodes_and_edges.py
View file

@ -8,9 +8,11 @@ from cognee.modules.engine.utils import (
generate_node_name,
)
from cognee.modules.ontology.base_ontology_resolver import BaseOntologyResolver
from cognee.modules.ontology.ontology_env_config import get_ontology_env_config
from cognee.shared.data_models import KnowledgeGraph
from cognee.modules.ontology.rdf_xml.RDFLibOntologyResolver import RDFLibOntologyResolver
from cognee.modules.ontology.get_default_ontology_resolver import get_default_ontology_resolver
from cognee.modules.ontology.get_default_ontology_resolver import get_default_ontology_resolver, \
get_ontology_resolver_from_env
def _create_node_key(node_id: str, category: str) -> str:
@ -322,7 +324,15 @@ def expand_with_nodes_and_edges(
existing_edges_map = {}
if ontology_resolver is None:
ontology_resolver = get_default_ontology_resolver()
ontology_config = get_ontology_env_config()
if (
ontology_config.ontology_file_path
and ontology_config.ontology_resolver
and ontology_config.matching_strategy
):
ontology_resolver = get_ontology_resolver_from_env(**ontology_config.to_dict())
else:
ontology_resolver = get_default_ontology_resolver()
added_nodes_map = {}
added_ontology_nodes_map = {}