Update crew.py

crew.py

@@ -15,18 +15,18 @@ from util import read_file, get_final_answer
 
 ## LLMs
 
-MANAGER_MODEL           = "gpt-4.1"
-AGENT_MODEL             = "gpt-4.1"
+MANAGER_MODEL           = "gpt-4.1-mini"
+AGENT_MODEL             = "gpt-4.1-mini"
 
 FINAL_ANSWER_MODEL      = "gemini-2.5-flash-preview-04-17"
 
 WEB_SEARCH_MODEL        = "gemini-2.5-flash-preview-04-17"
-IMAGE_ANALYSIS_MODEL    = "gemini-2.5-pro-preview-03-25"
+IMAGE_ANALYSIS_MODEL    = "gemini-2.5-flash-preview-04-17"
 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"
-CODE_GENERATION_MODEL   = "gemini-2.5-pro-preview-03-25"
-CODE_EXECUTION_MODEL    = "gemini-2.5-pro-preview-03-25"
+CODE_GENERATION_MODEL   = "gemini-2.5-flash-preview-04-17"
+CODE_EXECUTION_MODEL    = "gemini-2.5-flash-preview-04-17"
 
 # LLM evaluation
 
@@ -181,11 +181,11 @@ def run_crew(question, file_path):
 
     @tool("Code Generation Tool")
     def code_generation_tool(question: str) -> str:
-        """Given a question, generate code to answer the question.
+        """Given a question with data, generate code to answer the question.
 
            Args:
                question (str): Question to answer
-                
+
            Returns:
                str: Answer to the question
                
@@ -371,8 +371,10 @@ def run_crew(question, file_path):
     initial_answer = crew.kickoff(inputs={"question": question})
     final_answer = get_final_answer(FINAL_ANSWER_MODEL, question, str(initial_answer))
 
+    print("###")
     print(f"Question: {question}")
     print(f"Initial answer: {initial_answer}")
     print(f"Final answer: {final_answer}")
+    print("###")
     
     return final_answer