lean-interact 0.3.1__tar.gz → 0.3.3__tar.gz

{lean_interact-0.3.1 → lean_interact-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lean-interact
-Version: 0.3.1
+Version: 0.3.3
 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
@@ -43,7 +43,7 @@ Description-Content-Type: text/markdown
 
 - **🔗 Interactivity**: Execute Lean code and files directly from Python.
 - **🚀 Ease of Use**: LeanInteract abstracts the complexities of Lean setup and interaction.
-- **🔧 Compatibility**: Supports all Lean versions between `v4.7.0-rc1` and `v4.18.0`.
+- **🔧 Compatibility**: Supports all Lean versions between `v4.7.0-rc1` and `v4.19.0-rc2`.
 - **📦 Temporary Projects**: Easily instantiate temporary Lean environments.
   - Useful for experimenting with benchmarks depending on [Mathlib](https://github.com/leanprover-community/mathlib4) like [ProofNet#](https://huggingface.co/datasets/PAug/ProofNetSharp) and [MiniF2F](https://github.com/yangky11/miniF2F-lean4).
 

{lean_interact-0.3.1 → lean_interact-0.3.3}/README.md

@@ -6,7 +6,7 @@
 
 - **🔗 Interactivity**: Execute Lean code and files directly from Python.
 - **🚀 Ease of Use**: LeanInteract abstracts the complexities of Lean setup and interaction.
-- **🔧 Compatibility**: Supports all Lean versions between `v4.7.0-rc1` and `v4.18.0`.
+- **🔧 Compatibility**: Supports all Lean versions between `v4.7.0-rc1` and `v4.19.0-rc2`.
 - **📦 Temporary Projects**: Easily instantiate temporary Lean environments.
   - Useful for experimenting with benchmarks depending on [Mathlib](https://github.com/leanprover-community/mathlib4) like [ProofNet#](https://huggingface.co/datasets/PAug/ProofNetSharp) and [MiniF2F](https://github.com/yangky11/miniF2F-lean4).
 

{lean_interact-0.3.1 → lean_interact-0.3.3}/examples/beq_plus.py

@@ -2,7 +2,7 @@
 # requires-python = ">=3.10"
 # dependencies = [
 #     "datasets",
-#     "lean-interact @ file:///home/poiroux/Documents/EPFL/PhD/Lean/LeanInteract",
+#     "lean-interact",
 #     "rich",
 #     "tqdm",
 #     "scikit-learn",

{lean_interact-0.3.1 → lean_interact-0.3.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "lean-interact"
-version = "0.3.1"
+version = "0.3.3"
 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.1 → lean_interact-0.3.3}/src/lean_interact/server.py

@@ -64,6 +64,7 @@ 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 || exit")
@@ -89,17 +90,17 @@ class LeanServer:
         assert self._proc is not None
         if verbose:
             logger.info("Sending query: %s", json_query)
-        self._proc.sendline(json_query)
-        self._proc.sendline()
-        _ = self._proc.expect_exact("\r\n\r\n", timeout=timeout)
+        self._proc.send(json_query + "\n\n")
+        self._proc.expect_exact("\r\n\r\n", timeout=timeout)
         return self._proc.before or ""
 
     def _parse_repl_output(self, raw_output: str, verbose: bool) -> dict:
         """Clean up raw REPL output and parse JSON response."""
+        if verbose:
+            logger.info("Server raw output: `%s", raw_output)
         output = raw_output.replace("\r\n", "\n")
         output = output[output.find('{"') :] if '{"' in output else ""
         if verbose:
-            logger.info("Server raw output: `%s", raw_output)
             logger.info("Server cleaned output: `%s", output)
         try:
             return json.loads(output)

{lean_interact-0.3.1 → lean_interact-0.3.3}/src/lean_interact/utils.py

@@ -21,7 +21,7 @@ logger.addHandler(handler)
 ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
 DEFAULT_CACHE_DIR = os.path.join(ROOT_DIR, "cache")
 DEFAULT_REPL_GIT_URL = "https://github.com/augustepoiroux/repl"
-DEFAULT_REPL_VERSION = "v1.0.4"
+DEFAULT_REPL_VERSION = "v1.0.5"
 
 os.makedirs(DEFAULT_CACHE_DIR, exist_ok=True)
 

{lean_interact-0.3.1 → lean_interact-0.3.3}/tests/test_server.py

@@ -199,13 +199,6 @@ class TestLeanServer(unittest.TestCase):
         result = server.run(ProofStep(tactic="rfl", proof_state=0))
         self.assertEqual(result, ProofStepResponse(proof_state=1, goals=[]))
 
-    def test_run_multiple_commands(self):
-        server = AutoLeanServer(config=LeanREPLConfig(memory_hard_limit_mb=4096, verbose=True))
-
-        for i in range(100):
-            cmd = Command(cmd=f"theorem womp{i} (a{i} b c : Nat) : (a{i} + b) + c = c + a{i} + b := by sorry")
-            server.run(cmd)
-
     def test_lean_version(self):
         server = AutoLeanServer(config=LeanREPLConfig(lean_version="v4.14.0", verbose=True))
         result = server.run(Command(cmd="#eval Lean.versionString"))
@@ -411,6 +404,28 @@ 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_multiple_commands(self):
+        # Test this issue: https://github.com/leanprover-community/repl/issues/77
+        server = AutoLeanServer(config=LeanREPLConfig(memory_hard_limit_mb=4096, verbose=True))
+
+        with self.assertRaises(ConnectionAbortedError):
+            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")
+                server.run(cmd)
+
+    def test_run_lots_of_commands(self):
+        # Test this issue: https://github.com/leanprover-community/repl/issues/77
+        server = LeanServer(LeanREPLConfig(verbose=True))
+
+        init_env = server.run(Command(cmd="#eval 1"))
+        assert not isinstance(init_env, LeanError)
+        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 +450,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))
 

{lean_interact-0.3.1 → lean_interact-0.3.3}/uv.lock

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