mirror of
https://github.com/aljazceru/gpt-engineer.git
synced 2025-12-17 12:45:26 +01:00
e7df947b98
- Enforce an explicit markdown code block format - Add a token to split the output to clearly detect when the code blocks start - Save all non-code output to a `README.md` file - Update RegEx to extract and strip text more reliably and clean up the output - Update the identify prompts appropriately
43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
import re
|
|
|
|
def parse_chat(chat):# -> List[Tuple[str, str]]:
|
|
# Split the chat into sections by the '*CODEBLOCKSBELOW*' token
|
|
split_chat = chat.split('*CODEBLOCKSBELOW*')
|
|
|
|
# Check if the '*CODEBLOCKSBELOW*' token was found
|
|
is_token_found = len(split_chat) > 1
|
|
|
|
# If the '*CODEBLOCKSBELOW*' token is found, use the first part as README and second part as code blocks.
|
|
# Otherwise, treat README as optional and proceed with empty README and the entire chat as code blocks
|
|
readme = split_chat[0].strip() if is_token_found else 'No readme'
|
|
code_blocks = split_chat[1] if is_token_found else chat
|
|
|
|
# Get all ``` blocks and preceding filenames
|
|
regex = r"\[(.*?)\]\s*```.*?\n(.*?)```"
|
|
matches = re.finditer(regex, code_blocks, re.DOTALL)
|
|
|
|
files = []
|
|
for match in matches:
|
|
# Strip the filename of any non-allowed characters and convert / to \
|
|
path = re.sub(r'[<>"|?*]', '', match.group(1))
|
|
|
|
# Get the code
|
|
code = match.group(2)
|
|
|
|
# Add the file to the list
|
|
files.append((path, code))
|
|
|
|
# Add README to the list
|
|
files.append(('README.txt', readme))
|
|
|
|
# Return the files
|
|
return files
|
|
|
|
|
|
def to_files(chat, workspace):
|
|
workspace['all_output.txt'] = chat
|
|
|
|
files = parse_chat(chat)
|
|
for file_name, file_content in files:
|
|
workspace[file_name] = file_content
|