🛠️ apply pre-commit fixes

Author: getsantry[bot]

src/migrations/versions/d962b4664c6a_migration.py

@@ -5,42 +5,51 @@
 Create Date: 2025-03-17 18:08:58.255287
 
 """
-from alembic import op
-import sqlalchemy as sa
 
+import sqlalchemy as sa
+from alembic import op
 
 # revision identifiers, used by Alembic.
-revision = 'd962b4664c6a'
-down_revision = 'cac994b711d3'
+revision = "d962b4664c6a"
+down_revision = "cac994b711d3"
 branch_labels = None
 depends_on = None
 
 
 def upgrade():
     # ### commands auto generated by Alembic - please adjust! ###
-    op.create_table('codegen_unit_test_generation_pr_context_to_run_id',
-    sa.Column('id', sa.Integer(), nullable=False),
-    sa.Column('provider', sa.String(), nullable=False),
-    sa.Column('owner', sa.String(), nullable=False),
-    sa.Column('pr_id', sa.BigInteger(), nullable=False),
-    sa.Column('repo', sa.String(), nullable=False),
-    sa.Column('run_id', sa.Integer(), nullable=False),
-    sa.Column('iterations', sa.Integer(), nullable=False),
-    sa.Column('original_pr_url', sa.String(), nullable=False),
-    sa.ForeignKeyConstraint(['run_id'], ['run_state.id'], ondelete='CASCADE'),
-    sa.PrimaryKeyConstraint('id'),
-    sa.UniqueConstraint('provider', 'pr_id', 'repo', 'owner', 'original_pr_url')
+    op.create_table(
+        "codegen_unit_test_generation_pr_context_to_run_id",
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("provider", sa.String(), nullable=False),
+        sa.Column("owner", sa.String(), nullable=False),
+        sa.Column("pr_id", sa.BigInteger(), nullable=False),
+        sa.Column("repo", sa.String(), nullable=False),
+        sa.Column("run_id", sa.Integer(), nullable=False),
+        sa.Column("iterations", sa.Integer(), nullable=False),
+        sa.Column("original_pr_url", sa.String(), nullable=False),
+        sa.ForeignKeyConstraint(["run_id"], ["run_state.id"], ondelete="CASCADE"),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("provider", "pr_id", "repo", "owner", "original_pr_url"),
     )
-    with op.batch_alter_table('codegen_unit_test_generation_pr_context_to_run_id', schema=None) as batch_op:
-        batch_op.create_index('ix_autofix_repo_owner_pr_id', ['owner', 'repo', 'pr_id', 'original_pr_url'], unique=False)
+    with op.batch_alter_table(
+        "codegen_unit_test_generation_pr_context_to_run_id", schema=None
+    ) as batch_op:
+        batch_op.create_index(
+            "ix_autofix_repo_owner_pr_id",
+            ["owner", "repo", "pr_id", "original_pr_url"],
+            unique=False,
+        )
 
     # ### end Alembic commands ###
 
 
 def downgrade():
     # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('codegen_unit_test_generation_pr_context_to_run_id', schema=None) as batch_op:
-        batch_op.drop_index('ix_autofix_repo_owner_pr_id')
+    with op.batch_alter_table(
+        "codegen_unit_test_generation_pr_context_to_run_id", schema=None
+    ) as batch_op:
+        batch_op.drop_index("ix_autofix_repo_owner_pr_id")
 
-    op.drop_table('codegen_unit_test_generation_pr_context_to_run_id')
+    op.drop_table("codegen_unit_test_generation_pr_context_to_run_id")
     # ### end Alembic commands ###

src/seer/automation/codegen/prompts.py

@@ -327,10 +327,10 @@ class RetryUnitTestPrompts:
     @staticmethod
     def format_continue_unit_tests_prompt(code_coverage_info: str, test_result_info: str):
         return textwrap.dedent(
-            """\            
+            """\
             The tests you have generated so far are not sufficient to cover all the changes in the codebase. You need to continue generating unit tests to address the gaps in coverage and fix any failing tests.
 
-            To help you with this, you have access to code coverage information at a file level attached as a JSON in addtion to test result information also in a JSON format. 
+            To help you with this, you have access to code coverage information at a file level attached as a JSON in addtion to test result information also in a JSON format.
 
             Using the information and instructions provided, update the unit tests to ensure robust code coverage as well as fix any failing tests. Use the exact same format you used previously to regenerate tests. Your changes will be appended as a new commit to the branch of the existing PR.
 

src/seer/automation/codegen/retry_unit_test_github_pr_creator.py

@@ -1,6 +1,8 @@
 import logging
 import time
+
 from github.PullRequest import PullRequest
+
 from seer.automation.codebase.repo_client import RepoClient
 from seer.automation.models import FileChange
 from seer.db import DbPrContextToUnitTestGenerationRunIdMapping, Session

src/seer/automation/codegen/retry_unittest_coding_component.py

@@ -6,13 +6,13 @@
 from integrations.codecov.codecov_client import CodecovClient
 from seer.automation.agent.agent import AgentConfig, LlmAgent, RunConfig
 from seer.automation.agent.client import AnthropicProvider, LlmClient
-from seer.automation.autofix.tools import BaseTools
 from seer.automation.autofix.components.coding.models import PlanStepsPromptXml
 from seer.automation.autofix.components.coding.utils import (
     task_to_file_change,
     task_to_file_create,
     task_to_file_delete,
 )
+from seer.automation.autofix.tools import BaseTools
 from seer.automation.codebase.repo_client import RepoClientType
 from seer.automation.codegen.codegen_context import CodegenContext
 from seer.automation.codegen.models import CodeUnitTestOutput, CodeUnitTestRequest

src/seer/automation/codegen/unit_test_github_pr_creator.py

@@ -5,10 +5,7 @@
 
 from seer.automation.codebase.repo_client import RepoClient
 from seer.automation.models import FileChange
-from seer.db import (
-    DbPrContextToUnitTestGenerationRunIdMapping,
-    Session,
-)
+from seer.db import DbPrContextToUnitTestGenerationRunIdMapping, Session
 
 logger = logging.getLogger(__name__)
 

tests/automation/codegen/test_retry_unit_test_step.py

@@ -1,8 +1,9 @@
 import unittest
 from unittest.mock import MagicMock, patch
-from seer.automation.models import RepoDefinition
-from seer.automation.codegen.retry_unittest_step import RetryUnittestStep
+
 from seer.automation.codegen.retry_unit_test_github_pr_creator import RetryUnitTestGithubPrUpdater
+from seer.automation.codegen.retry_unittest_step import RetryUnittestStep
+from seer.automation.models import RepoDefinition
 from seer.automation.state import DbStateRunTypes