From d69d42b451d4bbd37de53964c478e1935275e045 Mon Sep 17 00:00:00 2001 From: Vasilije <8619304+Vasilije1990@users.noreply.github.com> Date: Mon, 11 Mar 2024 14:23:01 +0100 Subject: [PATCH] Add utils for graph visualization + classification nodes --- cognitive_architecture/api/v1/cognify/cognify.py | 4 ++++ .../modules/cognify/graph/add_node_connections.py | 2 +- 2 files changed, 5 insertions(+), 1 deletion(-) 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))