running in terminal with background task - added temp scenario examples

Author: marktayl1

data/agent_teams/hr.json

@@ -28,9 +28,7 @@
 from middleware.health_check import HealthCheckMiddleware
 from v3.api.router import app_v3
 # Semantic Kernel imports
-from v3.config.settings import orchestration_config
-from v3.magentic_agents.magentic_agent_factory import (cleanup_all_agents,
-                                                       get_agents)
+from v3.orchestration.orchestration_manager import OrchestrationManager
 
 # Check if the Application Insights Instrumentation Key is set in the environment variables
 connection_string = config.APPLICATIONINSIGHTS_CONNECTION_STRING
@@ -60,14 +58,6 @@
     logging.WARNING
 )
 
-
-# @asynccontextmanager
-# async def lifespan(app: FastAPI):
-#     """Lifespan event handler to create and clean up agents."""
-#     config.agents = await get_agents()
-#     yield
-#     await cleanup_all_agents()
-
 # Initialize the FastAPI app
 app = FastAPI()
 
@@ -671,7 +661,7 @@ async def get_plans(
         raise HTTPException(status_code=400, detail="no user")
 
     # Initialize agent team for this user session
-    await orchestration_config.get_current_orchestration(user_id=user_id)
+    await OrchestrationManager.get_current_orchestration(user_id=user_id)
 
     # Initialize memory context
     memory_store = await DatabaseFactory.get_database(user_id=user_id)

data/agent_teams/marketing.json

@@ -71,6 +71,10 @@ def __init__(self):
         # Example: http://127.0.0.1:8000/mcp
         self.MCP_SERVER_ENDPOINT = self._get_optional("MCP_SERVER_ENDPOINT")
 
+        test_team_json = self._get_optional("TEST_TEAM_JSON")
+
+        self.AGENT_TEAM_FILE = f"../../data/agent_teams/{test_team_json}.json"
+
         # Cached clients and resources
         self._azure_credentials = None
         self._cosmos_client = None

data/agent_teams/oil&gas.json

@@ -9,14 +9,13 @@
                                            Plan, PlanStatus)
 from common.utils.event_utils import track_event_if_configured
 from common.utils.utils_kernel import rai_success, rai_validate_team_config
-from fastapi import (APIRouter, Depends, File, HTTPException, Request,
-                     UploadFile)
+from fastapi import (APIRouter, BackgroundTasks, Depends, FastAPI, File,
+                     HTTPException, Request, UploadFile, WebSocket,
+                     WebSocketDisconnect)
 from kernel_agents.agent_factory import AgentFactory
 from semantic_kernel.agents.runtime import InProcessRuntime
 from v3.common.services.team_service import TeamService
-from v3.config.settings import orchestration_config
-from v3.models.models import MPlan, MStep
-from v3.models.orchestration_models import AgentType
+from v3.orchestration.orchestration_manager import OrchestrationManager
 
 app_v3 = APIRouter(
     prefix="/api/v3",
@@ -25,7 +24,7 @@
 
 # To do: change endpoint to process request
 @app_v3.post("/create_plan")
-async def process_request(input_task: InputTask, request: Request):
+async def process_request(background_tasks: BackgroundTasks, input_task: InputTask, request: Request):
     """
     Create a new plan without full processing.
 
@@ -121,59 +120,23 @@ async def process_request(input_task: InputTask, request: Request):
         input_task.session_id = str(uuid.uuid4())
 
     try:
-        # Initialize memory store
-        memory_store = await DatabaseFactory.get_database(user_id=user_id)
-
-        # Create a new Plan object
-        plan = MPlan()
-        #     session_id=input_task.session_id,
-        #     team_id=input_task.team_id,
-        #     user_id=user_id,
-        #     initial_goal=input_task.description,
-        #     overall_status=PlanStatus.in_progress,
-        #     source=AgentType.PLANNER.value,
-        # )
-
-        # setup and call the magentic orchestration
-        magentic_orchestration = await orchestration_config.get_current_orchestration(user_id)
-
-        runtime = InProcessRuntime()
-        runtime.start()
-
-        # invoke returns immediately, wait on result.get
-        orchestration_result = await magentic_orchestration.invoke(task=input_task.description,runtime=runtime)
-        team_result = await orchestration_result.get()
-
-        # Save the plan to the database
-        await memory_store.add_plan(plan)
-
-        # Log successful plan creation
-        track_event_if_configured(
-            "PlanCreated",
-            {
-                "status": f"Plan created with ID: {plan.id}",
-                "session_id": input_task.session_id,
-                "plan_id": plan.id,
-                "description": input_task.description,
-            },
-        )
+        background_tasks.add_task(OrchestrationManager.run_orchestration, user_id, input_task)
 
         return {
-            "plan_id": plan.id,
-            "status": "Plan created successfully",
+            "status": "Request started successfully",
             "session_id": input_task.session_id,
         }
 
     except Exception as e:
         track_event_if_configured(
-            "CreatePlanError",
+            "RequestStartFailed",
             {
                 "session_id": input_task.session_id,
                 "description": input_task.description,
                 "error": str(e),
             },
         )
-        raise HTTPException(status_code=400, detail=f"Error creating plan: {e}") from e
+        raise HTTPException(status_code=400, detail=f"Error starting request: {e}") from e
 
 
 @app_v3.post("/upload_team_config")

data/agent_teams/retail.json

@@ -3,15 +3,10 @@
 Handles Azure OpenAI, MCP, and environment setup.
 """
 
-import os
-
 from common.config.app_config import config
 from semantic_kernel.agents.orchestration.magentic import MagenticOrchestration
 from semantic_kernel.connectors.ai.open_ai import (
     AzureChatCompletion, OpenAIChatPromptExecutionSettings)
-from v3.magentic_agents.magentic_agent_factory import (cleanup_all_agents,
-                                                       get_agents)
-from v3.orchestration.manager import init_orchestration
 
 
 class AzureConfig:
@@ -64,13 +59,13 @@ class OrchestrationConfig:
 
     def __init__(self):
         self._orchestrations = {}
+        self.plans = {}       # job_id -> current plan
+        self.approvals = {}   # job_id -> True/False/None
+        self.sockets = {}     # job_id -> WebSocket
 
-    async def get_current_orchestration(self, user_id: str) -> MagenticOrchestration:
-        """Initialize or get existing orchestration instance."""
-        if user_id not in self._orchestrations:
-            agents = await get_agents()
-            self._orchestrations[user_id] = await init_orchestration(agents)
-        return self._orchestrations[user_id]
+    def get_current_orchestration(self, user_id: str) -> MagenticOrchestration:
+        """get existing orchestration instance."""
+        return self._orchestrations.get(user_id, None)
 
 
 # Global config instances