mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-30 12:24:29 +01:00
abb397e442
Co-authored-by: Reinier van der Leer <github@pwuts.nl> Co-authored-by: Nicholas Tindle <nick@ntindle.com> Co-authored-by: Nicholas Tindle <nicktindle@outlook.com> Co-authored-by: k-boikov <64261260+k-boikov@users.noreply.github.com> Co-authored-by: merwanehamadi <merwanehamadi@gmail.com> Co-authored-by: Merwane Hamadi <merwanehamadi@gmail.com> Co-authored-by: Richard Beales <rich@richbeales.net> Co-authored-by: Luke K <2609441+lc0rp@users.noreply.github.com> Co-authored-by: Luke K (pr-0f3t) <2609441+lc0rp@users.noreply.github.com> Co-authored-by: Erik Peterson <e@eriklp.com> Co-authored-by: Auto-GPT-Bot <github-bot@agpt.co> Co-authored-by: Benny van der Lans <49377421+bfalans@users.noreply.github.com> Co-authored-by: Jan <jan-github@phobia.de> Co-authored-by: Robin Richtsfeld <robin.richtsfeld@gmail.com> Co-authored-by: Marc Bornträger <marc.borntraeger@gmail.com> Co-authored-by: Stefan Ayala <stefanayala3266@gmail.com> Co-authored-by: javableu <45064273+javableu@users.noreply.github.com> Co-authored-by: DGdev91 <DGdev91@users.noreply.github.com> Co-authored-by: Kinance <kinance@gmail.com> Co-authored-by: digger yu <digger-yu@outlook.com> Co-authored-by: David <scenaristeur@gmail.com> Co-authored-by: gravelBridge <john.tian31@gmail.com> Fix Python CI "update cassettes" step (#4591) fix CI (#4596) Fix inverted logic for deny_command (#4563) fix current_score.json generation (#4601) Fix duckduckgo rate limiting (#4592) Fix debug code challenge (#4632) Fix issues with information retrieval challenge a (#4622) fix issues with env configuration and .env.template (#4630) Fix prompt issue causing 'No Command' issues and challenge to fail (#4623) Fix benchmark logs (#4653) Fix typo in docs/setup.md (#4613) Fix run.sh shebang (#4561) Fix autogpt docker image not working because missing prompt_settings (#4680) Fix execute_command coming from plugins (#4730)
75 lines
2.3 KiB
Python
75 lines
2.3 KiB
Python
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from autogpt.config.ai_config import AIConfig
|
|
from autogpt.setup import generate_aiconfig_automatic, prompt_user
|
|
from tests.utils import requires_api_key
|
|
|
|
|
|
@pytest.mark.vcr
|
|
@requires_api_key("OPENAI_API_KEY")
|
|
def test_generate_aiconfig_automatic_default(patched_api_requestor):
|
|
user_inputs = [""]
|
|
with patch("autogpt.utils.session.prompt", side_effect=user_inputs):
|
|
ai_config = prompt_user()
|
|
|
|
assert isinstance(ai_config, AIConfig)
|
|
assert ai_config.ai_name is not None
|
|
assert ai_config.ai_role is not None
|
|
assert 1 <= len(ai_config.ai_goals) <= 5
|
|
|
|
|
|
@pytest.mark.vcr
|
|
@requires_api_key("OPENAI_API_KEY")
|
|
def test_generate_aiconfig_automatic_typical(patched_api_requestor):
|
|
user_prompt = "Help me create a rock opera about cybernetic giraffes"
|
|
ai_config = generate_aiconfig_automatic(user_prompt)
|
|
|
|
assert isinstance(ai_config, AIConfig)
|
|
assert ai_config.ai_name is not None
|
|
assert ai_config.ai_role is not None
|
|
assert 1 <= len(ai_config.ai_goals) <= 5
|
|
|
|
|
|
@pytest.mark.vcr
|
|
@requires_api_key("OPENAI_API_KEY")
|
|
def test_generate_aiconfig_automatic_fallback(patched_api_requestor):
|
|
user_inputs = [
|
|
"T&GF£OIBECC()!*",
|
|
"Chef-GPT",
|
|
"an AI designed to browse bake a cake.",
|
|
"Purchase ingredients",
|
|
"Bake a cake",
|
|
"",
|
|
"",
|
|
]
|
|
with patch("autogpt.utils.session.prompt", side_effect=user_inputs):
|
|
ai_config = prompt_user()
|
|
|
|
assert isinstance(ai_config, AIConfig)
|
|
assert ai_config.ai_name == "Chef-GPT"
|
|
assert ai_config.ai_role == "an AI designed to browse bake a cake."
|
|
assert ai_config.ai_goals == ["Purchase ingredients", "Bake a cake"]
|
|
|
|
|
|
@pytest.mark.vcr
|
|
@requires_api_key("OPENAI_API_KEY")
|
|
def test_prompt_user_manual_mode(patched_api_requestor):
|
|
user_inputs = [
|
|
"--manual",
|
|
"Chef-GPT",
|
|
"an AI designed to browse bake a cake.",
|
|
"Purchase ingredients",
|
|
"Bake a cake",
|
|
"",
|
|
"",
|
|
]
|
|
with patch("autogpt.utils.session.prompt", side_effect=user_inputs):
|
|
ai_config = prompt_user()
|
|
|
|
assert isinstance(ai_config, AIConfig)
|
|
assert ai_config.ai_name == "Chef-GPT"
|
|
assert ai_config.ai_role == "an AI designed to browse bake a cake."
|
|
assert ai_config.ai_goals == ["Purchase ingredients", "Bake a cake"]
|