From 210d69a317118ecb8f4450e0ba347080769acaef Mon Sep 17 00:00:00 2001 From: Ben Wiener Date: Wed, 24 May 2023 15:00:01 -0400 Subject: [PATCH] Add iteration tracking to generate_number --- jsonformer/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jsonformer/main.py b/jsonformer/main.py index dd867d4..9c13471 100644 --- a/jsonformer/main.py +++ b/jsonformer/main.py @@ -80,7 +80,7 @@ class Jsonformer: if iterations > 3: raise ValueError("Failed to generate a valid number") - return self.generate_number(temperature=self.temperature * 1.3) + return self.generate_number(temperature=self.temperature * 1.3, iterations=iterations+1) def generate_boolean(self) -> bool: prompt = self.get_prompt()