langchain-agentx-python 1.9.2__tar.gz → 1.9.3__tar.gz

{langchain-agentx-python-1.9.2/langchain_agentx_python.egg-info → langchain-agentx-python-1.9.3}/PKG-INFO

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

{langchain-agentx-python-1.9.2 → langchain-agentx-python-1.9.3}/langchain_agentx/__init__.py

@@ -11,7 +11,7 @@ from langchain_agentx import create_loop_agent
 ```
 """
 
-__version__ = "1.9.2"
+__version__ = "1.9.3"
 
 from .loop import (  # noqa: F401
     create_loop_agent,

{langchain-agentx-python-1.9.2 → langchain-agentx-python-1.9.3}/langchain_agentx/loop/run_context.py

@@ -10,6 +10,8 @@ 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。
 """
 
 from __future__ import annotations
@@ -18,12 +20,15 @@ import asyncio
 import logging
 import threading
 import time
-from contextvars import ContextVar, Token
+from contextlib import asynccontextmanager
+from contextvars import ContextVar, Token, copy_context
 from dataclasses import dataclass, field
-from typing import Optional
+from typing import AsyncIterator, Awaitable, Optional, TypeVar
 
 logger = logging.getLogger(__name__)
 
+T = TypeVar("T")
+
 _current_run_context: ContextVar[Optional["RunContext"]] = ContextVar(
     "run_ctx", default=None
 )
@@ -124,8 +129,77 @@ def install_run_context(
     return ctx, token
 
 
-def reset_run_context(token: Token) -> None:
-    _current_run_context.reset(token)
+def reset_run_context(token: Token, *, owner: RunContext | None = None) -> None:
+    """恢复 RunContext ContextVar；跨 logical context 时 force-clear 或降级 warn。
+
+    P0：token.reset 失败不抛未捕获异常。
+    P1：若当前 logical context 仍可见 ``owner``（derived context 继承值但 token 无效），
+    显式 ``set(None)`` 清理本层，避免 workflow finally 后 run_ctx 悬挂。
+    """
+    try:
+        _current_run_context.reset(token)
+        return
+    except ValueError:
+        pass
+
+    current = get_run_context()
+    if owner is not None and current is owner:
+        _current_run_context.set(None)
+        logger.warning(
+            "run_ctx force-cleared: token reset failed in derived logical context "
+            "(task=%r)",
+            owner.root_task.get_name(),
+        )
+        return
+
+    if owner is not None and owner.root_task.done():
+        logger.debug(
+            "run_ctx reset skipped: owner task %r already done; context discarded",
+            owner.root_task.get_name(),
+        )
+        return
+
+    logger.warning(
+        "run_ctx reset skipped: token created in a different context",
+        exc_info=True,
+    )
+
+
+@asynccontextmanager
+async def run_context_scope(*, parent: RunContext | None = None):
+    """显式 bracket RunContext install/reset（宿主非 generator 路径）。"""
+    ctx, token = install_run_context(parent=parent)
+    try:
+        yield ctx
+    finally:
+        reset_run_context(token, owner=ctx)
+
+
+def create_task_with_copied_context(
+    coro: Awaitable[T],
+    *,
+    context=None,
+    name: str | None = None,
+) -> asyncio.Task[T]:
+    """在显式 context 快照中 spawn task（宿主 POST → 后台 ingest 正模式）。
+
+    默认 ``copy_context()``，保证 detached task 与创建点 contextvars 一致。
+    仍需在同一 task 内完整消费 workflow 流，避免跨 task aclose generator。
+    """
+    if context is None:
+        context = copy_context()
+    return asyncio.create_task(coro, context=context, name=name)
+
+
+async def drive_async_iterator_in_current_task(
+    iterator: AsyncIterator[T],
+) -> AsyncIterator[T]:
+    """在当前 asyncio task 内驱动 async iterator（禁止把 aclose 交给子 task）。
+
+    用于宿主 ingest 需额外并发但仍须保证 generator finally 与 install 同 task 的场景。
+    """
+    async for item in iterator:
+        yield item
 
 
 def register_spawned_task(
@@ -165,9 +239,12 @@ def register_spawned_thread(
 
 __all__ = [
     "RunContext",
+    "create_task_with_copied_context",
+    "drive_async_iterator_in_current_task",
     "get_run_context",
     "install_run_context",
     "register_spawned_task",
     "register_spawned_thread",
     "reset_run_context",
+    "run_context_scope",
 ]

{langchain-agentx-python-1.9.2 → langchain-agentx-python-1.9.3}/langchain_agentx/workflow/base.py

@@ -416,13 +416,13 @@ class BaseWorkflow(ABC):
             "workflow_path": workflow_path,
         }
         parent_ctx = get_run_context() if parent_workflow is not None else None
-        _, run_ctx_token = install_run_context(parent=parent_ctx)
+        run_ctx, run_ctx_token = install_run_context(parent=parent_ctx)
 
         try:
             return await graph.ainvoke(initial_state, config=effective_config)
         finally:
             try:
-                reset_run_context(run_ctx_token)
+                reset_run_context(run_ctx_token, owner=run_ctx)
             finally:
                 try:
                     await self._emit_workflow_end_best_effort(workflow_end_ctx)
@@ -576,7 +576,7 @@ class BaseWorkflow(ABC):
             "workflow_path": workflow_path,
         }
         parent_ctx = get_run_context() if parent_workflow is not None else None
-        _, run_ctx_token = install_run_context(parent=parent_ctx)
+        run_ctx, run_ctx_token = install_run_context(parent=parent_ctx)
         try:
             graph = self.compile()
             effective_config = self._resolve_effective_run_config(
@@ -598,7 +598,7 @@ class BaseWorkflow(ABC):
                 yield event
         finally:
             try:
-                reset_run_context(run_ctx_token)
+                reset_run_context(run_ctx_token, owner=run_ctx)
             finally:
                 try:
                     await self._emit_workflow_end_best_effort(workflow_end_ctx)

{langchain-agentx-python-1.9.2 → langchain-agentx-python-1.9.3/langchain_agentx_python.egg-info}/PKG-INFO

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

{langchain-agentx-python-1.9.2 → langchain-agentx-python-1.9.3}/pyproject.toml

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