langmem/guides/use_tools_in_custom_agent/

[giscus[bot]]

langmem/guides/use_tools_in_custom_agent/

Build language agents as graphs

https://langchain-ai.github.io/langmem/guides/use_tools_in_custom_agent/

[hliang-8]

Hi, thanks for the tutorial. I think there may be an issue with Langmem. I've copied the OpenAI API implementation verbatim, and I get this error

KeyError                                  Traceback (most recent call last)
Cell In[6], line 99
     93 memory_tools = [
     94     create_manage_memory_tool(namespace="memories"),
     95     create_search_memory_tool(namespace="memories"),
     96 ]
     98 # Run the agent
---> 99 result = run_agent(
    100     tools=memory_tools,
    101     user_input="Remember that I like cherry pie. Then remember that I dislike rocky road.",
    102 )
    103 print(result)
    104 # I've remembered that you like cherry pie and that you dislike rocky road...

Cell In[6], line 74, in run_agent(tools, user_input, max_steps)
     69     messages.append(
     70         {"role": "assistant", "content": message.content, "tool_calls": tool_calls}
     71     )
     73     for tool_call in tool_calls:
---> 74         tool_result = execute_tool(tools_by_name, tool_call.model_dump())
     75         messages.append(
     76             {
     77                 "role": "tool",
   (...)     80             }
...
    116 """
    117 config = get_config()
--> 118 return config[CONF][CONFIG_KEY_STORE]

KeyError: '__pregel_store'