fix: Python执行

yybht155 · Oct 25, 2024 · ff689b0 · ff689b0
1 parent 82886de
commit ff689b0
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/api/core/helper/code_executor/code_executor.py b/api/core/helper/code_executor/code_executor.py
@@ -117,3 +117,13 @@ def execute_workflow_code_template(cls, language: CodeLanguage, code: str, input
             raise e
 
         return template_transformer.transform_response(response)
+
+class Capturing(list):
+    def __enter__(self):
+        self._stdout = sys.stdout
+        sys.stdout = self._stringio = StringIO()
+        return self
+    def __exit__(self, *args):
+        self.extend(self._stringio.getvalue().splitlines())
+        del self._stringio    # free up some memory
+        sys.stdout = self._stdout