IQKiller / prompt_loader.py
AvikalpK's picture
๐Ÿš€ IQKiller: AI-Powered Job Analysis Platform
4655858
raw
history blame contribute delete
474 Bytes
import yaml
from typing import Dict, Any
class PromptLoader:
def __init__(self, prompt_file: str = "prompts/v1.yaml"):
with open(prompt_file, 'r') as f:
self.prompts = yaml.safe_load(f)
def get_prompt(self, prompt_name: str, **kwargs) -> str:
"""Get and format a prompt with variables"""
template = self.prompts.get(prompt_name, "")
return template.format(**kwargs)
# Global instance
prompt_loader = PromptLoader()