daed8d51f5
<!-- .github/pull_request_template.md --> ## Description <!-- Provide a clear description of the changes in this PR --> ## DCO Affirmation I affirm that all code in every commit of this pull request conforms to the terms of the Topoteretes Developer Certificate of Origin.
41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
from uuid import UUID
|
|
from cognee.infrastructure.databases.relational import get_relational_engine
|
|
from cognee.modules.data.models import Data
|
|
from cognee.modules.pipelines.models import PipelineRun, PipelineRunStatus
|
|
from typing import Any
|
|
|
|
|
|
async def log_pipeline_run_error(
|
|
pipeline_run_id: UUID,
|
|
pipeline_id: str,
|
|
pipeline_name: str,
|
|
dataset_id: UUID,
|
|
data: Any,
|
|
e: Exception,
|
|
):
|
|
if not data:
|
|
data_info = "None"
|
|
elif isinstance(data, list) and all(isinstance(item, Data) for item in data):
|
|
data_info = [str(item.id) for item in data]
|
|
else:
|
|
data_info = str(data)
|
|
|
|
pipeline_run = PipelineRun(
|
|
pipeline_run_id=pipeline_run_id,
|
|
pipeline_name=pipeline_name,
|
|
pipeline_id=pipeline_id,
|
|
status=PipelineRunStatus.DATASET_PROCESSING_ERRORED,
|
|
dataset_id=dataset_id,
|
|
run_info={
|
|
"data": data_info,
|
|
"error": str(e),
|
|
},
|
|
)
|
|
|
|
db_engine = get_relational_engine()
|
|
|
|
async with db_engine.get_async_session() as session:
|
|
session.add(pipeline_run)
|
|
await session.commit()
|
|
|
|
return pipeline_run
|