Upload /app/agent-zero-fleet/agent-zero-pentesting/orchestrator.py with huggingface_hub

app/agent-zero-fleet/agent-zero-pentesting/orchestrator.py

@@ -0,0 +1,48 @@
+FROM python:3.11-slim
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONUNBUFFERED=1
+ENV PYTHONIOENCODING=utf-8
+ENV AGENT_NAME="pentesting"
+ENV AGENT_ROLE="pentesting_engineer"
+ENV WORKSPACE_ROOT="/workspace"
+ENV PROJECTS_DIR="/workspace/projects"
+ENV SHARED_DIR="/workspace/shared"
+ENV CONFIG_DIR="/workspace/config"
+ENV LOGS_DIR="/workspace/logs"
+ENV TASK_QUEUE="/workspace/shared/task_queue"
+ENV COMPLETED_DIR="/workspace/shared/completed"
+ENV LITELLM_PROXY_HOST="0.0.0.0"
+ENV LITELLM_PROXY_PORT="8000"
+ENV OPENAI_API_BASE="http://localhost:8000/v1"
+ENV OPENAI_API_KEY="not-needed"
+ENV MODEL_NAME="huggingface/Qwen/Qwen2.5-72B-Instruct"
+ENV FALLBACK_MODELS="huggingface/meta-llama/Llama-3.1-70B-Instruct,huggingface/mistralai/Mixtral-8x22B-Instruct-v0.1"
+ENV HF_INFERENCE_API="https://api-inference.huggingface.co/models/"
+ENV HF_TOKEN="${HF_TOKEN}"
+ENV PERSISTENCE_ENABLED="true"
+ENV DATA_VOLUME="/data"
+ENV SQLITE_DB="/data/agent_memory.db"
+RUN apt-get update && apt-get install -y \
+    git curl wget jq sqlite3 \
+    build-essential libssl-dev \
+    && rm -rf /var/lib/apt/lists/*
+RUN pip install --no-cache-dir \
+    gradio==4.44.0 \
+    litellm==1.48.0 \
+    openai==1.54.0 \
+    httpx==0.27.2 \
+    sqlite-vec==0.1.6 \
+    sentence-transformers==3.2.0 \
+    schedule==1.2.2 \
+    psutil==6.1.0 \
+    pyyaml==6.0.2 \
+    requests==2.32.3 \
+    huggingface-hub==0.26.2
+RUN mkdir -p ${WORKSPACE_ROOT}/{projects,shared/task_queue,shared/completed,config,logs,data}
+RUN mkdir -p ${DATA_VOLUME}
+COPY . /app/
+WORKDIR /app
+HEALTHCHECK --interval=30s --timeout=10s --start-period=60s --retries=3 \
+    CMD curl -f http://localhost:7860/health || exit 1
+EXPOSE 7860 8000
+CMD ["python", "orchestrator.py"]