Update agents/crew.py

agents/crew.py

@@ -4,41 +4,24 @@
 # https://arize.com/docs/phoenix/integrations/python/crewai/crewai-tracing
 
 import logging, os
-import sys
-from contextlib import contextmanager
 
-@contextmanager
-def suppress_stdout_stderr():
-    with open(os.devnull, 'w') as fnull:
-        old_stdout = sys.stdout
-        old_stderr = sys.stderr
-        sys.stdout = fnull
-        sys.stderr = fnull
-        try:
-            yield
-        finally:
-            sys.stdout = old_stdout
-            sys.stderr = old_stderr
-
-os.environ["CREWAI_TRACING_ENABLED"] = "0"
-
-with suppress_stdout_stderr():
-    from agents.models.llms import (
-        LLM_CREW_MANAGER,
-        LLM_CREW_PLANNING,
-        LLM_MANAGER,
-        LLM_AGENT
-    )
-    from agents.tools.ai_tools import AITools
-    from agents.tools.deterministic_tools import DeterministicTools
-    from agents.tools.mcp_tools import MCPTools
-    from crewai import Agent, Crew, Task, Process
-    from crewai.agents.agent_builder.base_agent import BaseAgent
-    from crewai.project import CrewBase, agent, crew, task
-    from google import genai
-    from google.genai import types
-    from phoenix.otel import register
+os.environ["CREWAI_TRACING_ENABLED"] = "0" # Arize Phoenix
 
+from agents.models.llms import (
+    LLM_CREW_MANAGER,
+    LLM_CREW_PLANNING,
+    LLM_MANAGER,
+    LLM_AGENT
+)
+from agents.tools.ai_tools import AITools
+from agents.tools.deterministic_tools import DeterministicTools
+from agents.tools.mcp_tools import MCPTools
+from crewai import Agent, Crew, Task, Process
+from crewai.agents.agent_builder.base_agent import BaseAgent
+from crewai.project import CrewBase, agent, crew, task
+from google import genai
+from google.genai import types
+from phoenix.otel import register
 from typing import List
 from utils.utils import read_file_json, is_ext