From a874988db68c7024f4f359916fc939974759f495 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Mon, 19 May 2025 13:18:36 +0200
Subject: [PATCH] fix: Fixes pipeline run status migration (#838)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

…exist case

<!-- .github/pull_request_template.md -->

## Description
Fixes pipeline run status migration

## 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.
---
 .../1d0bb7fede17_add_pipeline_run_status.py        | 14 ++++----------
 alembic/versions/482cd6517ce4_add_default_user.py  |  5 ++++-
 2 files changed, 8 insertions(+), 11 deletions(-)

diff --git a/alembic/versions/1d0bb7fede17_add_pipeline_run_status.py b/alembic/versions/1d0bb7fede17_add_pipeline_run_status.py
index fe1fb689b..5e0e99dba 100644
--- a/alembic/versions/1d0bb7fede17_add_pipeline_run_status.py
+++ b/alembic/versions/1d0bb7fede17_add_pipeline_run_status.py
@@ -1,9 +1,8 @@
-"""Add  pipeline run status
+"""Add pipeline run status
 
 Revision ID: 1d0bb7fede17
 Revises: 482cd6517ce4
 Create Date: 2025-05-19 10:58:15.993314
-
 """
 
 from typing import Sequence, Union
@@ -18,21 +17,16 @@ from cognee.modules.pipelines.models.PipelineRun import PipelineRun, PipelineRun
 revision: str = "1d0bb7fede17"
 down_revision: Union[str, None] = "482cd6517ce4"
 branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = "482cd6517ce4"
 
 
-async def upgrade() -> None:
+def upgrade() -> None:
     db_engine = get_relational_engine()
 
     if db_engine.engine.dialect.name == "postgresql":
         op.execute(
-            f"ALTER TYPE {PipelineRun.status} ADD VALUE IF NOT EXISTS '{str(PipelineRunStatus.DATASET_PROCESSING_INITIATED)}'"
+            "ALTER TYPE pipelinerunstatus ADD VALUE IF NOT EXISTS 'DATASET_PROCESSING_INITIATED'"
         )
-    else:
-        op.execute(f"""
-            ALTER TABLE {PipelineRun}
-            MODIFY COLUMN {PipelineRun.status} ENUM('{PipelineRunStatus.DATASET_PROCESSING_INITIATED}', '{PipelineRunStatus.DATASET_PROCESSING_STARTED}', '{PipelineRunStatus.DATASET_PROCESSING_COMPLETED}', '{PipelineRunStatus.DATASET_PROCESSING_ERRORED}') NOT NULL
-        """)
 
 
 def downgrade() -> None:
diff --git a/alembic/versions/482cd6517ce4_add_default_user.py b/alembic/versions/482cd6517ce4_add_default_user.py
index 92429e1e4..078dce7fd 100644
--- a/alembic/versions/482cd6517ce4_add_default_user.py
+++ b/alembic/versions/482cd6517ce4_add_default_user.py
@@ -21,7 +21,10 @@ depends_on: Union[str, Sequence[str], None] = "8057ae7329c2"
 
 
 def upgrade() -> None:
-    await_only(create_default_user())
+    try:
+        await_only(create_default_user())
+    except Exception:
+        pass
 
 
 def downgrade() -> None: