wehub-resource-sync
66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
# 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
|