refactor: update exception handling in client creation tests to use ImportError

Author: Dhanushree-Microsoft

src/processor/src/tests/unit/libs/agent_framework/test_agent_framework_helper.py

@@ -112,7 +112,7 @@ def test_default_token_provider_when_no_credential(self):
     def test_azure_openai_chat_completion(self):
         fake_module = types.ModuleType("agent_framework.azure")
         with patch.dict(sys.modules, {"agent_framework.azure": fake_module}):
-            with pytest.raises(NotImplementedError, match="AzureOpenAIChatClient was removed"):
+            with pytest.raises(ImportError):
                 AgentFrameworkHelper.create_client(
                     ClientType.AzureOpenAIChatCompletion,
                     endpoint="https://x",
@@ -123,7 +123,7 @@ def test_azure_openai_chat_completion(self):
     def test_azure_openai_assistant(self):
         fake_module = types.ModuleType("agent_framework.azure")
         with patch.dict(sys.modules, {"agent_framework.azure": fake_module}):
-            with pytest.raises(NotImplementedError, match="AzureOpenAIAssistantsClient was removed"):
+            with pytest.raises(ImportError):
                 AgentFrameworkHelper.create_client(
                     ClientType.AzureOpenAIAssistant,
                     endpoint="https://x",
@@ -134,7 +134,7 @@ def test_azure_openai_assistant(self):
     def test_azure_openai_response(self):
         fake_module = types.ModuleType("agent_framework.azure")
         with patch.dict(sys.modules, {"agent_framework.azure": fake_module}):
-            with pytest.raises(NotImplementedError, match="AzureOpenAIResponsesClient was removed"):
+            with pytest.raises(ImportError):
                 AgentFrameworkHelper.create_client(
                     ClientType.AzureOpenAIResponse,
                     endpoint="https://x",