Return output of PythonAstREPLTool when falling back to exec() (#2780)

When the code ran by the PythonAstREPLTool contains multiple statements
it will fallback to exec() instead of using eval(). With this change, it
will also return the output of the code in the same way the
PythonREPLTool will.

tests/unit_tests/test_python.py

@@ -22,6 +22,17 @@ multiply()
 ```
 """
 
+_AST_SAMPLE_CODE_EXECUTE = """
+```
+def multiply(a, b):
+    return(5*6)
+a = 5
+b = 6
+
+multiply(a, b)
+```
+"""
+
 
 def test_python_repl() -> None:
     """Test functionality when globals/locals are not provided."""
@@ -80,6 +91,16 @@ def test_python_ast_repl_multiline() -> None:
     assert output == 30
 
 
+def test_python_ast_repl_multi_statement() -> None:
+    """Test correct functionality for ChatGPT multi statement commands."""
+    if sys.version_info < (3, 9):
+        pytest.skip("Python 3.9+ is required for this test")
+    tool = PythonAstREPLTool()
+    output = tool.run(_AST_SAMPLE_CODE_EXECUTE)
+    print(output)
+    assert output == 30
+
+
 def test_function() -> None:
     """Test correct functionality."""
     chain = PythonREPL()