changed formatting templates

chains/diagnoser_chain.py

@@ -12,19 +12,64 @@ class DiagnoserChain(BaseModel):
     async def run(self, user_query: str, exercise_format: str) -> str:
         """
         Runs the composite chain:
-          1. Standardizes the exercise description (if exercise_format isn't Raw).
+          1. Standardizes the exercise formatting (if exercise_format isn't Raw).
           2. Generates a diagnosis from the standardized format.
         """
-        # --- Step 1: Standardize the exercise description ---
+        # --- Step 1: Standardize the exercise formatting ---
         if exercise_format == "Raw (original)":
             standardized_exercise = user_query
         else:
             mapping = {
-                "Markdown": "Please format the exercise in Markdown.",
-                "XML": "Please format the exercise in XML, using '",
-                "Plaintext": "Please format the exercise in plain text."
+                "Markdown": (
+                    "Please format the exercise in Markdown, similarly to this example:\n\n"
+                    "**Theorie**  \n"
+                    "Eenzaamheid wordt door ieder persoon anders ervaren en is daarom subjectief.\n\n"
+                    "---\n\n"
+                    "**Vraag**  \n"
+                    "Wat is de meest passende definitie van eenzaamheid?\n\n"
+                    "1. Het gevoel geen connectie te hebben met anderen  \n"
+                    "2. Regelmatig in je eentje zijn  \n"
+                    "3. Beide bovenstaande  \n"
+                    "4. Geen van bovenstaande  \n\n"
+                    "**Correct antwoord:**  \n"
+                    "1. Het gevoel geen connectie te hebben met anderen."
+                ),
+                "XML": (
+                    "Please reformat in XML, following this example:\n"
+                    "<exercise>\n"
+                    "    <content>\n"
+                    "        <question>Theorie:\n"
+                    "Eenzaamheid wordt door ieder persoon anders ervaren en is daarom subjectief.\n\n"
+                    "Vraag:\n"
+                    "Wat is de meest passende definitie van eenzaamheid?</question>\n"
+                    "        <choices>\n"
+                    "            <choice id=\"1\">Het gevoel geen connectie te hebben met anderen</choice>\n"
+                    "            <choice id=\"2\">Regelmatig in je eentje zijn</choice>\n"
+                    "            <choice id=\"3\">Beide bovenstaande</choice>\n"
+                    "            <choice id=\"4\">Geen van bovenstaande</choice>\n"
+                    "        </choices>\n"
+                    "    </content>\n"
+                    "    <answer>\n"
+                    "        <correct-choice>1</correct-choice>\n"
+                    "        <explanation></explanation>\n"
+                    "    </answer>\n"
+                    "</exercise>"
+                ),
+                "Plaintext": (
+                    "Please reformat in plain text, following this example:\n\n"
+                    "Theorie\n"
+                    "Eenzaamheid wordt door ieder persoon anders ervaren en is daarom subjectief.\n\n"
+                    "Vraag\n"
+                    "Wat is de meest passende definitie van eenzaamheid?\n\n"
+                    "1. Het gevoel geen connectie te hebben met anderen\n"
+                    "2. Regelmatig in je eentje zijn\n"
+                    "3. Beide bovenstaande\n"
+                    "4. Geen van bovenstaande\n\n"
+                    "Correct antwoord:\n"
+                    "1. Het gevoel geen connectie te hebben met anderen."
+                )
             }
-            formatting_instructions = mapping.get(exercise_format, "Please format the exercise in Markdown.")
+            formatting_instructions = mapping.get(exercise_format, "Please reformat the given exercise to ease further processing.")
             prompt_std = await self.template_standardize.aformat_prompt(
                 user_input=user_query,
                 formatting_instructions=formatting_instructions

config/chain_configs.py

@@ -10,12 +10,12 @@ chain_configs = {
         "class": DiagnoserChain,
         "template_standardize": standardize_template,
         "template_diagnose": diagnose_template,
-        "llm_standardize": llms["gpt-4o-mini"],  # Always fixed
-        "llm_diagnose": llms["gpt-4o"],          # Default; can be replaced in UI
+        "llm_standardize": llms["GPT-4o-mini"],  # Always fixed
+        "llm_diagnose": llms["GPT-4o"],          # Default; can be replaced in UI
     },
     "distractors": {
         "class": DistractorsChain,
         "template": distractors_template,
-        "llm": llms["gpt-4o"],
+        "llm": llms["GPT-4o"],
     },
 }