clean up

Author: iamNarcisse

tests/test_prompt_engine.py

@@ -14,9 +14,7 @@ class TestPromptEngine(unittest.TestCase):
 
     def test_get_prompt_engine_response_success(self) -> None:
         try:
-            result = jigsaw.prompt_engine.run(
-                {"id": "b08921b8-0b30-409e-8257-06fa1620c7e6", "stream": True}
-            )
+            result = jigsaw.prompt_engine.get("b08921b8-0b30-409e-8257-06fa1620c7e6")
             assert result["success"] == True
         except JigsawStackError as e:
             assert e.message == "Failed to parse API response. Please try again."