Created multiple lists to store the tools available to the agents

crew.py

@@ -34,6 +34,41 @@ tracer_provider = register(
     project_name="gaia"
 )
 
+## Tools
+
+DOCUMENT_TOOLS = [
+    AITools.document_analysis_tool,
+    AITools.summarize_tool,
+    AITools.translate_tool
+]
+
+MEDIA_TOOLS = [
+    AITools.image_analysis_tool,
+    AITools.audio_analysis_tool,
+    AITools.video_analysis_tool,
+    AITools.youtube_analysis_tool
+]
+
+WEB_TOOLS = [
+    AITools.web_search_tool,
+    AITools.web_browser_tool
+]
+
+ARITHMETIC_TOOLS = [
+    ArithmeticTools.add,
+    ArithmeticTools.subtract,
+    ArithmeticTools.multiply,
+    ArithmeticTools.divide,
+    ArithmeticTools.modulus
+]
+
+CODE_TOOLS = [
+    AITools.code_generation_tool,
+    AITools.code_execution_tool
+]
+
+
+
 #CrewAIInstrumentor().instrument(tracer_provider=tracer_provider)
 
 @CrewBase
@@ -48,7 +83,7 @@ class GAIACrew():
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=2,
-            tools=[AITools.web_search_tool],
+            tools=WEB_TOOLS,
             verbose=True
         )
 
@@ -59,7 +94,7 @@ class GAIACrew():
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=3,
-            tools=[AITools.web_browser_tool],
+            tools=WEB_TOOLS,
             verbose=True
         )
     
@@ -70,7 +105,7 @@ class GAIACrew():
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=2,
-            tools=[AITools.image_analysis_tool],
+            tools=MEDIA_TOOLS,
             verbose=True
         )
 
@@ -92,7 +127,7 @@ class GAIACrew():
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=2,
-            tools=[AITools.video_analysis_tool],
+            tools=MEDIA_TOOLS,
             verbose=True
         )
 
@@ -103,7 +138,7 @@ class GAIACrew():
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=2,
-            tools=[AITools.youtube_analysis_tool],
+            tools=MEDIA_TOOLS,
             verbose=True
         )
 
@@ -114,7 +149,7 @@ class GAIACrew():
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=2,
-            tools=[AITools.document_analysis_tool],
+            tools=DOCUMENT_TOOLS,
             verbose=True
         )
 
@@ -125,7 +160,7 @@ class GAIACrew():
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=2,
-            tools=[ArithmeticTools.add, ArithmeticTools.subtract, ArithmeticTools.multiply, ArithmeticTools.divide, ArithmeticTools.modulus],
+            tools=ARITHMETIC_TOOLS,
             verbose=True
         )
 
@@ -136,7 +171,7 @@ class GAIACrew():
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=3,
-            tools=[AITools.code_generation_tool],
+            tools=CODE_TOOLS,
             verbose=True
         )
 
@@ -147,7 +182,7 @@ class GAIACrew():
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=3,
-            tools=[AITools.code_execution_tool],
+            tools=CODE_TOOLS,
             verbose=True
         )
     
@@ -158,7 +193,7 @@ class GAIACrew():
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=2,
-            tools=[AITools.translate_tool],
+            tools=DOCUMENT_TOOLS,
             verbose=True
         )
 
@@ -169,7 +204,7 @@ class GAIACrew():
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=2,
-            tools=[AITools.summarize_tool],
+            tools=DOCUMENT_TOOLS,
             verbose=True
         )