diff --git a/src/oss/langchain/short-term-memory.mdx b/src/oss/langchain/short-term-memory.mdx index a0f013e67..0d8d72a53 100644 --- a/src/oss/langchain/short-term-memory.mdx +++ b/src/oss/langchain/short-term-memory.mdx @@ -40,7 +40,7 @@ from langgraph.checkpoint.memory import InMemorySaver # [!code highlight] agent = create_agent( "gpt-5", - [get_user_info], + tools=[get_user_info], checkpointer=InMemorySaver(), # [!code highlight] ) @@ -91,7 +91,7 @@ with PostgresSaver.from_conn_string(DB_URI) as checkpointer: checkpointer.setup() # auto create tables in PostgresSql agent = create_agent( "gpt-5", - [get_user_info], + tools=[get_user_info], checkpointer=checkpointer, # [!code highlight] ) ``` @@ -124,7 +124,7 @@ class CustomAgentState(AgentState): # [!code highlight] agent = create_agent( "gpt-5", - [get_user_info], + tools=[get_user_info], state_schema=CustomAgentState, # [!code highlight] checkpointer=InMemorySaver(), ) @@ -241,8 +241,8 @@ def trim_messages(state: AgentState, runtime: Runtime) -> dict[str, Any] | None: } agent = create_agent( - model, - tools=tools, + your_model_here, + tools=your_tools_here, middleware=[trim_messages], checkpointer=InMemorySaver(), )