17 lines
600 B
Python
17 lines
600 B
Python
from cognee.infrastructure.databases.vector import get_vector_engine
|
|
from cognee.infrastructure.databases.graph.get_graph_engine import get_graph_engine
|
|
from cognee.infrastructure.databases.relational import get_relational_engine
|
|
|
|
|
|
async def prune_system(graph=True, vector=True, metadata=True):
|
|
if graph:
|
|
graph_engine = await get_graph_engine()
|
|
await graph_engine.delete_graph()
|
|
|
|
if vector:
|
|
vector_engine = get_vector_engine()
|
|
await vector_engine.prune()
|
|
|
|
if metadata:
|
|
db_engine = get_relational_engine()
|
|
await db_engine.delete_database()
|