让Langchain Chatbot接入多种知识源

Question

我正在使用 langchain 构建一个聊天机器人，我希望聊天机器人在矢量存储中拥有一个知识库，并且还可以访问维基百科。我创建了聊天机器人有一个矢量存储来从中获取数据的部分。请指导我如何构建聊天机器人可以访问维基百科的功能。

def get_answer(query):
    client = get_weaviate_client()
    db = Weaviate(client=client, index_name=INDEX_NAME, text_key="text")
    llm = ChatOpenAI(temperature=0)
    
    chain = RetrievalQA.from_chain_type(
        retriever=db.as_retriever(),
        llm=llm, chain_type="stuff",
        memory=ConversationBufferWindowMemory(k=3)
    )
    response = chain(
        {"query": query},
        return_only_outputs=True,
    )
    return response['result']

Answer 1

你需要 pip 安装维基百科

 pip install wikipedia

调用 API 的代码示例

 from langchain.tools import WikipediaQueryRun
 from langchain.utilities import WikipediaAPIWrapper

 wikipedia = WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper())
 print(wikipedia.run("HUNTER X HUNTER"))