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

feat: adds cognee version to task and pipeline based telemetry info

Browse source
This commit is contained in:
hajdul88 2025-10-24 15:12:35 +02:00
parent d9f3be4d62
commit 5ce79fb361
2 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -2,6 +2,7 @@ import inspect
from cognee.shared.logging_utils import get_logger from cognee.shared.logging_utils import get_logger
from cognee.modules.users.models import User from cognee.modules.users.models import User
from cognee.shared.utils import send_telemetry from cognee.shared.utils import send_telemetry
from cognee import __version__ as cognee_version
from ..tasks.task import Task from ..tasks.task import Task
@ -25,6 +26,7 @@ async def handle_task(
user_id=user.id, user_id=user.id,
additional_properties={ additional_properties={
"task_name": running_task.executable.__name__, "task_name": running_task.executable.__name__,
"cognee_version": cognee_version,
}, },
) )
@ -46,6 +48,7 @@ async def handle_task(
user_id=user.id, user_id=user.id,
additional_properties={ additional_properties={
"task_name": running_task.executable.__name__, "task_name": running_task.executable.__name__,
"cognee_version": cognee_version,
}, },
) )
except Exception as error: except Exception as error:
@ -58,6 +61,7 @@ async def handle_task(
user_id=user.id, user_id=user.id,
additional_properties={ additional_properties={
"task_name": running_task.executable.__name__, "task_name": running_task.executable.__name__,
"cognee_version": cognee_version,
}, },
) )
raise error raise error

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

@ -4,6 +4,7 @@ from cognee.modules.settings import get_current_settings
from cognee.modules.users.models import User from cognee.modules.users.models import User
from cognee.shared.logging_utils import get_logger from cognee.shared.logging_utils import get_logger
from cognee.shared.utils import send_telemetry from cognee.shared.utils import send_telemetry
from cognee import __version__ as cognee_version
from .run_tasks_base import run_tasks_base from .run_tasks_base import run_tasks_base
from ..tasks.task import Task from ..tasks.task import Task
@ -26,6 +27,7 @@ async def run_tasks_with_telemetry(
user.id, user.id,
additional_properties={ additional_properties={
"pipeline_name": str(pipeline_name), "pipeline_name": str(pipeline_name),
"cognee_version": cognee_version,
} }
| config, | config,
) )
@ -39,6 +41,7 @@ async def run_tasks_with_telemetry(
user.id, user.id,
additional_properties={ additional_properties={
"pipeline_name": str(pipeline_name), "pipeline_name": str(pipeline_name),
"cognee_version": cognee_version,
} }
| config, | config,
) )
@ -54,6 +57,7 @@ async def run_tasks_with_telemetry(
user.id, user.id,
additional_properties={ additional_properties={
"pipeline_name": str(pipeline_name), "pipeline_name": str(pipeline_name),
"cognee_version": cognee_version,
} }
| config, | config,
) )