mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2026-01-25 08:54:28 +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.0 KiB
Python
75 lines
2.0 KiB
Python
from autogpt.config.ai_config import AIConfig
|
|
|
|
"""
|
|
Test cases for the AIConfig class, which handles loads the AI configuration
|
|
settings from a YAML file.
|
|
"""
|
|
|
|
|
|
def test_goals_are_always_lists_of_strings(tmp_path):
|
|
"""Test if the goals attribute is always a list of strings."""
|
|
|
|
yaml_content = """
|
|
ai_goals:
|
|
- Goal 1: Make a sandwich
|
|
- Goal 2, Eat the sandwich
|
|
- Goal 3 - Go to sleep
|
|
- "Goal 4: Wake up"
|
|
ai_name: McFamished
|
|
ai_role: A hungry AI
|
|
api_budget: 0.0
|
|
"""
|
|
config_file = tmp_path / "ai_settings.yaml"
|
|
config_file.write_text(yaml_content)
|
|
|
|
ai_config = AIConfig.load(config_file)
|
|
|
|
assert len(ai_config.ai_goals) == 4
|
|
assert ai_config.ai_goals[0] == "Goal 1: Make a sandwich"
|
|
assert ai_config.ai_goals[1] == "Goal 2, Eat the sandwich"
|
|
assert ai_config.ai_goals[2] == "Goal 3 - Go to sleep"
|
|
assert ai_config.ai_goals[3] == "Goal 4: Wake up"
|
|
|
|
config_file.write_text("")
|
|
ai_config.save(config_file)
|
|
|
|
yaml_content2 = """ai_goals:
|
|
- 'Goal 1: Make a sandwich'
|
|
- Goal 2, Eat the sandwich
|
|
- Goal 3 - Go to sleep
|
|
- 'Goal 4: Wake up'
|
|
ai_name: McFamished
|
|
ai_role: A hungry AI
|
|
api_budget: 0.0
|
|
"""
|
|
assert config_file.read_text() == yaml_content2
|
|
|
|
|
|
def test_ai_config_file_not_exists(workspace):
|
|
"""Test if file does not exist."""
|
|
|
|
config_file = workspace.get_path("ai_settings.yaml")
|
|
|
|
ai_config = AIConfig.load(str(config_file))
|
|
assert ai_config.ai_name == ""
|
|
assert ai_config.ai_role == ""
|
|
assert ai_config.ai_goals == []
|
|
assert ai_config.api_budget == 0.0
|
|
assert ai_config.prompt_generator is None
|
|
assert ai_config.command_registry is None
|
|
|
|
|
|
def test_ai_config_file_is_empty(workspace):
|
|
"""Test if file does not exist."""
|
|
|
|
config_file = workspace.get_path("ai_settings.yaml")
|
|
config_file.write_text("")
|
|
|
|
ai_config = AIConfig.load(str(config_file))
|
|
assert ai_config.ai_name == ""
|
|
assert ai_config.ai_role == ""
|
|
assert ai_config.ai_goals == []
|
|
assert ai_config.api_budget == 0.0
|
|
assert ai_config.prompt_generator is None
|
|
assert ai_config.command_registry is None
|