mirror of
https://github.com/aljazceru/gpt-engineer.git
synced 2025-12-17 12:45:26 +01:00
Make gpt-engineer pip installable/runnable (#60)
This commit is contained in:
27
gpt_engineer/chat_to_files.py
Normal file
27
gpt_engineer/chat_to_files.py
Normal file
@@ -0,0 +1,27 @@
|
||||
import re
|
||||
|
||||
|
||||
def parse_chat(chat): # -> List[Tuple[str, str]]:
|
||||
# Get all ``` blocks
|
||||
regex = r"```(.*?)```"
|
||||
|
||||
matches = re.finditer(regex, chat, re.DOTALL)
|
||||
|
||||
files = []
|
||||
for match in matches:
|
||||
path = match.group(1).split("\n")[0]
|
||||
# Get the code
|
||||
code = match.group(1).split("\n")[1:]
|
||||
code = "\n".join(code)
|
||||
# Add the file to the list
|
||||
files.append((path, code))
|
||||
|
||||
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
|
||||
Reference in New Issue
Block a user