Refactor generate_ideas function integration and add auto_generate_ideas action for cluster idea generation

2025-11-10 12:29:15 +00:00
parent 488ab78f25
commit b1d86cd4b8
4 changed files with 105 additions and 15 deletions
--- a/backend/igny8_core/modules/planner/views.py
+++ b/backend/igny8_core/modules/planner/views.py
@@ -714,7 +714,102 @@ class ClusterViewSet(SiteSectorModelViewSet):
        # Save with all required fields explicitly
        serializer.save(account=account, site=site, sector=sector)
    
-    # REMOVED: auto_generate_ideas action - idea generation function removed
+    @action(detail=False, methods=['post'], url_path='auto_generate_ideas', url_name='auto_generate_ideas')
+    def auto_generate_ideas(self, request):
+        """Auto-generate ideas for clusters using AI - New unified framework"""
+        import logging
+        from igny8_core.ai.tasks import run_ai_task
+        from kombu.exceptions import OperationalError as KombuOperationalError
+        
+        logger = logging.getLogger(__name__)
+        
+        try:
+            # Get account
+            account = getattr(request, 'account', None)
+            account_id = account.id if account else None
+            
+            # Prepare payload
+            payload = {
+                'ids': request.data.get('ids', [])
+            }
+            
+            logger.info(f"auto_generate_ideas called with ids={payload['ids']}")
+            
+            # Validate basic input
+            if not payload['ids']:
+                return Response({
+                    'success': False,
+                    'error': 'No IDs provided'
+                }, status=status.HTTP_400_BAD_REQUEST)
+            
+            if len(payload['ids']) > 10:
+                return Response({
+                    'success': False,
+                    'error': 'Maximum 10 clusters allowed for idea generation'
+                }, status=status.HTTP_400_BAD_REQUEST)
+            
+            # Try to queue Celery task
+            try:
+                if hasattr(run_ai_task, 'delay'):
+                    task = run_ai_task.delay(
+                        function_name='auto_generate_ideas',
+                        payload=payload,
+                        account_id=account_id
+                    )
+                    logger.info(f"Task queued: {task.id}")
+                    return Response({
+                        'success': True,
+                        'task_id': str(task.id),
+                        'message': 'Idea generation started'
+                    }, status=status.HTTP_200_OK)
+                else:
+                    # Celery not available - execute synchronously
+                    logger.warning("Celery not available, executing synchronously")
+                    result = run_ai_task(
+                        function_name='auto_generate_ideas',
+                        payload=payload,
+                        account_id=account_id
+                    )
+                    if result.get('success'):
+                        return Response({
+                            'success': True,
+                            **result
+                        }, status=status.HTTP_200_OK)
+                    else:
+                        return Response({
+                            'success': False,
+                            'error': result.get('error', 'Idea generation failed')
+                        }, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+            except (KombuOperationalError, ConnectionError) as e:
+                # Broker connection failed - fall back to synchronous execution
+                logger.warning(f"Celery broker unavailable, falling back to synchronous execution: {str(e)}")
+                result = run_ai_task(
+                    function_name='auto_generate_ideas',
+                    payload=payload,
+                    account_id=account_id
+                )
+                if result.get('success'):
+                    return Response({
+                        'success': True,
+                        **result
+                    }, status=status.HTTP_200_OK)
+                else:
+                    return Response({
+                        'success': False,
+                        'error': result.get('error', 'Idea generation failed')
+                    }, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+            except Exception as e:
+                logger.error(f"Error in auto_generate_ideas: {str(e)}", exc_info=True)
+                return Response({
+                    'success': False,
+                    'error': str(e)
+                }, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+        except Exception as e:
+            logger.error(f"Unexpected error in auto_generate_ideas: {str(e)}", exc_info=True)
+            return Response({
+                'success': False,
+                'error': f'Unexpected error: {str(e)}'
+            }, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
    
    def list(self, request, *args, **kwargs):
        """