Move FunctionChoiceBehavior imports to top of test files

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

Author: SergeyMenshykh

python/tests/unit/agents/azure_ai_agent/test_agent_thread_actions.py

@@ -26,6 +26,7 @@
 
 from semantic_kernel.agents.azure_ai.agent_thread_actions import AgentThreadActions
 from semantic_kernel.agents.azure_ai.azure_ai_agent import AzureAIAgent
+from semantic_kernel.connectors.ai.function_choice_behavior import FunctionChoiceBehavior
 from semantic_kernel.contents import FunctionCallContent, FunctionResultContent, TextContent
 from semantic_kernel.contents.chat_message_content import ChatMessageContent
 from semantic_kernel.contents.utils.author_role import AuthorRole
@@ -362,7 +363,6 @@ async def test_agent_thread_actions_invoke_stream(ai_project_client, ai_agent_de
 
 import pytest
 
-from semantic_kernel.connectors.ai.function_choice_behavior import FunctionChoiceBehavior
 from semantic_kernel.exceptions.agent_exceptions import AgentInvokeException
 
 

python/tests/unit/agents/azure_ai_agent/test_azure_ai_agent.py

@@ -9,6 +9,7 @@
 from semantic_kernel.agents.agent import AgentResponseItem
 from semantic_kernel.agents.azure_ai.azure_ai_agent import AzureAIAgent, AzureAIAgentThread
 from semantic_kernel.agents.channels.agent_channel import AgentChannel
+from semantic_kernel.connectors.ai.function_choice_behavior import FunctionChoiceBehavior
 from semantic_kernel.contents.chat_history import ChatHistory
 from semantic_kernel.contents.chat_message_content import ChatMessageContent
 from semantic_kernel.contents.function_call_content import FunctionCallContent
@@ -401,8 +402,6 @@ def test_create_client_raises_if_no_endpoint():
 
 
 async def test_azure_ai_agent_get_response_passes_function_choice_behavior(ai_project_client, ai_agent_definition):
-    from semantic_kernel.connectors.ai.function_choice_behavior import FunctionChoiceBehavior
-
     agent = AzureAIAgent(client=ai_project_client, definition=ai_agent_definition)
     thread = AsyncMock(spec=AzureAIAgentThread)
     fcb = FunctionChoiceBehavior.Auto()
@@ -422,8 +421,6 @@ async def fake_invoke(*args, **kwargs):
 
 
 async def test_azure_ai_agent_invoke_passes_function_choice_behavior(ai_project_client, ai_agent_definition):
-    from semantic_kernel.connectors.ai.function_choice_behavior import FunctionChoiceBehavior
-
     agent = AzureAIAgent(client=ai_project_client, definition=ai_agent_definition)
     thread = AsyncMock(spec=AzureAIAgentThread)
     fcb = FunctionChoiceBehavior.Auto()
@@ -444,8 +441,6 @@ async def fake_invoke(*args, **kwargs):
 
 
 async def test_azure_ai_agent_invoke_stream_passes_function_choice_behavior(ai_project_client, ai_agent_definition):
-    from semantic_kernel.connectors.ai.function_choice_behavior import FunctionChoiceBehavior
-
     agent = AzureAIAgent(client=ai_project_client, definition=ai_agent_definition)
     thread = AsyncMock(spec=AzureAIAgentThread)
     fcb = FunctionChoiceBehavior.Auto()

python/tests/unit/agents/openai_assistant/test_assistant_thread_actions.py

@@ -55,6 +55,7 @@
 from semantic_kernel.agents.open_ai.function_action_result import FunctionActionResult
 from semantic_kernel.agents.open_ai.openai_assistant_agent import OpenAIAssistantAgent
 from semantic_kernel.agents.open_ai.run_polling_options import RunPollingOptions
+from semantic_kernel.connectors.ai.function_choice_behavior import FunctionChoiceBehavior
 from semantic_kernel.contents.chat_message_content import ChatMessageContent
 from semantic_kernel.contents.file_reference_content import FileReferenceContent
 from semantic_kernel.contents.function_call_content import FunctionCallContent
@@ -858,9 +859,6 @@ async def test_handle_streaming_requires_action_returns_none():
 # region Security tests for tools override and function_choice_behavior
 
 
-from semantic_kernel.connectors.ai.function_choice_behavior import FunctionChoiceBehavior
-
-
 async def test_validate_function_choice_behavior_rejects_required():
     """Required FCB is not supported for agent invocations."""
     with pytest.raises(AgentInvokeException, match="not supported"):

python/tests/unit/agents/openai_assistant/test_openai_assistant_agent.py

@@ -10,6 +10,7 @@
 from semantic_kernel.agents import AgentRegistry, AgentResponseItem, OpenAIAssistantAgent
 from semantic_kernel.agents.open_ai.openai_assistant_agent import AssistantAgentThread
 from semantic_kernel.agents.open_ai.run_polling_options import RunPollingOptions
+from semantic_kernel.connectors.ai.function_choice_behavior import FunctionChoiceBehavior
 from semantic_kernel.contents.chat_history import ChatHistory
 from semantic_kernel.contents.chat_message_content import ChatMessageContent
 from semantic_kernel.contents.function_call_content import FunctionCallContent
@@ -502,8 +503,6 @@ async def test_openai_assistant_agent_from_yaml_invalid_type():
 async def test_openai_assistant_agent_get_response_passes_function_choice_behavior(
     openai_client, assistant_definition
 ):
-    from semantic_kernel.connectors.ai.function_choice_behavior import FunctionChoiceBehavior
-
     agent = OpenAIAssistantAgent(client=openai_client, definition=assistant_definition)
     thread = AsyncMock(spec=AssistantAgentThread)
     fcb = FunctionChoiceBehavior.Auto()
@@ -523,8 +522,6 @@ async def fake_invoke(*args, **kwargs):
 
 
 async def test_openai_assistant_agent_invoke_passes_function_choice_behavior(openai_client, assistant_definition):
-    from semantic_kernel.connectors.ai.function_choice_behavior import FunctionChoiceBehavior
-
     agent = OpenAIAssistantAgent(client=openai_client, definition=assistant_definition)
     thread = AsyncMock(spec=AssistantAgentThread)
     fcb = FunctionChoiceBehavior.Auto()
@@ -547,8 +544,6 @@ async def fake_invoke(*args, **kwargs):
 async def test_openai_assistant_agent_invoke_stream_passes_function_choice_behavior(
     openai_client, assistant_definition
 ):
-    from semantic_kernel.connectors.ai.function_choice_behavior import FunctionChoiceBehavior
-
     agent = OpenAIAssistantAgent(client=openai_client, definition=assistant_definition)
     thread = AsyncMock(spec=AssistantAgentThread)
     fcb = FunctionChoiceBehavior.Auto()