Update src/prompts.py

src/prompts.py

@@ -286,7 +286,19 @@ Output:
 USER'S QUESTION: {query}
 Return only the JSON object. Do not provide explanations, markdown, or additional text.
 """
-
+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
+Output:
+{{
+  "country": ["Malawi"] or None,
+  "value_chain":["coffee"] or None,
+  "investment":["credit"] or None
+}}
+USER'S QUESTION: {query}
+Return only the JSON object. Do not provide explanations, markdown, or additional text.
+"""
 def build_stata_prompt(query:str, 
                        subcmd:str, 
                        dofx:str,