fix: self healing response parser

This commit is contained in:
Florian Hönicke
2023-05-26 17:45:10 +02:00
parent 43bcfc31e2
commit d3242d75cc
2 changed files with 4 additions and 4 deletions

View File

@@ -21,7 +21,7 @@ from dev_gpt.constants import FILE_AND_TAG_PAIRS, NUM_IMPLEMENTATION_STRATEGIES,
IMPLEMENTATION_FILE_TAG, LANGUAGE_PACKAGES, UNNECESSARY_PACKAGES, DOCKER_BASE_IMAGE_VERSION, SEARCH_PACKAGES, \
INDICATOR_TO_IMPORT_STATEMENT
from dev_gpt.options.generate.conversation_logger import Timer
from dev_gpt.options.generate.parser import json_parser
from dev_gpt.options.generate.parser import json_parser, self_healing_json_parser
from dev_gpt.options.generate.pm.pm import PM
from dev_gpt.options.generate.templates_user import template_generate_microservice_name, \
template_generate_possible_packages, \
@@ -451,7 +451,7 @@ pytest
response_format_example=response_format_suggest_solutions,
)
suggested_solutions = json_parser(json_string)['solutions.json']
suggested_solutions = self_healing_json_parser(json_string)['solutions.json']
if len(self.previous_errors) > 0:
was_error_seen_before = json.loads(

View File

@@ -5,7 +5,7 @@ from dev_gpt.options.generate.chains.question_answering import is_question_true
from dev_gpt.options.generate.chains.translation import translation
from dev_gpt.options.generate.chains.user_confirmation_feedback_loop import user_feedback_loop
from dev_gpt.options.generate.chains.get_user_input_if_needed import get_user_input_if_needed
from dev_gpt.options.generate.parser import identity_parser, json_parser
from dev_gpt.options.generate.parser import identity_parser, json_parser, self_healing_json_parser
from dev_gpt.options.generate.pm.task_tree_schema import TaskTree
from dev_gpt.options.generate.prompt_factory import make_prompt_friendly
from dev_gpt.options.generate.ui import get_random_employee
@@ -156,7 +156,7 @@ def construct_sub_task_tree(self, microservice_description):
)
sub_task_tree_updated = ask_gpt(
sub_task_tree_update_prompt,
json_parser,
self_healing_json_parser,
microservice_description=microservice_description,
# nlp_fns=nlp_fns,
sub_task_tree=sub_task_tree_dict, solutions=solutions