AI创想

标题: 六、、使用LangGraph自定义State [打印本页]

---

作者: 米落枫    时间: 昨天 22:39
标题: 六、、使用LangGraph自定义State
作者：CSDN博客
        State中除了消息列表（messages: Annotated[list, add_messages]），也可以加入其它字段，如下，加入name和birthday字段：

1. class State(TypedDict):
2.     messages: Annotated[list, add_messages]
3.     name: str
4.     birthday: str

复制代码

使用human_assistance 工具更新state

1. from langchain_core.messages import ToolMessage
2. from langchain_core.tools import InjectedToolCallId, tool
4. from langgraph.types import Command, interrupt
6. @tool
7. def human_assistance(
8.     name: str, birthday: str, tool_call_id: Annotated[str, InjectedToolCallId]
9. ) -> str:
10.     """Request assistance from a human."""
11.     human_response = interrupt(
12.         {
13.             "question": "Is this correct?",
14.             "name": name,
15.             "birthday": birthday,
16.         },
17.     )
18.     # If the information is correct, update the state as-is.
19.     if human_response.get("correct", "").lower().startswith("y"):
20.         verified_name = name
21.         verified_birthday = birthday
22.         response = "Correct"
23.     # Otherwise, receive information from the human reviewer.
24.     else:
25.         verified_name = human_response.get("name", name)
26.         verified_birthday = human_response.get("birthday", birthday)
27.         response = f"Made a correction: {human_response}"
29.     # This time we explicitly update the state with a ToolMessage inside
30.     # the tool.
31.     state_update = {
32.         "name": verified_name,
33.         "birthday": verified_birthday,
34.         "messages": [ToolMessage(response, tool_call_id=tool_call_id)],
35.     }
36.     # We return a Command object in the tool to update our state.
37.     return Command(update=state_update)

复制代码

完整代码：

1. import os
2. from typing import Annotated
3. from typing_extensions import

复制代码

原文地址：https://blog.csdn.net/Mrhiuser/article/details/151395406

---

欢迎光临 AI创想 (https://www.llms-ai.com/)

Powered by Discuz! X3.4