beswarm 0.2.99__py3-none-any.whl → 0.3.1__py3-none-any.whl

beswarm/agents/planact.py

@@ -164,9 +164,12 @@ class WorkerAgent(BaseAgent):
     def __init__(self, goal: str, tools_json: List, agent_config: Dict, work_dir: str, cache_messages: Union[bool, List[Dict]], broker: MessageBroker, listen_topic: str, publish_topic: str, status_topic: str):
         super().__init__(goal, tools_json, agent_config, work_dir, cache_messages, broker, listen_topic, publish_topic, status_topic)
 
-        if self.cache_messages and isinstance(self.cache_messages, list) and len(self.cache_messages) > 1:
-            first_user_message = replace_xml_content(self.cache_messages[1]["content"], "goal", goal)
-            self.config["cache_messages"] = self.cache_messages[0:1] + [{"role": "user", "content": first_user_message}] + self.cache_messages[2:]
+        if self.cache_messages and isinstance(self.cache_messages, Messages) and len(self.cache_messages) > 1:
+            if self.cache_messages.provider("goal"):
+                self.cache_messages.provider("goal").update(goal)
+            else:
+                self.cache_messages[1].insert(0, Goal(goal))
+            self.config["cache_messages"] = self.cache_messages
 
         self.agent = chatgpt(**self.config)
 

beswarm/cli.py

@@ -0,0 +1,68 @@
+import os
+import zipfile
+import argparse
+from tqdm import tqdm
+
+def zip_beswarm_folders(source_dir, zip_filename):
+    """
+    Traverses the source directory, finds all files within any .beswarm folder,
+    and adds them to a zip file, preserving the directory structure.
+
+    :param source_dir: The directory to search for .beswarm folders.
+    :param zip_filename: The name of the output zip file.
+    """
+    if not os.path.isdir(source_dir):
+        print(f"Error: Source directory '{source_dir}' not found.")
+        return
+
+    source_dir = os.path.abspath(source_dir)
+
+    files_to_add = []
+    # Walk the entire directory tree to find all files.
+    for root, dirs, files in os.walk(source_dir):
+        for file in files:
+            file_path = os.path.join(root, file)
+            # A file should be added if its path contains a '.beswarm' directory component.
+            if '.beswarm' in file_path.split(os.sep):
+                files_to_add.append(file_path)
+
+    if not files_to_add:
+        print("No files found within any .beswarm directories.")
+        return
+
+    print(f"Found {len(files_to_add)} files in .beswarm directories to zip.")
+
+    # Create the zip file and add all the collected files.
+    with zipfile.ZipFile(zip_filename, 'w', zipfile.ZIP_DEFLATED) as zipf:
+        for file_path in tqdm(files_to_add, desc="Zipping files", unit="file"):
+            # arcname is the path inside the zip file.
+            arcname = os.path.relpath(file_path, source_dir)
+            zipf.write(file_path, arcname)
+
+    print(f"\nSuccessfully created '{zip_filename}'")
+
+def main():
+    parser = argparse.ArgumentParser(description="Beswarm main command line interface.")
+    subparsers = parser.add_subparsers(dest='command', help='Available commands', required=True)
+
+    # Create the parser for the "debug" command
+    parser_debug = subparsers.add_parser('debug', help='Package all .beswarm folders from a directory into a zip file.')
+    parser_debug.add_argument(
+        "source_directory",
+        help="The source directory to search for .beswarm folders."
+    )
+    parser_debug.add_argument(
+        "-o", "--output",
+        default="beswarm_debug.zip",
+        help="The name of the output zip file (default: beswarm_debug.zip)."
+    )
+
+    args = parser.parse_args()
+
+    if args.command == 'debug':
+        zip_beswarm_folders(args.source_directory, args.output)
+    else:
+        parser.print_help()
+
+if __name__ == "__main__":
+    main()

{beswarm-0.2.99.dist-info → beswarm-0.3.1.dist-info}/METADATA

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

{beswarm-0.2.99.dist-info → beswarm-0.3.1.dist-info}/RECORD

@@ -1,12 +1,13 @@
 beswarm/__init__.py,sha256=HZjUOJtZR5QhMuDbq-wukQQn1VrBusNWai_ysGo-VVI,20
 beswarm/broker.py,sha256=64Y-djrKYaZfBQ8obwHOmr921QgZeu9BtScZWaYLfDo,9887
+beswarm/cli.py,sha256=m8fcEYznUq--iLULB6nGNecUGeT-wuuSRqiooINYDpY,2472
 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/agents/chatgroup.py,sha256=PzrmRcDKAbB7cxL16nMod_CzPosDV6bfTmXxQVuv-AQ,12012
-beswarm/agents/planact.py,sha256=OjxnYNu-LkxF2zIf7o75QB25d-yVtD4-FgDllTV_nZM,18456
+beswarm/agents/planact.py,sha256=QuhdpF3mVABdZ_IgXVGiMRHtAnDG4bHRMj8nTMjgq6g,18473
 beswarm/aient/aient/__init__.py,sha256=SRfF7oDVlOOAi6nGKiJIUK6B_arqYLO9iSMp-2IZZps,21
 beswarm/aient/aient/architext/architext/__init__.py,sha256=79Ih1151rfcqZdr7F8HSZSTs_iT2SKd1xCkehMsXeXs,19
 beswarm/aient/aient/architext/architext/core.py,sha256=_o5gI5XF6e4kQo4L4x8Q1Tb7C1U0GANm24gJmvAgoXE,30546
@@ -122,7 +123,8 @@ beswarm/tools/search_web.py,sha256=0fTeczXiOX_LJQGaLEGbuJtIPzofeuquGWEt3yDMtVw,1
 beswarm/tools/subtasks.py,sha256=NHDnmUhUPgDQKBACnpgErpFJRcsH0w_Q9VsyQjNvNHA,12658
 beswarm/tools/worker.py,sha256=mQ1qdrQ8MgL99byAbTvxfEByFFGN9mty3UHqHjARMQ8,2331
 beswarm/tools/write_csv.py,sha256=u0Hq18Ksfheb52MVtyLNCnSDHibITpsYBPs2ub7USYA,1466
-beswarm-0.2.99.dist-info/METADATA,sha256=sVKJHsH5_LbfXDOEtrGcrKTpa2NQEAMyTh3ZA84ffCI,3878
-beswarm-0.2.99.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-beswarm-0.2.99.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
-beswarm-0.2.99.dist-info/RECORD,,
+beswarm-0.3.1.dist-info/METADATA,sha256=gtgcjZIH2wvnqO27bdwVPiXesgsGupjr1wrArKzPOeY,3877
+beswarm-0.3.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+beswarm-0.3.1.dist-info/entry_points.txt,sha256=URK7Y4PDzBgxIecQnxsWTu4O-eaFa1CoAcNTWh5R7LM,45
+beswarm-0.3.1.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
+beswarm-0.3.1.dist-info/RECORD,,

beswarm-0.3.1.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+beswarm = beswarm.cli:main