From 7aae7f8f15510bed19d4df5432852e37cc8d68ed Mon Sep 17 00:00:00 2001 From: rahulgs12 Date: Wed, 3 May 2023 11:51:01 -0400 Subject: [PATCH] clean up example --- jsonformer/main.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/jsonformer/main.py b/jsonformer/main.py index fed8189..7272393 100644 --- a/jsonformer/main.py +++ b/jsonformer/main.py @@ -145,7 +145,10 @@ class Jsonformer: return self.generate_array(schema["items"], new_array) elif schema_type == "object": new_obj = {} - obj[key if key else -1] = new_obj + if key: + obj[key] = new_obj + else: + obj.append(new_obj) return self.generate_object(schema["properties"], new_obj) else: raise ValueError(f"Unsupported schema type: {schema_type}") @@ -179,7 +182,7 @@ class Jsonformer: if gen_marker_index != -1: progress = progress[:gen_marker_index] else: - print("Failed to find generation marker") + raise ValueError("Failed to find generation marker") prompt = template.format( prompt=self.prompt,