Update crew.py

crew.py

@@ -27,6 +27,7 @@ AUDIO_ANALYSIS_MODEL    = "gemini-2.5-flash-preview-04-17"
 VIDEO_ANALYSIS_MODEL    = "gemini-2.5-flash-preview-04-17"
 YOUTUBE_ANALYSIS_MODEL  = "gemini-2.5-flash-preview-04-17"
 DOCUMENT_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
+ARITHMETIC_MODEL        = "gemini-2.5-flash-preview-04-17"
 CODE_GENERATION_MODEL   = "gemini-2.5-flash-preview-04-17"
 CODE_EXECUTION_MODEL    = "gemini-2.5-flash-preview-04-17"
 
@@ -220,8 +221,8 @@ 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:
+    @tool("Arithmetic Tool")
+    def arithmetic_tool(question: str, a: float, b: float) -> float:
         """Given a question and two numbers, perform the calculation to answer the question.
     
            Args:
@@ -238,7 +239,7 @@ def run_crew(question, file_path):
             client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
             
             response = client.models.generate_content(
-                model=WEB_SEARCH_MODEL,
+                model=ARITHMETIC_MODEL,
                 contents=question,
                 config=types.GenerateContentConfig(
                     tools=[add, subtract, multiply, divide, modulus]
@@ -379,10 +380,10 @@ def run_crew(question, file_path):
         verbose=True
     )
 
-    math_agent = Agent(
-        role="Math Agent",
+    arithmetic_agent = Agent(
+        role="Arithmetic 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.",
+        backstory="As an expert arithmetic assistant, you perform the calculation to answer the question.",
         allow_delegation=False,
         llm=AGENT_MODEL,
         max_iter=2,
@@ -432,7 +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 the modulus**.\n"
+                    "- Arithmetic Agent requires a question and **two numbers to add, subtract, multiply, divide, or get the 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"
@@ -449,7 +450,7 @@ def run_crew(question, file_path):
                 video_analysis_agent, 
                 youtube_analysis_agent, 
                 document_analysis_agent, 
-                #math_agent,
+                arithmetic_agent,
                 code_generation_agent, 
                 code_execution_agent],
         manager_agent=manager_agent,