Openai API + langchain 分析小型pdf文档

最新推荐文章于 2025-08-06 13:15:18 发布

原创

最新推荐文章于 2025-08-06 13:15:18 发布 · 623 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#langchain #pdf #python

声明：该版代码在2024.08.23有效。

代码如下：

from langchain_community.document_loaders import PyPDFLoader
import getpass
import os
from langchain_openai import ChatOpenAI
from langchain_chroma import Chroma
from langchain_openai import OpenAIEmbeddings
from langchain_text_splitters import RecursiveCharacterTextSplitter
from langchain.chains import create_retrieval_chain
from langchain.chains.combine_documents import create_stuff_documents_chain
from langchain_core.prompts import ChatPromptTemplate

class QA:
    """
    A class to handle question-answering tasks on a given PDF document.

    Attributes:
        question (str): The question to be answered about the PDF.
        pdf_path (str): Path to the PDF file.
        model_name (str): Name of the model used for analysis.
        docs (list): Loaded PDF documents.
        vecstore (Chroma): The vector store object for storing document embeddings.

    Methods:
        set_environ(): Set environment