langchain-agentx-python 2.3.8__tar.gz → 2.3.9__tar.gz

{langchain-agentx-python-2.3.8/langchain_agentx_python.egg-info → langchain-agentx-python-2.3.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langchain-agentx-python
-Version: 2.3.8
+Version: 2.3.9
 Summary: LangChain/LangGraph-based agent utilities for CodeBaseX.
 Author-email: GoodMood2008 <GoodMood2008@users.noreply.github.com>
 License:                                  Apache License

{langchain-agentx-python-2.3.8 → langchain-agentx-python-2.3.9}/langchain_agentx/__init__.py

@@ -11,7 +11,7 @@ from langchain_agentx import create_loop_agent
 ```
 """
 
-__version__ = "2.3.8"
+__version__ = "2.3.9"
 
 from .mcp.call import McpToolCaller
 from .mcp.client_manager import McpClientManager, McpConnectedClient

{langchain-agentx-python-2.3.8 → langchain-agentx-python-2.3.9}/langchain_agentx/loop/loop_abort.py

@@ -45,11 +45,23 @@ class LoopAbortSignal(Exception):
 
 
 def is_cancel_requested() -> bool:
-    """当前 RunContext 是否已请求取消（无 context 则为 False）。"""
+    """当前或祖先 RunContext 是否已请求取消（无 context 则为 False）。
+
+    祖先链：nested Parallelization ctx 下 leaf 可读到 L0 ``cancel_requested``。
+    """
     from .run_context import get_run_context
 
     ctx = get_run_context()
-    return ctx is not None and bool(ctx.cancel_requested)
+    seen: set[int] = set()
+    while ctx is not None:
+        ident = id(ctx)
+        if ident in seen:
+            break
+        seen.add(ident)
+        if bool(ctx.cancel_requested):
+            return True
+        ctx = ctx.parent
+    return False
 
 
 def abort_signal_for_phase(

{langchain-agentx-python-2.3.8 → langchain-agentx-python-2.3.9}/langchain_agentx/loop/run_context.py

@@ -12,6 +12,7 @@ run_context.py — Workflow/Agent 结构化并发上下文（parent → child ca
     asyncio 为主；thread 仅 bounded join（无法强制中断 LLM）；anyio/trio 不保证 context 传播。
     P1：reset 失败时在仍可见 owner 的 derived logical context 内 force-clear；
     宿主 detached ingest 用 create_task_with_copied_context / run_context_scope。
+    nested Parallelization：child_contexts + 递归 cancel_tree；W4 可用 use_cancel_authority。
 """
 
 from __future__ import annotations
@@ -41,6 +42,10 @@ class RunContext:
     root_task: asyncio.Task
     child_tasks: set[asyncio.Task] = field(default_factory=set)
     child_threads: set[threading.Thread] = field(default_factory=set)
+    # nested Parallelization/Chaining 再 install 时登记；cancel_tree 先深取消其 child_tasks。
+    # 用 list：RunContext 含 set/Task 不可哈希，不能放 set。
+    child_contexts: list["RunContext"] = field(default_factory=list, repr=False)
+    parent: Optional["RunContext"] = field(default=None, repr=False, compare=False)
     cancel_requested: bool = False
     # D-EVT-6：宿主在 cancel_tree / killAll 时写入；finish 投影读取；缺省 unknown。
     # 取值：escape | workflow_stop | kill_agents | unknown；禁止从「是否有后台」猜测。
@@ -52,10 +57,42 @@ class RunContext:
     def register_child_thread(self, thread: threading.Thread) -> None:
         self.child_threads.add(thread)
 
-    async def cancel_tree(self, timeout: float = 10.0) -> None:
+    def register_child_context(self, child: "RunContext") -> None:
+        """登记 nested RunContext（install_run_context(parent=…)）。"""
+        if all(existing is not child for existing in self.child_contexts):
+            self.child_contexts.append(child)
+        child.parent = self
+
+    def cancel_authority(self) -> "RunContext":
+        """取消权威：最外层 RunContext（宿主 Stop / cancel_tree 目标）。"""
+        node: RunContext = self
+        seen: set[int] = set()
+        while node.parent is not None:
+            ident = id(node)
+            if ident in seen:
+                break
+            seen.add(ident)
+            node = node.parent
+        return node
+
+    def _mark_cancel_requested(self, *, source: str | None) -> None:
+        """本层 + 全部 nested 写入 cancel_requested（协作闸可立即看到）。"""
         self.cancel_requested = True
+        if source is not None and self.cancel_source is None:
+            self.cancel_source = source
+        for nested in list(self.child_contexts):
+            nested._mark_cancel_requested(source=source or self.cancel_source)
+
+    async def cancel_tree(self, timeout: float = 10.0) -> None:
+        """取消本层 child_tasks，并递归 nested child_contexts（W4 / leaf producer）。"""
+        self._mark_cancel_requested(source=self.cancel_source)
         deadline = time.monotonic() + timeout
 
+        # 深取消：先砍 nested 内 W4 producer，再 cancel 本层（含 nested.root_task）。
+        for nested in list(self.child_contexts):
+            remaining = max(0.0, deadline - time.monotonic())
+            await nested.cancel_tree(timeout=remaining)
+
         for child in list(self.child_tasks):
             child.cancel()
         if self.child_tasks:
@@ -128,6 +165,7 @@ def install_run_context(
         ctx = RunContext(root_task=current)
     if parent is not None:
         parent.register_child(ctx.root_task)
+        parent.register_child_context(ctx)
     token = _current_run_context.set(ctx)
     return ctx, token
 
@@ -205,15 +243,32 @@ async def drive_async_iterator_in_current_task(
         yield item
 
 
+def resolve_cancel_authority(
+    ctx: RunContext | None = None,
+) -> RunContext | None:
+    """解析取消权威 ctx；缺省取当前 ContextVar 的 outermost。"""
+    current = ctx if ctx is not None else get_run_context()
+    if current is None:
+        return None
+    return current.cancel_authority()
+
+
 def register_spawned_task(
     task: asyncio.Task,
     *,
     owner: RunContext | None = None,
+    use_cancel_authority: bool = False,
 ) -> None:
-    """将 spawn 出的 asyncio task 注册到 RunContext（无 context 时 no-op）。"""
+    """将 spawn 出的 asyncio task 注册到 RunContext（无 context 时 no-op）。
+
+    ``use_cancel_authority=True``：挂到 outermost（Workflow W4），避免仅挂在
+    nested Parallelization ctx 上导致 L0 ``cancel_tree`` 砍不到 producer。
+    """
     ctx = owner if owner is not None else get_run_context()
     if ctx is None:
         return
+    if use_cancel_authority:
+        ctx = ctx.cancel_authority()
     try:
         owner_loop = ctx.root_task.get_loop()
         task_loop = task.get_loop()
@@ -249,5 +304,6 @@ __all__ = [
     "register_spawned_task",
     "register_spawned_thread",
     "reset_run_context",
+    "resolve_cancel_authority",
     "run_context_scope",
 ]

{langchain-agentx-python-2.3.8 → langchain-agentx-python-2.3.9}/langchain_agentx/workflow/execution.py

@@ -445,7 +445,9 @@ class WorkflowAgentLoopExecutor:
                 await queue.put(None)
 
         task = asyncio.create_task(_producer())
-        register_spawned_task(task)
+        # W4：挂到 cancel authority（outermost），避免 Parallelization nested
+        # install 后 L0.cancel_tree 只看见 nested.root、砍不到 leaf producer。
+        register_spawned_task(task, use_cancel_authority=True)
         try:
             while True:
                 event = await queue.get()

{langchain-agentx-python-2.3.8 → langchain-agentx-python-2.3.9/langchain_agentx_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langchain-agentx-python
-Version: 2.3.8
+Version: 2.3.9
 Summary: LangChain/LangGraph-based agent utilities for CodeBaseX.
 Author-email: GoodMood2008 <GoodMood2008@users.noreply.github.com>
 License:                                  Apache License

{langchain-agentx-python-2.3.8 → langchain-agentx-python-2.3.9}/pyproject.toml

@@ -4,7 +4,7 @@
 
  [project]
  name = "langchain-agentx-python"
- version = "2.3.8"
+ version = "2.3.9"
  description = "LangChain/LangGraph-based agent utilities for CodeBaseX."
  readme = "README.md"
  requires-python = ">=3.11"