mirror of
https://github.com/microsoft/autogen.git
synced 2025-08-03 14:22:34 +00:00
bf08856389
* fix type and default value of the code_execution_config of UserProxAgent * fix type and default value of the code_execution_config of UserProxAgent * set default value of llm_config in UserProxyAgent to None * fixed tests * revert llm_config to False --------- Co-authored-by: Chi Wang <wang.chi@microsoft.com>
104 lines
3.0 KiB
Python
104 lines
3.0 KiB
Python
from autogen import UserProxyAgent
|
|
import pytest
|
|
from conftest import skip_openai
|
|
import os
|
|
|
|
from autogen.code_utils import (
|
|
is_docker_running,
|
|
in_docker_container,
|
|
)
|
|
|
|
try:
|
|
import openai
|
|
except ImportError:
|
|
skip = True
|
|
else:
|
|
skip = False or skip_openai
|
|
|
|
|
|
def docker_running():
|
|
return is_docker_running() or in_docker_container()
|
|
|
|
|
|
@pytest.mark.skipif(skip, reason="openai not installed")
|
|
def test_agent_setup_with_code_execution_off():
|
|
user_proxy = UserProxyAgent(
|
|
name="test_agent",
|
|
human_input_mode="NEVER",
|
|
code_execution_config=False,
|
|
)
|
|
|
|
assert user_proxy._code_execution_config is False
|
|
|
|
|
|
@pytest.mark.skipif(skip, reason="openai not installed")
|
|
def test_agent_setup_with_use_docker_false():
|
|
user_proxy = UserProxyAgent(
|
|
name="test_agent",
|
|
human_input_mode="NEVER",
|
|
code_execution_config={"use_docker": False},
|
|
)
|
|
|
|
assert user_proxy._code_execution_config["use_docker"] is False
|
|
|
|
|
|
@pytest.mark.skipif(skip, reason="openai not installed")
|
|
def test_agent_setup_with_env_variable_false_and_docker_running(monkeypatch):
|
|
monkeypatch.setenv("AUTOGEN_USE_DOCKER", "False")
|
|
|
|
user_proxy = UserProxyAgent(
|
|
name="test_agent",
|
|
human_input_mode="NEVER",
|
|
)
|
|
|
|
assert user_proxy._code_execution_config["use_docker"] is False
|
|
|
|
|
|
@pytest.mark.skipif(skip or (not docker_running()), reason="openai not installed OR docker not running")
|
|
def test_agent_setup_with_default_and_docker_running(monkeypatch):
|
|
monkeypatch.delenv("AUTOGEN_USE_DOCKER", raising=False)
|
|
|
|
assert os.getenv("AUTOGEN_USE_DOCKER") is None
|
|
|
|
user_proxy = UserProxyAgent(
|
|
name="test_agent",
|
|
human_input_mode="NEVER",
|
|
)
|
|
|
|
assert os.getenv("AUTOGEN_USE_DOCKER") is None
|
|
|
|
assert user_proxy._code_execution_config["use_docker"] is True
|
|
|
|
|
|
@pytest.mark.skipif(skip or (docker_running()), reason="openai not installed OR docker running")
|
|
def test_raises_error_agent_setup_with_default_and_docker_not_running(monkeypatch):
|
|
monkeypatch.delenv("AUTOGEN_USE_DOCKER", raising=False)
|
|
with pytest.raises(RuntimeError):
|
|
UserProxyAgent(
|
|
name="test_agent",
|
|
human_input_mode="NEVER",
|
|
)
|
|
|
|
|
|
@pytest.mark.skipif(skip or (docker_running()), reason="openai not installed OR docker running")
|
|
def test_raises_error_agent_setup_with_env_variable_true_and_docker_not_running(monkeypatch):
|
|
monkeypatch.setenv("AUTOGEN_USE_DOCKER", "True")
|
|
|
|
with pytest.raises(RuntimeError):
|
|
UserProxyAgent(
|
|
name="test_agent",
|
|
human_input_mode="NEVER",
|
|
)
|
|
|
|
|
|
@pytest.mark.skipif(skip or (not docker_running()), reason="openai not installed OR docker not running")
|
|
def test_agent_setup_with_env_variable_true_and_docker_running(monkeypatch):
|
|
monkeypatch.setenv("AUTOGEN_USE_DOCKER", "True")
|
|
|
|
user_proxy = UserProxyAgent(
|
|
name="test_agent",
|
|
human_input_mode="NEVER",
|
|
)
|
|
|
|
assert user_proxy._code_execution_config["use_docker"] is True
|