We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0cbb564 commit 650a349Copy full SHA for 650a349
1 file changed
tests/github_actions/rollout_worker.py
@@ -38,7 +38,7 @@ def main():
38
try:
39
completion_kwargs = {"model": args.model, "messages": messages}
40
41
- client = OpenAI(api_key=os.environ.get("FIREWORKS_API_KEY"))
+ client = OpenAI(base_url="https://api.fireworks.ai/inference/v1", api_key=os.environ.get("FIREWORKS_API_KEY"))
42
43
print("📡 Calling OpenAI completion...")
44
completion = client.chat.completions.create(**completion_kwargs)
0 commit comments