Update __init__.py

promptslab · Jul 29, 2023 · 3d815c5 · 3d815c5
1 parent 07ab6c2
commit 3d815c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/promptify/pipelines/__init__.py b/promptify/pipelines/__init__.py
@@ -94,7 +94,7 @@ def _get_output_from_cache_or_model(self, template):
 
             if self.structured_output:
                 output = self.model.model_output(
-                    response, max_completion_length=self.json_depth_limit
+                    response, json_depth_limit=self.json_depth_limit
                 )
             else:
                 output = response