Update src/prompts.py

src/prompts.py

@@ -243,51 +243,44 @@ STATA_PROMPT_TEMPLATE = {
             "explain": "Explain what the following stata error code or message means. If code is provided, "
                         "describe the error in context of the code."}
 }
-
 CATEGORIZER_PROMPT_TEMPLATE = """\
 You are a query-routing classifier for a {domain} research chatbot.
-Your task is to read the user's query and classify it into exactly one category and action. Return ONLY valid JSON.
+Your task is to read the user's query and classify it into exactly one category. Return ONLY valid JSON.
 Classification rules:
-1. Contact and Information Sharing
-- Use when the user wants to contact the WEAI team.
+1. Pillar 1
+- Use when the user shows interest in understanding whether the selected value chain aligns with the government's political incentives.
 - If user input indicates the issue is resolved output "proceed".
   Output:
   {{
-    "category": "contact_and_info",
-    "action": "contact",
+    "category": "pillar_1",
     "sub-action":"proceed"or"unresolved"
   }}
-- Use when the user wants to share, submit, upload, or send files/documents/data.
-- If user input indicates the issue is resolved output "proceed".
-  Output:
-  {{
-    "category": "contact_and_info",
-    "action": "upload",
+2. Pillar 2
+- Use when the user shows interest in understanding to what degree decisions are impacted by the lobbying of particular groups or by elite influence
+Output:
+{{
+  "category": "pillar_2",
     "sub-action":"proceed"or"unresolved"
-  }}
-2. Stata Implementation
-- Questions about Stata programming, do-files, code snippets, scripts, syntax, or debugging.
-- Classify as:
-  - "do" for do-files
-  - "code" for Stata code or programming help
-  - "error" for debugging or error messages
-- If user input contains the sub-action include all selected options.
-- Unless explicitly stated in the user input, always return "unresolved".
-- Always include sub-action options in a list unless it's "unresolved".
-- If user input indicates the issue is resolved output "proceed".
+}}
+3. Pillar 3
+- Use when the user shows interest in understanding if the value chain/or investment intervention can be feasibly implemented given the broader institutional and policy environment
 Output:
 {{
-  "category": "stata",
-  "action": "do" | "code" | "error",
-  "sub-action":"rewrite"or"explain"or"suggestfix"or"unresolved"or"proceed"
+  "category": "pillar_3",
+  "sub-action":"proceed"or"unresolved"
 }}
-3. General Knowledge
-- Any WEAI-related question that does not fit the above categories.
-- If user input indicates the issue is resolved output "proceed".
+4. Complete Analysis
+- Use if the input query indicates interest in all three pillars
+Output:
+{{
+  "category": "complete",
+  "sub-action":"proceed"or"unresolved"
+}}
+4. Qualitative Discussion
+- User shows interest in understanding if there are any political risks that could undermine the selected value chain or investment intervention
 Output:
 {{
-  "category": "general",
-  "action": "continue",
+  "category": "complete",
   "sub-action":"proceed"or"unresolved"
 }}
 USER'S QUESTION: {query}