Graph Visualization

from cognee.api.v1.visualize.visualize import visualize_graph

# Default: bounded subgraph around the highest-degree nodes
await visualize_graph("./graph.html")

# Seed the subgraph from a query
await visualize_graph("./graph.html", query="natural language processing")

# Show the subgraph behind a recall answer
result = await cognee.recall("What does Alice know?")
await visualize_graph("./graph.html", recall_result=result)

# Legacy whole-graph render
await visualize_graph("./graph.html", full=True)

from cognee.infrastructure.databases.graph import get_graph_engine

graph_engine = await get_graph_engine()

nodes, edges = await graph_engine.get_graph_data()
print(f"nodes={len(nodes)}, edges={len(edges)}")

metrics = await graph_engine.get_graph_metrics()
print(metrics)  # {'num_nodes': ..., 'num_edges': ..., ...}

import asyncio
import cognee
import os
from cognee.api.v1.visualize.visualize import visualize_graph


async def main():
    # Prune data and system metadata before running, only if we want "fresh" state.
    await cognee.forget(everything=True)

    await cognee.remember(
        ["Alice knows Bob.", "NLP is a subfield of CS."],
        self_improvement=False,
    )

    visualize_graph_path = os.path.join(
        os.path.dirname(__file__), ".artifacts", "graph_after_remember.html"
    )
    await visualize_graph(visualize_graph_path)


if __name__ == "__main__":
    asyncio.run(main())

import asyncio
import os

import cognee
from cognee.api.v1.visualize.visualize import visualize_graph

DEST = os.path.join(os.path.expanduser("~"), "semantic_memory_map.html")

# A few short, deliberately multi-topic passages so distinct clusters emerge:
# computing pioneers, jazz, and ocean science.
TEXT = """
Ada Lovelace worked with Charles Babbage on the Analytical Engine in London.
Alan Turing formalized computation and broke ciphers at Bletchley Park.
Grace Hopper built the first compiler and worked on the Harvard Mark I.

Miles Davis recorded Kind of Blue, a landmark modal jazz album, in New York.
John Coltrane played saxophone with the Miles Davis Quintet before A Love Supreme.
Bill Evans, the pianist on Kind of Blue, shaped its impressionistic harmony.

Marine biologists study coral reefs, which host a quarter of all ocean species.
Rising sea temperatures cause coral bleaching, threatening reef ecosystems.
Phytoplankton in the ocean produce a large share of the planet's oxygen.
"""


async def main():
    await cognee.prune.prune_data()
    await cognee.prune.prune_system(metadata=True)

    await cognee.add(TEXT)
    await cognee.cognify()

    html = await visualize_graph(destination_file_path=DEST)

    has_semantic = 'data-view="semantic"' in html
    has_positions = "window._semanticPositions = null" not in html
    print(f"\nSaved: {DEST}")
    print(f"Semantic tab present:   {has_semantic}")
    print(f"Semantic positions set: {has_positions}")
    print("\nOpen the file and click the Semantic tab (or append #semantic to the URL).")


if __name__ == "__main__":
    asyncio.run(main())

import asyncio
import cognee
import os
from cognee.api.v1.visualize.visualize import visualize_graph


async def main():
    await cognee.add(["Alice knows Bob.", "NLP is a subfield of CS."])
    await cognee.cognify()

    visualize_graph_path = os.path.join(
        os.path.dirname(__file__), ".artifacts", "graph_visualization.html"
    )
    await visualize_graph(visualize_graph_path)


if __name__ == "__main__":
    asyncio.run(main())