Add type check before S3 prefix check

aws-samples · Jul 5, 2023 · 8dbec37 · 8dbec37
2 parents a19211f + 8a7dfda
commit 8dbec37
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/textractor/textractor.py b/textractor/textractor.py
@@ -267,7 +267,7 @@ def start_document_text_detection(
             )
 
         # If the file is not already in S3
-        if not file_source.startswith("s3://"):
+        if not isinstance(file_source, str) or not file_source.startswith("s3://"):
             # Check if the user has given us a bucket to upload to
             if not s3_upload_path:
                 raise InputError(
@@ -760,7 +760,7 @@ def start_expense_analysis(
             )
 
         # If the file is not already in S3
-        if not file_source.startswith("s3://"):
+        if not isinstance(file_source, str) or not file_source.startswith("s3://"):
             # Check if the user has given us a bucket to upload to
             if not s3_upload_path:
                 raise InputError(