github/autogen
1
1
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-04-20 03:02:16 -04:00
Code Issues Packages Projects Releases Wiki Activity

refactor: [conversable_agent] remove list of func pointers (#2005)

Browse Source 
* refactor: [conversable_agent] remove list of func pointers

Ideally register_reply creates less state in multiple places and avoid having two copies of the same 'func pointer'.

* Update conversable_agent.py

---------

Co-authored-by: Chi Wang <wang.chi@microsoft.com>
This commit is contained in:
Eduardo Salinas
2024-03-19 14:38:08 -04:00
committed by GitHub
parent e35db7ef5f
commit e23bdfb38e
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
autogen/agentchat/conversable_agent.py
View File

@@ -161,7 +161,6 @@ class ConversableAgent(LLMAgent):
)
self._default_auto_reply = default_auto_reply
self._reply_func_list = []
self._ignore_async_func_in_sync_chat_list = []
self._human_input = []
self.reply_at_receive = defaultdict(bool)
self.register_reply([Agent, None], ConversableAgent.generate_oai_reply)
@@ -335,10 +334,9 @@ class ConversableAgent(LLMAgent):
"config": copy.copy(config),
"init_config": config,
"reset_config": reset_config,
"ignore_async_in_sync_chat": ignore_async_in_sync_chat and inspect.iscoroutinefunction(reply_func),
},
)
if ignore_async_in_sync_chat and inspect.iscoroutinefunction(reply_func):
self._ignore_async_func_in_sync_chat_list.append(reply_func)
@staticmethod
def _summary_from_nested_chats(
@@ -836,9 +834,9 @@ class ConversableAgent(LLMAgent):
Raises:
RuntimeError: if any async reply functions are registered.
"""
reply_functions = {f["reply_func"] for f in self._reply_func_list}.difference(
self._ignore_async_func_in_sync_chat_list
)
reply_functions = {
f["reply_func"] for f in self._reply_func_list if not f.get("ignore_async_in_sync_chat", False)
}
async_reply_functions = [f for f in reply_functions if inspect.iscoroutinefunction(f)]
if async_reply_functions != []: