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)