mirror of
https://github.com/microsoft/autogen.git
synced 2026-01-26 23:10:23 -05:00
Adding async support to get_human_input (#466)
* Adding async support to get_human_input * Adjust code for Code formatting testing fail * Adjust the test_async_get_human_input.py to run async on test * Adjust the test_async_get_human_input.py for pre-commit-check error * Adjust the test_async_get_human_input.py for pre-commit-check error v2 * Adjust remove unnecessary register_reply * Adjust test to use asyncio call * Adjust go back to not use asyncio
This commit is contained in:
35
test/agentchat/test_async_get_human_input.py
Normal file
35
test/agentchat/test_async_get_human_input.py
Normal file
@@ -0,0 +1,35 @@
|
||||
import asyncio
|
||||
import autogen
|
||||
import pytest
|
||||
from test_assistant_agent import KEY_LOC, OAI_CONFIG_LIST
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_async_get_human_input():
|
||||
try:
|
||||
import openai
|
||||
except ImportError:
|
||||
return
|
||||
config_list = autogen.config_list_from_json(OAI_CONFIG_LIST, KEY_LOC)
|
||||
|
||||
# create an AssistantAgent instance named "assistant"
|
||||
assistant = autogen.AssistantAgent(
|
||||
name="assistant",
|
||||
max_consecutive_auto_reply=2,
|
||||
llm_config={"request_timeout": 600, "seed": 41, "config_list": config_list, "temperature": 0},
|
||||
)
|
||||
|
||||
user_proxy = autogen.UserProxyAgent(name="user", human_input_mode="ALWAYS", code_execution_config=False)
|
||||
|
||||
async def custom_a_get_human_input(prompt):
|
||||
return "This is a test"
|
||||
|
||||
user_proxy.a_get_human_input = custom_a_get_human_input
|
||||
|
||||
user_proxy.register_reply([autogen.Agent, None], autogen.ConversableAgent.a_check_termination_and_human_reply)
|
||||
|
||||
await user_proxy.a_initiate_chat(assistant, clear_history=True, message="Hello.")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
test_async_get_human_input()
|
||||
Reference in New Issue
Block a user