Zeggai Abdellah commited on
Commit
d5f8cfd
·
1 Parent(s): e79b557

first create just the first tool for test

Browse files
Files changed (1) hide show
  1. rag_system.py +19 -19
rag_system.py CHANGED
@@ -231,29 +231,29 @@ class AgenticRAGSystem:
231
  """Create all the function tools"""
232
  tools = []
233
 
234
- # Main guide tool
235
- if self.guide_retriever:
236
- def guide_tool(query: str) -> str:
237
- """General-purpose retrieval tool for the Algerian National Vaccination Guide (2023)"""
238
- return self.section_tool_wrapper(
239
- self.guide_retriever,
240
- os.path.join(self.config.BASE_PATH, "Guide-pratique-de-mise-en-oeuvre-du-calendrier-national-de-vaccination-2023.json"),
241
- query
242
- )
243
 
244
- tools.append(FunctionTool.from_defaults(name="Guide_vector_tool", fn=guide_tool))
245
 
246
  # Section tools
247
  section_descriptions = {
248
- "section_1": "Programme Élargi de Vaccination - General national immunization program in Algeria",
249
- "section_2": "Maladies Ciblées - Diseases targeted by the national vaccination calendar",
250
- "section_3": "Vaccins du Calendrier - Vaccines themselves: types, administration methods, compositions",
251
- "section_4": "Rattrapage Vaccinal - Catch-up vaccination procedures and schedules",
252
- "section_5": "Populations Particulières - Vaccination of special populations (premature, immunosuppressed, etc.)",
253
- "section_6": "Chaîne du Froid - Vaccine cold chain logistics and storage",
254
- "section_7": "Sécurité des Injections - Safe injection practices",
255
- "section_8": "Séance de Vaccination & Vaccinovigilance - Vaccination sessions and adverse event monitoring",
256
- "section_9": "Planification des Séances - Planning of vaccination sessions",
257
  "section_10": "Mobilisation Sociale - Community mobilization and vaccine hesitancy"
258
  }
259
 
 
231
  """Create all the function tools"""
232
  tools = []
233
 
234
+ # # Main guide tool
235
+ # if self.guide_retriever:
236
+ # def guide_tool(query: str) -> str:
237
+ # """General-purpose retrieval tool for the Algerian National Vaccination Guide (2023)"""
238
+ # return self.section_tool_wrapper(
239
+ # self.guide_retriever,
240
+ # os.path.join(self.config.BASE_PATH, "Guide-pratique-de-mise-en-oeuvre-du-calendrier-national-de-vaccination-2023.json"),
241
+ # query
242
+ # )
243
 
244
+ # tools.append(FunctionTool.from_defaults(name="Guide_vector_tool", fn=guide_tool))
245
 
246
  # Section tools
247
  section_descriptions = {
248
+ # "section_1": "Programme Élargi de Vaccination - General national immunization program in Algeria",
249
+ # "section_2": "Maladies Ciblées - Diseases targeted by the national vaccination calendar",
250
+ # "section_3": "Vaccins du Calendrier - Vaccines themselves: types, administration methods, compositions",
251
+ # "section_4": "Rattrapage Vaccinal - Catch-up vaccination procedures and schedules",
252
+ # "section_5": "Populations Particulières - Vaccination of special populations (premature, immunosuppressed, etc.)",
253
+ # "section_6": "Chaîne du Froid - Vaccine cold chain logistics and storage",
254
+ # "section_7": "Sécurité des Injections - Safe injection practices",
255
+ # "section_8": "Séance de Vaccination & Vaccinovigilance - Vaccination sessions and adverse event monitoring",
256
+ # "section_9": "Planification des Séances - Planning of vaccination sessions",
257
  "section_10": "Mobilisation Sociale - Community mobilization and vaccine hesitancy"
258
  }
259