agentchat-task-agent 0.1.1__tar.gz → 0.1.2__tar.gz

{agentchat_task_agent-0.1.1/src/agentchat_task_agent.egg-info → agentchat_task_agent-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentchat-task-agent
-Version: 0.1.1
+Version: 0.1.2
 Summary: 自主任务 Agent：面向模糊长目标的多步自主执行引擎（LangGraph），零业务依赖
 Author: Zhuliqx
 License-Expression: MIT

{agentchat_task_agent-0.1.1 → agentchat_task_agent-0.1.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "agentchat-task-agent"
-version = "0.1.1"
+version = "0.1.2"
 description = "自主任务 Agent：面向模糊长目标的多步自主执行引擎（LangGraph），零业务依赖"
 readme = "README.md"
 license = "MIT"

{agentchat_task_agent-0.1.1 → agentchat_task_agent-0.1.2/src/agentchat_task_agent.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentchat-task-agent
-Version: 0.1.1
+Version: 0.1.2
 Summary: 自主任务 Agent：面向模糊长目标的多步自主执行引擎（LangGraph），零业务依赖
 Author: Zhuliqx
 License-Expression: MIT

{agentchat_task_agent-0.1.1 → agentchat_task_agent-0.1.2}/src/agentchat_task_agent.egg-info/SOURCES.txt

@@ -24,6 +24,7 @@ src/task_agent/telemetry.py
 src/task_agent/tools.py
 tests/test_cli.py
 tests/test_judge.py
+tests/test_llm.py
 tests/test_memory.py
 tests/test_resilience.py
 tests/test_task_agent.py

{agentchat_task_agent-0.1.1 → agentchat_task_agent-0.1.2}/src/task_agent/config.py

@@ -19,6 +19,7 @@ class TaskAgentConfig:
     hitl: bool = True             # 节点级人工确认（依赖 checkpointer，无则自动降级全自主）
     max_retries: int = 2          # verify 容错：单个子任务失败后自检的最大重试次数
     max_steps: int = 8            # replan 模式步数上限（防循环）
+    replan_retries: int = 1       # replan 解析不到下一步动作时的重试次数（空响应/非 JSON 兜底）
     llm_timeout: float = 60.0     # 单次 LLM 调用超时（节点级 timeout 用）
     llm_max_retries: int = 2      # 节点级瞬时错误重试次数
     findings_budget: int | None = None  # findings 保留条数上限；超限把历史压缩进 findings_summary（None=不限）
@@ -28,5 +29,7 @@ class TaskAgentConfig:
             raise ValueError(f"未知任务模式: {self.mode!r}（支持 replan / fixed）")
         if self.max_retries < 0 or self.max_steps <= 0:
             raise ValueError("max_retries >= 0 且 max_steps > 0")
+        if self.replan_retries < 0:
+            raise ValueError("replan_retries >= 0")
         if self.findings_budget is not None and self.findings_budget <= 0:
             raise ValueError("findings_budget 必须为正整数或 None（不限制）")

agentchat_task_agent-0.1.2/src/task_agent/llm.py

@@ -0,0 +1,30 @@
+"""LLM 最小协议与文本抽取工具（不绑定具体厂商 SDK）。"""
+from __future__ import annotations
+
+from typing import Any, Callable, Protocol
+
+
+class LLM(Protocol):
+    """宿主/默认实现只需提供 async ainvoke(prompt) -> 带 .content 的响应。"""
+
+    async def ainvoke(self, prompt: str) -> Any: ...
+
+
+LLMFactory = Callable[[], LLM]
+
+
+async def llm_text(llm: LLM, prompt: str, empty_retries: int = 2) -> str:
+    """单次 LLM 调用并抽取文本；失败直接抛异常（由节点级 retry/error_handler 处理）。
+
+    空/纯空白响应视为瞬时失败（上游 LLM 偶发空内容），重试 ``empty_retries`` 次；
+    重试耗尽仍为空则返回空串，由上层解析/降级逻辑决定语义（如 replan 的空
+    ``next_action`` 被当作"无下一步"）。
+    """
+    last = ""
+    for _ in range(empty_retries + 1):
+        resp = await llm.ainvoke(prompt)
+        text = resp.content if isinstance(resp.content, str) else str(resp.content)
+        if text and text.strip():
+            return text
+        last = text
+    return last

{agentchat_task_agent-0.1.1 → agentchat_task_agent-0.1.2}/src/task_agent/nodes.py

@@ -127,8 +127,13 @@ def make_nodes(runtime: Runtime) -> dict[str, Node]:
         goal = state["goal"]
         findings = state.get("findings") or []
         mem = await _memory_ctx(runtime, goal)
-        action, source = _parse_next_action(
-            await llm_text(
+        # 首次进入（step=0）语义上是初始规划，发 "plan"；后续才发 "replan"
+        kind = "replan" if int(state.get("step") or 0) > 0 else "plan"
+        # 解析不到下一步动作时重试（覆盖空响应与非 JSON 输出），
+        # 避免上游 LLM 偶发异常被误判为"任务已完成"而提前结束。
+        action, source = "", "default"
+        for _ in range(config.replan_retries + 1):
+            raw = await llm_text(
                 runtime.llm_factory(),
                 REPLAN_PROMPT.format(
                     goal=goal,
@@ -136,12 +141,14 @@ def make_nodes(runtime: Runtime) -> dict[str, Node]:
                     memory=mem,
                 ),
             )
-        )
+            action, source = _parse_next_action(raw)
+            if action:
+                break
         if not action:
             # 无下一步 → 视为可完成
-            runtime.emit("replan", {"action": "", "source": source})
+            runtime.emit(kind, {"action": "", "source": source})
             return {"current_action": "", "done": True, "retries": 0}
-        runtime.emit("replan", {"action": action, "source": source})
+        runtime.emit(kind, {"action": action, "source": source})
         return {"current_action": action, "expected_source": source, "retries": 0}
 
     async def execute_action_node(state: dict) -> dict:

agentchat_task_agent-0.1.2/tests/test_llm.py

@@ -0,0 +1,49 @@
+"""llm_text 空响应重试行为测试。"""
+from __future__ import annotations
+
+import asyncio
+from types import SimpleNamespace
+
+from task_agent.llm import llm_text
+
+
+class _FakeLLM:
+    """按调用顺序返回预设文本的假 LLM（超出后重复最后一条）。"""
+
+    def __init__(self, *contents: str):
+        self._contents = list(contents)
+        self.calls = 0
+
+    async def ainvoke(self, prompt):
+        self.calls += 1
+        return SimpleNamespace(
+            content=self._contents[min(self.calls - 1, len(self._contents) - 1)]
+        )
+
+
+def _run(coro):
+    return asyncio.run(coro)
+
+
+def test_llm_text_returns_content_without_retry():
+    llm = _FakeLLM("答案")
+    assert _run(llm_text(llm, "p")) == "答案"
+    assert llm.calls == 1
+
+
+def test_llm_text_retries_empty_then_succeeds():
+    llm = _FakeLLM("", "  ", "答案")
+    assert _run(llm_text(llm, "p")) == "答案"
+    assert llm.calls == 3  # 首次 + 2 次重试
+
+
+def test_llm_text_returns_empty_after_retries_exhausted():
+    llm = _FakeLLM("", "", "")
+    assert _run(llm_text(llm, "p")) == ""
+    assert llm.calls == 3
+
+
+def test_llm_text_respects_custom_retries():
+    llm = _FakeLLM("", "")
+    assert _run(llm_text(llm, "p", empty_retries=1)) == ""
+    assert llm.calls == 2

{agentchat_task_agent-0.1.1 → agentchat_task_agent-0.1.2}/tests/test_task_agent.py

@@ -464,3 +464,64 @@ def test_demo_flow_offline():
     assert result.get("final_answer")
     assert "2020" in result["final_answer"]
     assert len(result.get("findings") or []) >= 2
+
+
+def test_replan_empty_llm_response_is_not_premature_done():
+    """LLM 连续空响应后给出合法动作：不应被当作“没有下一步”提前结束。"""
+    llm = _SequenceLLM(
+        "", "", '{"next_action": "计算", "expected_source": "code"}'
+    )
+    nodes = _nodes(llm=llm)
+    out = asyncio.run(
+        nodes["replan_node"]({"goal": "计算 12 和 18 的最小公倍数", "findings": []})
+    )
+    assert out.get("current_action") == "计算"
+    assert not out.get("done", False)
+
+
+def test_replan_persistently_empty_falls_back_to_done():
+    """LLM 持续空响应（重试耗尽）才按“无下一步”结束。"""
+    llm = _SequenceLLM("", "", "")
+    nodes = _nodes(llm=llm)
+    out = asyncio.run(nodes["replan_node"]({"goal": "目标", "findings": []}))
+    assert out.get("done") is True
+    assert not out.get("current_action")
+
+
+def test_replan_unparseable_then_valid_action():
+    """非 JSON 输出不应被当作“没有下一步”：重试一次后解析出动作。"""
+    llm = _SequenceLLM("抱歉，我无法完成该任务。", '{"next_action": "计算", "expected_source": "code"}')
+    nodes = _nodes(llm=llm)
+    out = asyncio.run(
+        nodes["replan_node"]({"goal": "计算 12 和 18 的最小公倍数", "findings": []})
+    )
+    assert out.get("current_action") == "计算"
+    assert not out.get("done", False)
+
+
+def test_replan_persistently_unparseable_falls_back_to_done():
+    """持续非 JSON 输出（重试耗尽）才按“无下一步”结束。"""
+    llm = _SequenceLLM("抱歉，我无法完成该任务。", "无法解析")
+    nodes = _nodes(llm=llm)
+    out = asyncio.run(nodes["replan_node"]({"goal": "目标", "findings": []}))
+    assert out.get("done") is True
+    assert not out.get("current_action")
+
+
+def test_replan_emits_plan_then_replan():
+    """首次进入 replan 节点发 "plan"（初始规划），后续轮次才发 "replan"。"""
+    events: list[tuple[str, dict]] = []
+    llm = _SequenceLLM(
+        '{"next_action": "步骤一", "expected_source": "default"}',
+        '{"next_action": "步骤二", "expected_source": "default"}',
+    )
+    runtime = Runtime(
+        config=TaskAgentConfig(),
+        llm_factory=lambda: llm,
+        executor=_FakeExecutor(),
+        on_event=lambda kind, data: events.append((kind, data)),
+    )
+    nodes = make_nodes(runtime)
+    asyncio.run(nodes["replan_node"]({"goal": "目标", "findings": [], "step": 0}))
+    asyncio.run(nodes["replan_node"]({"goal": "目标", "findings": [], "step": 1}))
+    assert [k for k, _ in events] == ["plan", "replan"]

agentchat_task_agent-0.1.1/src/task_agent/llm.py

@@ -1,19 +0,0 @@
-"""LLM 最小协议与文本抽取工具（不绑定具体厂商 SDK）。"""
-from __future__ import annotations
-
-from typing import Any, Callable, Protocol
-
-
-class LLM(Protocol):
-    """宿主/默认实现只需提供 async ainvoke(prompt) -> 带 .content 的响应。"""
-
-    async def ainvoke(self, prompt: str) -> Any: ...
-
-
-LLMFactory = Callable[[], LLM]
-
-
-async def llm_text(llm: LLM, prompt: str) -> str:
-    """单次 LLM 调用并抽取文本；失败直接抛异常（由节点级 retry/error_handler 处理）。"""
-    resp = await llm.ainvoke(prompt)
-    return resp.content if isinstance(resp.content, str) else str(resp.content)