File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed
Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change 33from unittest .mock import AsyncMock , MagicMock , patch
44
55import pytest
6- import orchestrator as orch_module
6+
7+ import orchestrator
78from orchestrator import (_HARMFUL_PATTERNS_COMPILED ,
89 _SYSTEM_PROMPT_PATTERNS_COMPILED ,
910 PLANNING_INSTRUCTIONS , RAI_HARMFUL_CONTENT_RESPONSE ,
@@ -840,7 +841,7 @@ def test_get_orchestrator_singleton():
840841 mock_builder .return_value = mock_builder_instance
841842
842843 # Reset the singleton
843- orch_module ._orchestrator = None
844+ orchestrator ._orchestrator = None
844845
845846 instance1 = get_orchestrator ()
846847 instance2 = get_orchestrator ()
@@ -1517,7 +1518,7 @@ async def test_get_chat_client_foundry_mode():
15171518def test_foundry_not_available ():
15181519 """Test when Foundry SDK is not available."""
15191520 # Check that FOUNDRY_AVAILABLE is defined
1520- assert hasattr (orch_module , 'FOUNDRY_AVAILABLE' )
1521+ assert hasattr (orchestrator , 'FOUNDRY_AVAILABLE' )
15211522
15221523# Tests for workflow event handling (lines 736-799, 841-895)
15231524# Note: These are integration-level tests that verify the workflow event
You can’t perform that action at this time.
0 commit comments