PDF问答工具（基于openai API和streamlit）

橙意满满的西瓜大侠

于 2025-02-03 23:27:32 发布

阅读量616

点赞数 2

分类专栏： llm 文章标签： streamlit langchain 人工智能

本文链接：https://blog.youkuaiyun.com/2302_79795489/article/details/145434541

版权

utils.py：

from langchain_community.document_loaders import PyPDFLoader
from langchain_text_splitters import RecursiveCharacterTextSplitter
from langchain_openai.embeddings import OpenAIEmbeddings
from langchain_community.vectorstores import FAISS

from langchain_openai import ChatOpenAI
from langchain.chains import ConversationalRetrievalChain

def rag_tool(api_key,memory,uploaded_file,question):
    # 将文件内容写入到本地，才能有文件路径（用户上传的文件直接储存在内存里，无路径）
    file_content = uploaded_file.read() #返回文件的二进制数据
    temp_file_path = "temp.pdf"
    with open(temp_file_path,"wb") as fwb: #以二进制写入方式打开文件，并创建文件对象（与文件进行交互的接口）
        fwb.write(file_content) #写入上面读取到的文件内容

    # 加载
    loader = PyPDFLoader(temp_file_path) 
    docs = loader.load() 
    # 分割