update type

utils/chatgpt.py

@@ -3,7 +3,7 @@ import tiktoken
 
 
 class ChatGPTAPI:
-    def __init__(self, api_key='', max_input_length=1024):
+    def __init__(self, api_key: str = '', max_input_length: int = 1024):
         if not api_key:
             try:
                 api_key = open('data/api_key.txt', 'r').read()

utils/read_pdf.py

@@ -1,7 +1,9 @@
+from typing import List
+
 import pypdf
 
 
-def read_pdf(filepath) -> list[str]:
+def read_pdf(filepath: str) -> List[str]:
     outputs = []
     with open(filepath, 'rb') as f:
         pdf_reader = pypdf.PdfReader(f)