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

Merge remote-tracking branch 'origin/cognify_adjusment_crewai_demo' into cognify_adjusment_crewai_demo

Browse source
This commit is contained in:
Boris Arzentar 2025-06-04 16:39:10 +02:00
commit 042c122da0
No known key found for this signature in database
GPG key ID: D5CC274C784807B7
2 changed files with 10 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
cognee/complex_demos/crewai_demo/src/crewai_demo/hiring_crew.py
View file

@ -25,7 +25,6 @@ class HiringCrew:
def __init__(self, user, inputs): def __init__(self, user, inputs):
self.user = user self.user = user
self.inputs = inputs self.inputs = inputs
self
@agent @agent
def soft_skills_expert_agent(self) -> Agent: def soft_skills_expert_agent(self) -> Agent:

11
cognee/modules/data/deletion/prune_system.py
View file

@ -3,6 +3,7 @@ from cognee.infrastructure.databases.graph.get_graph_engine import get_graph_eng
from cognee.infrastructure.databases.relational import get_relational_engine from cognee.infrastructure.databases.relational import get_relational_engine
from cognee.modules.data.methods import delete_dataset, get_authorized_existing_datasets from cognee.modules.data.methods import delete_dataset, get_authorized_existing_datasets
from cognee.modules.users.exceptions import PermissionDeniedError from cognee.modules.users.exceptions import PermissionDeniedError
from sqlalchemy import delete
async def prune_system(user=None, graph=True, vector=True, metadata=False): async def prune_system(user=None, graph=True, vector=True, metadata=False):
@ -22,7 +23,15 @@ async def prune_system(user=None, graph=True, vector=True, metadata=False):
user_datasets = [] user_datasets = []
for dataset in user_datasets: for dataset in user_datasets:
await delete_dataset(dataset) # Delete ACLs related to dataset
from cognee.modules.users.models import ACL
from cognee.modules.data.models import Dataset
engine = get_relational_engine()
async with engine.get_async_session() as session:
await session.execute(delete(ACL).where(ACL.dataset_id == dataset.id))
await session.execute(delete(Dataset).where(Dataset.id == dataset.id))
await session.commit()
else: else:
if graph: if graph:
graph_engine = await get_graph_engine() graph_engine = await get_graph_engine()