Refactor enum values in migration scripts to lowercase

kodart · kodart · commit 751f4594a77e · 2025-11-08T02:00:07.000+05:00
- Updated enum values in multiple migration scripts to use lowercase for consistency with OpenAI API standards.
- Adjusted `finish_reason`, `item_type`, `score_type`, `evaluation_status`, and `optimization_status` enums accordingly.
diff --git a/backend/migrations/versions/2ab7f18e5ff7_restructure_traces_to_match_openai_api.py b/backend/migrations/versions/2ab7f18e5ff7_restructure_traces_to_match_openai_api.py
@@ -1,7 +1,7 @@
-"""restructure_traces_to_match_openai_api
+"""Restructure traces to match OpenAI API.
 
 Revision ID: 2ab7f18e5ff7
-Revises: 
+Revises:
 Create Date: 2025-10-17 15:28:15.330107
 
 """
@@ -19,27 +19,27 @@
 
 
 finish_reason_enum = postgresql.ENUM(
-    "STOP",
-    "LENGTH",
-    "TOOL_CALLS",
-    "CONTENT_FILTER",
-    "FUNCTION_CALL",
-    "ERROR",
+    "stop",
+    "length",
+    "tool_calls",
+    "content_filter",
+    "function_call",
+    "error",
     name="finish_reason",
     create_type=False,
 )
 
 item_type_enum = postgresql.ENUM(
-    "MESSAGE",
-    "FUNCTION_CALL",
-    "FUNCTION_RESULT",
-    "TOOL_CALL",
-    "TOOL_RESULT",
-    "IMAGE",
-    "VIDEO",
-    "AUDIO",
-    "MCP_TOOL_CALL",
-    "MCP_TOOL_RESULT",
+    "message",
+    "function_call",
+    "function_result",
+    "tool_call",
+    "tool_result",
+    "image",
+    "video",
+    "audio",
+    "mcp_tool_call",
+    "mcp_tool_result",
     name="item_type",
     create_type=False,
 )
diff --git a/backend/migrations/versions/53914fdc7590_add_optimization_model.py b/backend/migrations/versions/53914fdc7590_add_optimization_model.py
@@ -5,54 +5,88 @@
 Create Date: 2025-10-31 20:35:10.057534
 
 """
-from typing import Sequence, Union
+from collections.abc import Sequence
 
-from alembic import op
 import sqlalchemy as sa
+from alembic import op
 from sqlalchemy.dialects import postgresql
 
 # revision identifiers, used by Alembic.
-revision: str = '53914fdc7590'
-down_revision: Union[str, Sequence[str], None] = '37309a4b1b2f'
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+revision: str = "53914fdc7590"
+down_revision: str | Sequence[str] | None = "37309a4b1b2f"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+optimization_status_enum = postgresql.ENUM(
+    "pending",
+    "running",
+    "completed",
+    "failed",
+    name="optimization_status",
+    create_type=False,
+)
 
 
 def upgrade() -> None:
     """Upgrade schema."""
     # ### commands auto generated by Alembic - please adjust! ###
-    op.create_table('optimization',
-    sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
-    sa.Column('task_id', sa.Integer(), nullable=False),
-    sa.Column('status', sa.Enum('PENDING', 'RUNNING', 'COMPLETED', 'FAILED', name='optimization_status'), nullable=False),
-    sa.Column('started_at', sa.DateTime(timezone=True), nullable=True),
-    sa.Column('completed_at', sa.DateTime(timezone=True), nullable=True),
-    sa.Column('error', sa.Text(), nullable=True),
-    sa.Column('max_iterations', sa.Integer(), nullable=False),
-    sa.Column('changeable_fields', sa.JSON().with_variant(postgresql.JSONB(astext_type=sa.Text()), 'postgresql'), nullable=False),
-    sa.Column('max_consecutive_no_improvements', sa.Integer(), nullable=False),
-    sa.Column('iterations_run', sa.Integer(), nullable=False),
-    sa.Column('current_iteration', sa.Integer(), nullable=True),
-    sa.Column('best_implementation_id', sa.Integer(), nullable=True),
-    sa.Column('best_score', sa.Float(), nullable=True),
-    sa.Column('iterations', sa.JSON().with_variant(postgresql.JSONB(astext_type=sa.Text()), 'postgresql'), nullable=False),
-    sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
-    sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
-    sa.ForeignKeyConstraint(['best_implementation_id'], ['implementation.id'], name=op.f('fk_optimization_best_implementation_id_implementation'), ondelete='SET NULL'),
-    sa.ForeignKeyConstraint(['task_id'], ['task.id'], name=op.f('fk_optimization_task_id_task'), ondelete='CASCADE'),
-    sa.PrimaryKeyConstraint('id', name=op.f('pk_optimization'))
+    bind = op.get_bind()
+    optimization_status_enum.create(bind, checkfirst=True)
+
+    op.create_table(
+        "optimization",
+        sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
+        sa.Column("task_id", sa.Integer(), nullable=False),
+        sa.Column("status", optimization_status_enum, nullable=False),
+        sa.Column("started_at", sa.DateTime(timezone=True), nullable=True),
+        sa.Column("completed_at", sa.DateTime(timezone=True), nullable=True),
+        sa.Column("error", sa.Text(), nullable=True),
+        sa.Column("max_iterations", sa.Integer(), nullable=False),
+        sa.Column(
+            "changeable_fields",
+            sa.JSON().with_variant(postgresql.JSONB(astext_type=sa.Text()), "postgresql"),
+            nullable=False,
+        ),
+        sa.Column("max_consecutive_no_improvements", sa.Integer(), nullable=False),
+        sa.Column("iterations_run", sa.Integer(), nullable=False),
+        sa.Column("current_iteration", sa.Integer(), nullable=True),
+        sa.Column("best_implementation_id", sa.Integer(), nullable=True),
+        sa.Column("best_score", sa.Float(), nullable=True),
+        sa.Column(
+            "iterations",
+            sa.JSON().with_variant(postgresql.JSONB(astext_type=sa.Text()), "postgresql"),
+            nullable=False,
+        ),
+        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
+        sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
+        sa.ForeignKeyConstraint(
+            ["best_implementation_id"],
+            ["implementation.id"],
+            name=op.f("fk_optimization_best_implementation_id_implementation"),
+            ondelete="SET NULL",
+        ),
+        sa.ForeignKeyConstraint(
+            ["task_id"],
+            ["task.id"],
+            name=op.f("fk_optimization_task_id_task"),
+            ondelete="CASCADE",
+        ),
+        sa.PrimaryKeyConstraint("id", name=op.f("pk_optimization")),
     )
-    op.create_index('ix_optimization_started_at', 'optimization', ['started_at'], unique=False)
-    op.create_index('ix_optimization_status', 'optimization', ['status'], unique=False)
-    op.create_index('ix_optimization_task_id', 'optimization', ['task_id'], unique=False)
+    op.create_index("ix_optimization_started_at", "optimization", ["started_at"], unique=False)
+    op.create_index("ix_optimization_status", "optimization", ["status"], unique=False)
+    op.create_index("ix_optimization_task_id", "optimization", ["task_id"], unique=False)
     # ### end Alembic commands ###
 
 
 def downgrade() -> None:
     """Downgrade schema."""
     # ### commands auto generated by Alembic - please adjust! ###
-    op.drop_index('ix_optimization_task_id', table_name='optimization')
-    op.drop_index('ix_optimization_status', table_name='optimization')
-    op.drop_index('ix_optimization_started_at', table_name='optimization')
-    op.drop_table('optimization')
+    op.drop_index("ix_optimization_task_id", table_name="optimization")
+    op.drop_index("ix_optimization_status", table_name="optimization")
+    op.drop_index("ix_optimization_started_at", table_name="optimization")
+    op.drop_table("optimization")
+    bind = op.get_bind()
+    optimization_status_enum.drop(bind, checkfirst=True)
     # ### end Alembic commands ###
diff --git a/backend/migrations/versions/7a8b9c0d1e2f_add_execution_result_table.py b/backend/migrations/versions/7a8b9c0d1e2f_add_execution_result_table.py
@@ -17,12 +17,12 @@
 branch_labels: str | Sequence[str] | None = None
 depends_on: str | Sequence[str] | None = None
 finish_reason_enum = postgresql.ENUM(
-    "STOP",
-    "LENGTH",
-    "TOOL_CALLS",
-    "CONTENT_FILTER",
-    "FUNCTION_CALL",
-    "ERROR",
+    "stop",
+    "length",
+    "tool_calls",
+    "content_filter",
+    "function_call",
+    "error",
     name="finish_reason",
     create_type=False,
 )
diff --git a/backend/migrations/versions/cb5cec1b033d_add_grader_and_grade.py b/backend/migrations/versions/cb5cec1b033d_add_grader_and_grade.py
@@ -19,8 +19,8 @@
 
 
 score_type_enum = postgresql.ENUM(
-    "FLOAT",
-    "BOOLEAN",
+    "float",
+    "boolean",
     name="score_type",
     create_type=False,
 )
diff --git a/backend/migrations/versions/e85591d36b7d_evaluation_and_testing.py b/backend/migrations/versions/e85591d36b7d_evaluation_and_testing.py
@@ -19,10 +19,10 @@
 
 
 evaluation_status_enum = postgresql.ENUM(
-    "PENDING",
-    "RUNNING",
-    "COMPLETED",
-    "FAILED",
+    "pending",
+    "running",
+    "completed",
+    "failed",
     name="evaluation_status",
     create_type=False,
 )

Original file line number	Diff line number	Diff line change
`@@ -19,8 +19,8 @@`
`19`	`19`
`20`	`20`
`21`	`21`	`score_type_enum = postgresql.ENUM(`
`22`		`- "FLOAT",`
`23`		`- "BOOLEAN",`
	`22`	`+ "float",`
	`23`	`+ "boolean",`
`24`	`24`	`name="score_type",`
`25`	`25`	`create_type=False,`
`26`	`26`	`)`