mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2026-01-18 13:34:28 +01:00
2520ec6e08
* Rename PromptConfig to AIDirectives * Make PromptGenerator primarily responsible for generating prompt instead of AIConfig * Refactor PromptGenerator * Add `available` attribute to `Command` class, to determine availability based on `Agent` state
43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
from autogpt.config.ai_directives import AIDirectives
|
|
|
|
"""
|
|
Test cases for the PromptConfig class, which handles loads the Prompts configuration
|
|
settings from a YAML file.
|
|
"""
|
|
|
|
|
|
def test_prompt_config_loading(tmp_path):
|
|
"""Test if the prompt configuration loads correctly"""
|
|
|
|
yaml_content = """
|
|
constraints:
|
|
- A test constraint
|
|
- Another test constraint
|
|
- A third test constraint
|
|
resources:
|
|
- A test resource
|
|
- Another test resource
|
|
- A third test resource
|
|
best_practices:
|
|
- A test best-practice
|
|
- Another test best-practice
|
|
- A third test best-practice
|
|
"""
|
|
prompt_settings_file = tmp_path / "test_prompt_settings.yaml"
|
|
prompt_settings_file.write_text(yaml_content)
|
|
|
|
prompt_config = AIDirectives.from_file(prompt_settings_file)
|
|
|
|
assert len(prompt_config.constraints) == 3
|
|
assert prompt_config.constraints[0] == "A test constraint"
|
|
assert prompt_config.constraints[1] == "Another test constraint"
|
|
assert prompt_config.constraints[2] == "A third test constraint"
|
|
assert len(prompt_config.resources) == 3
|
|
assert prompt_config.resources[0] == "A test resource"
|
|
assert prompt_config.resources[1] == "Another test resource"
|
|
assert prompt_config.resources[2] == "A third test resource"
|
|
assert len(prompt_config.best_practices) == 3
|
|
assert prompt_config.best_practices[0] == "A test best-practice"
|
|
assert prompt_config.best_practices[1] == "Another test best-practice"
|
|
assert prompt_config.best_practices[2] == "A third test best-practice"
|