chore: import upstream snapshot with attribution

python/tests/integration/agents/bedrock_agent/conftest.py

@@ -0,0 +1,25 @@
+# Copyright (c) Microsoft. All rights reserved.
+
+from typing import Annotated
+
+import pytest
+
+from semantic_kernel.functions.kernel_function_decorator import kernel_function
+from semantic_kernel.kernel import Kernel
+
+
+class WeatherPlugin:
+    """Mock weather plugin."""
+
+    @kernel_function(description="Get real-time weather information.")
+    def current(self, location: Annotated[str, "The location to get the weather"]) -> str:
+        """Returns the current weather."""
+        return f"The weather in {location} is sunny."
+
+
+@pytest.fixture
+def kernel_with_dummy_function() -> Kernel:
+    kernel = Kernel()
+    kernel.add_plugin(WeatherPlugin(), plugin_name="weather")
+
+    return kernel

python/tests/integration/agents/bedrock_agent/test_bedrock_agent_integration.py

@@ -0,0 +1,141 @@
+# Copyright (c) Microsoft. All rights reserved.
+
+import uuid
+
+import pytest
+
+from semantic_kernel.agents.bedrock.bedrock_agent import BedrockAgent
+from semantic_kernel.contents.binary_content import BinaryContent
+from semantic_kernel.contents.chat_message_content import ChatMessageContent
+from semantic_kernel.contents.streaming_chat_message_content import StreamingChatMessageContent
+from semantic_kernel.contents.utils.author_role import AuthorRole
+
+
+class TestBedrockAgentIntegration:
+    @pytest.fixture(autouse=True)
+    async def setup_and_teardown(self, request):
+        """Setup and teardown for the test.
+
+        This is run for each test function, i.e. each test function will have its own instance of the agent.
+        """
+        try:
+            self.bedrock_agent = await BedrockAgent.create_and_prepare_agent(
+                f"semantic-kernel-integration-test-agent-{uuid.uuid4()}",
+                "You are a helpful assistant that help users with their questions.",
+            )
+            if hasattr(request, "param"):
+                if "enable_code_interpreter" in request.param:
+                    await self.bedrock_agent.create_code_interpreter_action_group()
+                if "kernel" in request.param:
+                    self.bedrock_agent.kernel = request.getfixturevalue(request.param.get("kernel"))
+                if "enable_kernel_function" in request.param:
+                    await self.bedrock_agent.create_kernel_function_action_group()
+        except Exception as e:
+            pytest.fail("Failed to create agent")
+            raise e
+        # Yield control to the test
+        yield
+        # Clean up
+        try:
+            await self.bedrock_agent.delete_agent()
+        except Exception as e:
+            pytest.fail(f"Failed to delete agent: {e}")
+            raise e
+
+    @pytest.mark.asyncio
+    async def test_invoke(self):
+        """Test invoke of the agent."""
+        async for response in self.bedrock_agent.invoke(messages="Hello"):
+            assert isinstance(response.message, ChatMessageContent)
+            assert response.message.role == AuthorRole.ASSISTANT
+            assert response.message.content is not None
+
+    @pytest.mark.asyncio
+    async def test_invoke_stream(self):
+        """Test invoke stream of the agent."""
+        async for response in self.bedrock_agent.invoke_stream(messages="Hello"):
+            assert isinstance(response.message, StreamingChatMessageContent)
+            assert response.message.role == AuthorRole.ASSISTANT
+            assert response.message.content is not None
+
+    @pytest.mark.asyncio
+    @pytest.mark.parametrize("setup_and_teardown", [{"enable_code_interpreter": True}], indirect=True)
+    async def test_code_interpreter(self):
+        """Test code interpreter."""
+        input_text = """
+Create a bar chart for the following data:
+Panda   5
+Tiger   8 
+Lion    3
+Monkey  6
+Dolphin  2
+"""
+        binary_item: BinaryContent | None = None
+        async for response in self.bedrock_agent.invoke(messages=input_text):
+            assert isinstance(response.message, ChatMessageContent)
+            assert response.message.role == AuthorRole.ASSISTANT
+            if not binary_item:
+                binary_item = next((item for item in response.message.items if isinstance(item, BinaryContent)), None)
+
+        assert binary_item
+
+    @pytest.mark.asyncio
+    @pytest.mark.parametrize("setup_and_teardown", [{"enable_code_interpreter": True}], indirect=True)
+    async def test_code_interpreter_stream(self):
+        """Test code interpreter streaming."""
+        input_text = """
+Create a bar chart for the following data:
+Panda   5
+Tiger   8 
+Lion    3
+Monkey  6
+Dolphin  2
+"""
+        binary_item: BinaryContent | None = None
+        async for response in self.bedrock_agent.invoke_stream(messages=input_text):
+            assert isinstance(response.message, StreamingChatMessageContent)
+            assert response.message.role == AuthorRole.ASSISTANT
+            binary_item = next((item for item in response.message.items if isinstance(item, BinaryContent)), None)
+        assert binary_item
+
+    @pytest.mark.asyncio
+    @pytest.mark.parametrize(
+        "setup_and_teardown",
+        [
+            {
+                "enable_kernel_function": True,
+                "kernel": "kernel_with_dummy_function",
+            },
+        ],
+        indirect=True,
+    )
+    async def test_function_calling(self):
+        """Test function calling."""
+        async for response in self.bedrock_agent.invoke(
+            messages="What is the weather in Seattle?",
+        ):
+            assert isinstance(response.message, ChatMessageContent)
+            assert response.message.role == AuthorRole.ASSISTANT
+            assert "sunny" in response.message.content
+
+    @pytest.mark.asyncio
+    @pytest.mark.parametrize(
+        "setup_and_teardown",
+        [
+            {
+                "enable_kernel_function": True,
+                "kernel": "kernel_with_dummy_function",
+            },
+        ],
+        indirect=True,
+    )
+    async def test_function_calling_stream(self):
+        """Test function calling streaming."""
+        full_message: str = ""
+        async for response in self.bedrock_agent.invoke_stream(
+            messages="What is the weather in Seattle?",
+        ):
+            assert isinstance(response.message, StreamingChatMessageContent)
+            assert response.message.role == AuthorRole.ASSISTANT
+            full_message += response.message.content
+        assert "sunny" in full_message