Update agents/crew.py

agents/crew.py

@@ -5,10 +5,10 @@
 
 import os
 from agents.llms import (
-    MANAGER_MODEL,
-    AGENT_MODEL,
-    FINAL_ANSWER_MODEL,
-    FINAL_ANSWER_PROMPT
+    LLM_MANAGER,
+    LLM_AGENT,
+    LLM_FINAL_ANSWER,
+    PROMPT_FINAL_ANSWER
 )
 from agents.tools.ai_tools import AITools
 from agents.tools.arithmetic_tools import ArithmeticTools
@@ -29,7 +29,7 @@ class GAIACrew():
         return Agent(
             config=self.agents_config["web_search_agent"],
             allow_delegation=False,
-            llm=AGENT_MODEL,
+            llm=LLM_AGENT,
             max_iter=2,
             tools=[AITools.web_search_tool],
             verbose=True
@@ -40,7 +40,7 @@ class GAIACrew():
         return Agent(
             config=self.agents_config["web_browser_agent"],
             allow_delegation=False,
-            llm=AGENT_MODEL,
+            llm=LLM_AGENT,
             max_iter=3,
             tools=[AITools.web_browser_tool],
             verbose=True
@@ -51,7 +51,7 @@ class GAIACrew():
         return Agent(
             config=self.agents_config["chess_analysis_agent"],
             allow_delegation=False,
-            llm=AGENT_MODEL,
+            llm=LLM_AGENT,
             max_iter=2,
             tools=[AITools.chess_analysis_tool],
             verbose=True
@@ -62,7 +62,7 @@ class GAIACrew():
         return Agent(
             config=self.agents_config["image_analysis_agent"],
             allow_delegation=False,
-            llm=AGENT_MODEL,
+            llm=LLM_AGENT,
             max_iter=2,
             tools=[AITools.image_analysis_tool],
             verbose=True
@@ -73,7 +73,7 @@ class GAIACrew():
         return Agent(
             config=self.agents_config["audio_analysis_agent"],
             allow_delegation=False,
-            llm=AGENT_MODEL,
+            llm=LLM_AGENT,
             max_iter=2,
             tools=[AITools.audio_analysis_tool],
             verbose=True
@@ -84,7 +84,7 @@ class GAIACrew():
         return Agent(
             config=self.agents_config["video_analysis_agent"],
             allow_delegation=False,
-            llm=AGENT_MODEL,
+            llm=LLM_AGENT,
             max_iter=2,
             tools=[AITools.video_analysis_tool],
             verbose=True
@@ -95,7 +95,7 @@ class GAIACrew():
         return Agent(
             config=self.agents_config["youtube_analysis_agent"],
             allow_delegation=False,
-            llm=AGENT_MODEL,
+            llm=LLM_AGENT,
             max_iter=2,
             tools=[AITools.youtube_analysis_tool],
             verbose=True
@@ -106,7 +106,7 @@ class GAIACrew():
         return Agent(
             config=self.agents_config["document_analysis_agent"],
             allow_delegation=False,
-            llm=AGENT_MODEL,
+            llm=LLM_AGENT,
             max_iter=2,
             tools=[AITools.document_analysis_tool],
             verbose=True
@@ -117,7 +117,7 @@ class GAIACrew():
         return Agent(
             config=self.agents_config["document_analysis_agent"],
             allow_delegation=False,
-            llm=AGENT_MODEL,
+            llm=LLM_AGENT,
             max_iter=2,
             tools=[ArithmeticTools.add, ArithmeticTools.subtract, ArithmeticTools.multiply, ArithmeticTools.divide, ArithmeticTools.modulus],
             verbose=True
@@ -128,7 +128,7 @@ class GAIACrew():
         return Agent(
             config=self.agents_config["code_generation_agent"],
             allow_delegation=False,
-            llm=AGENT_MODEL,
+            llm=LLM_AGENT,
             max_iter=3,
             tools=[AITools.code_generation_tool],
             verbose=True
@@ -139,7 +139,7 @@ class GAIACrew():
         return Agent(
             config=self.agents_config["code_execution_agent"],
             allow_delegation=False,
-            llm=AGENT_MODEL,
+            llm=LLM_AGENT,
             max_iter=3,
             tools=[AITools.code_execution_tool],
             verbose=True
@@ -150,7 +150,7 @@ class GAIACrew():
         return Agent(
             config=self.agents_config["manager_agent"],
             allow_delegation=True,
-            llm=MANAGER_MODEL,
+            llm=LLM_MANAGER,
             max_iter=5,
             verbose=True
         )
@@ -190,12 +190,12 @@ def run_crew(question, file_path):
     return final_answer
 
 def get_final_answer(question, answer):
-    prompt = FINAL_ANSWER_PROMPT.format(question=question, answer=answer)
+    prompt = PROMPT_FINAL_ANSWER.format(question=question, answer=answer)
 
     client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
 
     response = client.models.generate_content(
-        model=FINAL_ANSWER_MODEL, 
+        model=LLM_FINAL_ANSWER, 
         contents=[prompt]
     )