mirror of
https://github.com/microsoft/autogen.git
synced 2025-07-27 02:41:12 +00:00
2997c27d1b
* add react-flow * update tutorial, some ui improvement for agent thread layout * v1 for interactive viz for agent state * add Ui visualization of message based transitions * minor updates * fix node edges, support visualization of self loops * improve edges and layout, add support for selector group chat prompt * minor ui tweaks * ui and layout updates * ugrade dependencies to fix dependabot scan errors * persist sidebar, enable contentbar title mechanism #4191 * add support for user proxy agent, support human in put mode. #4011 * add UI support for human input mode via a userproxy agent #4011 * version update * fix db initialization bug * support for human input mode in UI, fix backend api route minor bugs * update pyproject toml and uv lock * readme update, support full screen mode for agent visualiation * update uv.lock
48 lines
1.8 KiB
Python
48 lines
1.8 KiB
Python
from typing import Callable, List, Optional, Sequence, Union, Awaitable
|
|
from inspect import iscoroutinefunction
|
|
|
|
from autogen_agentchat.agents import BaseChatAgent
|
|
from autogen_agentchat.base import Response
|
|
from autogen_agentchat.messages import ChatMessage, TextMessage
|
|
from autogen_core.base import CancellationToken
|
|
import asyncio
|
|
|
|
|
|
class UserProxyAgent(BaseChatAgent):
|
|
"""An agent that can represent a human user in a chat."""
|
|
|
|
def __init__(
|
|
self,
|
|
name: str,
|
|
description: Optional[str] = "a",
|
|
input_func: Optional[Union[Callable[..., str],
|
|
Callable[..., Awaitable[str]]]] = None
|
|
) -> None:
|
|
super().__init__(name, description=description)
|
|
self.input_func = input_func or input
|
|
self._is_async = iscoroutinefunction(
|
|
input_func) if input_func else False
|
|
|
|
@property
|
|
def produced_message_types(self) -> List[type[ChatMessage]]:
|
|
return [TextMessage]
|
|
|
|
async def _get_input(self, prompt: str) -> str:
|
|
"""Handle both sync and async input functions"""
|
|
if self._is_async:
|
|
return await self.input_func(prompt)
|
|
else:
|
|
return await asyncio.get_event_loop().run_in_executor(None, self.input_func, prompt)
|
|
|
|
async def on_messages(self, messages: Sequence[ChatMessage], cancellation_token: CancellationToken) -> Response:
|
|
|
|
try:
|
|
user_input = await self._get_input("Enter your response: ")
|
|
return Response(chat_message=TextMessage(content=user_input, source=self.name))
|
|
except Exception as e:
|
|
# Consider logging the error here
|
|
raise RuntimeError(f"Failed to get user input: {str(e)}") from e
|
|
|
|
async def on_reset(self, cancellation_token: CancellationToken) -> None:
|
|
pass
|