chore: import upstream snapshot with attribution

python/samples/demos/document_generator/agents/user_agent.py

@@ -0,0 +1,65 @@
+# Copyright (c) Microsoft. All rights reserved.
+
+import sys
+from collections.abc import AsyncIterable, Awaitable, Callable
+from typing import TYPE_CHECKING, Any
+
+if sys.version_info >= (3, 12):
+    from typing import override  # pragma: no cover
+else:
+    from typing_extensions import override  # pragma: no cover
+
+from samples.demos.document_generator.agents.custom_agent_base import CustomAgentBase, Services
+from samples.demos.document_generator.plugins.user_plugin import UserPlugin
+from semantic_kernel.contents import ChatMessageContent
+from semantic_kernel.functions import KernelArguments
+
+if TYPE_CHECKING:
+    from semantic_kernel.agents.agent import AgentResponseItem, AgentThread
+    from semantic_kernel.kernel import Kernel
+
+INSTRUCTION = """
+You are part of a chat with multiple agents working on a document.
+
+Your task is to summarize the user's feedback on the latest draft from the author agent.
+Present the draft to the user and summarize their feedback.
+
+Do not try to address the user's feedback in this chat.
+"""
+
+DESCRIPTION = """
+Select me if you want to ask the user to review the latest draft for publication.
+"""
+
+
+class UserAgent(CustomAgentBase):
+    def __init__(self):
+        super().__init__(
+            service=self._create_ai_service(Services.AZURE_OPENAI),
+            plugins=[UserPlugin()],
+            name="UserAgent",
+            instructions=INSTRUCTION.strip(),
+            description=DESCRIPTION.strip(),
+        )
+
+    @override
+    async def invoke(
+        self,
+        *,
+        messages: str | ChatMessageContent | list[str | ChatMessageContent] | None = None,
+        thread: "AgentThread | None" = None,
+        on_intermediate_message: Callable[[ChatMessageContent], Awaitable[None]] | None = None,
+        arguments: KernelArguments | None = None,
+        kernel: "Kernel | None" = None,
+        **kwargs: Any,
+    ) -> AsyncIterable["AgentResponseItem[ChatMessageContent]"]:
+        async for response in super().invoke(
+            messages=messages,
+            thread=thread,
+            on_intermediate_message=on_intermediate_message,
+            arguments=arguments,
+            kernel=kernel,
+            additional_user_message="Now present the latest draft to the user for feedback and summarize their feedback.",  # noqa: E501
+            **kwargs,
+        ):
+            yield response