Update crew.py

crew.py

@@ -11,6 +11,7 @@ from google import genai
 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 utils import read_file_json, read_docx_text, read_pptx_text, is_ext
 
 ## LLMs
@@ -219,6 +220,36 @@ def run_crew(question, file_path):
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
 
+    @tool("Math Tool")
+    def math_tool(question: str, a: float, b: float) -> float:
+        """Given a question and two numbers, perform the calculation to answer the question.
+    
+           Args:
+               question (str): Question to answer
+               a (float): Float a
+               b (float): Float b
+                
+           Returns:
+               float: Result float
+                
+           Raises:
+               RuntimeError: If processing fails"""
+        try:
+            client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
+            
+            response = client.models.generate_content(
+                model=WEB_SEARCH_MODEL,
+                contents=question,
+                config=types.GenerateContentConfig(
+                    tools=[add, subtract, multiply, divide, modulus]
+                )
+            )
+
+            return response.text
+        except Exception as e:
+            raise RuntimeError(f"Processing failed: {str(e)}")
+    ###
+    
     @tool("Code Generation Tool")
     def code_generation_tool(question: str, json_data: str) -> str:
         """Given a question and json data, generate and execute code to answer the question.
@@ -348,6 +379,17 @@ def run_crew(question, file_path):
         tools=[document_analysis_tool],
         verbose=True
     )
+
+    math_agent = Agent(
+        role="Math Agent",
+        goal="Given a question and two numbers, perform the calculation and answer the question: {question}",
+        backstory="As an expert math assistant, you perform the calculation to answer the question.",
+        allow_delegation=False,
+        llm=AGENT_MODEL,
+        max_iter=2,
+        tools=[math_tool],
+        verbose=True
+    )
     
     code_generation_agent = Agent(
         role="Code Generation Agent",
@@ -391,6 +433,7 @@ def run_crew(question, file_path):
                     "- Audio Analysis Agent requires a question and **.wav, .mp3, .aiff, .aac, .ogg, or .flac audio file**.\n"
                     "- Video Analysis Agent requires a question and **.mp4, .mpeg, .mov, .avi, .x-flv, .mpg, .webm, .wmv, or .3gpp video file**.\n"
                     "- Document Analysis Agent requires a question and **.docx, .pptx, .pdf, .txt, .html, css, .js, .md, .xml, or .rtf document file**.\n"
+                    "- Math Agent requires a question and **two numbers to add, subtract, multiply, divide, or get modulus**.\n"
                     "- YouTube Analysis Agent requires a question and **YouTube URL**.\n"
                     "- Code Generation Agent requires a question and **JSON data**.\n"
                     "- Code Execution Agent requires a question and **Python file**.\n"
@@ -407,6 +450,7 @@ def run_crew(question, file_path):
                 video_analysis_agent, 
                 youtube_analysis_agent, 
                 document_analysis_agent, 
+                #math_agent,
                 code_generation_agent, 
                 code_execution_agent],
         manager_agent=manager_agent,