release: v0.0.14 and prep for langgraph v0.6 (#92)

Tests passing locally, waiting for v0.6 to merge.

Author: sydney-runkle

langgraph_swarm/swarm.py

@@ -1,8 +1,10 @@
 from typing import Literal, Optional, Union, cast, get_args, get_origin
+from warnings import warn
 
+from langgraph._internal._typing import DeprecatedKwargs
 from langgraph.graph import START, MessagesState, StateGraph
 from langgraph.pregel import Pregel
-from typing_extensions import Any, TypeVar
+from typing_extensions import Any, TypeVar, Unpack
 
 from langgraph_swarm.handoff import get_handoff_destinations
 
@@ -143,12 +145,13 @@ def route_to_active_agent(state: dict) -> str:
     return builder
 
 
-def create_swarm(
+def create_swarm(  # noqa: D417
     agents: list[Pregel],
     *,
     default_active_agent: str,
     state_schema: StateSchemaType = SwarmState,
-    config_schema: type[Any] | None = None,
+    context_schema: type[Any] | None = None,
+    **deprecated_kwargs: Unpack[DeprecatedKwargs],
 ) -> StateGraph:
     """Create a multi-agent swarm.
 
@@ -159,8 +162,7 @@ def create_swarm(
             or any other [Pregel](https://langchain-ai.github.io/langgraph/reference/pregel/#langgraph.pregel.Pregel) object.
         default_active_agent: Name of the agent to route to by default (if no agents are currently active).
         state_schema: State schema to use for the multi-agent graph.
-        config_schema: An optional schema for configuration.
-            Use this to expose configurable parameters via `swarm.config_specs`.
+        context_schema: Specifies the schema for the context object that will be passed to the workflow.
 
     Returns:
         A multi-agent swarm StateGraph.
@@ -208,14 +210,22 @@ def add(a: int, b: int) -> int:
         ```
 
     """
+    if (config_schema := deprecated_kwargs.get("config_schema")) is not None:
+        warn(
+            "`config_schema` is deprecated. Please use `context_schema` instead.",
+            DeprecationWarning,
+            stacklevel=2,
+        )
+        context_schema = config_schema  # type: ignore[assignment]
+
     active_agent_annotation = state_schema.__annotations__.get("active_agent")
     if active_agent_annotation is None:
         msg = "Missing required key 'active_agent' in state_schema"
         raise ValueError(msg)
 
     agent_names = [agent.name for agent in agents]
     state_schema = _update_state_schema_agent_names(state_schema, agent_names)
-    builder = StateGraph(state_schema, config_schema)
+    builder = StateGraph(state_schema, context_schema)
     add_active_agent_router(
         builder,
         route_to=agent_names,

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "pdm.backend"
 
 [project]
 name = "langgraph-swarm"
-version = "0.0.13"
+version = "0.0.14"
 description = "An implementation of a multi-agent swarm using LangGraph"
 authors = [
     {name = "Vadym Barda", email = "[email protected] "}
@@ -14,7 +14,7 @@ license-files = ["LICENSE"]
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
-    "langgraph>=0.5",
+    "langgraph>=0.6.0,<0.7.0",
     "langchain-core>=0.3.40,<0.4.0"
 ]
 
@@ -68,7 +68,7 @@ check_untyped_defs = true
 "tests/**/*.py" = [
     # at least this three should be fine in tests:
     "S101", # asserts allowed in tests...
-    "ARG", # Unused function args -> fixtures nevertheless are functionally relevant...
+    "ARG", # Unused function args -> fixtures nevertheless are functionally relevant
     "FBT", # Don't care about booleans as positional arguments in tests, e.g. via @pytest.mark.parametrize()
     "D104",
     # The below are debateable
@@ -82,4 +82,3 @@ check_untyped_defs = true
     "D105", # Missing docstring in magic method
     "D415", # First line should end with a period, question mark, or exclamation point
 ]
-