上传所有文件

prompt_scripts/__init__.py

@@ -0,0 +1,21 @@
+"""
+Description :   description
+Author      :   Ruidi Qiu (r.qiu@tum.de)
+Time        :   2024/3/23 11:30:00
+LastEdited  :   2024/7/24 11:43:13
+"""
+
+from .base_script import BaseScript, BaseScriptStage
+from .script_pychecker import WF_pychecker
+from .script_directgen import WF_directgen
+
+SCRIPTS_SELECTER = {
+    "pychecker": WF_pychecker,
+    "directgen": WF_directgen
+}
+
+def get_script(script_name:str) -> BaseScript:
+    if script_name in SCRIPTS_SELECTER:
+        return SCRIPTS_SELECTER[script_name]
+    else:
+        raise ValueError(f"script name {script_name} is not supported")