Skip to content

Commit 45cbb23

Browse files
committed
Update optillm.py
add plugin_approaches as well to args
1 parent 572a0c5 commit 45cbb23

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

optillm.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -522,7 +522,7 @@ def parse_args():
522522
# Define arguments and their corresponding environment variables
523523
args_env = [
524524
("--optillm-api-key", "OPTILLM_API_KEY", str, "", "Optional API key for client authentication to optillm"),
525-
("--approach", "OPTILLM_APPROACH", str, "auto", "Inference approach to use", known_approaches),
525+
("--approach", "OPTILLM_APPROACH", str, "auto", "Inference approach to use", known_approaches + list(plugin_approaches.keys())),
526526
("--mcts-simulations", "OPTILLM_SIMULATIONS", int, 2, "Number of MCTS simulations"),
527527
("--mcts-exploration", "OPTILLM_EXPLORATION", float, 0.2, "Exploration weight for MCTS"),
528528
("--mcts-depth", "OPTILLM_DEPTH", int, 1, "Simulation depth for MCTS"),
@@ -571,10 +571,10 @@ def parse_args():
571571

572572
def main():
573573
global server_config
574-
args = parse_args()
575-
576574
# Call this function at the start of main()
577575
load_plugins()
576+
args = parse_args()
577+
578578
# Update server_config with all argument values
579579
server_config.update(vars(args))
580580

0 commit comments

Comments
 (0)