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 |
|