Adjust command formatting in prompt

This commit is contained in:
Reinier van der Leer
2023-08-23 01:04:39 +02:00
parent d8f02dc79f
commit a660619ea8
3 changed files with 3 additions and 3 deletions

View File

@@ -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)})"

View File

@@ -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:
"""

View File

@@ -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])"
)