From 21e3c397b40e5f71620159cedec094679a07b9a8 Mon Sep 17 00:00:00 2001 From: escesare Date: Fri, 28 Apr 2023 17:12:07 -0400 Subject: [PATCH] Use correct keys for output parsing except block --- promptify/parser/parser.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/promptify/parser/parser.py b/promptify/parser/parser.py index 2af9c91..f42e02d 100644 --- a/promptify/parser/parser.py +++ b/promptify/parser/parser.py @@ -234,14 +234,14 @@ def fit(self, json_str: str, max_completion_length: int = 5) -> Dict[str, Any]: ) return { "status": "completed", - "object_type": type(output["completion"]), - "data": output, + "object_type": type(output), + "data": {"completion": output, "suggestions": []}, } except Exception as e: return { "status": "failed", "object_type": None, - "data": {"error_message": str(e)}, + "data": {"completion": str(e)}, } def find_max_length(self, data_list: List[Any]) -> Dict[str, List[Any]]: