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

Merge branch 'dev' into read-only-file-system-support

Browse source
This commit is contained in:
Igor Ilic 2025-10-27 16:41:36 +01:00 • committed by GitHub
commit 8ba1aa5b7c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 14 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
cognee/modules/pipelines/operations/run_tasks_base.py
View file

@ -27,6 +27,7 @@ async def handle_task(
additional_properties={ additional_properties={
"task_name": running_task.executable.__name__, "task_name": running_task.executable.__name__,
"cognee_version": cognee_version, "cognee_version": cognee_version,
"tenant_id": str(user.tenant_id) if user.tenant_id else "Single User Tenant",
}, },
) )
@ -49,6 +50,7 @@ async def handle_task(
additional_properties={ additional_properties={
"task_name": running_task.executable.__name__, "task_name": running_task.executable.__name__,
"cognee_version": cognee_version, "cognee_version": cognee_version,
"tenant_id": str(user.tenant_id) if user.tenant_id else "Single User Tenant",
}, },
) )
except Exception as error: except Exception as error:
@ -62,6 +64,7 @@ async def handle_task(
additional_properties={ additional_properties={
"task_name": running_task.executable.__name__, "task_name": running_task.executable.__name__,
"cognee_version": cognee_version, "cognee_version": cognee_version,
"tenant_id": str(user.tenant_id) if user.tenant_id else "Single User Tenant",
}, },
) )
raise error raise error

3
cognee/modules/pipelines/operations/run_tasks_with_telemetry.py
View file

@ -28,6 +28,7 @@ async def run_tasks_with_telemetry(
additional_properties={ additional_properties={
"pipeline_name": str(pipeline_name), "pipeline_name": str(pipeline_name),
"cognee_version": cognee_version, "cognee_version": cognee_version,
"tenant_id": str(user.tenant_id) if user.tenant_id else "Single User Tenant",
} }
| config, | config,
) )
@ -42,6 +43,7 @@ async def run_tasks_with_telemetry(
additional_properties={ additional_properties={
"pipeline_name": str(pipeline_name), "pipeline_name": str(pipeline_name),
"cognee_version": cognee_version, "cognee_version": cognee_version,
"tenant_id": str(user.tenant_id) if user.tenant_id else "Single User Tenant",
} }
| config, | config,
) )
@ -58,6 +60,7 @@ async def run_tasks_with_telemetry(
additional_properties={ additional_properties={
"pipeline_name": str(pipeline_name), "pipeline_name": str(pipeline_name),
"cognee_version": cognee_version, "cognee_version": cognee_version,
"tenant_id": str(user.tenant_id) if user.tenant_id else "Single User Tenant",
} }
| config, | config,
) )

10
cognee/modules/search/methods/search.py
View file

@ -67,7 +67,10 @@ async def search(
send_telemetry( send_telemetry(
"cognee.search EXECUTION STARTED", "cognee.search EXECUTION STARTED",
user.id, user.id,
additional_properties={"cognee_version": cognee_version}, additional_properties={
"cognee_version": cognee_version,
"tenant_id": str(user.tenant_id) if user.tenant_id else "Single User Tenant",
},
) )
# Use search function filtered by permissions if access control is enabled # Use search function filtered by permissions if access control is enabled
@ -108,7 +111,10 @@ async def search(
send_telemetry( send_telemetry(
"cognee.search EXECUTION COMPLETED", "cognee.search EXECUTION COMPLETED",
user.id, user.id,
additional_properties={"cognee_version": cognee_version}, additional_properties={
"cognee_version": cognee_version,
"tenant_id": str(user.tenant_id) if user.tenant_id else "Single User Tenant",
},
) )
await log_result( await log_result(