feat: check config keys are set properly

app_langchain.py

@@ -3,6 +3,10 @@ import yaml
 import requests
 import re
 import os
+
+from langchain_core.prompts import PromptTemplate
+import streamlit as st
+
 from src.pdfParser import get_pdf_text
 
 # Get HuggingFace API key
@@ -11,12 +15,39 @@ api_key = os.getenv(api_key_name)
 if api_key is None:
     st.error(f"Failed to read `{api_key_name}`. Ensure the token is correctly located")
 
+# Load in model configuration and check the required keys are present
+model_config_dir = "config/model_config.yml"
+config_keys = ["system_message", "model_id", "template"]
 
-with open("config/model_config.yml", "r") as file:
+with open(model_config_dir, "r") as file:
     model_config = yaml.safe_load(file)
 
+for var in model_config.keys():
+    if var not in config_keys:
+        raise ValueError(f"`{var}` key missing from `{model_config_dir}`")
+    
 system_message = model_config["system_message"]
 model_id = model_config["model_id"]
+template = model_config["template"]
+
+prompt_template = PromptTemplate(
+    template=template,
+    input_variables=["system_message", "user_message"]
+)
+
+
+
+
+
+
+
+
+
+
+
+
+
+
 
 
 def query(payload, model_id):