Register async human input handler (microsoft#794)

* Update conversable_agent.py * Add files via upload * Delete notebook/Async_human_input.ipynb * Add files via upload * refactor:formatter * feat:updated position --------- Co-authored-by: Chi Wang <[email protected]>
whiskyboy · Dec 3, 2023 · c2386bf · c2386bf
1 parent a9dd2fe
commit c2386bf
Show file tree

Hide file tree

Showing 4 changed files with 379 additions and 2 deletions.
diff --git a/autogen/agentchat/contrib/llava_agent.py b/autogen/agentchat/contrib/llava_agent.py
@@ -54,7 +54,7 @@ def __init__(
         )
 
         assert self.llm_config is not None, "llm_config must be provided."
-        self.register_reply([Agent, None], reply_func=LLaVAAgent._image_reply, position=1)
+        self.register_reply([Agent, None], reply_func=LLaVAAgent._image_reply, position=2)
 
     def _image_reply(self, messages=None, sender=None, config=None):
         # Note: we did not use "llm_config" yet.

diff --git a/autogen/agentchat/contrib/retrieve_user_proxy_agent.py b/autogen/agentchat/contrib/retrieve_user_proxy_agent.py
@@ -206,7 +206,7 @@ def retrieve_docs(self, problem: str, n_results: int = 20, search_string: str =
         self._is_termination_msg = (
             self._is_termination_msg_retrievechat if is_termination_msg is None else is_termination_msg
         )
-        self.register_reply(Agent, RetrieveUserProxyAgent._generate_retrieve_user_reply, position=1)
+        self.register_reply(Agent, RetrieveUserProxyAgent._generate_retrieve_user_reply, position=2)
 
     def _is_termination_msg_retrievechat(self, message):
         """Check if a message is a termination message.

diff --git a/autogen/agentchat/conversable_agent.py b/autogen/agentchat/conversable_agent.py
@@ -133,6 +133,7 @@ def __init__(
         self.register_reply([Agent, None], ConversableAgent.generate_function_call_reply)
         self.register_reply([Agent, None], ConversableAgent.generate_async_function_call_reply)
         self.register_reply([Agent, None], ConversableAgent.check_termination_and_human_reply)
+        self.register_reply([Agent, None], ConversableAgent.a_check_termination_and_human_reply)
 
     def register_reply(
         self,