fix: handle \n on start of input

pull/16/head
hanchchch 1 year ago
parent c0ef2c5ef1
commit 9d40a95ecc

@ -74,16 +74,16 @@ class CodeEditor(BaseToolSet):
name="CodeEditor.WRITE",
description="Write code to create a new tool. "
"If the code is completed, use the Terminal tool to execute it, if not, append the code through the CodeEditor.APPEND tool. "
"Input should be formatted like below.\n"
"<filename>\n<code>\n"
"Here is an example:\n"
"Input should be formatted like: "
"<filename>\n<code>\n\n"
"Here is an example: "
"test.py\nmessage = 'hello world'\nprint(message)\n"
"\n"
"The output will be last 3 lines you wrote.",
)
def write(self, inputs: str) -> str:
try:
code = CodeWriter.write(inputs)
code = CodeWriter.write(inputs.lstrip())
output = "Last 3 line was:\n" + "\n".join(code.split("\n")[-3:])
except Exception as e:
output = str(e)

Loading…
Cancel
Save