使用LangChain ConversationChain实现记住对话记录

ConversationChain： 顾名思义就是可以记住历史对话。

import ChatGLM
from langchain.chains import LLMChain
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate
from langchain.chains import SimpleSequentialChain
from langchain_core.runnables import RunnablePassthrough
from operator import itemgetter
from langchain.chains import ConversationChain

llm = ChatGLM.ChatGLM_LLM()

prompt = ChatPromptTemplate.from_template("""
 请用中文返回,不多于40个字,
current conversation:{history}
        Human:{input}
"""
    )

conversation = ConversationChain(llm=llm, prompt=prompt)

print(conversation.invoke("who is Obama"))

print(conversation.invoke("who is George Washington"))

print(conversation.invoke("what is my first question"))

输出第一次的问题： who is Obama