update the system prompt

prepare_env.py

@@ -310,22 +310,33 @@ def create_section_tools(embedding_function, llm):
     # General-purpose tool (entire Algerian guide)
     def guide_retrieval_tool(query: str) -> str:
         """
-        General-purpose retrieval tool for the entire Algerian National Vaccination Guide (2023).
-
-        Use ONLY when a query spans multiple unrelated sections or cannot be confidently routed
-        to a more specific tool. This tool provides a fallback when the intent is ambiguous
-        or multi-topic (e.g., combining schedule, cold chain, and public outreach in one query).
-
-        Do NOT use this tool for clearly scoped questions related to vaccination schedules, disease profiles,
-        vaccine logistics, or procedural workflows — use the specific section tools instead.
-
-        Secondary source: WHO Immunization Guide (use `immunization_tool` when broader context is required).
+        General-purpose retrieval tool that searches across the entire Algerian National Vaccination Guide (2023).
+        This tool is used to handle broad or ambiguous queries that may span multiple sections, or when no specific
+        section tool is confidently matched.
+
+        It can respond to questions covering all major topics in the guide, including:
+        - The national vaccination program (PEV)
+        - Targeted vaccine-preventable diseases
+        - Vaccine types and administration methods
+        - Catch-up vaccination schedules
+        - Vaccination of special populations (e.g., premature, immunocompromised)
+        - Cold chain logistics and storage protocols
+        - Injection safety practices
+        - Session planning and vaccine stock management
+        - Community mobilization and vaccine hesitancy
+        - Operational workflows and vaccinovigilance procedures
+
+        This tool provides coverage across Sections 1 to 10 and complements the section-specific tools.
+        It is especially useful when:
+            - A question involves content from multiple sections
+            - Tool routing is uncertain or ambiguous
+            - A complete or exploratory answer is needed from the full document
 
         Args:
-            query (str): A general, complex, or cross-sectional vaccination query.
+            query (str): A question related to any topic within the Algerian National Vaccination Guide.
 
         Returns:
-            str: Synthesized answer from the entire national guide.
+            str: A synthesized response generated from relevant content retrieved across all sections.
         """
         print(f"🏥 GUIDE TOOL CALLED: {query[:50]}...")
         if not guide_retriever:
@@ -339,13 +350,8 @@ def create_section_tools(embedding_function, llm):
 
     def immunization_tool(query: str) -> str:
         """
-        WHO Immunization in Practice 2015 retrieval tool.
+        A comprehensive tool for querying global immunization practices, based on the 2015 WHO 'Immunization in Practice: A Practical Guide for Health Staff'. This tool provides detailed information on vaccine-preventable diseases (e.g., diphtheria, measles, polio, hepatitis B, Hib), vaccination schedules, cold chain management, safe injection practices, microplanning, immunization session management, monitoring, surveillance, and community engagement strategies. It is ideal for questions about global immunization programs, disease prevention, vaccine administration, and operational guidelines for health workers, with a focus on practical implementation at health facility and sub-national levels
 
-        Use ONLY when global guidance or procedural context is needed that is not covered or
-        is unclear in the Algerian guide. This is a secondary reference for training standards,
-        general immunization logistics, and vaccine delivery practices.
-
-        Do NOT use this tool to answer country-specific policy or scheduling questions.
 
         Args:
             query (str): A question seeking global immunization practices.
@@ -366,7 +372,7 @@ def create_section_tools(embedding_function, llm):
     # Section-Specific Tools - Fixed implementation
     def section_one_tool(query: str) -> str:
         """
-        Section 1: Programme Élargi de Vaccination (PEV)
+        Section 1: Programme Élargi de Vaccination (PEV) from the Algerian National Vaccination Guide
 
         Use for queries about the national immunization program structure: its objectives,
         history, evaluation, and rationale for updates to the Algerian calendar.
@@ -391,7 +397,7 @@ def create_section_tools(embedding_function, llm):
 
     def section_two_tool(query: str) -> str:
         """
-        Section 2: Maladies Ciblées
+        Section 2: Maladies Ciblées from the Algerian National Vaccination Guide
 
         Use ONLY for questions about the characteristics of vaccine-preventable diseases:
         symptoms, transmission, complications, and prevention.
@@ -416,7 +422,7 @@ def create_section_tools(embedding_function, llm):
 
     def section_three_tool(query: str) -> str:
         """
-        Section 3: Vaccins du Calendrier
+        Section 3: Vaccins du Calendrier from the Algerian National Vaccination Guide
 
         Use ONLY for questions about the vaccines themselves: their types, compositions,
         methods of administration, and how they work.
@@ -441,7 +447,7 @@ def create_section_tools(embedding_function, llm):
 
     def section_four_tool(query: str) -> str:
         """
-        Section 4: Rattrapage Vaccinal
+        Section 4: Rattrapage Vaccinal from the Algerian National Vaccination Guide
 
         Use ONLY when the question involves missed or delayed vaccinations and how to reschedule them
         based on the child's current age.
@@ -466,7 +472,7 @@ def create_section_tools(embedding_function, llm):
 
     def section_five_tool(query: str) -> str:
         """
-        Section 5: Populations Particulières
+        Section 5: Populations Particulières from the Algerian National Vaccination Guide
 
         Use ONLY for vaccination questions concerning special populations:
         preterm infants, immunosuppressed patients, chronic illness, or allergy conditions.
@@ -491,7 +497,7 @@ def create_section_tools(embedding_function, llm):
 
     def section_six_tool(query: str) -> str:
         """
-        Section 6: Chaîne du Froid
+        Section 6: Chaîne du Froid from the Algerian National Vaccination Guide
 
         Use ONLY for questions about vaccine storage, transport, cold chain equipment,
         temperature monitoring, or cold chain failures.
@@ -515,8 +521,8 @@ def create_section_tools(embedding_function, llm):
             return f"Error accessing section 6: {str(e)}"
 
     def section_seven_tool(query: str) -> str:
-        """
-        Section 7: Sécurité des Injections
+        """ 
+        Section 7: Sécurité des Injections from the Algerian National Vaccination Guide
 
         Use ONLY for questions related to the safe administration of vaccines:
         equipment use, technique, safety precautions, and waste disposal.
@@ -541,7 +547,7 @@ def create_section_tools(embedding_function, llm):
 
     def section_eight_tool(query: str) -> str:
         """
-        Section 8: Séance de Vaccination & Vaccinovigilance
+        Section 8: Séance de Vaccination & Vaccinovigilance from the Algerian National Vaccination Guide
 
         Use ONLY for questions about running a vaccination session, preparing the setting,
         recording injections, and monitoring for adverse events (AEFI).
@@ -566,7 +572,7 @@ def create_section_tools(embedding_function, llm):
 
     def section_nine_tool(query: str) -> str:
         """
-        Section 9: Planification des Séances de Vaccination
+        Section 9: Planification des Séances de Vaccination from the Algerian National Vaccination Guide
 
         Use ONLY for planning and logistics questions: session scheduling, stock estimation,
         and operational preparation at the facility level.
@@ -591,7 +597,7 @@ def create_section_tools(embedding_function, llm):
 
     def section_ten_tool(query: str) -> str:
         """
-        Section 10: Mobilisation Sociale
+        Section 10: Mobilisation Sociale from the Algerian National Vaccination Guide
 
         Use ONLY for questions about communication strategies, overcoming vaccine hesitancy,
         rumor management, or community outreach.

rag_pipeline.py

@@ -116,7 +116,7 @@ def create_safe_custom_prompt(tools, llm, is_fallback=False):
     if is_fallback:
         custom_instructions = """
 ## MEDICAL ASSISTANT ROLE - FALLBACK MODE
-You are a helpful and knowledgeable AI-powered vaccine assistant designed to support doctors in clinical decision-making.
+You are a helpful and knowledgeable AI-powered vaccine assistant designed to support the Algerian doctors in clinical decision-making.
 You are operating in FALLBACK MODE with access to only the most essential and comprehensive tools.
 You provide evidence-based guidance using only information from official vaccine medical documents.
 Answer the doctor's question accurately and concisely using only the provided information.
@@ -147,7 +147,7 @@ Answer the doctor's question accurately and concisely using only the provided in
     else:
         custom_instructions = """
 ## MEDICAL ASSISTANT ROLE
-You are a helpful and knowledgeable AI-powered vaccine assistant designed to support doctors in clinical decision-making.
+You are a helpful and knowledgeable AI-powered vaccine assistant designed to support the Algerian doctors in clinical decision-making.
 You provide evidence-based guidance using only information from official vaccine medical documents.
 Answer the doctor's question accurately and concisely using only the provided information.