diff --git a/cognee/modules/pipelines/operations/run_tasks_base.py b/cognee/modules/pipelines/operations/run_tasks_base.py index e5f577848..ee2ccfd8c 100644 --- a/cognee/modules/pipelines/operations/run_tasks_base.py +++ b/cognee/modules/pipelines/operations/run_tasks_base.py @@ -2,6 +2,7 @@ import inspect from cognee.shared.logging_utils import get_logger from cognee.modules.users.models import User from cognee.shared.utils import send_telemetry +from cognee import __version__ as cognee_version from ..tasks.task import Task @@ -25,6 +26,7 @@ async def handle_task( user_id=user.id, additional_properties={ "task_name": running_task.executable.__name__, + "cognee_version": cognee_version, }, ) @@ -46,6 +48,7 @@ async def handle_task( user_id=user.id, additional_properties={ "task_name": running_task.executable.__name__, + "cognee_version": cognee_version, }, ) except Exception as error: @@ -58,6 +61,7 @@ async def handle_task( user_id=user.id, additional_properties={ "task_name": running_task.executable.__name__, + "cognee_version": cognee_version, }, ) 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 bcee4cec6..9a52bf854 100644 --- a/cognee/modules/pipelines/operations/run_tasks_with_telemetry.py +++ b/cognee/modules/pipelines/operations/run_tasks_with_telemetry.py @@ -4,6 +4,7 @@ from cognee.modules.settings import get_current_settings from cognee.modules.users.models import User from cognee.shared.logging_utils import get_logger from cognee.shared.utils import send_telemetry +from cognee import __version__ as cognee_version from .run_tasks_base import run_tasks_base from ..tasks.task import Task @@ -26,6 +27,7 @@ async def run_tasks_with_telemetry( user.id, additional_properties={ "pipeline_name": str(pipeline_name), + "cognee_version": cognee_version, } | config, ) @@ -39,6 +41,7 @@ async def run_tasks_with_telemetry( user.id, additional_properties={ "pipeline_name": str(pipeline_name), + "cognee_version": cognee_version, } | config, ) @@ -54,6 +57,7 @@ async def run_tasks_with_telemetry( user.id, additional_properties={ "pipeline_name": str(pipeline_name), + "cognee_version": cognee_version, } | config, )