From d3242d75cca38d51b9feeea0e31a91b125dbd1ab Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Florian=20Ho=CC=88nicke?= Date: Fri, 26 May 2023 17:45:10 +0200 Subject: [PATCH] =?UTF-8?q?=E2=8F=AA=20fix:=20self=20healing=20response=20?= =?UTF-8?q?parser?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- dev_gpt/options/generate/generator.py | 4 ++-- dev_gpt/options/generate/pm/pm.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/dev_gpt/options/generate/generator.py b/dev_gpt/options/generate/generator.py index 8ac2ca9..1ccee63 100644 --- a/dev_gpt/options/generate/generator.py +++ b/dev_gpt/options/generate/generator.py @@ -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( diff --git a/dev_gpt/options/generate/pm/pm.py b/dev_gpt/options/generate/pm/pm.py index 17468cb..8f6db76 100644 --- a/dev_gpt/options/generate/pm/pm.py +++ b/dev_gpt/options/generate/pm/pm.py @@ -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