Revert experimental change

ag2ai · Feb 24, 2025 · 13886ce · 13886ce
1 parent 961ed2e
commit 13886ce
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/test/agentchat/realtime_agent/test_e2e.py b/test/agentchat/realtime_agent/test_e2e.py
@@ -14,15 +14,15 @@
 from pytest import FixtureRequest
 
 from autogen.agentchat.realtime.experimental import RealtimeAgent, RealtimeObserver, WebSocketAudioAdapter
+from autogen.import_utils import skip_on_missing_imports
 
-# from autogen.import_utils import skip_on_missing_imports
 from ...conftest import Credentials
 from .realtime_test_utils import text_to_speech, trace
 
 logger = getLogger(__name__)
 
 
-# @skip_on_missing_imports("websockets", "websockets")
+@skip_on_missing_imports("websockets", "websockets")
 class TestE2E:
     async def _test_e2e(self, credentials_llm: Credentials, credentials_openai: Credentials) -> None:
         """End-to-end test for the RealtimeAgent.

diff --git a/test/agentchat/realtime_agent/test_swarm_start.py b/test/agentchat/realtime_agent/test_swarm_start.py
@@ -16,8 +16,7 @@
 from autogen import ConversableAgent
 from autogen.agentchat.realtime.experimental import RealtimeAgent, RealtimeObserver, WebSocketAudioAdapter
 from autogen.agentchat.realtime.experimental.realtime_swarm import register_swarm
-
-# from autogen.import_utils import skip_on_missing_imports
+from autogen.import_utils import skip_on_missing_imports
 from autogen.tools.dependency_injection import Field as AG2Field
 
 from ...conftest import Credentials
@@ -26,7 +25,7 @@
 logger = getLogger(__name__)
 
 
-# @skip_on_missing_imports("websockets", "websockets")
+@skip_on_missing_imports("websockets", "websockets")
 class TestSwarmE2E:
     async def _test_e2e(self, credentials_llm_realtime: Credentials, credentials_gpt_4o_mini: Credentials) -> None:
         """End-to-end test for the RealtimeAgent.