aljaz/gpt-engineer
1
0
Fork 0
mirror of https://github.com/aljazceru/gpt-engineer.git synced 2025-12-17 12:45:26 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix linting

Browse Source
This commit is contained in:
Enzo Martin
2023-06-18 10:34:35 +02:00
parent 2f2fef50fe
commit 8b3862d94d
3 changed files with 20 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
gpt_engineer/chat_to_files.py
View File

@@ -1,15 +1,17 @@
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
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'
# 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
@@ -19,7 +21,7 @@ def parse_chat(chat):# -> List[Tuple[str, str]]:
files = []
for match in matches:
# Strip the filename of any non-allowed characters and convert / to \
path = re.sub(r'[<>"|?*]', '', match.group(1))
path = re.sub(r'[<>"|?*]', "", match.group(1))
# Get the code
code = match.group(2)
@@ -28,14 +30,14 @@ def parse_chat(chat):# -> List[Tuple[str, str]]:
files.append((path, code))
# Add README to the list
files.append(('README.txt', readme))
files.append(("README.txt", readme))
# Return the files
return files
def to_files(chat, workspace):
workspace['all_output.txt'] = chat
workspace["all_output.txt"] = chat
files = parse_chat(chat)
for file_name, file_content in files: