昨天我們用了40行代碼搭建出了智能問答小客服，已經(jīng)清晰LangChain的結(jié)構(gòu)了，今天我們來嘗試用25行代碼實現(xiàn)出一個簡易聊天機器人。希望通過這個例子能幫助大家更加理解LangChain。

框架設(shè)計

組件一覽：

組件類型子類型HumanMessage索引組件文本加載器LLMChain鏈組件對話鏈ChatOpenAI模型組件LLMConversationBufferMemory索引組件儲存

準備工作

安裝依賴

設(shè)置密鑰

導入庫

開始編碼

ChatOpenAI響應信息

記憶

創(chuàng)建對話內(nèi)存實例,添加用戶消息和AI消息

對話

調(diào)用ChatOpenAI作為LLM。

使用ChatPromptTemplate創(chuàng)建一個名為prompt的聊天提示，其中包含系統(tǒng)消息、消息占位符和人類消息。

ConversationBufferMemory是一個對話內(nèi)存的類，用于存儲和管理對話的歷史記錄。通過設(shè)置memory_key參數(shù)，可以指定對話內(nèi)存的鍵值，以便在后續(xù)的代碼中引用。設(shè)置return_messages參數(shù)為True，表示在對話內(nèi)存中存儲的消息將被返回，以便在后續(xù)的代碼中使用。

LLMChain是一個對話鏈的類，用于構(gòu)建和管理聊天機器人的對話流程。通過設(shè)置llm參數(shù)為llm，prompt參數(shù)為prompt，verbose參數(shù)為True，memory參數(shù)為memory，可以創(chuàng)建一個具有指定參數(shù)的對話鏈實例。這個對話鏈實例將用于后續(xù)的對話交互，包括向聊天機器人提問并獲取響應。

運行

使用conversation向LLMChain實例傳遞一個包含問題的字典，然后將返回的響應存儲在response變量中，使用不同的問題進行對話。

執(zhí)行結(jié)果演示

完整代碼

完整代碼在githhub搜qq31682216/chatgpt_all

小結(jié)

以上就是我在配置并使用LangChian實現(xiàn)一個簡易聊天機器人分享，希望可以幫到各位！歡迎關(guān)注或發(fā)私信與我共同討論更多大模型領(lǐng)域知識~