Skip to content

Commit 5b4b63e

Browse files
committed
put rollout.yml on main to test gharolloutprocessor
1 parent a3baa0a commit 5b4b63e

File tree

1 file changed

+58
-0
lines changed

1 file changed

+58
-0
lines changed

.github/workflows/rollout.yml

Lines changed: 58 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,58 @@
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

0 commit comments

Comments
 (0)