|
8 | 8 | import sys |
9 | 9 | from pathlib import Path |
10 | 10 |
|
| 11 | +from fireworks import Fireworks |
| 12 | + |
11 | 13 | from .cli_commands.common import setup_logging |
12 | 14 | from .cli_commands.utils import add_args_from_callable_signature |
13 | | -from .fireworks_client import create_fireworks_client |
14 | 15 |
|
15 | 16 | logger = logging.getLogger(__name__) |
16 | 17 |
|
@@ -82,7 +83,11 @@ def _configure_parser(parser: argparse.ArgumentParser) -> argparse.ArgumentParse |
82 | 83 | ) |
83 | 84 |
|
84 | 85 | # Auto-generate flags from SDK Fireworks().evaluators.create() signature |
85 | | - create_evaluator_fn = create_fireworks_client().evaluators.create |
| 86 | + # Note: We use Fireworks() directly here instead of create_fireworks_client() |
| 87 | + # because we only need the method signature for introspection, not a fully |
| 88 | + # authenticated client. create_fireworks_client() would trigger an HTTP request |
| 89 | + # to verify the API key, causing delays even for --help invocations. |
| 90 | + create_evaluator_fn = Fireworks().evaluators.create |
86 | 91 |
|
87 | 92 | upload_skip_fields = { |
88 | 93 | "__top_level__": { |
@@ -191,7 +196,11 @@ def _configure_parser(parser: argparse.ArgumentParser) -> argparse.ArgumentParse |
191 | 196 | "loss_config.method": "RL loss method for underlying trainers. One of {grpo,dapo}.", |
192 | 197 | } |
193 | 198 |
|
194 | | - create_rft_job_fn = create_fireworks_client().reinforcement_fine_tuning_jobs.create |
| 199 | + # Note: We use Fireworks() directly here instead of create_fireworks_client() |
| 200 | + # because we only need the method signature for introspection, not a fully |
| 201 | + # authenticated client. create_fireworks_client() would trigger an HTTP request |
| 202 | + # to verify the API key, causing delays even for --help invocations. |
| 203 | + create_rft_job_fn = Fireworks().reinforcement_fine_tuning_jobs.create |
195 | 204 |
|
196 | 205 | add_args_from_callable_signature( |
197 | 206 | rft_parser, |
|
0 commit comments