Update agents/crew.py

agents/crew.py

@@ -3,7 +3,7 @@
 # https://docs.crewai.com/introduction
 # https://arize.com/docs/phoenix/integrations/python/crewai/crewai-tracing
 
-import os
+import logging, os
 from agents.models.llms import (
     LLM_CREW_MANAGER,
     LLM_CREW_PLANNING,
@@ -54,6 +54,8 @@ tracer_provider = register(
     set_global_tracer_provider=True
 )
 
+logging.getLogger("openinference").setLevel(logging.CRITICAL)
+
 @CrewBase
 class GAIACrew():
     agents: List[BaseAgent]
@@ -229,12 +231,13 @@ class GAIACrew():
         return Crew(
             agents=self.agents,
             tasks=self.tasks,
-            process=Process.sequential, # TODO
-            manager_llm=LLM_CREW_MANAGER, # TODO
+            process=Process.sequential,
+            manager_llm=LLM_CREW_MANAGER,
             planning_llm=LLM_CREW_PLANNING,
             planning=PLANNING_CREW,
             memory=MEMORY_CREW,
-            verbose=VERBOSE_CREW
+            verbose=VERBOSE_CREW,
+            tracing=False # Arize Phoenix
         )
 
 def run_crew(question, file_path):
@@ -250,7 +253,10 @@ def run_crew(question, file_path):
     print(f"🤖 Initial question: {question}")
     print(f"🤖 Final question: {final_question}")
 
-    answer = GAIACrew().crew().kickoff(inputs={"question": final_question})   
+    print(f"🤖 Crew started")
+    answer = GAIACrew().crew().kickoff(inputs={"question": final_question})
+    print(f"🤖 Crew completed")
+
     final_answer = AITools.get_final_answer(question, str(answer))
 
     print(f"🤖 Initial answer: {answer}")