feat: parando de esperar todo processamento terminar para enviar a resposta da requisição

_utils/bubble_integrations/enviar_resposta_final.py

@@ -0,0 +1,14 @@
+import requests
+import os
+
+def enviar_resposta_final(doc_id: str, form_response_id: str, version: str, texto_completo: str, error: bool = False):
+    url = f"https://vella.app.br/version-{version}/api/1.1/wf/texto_completo"
+    headers = {"Authorization": f"Bearer {os.environ.get("BUBBLE_TOKEN")}"}
+
+    body = {
+        "doc_id": doc_id,
+        "form_response_id": form_response_id,
+        "texto_completo": texto_completo,
+        "erro": error,
+    }
+    requests.post(url, body, headers=headers)

_utils/custom_exception_handler.py

@@ -0,0 +1,31 @@
+# myapp/exception_handler.py
+
+from typing import Dict
+from rest_framework.views import exception_handler
+import logging
+
+from _utils.bubble_integrations.enviar_resposta_final import enviar_resposta_final
+
+logger = logging.getLogger(__name__)
+
+
+def custom_exception_handler(exc, context):
+    # Call REST framework's default exception handler first
+    response = exception_handler(exc, context)
+    serializer: Dict = context["view"].serializer
+
+    enviar_resposta_final(
+        serializer.get("doc_id", ""),
+        serializer.get("form_response_id", ""),
+        serializer.get("version", ""),
+        serializer.get("texto_completo", ""),
+        True,
+    )
+
+    return response
+
+
+def your_additional_function(exc, context):
+    # Implement your custom logic here
+    # For example, send an email notification or perform other actions
+    pass

_utils/gerar_relatorio_modelo_usuario/utils.py

@@ -29,7 +29,7 @@ def gerar_resposta_compilada(serializer):
         "chunk_size": serializer["chunk_size"],
         "chunk_overlap": serializer["chunk_overlap"],
         # "prompt_auxiliar": serializer["prompt_auxiliar"],
-        "prompt_gerar_documento": serializer["prompt_gerar_documento"],
+        "prompt_gerar_documento": serializer["prompt_gerar_documento"][0:200],
     }
 
 

_utils/resumo_completo_cursor.py

@@ -1,4 +1,5 @@
 import os
+from _utils.bubble_integrations.enviar_resposta_final import enviar_resposta_final
 from _utils.gerar_relatorio_modelo_usuario.prompts import prompt_auxiliar_SEM_CONTEXT
 from _utils.gerar_relatorio_modelo_usuario.GerarDocumento import (
     GerarDocumento,
@@ -101,7 +102,6 @@ async def get_llm_summary_answer_by_cursor_complete(
     )
 
     llm_ultimas_requests = serializer["llm_ultimas_requests"]
-    # Generate enhanced summary
     print("\nCOMEÇANDO A FAZER ÚLTIMA REQUISIÇÃO")
     structured_summaries = await summarizer.gerar_documento_final(
         vector_store, bm25, chunk_ids, llm_ultimas_requests, prompt_auxiliar_SEM_CONTEXT
@@ -117,9 +117,22 @@ async def get_llm_summary_answer_by_cursor_complete(
 
     for x in structured_summaries:
         texto_completo = texto_completo + x["content"] + "\n"
+        x["source"]["text"] = x["source"]["text"][0:200]
+        x["source"]["context"] = x["source"]["context"][0:200]
+
+    if isBubble:
+        print("COMEÇANDO A REQUISIÇÃO FINAL PARA O BUBBLE")
+        enviar_resposta_final(
+            serializer["doc_id"],
+            serializer["form_response_id"],
+            serializer["version"],
+            texto_completo,
+            False,
+        )
+        print("TERMINOU A REQUISIÇÃO FINAL PARA O BUBBLE")
 
     return {
-        "resultado": structured_summaries,
         "texto_completo": texto_completo,
+        "resultado": structured_summaries,
         "parametros-utilizados": gerar_resposta_compilada(serializer),
     }

gerar_documento/serializer.py

@@ -49,7 +49,9 @@ class GerarDocumentoSerializer(ResumoCursorSerializer):
     llm_ultimas_requests = serializers.CharField(
         required=False, default="gemini-2.0-flash"
     )
-
+    doc_id = serializers.CharField(required=True)
+    form_response_id = serializers.CharField(required=True)
+    version = serializers.CharField(required=True)
 
 class GerarDocumentoComPDFProprioSerializer(ResumoCursorSerializer):
     system_prompt = None

gerar_documento/views.py

@@ -16,10 +16,12 @@ from .serializer import (
     GerarDocumentoComPDFProprioSerializer,
     GerarDocumentoSerializer,
 )
+import asyncio
 
 
 class GerarDocumentoView(AsyncAPIView):
     # parser_classes = [MultiPartParser]
+    serializer = {}
 
     @extend_schema(
         request=GerarDocumentoSerializer,
@@ -30,31 +32,41 @@ class GerarDocumentoView(AsyncAPIView):
         if serializer.is_valid(raise_exception=True):
             if not serializer.validated_data:
                 raise ValueError("Erro no validated_data")
-            data = cast(Dict[str, Any], serializer.validated_data)
-            print("\ndata: ", data)
+            
+            async def proccess_data_after_response():
+                # await asyncio.sleep(0)
+                data = cast(Dict[str, Any], serializer.validated_data)
+                print("\ndata: ", data)
+                self.serializer = data
 
-            # data["prompt_auxiliar"] = (
-            #     prompt_auxiliar_inicio + "\n" + data["prompt_auxiliar"]
-            # )
+                # data["prompt_auxiliar"] = (
+                #     prompt_auxiliar_inicio + "\n" + data["prompt_auxiliar"]
+                # )
 
-            # listaPDFs = handle_pdf_files_from_serializer(data["files"])
-            listaPDFs = [l["link_arquivo"] for l in data["files"]]
+                # listaPDFs = handle_pdf_files_from_serializer(data["files"])
+                listaPDFs = [l["link_arquivo"] for l in data["files"]]
 
-            print("\n\nlistaPDFs: ", listaPDFs)
+                print("\n\nlistaPDFs: ", listaPDFs)
 
-            resposta_llm = await get_llm_summary_answer_by_cursor_complete(
-                data, listaPDFs, True
-            )
-            print("\n\nresposta_llm: ", resposta_llm)
+                resposta_llm = await get_llm_summary_answer_by_cursor_complete(
+                    data, listaPDFs, True
+                )
+                print("\n\nresposta_llm: ", resposta_llm)
 
-            # remove_pdf_temp_files(listaPDFs)
+                # remove_pdf_temp_files(listaPDFs)
 
-            print("PRÓXIMA LINHA ENVIA A RESPOSTA A QUEM FEZ A REQUISIÇÃO")
-            return Response({"resposta": resposta_llm})
+                # print("PRÓXIMA LINHA ENVIA A RESPOSTA A QUEM FEZ A REQUISIÇÃO")
+            
+            # asyncio.create_task(proccess_data_after_response())
+            loop = asyncio.get_running_loop()
+            loop.run_in_executor(None, lambda: asyncio.run(proccess_data_after_response()))
+            
+            return Response({"resposta": "Requisição está sendo processada em segundo plano"})
 
 
 class GerarDocumentoComPDFProprioView(AsyncAPIView):
     parser_classes = [MultiPartParser]
+    serializer = {}
 
     @extend_schema(
         request=GerarDocumentoComPDFProprioSerializer,
@@ -65,6 +77,7 @@ class GerarDocumentoComPDFProprioView(AsyncAPIView):
         if serializer.is_valid(raise_exception=True):
             data = cast(Dict[str, Any], serializer.validated_data)
             print("\n\ndata: ", data)
+            self.serializer = data
 
             listaPDFs = handle_pdf_files_from_serializer(data["files"])
 

setup/settings.py

@@ -142,6 +142,7 @@ CORS_ORIGIN_WHITELIST = [
 ]
 
 REST_FRAMEWORK = {
+    'EXCEPTION_HANDLER': "_utils.custom_exception_handler.custom_exception_handler",
     "DEFAULT_PAGINATION_CLASS": "rest_framework.pagination.PageNumberPagination",
     "PAGE_SIZE": 10,
     "DEFAULT_SCHEMA_CLASS": "drf_spectacular.openapi.AutoSchema",