mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2026-01-13 19:24:18 +01:00
2eb346e06e
* Clean up prompt generation * Rename Performance Evaluations to Best Practices * Move specification of response format from system prompt to Agent.construct_base_prompt * Clean up PromptGenerator class * Add debug logging to AIConfig autogeneration * Clarify prompting and add support for multiple thought processes to Agent
43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
from autogpt.config.prompt_config import PromptConfig
|
|
|
|
"""
|
|
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 = PromptConfig(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"
|