open-swarm 0.1.1745125895__tar.gz → 0.1.1745125904__tar.gz

{open_swarm-0.1.1745125895 → open_swarm-0.1.1745125904}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: open-swarm
-Version: 0.1.1745125895
+Version: 0.1.1745125904
 Summary: Open Swarm: Orchestrating AI Agent Swarms with Django
 Project-URL: Homepage, https://github.com/yourusername/open-swarm
 Project-URL: Documentation, https://github.com/yourusername/open-swarm/blob/main/README.md

{open_swarm-0.1.1745125895 → open_swarm-0.1.1745125904}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "open-swarm"
-version = "0.1.1745125895"
+version = "0.1.1745125904"
 description = "Open Swarm: Orchestrating AI Agent Swarms with Django"
 readme = "README.md"
 requires-python = ">=3.10"

open_swarm-0.1.1745125904/src/swarm/utils/ansi_box.py

@@ -0,0 +1,34 @@
+import shutil
+
+ANSI_COLORS = {
+    'cyan': '\033[96m',
+    'green': '\033[92m',
+    'yellow': '\033[93m',
+    'magenta': '\033[95m',
+    'blue': '\033[94m',
+    'red': '\033[91m',
+    'white': '\033[97m',
+    'grey': '\033[90m',
+    'reset': '\033[0m',
+}
+
+
+def ansi_box(text, color='cyan', emoji='🤖', width=None):
+    """
+    Draw a fancy ANSI box around the given text, with color and emoji.
+    """
+    lines = [line.rstrip() for line in text.strip('\n').split('\n')]
+    max_len = max(len(line) for line in lines)
+    if width is None:
+        try:
+            width = min(shutil.get_terminal_size((80, 20)).columns, max_len + 6)
+        except Exception:
+            width = max_len + 6
+    box_width = max(width, max_len + 6)
+    color_code = ANSI_COLORS.get(color, ANSI_COLORS['cyan'])
+    reset = ANSI_COLORS['reset']
+    top = f"{color_code}╔{'═' * (box_width-2)}╗{reset}"
+    title = f"{color_code}║ {emoji} {' ' * (box_width-6)}║{reset}"
+    content = [f"{color_code}║ {line.ljust(box_width-4)} ║{reset}" for line in lines]
+    bottom = f"{color_code}╚{'═' * (box_width-2)}╝{reset}"
+    return '\n'.join([top, title] + content + [bottom])

open_swarm-0.1.1745125904/src/swarm/ux/ansi_box.py

@@ -0,0 +1,43 @@
+import sys
+
+def ansi_box(title, content, count=None, params=None, style='default', emoji=None):
+    """
+    Print a visually distinct ANSI box summarizing search/analysis results.
+    - title: e.g. 'Searched filesystem', 'Analyzed code'
+    - content: main result string or list of strings
+    - count: result count (optional)
+    - params: dict of search parameters (optional)
+    - style: 'default', 'success', 'warning', etc.
+    - emoji: optional emoji prefix
+    """
+    border = {
+        'default': '━',
+        'success': '━',
+        'warning': '━',
+    }.get(style, '━')
+    color = {
+        'default': '\033[36m',  # Cyan
+        'success': '\033[32m',  # Green
+        'warning': '\033[33m',  # Yellow
+    }.get(style, '\033[36m')
+    reset = '\033[0m'
+    box_width = 80
+    lines = []
+    head = f"{emoji + ' ' if emoji else ''}{title}"
+    if count is not None:
+        head += f" | Results: {count}"
+    if params:
+        head += f" | Params: {params}"
+    lines.append(color + border * box_width + reset)
+    lines.append(color + f"{head[:box_width]:^{box_width}}" + reset)
+    lines.append(color + border * box_width + reset)
+    if isinstance(content, str):
+        content = [content]
+    for line in content:
+        for l in str(line).split('\n'):
+            lines.append(f"{l[:box_width]:<{box_width}}")
+    lines.append(color + border * box_width + reset)
+    print("\n".join(lines))
+
+# Example usage:
+# ansi_box('Searched filesystem', 'Found 12 files', count=12, params={'pattern': '*.py'}, style='success', emoji='💾')

open_swarm-0.1.1745125904/src/swarm/ux/spinner.py

@@ -0,0 +1,53 @@
+import sys
+import threading
+import time
+
+class Spinner:
+    """
+    Displays spinner states: Generating., Generating.., Generating..., Running...,
+    and switches to 'Taking longer than expected' after a timeout.
+    """
+    def __init__(self, base_message="Generating", long_wait_timeout=8):
+        self.base_message = base_message
+        self.states = [".", "..", "...", "..", "."]
+        self.running = False
+        self.thread = None
+        self.long_wait_timeout = long_wait_timeout
+        self._long_wait = False
+
+    def start(self):
+        self.running = True
+        self.thread = threading.Thread(target=self._spin, daemon=True)
+        self.thread.start()
+
+    def stop(self):
+        self.running = False
+        if self.thread:
+            self.thread.join()
+        sys.stdout.write("\r" + " " * 80 + "\r")
+        sys.stdout.flush()
+
+    def _spin(self):
+        idx = 0
+        start_time = time.time()
+        while self.running:
+            if not self._long_wait and (time.time() - start_time > self.long_wait_timeout):
+                self._long_wait = True
+            if self._long_wait:
+                msg = f"{self.base_message}... Taking longer than expected"
+            else:
+                msg = f"{self.base_message}{self.states[idx % len(self.states)]}"
+            sys.stdout.write(f"\r{msg}")
+            sys.stdout.flush()
+            time.sleep(0.4)
+            idx += 1
+
+    def set_message(self, message):
+        self.base_message = message
+        self._long_wait = False
+
+# Example usage:
+# spinner = Spinner()
+# spinner.start()
+# ... do work ...
+# spinner.stop()