feat: mudando contextuals para ser Gemini-2.0-flash

_utils/LLMs/LLM_class.py

@@ -22,7 +22,7 @@ class LLM:
             model=model,
         )
 
-    def google_gemini(self, model="gemini-1.5-flash"):
+    def google_gemini(self, model="gemini-2.0-flash"):
         return ChatGoogleGenerativeAI(
             api_key=SecretStr(google_api_key),
             model=model,

_utils/gerar_relatorio_modelo_usuario/contextual_retriever.py

@@ -1,4 +1,5 @@
 import os
+from _utils.LLMs.LLM_class import LLM
 from _utils.gerar_relatorio_modelo_usuario.utils import (
     get_response_from_auxiliar_contextual_prompt,
     validate_many_chunks_in_one_request,
@@ -19,6 +20,7 @@ from _utils.models.gerar_relatorio import (
     DocumentChunk,
     RetrievalConfig,
 )
+from langchain_core.messages import HumanMessage
 
 lista_contador = []
 
@@ -84,10 +86,12 @@ class ContextualRetriever:
                     f"\n\nTENTATIVA FORMATAÇÃO CHUNKS NÚMERO {attempt + 1}"
                 )
                 print("\nCOMEÇANDO UMA REQUISIÇÃO DO CONTEXTUAL")
-                raw_response = await agpt_answer(prompt)
+                llms = LLM()
+                # raw_response = await agpt_answer(prompt)
+                raw_response = await llms.google_gemini("gemini-2.0-flash-lite-preview-02-05").ainvoke([HumanMessage(content=prompt)])
+                
                 print("\nTERMINOU UMA REQUISIÇÃO DO CONTEXTUAL")
                 response = cast(str, raw_response)
-                # llms = LLM()
                 # response = await llms.deepseek().ainvoke([HumanMessage(content=prompt)])
                 # return cast(str, response.content)
 

_utils/splitters/Splitter_class.py

@@ -108,7 +108,7 @@ class Splitter:
             }
 
             # char_count += len(text)
-            print("\nTERMINOU DE ORGANIZAR PDFS EM CHUNKS")
+        print("\nTERMINOU DE ORGANIZAR PDFS EM CHUNKS")
 
         return chunks, initial_chunks
 

tests/test_custom_exception_handler.py

@@ -0,0 +1,24 @@
+import pytest
+import os
+from _utils.splitters.Splitter_class import Splitter
+from _utils.models.gerar_relatorio import (
+    DocumentChunk,
+)
+from rest_framework import status
+from rest_framework.exceptions import ValidationError
+from rest_framework.test import APIClient
+from django.core.handlers.wsgi import WSGIRequest
+
+
+# class TestCustomExceptionHandler:
+#     @pytest.mark.django_db
+#     @pytest.mark.asyncio
+#     def test_function_that_raises(self):
+#         client = APIClient()
+        
+#         invalid_data = {
+#             "name": "wrong property"
+#         }
+
+#         response = client.post('/gerar-documento', invalid_data, format='json')
+#         assert response.status_code == status.HTTP_400_BAD_REQUEST