Update crew.py

crew.py

@@ -14,6 +14,8 @@ from google.genai import types
 from openinference.instrumentation.crewai import CrewAIInstrumentor
 from phoenix.otel import register
 from tools import add, subtract, multiply, divide, modulus
+from tools.ai_tools import AITools
+from tools.arithmetic_tools import ArithmeticTools
 from typing import List
 from utils import read_file_json, read_docx_text, read_pptx_text, is_ext
 
@@ -50,7 +52,7 @@ class GAIACrew():
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=2,
-            tools=[web_search_tool],
+            tools=[AITools.web_search_tool],
             verbose=True
         )
     
@@ -61,7 +63,7 @@ class GAIACrew():
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=2,
-            tools=[image_analysis_tool],
+            tools=[AITools.image_analysis_tool],
             verbose=True
         )
 
@@ -72,7 +74,7 @@ class GAIACrew():
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=2,
-            tools=[audio_analysis_tool],
+            tools=[AITools.audio_analysis_tool],
             verbose=True
         )
 
@@ -83,7 +85,7 @@ class GAIACrew():
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=2,
-            tools=[video_analysis_tool],
+            tools=[AITools.video_analysis_tool],
             verbose=True
         )
 
@@ -94,7 +96,7 @@ class GAIACrew():
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=2,
-            tools=[youtube_analysis_tool],
+            tools=[AITools.youtube_analysis_tool],
             verbose=True
         )
 
@@ -105,18 +107,18 @@ class GAIACrew():
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=2,
-            tools=[document_analysis_tool],
+            tools=[AITools.document_analysis_tool],
             verbose=True
         )
 
     @agent
-    def document_analysis_agent(self) -> Agent:
+    def arithmetic_agent(self) -> Agent:
         return Agent(
             config=self.agents_config["document_analysis_agent"],
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=2,
-            tools=[arithmetic_tool],
+            tools=[ArithmeticTools.add, ArithmeticTools.subtract, ArithmeticTools.multiply, ArithmeticTools.divide, ArithmeticTools.modulus],
             verbose=True
         )
 
@@ -127,7 +129,7 @@ class GAIACrew():
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=3,
-            tools=[code_execution_tool],
+            tools=[AITools.code_execution_tool],
             verbose=True
         )
 
@@ -138,7 +140,7 @@ class GAIACrew():
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=3,
-            tools=[code_generation_tool],
+            tools=[AITools.code_generation_tool],
             verbose=True
         )