Spaces:
Running
Running
luanpoppe
commited on
Commit
·
82e7990
1
Parent(s):
096dfd3
fix: uuid do axiom
Browse files- gerar_documento/views.py +10 -8
- 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
|
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
|
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
|
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 |
|