Refactor prompt handling WIP

src/bpmn_assistant/prompts/__init__.py

@@ -0,0 +1,3 @@
+from .prompt_template_processor import PromptTemplateProcessor
+
+__all__ = ["PromptTemplateProcessor"]

src/bpmn_assistant/prompts/make_final_comment.txt → src/bpmn_assistant/prompts/make_final_comment.jinja2

@@ -1,6 +1,6 @@
 Message history:
 
-::message_history
+{{ message_history }}
 
 ---
 
@@ -12,7 +12,7 @@ The last user message indicates that the user wanted to create or modify a BPMN
 
 You have made the requested modification to the BPMN process, and this is the updated BPMN process in JSON format:
 ```json
-::process
+{{ process }}
 ```
 
 Make a final comment to the user, explaining that the BPMN process has been successfully created or modified.

src/bpmn_assistant/prompts/prompt_template_processor.py

@@ -0,0 +1,45 @@
+import os
+
+from jinja2 import Environment, FileSystemLoader, select_autoescape
+
+
+class PromptTemplateProcessor:
+    def __init__(self, prompts_dir=os.path.dirname(os.path.abspath(__file__))):
+        """
+        Initialize the template processor with a directory containing prompt templates.
+
+        Args:
+            prompts_dir (str): Path to the directory containing prompt templates
+        """
+        self.env = Environment(
+            loader=FileSystemLoader(prompts_dir),
+            autoescape=select_autoescape(),
+            trim_blocks=True,
+            lstrip_blocks=True,
+        )
+
+    def render_template(self, template_name, **kwargs):
+        """
+        Render a template with the provided variables.
+
+        Args:
+            template_name (str): Name of the template file
+            **kwargs: Variables to pass to the template
+
+        Returns:
+            str: Rendered template string
+        """
+        template = self.env.get_template(template_name)
+        return template.render(**kwargs)
+
+
+if __name__ == "__main__":
+    processor = PromptTemplateProcessor()
+
+    variables = {
+        "message_history": "This should be the message history!",
+        # "process": "This should be the BPMN process, which is optional!",
+    }
+
+    prompt = processor.render_template("respond_to_query.jinja2", **variables)
+    print(prompt)