Update agents/crew.py

agents/crew.py

@@ -23,18 +23,28 @@ from phoenix.otel import register
 from typing import List
 from utils.utils import read_file_json, is_ext
 
-VERBOSE_CREW    = True
-VERBOSE_MANAGER = False
-VERBOSE_AGENT   = False
+# Configuration
+
+MEMORY_CREW                  = True
+PLANNING_CREW                = True
+VERBOSE_CREW                 = True
+
+MAX_ITER_MANAGER               = 5
+MAX_REASONING_ATTEMPTS_MANAGER = 5
+REASONING_MANAGER              = True
+VERBOSE_MANAGER                = True
+
+MAX_ITER_AGENT                 = 3
+MAX_REASONING_ATTEMPTS_AGENT   = 3
+REASONING_AGENT                = True
+VERBOSE_AGENT                  = True
 
 # Observability
 
 os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
 os.environ["PHOENIX_API_KEY"] = os.getenv("PHOENIX_API_KEY")
-PHOENIX_API_KEY = os.environ["PHOENIX_API_KEY"]
-
 os.environ["PHOENIX_COLLECTOR_ENDPOINT"] = "https://app.phoenix.arize.com/s/bstraehle/v1/traces"
-os.environ["PHOENIX_CLIENT_HEADERS"] = f"api-key={PHOENIX_API_KEY}"
+os.environ["PHOENIX_CLIENT_HEADERS"] = f"api-key={os.environ["PHOENIX_API_KEY"]}"
 
 tracer_provider = register(
     project_name="gaia",
@@ -58,8 +68,10 @@ class GAIACrew():
             config=self.agents_config["web_search_agent"],
             allow_delegation=False,
             llm=LLM_AGENT,
-            max_iter=2,
+            max_iter=MAX_ITER_AGENT,
             tools=[AITools.web_search_tool],
+            reasoning=REASONING_AGENT,
+            max_reasoning_attempts=MAX_REASONING_ATTEMPTS_AGENT,
             verbose=VERBOSE_AGENT
         )
 
@@ -69,8 +81,10 @@ class GAIACrew():
             config=self.agents_config["web_browser_agent"],
             allow_delegation=False,
             llm=LLM_AGENT,
-            max_iter=5,
+            max_iter=MAX_ITER_AGENT,
             tools=[AITools.web_browser_tool],
+            reasoning=REASONING_AGENT,
+            max_reasoning_attempts=MAX_REASONING_ATTEMPTS_AGENT,
             verbose=VERBOSE_AGENT
         )
 
@@ -80,10 +94,10 @@ class GAIACrew():
             config=self.agents_config["chess_analysis_agent"],
             allow_delegation=False,
             llm=LLM_AGENT,
-            max_iter=2,
+            max_iter=MAX_ITER_AGENT,
             tools=[AITools.img_to_fen_tool, ChessTools.get_best_move_tool],
-            reasoning=True, ###
-            max_reasoning_attempts=3, ###
+            reasoning=REASONING_AGENT,
+            max_reasoning_attempts=MAX_REASONING_ATTEMPTS_AGENT,
             verbose=VERBOSE_AGENT
         )
 
@@ -93,8 +107,10 @@ class GAIACrew():
             config=self.agents_config["image_analysis_agent"],
             allow_delegation=False,
             llm=LLM_AGENT,
-            max_iter=2,
+            max_iter=MAX_ITER_AGENT,
             tools=[AITools.image_analysis_tool],
+            reasoning=REASONING_AGENT,
+            max_reasoning_attempts=MAX_REASONING_ATTEMPTS_AGENT,
             verbose=VERBOSE_AGENT
         )
 
@@ -104,8 +120,10 @@ class GAIACrew():
             config=self.agents_config["audio_analysis_agent"],
             allow_delegation=False,
             llm=LLM_AGENT,
-            max_iter=2,
+            max_iter=MAX_ITER_AGENT,
             tools=[AITools.audio_analysis_tool],
+            reasoning=REASONING_AGENT,
+            max_reasoning_attempts=MAX_REASONING_ATTEMPTS_AGENT,
             verbose=VERBOSE_AGENT
         )
 
@@ -115,8 +133,10 @@ class GAIACrew():
             config=self.agents_config["video_analysis_agent"],
             allow_delegation=False,
             llm=LLM_AGENT,
-            max_iter=2,
+            max_iter=MAX_ITER_AGENT,
             tools=[AITools.video_analysis_tool],
+            reasoning=REASONING_AGENT,
+            max_reasoning_attempts=MAX_REASONING_ATTEMPTS_AGENT,
             verbose=VERBOSE_AGENT
         )
 
@@ -126,8 +146,10 @@ class GAIACrew():
             config=self.agents_config["youtube_analysis_agent"],
             allow_delegation=False,
             llm=LLM_AGENT,
-            max_iter=2,
+            max_iter=MAX_ITER_AGENT,
             tools=[AITools.youtube_analysis_tool],
+            reasoning=REASONING_AGENT,
+            max_reasoning_attempts=MAX_REASONING_ATTEMPTS_AGENT,
             verbose=VERBOSE_AGENT
         )
 
@@ -137,8 +159,10 @@ class GAIACrew():
             config=self.agents_config["document_analysis_agent"],
             allow_delegation=False,
             llm=LLM_AGENT,
-            max_iter=2,
+            max_iter=MAX_ITER_AGENT,
             tools=[AITools.document_analysis_tool],
+            reasoning=REASONING_AGENT,
+            max_reasoning_attempts=MAX_REASONING_ATTEMPTS_AGENT,
             verbose=VERBOSE_AGENT
         )
 
@@ -148,12 +172,14 @@ class GAIACrew():
             config=self.agents_config["document_analysis_agent"],
             allow_delegation=False,
             llm=LLM_AGENT,
-            max_iter=2,
+            max_iter=MAX_ITER_AGENT,
             tools=[ArithmeticTools.add_tool,
                    ArithmeticTools.subtract_tool,
                    ArithmeticTools.multiply_tool,
                    ArithmeticTools.divide_tool,
                    ArithmeticTools.modulus_tool],
+            reasoning=REASONING_AGENT,
+            max_reasoning_attempts=MAX_REASONING_ATTEMPTS_AGENT,
             verbose=VERBOSE_AGENT
         )
 
@@ -163,8 +189,10 @@ class GAIACrew():
             config=self.agents_config["code_generation_agent"],
             allow_delegation=False,
             llm=LLM_AGENT,
-            max_iter=3,
+            max_iter=MAX_ITER_AGENT,
             tools=[AITools.code_generation_tool],
+            reasoning=REASONING_AGENT,
+            max_reasoning_attempts=MAX_REASONING_ATTEMPTS_AGENT,
             verbose=VERBOSE_AGENT
         )
 
@@ -174,8 +202,10 @@ class GAIACrew():
             config=self.agents_config["code_execution_agent"],
             allow_delegation=False,
             llm=LLM_AGENT,
-            max_iter=3,
+            max_iter=MAX_ITER_AGENT,
             tools=[AITools.code_execution_tool],
+            reasoning=REASONING_AGENT,
+            max_reasoning_attempts=MAX_REASONING_ATTEMPTS_AGENT,
             verbose=VERBOSE_AGENT
         )
    
@@ -185,8 +215,10 @@ class GAIACrew():
             config=self.agents_config["manager_agent"],
             allow_delegation=True,
             llm=LLM_MANAGER,
-            max_iter=5,
+            max_iter=MAX_ITER_MANAGER,
             tools=[AITools.final_answer_tool],
+            reasoning=REASONING_MANAGER,
+            max_reasoning_attempts=MAX_REASONING_ATTEMPTS_MANAGER,
             verbose=VERBOSE_MANAGER
         )
 
@@ -201,8 +233,8 @@ class GAIACrew():
         return Crew(
             agents=self.agents,
             tasks=self.tasks,
-            memory=True, ###
-            planning=True, ###
+            memory=MEMORY_CREW,
+            planning=PLANNING_CREW,
             verbose=VERBOSE_CREW
         )