vella-backend-tests / setup /easy_imports.py
luanpoppe
fix: adicionando mais uma possibilidade de carregar o pdf caso dê erro com o pypdf
ab79998
raw
history blame
1.2 kB
from adrf.views import APIView as AsyncAPIView
from drf_spectacular.utils import extend_schema
from rest_framework.views import APIView
from rest_framework.response import Response
from rest_framework.parsers import MultiPartParser
from langchain.text_splitter import RecursiveCharacterTextSplitter
from langchain_huggingface import HuggingFaceEmbeddings
# from langchain_community.embeddings import HuggingFaceEmbeddings
from langchain.prompts import PromptTemplate
from langchain_core.runnables import RunnablePassthrough
from langchain_core.prompts import ChatPromptTemplate
from langchain_community.document_loaders import (
PyPDFLoader,
Docx2txtLoader,
TextLoader,
PyMuPDFLoader,
)
from langchain_community.vectorstores import Chroma
from langchain_google_genai import ChatGoogleGenerativeAI
# from langchain_community.chat_models import ChatOpenAI
from langchain_openai import ChatOpenAI, OpenAIEmbeddings
from langchain.schema import Document
from langchain.chains import create_extraction_chain
from langchain.chains.retrieval import create_retrieval_chain
from langchain.chains.combine_documents import create_stuff_documents_chain
from rank_bm25 import BM25Okapi