File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree 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+ import orchestrator
77from orchestrator import (_HARMFUL_PATTERNS_COMPILED ,
88 _SYSTEM_PROMPT_PATTERNS_COMPILED ,
99 PLANNING_INSTRUCTIONS , RAI_HARMFUL_CONTENT_RESPONSE ,
@@ -840,7 +840,7 @@ def test_get_orchestrator_singleton():
840840 mock_builder .return_value = mock_builder_instance
841841
842842 # Reset the singleton
843- orch_module ._orchestrator = None
843+ orchestrator ._orchestrator = None
844844
845845 instance1 = get_orchestrator ()
846846 instance2 = get_orchestrator ()
@@ -1517,7 +1517,7 @@ async def test_get_chat_client_foundry_mode():
15171517def test_foundry_not_available ():
15181518 """Test when Foundry SDK is not available."""
15191519 # Check that FOUNDRY_AVAILABLE is defined
1520- assert hasattr (orch_module , 'FOUNDRY_AVAILABLE' )
1520+ assert hasattr (orchestrator , 'FOUNDRY_AVAILABLE' )
15211521
15221522# Tests for workflow event handling (lines 736-799, 841-895)
15231523# Note: These are integration-level tests that verify the workflow event
You can’t perform that action at this time.
0 commit comments