Update config.py

config.py

@@ -8,8 +8,13 @@ HF_TOKEN = os.environ.get("HF_TOKEN")
 GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY") 
 
+
 CURRENT_DIR = Path(__file__).parent
-_PROMPTS = safe_load(CURRENT_DIR.joinpath("prompts.yaml").read_text())
+# Try both extensions
+_prompts_path = CURRENT_DIR.joinpath("prompts.yaml")
+if not _prompts_path.exists():
+    _prompts_path = CURRENT_DIR.joinpath("prompts.yml")
+_PROMPTS = safe_load(_prompts_path.read_text(encoding="utf-8"))
 
 
 def get_prompt(prompt_key: str, **kwargs: str) -> str: