Skip to content

Commit 650a349

Browse files
committed
fix rollout worker
1 parent 0cbb564 commit 650a349

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tests/github_actions/rollout_worker.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def main():
3838
try:
3939
completion_kwargs = {"model": args.model, "messages": messages}
4040

41-
client = OpenAI(api_key=os.environ.get("FIREWORKS_API_KEY"))
41+
client = OpenAI(base_url="https://api.fireworks.ai/inference/v1", api_key=os.environ.get("FIREWORKS_API_KEY"))
4242

4343
print("📡 Calling OpenAI completion...")
4444
completion = client.chat.completions.create(**completion_kwargs)

0 commit comments

Comments
 (0)