mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-18 06:24:20 +01:00
formatting
This commit is contained in:
@@ -1,9 +1,10 @@
|
||||
from promptgenerator import PromptGenerator
|
||||
|
||||
|
||||
def get_prompt():
|
||||
"""
|
||||
This function generates a prompt string that includes various constraints, commands, resources, and performance evaluations.
|
||||
|
||||
|
||||
Returns:
|
||||
str: The generated prompt string.
|
||||
"""
|
||||
@@ -58,5 +59,5 @@ def get_prompt():
|
||||
|
||||
# Generate the prompt string
|
||||
prompt_string = prompt_generator.generate_prompt_string()
|
||||
|
||||
|
||||
return prompt_string
|
||||
|
||||
@@ -7,6 +7,7 @@ import os
|
||||
sys.path.append(os.path.abspath("../scripts"))
|
||||
from promptgenerator import PromptGenerator
|
||||
|
||||
|
||||
# Create a test class for the PromptGenerator, subclassed from unittest.TestCase
|
||||
class promptgenerator_tests(unittest.TestCase):
|
||||
|
||||
@@ -81,7 +82,7 @@ class promptgenerator_tests(unittest.TestCase):
|
||||
self.assertIn(constraint, prompt_string)
|
||||
for command in commands:
|
||||
self.assertIn(command["name"], prompt_string)
|
||||
|
||||
|
||||
# Check for each key-value pair in the command args dictionary
|
||||
for key, value in command["args"].items():
|
||||
self.assertIn(f'"{key}": "{value}"', prompt_string)
|
||||
@@ -94,6 +95,7 @@ class promptgenerator_tests(unittest.TestCase):
|
||||
self.assertIn("resources", prompt_string.lower())
|
||||
self.assertIn("performance evaluation", prompt_string.lower())
|
||||
|
||||
|
||||
# Run the tests when this script is executed
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user