Spaces:
Running
Running
luanpoppe
commited on
Commit
·
78aaed0
1
Parent(s):
0cb12f9
refactor: removendo comentários
Browse files
_utils/gerar_documento_utils/tests/test_contextual_retriever_inside.py
CHANGED
@@ -70,41 +70,6 @@ lista_chunks_errados = [
|
|
70 |
class TestContextualRetrieverUtils:
|
71 |
context_retrieval_utils = ContextualRetrieverUtils()
|
72 |
|
73 |
-
# @pytest.mark.asyncio
|
74 |
-
# async def test_valida_retries_automaticos_do_contextual(self, monkeypatch):
|
75 |
-
# call_count = 0 # Track how many times the function is called
|
76 |
-
|
77 |
-
# # Define the custom side effect
|
78 |
-
# def custom_side_effect(response, document_ids):
|
79 |
-
# nonlocal call_count
|
80 |
-
# print(f"MOCK CALL {call_count+1} RECEIVED")
|
81 |
-
# if call_count < 2:
|
82 |
-
# call_count += 1
|
83 |
-
# print("MOCK RETORNADO FALSE")
|
84 |
-
# return False # Mocked return value for first two calls
|
85 |
-
# return validate_many_chunks_in_one_request(
|
86 |
-
# response, document_ids
|
87 |
-
# ) # Call the real function after two mocks
|
88 |
-
|
89 |
-
# monkeypatch.setattr(
|
90 |
-
# "_utils.gerar_documento_utils.contextual_retriever.validate_many_chunks_in_one_request",
|
91 |
-
# custom_side_effect,
|
92 |
-
# )
|
93 |
-
|
94 |
-
# mock_agpt_answer = AsyncMock(
|
95 |
-
# return_value=lista_respostas_contextuals_cenario_02[0]
|
96 |
-
# )
|
97 |
-
# monkeypatch.setattr(
|
98 |
-
# "_utils.gerar_documento_utils.contextual_retriever.agpt_answer",
|
99 |
-
# mock_agpt_answer,
|
100 |
-
# )
|
101 |
-
|
102 |
-
# resultado = await self.context_retrieval.llm_call_uma_lista_de_20_chunks(
|
103 |
-
# cenario_01_lista_com_20_chunks, cenario_01_resumo_auxiliar
|
104 |
-
# )
|
105 |
-
# assert all(len(item) == 3 for item in resultado)
|
106 |
-
# assert call_count == 2 # Ensure only the first two calls were mocked
|
107 |
-
|
108 |
@pytest.mark.asyncio
|
109 |
def test_check_regex_patterns_success(self):
|
110 |
for context in resposta_llm_com_20_chunks_fixture_array:
|
|
|
70 |
class TestContextualRetrieverUtils:
|
71 |
context_retrieval_utils = ContextualRetrieverUtils()
|
72 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
73 |
@pytest.mark.asyncio
|
74 |
def test_check_regex_patterns_success(self):
|
75 |
for context in resposta_llm_com_20_chunks_fixture_array:
|