From 3914f4d6ac4ab1bc3da5174c559ad43b13f005e0 Mon Sep 17 00:00:00 2001 From: Dev Khant Date: Tue, 24 Sep 2024 11:16:33 +0530 Subject: [PATCH] Fix langgraph doc (#1898) --- docs/integrations/langgraph.mdx | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/integrations/langgraph.mdx b/docs/integrations/langgraph.mdx index 06c3cd1d..cb0a2f58 100644 --- a/docs/integrations/langgraph.mdx +++ b/docs/integrations/langgraph.mdx @@ -29,7 +29,7 @@ from typing import Annotated, TypedDict, List from langgraph.graph import StateGraph, START from langgraph.graph.message import add_messages from langchain_openai import ChatOpenAI -from mem0 import Memory +from mem0 import MemoryClient from langchain_core.messages import SystemMessage, HumanMessage, AIMessage # Configuration @@ -38,7 +38,7 @@ MEM0_API_KEY = 'your-mem0-key' # Replace with your actual Mem0 API key # Initialize LangChain and Mem0 llm = ChatOpenAI(model="gpt-4", api_key=OPENAI_API_KEY) -mem0 = Memory(api_key=MEM0_API_KEY) +mem0 = MemoryClient(api_key=MEM0_API_KEY) ``` ## Define State and Graph