From 61471fce3848a0b3bfde40d12b450abfa26257b9 Mon Sep 17 00:00:00 2001 From: ksridhar Date: Tue, 28 Jan 2025 10:59:41 -0500 Subject: [PATCH 1/3] put default values for response json --- src/allie/flowkit/endpoints/mechscriptbot.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/allie/flowkit/endpoints/mechscriptbot.py b/src/allie/flowkit/endpoints/mechscriptbot.py index 72c115d..abdfd21 100644 --- a/src/allie/flowkit/endpoints/mechscriptbot.py +++ b/src/allie/flowkit/endpoints/mechscriptbot.py @@ -67,16 +67,16 @@ async def triggermechscriptbot(request: MechScriptBotRequest, api_key: str = Hea response_dict = requests.get(url=url, json=request_dict_copy).json() - output = f"```{response_dict['output']}" + output = f"```{response_dict.get('output', '')}" - human_new_mem, ai_new_mem = tuple(response_dict["new_memory"]) + human_new_mem, ai_new_mem = tuple(response_dict.get("new_memory", [])) updated_human_memory = request.full_human_memory + [human_new_mem] updated_ai_memory = request.full_ai_memory + [ai_new_mem] - new_variables_list = [f"{var_name}:{var_type}" for var_name, var_type in (response_dict["new_variables"]).items()] + new_variables_list = [f"{var_name}:{var_type}" for var_name, var_type in (response_dict.get("new_variables", {})).items()] updated_variables = request.full_variables + new_variables_list - updated_mechanical_objects = request.full_mechanical_objects + response_dict["new_mechanical_objects"] + updated_mechanical_objects = request.full_mechanical_objects + response_dict.get("new_mechanical_objects", []) return MechScriptBotResponse( output=output, From ae6a453368bff801ffaf20ca235bea3284f0f969 Mon Sep 17 00:00:00 2001 From: ksridhar Date: Tue, 28 Jan 2025 11:06:15 -0500 Subject: [PATCH 2/3] fix style issue with long line --- src/allie/flowkit/endpoints/mechscriptbot.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/allie/flowkit/endpoints/mechscriptbot.py b/src/allie/flowkit/endpoints/mechscriptbot.py index abdfd21..6064d19 100644 --- a/src/allie/flowkit/endpoints/mechscriptbot.py +++ b/src/allie/flowkit/endpoints/mechscriptbot.py @@ -73,7 +73,10 @@ async def triggermechscriptbot(request: MechScriptBotRequest, api_key: str = Hea updated_human_memory = request.full_human_memory + [human_new_mem] updated_ai_memory = request.full_ai_memory + [ai_new_mem] - new_variables_list = [f"{var_name}:{var_type}" for var_name, var_type in (response_dict.get("new_variables", {})).items()] + new_variables_list = [ + f"{var_name}:{var_type}" + for var_name, var_type in (response_dict.get("new_variables", {})).items() + ] updated_variables = request.full_variables + new_variables_list updated_mechanical_objects = request.full_mechanical_objects + response_dict.get("new_mechanical_objects", []) From 2b262ed07cbbed81efe1f7cabf8adf901698a0d9 Mon Sep 17 00:00:00 2001 From: laurasgkadri98 <126669056+laurasgkadri98@users.noreply.github.com> Date: Tue, 28 Jan 2025 17:26:50 +0100 Subject: [PATCH 3/3] fix pre-commit errors --- src/allie/flowkit/endpoints/mechscriptbot.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/allie/flowkit/endpoints/mechscriptbot.py b/src/allie/flowkit/endpoints/mechscriptbot.py index 6064d19..4bdbd45 100644 --- a/src/allie/flowkit/endpoints/mechscriptbot.py +++ b/src/allie/flowkit/endpoints/mechscriptbot.py @@ -74,8 +74,7 @@ async def triggermechscriptbot(request: MechScriptBotRequest, api_key: str = Hea updated_ai_memory = request.full_ai_memory + [ai_new_mem] new_variables_list = [ - f"{var_name}:{var_type}" - for var_name, var_type in (response_dict.get("new_variables", {})).items() + f"{var_name}:{var_type}" for var_name, var_type in (response_dict.get("new_variables", {})).items() ] updated_variables = request.full_variables + new_variables_list