diff --git a/cognee/modules/pipelines/operations/run_tasks_base.py b/cognee/modules/pipelines/operations/run_tasks_base.py index ee2ccfd8c..79d37a451 100644 --- a/cognee/modules/pipelines/operations/run_tasks_base.py +++ b/cognee/modules/pipelines/operations/run_tasks_base.py @@ -27,6 +27,7 @@ async def handle_task( additional_properties={ "task_name": running_task.executable.__name__, "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={ "task_name": running_task.executable.__name__, "cognee_version": cognee_version, + "tenant_id": str(user.tenant_id) if user.tenant_id else "Single User Tenant", }, ) except Exception as error: @@ -62,6 +64,7 @@ async def handle_task( additional_properties={ "task_name": running_task.executable.__name__, "cognee_version": cognee_version, + "tenant_id": str(user.tenant_id) if user.tenant_id else "Single User Tenant", }, ) raise error diff --git a/cognee/modules/pipelines/operations/run_tasks_with_telemetry.py b/cognee/modules/pipelines/operations/run_tasks_with_telemetry.py index 9a52bf854..ae968c7a5 100644 --- a/cognee/modules/pipelines/operations/run_tasks_with_telemetry.py +++ b/cognee/modules/pipelines/operations/run_tasks_with_telemetry.py @@ -28,6 +28,7 @@ async def run_tasks_with_telemetry( additional_properties={ "pipeline_name": str(pipeline_name), "cognee_version": cognee_version, + "tenant_id": str(user.tenant_id) if user.tenant_id else "Single User Tenant", } | config, ) @@ -42,6 +43,7 @@ async def run_tasks_with_telemetry( additional_properties={ "pipeline_name": str(pipeline_name), "cognee_version": cognee_version, + "tenant_id": str(user.tenant_id) if user.tenant_id else "Single User Tenant", } | config, ) @@ -58,6 +60,7 @@ async def run_tasks_with_telemetry( additional_properties={ "pipeline_name": str(pipeline_name), "cognee_version": cognee_version, + "tenant_id": str(user.tenant_id) if user.tenant_id else "Single User Tenant", } | config, ) diff --git a/cognee/modules/search/methods/search.py b/cognee/modules/search/methods/search.py index 93c0ef5c8..aab004924 100644 --- a/cognee/modules/search/methods/search.py +++ b/cognee/modules/search/methods/search.py @@ -67,7 +67,10 @@ async def search( send_telemetry( "cognee.search EXECUTION STARTED", 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 @@ -108,7 +111,10 @@ async def search( send_telemetry( "cognee.search EXECUTION COMPLETED", 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(