from langchain_google_genai import ChatGoogleGenerativeAI
from langchain.memory import ConversationTokenBufferMemory
from langchain.chains import ConversationChain
import os
llm = ChatGoogleGenerativeAI(
model="gemini-1.5-flash",
api_key=os.getenv("GEMINI_API_KEY")
)
memory = ConversationTokenBufferMemory(
llm=llm,
max_token_limit=150
)
conversation = ConversationChain(
llm=llm,
memory=memory,
verbose=True
)
conversation.invoke("My name is John")
conversation.invoke("I live in Toronto and work as a software engineer")
conversation.invoke("What do you remember about me?")