Update prompt_template.py

prompt_template.py

@@ -3,8 +3,6 @@ import yaml
 from dataclasses import dataclass, field
 from typing import Dict, List, Set, Optional
 
-from openai import OpenAI
-
 @dataclass
 class PromptTemplate:
     """
@@ -84,3 +82,37 @@ class PromptTemplate:
             {"role": "system", "content": self.system_prompt},
             {"role": "user", "content": formatted_user_message}
         ]
+
+
+def load_prompt(yaml_path: str, version: str = None) -> tuple[PromptTemplate, dict]:
+    """
+    Load prompt configuration from YAML file.
+
+    Args:
+        yaml_path: Path to YAML configuration file
+        version: Specific version to load (defaults to 'current_version')
+
+    Returns:
+        tuple: (PromptTemplate instance, generation parameters dictionary)
+
+    Example:
+        prompt, params = load_prompt('prompts.yaml', version='v2')
+    """
+    with open(yaml_path, 'r') as f:
+        data = yaml.safe_load(f)
+
+    # Use specified version or fall back to current_version
+    version_to_use = version or data.get('current_version')
+    if version_to_use not in data:
+        raise KeyError(f"Version '{version_to_use}' not found in {yaml_path}")
+
+    version_data = data[version_to_use]
+
+    prompt = PromptTemplate(
+        system_prompt=version_data['system_prompt'],
+        user_template=version_data['user_template']
+    )
+
+    generation_params = version_data.get('generation_params', {})
+
+    return prompt, generation_params