memstack-skill-loader 4.1.1__tar.gz → 4.3.0__tar.gz

{memstack_skill_loader-4.1.1/src/memstack_skill_loader.egg-info → memstack_skill_loader-4.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: memstack-skill-loader
-Version: 4.1.1
+Version: 4.3.0
 Summary: MCP server that vector-indexes MemStack Pro skills for on-demand loading
 Requires-Python: >=3.10
 Requires-Dist: mcp>=1.0.0
@@ -8,3 +8,4 @@ Requires-Dist: lancedb>=0.6.0
 Requires-Dist: sentence-transformers>=2.2.0
 Requires-Dist: pyarrow>=14.0.0
 Requires-Dist: httpx>=0.24.0
+Requires-Dist: aiohttp>=3.9

{memstack_skill_loader-4.1.1 → memstack_skill_loader-4.3.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "memstack-skill-loader"
-version = "4.1.1"
+version = "4.3.0"
 description = "MCP server that vector-indexes MemStack Pro skills for on-demand loading"
 requires-python = ">=3.10"
 dependencies = [
@@ -13,6 +13,7 @@ dependencies = [
     "sentence-transformers>=2.2.0",
     "pyarrow>=14.0.0",
     "httpx>=0.24.0",
+    "aiohttp>=3.9",
 ]
 
 [project.scripts]

{memstack_skill_loader-4.1.1 → memstack_skill_loader-4.3.0}/src/memstack_skill_loader/__init__.py

@@ -1,3 +1,3 @@
 """MemStack Skill Loader — MCP server for semantic skill search."""
 
-__version__ = "4.1.1"
+__version__ = "4.2.0"

memstack_skill_loader-4.3.0/src/memstack_skill_loader/__main__.py

@@ -0,0 +1,183 @@
+"""Entry point for python -m memstack_skill_loader."""
+
+import asyncio
+import os
+import signal
+import sys
+import time
+from pathlib import Path
+
+from .server import run
+
+
+def _run_agents(task: str) -> None:
+    """Start agent runner and print live message log until done."""
+    from . import agent_runner
+
+    print(f"\033[1;36m{'='*60}\033[0m")
+    print(f"\033[1;36m  Agent Runner — Task Orchestration\033[0m")
+    print(f"\033[1;36m{'='*60}\033[0m")
+    print(f"\n  Task: {task}")
+    print(f"  Working dir: {sys.argv[3] if len(sys.argv) > 3 else '.'}")
+    print(f"\n  Press Ctrl+C to stop all agents.\n")
+
+    working_dir = sys.argv[3] if len(sys.argv) > 3 else None
+    result = agent_runner.start_run(task=task, working_dir=working_dir)
+
+    if "error" in result:
+        print(f"\033[31mError: {result['error']}\033[0m", file=sys.stderr)
+        sys.exit(1)
+
+    print(f"  Session: {result['session_id']}\n")
+    print(f"\033[90m{'─'*60}\033[0m")
+
+    seen_messages = 0
+    try:
+        while True:
+            time.sleep(2)
+            status = agent_runner.get_status()
+
+            # Print new messages
+            messages = status.get("messages", [])
+            for msg in messages[seen_messages:]:
+                from_name = msg.get("from", "?")
+                to_name = msg.get("to", "?")
+                content = msg.get("content", "")[:200]
+                ts = msg.get("timestamp", "")
+
+                color = "\033[35m" if from_name == "manager" else \
+                        "\033[36m" if from_name == "builder" else \
+                        "\033[32m" if from_name == "reviewer" else "\033[90m"
+
+                print(f"  {color}{from_name}\033[0m → \033[37m{to_name}\033[0m")
+                for line in content.split("\n")[:5]:
+                    print(f"    {line}")
+                print()
+
+            seen_messages = len(messages)
+
+            # Check if done
+            session_status = status.get("status", "")
+            if session_status in ("completed", "error", "stopped"):
+                print(f"\033[90m{'─'*60}\033[0m")
+                color = "\033[32m" if session_status == "completed" else "\033[31m"
+                print(f"\n  {color}Status: {session_status}\033[0m")
+                result_text = status.get("result", "")
+                if result_text:
+                    print(f"  {result_text[:300]}")
+                print()
+                break
+
+    except KeyboardInterrupt:
+        print("\n\n  Stopping agents...")
+        agent_runner.stop_all()
+        print("  All agents stopped.")
+
+
+def _stop() -> None:
+    """Stop the MemStack dashboard and any agent runner claude.exe processes."""
+    pid_file = Path.home() / ".memstack" / "dashboard.pid"
+    if not pid_file.exists():
+        print("No MemStack dashboard running.")
+        return
+
+    try:
+        pid = int(pid_file.read_text().strip())
+    except (ValueError, OSError):
+        print("No MemStack dashboard running (corrupt PID file).")
+        pid_file.unlink(missing_ok=True)
+        return
+
+    # Check if the process is alive
+    try:
+        os.kill(pid, 0)
+    except OSError:
+        print("No MemStack dashboard running (process not found).")
+        pid_file.unlink(missing_ok=True)
+        return
+
+    # Kill any claude.exe processes spawned by agent runner
+    killed_agents = 0
+    try:
+        import subprocess
+        if sys.platform == "win32":
+            result = subprocess.run(
+                ["wmic", "process", "where",
+                 f"ParentProcessId={pid} and Name='claude.exe'",
+                 "get", "ProcessId"],
+                capture_output=True, text=True, timeout=10,
+            )
+            for line in result.stdout.splitlines():
+                line = line.strip()
+                if line.isdigit():
+                    try:
+                        os.kill(int(line), signal.SIGTERM)
+                        killed_agents += 1
+                    except OSError:
+                        pass
+        else:
+            result = subprocess.run(
+                ["pgrep", "-P", str(pid), "-x", "claude"],
+                capture_output=True, text=True, timeout=10,
+            )
+            for line in result.stdout.splitlines():
+                line = line.strip()
+                if line.isdigit():
+                    try:
+                        os.kill(int(line), signal.SIGTERM)
+                        killed_agents += 1
+                    except OSError:
+                        pass
+    except Exception:
+        pass
+
+    # Kill the dashboard process
+    try:
+        if sys.platform == "win32":
+            os.kill(pid, signal.SIGTERM)
+        else:
+            os.kill(pid, signal.SIGTERM)
+        print(f"MemStack dashboard stopped (PID {pid}).")
+        if killed_agents:
+            print(f"Stopped {killed_agents} agent process(es).")
+    except OSError as exc:
+        print(f"Failed to stop dashboard (PID {pid}): {exc}")
+
+    pid_file.unlink(missing_ok=True)
+
+
+def main():
+    if len(sys.argv) > 1 and sys.argv[1] == "stop":
+        _stop()
+        return
+    if len(sys.argv) > 1 and sys.argv[1] == "dashboard":
+        import argparse
+        parser = argparse.ArgumentParser(prog="memstack dashboard")
+        parser.add_argument("--with-proxy", action="store_true", help="Start TokenStack proxy on background thread")
+        parser.add_argument("--proxy-port", type=int, default=8787, help="Proxy port (default 8787)")
+        args = parser.parse_args(sys.argv[2:])
+        from .dashboard import start_dashboard
+        start_dashboard(with_proxy=args.with_proxy, proxy_port=args.proxy_port)
+    elif len(sys.argv) > 2 and sys.argv[1] == "run":
+        task = " ".join(sys.argv[2:])
+        _run_agents(task)
+    elif len(sys.argv) > 1 and sys.argv[1] == "proxy":
+        import argparse
+        parser = argparse.ArgumentParser(prog="memstack proxy")
+        parser.add_argument("--port", type=int, default=8787)
+        parser.add_argument("--host", default="127.0.0.1")
+        parser.add_argument("--verbose", action="store_true", help="Show parser diagnostics")
+        parser.add_argument("--pro", action="store_true", help="Force pro tier transforms (testing without license file)")
+        args = parser.parse_args(sys.argv[2:])
+        from .proxy.server import start_proxy
+        start_proxy(host=args.host, port=args.port, verbose=args.verbose, force_pro=args.pro)
+    elif len(sys.argv) > 1 and sys.argv[1] == "run":
+        print("Usage: python -m memstack_skill_loader run \"Your task description\"",
+              file=sys.stderr)
+        sys.exit(1)
+    else:
+        asyncio.run(run())
+
+
+if __name__ == "__main__":
+    main()