Spaces:
Sleeping
Sleeping
Zeggai Abdellah
commited on
Commit
·
d5f8cfd
1
Parent(s):
e79b557
first create just the first tool for test
Browse files- 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 |
-
|
| 237 |
-
|
| 238 |
-
|
| 239 |
-
|
| 240 |
-
|
| 241 |
-
|
| 242 |
-
|
| 243 |
|
| 244 |
-
|
| 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 |
|