如何使用LangChain将对话记忆添加到ChatGPT API中

ChatGPT中文站 — Photo by Andrew Neel on Unsplash

这个快速教程介绍了如何使用 LangChain 与 ChatGPT API（gpt-3.5-turbo）一起使用，以添加对话记忆（摘要缓存记忆）。

pip install langchain openai

import openai

from langchain.chat_models import ChatOpenAI
from langchain.chains import ConversationChain
from langchain.memory import ConversationSummaryBufferMemory

openai.api_key = "your-openai-api-key"

chat = ChatOpenAI()
conversation = ConversationChain(
    llm=chat,
    memory=ConversationSummaryBufferMemory(
        llm=ChatOpenAI(), max_token_limit=2048
    ),
    verbose=False,
)

conversation.predict(input="Hello world!")