From be10abb3983ee5fbe050e0160ca6d7bb18bf71b6 Mon Sep 17 00:00:00 2001
From: Olindo Task <olindost.sol@gmail.com>
Date: Tue, 21 Oct 2025 12:39:49 +0000
Subject: [PATCH] feat: Add base AIOrchestrator class

---
 backend/app/core/orchestrator.py | 87 +++++++++++++-------------------
 1 file changed, 35 insertions(+), 52 deletions(-)

diff --git a/backend/app/core/orchestrator.py b/backend/app/core/orchestrator.py
index a3baf2e..13365fe 100644
--- a/backend/app/core/orchestrator.py
+++ b/backend/app/core/orchestrator.py
@@ -1,56 +1,39 @@
 import asyncio
-import logging
-from typing import Dict, Callable, Awaitable
-from backend.app.services.report_service import save_report_data
 
-logger = logging.getLogger(__name__)
+class AIOrchestrator:
+    """
+    Base class for coordinating multiple AI agents.
+    Designed to handle parallel asynchronous agent calls.
+    """
 
-class Orchestrator:
     def __init__(self):
-        self.registered_agents: Dict[str, Callable[[str, str], Awaitable[Dict]]] = {}
-
-    def register_agent(self, name: str, agent_func: Callable[[str, str], Awaitable[Dict]]):
-        self.registered_agents[name] = agent_func
-
-    async def execute_agents_concurrently(self, report_id: str, token_id: str):
-        agent_tasks = []
-        agent_names = []
-
-        for name, agent_func in self.registered_agents.items():
-            agent_names.append(name)
-            agent_tasks.append(self._run_agent_safely(name, agent_func, report_id, token_id))
-
-        results = await asyncio.gather(*agent_tasks, return_exceptions=True)
-
-        aggregated_results = {}
-        for i, result in enumerate(results):
-            agent_name = agent_names[i]
-            if isinstance(result, Exception):
-                logger.error("Agent '%s' failed with error: %s", agent_name, result, exc_info=isinstance(result, BaseException))
-                aggregated_results[agent_name] = {'status': 'failed', 'error': str(result)}
-            else:
-                aggregated_results[agent_name] = {'status': 'completed', 'data': result}
-
-        failed_count = sum(1 for r in aggregated_results.values() if r['status'] == 'failed')
-        total = len(aggregated_results)
-        if failed_count == total:
-            overall_status = 'failed'
-        elif failed_count > 0:
-            overall_status = 'partial_success'
-        else:
-            overall_status = 'completed'
-
-        await save_report_data(report_id, {
-            'agent_results': aggregated_results,
-            'status': overall_status,
-            'summary': {'total': total, 'success': total - failed_count, 'failed': failed_count}
-        })
-
-    async def _run_agent_safely(self, name: str, agent_func: Callable[[str, str], Awaitable[Dict]], report_id: str, token_id: str) -> Dict:
-        try:
-            return await agent_func(report_id, token_id)
-        except Exception as e:
-            logger.error("Error running agent '%s': %s", name, e, exc_info=True)
-            raise  # Re-raise to be caught by asyncio.gather
-
-orchestrator = Orchestrator()
+        self.agents = []
+
+    def register_agent(self, agent):
+        """
+        Registers an AI agent with the orchestrator.
+        Args:
+            agent: An instance of an AI agent.
+        """
+        raise NotImplementedError
+
+    async def execute_agents(self, *args, **kwargs):
+        """
+        Executes all registered AI agents in parallel asynchronously.
+        Args:
+            *args: Variable length argument list for agent execution.
+            **kwargs: Arbitrary keyword arguments for agent execution.
+        Returns:
+            A list of results from each agent.
+        """
+        raise NotImplementedError
+
+    def aggregate_results(self, results):
+        """
+        Aggregates the results from the executed AI agents.
+        Args:
+            results (list): A list of results from the executed agents.
+        Returns:
+            The aggregated result.
+        """
+        raise NotImplementedError
\ No newline at end of file