beswarm 0.2.13__py3-none-any.whl → 0.2.14__py3-none-any.whl

beswarm/tools/worker.py

@@ -1,5 +1,6 @@
 import os
 import re
+import sys
 import copy
 import json
 import difflib
@@ -7,6 +8,19 @@ import platform
 from pathlib import Path
 from datetime import datetime
 
+class Tee:
+    def __init__(self, *files):
+        self.files = files
+
+    def write(self, obj):
+        for f in self.files:
+            f.write(obj)
+            f.flush()
+
+    def flush(self):
+        for f in self.files:
+            f.flush()
+
 from ..aient.src.aient.models import chatgpt
 from ..aient.src.aient.plugins import register_tool, get_function_call_list, registry
 from ..prompt import worker_system_prompt, instruction_system_prompt
@@ -34,11 +48,12 @@ async def worker(goal, tools, work_dir, cache_messages=None):
     """
     DEBUG = os.getenv("DEBUG", "false").lower() in ("true", "1", "t", "yes")
     if DEBUG:
-        import sys
         log_file = open(Path(work_dir) / ".beswarm" / "history.log", "a", encoding="utf-8")
         log_file.write(f"========== {datetime.now().strftime('%Y-%m-%d %H:%M:%S')} ==========\n")
-        sys.stdout = log_file
-        sys.stderr = log_file
+        original_stdout = sys.stdout
+        original_stderr = sys.stderr
+        sys.stdout = Tee(original_stdout, log_file)
+        sys.stderr = Tee(original_stderr, log_file)
 
     start_time = datetime.now()
     os.chdir(Path(work_dir).absolute())
@@ -235,11 +250,12 @@ async def worker(goal, tools, work_dir, cache_messages=None):
 async def worker_gen(goal, tools, work_dir, cache_messages=None):
     DEBUG = os.getenv("DEBUG", "false").lower() in ("true", "1", "t", "yes")
     if DEBUG:
-        import sys
         log_file = open(Path(work_dir) / ".beswarm" / "history.log", "a", encoding="utf-8")
         log_file.write(f"========== {datetime.now().strftime('%Y-%m-%d %H:%M:%S')} ==========\n")
-        sys.stdout = log_file
-        sys.stderr = log_file
+        original_stdout = sys.stdout
+        original_stderr = sys.stderr
+        sys.stdout = Tee(original_stdout, log_file)
+        sys.stderr = Tee(original_stderr, log_file)
 
     start_time = datetime.now()
     os.chdir(Path(work_dir).absolute())

{beswarm-0.2.13.dist-info → beswarm-0.2.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beswarm
-Version: 0.2.13
+Version: 0.2.14
 Summary: MAS
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{beswarm-0.2.13.dist-info → beswarm-0.2.14.dist-info}/RECORD

@@ -134,8 +134,8 @@ beswarm/tools/request_input.py,sha256=gXNAJPOJektMqxJVyzNTFOeMQ7xUkO-wWMYH-r2Rdw
 beswarm/tools/screenshot.py,sha256=u6t8FCgW5YHJ_Oc4coo8e0F3wTusWE_-H8dFh1rBq9Q,1011
 beswarm/tools/search_arxiv.py,sha256=GpuIOYX8T0iRC-X-hmuR9AUJVn15WWZq864DaoC7BUc,8004
 beswarm/tools/search_web.py,sha256=w0T0aCqOVlb6Of5hn_TtpnrGXo6bMtw2aKZdkrYjycI,12069
-beswarm/tools/worker.py,sha256=Lahfv7XkayjdxtzsK3jSjhfVHHjkOhubR3gJhHBq6Mc,23035
-beswarm-0.2.13.dist-info/METADATA,sha256=QV1OaMpF4nCTEIcl5eRJ2dD1GJCd27rUJru-eGGeklg,3847
-beswarm-0.2.13.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-beswarm-0.2.13.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
-beswarm-0.2.13.dist-info/RECORD,,
+beswarm/tools/worker.py,sha256=ZFgSe17Je9NJuITvojZ89uEk2p-pXWJIy9GHNvMDAVo,23491
+beswarm-0.2.14.dist-info/METADATA,sha256=EDT-5iFXFjVPqxMYNwTTvF7Ob0yvMarAd5zPKTpaPlc,3847
+beswarm-0.2.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+beswarm-0.2.14.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
+beswarm-0.2.14.dist-info/RECORD,,