Update agents/crew.py

agents/crew.py

@@ -4,24 +4,41 @@
 # https://arize.com/docs/phoenix/integrations/python/crewai/crewai-tracing
 
 import logging, os
+import sys
+from contextlib import contextmanager
 
-os.environ["CREWAI_TRACING_ENABLED"] = "0" # Arize Phoenix
+@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
 
-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