remove vLLM option in main

Author: EYH0602

src/main.py

@@ -21,7 +21,6 @@ def main(
     effort: str | None = None,
     output_file: str | None = None,
     log_file: str = "evaluation_log.jsonl",
-    use_vllm_server: bool = False,
 ):
     """
     Run an experiment using various AI models to generate and evaluate type signatures.
@@ -45,7 +44,7 @@ def main(
         output_file = os.path.abspath(f"result/{model}.txt")
     logging.info(f"Writing generation results in {output_file}.")
 
-    client = router(model, pure, effort, use_vllm_server)
+    client = router(model, pure, effort)
     assert client, f"Failed to create client for {model}."
 
     tasks = load_from_hf("pure" if pure else "base")

src/tfbench/lm/utils.py

@@ -47,7 +47,6 @@ def router(
     model_name: str,
     pure: bool,
     effort: str | None = None,
-    use_vllm_server: bool = False,
 ) -> LM | None:
     """Route the model name to the appropriate LM class."""
     if model_name in OAI_MODELS: