Code agent: allow function persistence between steps (#31769)

* Code agent: allow function persistence between steps
2024-07-05 11:09:11 +02:00
parent eef0507f3d
commit 1556025271
5 changed files with 63 additions and 11 deletions
--- a/tests/agents/test_python_interpreter.py
+++ b/tests/agents/test_python_interpreter.py
@@ -660,7 +660,6 @@ add_one(1, 1)
 """
        state = {}
        result = evaluate_python_code(code, {"print": print, "range": range, "ord": ord, "chr": chr}, state=state)
-        print(state)
        assert result == 2

        # test returning None
@@ -672,5 +671,4 @@ returns_none(1)
 """
        state = {}
        result = evaluate_python_code(code, {"print": print, "range": range, "ord": ord, "chr": chr}, state=state)
-        print(state)
        assert result is None