beswarm 0.3.11__py3-none-any.whl → 0.3.12__py3-none-any.whl

beswarm/aient/aient/core/request.py

@@ -295,7 +295,7 @@ async def get_gemini_payload(request, engine, provider, api_key=None):
             if key == request.model:
                 for k, v in value.items():
                     payload[k] = v
-            elif all(_model not in request.model.lower() for _model in ["gemini", "gpt", "claude", "deepseek"]) and "-" not in key:
+            elif all(_model not in request.model.lower() for _model in model_dict.keys()) and "-" not in key:
                 payload[key] = value
 
     return url, headers, payload
@@ -591,7 +591,7 @@ async def get_vertex_gemini_payload(request, engine, provider, api_key=None):
             if key == request.model:
                 for k, v in value.items():
                     payload[k] = v
-            elif all(_model not in request.model.lower() for _model in ["gemini", "gpt", "claude"]):
+            elif all(_model not in request.model.lower() for _model in model_dict.keys()) and "-" not in key:
                 payload[key] = value
 
     return url, headers, payload
@@ -1149,7 +1149,7 @@ async def get_gpt_payload(request, engine, provider, api_key=None):
             if key == request.model:
                 for k, v in value.items():
                     payload[k] = v
-            elif all(_model not in request.model.lower() for _model in ["gemini", "gpt", "claude"]):
+            elif all(_model not in request.model.lower() for _model in model_dict.keys()) and "-" not in key:
                 payload[key] = value
 
     return url, headers, payload
@@ -1247,7 +1247,7 @@ async def get_azure_payload(request, engine, provider, api_key=None):
             if key == request.model:
                 for k, v in value.items():
                     payload[k] = v
-            elif all(_model not in request.model.lower() for _model in ["gemini", "gpt", "claude"]):
+            elif all(_model not in request.model.lower() for _model in model_dict.keys()) and "-" not in key:
                 payload[key] = value
 
     return url, headers, payload
@@ -1367,7 +1367,7 @@ async def get_azure_databricks_payload(request, engine, provider, api_key=None):
             if key == request.model:
                 for k, v in value.items():
                     payload[k] = v
-            elif all(_model not in request.model.lower() for _model in ["gemini", "gpt", "claude"]):
+            elif all(_model not in request.model.lower() for _model in model_dict.keys()) and "-" not in key:
                 payload[key] = value
 
     return url, headers, payload
@@ -1454,7 +1454,7 @@ async def get_openrouter_payload(request, engine, provider, api_key=None):
             if key == request.model:
                 for k, v in value.items():
                     payload[k] = v
-            elif all(_model not in request.model.lower() for _model in ["gemini", "gpt", "claude"]):
+            elif all(_model not in request.model.lower() for _model in model_dict.keys()) and "-" not in key:
                 payload[key] = value
 
     return url, headers, payload
@@ -1820,7 +1820,7 @@ async def get_claude_payload(request, engine, provider, api_key=None):
             if key == request.model:
                 for k, v in value.items():
                     payload[k] = v
-            elif all(_model not in request.model.lower() for _model in ["gemini", "gpt", "claude"]):
+            elif all(_model not in request.model.lower() for _model in model_dict.keys()) and "-" not in key:
                 payload[key] = value
 
     return url, headers, payload

beswarm/utils.py

@@ -2,6 +2,9 @@ import re
 import sys
 import logging
 from pathlib import Path
+import queue
+import atexit
+from logging.handlers import QueueHandler, QueueListener
 
 def extract_xml_content(text, xml_tag):
     result = ''
@@ -167,9 +170,17 @@ async def {tool_name}({params_str}):
             print(f"Failed to create or register function for tool '{tool_name}': {e}")
             traceback.print_exc()
 
+# For asynchronous logging
+_listeners = []
+_atexit_registered = False
+
+def _stop_all_listeners():
+    for listener in _listeners:
+        listener.stop()
+
 def setup_logger(logger_name: str, log_file: Path):
     """
-    用最简单的方式设置一个 logger，使其同时输出到文件和终端。
+    用异步方式设置一个 logger，使其同时输出到文件和终端，避免IO阻塞。
 
     Args:
         logger_name (str): Logger 的唯一名称。
@@ -178,35 +189,52 @@ def setup_logger(logger_name: str, log_file: Path):
     Returns:
         logging.Logger: 配置好的 logger 实例。
     """
+    global _atexit_registered
+
     # 1. 获取 Logger 实例
     logger = logging.getLogger(logger_name)
     logger.setLevel(logging.INFO) # 设置 logger 的最低处理级别为 INFO
     logger.propagate = False
 
+    # 如果已经配置了异步 handler，直接返回
+    if any(isinstance(h, QueueHandler) for h in logger.handlers):
+        return logger
+
+    # 如果有其他 handler，说明被其他方式配置过，也直接返回
     if logger.hasHandlers():
         return logger
 
-    # 2. 创建文件 Handler
+    # 2. 创建用于 Listener 的 Handler（文件和终端）
     # 确保日志文件所在的目录存在
     log_file.parent.mkdir(parents=True, exist_ok=True)
     file_handler = logging.FileHandler(log_file, mode='a', encoding='utf-8')
-
-    # 3. 创建终端 Handler
-    # 使用 sys.stdout 可以确保输出在标准输出流，与 print() 行为一致
     stream_handler = logging.StreamHandler(sys.stdout)
 
-    # 4. 创建一个通用的 Formatter
+    # 3. 创建一个通用的 Formatter
     formatter = logging.Formatter('%(asctime)s - %(name)s - [%(levelname)s] - %(message)s')
 
-    # 5. 为两个 Handler 设置 Formatter
+    # 4. 为两个 Handler 设置 Formatter
     file_handler.setFormatter(formatter)
     stream_handler.setFormatter(formatter)
 
-    # 6. 将两个 Handler 添加到 Logger
-    # DEBUG = os.getenv("DEBUG", "false").lower() in ("true", "1", "t", "yes")
-    logger.addHandler(file_handler)
-    logger.addHandler(stream_handler)
-    logger.propagate = False
+    # 5. 设置 Queue、QueueHandler 和 QueueListener
+    log_queue = queue.Queue(-1)
+    # listener 在后台线程中从 queue 读取日志，并分发给 file_handler 和 stream_handler
+    listener = QueueListener(log_queue, file_handler, stream_handler, respect_handler_level=True)
+
+    # queue_handler 是一个非阻塞的 handler，它把日志消息放到 queue 中
+    queue_handler = QueueHandler(log_queue)
+
+    # 6. 将 QueueHandler 添加到 Logger
+    logger.addHandler(queue_handler)
+
+    # 7. 启动 listener 并注册 atexit 钩子以确保程序退出时停止
+    _listeners.append(listener)
+    listener.start()
+
+    if not _atexit_registered:
+        atexit.register(_stop_all_listeners)
+        _atexit_registered = True
 
     return logger
 

{beswarm-0.3.11.dist-info → beswarm-0.3.12.dist-info}/METADATA

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

{beswarm-0.3.11.dist-info → beswarm-0.3.12.dist-info}/RECORD

@@ -5,7 +5,7 @@ beswarm/core.py,sha256=jKStpTTOu6Ojond_i-okTZLrvSAJ4yUoTZwtDfFTiRs,553
 beswarm/knowledge_graph.py,sha256=oiOMknAJzGrOHc2AyQgvrCcZAkGLhFnsnvSBdfFBWMw,14831
 beswarm/prompt.py,sha256=45onnyoY9plKM86KQefbPw5z9QJMn-mVnjlFQZcrjz0,34373
 beswarm/taskmanager.py,sha256=2Uz_cthW9rWkQMJdzgsXAMlfN8Ni2Qj_DOq_L-p6XZc,12662
-beswarm/utils.py,sha256=0J-b38P5QGT-A_38co7FjzaUNJykaskI7mbbcQ4w_68,8215
+beswarm/utils.py,sha256=mtWzksvdar7dR8oRGvFBEAFXjigk4m1MD1arrx9wEuU,9157
 beswarm/agents/chatgroup.py,sha256=PzrmRcDKAbB7cxL16nMod_CzPosDV6bfTmXxQVuv-AQ,12012
 beswarm/agents/planact.py,sha256=dhPkMUaZLc1dO0TrI1Orr_i60iLzJHQIYBI_QjjSJwI,18723
 beswarm/aient/aient/__init__.py,sha256=SRfF7oDVlOOAi6nGKiJIUK6B_arqYLO9iSMp-2IZZps,21
@@ -17,7 +17,7 @@ beswarm/aient/aient/architext/test/test_save_load.py,sha256=o8DqH6gDYZkFkQy-a7bl
 beswarm/aient/aient/core/__init__.py,sha256=NxjebTlku35S4Dzr16rdSqSTWUvvwEeACe8KvHJnjPg,34
 beswarm/aient/aient/core/log_config.py,sha256=kz2_yJv1p-o3lUQOwA3qh-LSc3wMHv13iCQclw44W9c,274
 beswarm/aient/aient/core/models.py,sha256=KMlCRLjtq1wQHZTJGqnbWhPS2cHq6eLdnk7peKDrzR8,7490
-beswarm/aient/aient/core/request.py,sha256=u4ebaBDJTJbi-JfdXbU1Z_HkmDOk40WQH65aqTeg7Pw,77804
+beswarm/aient/aient/core/request.py,sha256=u9wkesp0JMQoJdLoDCNQQgiAB7a_W4Hs38uX6Ppqpi8,77836
 beswarm/aient/aient/core/response.py,sha256=VYpXfF6RO3Y-fTZMGV2p-bcrd73BPAKlz33gQkOcqjE,38462
 beswarm/aient/aient/core/utils.py,sha256=9T6Ze9sMnsX4NBWeYCgY3AlZdhh6HFV1LI5SojzZars,31751
 beswarm/aient/aient/core/test/test_base_api.py,sha256=pWnycRJbuPSXKKU9AQjWrMAX1wiLC_014Qc9hh5C2Pw,524
@@ -121,8 +121,8 @@ beswarm/tools/subtasks.py,sha256=Fsf_542CkECcwFwxD-F38_IUsmk06xe1P7e6pBPTy4Y,128
 beswarm/tools/worker.py,sha256=mQ1qdrQ8MgL99byAbTvxfEByFFGN9mty3UHqHjARMQ8,2331
 beswarm/tools/write_csv.py,sha256=u0Hq18Ksfheb52MVtyLNCnSDHibITpsYBPs2ub7USYA,1466
 beswarm/tools/write_file.py,sha256=L2coBqz-aRFxPBvJBrbWbUJhu7p3oKAAGb9R144bFtk,4926
-beswarm-0.3.11.dist-info/METADATA,sha256=Lg7pycqsatSZLBeuoHVbeiQ6gMthrI87seEw3RLaOtk,3878
-beswarm-0.3.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-beswarm-0.3.11.dist-info/entry_points.txt,sha256=URK7Y4PDzBgxIecQnxsWTu4O-eaFa1CoAcNTWh5R7LM,45
-beswarm-0.3.11.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
-beswarm-0.3.11.dist-info/RECORD,,
+beswarm-0.3.12.dist-info/METADATA,sha256=lsjwHlJwRdbZPGPqsihISy5xKb1Rim2bsiTcQaYJXVo,3878
+beswarm-0.3.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+beswarm-0.3.12.dist-info/entry_points.txt,sha256=URK7Y4PDzBgxIecQnxsWTu4O-eaFa1CoAcNTWh5R7LM,45
+beswarm-0.3.12.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
+beswarm-0.3.12.dist-info/RECORD,,