luanpoppe commited on
Commit
82e7990
·
1 Parent(s): 096dfd3

fix: uuid do axiom

Browse files
Files changed (2) hide show
  1. gerar_documento/views.py +10 -8
  2. setup/logging.py +3 -2
gerar_documento/views.py CHANGED
@@ -32,15 +32,16 @@ from _utils.langchain_utils.Splitter_class import Splitter
32
  class GerarDocumentoView(AsyncAPIView):
33
  # parser_classes = [MultiPartParser]
34
  serializer = {}
 
35
 
36
  @extend_schema(
37
  request=GerarDocumentoSerializer,
38
  )
39
  async def post(self, request):
40
- axiom_instance = Axiom()
41
  print(f"\n\nDATA E HORA DA REQUISIÇÃO: {datetime.now()}")
42
- axiom_instance.send_axiom("COMEÇOU NOVA REQUISIÇÃO")
43
- axiom_instance.send_axiom(f"request.data: {request.data}")
44
  serializer = GerarDocumentoSerializer(data=request.data)
45
  if serializer.is_valid(raise_exception=True):
46
  obj = serializer.get_obj() # type: ignore
@@ -54,12 +55,12 @@ class GerarDocumentoView(AsyncAPIView):
54
 
55
  listaPDFs = [l["link_arquivo"] for l in data["files"]]
56
 
57
- axiom_instance.send_axiom(f"listaPDFs: {listaPDFs}")
58
 
59
  resposta_llm = await gerar_documento(
60
- obj, listaPDFs, axiom_instance, isBubble=True
61
  )
62
- axiom_instance.send_axiom(f"resposta_llm: {resposta_llm}")
63
 
64
  # remove_pdf_temp_files(listaPDFs)
65
 
@@ -83,6 +84,7 @@ class GerarDocumentoComPDFProprioView(AsyncAPIView):
83
  request=GerarDocumentoComPDFProprioSerializer,
84
  )
85
  async def post(self, request):
 
86
  print(f"\n\nDATA E HORA DA REQUISIÇÃO: {datetime.now()}")
87
  self.axiom_instance.send_axiom("COMEÇOU NOVA REQUISIÇÃO")
88
  self.axiom_instance.send_axiom(f"request.data: {request.data}")
@@ -128,7 +130,7 @@ class GerarEmentaView(AsyncAPIView):
128
  listaPDFs = [l["link_arquivo"] for l in data["files"]]
129
  print("\n\nlistaPDFs: ", listaPDFs)
130
 
131
- all_PDFs_chunks, full_text_as_array, full_text_as_string = (
132
  await get_full_text_and_all_PDFs_chunks(
133
  listaPDFs,
134
  Splitter(data["chunk_size"], data["chunk_overlap"]),
@@ -180,7 +182,7 @@ class GerarEmentaComPDFProprioView(AsyncAPIView):
180
  listaPDFs = [l["link_arquivo"] for l in data["files"]]
181
  print("\n\nlistaPDFs: ", listaPDFs)
182
 
183
- all_PDFs_chunks, full_text_as_array, full_text_as_string = (
184
  await get_full_text_and_all_PDFs_chunks(
185
  listaPDFs,
186
  Splitter(data["chunk_size"], data["chunk_overlap"]),
 
32
  class GerarDocumentoView(AsyncAPIView):
33
  # parser_classes = [MultiPartParser]
34
  serializer = {}
35
+ axiom_instance = Axiom()
36
 
37
  @extend_schema(
38
  request=GerarDocumentoSerializer,
39
  )
40
  async def post(self, request):
41
+ self.axiom_instance.generate_new_uuid()
42
  print(f"\n\nDATA E HORA DA REQUISIÇÃO: {datetime.now()}")
43
+ self.axiom_instance.send_axiom("COMEÇOU NOVA REQUISIÇÃO")
44
+ self.axiom_instance.send_axiom(f"request.data: {request.data}")
45
  serializer = GerarDocumentoSerializer(data=request.data)
46
  if serializer.is_valid(raise_exception=True):
47
  obj = serializer.get_obj() # type: ignore
 
55
 
56
  listaPDFs = [l["link_arquivo"] for l in data["files"]]
57
 
58
+ self.axiom_instance.send_axiom(f"listaPDFs: {listaPDFs}")
59
 
60
  resposta_llm = await gerar_documento(
61
+ obj, listaPDFs, self.axiom_instance, isBubble=True
62
  )
63
+ self.axiom_instance.send_axiom(f"resposta_llm: {resposta_llm}")
64
 
65
  # remove_pdf_temp_files(listaPDFs)
66
 
 
84
  request=GerarDocumentoComPDFProprioSerializer,
85
  )
86
  async def post(self, request):
87
+ self.axiom_instance.generate_new_uuid()
88
  print(f"\n\nDATA E HORA DA REQUISIÇÃO: {datetime.now()}")
89
  self.axiom_instance.send_axiom("COMEÇOU NOVA REQUISIÇÃO")
90
  self.axiom_instance.send_axiom(f"request.data: {request.data}")
 
130
  listaPDFs = [l["link_arquivo"] for l in data["files"]]
131
  print("\n\nlistaPDFs: ", listaPDFs)
132
 
133
+ all_PDFs_chunks, full_text_as_array = (
134
  await get_full_text_and_all_PDFs_chunks(
135
  listaPDFs,
136
  Splitter(data["chunk_size"], data["chunk_overlap"]),
 
182
  listaPDFs = [l["link_arquivo"] for l in data["files"]]
183
  print("\n\nlistaPDFs: ", listaPDFs)
184
 
185
+ all_PDFs_chunks, full_text_as_array = (
186
  await get_full_text_and_all_PDFs_chunks(
187
  listaPDFs,
188
  Splitter(data["chunk_size"], data["chunk_overlap"]),
setup/logging.py CHANGED
@@ -47,10 +47,11 @@ def send_axiom(mensagem):
47
 
48
  @dataclass
49
  class Axiom:
50
- uuid = uuid.uuid4()
51
-
52
  array_of_logs = []
53
 
 
 
 
54
  def add_axiom(self, mensagem):
55
  self.array_of_logs.append(mensagem)
56
 
 
47
 
48
  @dataclass
49
  class Axiom:
 
 
50
  array_of_logs = []
51
 
52
+ def generate_new_uuid(self):
53
+ self.uuid = uuid.uuid4()
54
+
55
  def add_axiom(self, mensagem):
56
  self.array_of_logs.append(mensagem)
57