diff --git a/web/api/views.py b/web/api/views.py index d89be23d..b2c0e47d 100644 --- a/web/api/views.py +++ b/web/api/views.py @@ -170,7 +170,7 @@ class LLMVulnerabilityReportGenerator(APIView): 'status': False, 'error': 'Missing GET param Vulnerability `id`' }) - task = gpt_vulnerability_description.apply_async(args=(vulnerability_id,)) + task = llm_vulnerability_description.apply_async(args=(vulnerability_id,)) response = task.wait() return Response(response) diff --git a/web/celery-entrypoint.sh b/web/celery-entrypoint.sh index 923115bc..70f6ab74 100755 --- a/web/celery-entrypoint.sh +++ b/web/celery-entrypoint.sh @@ -186,7 +186,7 @@ watchmedo auto-restart --recursive --pattern="*.py" --directory="/usr/src/app/re watchmedo auto-restart --recursive --pattern="*.py" --directory="/usr/src/app/reNgine/" -- celery -A reNgine.tasks worker --pool=gevent --concurrency=50 --loglevel=$loglevel -Q run_command_queue -n run_command_worker & watchmedo auto-restart --recursive --pattern="*.py" --directory="/usr/src/app/reNgine/" -- celery -A reNgine.tasks worker --pool=gevent --concurrency=10 --loglevel=$loglevel -Q query_reverse_whois_queue -n query_reverse_whois_worker & watchmedo auto-restart --recursive --pattern="*.py" --directory="/usr/src/app/reNgine/" -- celery -A reNgine.tasks worker --pool=gevent --concurrency=10 --loglevel=$loglevel -Q query_ip_history_queue -n query_ip_history_worker & -watchmedo auto-restart --recursive --pattern="*.py" --directory="/usr/src/app/reNgine/" -- celery -A reNgine.tasks worker --pool=gevent --concurrency=30 --loglevel=$loglevel -Q gpt_queue -n gpt_worker & +watchmedo auto-restart --recursive --pattern="*.py" --directory="/usr/src/app/reNgine/" -- celery -A reNgine.tasks worker --pool=gevent --concurrency=30 --loglevel=$loglevel -Q llm_queue -n llm_worker & watchmedo auto-restart --recursive --pattern="*.py" --directory="/usr/src/app/reNgine/" -- celery -A reNgine.tasks worker --pool=gevent --concurrency=10 --loglevel=$loglevel -Q dorking_queue -n dorking_worker & watchmedo auto-restart --recursive --pattern="*.py" --directory="/usr/src/app/reNgine/" -- celery -A reNgine.tasks worker --pool=gevent --concurrency=10 --loglevel=$loglevel -Q osint_discovery_queue -n osint_discovery_worker & watchmedo auto-restart --recursive --pattern="*.py" --directory="/usr/src/app/reNgine/" -- celery -A reNgine.tasks worker --pool=gevent --concurrency=10 --loglevel=$loglevel -Q h8mail_queue -n h8mail_worker & diff --git a/web/reNgine/tasks.py b/web/reNgine/tasks.py index 785c1acd..bf44e313 100644 --- a/web/reNgine/tasks.py +++ b/web/reNgine/tasks.py @@ -4772,8 +4772,8 @@ def query_ip_history(domain): return get_domain_historical_ip_address(domain) -@app.task(name='gpt_vulnerability_description', bind=False, queue='gpt_queue') -def gpt_vulnerability_description(vulnerability_id): +@app.task(name='llm_vulnerability_description', bind=False, queue='llm_queue') +def llm_vulnerability_description(vulnerability_id): """Generate and store Vulnerability Description using GPT. Args: