[fix] ResumeParserDataclass address can be null

edenai_apis/apis/openai/openai_doc_parsing_api.py

@@ -1,6 +1,6 @@
 import json
 import os
-import fitz 
+import fitz
 from time import sleep
 
 from edenai_apis.features.ocr import (
@@ -13,17 +13,19 @@
 from edenai_apis.features import OcrInterface
 from edenai_apis.utils.exception import ProviderException
 
+
 def extract_text_from_pdf(pdf_path):
-    try :
+    try:
         doc = fitz.open(pdf_path)
-    except :
+    except:
         raise FileNotFoundError
     text = ""
     for page_num in range(len(doc)):
         page = doc.load_page(page_num)
         text += page.get_text()
     return text
 
+
 class OpenaiDocParsingApi(OcrInterface):
 
     def __assistant_parser(
@@ -42,14 +44,13 @@ def __assistant_parser(
             model="gpt-4o",
         )
 
-
         input_file_text = extract_text_from_pdf(input_file)
 
         thread = self.client.beta.threads.create(
             messages=[
                 {
                     "role": "user",
-                    "content": message_text+input_file_text,
+                    "content": message_text + input_file_text,
                 }
             ]
         )
@@ -69,7 +70,7 @@ def __assistant_parser(
 
         try:
             standardized_response = json.loads(
-                   json.loads(messages.data[0].content[0].json())["text"]["value"]
+                json.loads(messages.data[0].content[0].json())["text"]["value"]
             )
         except json.JSONDecodeError as exc:
             raise ProviderException(

edenai_apis/features/ocr/resume_parser/resume_parser_dataclass.py

@@ -70,7 +70,7 @@ class ResumePersonalName(BaseModel):
 
 class ResumePersonalInfo(BaseModel):
     name: ResumePersonalName
-    address: ResumeLocation
+    address: Optional[ResumeLocation]
     self_summary: Optional[StrictStr]  # all
     objective: Optional[StrictStr]  # Hireability
     date_of_birth: Optional[StrictStr]  # Affinda