mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-06-06 20:15:17 +02:00
1d2519730e
filesystem_state.py (the multi-agent graph state) and state_reducers.py (its merge reducers) are consumed only by multi_agent_chat (filesystem tools/middleware, kb projection, and the MAC-only shared middleware) plus two unit tests -- no external app code. Relocate them into a dedicated multi_agent_chat/shared/state/ package (filesystem_state.py + reducers.py) and repoint every importer. No behavior change; import-all + the full unit/middleware + unit/agents suites (1066 tests) stay green.
71 lines
2.4 KiB
Python
71 lines
2.4 KiB
Python
"""Helper for wrapping a tool result with a Receipt in a ``Command(update=...)``.
|
|
|
|
Most mutating subagent tools historically returned a plain ``dict`` payload
|
|
which deepagents serialised straight into the ``ToolMessage`` content. To
|
|
participate in the verification teaching from
|
|
``multi_agent_chat/subagents/shared/snippets/verifiable_handle.md`` those
|
|
tools now also need to write a :class:`Receipt` into the parent's
|
|
``state['receipts']`` list (declared on
|
|
:class:`~app.agents.multi_agent_chat.shared.state.filesystem_state.SurfSenseFilesystemState`
|
|
and backed by the append reducer).
|
|
|
|
:func:`with_receipt` wraps both behaviours: it returns the tool payload as
|
|
a JSON-encoded ``ToolMessage`` AND appends the receipt to state in a single
|
|
:class:`~langgraph.types.Command`. Use it at every ``return`` site of a
|
|
mutating tool — including failure paths (emit a receipt with
|
|
``status="failed"`` and the error message in ``error``).
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
from typing import Any
|
|
|
|
from langchain_core.messages import ToolMessage
|
|
from langgraph.types import Command
|
|
|
|
from app.agents.shared.receipt import Receipt
|
|
|
|
|
|
def _content_to_text(payload: dict[str, Any] | str) -> str:
|
|
"""Serialise a tool payload to ``ToolMessage`` content.
|
|
|
|
Dicts go through ``json.dumps`` (matching deepagents' default tool-result
|
|
serialisation); strings are passed through. Anything else is coerced via
|
|
``str`` so we never raise here — a mis-typed tool return would already
|
|
have failed inside the tool body.
|
|
"""
|
|
if isinstance(payload, str):
|
|
return payload
|
|
if isinstance(payload, dict):
|
|
return json.dumps(payload, default=str)
|
|
return str(payload)
|
|
|
|
|
|
def with_receipt(
|
|
*,
|
|
payload: dict[str, Any] | str,
|
|
receipt: Receipt,
|
|
tool_call_id: str,
|
|
) -> Command:
|
|
"""Return a Command that ships ``payload`` as a ToolMessage AND appends ``receipt``.
|
|
|
|
The append happens via the ``_list_append_reducer`` on the ``receipts``
|
|
field of :class:`~app.agents.multi_agent_chat.shared.state.filesystem_state.SurfSenseFilesystemState`,
|
|
so concurrent subagent batches (item 4 in the plan) won't clobber each
|
|
other's receipts.
|
|
"""
|
|
return Command(
|
|
update={
|
|
"messages": [
|
|
ToolMessage(
|
|
content=_content_to_text(payload),
|
|
tool_call_id=tool_call_id,
|
|
)
|
|
],
|
|
"receipts": [receipt],
|
|
}
|
|
)
|
|
|
|
|
|
__all__ = ["with_receipt"]
|