This commit is contained in:
Carl Thomé
2023-06-18 15:04:30 +02:00
parent 742b56804e
commit 502756bb58
2 changed files with 9 additions and 1 deletions

View File

@@ -1,5 +1,6 @@
import re import re
def parse_chat(chat): # -> List[Tuple[str, str]]: def parse_chat(chat): # -> List[Tuple[str, str]]:
# Split the chat into sections by the "*CODEBLOCKSBELOW*" token # Split the chat into sections by the "*CODEBLOCKSBELOW*" token
split_chat = chat.split("*CODEBLOCKSBELOW*") split_chat = chat.split("*CODEBLOCKSBELOW*")

View File

@@ -199,7 +199,14 @@ STEPS = {
"benchmark": [gen_spec, gen_unit_tests, gen_code, gen_entrypoint], "benchmark": [gen_spec, gen_unit_tests, gen_code, gen_entrypoint],
"simple": [simple_gen, gen_entrypoint, execute_entrypoint], "simple": [simple_gen, gen_entrypoint, execute_entrypoint],
"clarify": [clarify, gen_clarified_code, gen_entrypoint, execute_entrypoint], "clarify": [clarify, gen_clarified_code, gen_entrypoint, execute_entrypoint],
"respec": [gen_spec, respec, gen_unit_tests, gen_code, gen_entrypoint, execute_entrypoint], "respec": [
gen_spec,
respec,
gen_unit_tests,
gen_code,
gen_entrypoint,
execute_entrypoint,
],
"execute_only": [execute_entrypoint], "execute_only": [execute_entrypoint],
"use_feedback": [use_feedback], "use_feedback": [use_feedback],
} }