Update src/prompts.py

src/prompts.py

@@ -266,6 +266,9 @@ INIT_PROMPT_TEMPLATE = """\
 You are a query-routing classifier for a {domain} research chatbot.
 Your task is to read the user's query and extract the country, value chain, and investment area in which the PERI analysis is to be run. Return ONLY valid JSON.
 Place all output values in a list and if not specified in the user query return None
+
+Match labels included in the user queries with options included in this items list:
+{approved_options}
 Output:
 {{
   "country": ["Malawi"] or None,
@@ -275,6 +278,21 @@ Output:
 USER'S QUESTION: {query}
 Return only the JSON object. Do not provide explanations, markdown, or additional text.
 """
+def dict_to_string(d, indent=0):
+    result = ""
+    for key, value in d.items():
+        # Add indentation based on nesting depth
+        spacing = "  " * indent
+        
+        if isinstance(value, dict):
+            result += f"{spacing}{key}:\n"
+            # Recursively call the function for nested dictionaries
+            result += dict_to_string(value, indent + 1)
+        else:
+            result += f"{spacing}{key}: {value}\n"
+            
+    return result
+
 def build_stata_prompt(query:str, 
                        subcmd:str, 
                        dofx:str, 
@@ -292,7 +310,8 @@ def build_categorizer_prompt(query:str,
                              ):
     return CATEGORIZER_PROMPT_TEMPLATE.format(
         domain=domain,
-        query=query
+        query=query,
+        approved_options=dict_to_string(APPROVED_OPTIONS, indent=4)
     )
 
 def build_init_prompt(query:str,