From bd5502811549067546be57ce90325f1759dc021c Mon Sep 17 00:00:00 2001 From: sabaimran Date: Sat, 9 Nov 2024 19:17:18 -0800 Subject: [PATCH] Fix randint import from random when creating filenames for tmp --- src/khoj/processor/content/docx/docx_to_entries.py | 4 ++-- src/khoj/processor/content/pdf/pdf_to_entries.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/khoj/processor/content/docx/docx_to_entries.py b/src/khoj/processor/content/docx/docx_to_entries.py index 9ca9ef1f..55dd8bac 100644 --- a/src/khoj/processor/content/docx/docx_to_entries.py +++ b/src/khoj/processor/content/docx/docx_to_entries.py @@ -1,7 +1,7 @@ import logging import os from datetime import datetime -from random import random +from random import randint from typing import Dict, List, Tuple from langchain_community.document_loaders import Docx2txtLoader @@ -95,7 +95,7 @@ class DocxToEntries(TextToEntries): """Extract text from specified DOCX file""" try: timestamp_now = datetime.utcnow().timestamp() - random_suffix = random.randint(0, 1000) + random_suffix = randint(0, 1000) tmp_file = f"tmp_docx_file_{timestamp_now}_{random_suffix}.docx" docx_entry_by_pages = [] with open(tmp_file, "wb") as f: diff --git a/src/khoj/processor/content/pdf/pdf_to_entries.py b/src/khoj/processor/content/pdf/pdf_to_entries.py index a5de1335..311ac807 100644 --- a/src/khoj/processor/content/pdf/pdf_to_entries.py +++ b/src/khoj/processor/content/pdf/pdf_to_entries.py @@ -2,7 +2,7 @@ import base64 import logging import os from datetime import datetime -from random import random +from random import randint from typing import Dict, List, Tuple from langchain_community.document_loaders import PyMuPDFLoader @@ -99,7 +99,7 @@ class PdfToEntries(TextToEntries): try: # Write the PDF file to a temporary file, as it is stored in byte format in the pdf_file object and the PDF Loader expects a file path timestamp_now = datetime.utcnow().timestamp() - random_suffix = random.randint(0, 1000) + random_suffix = randint(0, 1000) tmp_file = f"tmp_pdf_file_{timestamp_now}_{random_suffix}.pdf" pdf_entry_by_pages = [] with open(f"{tmp_file}", "wb") as f: