update tool description

prepare_env.py

@@ -225,66 +225,227 @@ def create_section_tools(embedding_function, llm):
         guide_retriever = create_retriever(guide_vstore, guide_docs, llm)
     else:
         guide_retriever = None
-    
-    # Define tool functions
+    # General-purpose tool (entire Algerian guide)
     def guide_retrieval_tool(query: str) -> str:
-        """General-purpose retrieval tool for the entire Algerian National Vaccination Guide"""
+        """
+        General-purpose retrieval tool for the entire Algerian National Vaccination Guide (2023).
+
+        Use this tool when a query spans multiple sections or cannot be routed confidently to a specific tool.
+        This is the fallback and all-encompassing tool to retrieve any vaccination-related information 
+        from the national guide.
+
+        Secondary source: The WHO Immunization Guide can be queried separately via `immunization_tool`.
+
+        Args:
+            query (str): A general or complex question related to vaccination policy, schedules, or practice.
+
+        Returns:
+            str: Synthesized response based on the full Algerian guide.
+        """
         if not guide_retriever:
             return "Guide retriever not available"
         return section_tool_wrapper(guide_retriever, guide_path, query)
-    
-    # Immunization in Practice_WHO_eng_2015 retriever
+
+    # Primary + Secondary Document Paths
     immunization_path = './data/Immunization_in_Practice_WHO_eng_2015.json'
+
+    # WHO Immunization in Practice Tool
     if os.path.exists(immunization_path):
-        immunization_vstore, immunization_docs = create_vectorstore_from_json(immunization_path, "Immunization_in_Practice_WHO_eng_2015", embedding_function)
+        immunization_vstore, immunization_docs = create_vectorstore_from_json(
+            immunization_path,
+            "Immunization_in_Practice_WHO_eng_2015",
+            embedding_function
+        )
         immunization_retriever = create_retriever(immunization_vstore, immunization_docs, llm)
     else:
         immunization_retriever = None
-    
+
     def immunization_tool(query: str) -> str:
-        """Immunization in Practice WHO 2015 retrieval tool"""
+        """
+        WHO Immunization in Practice 2015 retrieval tool.
+
+        Use this tool to provide global best practices and operational guidance on immunization,
+        especially when context or clarification is needed beyond the Algerian national guide.
+        This can serve as a secondary source for training, logistics, and procedural reference.
+
+        Args:
+            query (str): A question related to immunization practice in general.
+
+        Returns:
+            str: Retrieved guidance from the WHO Immunization in Practice manual (2015).
+        """
         if not immunization_retriever:
             return "Immunization in Practice retriever not available"
         return section_tool_wrapper(immunization_retriever, immunization_path, query)
-    
+
+
+
+    # Section-Specific Tools (Primary: Algerian National Vaccination Guide)
+
     def section_one_tool(query: str) -> str:
-        """Section 1: Programme Élargi de Vaccination"""
+        """
+        Section 1: Programme Élargi de Vaccination (PEV)
+
+        Use this tool to retrieve information about the Algerian immunization program:
+        its objectives, historical background, strengths and weaknesses, and justification
+        for calendar updates.
+
+        Primary source: Algerian National Vaccination Guide, Section 1.
+        Secondary source for operational benchmarks: WHO Immunization in Practice (optional).
+
+        Args:
+            query (str): A question about Algeria’s national immunization strategy.
+
+        Returns:
+            str: Relevant content from Section 1 of the guide.
+        """
         return section_tool_wrapper(section_retrievers['one'], section_paths['one'], query)
-    
+
+
     def section_two_tool(query: str) -> str:
-        """Section 2: Maladies Ciblées"""
+        """
+        Section 2: Maladies Ciblées par la Vaccination
+
+        Use this tool for questions about the diseases targeted by the national vaccination calendar:
+        symptoms, transmission, complications, and prevention strategies.
+
+        Primary source: Algerian National Guide, Section 2.
+        Secondary source: WHO guide may support contextual insights.
+
+        Args:
+            query (str): A question about a vaccine-preventable disease (e.g. polio, rougeole).
+
+        Returns:
+            str: Disease-specific guidance from Section 2.
+        """
         return section_tool_wrapper(section_retrievers['two'], section_paths['two'], query)
-    
+
+
     def section_three_tool(query: str) -> str:
-        """Section 3: Vaccins du Calendrier"""
+        """
+        Section 3: Vaccins du Calendrier
+
+        Use this tool to retrieve technical and procedural information about the vaccines used in the calendar:
+        names, contents, administration method, and dosing details.
+
+        Args:
+            query (str): A question about a specific vaccine's type or method of use.
+
+        Returns:
+            str: Vaccine information from Section 3.
+        """
         return section_tool_wrapper(section_retrievers['three'], section_paths['three'], query)
-    
+
+
     def section_four_tool(query: str) -> str:
-        """Section 4: Rattrapage Vaccinal"""
+        """
+        Section 4: Rattrapage Vaccinal
+
+        Use this tool to determine catch-up strategies for children who missed or delayed one or more doses.
+        It provides age-adjusted rescheduling rules and justifications.
+
+        Args:
+            query (str): A question about how to manage missed vaccinations.
+
+        Returns:
+            str: Catch-up guidelines from Section 4.
+        """
         return section_tool_wrapper(section_retrievers['four'], section_paths['four'], query)
-    
+
+
     def section_five_tool(query: str) -> str:
-        """Section 5: Populations Particulières"""
+        """
+        Section 5: Vaccination des Populations Particulières
+
+        Use this tool to retrieve recommendations for specific medical contexts:
+        preterm infants, immunocompromised children, allergies (e.g. eggs), and chronic diseases.
+
+        Args:
+            query (str): A question about vaccination adaptations for vulnerable groups.
+
+        Returns:
+            str: Guidelines from Section 5.
+        """
         return section_tool_wrapper(section_retrievers['five'], section_paths['five'], query)
-    
+
+
     def section_six_tool(query: str) -> str:
-        """Section 6: Chaîne du Froid"""
+        """
+        Section 6: Chaîne du Froid
+
+        Use this tool for logistics, storage conditions, temperature monitoring,
+        and emergency procedures in case of cold chain failure.
+
+        Args:
+            query (str): A question about how vaccines should be stored and transported.
+
+        Returns:
+            str: Operational cold chain standards from Section 6.
+        """
         return section_tool_wrapper(section_retrievers['six'], section_paths['six'], query)
-    
+
+
     def section_seven_tool(query: str) -> str:
-        """Section 7: Sécurité des Injections"""
+        """
+        Section 7: Sécurité des Injections
+
+        Use this tool to ensure injection safety: handling equipment, preventing needle-stick injuries,
+        and disposing of biomedical waste.
+
+        Args:
+            query (str): A question about safe injection practices.
+
+        Returns:
+            str: Procedures and guidelines from Section 7.
+        """
         return section_tool_wrapper(section_retrievers['seven'], section_paths['seven'], query)
-    
+
+
     def section_eight_tool(query: str) -> str:
-        """Section 8: Séance de Vaccination & Vaccinovigilance"""
+        """
+        Section 8: Tenue d'une Séance de Vaccination & Vaccinovigilance
+
+        Use this tool to plan and monitor vaccination sessions, including material preparation,
+        injection recording, and handling of adverse events post-immunization (AEFI).
+
+        Args:
+            query (str): A question about session operations or vaccine side effect monitoring.
+
+        Returns:
+            str: Guidelines from Section 8.
+        """
         return section_tool_wrapper(section_retrievers['eight'], section_paths['eight'], query)
-    
+
+
     def section_nine_tool(query: str) -> str:
-        """Section 9: Planification des Séances de Vaccination"""
+        """
+        Section 9: Planification des Séances de Vaccination
+
+        Use this tool to support logistical planning: mapping, resource estimation,
+        scheduling, and stock management.
+
+        Args:
+            query (str): A question about planning and organizing vaccination sessions.
+
+        Returns:
+            str: Recommendations from Section 9.
+        """
         return section_tool_wrapper(section_retrievers['nine'], section_paths['nine'], query)
-    
+
+
     def section_ten_tool(query: str) -> str:
-        """Section 10: Mobilisation Sociale"""
+        """
+        Section 10: Mobilisation Sociale
+
+        Use this tool for strategies to increase public engagement, combat vaccine hesitancy,
+        and manage misinformation.
+
+        Args:
+            query (str): A question about public communication and trust-building around vaccines.
+
+        Returns:
+            str: Social mobilization approaches from Section 10.
+        """
         return section_tool_wrapper(section_retrievers['ten'], section_paths['ten'], query)
     
     # Create FunctionTool objects