remove unused imports

src/seer/automation/autofix/components/planner/models.py

@@ -3,7 +3,6 @@
 from pydantic import BaseModel, StringConstraints
 from pydantic_xml import attr, element
 
-from seer.automation.autofix.components.assessment.models import ProblemDiscoveryOutput
 from seer.automation.autofix.components.root_cause.models import RootCauseAnalysisItem
 from seer.automation.component import BaseComponentOutput, BaseComponentRequest
 from seer.automation.models import EventDetails, PromptXmlModel

src/seer/automation/autofix/components/reranker/models.py

@@ -1,7 +1,5 @@
 import json
-import textwrap
 
-from pydantic import BaseModel
 from pydantic_xml import element
 
 from seer.automation.codebase.models import BaseDocumentChunk

src/seer/automation/autofix/steps/create_index_step.py

@@ -1,4 +1,4 @@
-from typing import Any, Type
+from typing import Any
 
 import sentry_sdk
 

src/seer/automation/autofix/tools.py

@@ -1,11 +1,8 @@
-import json
 import logging
 import textwrap
 
 from sentry_sdk.ai.monitoring import ai_track
 
-from seer.automation.agent.client import GptClient
-from seer.automation.agent.models import Message, Usage
 from seer.automation.agent.tools import FunctionTool
 from seer.automation.autofix.autofix_context import AutofixContext
 from seer.automation.autofix.components.retriever import RetrieverRequest
@@ -19,7 +16,6 @@
 from seer.automation.autofix.utils import find_original_snippet
 from seer.automation.codebase.codebase_index import CodebaseIndex
 from seer.automation.models import FileChange
-from seer.automation.state import State
 
 logger = logging.getLogger("autofix")
 

src/seer/automation/codebase/ast.py

@@ -1,5 +1,5 @@
 from pydantic import BaseModel
-from tree_sitter import Node, Tree
+from tree_sitter import Node
 
 
 class AstDeclaration(BaseModel):

src/seer/automation/component.py

@@ -5,7 +5,6 @@
 
 from seer.automation.models import PromptXmlModel
 from seer.automation.pipeline import PipelineContext
-from seer.automation.state import State
 
 
 class BaseComponentRequest(BaseModel):

src/seer/automation/state.py

@@ -2,7 +2,7 @@
 import contextlib
 import dataclasses
 import functools
-from typing import Any, Callable, Generic, Iterator, Type, TypeVar, cast
+from typing import Any, Generic, Iterator, Type, TypeVar, cast
 
 from celery import Task
 from pydantic import BaseModel

src/seer/automation/utils.py

@@ -1,7 +1,6 @@
 import functools
 import logging
 import os
-from typing import Any
 
 import billiard  # type: ignore[import-untyped]
 import torch

src/seer/db.py

@@ -23,7 +23,7 @@
 )
 from sqlalchemy.dialects.postgresql import insert
 from sqlalchemy.ext.asyncio import async_sessionmaker
-from sqlalchemy.orm import DeclarativeBase, Mapped, deferred, mapped_column, sessionmaker
+from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, sessionmaker
 
 
 class Base(DeclarativeBase):

src/seer/generate_schemas.py

@@ -1,6 +1,6 @@
 import json
 import os.path
-from typing import Dict, Generator, Iterable, Iterator, List, Optional, Set, Union
+from typing import Generator, Iterator, List, Optional, Set, Union
 
 root = os.path.abspath(os.path.join(__file__, ".."))
 

src/seer/grouping/grouping.py

@@ -1,9 +1,8 @@
 import difflib
 import logging
-from typing import List, Mapping, Optional
+from typing import List, Optional
 
 import numpy as np
-import pandas as pd
 import sentry_sdk
 import torch
 from pydantic import BaseModel, ValidationInfo, field_validator

src/seer/tasks.py

@@ -13,7 +13,6 @@
 from dateutil.relativedelta import relativedelta
 from sentry_sdk.integrations.asyncio import AsyncioIntegration
 from sqlalchemy import func, select, text
-from sqlalchemy.ext.asyncio import async_sessionmaker
 
 from seer.db import AsyncSession, ProcessRequest
 from seer.task_factory import AsyncTaskFactory, _async_task_factories

src/seer/utils.py

@@ -1,7 +1,6 @@
 import contextlib
 import functools
 import json
-import sys
 import weakref
 from enum import Enum
 from queue import Empty, Full, Queue

tests/automation/agent/test_agent.py

@@ -1,8 +1,8 @@
 import unittest
-from unittest.mock import MagicMock, patch
+from unittest.mock import MagicMock
 
 from seer.automation.agent.agent import GptAgent
-from seer.automation.agent.models import Message, ToolCall
+from seer.automation.agent.models import ToolCall
 from seer.automation.agent.tools import FunctionTool
 
 

tests/automation/autofix/steps/test_create_missing_indexes_chain.py

@@ -5,7 +5,6 @@
 
 from seer.automation.autofix.models import AutofixRequest
 from seer.automation.autofix.steps.create_missing_indexes_chain import CreateMissingIndexesStep
-from seer.automation.autofix.steps.update_index_step import UpdateIndexStep
 from seer.automation.models import RepoDefinition
 from seer.automation.steps import ParallelizedChainStepRequest
 

tests/automation/autofix/test_autofix_context.py

@@ -16,7 +16,7 @@
 from seer.automation.autofix.state import ContinuationState
 from seer.automation.codebase.models import QueryResultDocumentChunk, RepositoryInfo
 from seer.automation.models import FileChange, IssueDetails, SentryEventData
-from seer.automation.state import DbState, LocalMemoryState
+from seer.automation.state import LocalMemoryState
 from seer.db import DbPrIdToAutofixRunIdMapping, Session
 
 

tests/automation/codebase/test_parser.py

@@ -1,5 +1,5 @@
 import unittest
-from unittest.mock import MagicMock, patch
+from unittest.mock import MagicMock
 
 from sentence_transformers import SentenceTransformer
 from tree_sitter import Node, Parser

tests/automation/codebase/test_repo_client.py

@@ -5,7 +5,7 @@
 from pydantic import ValidationError
 
 from seer.automation.codebase.repo_client import RepoClient
-from seer.automation.models import InitializationError, RepoDefinition
+from seer.automation.models import RepoDefinition
 
 
 class TestRepoClient(unittest.TestCase):

tests/automation/test_steps.py

@@ -1,6 +1,5 @@
 import unittest
-from typing import Type
-from unittest.mock import MagicMock, patch
+from unittest.mock import MagicMock
 
 from src.seer.automation.steps import (
     ConditionalStep,

tests/conftest.py

@@ -1,9 +1,7 @@
-import asyncio
 import os
 
 import pytest
 from sqlalchemy import text
-from sqlalchemy.ext.asyncio import create_async_engine
 
 from seer.bootup import CELERY_CONFIG, bootup
 from seer.db import Session, db

tests/generators.py

@@ -1,7 +1,6 @@
 import contextlib
 import dataclasses
 import datetime
-import hashlib
 from typing import Annotated, Iterator
 from unittest import mock
 

tests/test_rpc.py

@@ -12,9 +12,7 @@
 
 import pytest
 from aiohttp import ClientResponseError
-from johen import gen
 from johen.pytest import parametrize
-from pydantic import BaseModel
 from requests.models import HTTPError
 
 from seer.automation.autofix.models import AutofixCompleteArgs, AutofixStepUpdateArgs