add liifespan of an agent

Author: Shreyas-Microsoft

src/backend/app.py

@@ -1,13 +1,23 @@
 """Create and configure the FastAPI application."""
+from contextlib import asynccontextmanager
+
 from api.api_routes import router as backend_router
 
+from azure.identity.aio import DefaultAzureCredential
+
+from common.config.config import app_config
 from common.logger.app_logger import AppLogger
 
 from dotenv import load_dotenv
 
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 
+from semantic_kernel.agents.azure_ai.azure_ai_agent import AzureAIAgent  # pylint: disable=E0611
+
+from sql_agents.agents.agent_config import AgentBaseConfig
+from sql_agents.helpers.agents_manager import SqlAgents
+
 import uvicorn
 # from agent_services.agents_routes import router as agents_router
 
@@ -17,10 +27,77 @@
 # Configure logging
 logger = AppLogger("app")
 
+# Global variables for agents
+sql_agents: SqlAgents = None
+azure_client = None
+
+
+def get_sql_agents() -> SqlAgents:
+    """Get the global SQL agents instance."""
+    return sql_agents
+
+
+async def update_agent_config(convert_from: str, convert_to: str):
+    """Update the global agent configuration for different SQL conversion types."""
+    global sql_agents
+    if sql_agents and sql_agents.agent_config:
+        sql_agents.agent_config.sql_from = convert_from
+        sql_agents.agent_config.sql_to = convert_to
+        logger.logger.info(f"Updated agent configuration: {convert_from} -> {convert_to}")
+    else:
+        logger.logger.warning("SQL agents not initialized, cannot update configuration")
+
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    """Manage application lifespan - startup and shutdown."""
+    global sql_agents, azure_client
+    
+    # Startup
+    try:
+        logger.logger.info("Initializing SQL agents...")
+        
+        # Create Azure credentials and client
+        creds = DefaultAzureCredential()
+        azure_client = AzureAIAgent.create_client(
+            credential=creds, 
+            endpoint=app_config.ai_project_endpoint
+        )
+        
+        # Setup agent configuration with default conversion settings
+        agent_config = AgentBaseConfig(
+            project_client=azure_client,
+            sql_from="informix",  # Default source dialect
+            sql_to="tsql"         # Default target dialect
+        )
+        
+        # Create SQL agents
+        sql_agents = await SqlAgents.create(agent_config)
+        logger.logger.info("SQL agents initialized successfully.")
+        
+    except Exception as exc:
+        logger.logger.error("Failed to initialize SQL agents: %s", exc)
+        # Don't raise the exception to allow the app to start even if agents fail
+    
+    yield  # Application runs here
+    
+    # Shutdown
+    try:
+        if sql_agents:
+            logger.logger.info("Cleaning up SQL agents...")
+            await sql_agents.delete_agents()
+            logger.logger.info("SQL agents cleaned up successfully.")
+        
+        if azure_client:
+            await azure_client.close()
+            
+    except Exception as exc:
+        logger.logger.error("Error during agent cleanup: %s", exc)
+
 
 def create_app() -> FastAPI:
     """Create and return the FastAPI application instance."""
-    app = FastAPI(title="Code Gen Accelerator", version="1.0.0")
+    app = FastAPI(title="Code Gen Accelerator", version="1.0.0", lifespan=lifespan)
 
     # Configure CORS
     app.add_middleware(

src/backend/sql_agents/process_batch.py

@@ -7,10 +7,8 @@
 import logging
 
 from api.status_updates import send_status_update
+from app import get_sql_agents, update_agent_config
 
-from azure.identity.aio import DefaultAzureCredential
-
-from common.config.config import app_config
 from common.models.api import (
     FileProcessUpdate,
     FileRecord,
@@ -23,14 +21,10 @@
 
 from fastapi import HTTPException
 
-
-from semantic_kernel.agents.azure_ai.azure_ai_agent import AzureAIAgent  # pylint: disable=E0611
 from semantic_kernel.contents import AuthorRole
 from semantic_kernel.exceptions.service_exceptions import ServiceResponseException
 
-from sql_agents.agents.agent_config import AgentBaseConfig
 from sql_agents.convert_script import convert_script
-from sql_agents.helpers.agents_manager import SqlAgents
 from sql_agents.helpers.models import AgentType
 from sql_agents.helpers.utils import is_text
 
@@ -57,22 +51,20 @@ async def process_batch_async(
     except Exception as exc:
         logger.error("Error updating batch status. %s", exc)
 
-    # Add client and auto cleanup
-    async with (
-        DefaultAzureCredential() as creds,
-        AzureAIAgent.create_client(credential=creds, endpoint=app_config.ai_project_endpoint) as client,
-    ):
-
-        # setup all agent settings and agents per batch
-        agent_config = AgentBaseConfig(
-            project_client=client, sql_from=convert_from, sql_to=convert_to
-        )
-        sql_agents = await SqlAgents.create(agent_config)
-
-        # Walk through each file name and retrieve it from blob storage
-        # Send file to the agents for processing
-        # Send status update to the client of type in progress, completed, or failed
-        for file in batch_files:
+    # Get the global SQL agents instance
+    sql_agents = get_sql_agents()
+    if not sql_agents:
+        logger.error("SQL agents not initialized. Application may not have started properly.")
+        await batch_service.update_batch(batch_id, ProcessStatus.FAILED)
+        return
+
+    # Update agent configuration for this batch's conversion requirements
+    await update_agent_config(convert_from, convert_to)
+
+    # Walk through each file name and retrieve it from blob storage
+    # Send file to the agents for processing
+    # Send status update to the client of type in progress, completed, or failed
+    for file in batch_files:
             # Get the file from blob storage
             try:
                 file_record = FileRecord.fromdb(file)
@@ -145,16 +137,14 @@ async def process_batch_async(
                 # insert data base write to file record stating invalid file
                 await process_error(exc, file_record, batch_service)
 
-        # Cleanup the agents
-        await sql_agents.delete_agents()
-
-        try:
-            await batch_service.batch_files_final_update(batch_id)
-            await batch_service.update_batch(batch_id, ProcessStatus.COMPLETED)
-        except Exception as exc:
-            await batch_service.update_batch(batch_id, ProcessStatus.FAILED)
-            logger.error("Error updating batch status. %s", exc)
-        logger.info("Batch processing complete.")
+    # Update batch status to completed or failed
+    try:
+        await batch_service.batch_files_final_update(batch_id)
+        await batch_service.update_batch(batch_id, ProcessStatus.COMPLETED)
+    except Exception as exc:
+        await batch_service.update_batch(batch_id, ProcessStatus.FAILED)
+        logger.error("Error updating batch status. %s", exc)
+    logger.info("Batch processing complete.")
 
 
 async def process_error(