Update agents/crew.py

agents/crew.py

@@ -12,7 +12,7 @@ from agents.models.llms import (
     THINKING_LEVEL_FINAL_ANSWER
 )
 from agents.models.prompts import PROMPT_FINAL_ANSWER
-from agents.tools.ai_tools import AITools
+from agents.tools.ai_tools import AITools, get_final_answer
 from agents.tools.arithmetic_tools import ArithmeticTools
 from agents.tools.chess_tools import ChessTools
 from crewai import Agent, Crew, Task
@@ -209,28 +209,11 @@ def run_crew(question, file_path):
             final_question = f"{question} File path: {file_path}."
 
     answer = GAIACrew().crew().kickoff(inputs={"question": final_question})   
-    #final_answer = get_final_answer(question, str(answer))
+    final_answer = get_final_answer(question, str(answer))
 
     print(f"🤖 Initial question: {question}")
     print(f"🤖 Final question: {final_question}")
-    #print(f"🤖 Initial answer: {answer}")
+    print(f"🤖 Initial answer: {answer}")
     print(f"🤖 Final answer: {final_answer}")
     
-    return final_answer
-
-def get_final_answer(question, answer):
-    client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
-
-    prompt = PROMPT_FINAL_ANSWER.format(question=question, answer=answer)
-
-    response = client.models.generate_content(
-        model=LLM_FINAL_ANSWER, 
-        contents=[prompt],
-        config=types.GenerateContentConfig(
-            thinking_config=types.ThinkingConfig(
-                thinking_level=THINKING_LEVEL_FINAL_ANSWER
-            )
-        )
-    )
-    
-    return response.text
+    return final_answer