step1: merge mk_prompt and sample

step1_mk_prompt.py → step1_sample_apps.py

@@ -64,4 +64,12 @@ def mk_sample_prompt(model_path, apps_path, output_path):
 
 if __name__ == "__main__":
     cfg = read_config()
-    mk_sample_prompt(cfg["model"], cfg["apps"], cfg["sample"]["sample_prompt_path"])
+    mk_sample_prompt(cfg["model"], cfg["apps"], cfg["sample"]["sample_prompt_path"])
\ No newline at end of file
+
+    from utils_vllm import vllm_chatcomplete
+    vllm_chatcomplete(
+        cfg["model"],
+        cfg["sample"]["sample_prompt_path"],
+        cfg["sample"]["sample_result_path"],
+        cfg["sample"]["sampling_params"],
+    )
\ No newline at end of file

step1_sample_code.py

-from utils_vllm import vllm_chatcomplete
-from utils import read_config
-
-
-if __name__ == "__main__":
-    cfg = read_config()
-    vllm_chatcomplete(
-        cfg["model"],
-        cfg["sample"]["sample_prompt_path"],
-        cfg["sample"]["sample_result_path"],
-        cfg["sample"]["sampling_params"],
-    )