File tree Expand file tree Collapse file tree 1 file changed +58
-0
lines changed
Expand file tree Collapse file tree 1 file changed +58
-0
lines changed Original file line number Diff line number Diff line change 1+ name : Eval Protocol Rollout
2+
3+ on :
4+ workflow_dispatch :
5+ inputs :
6+ model :
7+ description : ' Model to use for the rollout'
8+ required : true
9+ type : string
10+ rollout_id :
11+ description : ' Rollout ID for tracking'
12+ required : true
13+ type : string
14+ messages_b64 :
15+ description : ' Base64 encoded JSON messages array'
16+ required : true
17+ type : string
18+ tools_b64 :
19+ description : ' Base64 encoded JSON tools array (optional)'
20+ required : false
21+ type : string
22+
23+ jobs :
24+ rollout :
25+ runs-on : ubuntu-latest
26+ name : rollout-${{ inputs.rollout_id }}
27+
28+ steps :
29+ - name : Checkout code
30+ uses : actions/checkout@v4
31+
32+ - name : Set up Python
33+ uses : actions/setup-python@v5
34+ with :
35+ python-version : ' 3.11'
36+
37+ - name : Install dependencies
38+ run : |
39+ python -m pip install --upgrade pip
40+ pip install -e .
41+
42+ - name : Run rollout script
43+ env :
44+ FIREWORKS_API_KEY : ${{ secrets.FIREWORKS_API_KEY }}
45+ run : |
46+ python tests/github_actions/rollout_worker.py \
47+ --model "${{ inputs.model }}" \
48+ --rollout-id "${{ inputs.rollout_id }}" \
49+ --messages-b64 "${{ inputs.messages_b64 }}" \
50+ ${{ inputs.tools_b64 && format('--tools-b64 "{0}"', inputs.tools_b64) || '' }}
51+
52+ - name : Upload rollout trace
53+ uses : actions/upload-artifact@v4
54+ if : always() # Upload even if the rollout failed
55+ with :
56+ name : rollout-trace-${{ inputs.rollout_id }}
57+ path : rollout_trace_${{ inputs.rollout_id }}.json
58+ retention-days : 7
You can’t perform that action at this time.
0 commit comments