Spaces:
Running
Running
File size: 1,162 Bytes
ab79998 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
import pytest
from _utils.bubble_integrations.obter_arquivo import (
get_pdf_from_bubble,
)
from langchain_core.documents import Document
from _utils.gerar_documento_utils.contextual_retriever import (
ContextualRetriever,
ContextualRetrieverUtils,
)
from _utils.gerar_documento_utils.llm_calls import agemini_answer
from _utils.models.gerar_documento import RetrievalConfig
from _utils.models.gerar_documento import (
ContextualizedChunk,
DocumentChunk,
RetrievalConfig,
)
from gerar_documento.serializer import (
GerarDocumentoComPDFProprioSerializer,
GerarDocumentoComPDFProprioSerializerData,
)
from setup.logging import Axiom
from langchain_community.document_loaders import PyPDFLoader, PyMuPDFLoader
class TestObterArquivo:
@pytest.mark.asyncio
async def test_get_pdf_from_bubble(self):
file_url = "https://vella.app.br/version-5265j/fileupload/f1745094959601x803512841326306200/000_I%CC%81ntegra%20%200015348-89.2020.4.06.0981.pdf"
response = await get_pdf_from_bubble(file_url, False)
assert len(response) > 0
for page in response:
assert isinstance(page, Document)
|