core: add RemoveMessage (#23636)

This change adds a new message type `RemoveMessage`. This will enable
`langgraph` users to manually modify graph state (or have the graph
nodes modify the state) to remove messages by `id`

Examples:

* allow users to delete messages from state by calling

```python
graph.update_state(config, values=[RemoveMessage(id=state.values[-1].id)])
```

* allow nodes to delete messages

```python
graph.add_node("delete_messages", lambda state: [RemoveMessage(id=state[-1].id)])
```
This commit is contained in:
Vadym Barda
2024-06-28 17:40:02 -04:00
committed by GitHub
parent 8fce8c6771
commit e8d77002ea
5 changed files with 39 additions and 0 deletions

View File

@@ -21,6 +21,7 @@ EXPECTED_ALL = [
"ToolCallChunk",
"ToolMessage",
"ToolMessageChunk",
"RemoveMessage",
"convert_to_messages",
"get_buffer_string",
"merge_content",