agent folder organizing

Author: marktayl1

src/backend/api/api_routes.py

@@ -4,10 +4,6 @@
 import io
 import zipfile
 
-from api.auth.auth_utils import get_authenticated_user
-from api.status_updates import app_connection_manager, close_connection
-from common.logger.app_logger import AppLogger
-from common.services.batch_service import BatchService
 from fastapi import (
     APIRouter,
     File,
@@ -20,11 +16,16 @@
 )
 from fastapi.responses import Response
 
+from api.auth.auth_utils import get_authenticated_user
+from api.status_updates import app_connection_manager, close_connection
+from common.logger.app_logger import AppLogger
+from common.services.batch_service import BatchService
+
 router = APIRouter()
 logger = AppLogger("APIRoutes")
 
 # start processing the batch
-from sql_agents_start import process_batch_async
+from sql_agents.sql_agents_start import process_batch_async
 
 
 @router.post("/start-processing")

src/backend/sql_agents/agent_base.py …/backend/sql_agents/agents/agent_base.pysrc/backend/sql_agents/agent_base.py renamed to src/backend/sql_agents/agents/agent_base.py src/backend/sql_agents/agent_base.py renamed to src/backend/sql_agents/agents/agent_base.py

@@ -11,7 +11,7 @@
 from semantic_kernel.agents.azure_ai.azure_ai_agent import AzureAIAgent
 from semantic_kernel.functions import KernelArguments
 
-from sql_agents.agent_config import AgentBaseConfig
+from sql_agents.agents.agent_config import AgentBaseConfig
 from sql_agents.helpers.models import AgentType
 from sql_agents.helpers.utils import get_prompt
 

src/backend/sql_agents/agent_config.py …ackend/sql_agents/agents/agent_config.pysrc/backend/sql_agents/agent_config.py renamed to src/backend/sql_agents/agents/agent_config.py src/backend/sql_agents/agent_config.py renamed to src/backend/sql_agents/agents/agent_config.py

@@ -1,18 +1,18 @@
 """Factory for creating SQL migration agents."""
 
 import logging
-from typing import Any, Dict, Optional, Type, TypeVar
+from typing import Type, TypeVar
 
 from semantic_kernel.agents.azure_ai.azure_ai_agent import AzureAIAgent
 
-from sql_agents.agent_base import BaseSQLAgent
-from sql_agents.agent_config import AgentBaseConfig
-from sql_agents.fixer.agent import FixerAgent
+from sql_agents.agents.agent_base import BaseSQLAgent
+from sql_agents.agents.agent_config import AgentBaseConfig
+from sql_agents.agents.fixer.agent import FixerAgent
+from sql_agents.agents.migrator.agent import MigratorAgent
+from sql_agents.agents.picker.agent import PickerAgent
+from sql_agents.agents.semantic_verifier.agent import SemanticVerifierAgent
+from sql_agents.agents.syntax_checker.agent import SyntaxCheckerAgent
 from sql_agents.helpers.models import AgentType
-from sql_agents.migrator.agent import MigratorAgent
-from sql_agents.picker.agent import PickerAgent
-from sql_agents.semantic_verifier.agent import SemanticVerifierAgent
-from sql_agents.syntax_checker.agent import SyntaxCheckerAgent
 
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.DEBUG)

src/backend/sql_agents/agent_factory.py …ckend/sql_agents/agents/agent_factory.pysrc/backend/sql_agents/agent_factory.py renamed to src/backend/sql_agents/agents/agent_factory.py src/backend/sql_agents/agent_factory.py renamed to src/backend/sql_agents/agents/agent_factory.py

@@ -2,8 +2,8 @@
 
 import logging
 
-from sql_agents.agent_base import BaseSQLAgent
-from sql_agents.fixer.response import FixerResponse
+from sql_agents.agents.agent_base import BaseSQLAgent
+from sql_agents.agents.fixer.response import FixerResponse
 from sql_agents.helpers.models import AgentType
 
 logger = logging.getLogger(__name__)

src/backend/sql_agents/fixer/__init__.py …kend/sql_agents/agents/fixer/__init__.pysrc/backend/sql_agents/fixer/__init__.py renamed to src/backend/sql_agents/agents/fixer/__init__.py src/backend/sql_agents/fixer/__init__.py renamed to src/backend/sql_agents/agents/fixer/__init__.py

@@ -4,8 +4,8 @@
 
 from semantic_kernel.agents.azure_ai.azure_ai_agent import AzureAIAgent
 
-from sql_agents.agent_config import AgentBaseConfig
-from sql_agents.agent_factory import SQLAgentFactory
+from sql_agents.agents.agent_config import AgentBaseConfig
+from sql_agents.agents.agent_factory import SQLAgentFactory
 from sql_agents.helpers.models import AgentType
 
 logger = logging.getLogger(__name__)