lean-interact 0.3.0__tar.gz → 0.3.2__tar.gz

{lean_interact-0.3.0 → lean_interact-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lean-interact
-Version: 0.3.0
+Version: 0.3.2
 Summary: LeanInteract is a Python package that allows you to interact with the Lean theorem prover.
 Author-email: Auguste Poiroux <auguste.poiroux@epfl.ch>
 License: MIT License

{lean_interact-0.3.0 → lean_interact-0.3.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "lean-interact"
-version = "0.3.0"
+version = "0.3.2"
 description = "LeanInteract is a Python package that allows you to interact with the Lean theorem prover."
 keywords = ["Lean", "theorem proving", "autoformalization", "REPL"]
 license = { file = "LICENSE" }

{lean_interact-0.3.0 → lean_interact-0.3.2}/src/lean_interact/server.py

@@ -64,9 +64,10 @@ class LeanServer:
             echo=False,
             preexec_fn=lambda: _limit_memory(self.config.memory_hard_limit_mb),
         )
+        self._proc.delaybeforesend = None
         # `stty -icanon` is required to handle arbitrary long inputs in the Lean REPL
         self._proc.sendline("stty -icanon")
-        self._proc.sendline(f"lake env {self.config._cache_repl_dir}/.lake/build/bin/repl")
+        self._proc.sendline(f"lake env {self.config._cache_repl_dir}/.lake/build/bin/repl || exit")
 
     def is_alive(self) -> bool:
         return hasattr(self, "_proc") and self._proc is not None and self._proc.isalive()
@@ -131,8 +132,9 @@ class LeanServer:
             self.kill()
             raise ConnectionAbortedError(
                 "The Lean server closed unexpectedly. Possible reasons (not exhaustive):\n"
+                "- An uncaught exception in the Lean REPL (for example, an unexistent file has been requested)\n"
                 "- Not enough memory and/or compute available\n"
-                "- Your cached Lean REPL is corrupted. In this case, clear the cache"
+                "- The cached Lean REPL is corrupted. In this case, clear the cache"
                 " using the `clear_cache` (`from pyleanrepl import clear_cache`) method."
             ) from e
 

{lean_interact-0.3.0 → lean_interact-0.3.2}/tests/test_server.py

@@ -411,6 +411,18 @@ class TestLeanServer(unittest.TestCase):
         step2 = server.run(ProofStep(tactic="rfl", proof_state=step1.proof_state))
         self.assertEqual(step2, ProofStepResponse(proof_state=2, goals=[]))
 
+    def test_run_lots_of_commands(self):
+        server = LeanServer(LeanREPLConfig(verbose=True))
+
+        init_env = server.run(Command(cmd="#eval 1"))
+        assert isinstance(init_env, CommandResponse)
+        for i in range(1000):
+            cmd = Command(
+                cmd=f"theorem womp{i} (a{i} b c : Nat) : (a{i} + b) + c = c + a{i} + b := by sorry", env=init_env.env
+            )
+            result = server.run(cmd)
+            self.assertIsInstance(result, CommandResponse)
+
     def test_bug_increasing_memory(self):
         mem_limit = 512
         server = AutoLeanServer(config=LeanREPLConfig(memory_hard_limit_mb=mem_limit, verbose=True))
@@ -435,6 +447,8 @@ class TestLeanServer(unittest.TestCase):
                 result_queue.put(("success", result))
             except TimeoutError as e:
                 result_queue.put(("timeout", e))
+            except ConnectionAbortedError as e:
+                result_queue.put(("connection_aborted", e))  # out of memory
             except Exception as e:
                 result_queue.put(("error", e))
 
@@ -549,13 +563,11 @@ class TestLeanServer(unittest.TestCase):
         # Try to unpickle invalid data
         temp_pickle_file = tempfile.mkstemp(suffix=".olean")[1]
 
-        # TODO: Fix this in Lean REPL, it should crash instead of hanging
-
-        with self.assertRaises(TimeoutError):
-            server.run(UnpickleEnvironment(unpickle_env_from=temp_pickle_file), timeout=5)
-
-        with self.assertRaises(TimeoutError):
-            server.run(UnpickleProofState(unpickle_proof_state_from=temp_pickle_file), timeout=5)
+        # Try to unpickle invalid data
+        with self.assertRaises(ConnectionAbortedError):
+            server.run(UnpickleEnvironment(unpickle_env_from=temp_pickle_file))
+        with self.assertRaises(ConnectionAbortedError):
+            server.run(UnpickleProofState(unpickle_proof_state_from=temp_pickle_file))
 
         # delete the temp file
         os.remove(temp_pickle_file)

{lean_interact-0.3.0 → lean_interact-0.3.2}/uv.lock

@@ -351,7 +351,7 @@ wheels = [
 
 [[package]]
 name = "lean-interact"
-version = "0.3.0"
+version = "0.3.2"
 source = { editable = "." }
 dependencies = [
     { name = "gitpython" },