mirror of
https://github.com/microsoft/autogen.git
synced 2025-08-04 06:42:35 +00:00
c80df8acab
* --skip-openai * All tests pass * Update build.yml * Update Contribute.md * Fix for failing Ubuntu tests * More tests skipped, fixing 3.10 build * Apply suggestions from code review Co-authored-by: Qingyun Wu <qingyun0327@gmail.com> * Added more comments * fixed test__wrap_function_* --------- Co-authored-by: Qingyun Wu <qingyun0327@gmail.com> Co-authored-by: Davor Runje <davor@airt.ai>
41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
import asyncio
|
|
import autogen
|
|
import pytest
|
|
from conftest import skip_openai
|
|
from test_assistant_agent import KEY_LOC, OAI_CONFIG_LIST
|
|
|
|
try:
|
|
from openai import OpenAI
|
|
except ImportError:
|
|
skip = True
|
|
else:
|
|
skip = False or skip_openai
|
|
|
|
|
|
@pytest.mark.skipif(skip, reason="openai not installed OR requested to skip")
|
|
@pytest.mark.asyncio
|
|
async def test_async_get_human_input():
|
|
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={"timeout": 600, "cache_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()
|