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

Add utils for graph visualization + classification nodes

Browse source
This commit is contained in:
Vasilije 2024-03-11 14:23:01 +01:00
parent 7e964bcb23
commit d69d42b451
2 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cognitive_architecture/api/v1/cognify/cognify.py
View file

@ -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.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.batch_search import adapted_qdrant_batch_search
from cognitive_architecture.modules.cognify.vector.load_propositions import add_propositions 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 environment variables from .env file
load_dotenv() load_dotenv()
@ -199,6 +200,9 @@ async def cognify(input_text:str):
relationship_d = graph_ready_output(results) relationship_d = graph_ready_output(results)
CONNECTED_GRAPH = connect_nodes_in_graph(F, relationship_d) CONNECTED_GRAPH = connect_nodes_in_graph(F, relationship_d)
out = await render_graph(CONNECTED_GRAPH, graph_type='networkx')
print(out)
return CONNECTED_GRAPH return CONNECTED_GRAPH
# #

2
cognitive_architecture/modules/cognify/graph/add_node_connections.py
View file

@ -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 graph = graph_client.graph
node_descriptions = extract_node_descriptions(graph.nodes(data=True)) node_descriptions = extract_node_descriptions(graph.nodes(data=True))