mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-10-29 11:27:43 +08:00
6d4ce83837
严格检查全部宿主源码，清零类型错误预算并移除按模块类型豁免。 按原因、所属方法、错误码及语法指纹锁定必要兼容例外和站点类型桩， CI 阻断新增忽略、检查范围退化、本地桩替代真实接口及新的类型错误。 修复空值、异步和供应方类型边界，保持插件与 API 兼容合同， 并补充存储、订阅、通知、Agent 和工作流边界的回归覆盖。
92 lines
3.4 KiB
Python
92 lines
3.4 KiB
Python
"""
|
|
修复消息历史中缺少结果的工具调用，保留执行结果未知的事实。
|
|
"""
|
|
|
|
from collections.abc import Sequence
|
|
from typing import Any, Optional
|
|
|
|
from langchain.agents.middleware import AgentMiddleware, AgentState
|
|
from langchain_core.messages import AIMessage, BaseMessage, ToolCall, ToolMessage
|
|
from langgraph.runtime import Runtime
|
|
from langgraph.types import Overwrite
|
|
|
|
|
|
class PatchToolCallsMiddleware(AgentMiddleware):
|
|
"""
|
|
修复消息历史中悬空工具调用的中间件。
|
|
"""
|
|
|
|
@staticmethod
|
|
def _build_cancelled_tool_message(tool_call: ToolCall) -> ToolMessage:
|
|
"""
|
|
缺少回执只表示结果未知，不能据此重放可能已生效的写操作。
|
|
"""
|
|
tool_name = tool_call.get("name") or "unknown_tool"
|
|
tool_call_id = tool_call.get("id") or ""
|
|
tool_msg = (
|
|
f"Tool call {tool_name} with id {tool_call_id} was "
|
|
"interrupted before its result was recorded; its outcome is unknown. "
|
|
"It may already have changed external state. Before retrying any mutation, "
|
|
"verify the actual state with read-only tools and avoid duplicate side effects."
|
|
)
|
|
return ToolMessage(
|
|
content=tool_msg,
|
|
name=tool_name,
|
|
tool_call_id=tool_call_id,
|
|
status="error",
|
|
)
|
|
|
|
@classmethod
|
|
def _normalize_messages(cls, messages: Sequence[BaseMessage]) -> list[BaseMessage]:
|
|
"""
|
|
规范化工具调用消息顺序，满足 OpenAI tool_calls 协议要求。
|
|
"""
|
|
if not messages or len(messages) == 0:
|
|
return list(messages)
|
|
|
|
tool_messages = {
|
|
msg.tool_call_id: msg
|
|
for msg in messages
|
|
if isinstance(msg, ToolMessage) and msg.tool_call_id
|
|
}
|
|
patched_messages: list[BaseMessage] = []
|
|
for msg in messages:
|
|
if isinstance(msg, ToolMessage):
|
|
continue
|
|
|
|
patched_messages.append(msg)
|
|
if not isinstance(msg, AIMessage) or not msg.tool_calls:
|
|
continue
|
|
|
|
for tool_call in msg.tool_calls:
|
|
tool_call_id = tool_call.get("id")
|
|
corresponding_tool_msg = tool_messages.get(tool_call_id or "")
|
|
if corresponding_tool_msg:
|
|
patched_messages.append(corresponding_tool_msg)
|
|
else:
|
|
patched_messages.append(cls._build_cancelled_tool_message(tool_call))
|
|
|
|
return patched_messages
|
|
|
|
def before_agent(self, state: AgentState, runtime: Runtime[Any]) -> Optional[dict[str, Any]]: # noqa: ARG002
|
|
"""
|
|
在代理运行之前，处理任何 AIMessage 中悬空或乱序的工具调用。
|
|
"""
|
|
messages = state["messages"]
|
|
patched_messages = self._normalize_messages(messages)
|
|
if patched_messages == messages:
|
|
return None
|
|
|
|
return {"messages": Overwrite(patched_messages)}
|
|
|
|
async def abefore_agent(self, state: AgentState, runtime: Runtime[Any]) -> Optional[dict[str, Any]]: # noqa: ARG002
|
|
"""
|
|
在代理异步运行之前，处理任何 AIMessage 中悬空或乱序的工具调用。
|
|
"""
|
|
messages = state["messages"]
|
|
patched_messages = self._normalize_messages(messages)
|
|
if patched_messages == messages:
|
|
return None
|
|
|
|
return {"messages": Overwrite(patched_messages)}
|