feat: adicionando mais testes do contextual

_utils/gerar_documento_utils/llm_calls.py

@@ -1,5 +1,5 @@
 import os
-from typing import Literal
+from typing import Literal, cast
 
 from pydantic import SecretStr
 from _utils.langchain_utils.LLM_class import LLM
@@ -62,7 +62,7 @@ async def agemini_answer(
     model: Literal[
         "gemini-2.5-pro-exp-03-25", "gemini-2.0-flash", "gemini-2.0-flash-lite"
     ] = "gemini-2.0-flash",
-):
+) -> str:
     gemini = llm.google_gemini(model)
     resposta = await gemini.ainvoke([HumanMessage(content=prompt)])
-    return resposta.content
+    return cast(str, resposta.content)

_utils/gerar_documento_utils/tests/test_contextual_retriever_inside.py

@@ -19,7 +19,10 @@ from .fixtures.resposta_llm_com_20_chunks import (
     resposta_llm_com_20_chunks_fixture_array,
     lista_20_ids_fixture,
 )
-from .fixtures.lista_com_20_chunks import lista_com_20_chunks_fixture
+from .fixtures.lista_com_20_chunks import (
+    lista_com_20_chunks_fixture,
+    lista_de_listas_com_60_chunks_fixture,
+)
 from .fixtures.resumo_auxiliar import resumo_auxiliar_fixture
 from .fixtures import mock_llm_call_uma_lista_de_20_chunks
 
@@ -399,3 +402,23 @@ class TestContextuRetriever:
 
         print("resultado: ", resultado)
         assert resultado == []
+
+    @pytest.mark.asyncio
+    async def test_contextualize_all_chunks_success(
+        self,
+    ):
+        resultado = await self.context_retrieval.contextualize_all_chunks(
+            lista_de_listas_com_60_chunks_fixture,
+            resumo_auxiliar_fixture,
+            self.axiom,
+        )
+
+        print("resultado: ", resultado)
+        assert resultado
+        assert len(resultado) == 60
+        for r in resultado:
+            assert isinstance(r, ContextualizedChunk)
+            # assert isinstance(r[1], str)
+            # assert isinstance(r[2], str)
+            # assert len(r[1]) > 0
+            # assert len(r[2]) > 0