771cf41fea
- Modified `.env.example` to set the default terminal environment to 'singularity' and updated Docker and Singularity image references for better compatibility. - Enhanced `run_mixed_tasks.sh` and `run_terminal_tasks.sh` scripts to utilize the new Singularity setup, including improved logging and cache directory management. - Introduced functionality in `terminal_tool.py` to automatically build and cache SIF images from Docker URLs, streamlining the execution environment setup. - Updated logging messages for clarity on image usage and cache directory paths.
51 lines
2.3 KiB
Bash
Executable File
51 lines
2.3 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# Terminal-focused data generation run
|
|
# Uses nous-terminal-tasks.jsonl (597 tasks)
|
|
# Distribution: terminal 97%, web 15%, browser 0%, vision 8%, image_gen 3%
|
|
|
|
# Create logs directory if it doesn't exist
|
|
mkdir -p logs
|
|
|
|
# Generate log filename with timestamp
|
|
LOG_FILE="logs/terminal_tasks_$(date +%Y%m%d_%H%M%S).log"
|
|
|
|
echo "📝 Logging output to: $LOG_FILE"
|
|
echo "💻 Running terminal-focused tasks with terminal_tasks distribution"
|
|
|
|
# Set terminal environment
|
|
# SIF images are automatically built/cached by terminal_tool.py
|
|
export TERMINAL_ENV=singularity
|
|
export TERMINAL_SINGULARITY_IMAGE="docker://nikolaik/python-nodejs:python3.11-nodejs20"
|
|
export TERMINAL_TIMEOUT=300
|
|
|
|
# Set up Apptainer cache directories (use /scratch if available, otherwise /tmp)
|
|
if [ -d "/scratch" ] && [ -w "/scratch" ]; then
|
|
CACHE_BASE="/scratch/$USER/.apptainer"
|
|
else
|
|
CACHE_BASE="/tmp/$USER/.apptainer"
|
|
fi
|
|
export APPTAINER_CACHEDIR="$CACHE_BASE"
|
|
export APPTAINER_TMPDIR="$CACHE_BASE/tmp"
|
|
mkdir -p "$APPTAINER_CACHEDIR" "$APPTAINER_TMPDIR"
|
|
|
|
echo "📁 Apptainer cache: $APPTAINER_CACHEDIR"
|
|
echo "🐳 Image: $TERMINAL_SINGULARITY_IMAGE (auto-converted to SIF on first use)"
|
|
|
|
python batch_runner.py \
|
|
--dataset_file="nous-terminal-tasks.jsonl" \
|
|
--batch_size=5 \
|
|
--run_name="terminal_tasks-kimi-k2.5" \
|
|
--distribution="terminal_tasks" \
|
|
--model="moonshotai/kimi-k2.5" \
|
|
--verbose \
|
|
--base_url="https://openrouter.ai/api/v1" \
|
|
--num_workers=80 \
|
|
--max_turns=60 \
|
|
--providers_ignored="Novita" \
|
|
--resume \
|
|
--ephemeral_system_prompt="You have access to a terminal tool for executing commands and completing coding, system administration, and computing tasks. Use the terminal to write code, run scripts, install packages (use --break-system-packages with pip if needed), manipulate files, and verify your work. Always test and validate code you create. Do not use interactive tools like vim, nano, or python REPL. If git output is large, pipe to cat. When web search is available, use it to look up documentation, APIs, or best practices. If browser tools are available, use them for web interactions that require page manipulation. Do not use the terminal to communicate with the user - only your final response will be shown to them." \
|
|
2>&1 | tee "$LOG_FILE"
|
|
|
|
echo "✅ Log saved to: $LOG_FILE"
|