Merge pull request #1132 from llmware-ai/update-parser-table-config

update table strategy config
llmware-ai · Dec 19, 2024 · ff2f0e2 · ff2f0e2
2 parents 53e34be + db27c6f
commit ff2f0e2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/llmware/parsers.py b/llmware/parsers.py
@@ -2687,8 +2687,12 @@ def parse_one_pdf (self, fp, fn,  save_history=True):
         else:
             strip_header = c_int(0)
 
+        # update - expose configs for table extraction strategy
         if self.get_tables:
-            table_extract = c_int(1)
+            if 0 <= self.table_strategy <= 2:
+                table_extract = c_int(self.table_strategy)
+            else:
+                table_extract = c_int(1)
         else:
             table_extract = c_int(0)