python-3.x 使用ChatGPT API仅使用自定义知识而不是一般知识来训练Chatbot

jvlzgdj9  于 2023-08-08  发布在  Python
关注(0)|答案(1)|浏览(166)

所以现在我正在用Python编写一个使用ChatGPT API的聊天机器人,我遇到的问题是,我希望聊天机器人只具有从PDF文件中训练的自定义知识,而不是一般知识。
所以我用来嵌入PDF文件的函数如下:

# Function to embed the text
def get_vectorstore(text_chunks, api_key):
    embeddings = OpenAIEmbeddings(openai_api_key=api_key)
    vectorstore = FAISS.from_texts(text_chunks, embedding=embeddings)
    return vectorstore

字符串
我用来训练聊天机器人的代码是下面的?

# Train the bot with the PDF information
def get_conversation_chain(vectorstore, api_key):
    llm = ChatOpenAI(openai_api_key=api_key)
    memory = ConversationBufferMemory(memory_key='chat_history',  return_messages=True)
    conversation_chain = ConversationalRetrievalChain.from_llm(
        llm = llm,
        retriever = vectorstore.as_retriever(),
        memory = memory
    )
    return conversation_chain


关于如何只使用自定义知识训练聊天机器人有什么想法?
提前感谢!

xbp102n0

xbp102n01#

使用ChatGPT将任何PDF或文档转换为功能齐全的聊天机器人可能无法直接开箱即用。但是,您可以遵循一个结构化的流程,使用工具和技术的组合来实现这种转换。检查此link以获取所有信息

相关问题