99
1010
1111class _DocumentExtraction (GraphQLRequest ):
12-
1312 query = """
1413 mutation($files: [FileInput], $jsonConfig: JSONString, $ocrEngine: OCREngine) {
1514 documentExtraction(files: $files, jsonConfig: $jsonConfig, ocrEngine: $ocrEngine) {
@@ -22,7 +21,12 @@ def __init__(self, files, json_config={"preset_config": "legacy"}, ocr_engine=No
2221 if json_config and type (json_config ) == dict :
2322 json_config = json .dumps (json_config )
2423 super ().__init__ (
25- query = self .query , variables = {"files" : files , "jsonConfig" : json_config , "ocrEngine" : ocr_engine }
24+ query = self .query ,
25+ variables = {
26+ "files" : files ,
27+ "jsonConfig" : json_config ,
28+ "ocrEngine" : ocr_engine ,
29+ },
2630 )
2731
2832 def process_response (self , response ):
@@ -67,7 +71,11 @@ class DocumentExtraction(RequestChain):
6771 """
6872
6973 def __init__ (
70- self , files : List [str ], json_config : dict = None , upload_batch_size : int = None , ocr_engine : str = "OMNIPAGE"
74+ self ,
75+ files : List [str ],
76+ json_config : dict = None ,
77+ upload_batch_size : int = None ,
78+ ocr_engine : str = "OMNIPAGE" ,
7179 ):
7280 self .files = files
7381 self .json_config = json_config
@@ -83,4 +91,8 @@ def requests(self):
8391 )
8492 else :
8593 yield UploadDocument (files = self .files )
86- yield _DocumentExtraction (files = self .previous , json_config = self .json_config , ocr_engine = self .ocr_engine )
94+ yield _DocumentExtraction (
95+ files = self .previous ,
96+ json_config = self .json_config ,
97+ ocr_engine = self .ocr_engine ,
98+ )
0 commit comments