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

fix: Remove backend lock from UI

Browse source 
Removed lock that prevented using multiple datasets in cognify

Fix COG-912
This commit is contained in:
Igor Ilic 2024-12-17 11:52:30 +01:00
parent 630ab556db
commit c2d2b26319
1 changed files with 10 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
cognee/api/v1/cognify/cognify_v2.py
View file

@ -69,7 +69,7 @@ async def run_cognify_pipeline(dataset: Dataset, user: User, graph_model: BaseMo
send_telemetry("cognee.cognify EXECUTION STARTED", user.id) send_telemetry("cognee.cognify EXECUTION STARTED", user.id)
async with update_status_lock: #async with update_status_lock: TODO: Add UI lock to prevent multiple backend requests
task_status = await get_pipeline_status([dataset_id]) task_status = await get_pipeline_status([dataset_id])
if dataset_id in task_status and task_status[dataset_id] == PipelineRunStatus.DATASET_PROCESSING_STARTED: if dataset_id in task_status and task_status[dataset_id] == PipelineRunStatus.DATASET_PROCESSING_STARTED:
@ -80,6 +80,7 @@ async def run_cognify_pipeline(dataset: Dataset, user: User, graph_model: BaseMo
"dataset_name": dataset_name, "dataset_name": dataset_name,
"files": document_ids_str, "files": document_ids_str,
}) })
try: try:
cognee_config = get_cognify_config() cognee_config = get_cognify_config()