Refactored prompt-template related code (new package interprompt)

scripts/gen_prompt_factory.py

@@ -1,45 +1,22 @@
+"""
+Autogenerates the `prompt_factory.py` module 
+"""
+
+from pathlib import Path
+
 from sensai.util import logging
 
-from serena.llm.multilang_prompt import MultiLangPromptTemplateCollection
+from interprompt import autogenerate_prompt_factory_module
+from serena.constants import PROMPT_TEMPLATES_DIR, REPO_ROOT
 
 log = logging.getLogger(__name__)
 
 
 def main():
-    coll = MultiLangPromptTemplateCollection()
-
-    package = "src/serena"
-
-    # collect methods to generate
-    indent = "    "
-    methods = []
-    for mpt in coll.prompt_templates.values():
-        prompt_name = mpt.name
-        params = mpt.get_parameters()
-        if len(params) == 0:
-            params_str = ""
-        else:
-            params_str = ", *, " + ", ".join(params)
-        methods.append(
-            f"def create_{prompt_name}(self{params_str}) -> str:"
-            + f"\n{indent}{indent}return self._format_prompt('{prompt_name}', locals())\n\n{indent}"
-        )
-    for mpl in coll.prompt_lists.values():
-        prompt_name = mpl.name
-        methods.append(
-            f"def get_list_{prompt_name}(self) -> PromptList:" + f"\n{indent}{indent}return self._get_list('{prompt_name}')\n\n{indent}"
-        )
-
-    # write prompt factory with added methods
-    with open("code_templates/prompt_factory_template.py") as f:
-        code = f.read()
-    methods_str = "".join(methods)
-    code = code.replace("# methods", methods_str)
-
-    prompt_factory_module = f"{package}/llm/prompt_factory.py"
-    with open(prompt_factory_module, "w") as f:
-        f.write(code)
-    log.info(f"Prompt factory generated successfully in {prompt_factory_module}")
+    autogenerate_prompt_factory_module(
+        prompts_dir=PROMPT_TEMPLATES_DIR,
+        target_module_path=str(Path(REPO_ROOT) / "src" / "serena" / "prompt_factory.py"),
+    )
 
 
 if __name__ == "__main__":