DeerFlowClient 增加state schema参数

[hibari1015]

自定义的middleware中会依赖一些自定义的state字段，希望可以加个state schema参数限制继承原本的state类型，不然的话自定义的middleware太受限了

[willem-bd]

你这个需求其实现在已经能满足，不需要改 DeerFlowClient。

原因是 LangChain 的 create_agent 会把每个 middleware 自身的 state_schema 合并进整张图的 state schema（不只是用 DeerFlowClient 里硬编码的那个 ThreadState）。DeerFlowClient 把 state_schema=ThreadState 作为基类传进去，但每个 middleware 的 state_schema 字段会被 union 进来，成为真正的 graph channel（会被 checkpointer 持久化）。

源码依据（langchain/agents/factory.py）：

state_schemas: set[type] = {m.state_schema for m in middleware}   # 收集每个 middleware 的 schema
base_state = state_schema if state_schema is not None else AgentState
state_schemas.add(base_state)
resolved_state_schema, input_schema, output_schema = _resolve_schemas(state_schemas)  # 按 field 合并

所以推荐做法是：在自定义 middleware 上设置 state_schema，并在类型上继承 ThreadState（这样既能拿到原始字段，又能加自定义字段）：

from typing import NotRequired, TypedDict
from langchain.agents.middleware import AgentMiddleware
from deerflow.agents.thread_state import ThreadState

# 继承原来的 state 类型，扩展自定义字段
class MyState(ThreadState):
    my_custom_field: NotRequired[str | None]

class MyMiddleware(AgentMiddleware[MyState]):
    state_schema = MyState          # ← 关键：这个字段会被合并进 graph state

    def before_model(self, state, runtime):
        # 既可读自定义字段，也能读到 ThreadState 的字段（因为继承了）
        val = state.get("my_custom_field")
        return {"my_custom_field": "..."}   # 写回的是真实 channel，会被 checkpoint 持久化

然后照常传给 client：

client = DeerFlowClient(middlewares=[MyMiddleware()])

这里要分清两个层面：

1. 运行时 channel：靠 middleware 的 state_schema 属性。只要在这里声明，字段就会进入图的 state，可以读/写/持久化。
2. 静态类型：靠 AgentMiddleware[MyState] 的泛型参数 StateT，让 before_model / after_model 里的 state 有正确的类型提示。

两个层面是独立的，所以即使 middleware 不带泛型，运行时也照样能跑；带上泛型只是让 IDE / mypy 更舒服。

几点注意事项：

• 自定义字段如果没有 Annotated[..., reducer]，默认是 LastValue（last-write-wins）。如果你的字段会被多个节点/步骤并发写，需要自己写 reducer，例如 my_field: Annotated[list[str], my_merge_reducer]，和 ThreadState 里 merge_artifacts / merge_delegations 的写法一致。
• 不要覆盖 ThreadState 已有的字段名（比如重定义 messages、artifacts）。合并是按字段名 union、后者覆盖前者的，而且 set 遍历顺序不确定，重定义核心字段会出问题。只新增字段即可。

---

补充一点：如果你想要的自定义字段不属于任何 middleware（比如只想在某个 tool 或自定义 node 里读写），那目前确实没有入口——DeerFlowClient._ensure_agent 把 state_schema 写死成了 ThreadState，没暴露参数。对应 LangChain 原生 create_agent(state_schema=...) 参数的设计意图也是“无需写 middleware 就能加 state 字段”。

如果你有这种场景，可以考虑给 DeerFlowClient.__init__ / build_middlewares 加一个 state_schema: type[ThreadState] | None = None（默认仍是 ThreadState，校验是 ThreadState 的子类），然后在 _ensure_agent 里用它替换硬编码值。这个改动很小，欢迎提 PR。

[hibari1015]

是的 也有tool更新的场景，middleware 里带的tool