From 3110a5ac8af3a91caa1531ed2e89ea733be7f651 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Florian=20Ho=CC=88nicke?= Date: Thu, 25 May 2023 15:08:57 +0200 Subject: [PATCH] =?UTF-8?q?=F0=9F=92=81=20fix:=20pm?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../generate/chains/user_confirmation_feedback_loop.py | 7 +++---- dev_gpt/options/generate/pm/pm.py | 2 +- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/dev_gpt/options/generate/chains/user_confirmation_feedback_loop.py b/dev_gpt/options/generate/chains/user_confirmation_feedback_loop.py index 4b114d1..9b92a79 100644 --- a/dev_gpt/options/generate/chains/user_confirmation_feedback_loop.py +++ b/dev_gpt/options/generate/chains/user_confirmation_feedback_loop.py @@ -2,15 +2,14 @@ from dev_gpt.apis.gpt import ask_gpt from dev_gpt.options.generate.parser import identity_parser -def user_feedback_loop(context, current_description): +def user_feedback_loop(current_description): while (user_feedback := get_user_feedback(current_description)): - context['user_feedback'] = user_feedback current_description = ask_gpt( add_feedback_prompt, identity_parser, - **context + microservice_description=current_description, + user_feedback=user_feedback, ) - del context['user_feedback'] return current_description def get_user_feedback(microservice_description): diff --git a/dev_gpt/options/generate/pm/pm.py b/dev_gpt/options/generate/pm/pm.py index 83a3d98..5361983 100644 --- a/dev_gpt/options/generate/pm/pm.py +++ b/dev_gpt/options/generate/pm/pm.py @@ -43,7 +43,7 @@ Description of the microservice: def refine_description(self, microservice_description): context = {'microservice_description': microservice_description} auto_refine_description(context) - microservice_description = user_feedback_loop(context, context['microservice_description']) + microservice_description = user_feedback_loop(context['microservice_description']) test_description = ask_gpt( generate_test_assertion_prompt,