文心一言的key申請，參考：
LangChain 三行代碼實現(xiàn) 文心一言 Prompt+LLM 大模型問答
http://www.itdecent.cn/p/8ec9ea907446
上干貨：
from langchain_community.llms import QianfanLLMEndpoint
from langchain.vectorstores import Chroma
from langchain.chains import ConversationalRetrievalChain
from langchain.memory import ConversationBufferMemory
from langchain.embeddings.baidu_qianfan_endpoint import QianfanEmbeddingsEndpoint
from langchain_text_splitters import RecursiveCharacterTextSplitter
from langchain.document_loaders import UnstructuredFileLoader
WENXIN_APP_Key = "your key"
WENXIN_APP_SECRET = "your secret"
# 1 構(gòu)建 檢索向量庫
# 1.1 加載長文檔
loader = UnstructuredFileLoader("test_data/test.txt")
document = loader.load()
# 1.2 chunk
text_splitter = RecursiveCharacterTextSplitter(chunk_size=384, chunk_overlap=0, separators=["\n\n", "\n", " ", "", "。", "，"])
documents = text_splitter.split_documents(document)
# 1.3 向量模型，向量庫
embeddings = QianfanEmbeddingsEndpoint(qianfan_ak=WENXIN_APP_Key,qianfan_sk=WENXIN_APP_SECRET,model="Embedding-V1")
retriever = Chroma.from_documents(documents, embeddings).as_retriever()
# 2 定義問答大模型和消息記錄緩存
memory = ConversationBufferMemory(memory_key = "chat_history", return_messages=True)
llm = QianfanLLMEndpoint(model="ERNIE-Bot", qianfan_ak=WENXIN_APP_Key, qianfan_sk=WENXIN_APP_SECRET)
qa = ConversationalRetrievalChain.from_llm(llm, retriever, memory=memory)
result = qa({"question": "三級凍傷是什么意思"})
print(result)
