updated helper to use json_schema

Will Kukkamalla · Will Kukkamalla · commit 93dbd4b1e150 · 2025-07-09T09:34:36.000-04:00
diff --git a/tests/test_helper.py b/tests/test_helper.py
@@ -73,6 +73,7 @@ def fake_generate_response(
         test_output: Optional[Path] = None,
         scope: Optional[str] = None,
         llama_mode: Optional[str] = None,
+        json_schema: Optional[str] = None,
     ):
         all_prompts.append((test_name, "OpenAIModel.generate_response", prompt))
         return prompt, f"[MOCKED RESPONSE] \n {prompt}"
