machinaos 0.0.1 → 0.0.6

package/server/services/node_executor.py

@@ -1,375 +1,372 @@
-"""Node Executor - Single node execution with handler dispatch.
-
-Uses a registry pattern for clean handler dispatch without if-else chains.
-"""
-
-import asyncio
-import time
-import uuid
-from dataclasses import dataclass
-from datetime import datetime
-from functools import partial
-from typing import Dict, Any, Optional, Callable, TYPE_CHECKING
-
-from core.logging import get_logger
-from constants import (
-    ANDROID_SERVICE_NODE_TYPES,
-    AI_MODEL_TYPES,
-    AI_CHAT_MODEL_TYPES,
-    GOOGLE_MAPS_TYPES,
-    detect_ai_provider,
-)
-from models.nodes import validate_node_params
-from pydantic import ValidationError
-from services import event_waiter
-from services.handlers import (
-    handle_ai_agent, handle_chat_agent, handle_ai_chat_model, handle_simple_memory,
-    handle_android_device_setup, handle_android_service,
-    handle_python_executor, handle_javascript_executor,
-    handle_http_request, handle_webhook_response, handle_trigger_node,
-    handle_create_map, handle_add_locations, handle_nearby_places,
-    handle_text_generator, handle_file_handler,
-    handle_chat_send, handle_chat_history,
-    handle_start, handle_cron_scheduler, handle_timer, handle_console,
-    handle_whatsapp_send, handle_whatsapp_connect, handle_whatsapp_db,
-    handle_http_scraper, handle_file_downloader, handle_document_parser,
-    handle_text_chunker, handle_embedding_generator, handle_vector_store,
-)
-
-if TYPE_CHECKING:
-    from core.config import Settings
-    from core.database import Database
-    from services.ai import AIService
-    from services.maps import MapsService
-    from services.text import TextService
-    from services.android_service import AndroidService
-
-logger = get_logger(__name__)
-
-
-@dataclass
-class ExecutionResult:
-    """Standardized execution result."""
-    success: bool
-    node_id: str
-    node_type: str
-    result: Optional[Dict] = None
-    error: Optional[str] = None
-    execution_id: str = ""
-    execution_time: float = 0.0
-    timestamp: str = ""
-
-    def to_dict(self) -> Dict[str, Any]:
-        d = {
-            "success": self.success,
-            "node_id": self.node_id,
-            "node_type": self.node_type,
-            "execution_id": self.execution_id,
-            "execution_time": self.execution_time,
-            "timestamp": self.timestamp or datetime.now().isoformat(),
-        }
-        if self.success:
-            d["result"] = self.result or {}
-        else:
-            d["error"] = self.error
-        return d
-
-
-class NodeExecutor:
-    """Executes individual workflow nodes using registry-based dispatch."""
-
-    def __init__(
-        self,
-        database: "Database",
-        ai_service: "AIService",
-        maps_service: "MapsService",
-        text_service: "TextService",
-        android_service: "AndroidService",
-        settings: "Settings",
-        output_store: Optional[Callable] = None,
-    ):
-        self.database = database
-        self.ai_service = ai_service
-        self.maps_service = maps_service
-        self.text_service = text_service
-        self.android_service = android_service
-        self.settings = settings
-        self._output_store = output_store
-        self._handlers = self._build_handler_registry()
-
-    def _build_handler_registry(self) -> Dict[str, Callable]:
-        """Build handler registry with service dependencies bound via partial."""
-        registry = {
-            # Workflow control
-            'start': handle_start,
-            'cronScheduler': handle_cron_scheduler,
-            'timer': handle_timer,
-            # AI
-            'aiAgent': partial(handle_ai_agent, ai_service=self.ai_service, database=self.database),
-            'chatAgent': partial(handle_chat_agent, ai_service=self.ai_service, database=self.database),
-            'simpleMemory': handle_simple_memory,
-            # Maps
-            'createMap': partial(handle_create_map, maps_service=self.maps_service),
-            'addLocations': partial(handle_add_locations, maps_service=self.maps_service),
-            'showNearbyPlaces': partial(handle_nearby_places, maps_service=self.maps_service),
-            # Text
-            'textGenerator': partial(handle_text_generator, text_service=self.text_service),
-            'fileHandler': partial(handle_file_handler, text_service=self.text_service),
-            # WhatsApp
-            'whatsappSend': handle_whatsapp_send,
-            'whatsappConnect': handle_whatsapp_connect,
-            'whatsappDb': handle_whatsapp_db,
-            # Chat
-            'chatSend': handle_chat_send,
-            'chatHistory': handle_chat_history,
-            # HTTP
-            'httpRequest': handle_http_request,
-            # Android setup
-            'androidDeviceSetup': partial(handle_android_device_setup, settings=self.settings),
-            # Document processing
-            'httpScraper': handle_http_scraper,
-            'fileDownloader': handle_file_downloader,
-            'documentParser': handle_document_parser,
-            'textChunker': handle_text_chunker,
-            'embeddingGenerator': handle_embedding_generator,
-            'vectorStore': handle_vector_store,
-            # Note: 'console' handled in _dispatch with connected_outputs
-        }
-
-        # Register AI chat models
-        for node_type in AI_CHAT_MODEL_TYPES:
-            registry[node_type] = partial(handle_ai_chat_model, ai_service=self.ai_service)
-
-        # Register Android services
-        for node_type in ANDROID_SERVICE_NODE_TYPES:
-            registry[node_type] = partial(handle_android_service, android_service=self.android_service)
-
-        return registry
-
-    async def execute(
-        self,
-        node_id: str,
-        node_type: str,
-        parameters: Dict[str, Any],
-        context: Dict[str, Any],
-        resolve_params_fn: Optional[Callable] = None,
-    ) -> Dict[str, Any]:
-        """Execute a single workflow node."""
-        start_time = time.time()
-        session_id = context.get('session_id', 'default')
-        execution_id = context.get('execution_id') or str(uuid.uuid4())[:8]
-
-        try:
-            # Load, validate, enhance parameters
-            params = await self._prepare_parameters(node_id, node_type, parameters, session_id)
-
-            # Resolve templates if resolver provided
-            nodes = context.get('nodes')
-            edges = context.get('edges')
-            logger.debug(f"[NodeExecutor] Template resolution check: resolve_fn={resolve_params_fn is not None}, nodes={len(nodes) if nodes else 'None'}, edges={len(edges) if edges else 'None'}")
-
-            if resolve_params_fn and nodes is not None and edges is not None:
-                logger.debug(f"[NodeExecutor] Before resolution: params={list(params.keys())}")
-                params = await resolve_params_fn(params, node_id, nodes, edges, session_id)
-                logger.debug(f"[NodeExecutor] After resolution: params keys={list(params.keys())}")
-
-            # Build handler context
-            handler_ctx = {
-                **context,
-                "start_time": start_time,
-                "execution_id": execution_id,
-            }
-            logger.info("NodeExecutor context", node_id=node_id, workflow_id=context.get('workflow_id'))
-
-            # Execute via registry or special handlers
-            result = await self._dispatch(node_id, node_type, params, handler_ctx)
-            result['execution_id'] = execution_id
-
-            # Store output if successful
-            if result.get('success') and self._output_store:
-                output_data = result.get('result', {})
-
-                # For Android service nodes, extract the nested 'data' field for cleaner template access
-                # This allows {{batterymonitor.battery_level}} instead of {{batterymonitor.data.battery_level}}
-                if node_type in ANDROID_SERVICE_NODE_TYPES and isinstance(output_data, dict):
-                    # Flatten: promote 'data' contents to top level while preserving metadata
-                    nested_data = output_data.get('data', {})
-                    if isinstance(nested_data, dict):
-                        # Merge nested data with metadata (service_id, action, timestamp, etc.)
-                        output_data = {**output_data, **nested_data}
-                        logger.debug(f"[NodeExecutor] Flattened Android output for {node_id}: keys={list(output_data.keys())}")
-
-                await self._output_store(session_id, node_id, "output_0", output_data)
-
-            return result
-
-        except asyncio.CancelledError:
-            return ExecutionResult(False, node_id, node_type, error="Cancelled",
-                                   execution_id=execution_id, execution_time=time.time()-start_time).to_dict()
-        except Exception as e:
-            logger.error("Node execution error", node_id=node_id, error=str(e))
-            return ExecutionResult(False, node_id, node_type, error=str(e),
-                                   execution_id=execution_id, execution_time=time.time()-start_time).to_dict()
-
-    async def _prepare_parameters(self, node_id: str, node_type: str, params: Dict, session_id: str) -> Dict:
-        """Load from DB, validate, inject API keys."""
-        # Merge with DB parameters (DB provides defaults, frontend can override)
-        db_params = await self.database.get_node_parameters(node_id) or {}
-        merged = {**db_params, **params} if params else db_params
-
-        # Validate
-        try:
-            validated = validate_node_params(node_type, merged)
-            merged = {**merged, **validated.model_dump(by_alias=True, exclude_unset=True)}
-        except ValidationError as e:
-            logger.warning("Validation warning", node_type=node_type, errors=str(e))
-
-        # Inject API keys
-        return await self._inject_api_keys(node_type, merged)
-
-    async def _inject_api_keys(self, node_type: str, params: Dict) -> Dict:
-        """Auto-inject API keys for AI and Maps nodes."""
-        result = params.copy()
-
-        if node_type in AI_MODEL_TYPES:
-            provider = detect_ai_provider(node_type, params)
-            if not result.get('api_key') and not result.get('apiKey'):
-                key = await self.ai_service.auth.get_api_key(provider, "default")
-                if key:
-                    result['api_key'] = key
-            if not result.get('model'):
-                models = await self.ai_service.auth.get_stored_models(provider, "default")
-                if models:
-                    result['model'] = models[0]
-
-        elif node_type in GOOGLE_MAPS_TYPES:
-            if not result.get('api_key'):
-                # Try database first, then fall back to environment variable
-                key = await self.ai_service.auth.get_api_key("google_maps", "default")
-                if key:
-                    result['api_key'] = key
-                elif self.settings.google_maps_api_key:
-                    result['api_key'] = self.settings.google_maps_api_key
-
-        return result
-
-    async def _dispatch(self, node_id: str, node_type: str, params: Dict, context: Dict) -> Dict:
-        """Dispatch to handler from registry or special handlers."""
-
-        # Check registry first
-        handler = self._handlers.get(node_type)
-        if handler:
-            return await handler(node_id, node_type, params, context)
-
-        # Special handlers needing connected outputs
-        if node_type in ('pythonExecutor', 'javascriptExecutor', 'webhookResponse', 'console'):
-            outputs, source_nodes = await self._get_connected_outputs_with_info(context, node_id)
-            if node_type == 'console':
-                return await handle_console(node_id, node_type, params, context, outputs, source_nodes)
-            handlers = {
-                'pythonExecutor': handle_python_executor,
-                'javascriptExecutor': handle_javascript_executor,
-                'webhookResponse': handle_webhook_response,
-            }
-            return await handlers[node_type](node_id, node_type, params, context, outputs)
-
-        # Trigger nodes
-        if event_waiter.is_trigger_node(node_type):
-            return await handle_trigger_node(node_id, node_type, params, context)
-
-        # Fallback
-        return {
-            "success": True,
-            "node_id": node_id,
-            "node_type": node_type,
-            "result": {"message": f"Node {node_id} executed", "parameters": params},
-            "execution_time": time.time() - context.get('start_time', time.time()),
-            "timestamp": datetime.now().isoformat()
-        }
-
-    async def _get_connected_outputs(self, context: Dict, node_id: str) -> Dict[str, Any]:
-        """Get outputs from connected upstream nodes."""
-        get_output = context.get('get_output_fn')
-        if not get_output:
-            return {}
-
-        nodes = context.get('nodes', [])
-        edges = context.get('edges', [])
-        session_id = context.get('session_id', 'default')
-        result = {}
-
-        for edge in edges:
-            if edge.get('target') == node_id:
-                source_id = edge.get('source')
-                output = await get_output(session_id, source_id, "output_0")
-                if output:
-                    source = next((n for n in nodes if n.get('id') == source_id), {})
-                    result[source.get('type', 'unknown')] = output
-
-        return result
-
-    def _get_source_nodes_info(self, context: Dict, node_id: str) -> list:
-        """Get source node info (id, type, label) for edges targeting this node.
-
-        This is used for display purposes (e.g., showing source in Console panel).
-        Does NOT filter by output availability - just returns edge source info.
-        """
-        nodes = context.get('nodes', [])
-        edges = context.get('edges', [])
-        source_nodes = []
-
-        for edge in edges:
-            if edge.get('target') == node_id:
-                source_id = edge.get('source')
-                source = next((n for n in nodes if n.get('id') == source_id), {})
-                source_type = source.get('type', 'unknown')
-                source_data = source.get('data', {})
-                source_label = source_data.get('label') or source_type
-                source_nodes.append({
-                    'id': source_id,
-                    'type': source_type,
-                    'label': source_label
-                })
-
-        return source_nodes
-
-    async def _get_connected_outputs_with_info(self, context: Dict, node_id: str) -> tuple:
-        """Get outputs from connected upstream nodes with source node info.
-
-        Returns:
-            Tuple of (outputs dict, source_nodes list with id/type/label info)
-        """
-        get_output = context.get('get_output_fn')
-        if not get_output:
-            logger.warning(f"[_get_connected_outputs_with_info] No get_output_fn in context for {node_id}")
-            return {}, []
-
-        nodes = context.get('nodes', [])
-        edges = context.get('edges', [])
-        session_id = context.get('session_id', 'default')
-        outputs = {}
-        source_nodes = []
-
-        logger.debug(f"[_get_connected_outputs_with_info] node_id={node_id}, edges={len(edges)}, session={session_id}")
-
-        for edge in edges:
-            if edge.get('target') == node_id:
-                source_id = edge.get('source')
-                logger.debug(f"[_get_connected_outputs_with_info] Found edge from {source_id} to {node_id}")
-                output = await get_output(session_id, source_id, "output_0")
-                logger.debug(f"[_get_connected_outputs_with_info] Output from {source_id}: {'FOUND' if output else 'NOT FOUND'}")
-                if output:
-                    source = next((n for n in nodes if n.get('id') == source_id), {})
-                    source_type = source.get('type', 'unknown')
-                    outputs[source_type] = output
-                    # Get label from node data if available
-                    source_data = source.get('data', {})
-                    source_label = source_data.get('label') or source_type
-                    source_nodes.append({
-                        'id': source_id,
-                        'type': source_type,
-                        'label': source_label
-                    })
-
-        logger.debug(f"[_get_connected_outputs_with_info] Returning {len(outputs)} outputs, {len(source_nodes)} source_nodes")
-        return outputs, source_nodes
+"""Node Executor - Single node execution with handler dispatch.
+
+Uses a registry pattern for clean handler dispatch without if-else chains.
+"""
+
+import asyncio
+import time
+import uuid
+from dataclasses import dataclass
+from datetime import datetime
+from functools import partial
+from typing import Dict, Any, Optional, Callable, TYPE_CHECKING
+
+from core.logging import get_logger
+from constants import (
+    ANDROID_SERVICE_NODE_TYPES,
+    AI_MODEL_TYPES,
+    AI_CHAT_MODEL_TYPES,
+    GOOGLE_MAPS_TYPES,
+    detect_ai_provider,
+)
+from models.nodes import validate_node_params
+from pydantic import ValidationError
+from services import event_waiter
+from services.handlers import (
+    handle_ai_agent, handle_chat_agent, handle_ai_chat_model, handle_simple_memory,
+    handle_android_service,
+    handle_python_executor, handle_javascript_executor,
+    handle_http_request, handle_webhook_response, handle_trigger_node,
+    handle_create_map, handle_add_locations, handle_nearby_places,
+    handle_text_generator, handle_file_handler,
+    handle_chat_send, handle_chat_history,
+    handle_start, handle_cron_scheduler, handle_timer, handle_console,
+    handle_whatsapp_send, handle_whatsapp_db,
+    handle_http_scraper, handle_file_downloader, handle_document_parser,
+    handle_text_chunker, handle_embedding_generator, handle_vector_store,
+)
+
+if TYPE_CHECKING:
+    from core.config import Settings
+    from core.database import Database
+    from services.ai import AIService
+    from services.maps import MapsService
+    from services.text import TextService
+    from services.android_service import AndroidService
+
+logger = get_logger(__name__)
+
+
+@dataclass
+class ExecutionResult:
+    """Standardized execution result."""
+    success: bool
+    node_id: str
+    node_type: str
+    result: Optional[Dict] = None
+    error: Optional[str] = None
+    execution_id: str = ""
+    execution_time: float = 0.0
+    timestamp: str = ""
+
+    def to_dict(self) -> Dict[str, Any]:
+        d = {
+            "success": self.success,
+            "node_id": self.node_id,
+            "node_type": self.node_type,
+            "execution_id": self.execution_id,
+            "execution_time": self.execution_time,
+            "timestamp": self.timestamp or datetime.now().isoformat(),
+        }
+        if self.success:
+            d["result"] = self.result or {}
+        else:
+            d["error"] = self.error
+        return d
+
+
+class NodeExecutor:
+    """Executes individual workflow nodes using registry-based dispatch."""
+
+    def __init__(
+        self,
+        database: "Database",
+        ai_service: "AIService",
+        maps_service: "MapsService",
+        text_service: "TextService",
+        android_service: "AndroidService",
+        settings: "Settings",
+        output_store: Optional[Callable] = None,
+    ):
+        self.database = database
+        self.ai_service = ai_service
+        self.maps_service = maps_service
+        self.text_service = text_service
+        self.android_service = android_service
+        self.settings = settings
+        self._output_store = output_store
+        self._handlers = self._build_handler_registry()
+
+    def _build_handler_registry(self) -> Dict[str, Callable]:
+        """Build handler registry with service dependencies bound via partial."""
+        registry = {
+            # Workflow control
+            'start': handle_start,
+            'cronScheduler': handle_cron_scheduler,
+            'timer': handle_timer,
+            # AI
+            'aiAgent': partial(handle_ai_agent, ai_service=self.ai_service, database=self.database),
+            'chatAgent': partial(handle_chat_agent, ai_service=self.ai_service, database=self.database),
+            'simpleMemory': handle_simple_memory,
+            # Maps
+            'createMap': partial(handle_create_map, maps_service=self.maps_service),
+            'addLocations': partial(handle_add_locations, maps_service=self.maps_service),
+            'showNearbyPlaces': partial(handle_nearby_places, maps_service=self.maps_service),
+            # Text
+            'textGenerator': partial(handle_text_generator, text_service=self.text_service),
+            'fileHandler': partial(handle_file_handler, text_service=self.text_service),
+            # WhatsApp
+            'whatsappSend': handle_whatsapp_send,
+            'whatsappDb': handle_whatsapp_db,
+            # Chat
+            'chatSend': handle_chat_send,
+            'chatHistory': handle_chat_history,
+            # HTTP
+            'httpRequest': handle_http_request,
+            # Document processing
+            'httpScraper': handle_http_scraper,
+            'fileDownloader': handle_file_downloader,
+            'documentParser': handle_document_parser,
+            'textChunker': handle_text_chunker,
+            'embeddingGenerator': handle_embedding_generator,
+            'vectorStore': handle_vector_store,
+            # Note: 'console' handled in _dispatch with connected_outputs
+        }
+
+        # Register AI chat models
+        for node_type in AI_CHAT_MODEL_TYPES:
+            registry[node_type] = partial(handle_ai_chat_model, ai_service=self.ai_service)
+
+        # Register Android services
+        for node_type in ANDROID_SERVICE_NODE_TYPES:
+            registry[node_type] = partial(handle_android_service, android_service=self.android_service)
+
+        return registry
+
+    async def execute(
+        self,
+        node_id: str,
+        node_type: str,
+        parameters: Dict[str, Any],
+        context: Dict[str, Any],
+        resolve_params_fn: Optional[Callable] = None,
+    ) -> Dict[str, Any]:
+        """Execute a single workflow node."""
+        start_time = time.time()
+        session_id = context.get('session_id', 'default')
+        execution_id = context.get('execution_id') or str(uuid.uuid4())[:8]
+
+        try:
+            # Load, validate, enhance parameters
+            params = await self._prepare_parameters(node_id, node_type, parameters, session_id)
+
+            # Resolve templates if resolver provided
+            nodes = context.get('nodes')
+            edges = context.get('edges')
+            logger.debug(f"[NodeExecutor] Template resolution check: resolve_fn={resolve_params_fn is not None}, nodes={len(nodes) if nodes else 'None'}, edges={len(edges) if edges else 'None'}")
+
+            if resolve_params_fn and nodes is not None and edges is not None:
+                logger.debug(f"[NodeExecutor] Before resolution: params={list(params.keys())}")
+                params = await resolve_params_fn(params, node_id, nodes, edges, session_id)
+                logger.debug(f"[NodeExecutor] After resolution: params keys={list(params.keys())}")
+
+            # Build handler context
+            handler_ctx = {
+                **context,
+                "start_time": start_time,
+                "execution_id": execution_id,
+            }
+            logger.info("NodeExecutor context", node_id=node_id, workflow_id=context.get('workflow_id'))
+
+            # Execute via registry or special handlers
+            result = await self._dispatch(node_id, node_type, params, handler_ctx)
+            result['execution_id'] = execution_id
+
+            # Store output if successful
+            if result.get('success') and self._output_store:
+                output_data = result.get('result', {})
+
+                # For Android service nodes, extract the nested 'data' field for cleaner template access
+                # This allows {{batterymonitor.battery_level}} instead of {{batterymonitor.data.battery_level}}
+                if node_type in ANDROID_SERVICE_NODE_TYPES and isinstance(output_data, dict):
+                    # Flatten: promote 'data' contents to top level while preserving metadata
+                    nested_data = output_data.get('data', {})
+                    if isinstance(nested_data, dict):
+                        # Merge nested data with metadata (service_id, action, timestamp, etc.)
+                        output_data = {**output_data, **nested_data}
+                        logger.debug(f"[NodeExecutor] Flattened Android output for {node_id}: keys={list(output_data.keys())}")
+
+                await self._output_store(session_id, node_id, "output_0", output_data)
+
+            return result
+
+        except asyncio.CancelledError:
+            return ExecutionResult(False, node_id, node_type, error="Cancelled",
+                                   execution_id=execution_id, execution_time=time.time()-start_time).to_dict()
+        except Exception as e:
+            logger.error("Node execution error", node_id=node_id, error=str(e))
+            return ExecutionResult(False, node_id, node_type, error=str(e),
+                                   execution_id=execution_id, execution_time=time.time()-start_time).to_dict()
+
+    async def _prepare_parameters(self, node_id: str, node_type: str, params: Dict, session_id: str) -> Dict:
+        """Load from DB, validate, inject API keys."""
+        # Merge with DB parameters (DB provides defaults, frontend can override)
+        db_params = await self.database.get_node_parameters(node_id) or {}
+        merged = {**db_params, **params} if params else db_params
+
+        # Validate
+        try:
+            validated = validate_node_params(node_type, merged)
+            merged = {**merged, **validated.model_dump(by_alias=True, exclude_unset=True)}
+        except ValidationError as e:
+            logger.warning("Validation warning", node_type=node_type, errors=str(e))
+
+        # Inject API keys
+        return await self._inject_api_keys(node_type, merged)
+
+    async def _inject_api_keys(self, node_type: str, params: Dict) -> Dict:
+        """Auto-inject API keys for AI and Maps nodes."""
+        result = params.copy()
+
+        if node_type in AI_MODEL_TYPES:
+            provider = detect_ai_provider(node_type, params)
+            if not result.get('api_key') and not result.get('apiKey'):
+                key = await self.ai_service.auth.get_api_key(provider, "default")
+                if key:
+                    result['api_key'] = key
+            if not result.get('model'):
+                models = await self.ai_service.auth.get_stored_models(provider, "default")
+                if models:
+                    result['model'] = models[0]
+
+        elif node_type in GOOGLE_MAPS_TYPES:
+            if not result.get('api_key'):
+                # Try database first, then fall back to environment variable
+                key = await self.ai_service.auth.get_api_key("google_maps", "default")
+                if key:
+                    result['api_key'] = key
+                elif self.settings.google_maps_api_key:
+                    result['api_key'] = self.settings.google_maps_api_key
+
+        return result
+
+    async def _dispatch(self, node_id: str, node_type: str, params: Dict, context: Dict) -> Dict:
+        """Dispatch to handler from registry or special handlers."""
+
+        # Check registry first
+        handler = self._handlers.get(node_type)
+        if handler:
+            return await handler(node_id, node_type, params, context)
+
+        # Special handlers needing connected outputs
+        if node_type in ('pythonExecutor', 'javascriptExecutor', 'webhookResponse', 'console'):
+            outputs, source_nodes = await self._get_connected_outputs_with_info(context, node_id)
+            if node_type == 'console':
+                return await handle_console(node_id, node_type, params, context, outputs, source_nodes)
+            handlers = {
+                'pythonExecutor': handle_python_executor,
+                'javascriptExecutor': handle_javascript_executor,
+                'webhookResponse': handle_webhook_response,
+            }
+            return await handlers[node_type](node_id, node_type, params, context, outputs)
+
+        # Trigger nodes
+        if event_waiter.is_trigger_node(node_type):
+            return await handle_trigger_node(node_id, node_type, params, context)
+
+        # Fallback
+        return {
+            "success": True,
+            "node_id": node_id,
+            "node_type": node_type,
+            "result": {"message": f"Node {node_id} executed", "parameters": params},
+            "execution_time": time.time() - context.get('start_time', time.time()),
+            "timestamp": datetime.now().isoformat()
+        }
+
+    async def _get_connected_outputs(self, context: Dict, node_id: str) -> Dict[str, Any]:
+        """Get outputs from connected upstream nodes."""
+        get_output = context.get('get_output_fn')
+        if not get_output:
+            return {}
+
+        nodes = context.get('nodes', [])
+        edges = context.get('edges', [])
+        session_id = context.get('session_id', 'default')
+        result = {}
+
+        for edge in edges:
+            if edge.get('target') == node_id:
+                source_id = edge.get('source')
+                output = await get_output(session_id, source_id, "output_0")
+                if output:
+                    source = next((n for n in nodes if n.get('id') == source_id), {})
+                    result[source.get('type', 'unknown')] = output
+
+        return result
+
+    def _get_source_nodes_info(self, context: Dict, node_id: str) -> list:
+        """Get source node info (id, type, label) for edges targeting this node.
+
+        This is used for display purposes (e.g., showing source in Console panel).
+        Does NOT filter by output availability - just returns edge source info.
+        """
+        nodes = context.get('nodes', [])
+        edges = context.get('edges', [])
+        source_nodes = []
+
+        for edge in edges:
+            if edge.get('target') == node_id:
+                source_id = edge.get('source')
+                source = next((n for n in nodes if n.get('id') == source_id), {})
+                source_type = source.get('type', 'unknown')
+                source_data = source.get('data', {})
+                source_label = source_data.get('label') or source_type
+                source_nodes.append({
+                    'id': source_id,
+                    'type': source_type,
+                    'label': source_label
+                })
+
+        return source_nodes
+
+    async def _get_connected_outputs_with_info(self, context: Dict, node_id: str) -> tuple:
+        """Get outputs from connected upstream nodes with source node info.
+
+        Returns:
+            Tuple of (outputs dict, source_nodes list with id/type/label info)
+        """
+        get_output = context.get('get_output_fn')
+        if not get_output:
+            logger.warning(f"[_get_connected_outputs_with_info] No get_output_fn in context for {node_id}")
+            return {}, []
+
+        nodes = context.get('nodes', [])
+        edges = context.get('edges', [])
+        session_id = context.get('session_id', 'default')
+        outputs = {}
+        source_nodes = []
+
+        logger.debug(f"[_get_connected_outputs_with_info] node_id={node_id}, edges={len(edges)}, session={session_id}")
+
+        for edge in edges:
+            if edge.get('target') == node_id:
+                source_id = edge.get('source')
+                logger.debug(f"[_get_connected_outputs_with_info] Found edge from {source_id} to {node_id}")
+                output = await get_output(session_id, source_id, "output_0")
+                logger.debug(f"[_get_connected_outputs_with_info] Output from {source_id}: {'FOUND' if output else 'NOT FOUND'}")
+                if output:
+                    source = next((n for n in nodes if n.get('id') == source_id), {})
+                    source_type = source.get('type', 'unknown')
+                    outputs[source_type] = output
+                    # Get label from node data if available
+                    source_data = source.get('data', {})
+                    source_label = source_data.get('label') or source_type
+                    source_nodes.append({
+                        'id': source_id,
+                        'type': source_type,
+                        'label': source_label
+                    })
+
+        logger.debug(f"[_get_connected_outputs_with_info] Returning {len(outputs)} outputs, {len(source_nodes)} source_nodes")
+        return outputs, source_nodes