diff --git a/cognitive_architecture/api/v1/cognify/cognify.py b/cognitive_architecture/api/v1/cognify/cognify.py index fee11b00e..6c7850f50 100644 --- a/cognitive_architecture/api/v1/cognify/cognify.py +++ b/cognitive_architecture/api/v1/cognify/cognify.py @@ -22,6 +22,7 @@ from cognitive_architecture.modules.cognify.graph.add_propositions import append from cognitive_architecture.modules.cognify.llm.add_node_connection_embeddings import process_items from cognitive_architecture.modules.cognify.vector.batch_search import adapted_qdrant_batch_search from cognitive_architecture.modules.cognify.vector.load_propositions import add_propositions +from cognitive_architecture.utils import render_graph # Load environment variables from .env file load_dotenv() @@ -199,6 +200,9 @@ async def cognify(input_text:str): relationship_d = graph_ready_output(results) CONNECTED_GRAPH = connect_nodes_in_graph(F, relationship_d) + + out = await render_graph(CONNECTED_GRAPH, graph_type='networkx') + print(out) return CONNECTED_GRAPH # diff --git a/cognitive_architecture/modules/cognify/graph/add_node_connections.py b/cognitive_architecture/modules/cognify/graph/add_node_connections.py index 0fb2b4e3a..66032d682 100644 --- a/cognitive_architecture/modules/cognify/graph/add_node_connections.py +++ b/cognitive_architecture/modules/cognify/graph/add_node_connections.py @@ -12,7 +12,7 @@ def extract_node_descriptions(data): -def add_node_connection(graph_client, vector_database_client, data): +async def add_node_connection(graph_client, vector_database_client, data): graph = graph_client.graph node_descriptions = extract_node_descriptions(graph.nodes(data=True))