update tools discription

prepare_env.py

@@ -269,13 +269,15 @@ def section_tool_wrapper(retriever, section_path_chunks, query):
         return f"Error retrieving documents: {str(e)}"
 
 def create_section_tools(embedding_function, llm):
-    """Create all section-specific retrieval tools"""
-    print("🛠️ Creating section-specific retrieval tools...")
-    
+    """
+    Create all section-specific retrieval tools with improved descriptions for accurate routing.
+    """
+    print("🛠️ Creating section-specific retrieval tools with enhanced descriptions...")
+
     # Define section paths - Fixed path structure
     section_paths = {
         'one': './data/section_one_chunks.json',
-        'two': './data/section_two_chunks.json', 
+        'two': './data/section_two_chunks.json',
         'three': './data/section_three_chunks.json',
         'four': './data/section_four_chunks.json',
         'five': './data/section_five_chunks.json',
@@ -285,7 +287,7 @@ def create_section_tools(embedding_function, llm):
         'nine': './data/section_nine_chunks.json',
         'ten': './data/section_ten_chunks.json'
     }
-    
+
     # Create retrievers for each section
     section_retrievers = {}
     for section, path in section_paths.items():
@@ -301,7 +303,7 @@ def create_section_tools(embedding_function, llm):
         except Exception as e:
             print(f"❌ Error creating retriever for section {section}: {e}")
             section_retrievers[section] = None
-    
+
     # Create main guide retriever
     guide_path = './data/Guide-pratique-de-mise-en-oeuvre-du-calendrier-national-de-vaccination-2023.json'
     guide_retriever = None
@@ -334,77 +336,48 @@ def create_section_tools(embedding_function, llm):
     except Exception as e:
         print(f"❌ Error creating immunization retriever: {e}")
 
-    # General-purpose tool (entire Algerian guide)
-    def guide_retrieval_tool(query: str) -> str:
+    # --- Tool Definitions with Improved Descriptions ---
+
+    def general_guide_tool(query: str) -> str:
         """
-        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
+        A general-purpose tool for the Algerian National Vaccination Guide.
+        **Use this tool as a fallback** if no other specific tool seems appropriate, or for very broad, multi-topic questions
+        (e.g., 'Summarize the Algerian vaccination policy and its safety measures').
+        **Always prefer a more specific tool if the query matches its description** (e.g., use 'cold_chain_tool' for temperature questions).
 
         Args:
-            query (str): A question related to any topic within the Algerian National Vaccination Guide.
+            query (str): A broad or ambiguous question about the Algerian National Vaccination Guide.
 
         Returns:
-            str: A synthesized response generated from relevant content retrieved across all sections.
+            str: Content retrieved from the entire guide.
         """
-        print(f"🏥 GUIDE TOOL CALLED: {query[:50]}...")
+        print(f"🏥 GENERAL GUIDE TOOL CALLED (FALLBACK): {query[:50]}...")
         if not guide_retriever:
-            print("❌ Guide retriever not available - main guide file may be missing")
             return "Guide retriever not available - main guide file may be missing"
-        try:
-            return section_tool_wrapper(guide_retriever, guide_path, query)
-        except Exception as e:
-            print(f"❌ Error accessing guide retriever: {str(e)}")
-            return f"Error accessing guide retriever: {str(e)}"
+        return section_tool_wrapper(guide_retriever, guide_path, query)
 
-    def immunization_tool(query: str) -> str:
+    def who_immunization_tool(query: str) -> str:
         """
-        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
-
+        Provides information from the WHO's 'Immunization in Practice' guide. Use this for questions about
+        **global immunization standards**, international best practices, or for comparing Algerian policy to
+        general WHO recommendations on topics like cold chain, safety, and disease control.
 
         Args:
-            query (str): A question seeking global immunization practices.
+            query (str): A question seeking global or general immunization practices.
 
         Returns:
             str: Content from the WHO Immunization in Practice guide.
         """
         print(f"🌍 WHO TOOL CALLED: {query[:50]}...")
         if not immunization_retriever:
-            print("❌ Immunization in Practice retriever not available - WHO guide file may be missing")
             return "Immunization in Practice retriever not available - WHO guide file may be missing"
-        try:
-            return section_tool_wrapper(immunization_retriever, immunization_path, query)
-        except Exception as e:
-            print(f"❌ Error accessing immunization retriever: {str(e)}")
-            return f"Error accessing immunization retriever: {str(e)}"
+        return section_tool_wrapper(immunization_retriever, immunization_path, query)
 
-    # Section-Specific Tools - Fixed implementation
-    def section_one_tool(query: str) -> str:
+    def program_overview_tool(query: str) -> str:
         """
-        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.
-
-        Do NOT use for vaccine schedules, disease information, or administration techniques.
+        (Section 1) The primary tool for questions about the **history, objectives, and structure** of Algeria's
+        national immunization program (PEV - Programme Élargi de Vaccination). Use this for topics like
+        the program's rationale, key achievements, and the reasons for updates to the vaccination calendar.
 
         Args:
             query (str): A question about the foundation or evolution of the PEV.
@@ -412,24 +385,16 @@ def create_section_tools(embedding_function, llm):
         Returns:
             str: Response from Section 1.
         """
-        print(f"📋 SECTION 1 TOOL CALLED: {query[:50]}...")
+        print(f"📋 PROGRAM OVERVIEW (S1) TOOL CALLED: {query[:50]}...")
         if not section_retrievers.get('one'):
-            print("❌ Section 1 retriever not available - file may be missing")
-            return "Section 1 retriever not available - file may be missing"
-        try:
-            return section_tool_wrapper(section_retrievers['one'], section_paths['one'], query)
-        except Exception as e:
-            print(f"❌ Error accessing section 1: {str(e)}")
-            return f"Error accessing section 1: {str(e)}"
+            return "Section 1 retriever not available"
+        return section_tool_wrapper(section_retrievers['one'], section_paths['one'], query)
 
-    def section_two_tool(query: str) -> str:
+    def disease_info_tool(query: str) -> str:
         """
-        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.
-
-        Do NOT use for questions about vaccines, administration schedules, or procedures.
+        (Section 2) The definitive tool for information on **specific vaccine-preventable diseases**.
+        Use this to find details on **symptoms, transmission methods, complications**, and prevention
+        strategies for diseases like Diphtheria, Measles, Polio, Tetanus, etc.
 
         Args:
             query (str): A question about a disease covered by the national vaccination program.
@@ -437,99 +402,67 @@ def create_section_tools(embedding_function, llm):
         Returns:
             str: Disease-specific content from Section 2.
         """
-        print(f"🦠 SECTION 2 TOOL CALLED: {query[:50]}...")
+        print(f"🦠 DISEASE INFO (S2) TOOL CALLED: {query[:50]}...")
         if not section_retrievers.get('two'):
-            print("❌ Section 2 retriever not available - file may be missing")
-            return "Section 2 retriever not available - file may be missing"
-        try:
-            return section_tool_wrapper(section_retrievers['two'], section_paths['two'], query)
-        except Exception as e:
-            print(f"❌ Error accessing section 2: {str(e)}")
-            return f"Error accessing section 2: {str(e)}"
+            return "Section 2 retriever not available"
+        return section_tool_wrapper(section_retrievers['two'], section_paths['two'], query)
 
-    def section_three_tool(query: str) -> str:
+    def vaccine_properties_tool(query: str) -> str:
         """
-        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.
-
-        Do NOT use for schedule timing, catch-up protocols, or disease information.
+        (Section 3) The specific tool for questions about the **vaccines themselves**: their types (e.g., BCG, ROR,
+        DTCaVPI), composition, whether they are live or inactivated, and the correct **method of administration**
+        (e.g., intradermal, intramuscular, oral).
 
         Args:
-            query (str): A question about a vaccine's formulation or method of delivery.
+            query (str): A question about a vaccine's formulation or how it is administered.
 
         Returns:
-            str: Vaccine info from Section 3.
+            str: Vaccine-specific info from Section 3.
         """
-        print(f"💉 SECTION 3 TOOL CALLED: {query[:50]}...")
+        print(f"💉 VACCINE PROPERTIES (S3) TOOL CALLED: {query[:50]}...")
         if not section_retrievers.get('three'):
-            print("❌ Section 3 retriever not available - file may be missing")
-            return "Section 3 retriever not available - file may be missing"
-        try:
-            return section_tool_wrapper(section_retrievers['three'], section_paths['three'], query)
-        except Exception as e:
-            print(f"❌ Error accessing section 3: {str(e)}")
-            return f"Error accessing section 3: {str(e)}"
+            return "Section 3 retriever not available"
+        return section_tool_wrapper(section_retrievers['three'], section_paths['three'], query)
 
-    def section_four_tool(query: str) -> str:
+    def catch_up_vaccination_tool(query: str) -> str:
         """
-        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.
-
-        Do NOT use for standard schedules (on-time), vaccine properties, or cold chain issues.
+        (Section 4) Specialized tool for **missed or delayed vaccinations (rattrapage vaccinal)**.
+        Use this for questions about creating a **catch-up schedule** for a child who is behind
+        on their shots, based on their age and vaccination history.
 
         Args:
-            query (str): A question about catch-up vaccination based on delay or omission.
+            query (str): A question about catch-up vaccination due to a delay or missed dose.
 
         Returns:
-            str: Catch-up guidance from Section 4.
+            str: Catch-up schedule guidance from Section 4.
         """
-        print(f"🔄 SECTION 4 TOOL CALLED: {query[:50]}...")
+        print(f"🔄 CATCH-UP (S4) TOOL CALLED: {query[:50]}...")
         if not section_retrievers.get('four'):
-            print("❌ Section 4 retriever not available - file may be missing")
-            return "Section 4 retriever not available - file may be missing"
-        try:
-            return section_tool_wrapper(section_retrievers['four'], section_paths['four'], query)
-        except Exception as e:
-            print(f"❌ Error accessing section 4: {str(e)}")
-            return f"Error accessing section 4: {str(e)}"
+            return "Section 4 retriever not available"
+        return section_tool_wrapper(section_retrievers['four'], section_paths['four'], query)
 
-    def section_five_tool(query: str) -> str:
+    def special_populations_tool(query: str) -> str:
         """
-        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.
-
-        Do NOT use for general population, standard calendar, or vaccine preparation.
+        (Section 5) The designated tool for vaccination guidelines concerning **special populations**.
+        Use for questions about vaccinating preterm infants, allergic children, or patients with
+        immunosuppression, chronic illnesses (cardiac, pulmonary), or other specific health conditions.
 
         Args:
-            query (str): A question about tailored vaccination for vulnerable groups.
+            query (str): A question about tailored vaccination for a vulnerable or special group.
 
         Returns:
             str: Custom recommendations from Section 5.
         """
-        print(f"👥 SECTION 5 TOOL CALLED: {query[:50]}...")
+        print(f"👥 SPECIAL POPULATIONS (S5) TOOL CALLED: {query[:50]}...")
         if not section_retrievers.get('five'):
-            print("❌ Section 5 retriever not available - file may be missing")
-            return "Section 5 retriever not available - file may be missing"
-        try:
-            return section_tool_wrapper(section_retrievers['five'], section_paths['five'], query)
-        except Exception as e:
-            print(f"❌ Error accessing section 5: {str(e)}")
-            return f"Error accessing section 5: {str(e)}"
+            return "Section 5 retriever not available"
+        return section_tool_wrapper(section_retrievers['five'], section_paths['five'], query)
 
-    def section_six_tool(query: str) -> str:
+    def cold_chain_tool(query: str) -> str:
         """
-        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.
-
-        Do NOT use for dose timing, administration methods, or disease information.
+        (Section 6) The definitive tool for all questions about the **cold chain**, including vaccine **storage
+        temperatures**, transport protocols, refrigerators, temperature monitoring (like PCV pastilles),
+        and procedures for handling cold chain failures or power outages.
 
         Args:
             query (str): A logistics-related question about vaccine temperature management.
@@ -537,99 +470,67 @@ def create_section_tools(embedding_function, llm):
         Returns:
             str: Cold chain instructions from Section 6.
         """
-        print(f"❄️ SECTION 6 TOOL CALLED: {query[:50]}...")
+        print(f"❄️ COLD CHAIN (S6) TOOL CALLED: {query[:50]}...")
         if not section_retrievers.get('six'):
-            print("❌ Section 6 retriever not available - file may be missing")
-            return "Section 6 retriever not available - file may be missing"
-        try:
-            return section_tool_wrapper(section_retrievers['six'], section_paths['six'], query)
-        except Exception as e:
-            print(f"❌ Error accessing section 6: {str(e)}")
-            return f"Error accessing section 6: {str(e)}"
-
-    def section_seven_tool(query: str) -> str:
-        """ 
-        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.
+            return "Section 6 retriever not available"
+        return section_tool_wrapper(section_retrievers['six'], section_paths['six'], query)
 
-        Do NOT use for vaccine types, schedules, or cold chain issues.
+    def injection_safety_tool(query: str) -> str:
+        """
+        (Section 7) The primary tool for questions related to the **safe administration of injections**.
+        Use for topics like sterile equipment, proper injection techniques, preventing needlestick injuries,
+        and safe disposal of medical waste (DASRI).
 
         Args:
-            query (str): A question about how to inject vaccines safely.
+            query (str): A question about how to perform vaccine injections safely.
 
         Returns:
             str: Best practices from Section 7.
         """
-        print(f"🛡️ SECTION 7 TOOL CALLED: {query[:50]}...")
+        print(f"🛡️ INJECTION SAFETY (S7) TOOL CALLED: {query[:50]}...")
         if not section_retrievers.get('seven'):
-            print("❌ Section 7 retriever not available - file may be missing")
-            return "Section 7 retriever not available - file may be missing"
-        try:
-            return section_tool_wrapper(section_retrievers['seven'], section_paths['seven'], query)
-        except Exception as e:
-            print(f"❌ Error accessing section 7: {str(e)}")
-            return f"Error accessing section 7: {str(e)}"
+            return "Section 7 retriever not available"
+        return section_tool_wrapper(section_retrievers['seven'], section_paths['seven'], query)
 
-    def section_eight_tool(query: str) -> str:
+    def session_management_tool(query: str) -> str:
         """
-        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).
-
-        Do NOT use for disease, vaccine, or scheduling details.
+        (Section 8) Use this tool for questions about the **operational conduct of a vaccination session**
+        and **vaccinovigilance**. This includes preparing the session, material setup, registering vaccination
+        acts, and monitoring/reporting adverse events post-vaccination (MPVI).
 
         Args:
-            query (str): A question about operational conduct during vaccination.
+            query (str): A question about running a vaccination session or post-vaccine monitoring.
 
         Returns:
             str: Workflow and safety monitoring details from Section 8.
         """
-        print(f"📊 SECTION 8 TOOL CALLED: {query[:50]}...")
+        print(f"📊 SESSION MGMT (S8) TOOL CALLED: {query[:50]}...")
         if not section_retrievers.get('eight'):
-            print("❌ Section 8 retriever not available - file may be missing")
-            return "Section 8 retriever not available - file may be missing"
-        try:
-            return section_tool_wrapper(section_retrievers['eight'], section_paths['eight'], query)
-        except Exception as e:
-            print(f"❌ Error accessing section 8: {str(e)}")
-            return f"Error accessing section 8: {str(e)}"
+            return "Section 8 retriever not available"
+        return section_tool_wrapper(section_retrievers['eight'], section_paths['eight'], query)
 
-    def section_nine_tool(query: str) -> str:
+    def planning_and_logistics_tool(query: str) -> str:
         """
-        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.
-
-        Do NOT use for vaccine info, schedules, or injection techniques.
+        (Section 9) This tool is for **planning vaccination sessions and managing logistics**. Use it for
+        questions about creating operational maps, estimating vaccine and supply needs, managing stock,
+        and reducing vaccine wastage.
 
         Args:
-            query (str): A question about how to plan or organize vaccination services.
+            query (str): A question about organizing vaccination services or managing stock.
 
         Returns:
             str: Planning and stock guidance from Section 9.
         """
-        print(f"📅 SECTION 9 TOOL CALLED: {query[:50]}...")
+        print(f"📅 PLANNING & LOGISTICS (S9) TOOL CALLED: {query[:50]}...")
         if not section_retrievers.get('nine'):
-            print("❌ Section 9 retriever not available - file may be missing")
-            return "Section 9 retriever not available - file may be missing"
-        try:
-            return section_tool_wrapper(section_retrievers['nine'], section_paths['nine'], query)
-        except Exception as e:
-            print(f"❌ Error accessing section 9: {str(e)}")
-            return f"Error accessing section 9: {str(e)}"
+            return "Section 9 retriever not available"
+        return section_tool_wrapper(section_retrievers['nine'], section_paths['nine'], query)
 
-    def section_ten_tool(query: str) -> str:
+    def communication_tool(query: str) -> str:
         """
-        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.
-
-        Do NOT use for medical, logistical, or procedural topics.
+        (Section 10) The specific tool for **social mobilization and communication**. Use this for
+        questions about communication strategies, addressing **vaccine hesitancy**, managing rumors,
+        and community outreach to promote vaccination.
 
         Args:
             query (str): A question about public engagement or communication for vaccination.
@@ -637,34 +538,29 @@ def create_section_tools(embedding_function, llm):
         Returns:
             str: Public mobilization strategies from Section 10.
         """
-        print(f"📢 SECTION 10 TOOL CALLED: {query[:50]}...")
+        print(f"📢 COMMUNICATION (S10) TOOL CALLED: {query[:50]}...")
         if not section_retrievers.get('ten'):
-            print("❌ Section 10 retriever not available - file may be missing")
-            return "Section 10 retriever not available - file may be missing"
-        try:
-            return section_tool_wrapper(section_retrievers['ten'], section_paths['ten'], query)
-        except Exception as e:
-            print(f"❌ Error accessing section 10: {str(e)}")
-            return f"Error accessing section 10: {str(e)}"
+            return "Section 10 retriever not available"
+        return section_tool_wrapper(section_retrievers['ten'], section_paths['ten'], query)
 
-    # Create FunctionTool objects
+    # Create FunctionTool objects with new, clearer names
     tools = [
-        FunctionTool.from_defaults(name="Guide_vector_tool", fn=guide_retrieval_tool),
-        FunctionTool.from_defaults(name="Immunization_in_Practice_tool", fn=immunization_tool),
+        FunctionTool.from_defaults(name="general_guide_tool", fn=general_guide_tool),
+        FunctionTool.from_defaults(name="who_immunization_tool", fn=who_immunization_tool),
         # Section-specific tools
-        FunctionTool.from_defaults(name="section_one_vector_query_tool", fn=section_one_tool),
-        FunctionTool.from_defaults(name="section_two_vector_query_tool", fn=section_two_tool),
-        FunctionTool.from_defaults(name="section_three_vector_query_tool", fn=section_three_tool),
-        FunctionTool.from_defaults(name="section_four_vector_query_tool", fn=section_four_tool),
-        FunctionTool.from_defaults(name="section_five_vector_query_tool", fn=section_five_tool),
-        FunctionTool.from_defaults(name="section_six_vector_query_tool", fn=section_six_tool),
-        FunctionTool.from_defaults(name="section_seven_vector_query_tool", fn=section_seven_tool),
-        FunctionTool.from_defaults(name="section_eight_vector_query_tool", fn=section_eight_tool),
-        FunctionTool.from_defaults(name="section_nine_vector_query_tool", fn=section_nine_tool),
-        FunctionTool.from_defaults(name="section_ten_vector_query_tool", fn=section_ten_tool),
+        FunctionTool.from_defaults(name="program_overview_tool", fn=program_overview_tool),
+        FunctionTool.from_defaults(name="disease_info_tool", fn=disease_info_tool),
+        FunctionTool.from_defaults(name="vaccine_properties_tool", fn=vaccine_properties_tool),
+        FunctionTool.from_defaults(name="catch_up_vaccination_tool", fn=catch_up_vaccination_tool),
+        FunctionTool.from_defaults(name="special_populations_tool", fn=special_populations_tool),
+        FunctionTool.from_defaults(name="cold_chain_tool", fn=cold_chain_tool),
+        FunctionTool.from_defaults(name="injection_safety_tool", fn=injection_safety_tool),
+        FunctionTool.from_defaults(name="session_management_tool", fn=session_management_tool),
+        FunctionTool.from_defaults(name="planning_and_logistics_tool", fn=planning_and_logistics_tool),
+        FunctionTool.from_defaults(name="communication_tool", fn=communication_tool),
     ]
-    
-    print(f"✅ Created {len(tools)} section tools")
+
+    print(f"✅ Created {len(tools)} tools with improved routing descriptions")
     return tools
 
 def prepare_environment():

rag_pipeline.py

@@ -122,7 +122,7 @@ You provide evidence-based guidance using only information from official vaccine
 Answer the doctor's question accurately and concisely using only the provided information.
 
 ## FALLBACK MODE INSTRUCTIONS
-- You have access to only 2 powerful tools: Guide_vector_tool (Algerian National Vaccination Guide) and Immunization_in_Practice_tool (WHO global guidance).
+- You have access to only 2 powerful tools: general_guide_tool (Algerian National Vaccination Guide) and who_immunization_tool (WHO global guidance).
 - **MANDATORY TOOL USAGE**: Always use the relevant tool(s) to search for information before answering, even if you initially think no information is available.
 - Be direct and efficient - search once with each tool if needed, then provide your answer.
 - Do not overthink or search repeatedly - these tools are comprehensive.
@@ -237,7 +237,7 @@ def create_fallback_tools(all_tools):
     
     for tool in all_tools:
         tool_name = tool.metadata.name if hasattr(tool, 'metadata') else str(tool)
-        if tool_name in ["Guide_vector_tool", "Immunization_in_Practice_tool"]:
+        if tool_name in ["general_guide_tool", "who_immunization_tool"]:
             fallback_tools.append(tool)
             tool_names_found.append(tool_name)