strands-swarms 0.1.0__tar.gz → 0.1.1__tar.gz

strands_swarms-0.1.1/.github/workflows/ci.yml

@@ -0,0 +1,50 @@
+name: CI
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+jobs:
+  lint:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install ruff mypy
+          pip install -e .
+      - name: Lint with ruff
+        run: ruff check src/
+      - name: Type check with mypy
+        run: mypy src/
+
+  test:
+    runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: ["3.10", "3.11", "3.12", "3.13"]
+    steps:
+      - uses: actions/checkout@v4
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install -e ".[dev]"
+      - name: Run tests
+        run: |
+          if [ -d "tests" ]; then
+            pytest -v
+          else
+            echo "No tests directory found, skipping tests"
+          fi

{strands_swarms-0.1.0 → strands_swarms-0.1.1}/PKG-INFO

@@ -1,9 +1,10 @@
 Metadata-Version: 2.4
 Name: strands-swarms
-Version: 0.1.0
+Version: 0.1.1
 Summary: Task-based workflow orchestration for Strands Agents
-Project-URL: Homepage, https://github.com/strands-agents/strands-swarms
-Project-URL: Issues, https://github.com/strands-agents/strands-swarms/issues
+Project-URL: Homepage, https://github.com/JackXu0/strands-swarms
+Project-URL: Issues, https://github.com/JackXu0/strands-swarms/issues
+Author-email: Zhuocheng Xu <zhuocheng.xu@icloud.com>
 License: Apache-2.0
 License-File: LICENSE
 Classifier: Development Status :: 3 - Alpha
@@ -27,6 +28,10 @@ Description-Content-Type: text/markdown
 
 # strands-swarms
 
+[![CI](https://github.com/JackXu0/strands-swarms/actions/workflows/ci.yml/badge.svg)](https://github.com/JackXu0/strands-swarms/actions/workflows/ci.yml)
+[![PyPI](https://img.shields.io/pypi/v/strands-swarms)](https://pypi.org/project/strands-swarms/)
+[![License](https://img.shields.io/badge/License-Apache%202.0-blue.svg)](https://opensource.org/licenses/Apache-2.0)
+
 **Turn natural language into multi-agent workflows — automatically.**
 
 Give `DynamicSwarm` a query, and it automatically plans the workflow, spawns specialized agents, and executes tasks with dependencies. No manual graph configuration or agent wiring required.
@@ -99,7 +104,7 @@ Query: "Research AI trends and write a summary report"
 ## Installation
 
 ```bash
-pip install git+https://github.com/JackXu0/strands-swarms.git
+pip install strands-swarms
 ```
 
 ## Quick Start

{strands_swarms-0.1.0 → strands_swarms-0.1.1}/README.md

@@ -1,5 +1,9 @@
 # strands-swarms
 
+[![CI](https://github.com/JackXu0/strands-swarms/actions/workflows/ci.yml/badge.svg)](https://github.com/JackXu0/strands-swarms/actions/workflows/ci.yml)
+[![PyPI](https://img.shields.io/pypi/v/strands-swarms)](https://pypi.org/project/strands-swarms/)
+[![License](https://img.shields.io/badge/License-Apache%202.0-blue.svg)](https://opensource.org/licenses/Apache-2.0)
+
 **Turn natural language into multi-agent workflows — automatically.**
 
 Give `DynamicSwarm` a query, and it automatically plans the workflow, spawns specialized agents, and executes tasks with dependencies. No manual graph configuration or agent wiring required.
@@ -72,7 +76,7 @@ Query: "Research AI trends and write a summary report"
 ## Installation
 
 ```bash
-pip install git+https://github.com/JackXu0/strands-swarms.git
+pip install strands-swarms
 ```
 
 ## Quick Start

{strands_swarms-0.1.0 → strands_swarms-0.1.1}/pyproject.toml

@@ -4,8 +4,11 @@ build-backend = "hatchling.build"
 
 [project]
 name = "strands-swarms"
-version = "0.1.0"
+version = "0.1.1"
 description = "Task-based workflow orchestration for Strands Agents"
+authors = [
+    {name = "Zhuocheng Xu", email = "zhuocheng.xu@icloud.com"}
+]
 readme = "README.md"
 requires-python = ">=3.10"
 license = {text = "Apache-2.0"}
@@ -34,8 +37,8 @@ dev = [
 ]
 
 [project.urls]
-Homepage = "https://github.com/strands-agents/strands-swarms"
-Issues = "https://github.com/strands-agents/strands-swarms/issues"
+Homepage = "https://github.com/JackXu0/strands-swarms"
+Issues = "https://github.com/JackXu0/strands-swarms/issues"
 
 [tool.hatch.build.targets.wheel]
 packages = ["src/strands_swarms"]

{strands_swarms-0.1.0 → strands_swarms-0.1.1}/src/strands_swarms/__init__.py

@@ -33,32 +33,32 @@ Example:
     result = swarm.execute("Research AI trends and summarize")
 """
 
-from .swarm import DynamicSwarm, DynamicSwarmResult
-from .orchestrator import create_orchestrator_agent
+# Re-export strands types for convenience
+from strands.hooks import HookProvider, HookRegistry
+from strands.multiagent.base import Status
+
 from .events import (
-    # Planning/Orchestration events
-    SwarmStartedEvent,
-    PlanningStartedEvent,
     AgentSpawnedEvent,
-    TaskCreatedEvent,
-    PlanningCompletedEvent,
+    ExecutionCompletedEvent,
     # Execution events
     ExecutionStartedEvent,
-    TaskStartedEvent,
-    TaskCompletedEvent,
-    TaskFailedEvent,
-    ExecutionCompletedEvent,
-    SwarmCompletedEvent,
-    SwarmFailedEvent,
+    PlanningCompletedEvent,
+    PlanningStartedEvent,
     # Hook provider
     PrintingHookProvider,
+    SwarmCompletedEvent,
+    SwarmFailedEvent,
+    # Planning/Orchestration events
+    SwarmStartedEvent,
+    TaskCompletedEvent,
+    TaskCreatedEvent,
+    TaskFailedEvent,
+    TaskStartedEvent,
 )
+from .orchestrator import create_orchestrator_agent
+from .swarm import DynamicSwarm, DynamicSwarmResult
 
-# Re-export strands types for convenience
-from strands.hooks import HookProvider, HookRegistry
-from strands.multiagent.base import Status
-
-__version__ = "0.1.0"
+__version__ = "0.1.1"
 
 __all__ = [
     # Main API

{strands_swarms-0.1.0 → strands_swarms-0.1.1}/src/strands_swarms/events.py

@@ -9,11 +9,10 @@ from __future__ import annotations
 
 import re
 from dataclasses import dataclass, field
-from typing import Any
+from typing import Any, Callable
 
 from strands.hooks.registry import BaseHookEvent, HookProvider, HookRegistry
 
-
 # =============================================================================
 # Planning Events
 # =============================================================================
@@ -264,7 +263,7 @@ def _filter_internal_tags(text: str) -> str:
     return _INTERNAL_TAG_PATTERN.sub('', text)
 
 
-def create_colored_callback_handler(color: str, agent_name: str):
+def create_colored_callback_handler(color: str, agent_name: str) -> Callable[..., None]:
     """Create a callback handler that prints agent output with a specific color.
     
     This ensures all output from a specific agent (text, reasoning, tool calls)

{strands_swarms-0.1.0 → strands_swarms-0.1.1}/src/strands_swarms/swarm.py

@@ -14,24 +14,24 @@ from __future__ import annotations
 import asyncio
 from dataclasses import dataclass, field
 from textwrap import dedent
-from typing import Any, Callable, TYPE_CHECKING
+from typing import TYPE_CHECKING, Any, Callable
 
 from strands import Agent
-from strands.multiagent.base import Status, MultiAgentResult
-from strands.multiagent.graph import Graph, GraphBuilder, GraphResult
 from strands.hooks import HookProvider, HookRegistry
+from strands.multiagent.base import MultiAgentResult, Status
+from strands.multiagent.graph import Graph, GraphBuilder, GraphResult
 
 from .events import (
     AGENT_COLORS,
-    SwarmStartedEvent,
-    PlanningStartedEvent,
-    ExecutionStartedEvent,
-    TaskStartedEvent,
-    TaskCompletedEvent,
     ExecutionCompletedEvent,
+    ExecutionStartedEvent,
+    PlanningStartedEvent,
+    PrintingHookProvider,
     SwarmCompletedEvent,
     SwarmFailedEvent,
-    PrintingHookProvider,
+    SwarmStartedEvent,
+    TaskCompletedEvent,
+    TaskStartedEvent,
     create_colored_callback_handler,
 )
 
@@ -301,7 +301,7 @@ def build_swarm(
             name=definition.name,
             system_prompt=definition.build_system_prompt(),
             model=model,
-            tools=tools if tools else None,
+            tools=tools if tools else None,  # type: ignore[arg-type]
             callback_handler=callback_handler,
         )
 
@@ -522,7 +522,7 @@ class DynamicSwarm:
             DynamicSwarmResult containing planning and execution results.
         """
         # Import here to avoid circular import
-        from .orchestrator import set_swarm_config, create_orchestrator_agent
+        from .orchestrator import set_swarm_config
 
         # Create swarm config for this execution
         config = SwarmConfig(
@@ -599,6 +599,7 @@ class DynamicSwarm:
             # Orchestrator Phase 3: Generate Final Response
             # Uses the SAME orchestrator agent from planning (continued conversation)
             # =================================================================
+            assert planning_result.orchestrator is not None, "Orchestrator should exist after successful planning"
             final_response = await self._run_completion(
                 query, config, execution_result, 
                 orchestrator=planning_result.orchestrator