From a660619ea823e70c0c6a6f864adf3bb3f0965d17 Mon Sep 17 00:00:00 2001 From: Reinier van der Leer Date: Wed, 23 Aug 2023 01:04:39 +0200 Subject: [PATCH] Adjust command formatting in prompt --- autogpt/models/command.py | 2 +- autogpt/prompts/generator.py | 2 +- tests/unit/test_commands.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/autogpt/models/command.py b/autogpt/models/command.py index c9cccf84..a3369bb9 100644 --- a/autogpt/models/command.py +++ b/autogpt/models/command.py @@ -58,4 +58,4 @@ class Command: f"{param.name}: {param.type if param.required else f'Optional[{param.type}]'}" for param in self.parameters ] - return f"{self.name}: {self.description}, params: ({', '.join(params)})" + return f"{self.name}: {self.description}. Params: ({', '.join(params)})" diff --git a/autogpt/prompts/generator.py b/autogpt/prompts/generator.py index a5b69fdc..755a3689 100644 --- a/autogpt/prompts/generator.py +++ b/autogpt/prompts/generator.py @@ -53,7 +53,7 @@ class PromptGenerator: params_string = ", ".join( f'"{key}": "{value}"' for key, value in self.params.items() ) - return f'{self.label}: "{self.name}", params: ({params_string})' + return f'{self.label}: "{self.name}". Params: ({params_string})' def add_constraint(self, constraint: str) -> None: """ diff --git a/tests/unit/test_commands.py b/tests/unit/test_commands.py index cf4e44ad..25867f21 100644 --- a/tests/unit/test_commands.py +++ b/tests/unit/test_commands.py @@ -33,7 +33,7 @@ def test_command_creation(): assert cmd.description == "Example command" assert cmd.method == example_command_method assert ( - str(cmd) == "example: Example command, params: (arg1: int, arg2: Optional[str])" + str(cmd) == "example: Example command. Params: (arg1: int, arg2: Optional[str])" )