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]]: