fix

_utils/gerar_relatorio_modelo_usuario/EnhancedDocumentSummarizer.py

@@ -197,28 +197,18 @@ class EnhancedDocumentSummarizer(DocumentSummarizer):
                         }
                     )
 
-            url_request = f"{api_url}/modelo/{self.id_modelo_do_usuario}"
-            print("url_request: ", url_request)
-            resposta = requests.get(url_request)
-            print("resposta: ", resposta)
-
-            if resposta.status_code != 200:
-                return Response(
-                    {
-                        "error": "Ocorreu um problema. Pode ser que o modelo não tenha sido encontrado. Tente novamente e/ou entre em contato com a equipe técnica"
-                    }
-                )
-
-            modelo_buscado = resposta.json()["modelo"]
-            # from modelos_usuarios.models import ModeloUsuarioModel
-
-            # # try:
-            # modelo_buscado = ModeloUsuarioModel.objects.get(
-            #     pk=self.id_modelo_do_usuario
-            # )
-            # serializer = ModeloUsuarioSerializer(modelo_buscado)
-            # print("serializer.data: ", serializer.data)
-
+            # url_request = f"{api_url}/modelo/{self.id_modelo_do_usuario}"
+            # try:
+            #     print("url_request: ", url_request)
+            #     resposta = requests.get(url_request)
+            #     print("resposta: ", resposta)
+            #     if resposta.status_code != 200:
+            #         print("Entrou no if de erro")
+            #         return Response(
+            #             {
+            #                 "error": "Ocorreu um problema. Pode ser que o modelo não tenha sido encontrado. Tente novamente e/ou entre em contato com a equipe técnica"
+            #             }
+            #         )
             # except:
             #     return Response(
             #         {
@@ -226,7 +216,24 @@ class EnhancedDocumentSummarizer(DocumentSummarizer):
             #         }
             #     )
 
-            # print("modelo_buscado: ", modelo_buscado)
+            # modelo_buscado = resposta.json()["modelo"]
+            from modelos_usuarios.models import ModeloUsuarioModel
+
+            try:
+                modelo_buscado = ModeloUsuarioModel.objects.get(
+                    pk=self.id_modelo_do_usuario
+                )
+                serializer = ModeloUsuarioSerializer(modelo_buscado)
+                print("serializer.data: ", serializer.data)
+
+            except:
+                return Response(
+                    {
+                        "error": "Ocorreu um problema. Pode ser que o modelo não tenha sido encontrado. Tente novamente e/ou entre em contato com a equipe técnica"
+                    }
+                )
+
+            print("modelo_buscado: ", serializer.data["modelo"])
 
             llm = ChatOpenAI(
                 temperature=self.gpt_temperature,

_utils/resumo_completo_cursor.py

@@ -34,7 +34,7 @@ os.environ.get("LANGCHAIN_API_KEY")
 os.environ["LANGCHAIN_PROJECT"] = "VELLA"
 
 
-async def get_llm_summary_answer_by_cursor_complete(
+def get_llm_summary_answer_by_cursor_complete(
     serializer, listaPDFs=None, contexto=None
 ):
     """Parâmetro "contexto" só deve ser passado quando quiser utilizar o teste com ragas, e assim, não quiser passar PDFs"""
@@ -89,12 +89,12 @@ async def get_llm_summary_answer_by_cursor_complete(
         full_text = " ".join([page.page_content for page in pages])
     # Contextualize chunks
     if serializer["should_have_contextual_chunks"]:
-        contextualized_chunks = (
-            await summarizer.contextual_retriever.contextualize_all_chunks(
-                pages, allPdfsChunks
-            )
-        )
-        chunks_passados = contextualized_chunks
+        # contextualized_chunks = (
+        #     await summarizer.contextual_retriever.contextualize_all_chunks(
+        #         pages, allPdfsChunks
+        #     )
+        # )
+        # chunks_passados = contextualized_chunks
         is_contextualized_chunk = True
     else:
         chunks_passados = allPdfsChunks

gerar_relatorio_modelo_usuario/views.py

@@ -13,13 +13,13 @@ from rest_framework.parsers import MultiPartParser
 from drf_spectacular.utils import extend_schema
 
 
-class ResumoSimplesCursorCompletoView(AsyncAPIView):
+class ResumoSimplesCursorCompletoView(APIView):
     parser_classes = [MultiPartParser]
 
     @extend_schema(
         request=ResumoCursorCompeltoSerializer,
     )
-    async def post(self, request):
+    def post(self, request):
         serializer = ResumoCursorCompeltoSerializer(data=request.data)
         if serializer.is_valid(raise_exception=True):
             print("\n\n\n")
@@ -44,9 +44,7 @@ class ResumoSimplesCursorCompletoView(AsyncAPIView):
                     listaPDFs.append(temp_file_path)
             print("listaPDFs: ", listaPDFs)
 
-            resposta_llm = await get_llm_summary_answer_by_cursor_complete(
-                data, listaPDFs
-            )
+            resposta_llm = get_llm_summary_answer_by_cursor_complete(data, listaPDFs)
 
             final = resposta_llm
             print("\n\n\n")