Skip to content

Commit 6fdfb49

Browse files
authored
Merge pull request #203 from edenai/fix/rm-mindee-query-params
[fix] rm obsolete queryparams mindee api
2 parents 3707ad0 + 61638ca commit 6fdfb49

File tree

1 file changed

+13
-24
lines changed

1 file changed

+13
-24
lines changed

edenai_apis/apis/mindee/mindee_api.py

Lines changed: 13 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import json
22
from collections import defaultdict
33
from io import BufferedReader
4-
from typing import Dict, Optional, Sequence, TypeVar
4+
from typing import Dict, Optional, Sequence, TypeVar, TypedDict
55

66
import requests
77

@@ -61,8 +61,9 @@
6161
AsyncResponseType,
6262
)
6363

64-
ParamsApi = TypeVar("ParamsApi")
65-
64+
class RequestParams(TypedDict):
65+
headers: Dict[str, str]
66+
files: Dict[str, BufferedReader]
6667

6768
class MindeeApi(ProviderInterface, OcrInterface):
6869
provider_name = "mindee"
@@ -90,32 +91,22 @@ def __init__(self, api_keys: Dict = {}) -> None:
9091
)
9192

9293
def _get_api_attributes(
93-
self, file: BufferedReader, language: Optional[str] = None
94-
) -> ParamsApi:
95-
params: ParamsApi = {
96-
"headers": {"Authorization": self.api_key},
97-
"files": {"document": file},
98-
"params": {
99-
"local": {
100-
"langage": language.split("-")[0],
101-
"country": language.split("-")[1],
102-
}
103-
}
104-
if language
105-
else None,
106-
}
107-
return params
94+
self, file: BufferedReader
95+
) -> RequestParams:
96+
return RequestParams(
97+
headers={"Authorization": self.api_key},
98+
files={"document": file},
99+
)
108100

109101
def ocr__receipt_parser(
110102
self, file: str, language: str, file_url: str = ""
111103
) -> ResponseType[ReceiptParserDataClass]:
112104
file_ = open(file, "rb")
113-
args = self._get_api_attributes(file_, language)
105+
args = self._get_api_attributes(file_)
114106
response = requests.post(
115107
self.url_receipt,
116108
headers=args["headers"],
117109
files=args["files"],
118-
params=args["params"],
119110
)
120111
original_response = response.json()
121112

@@ -210,8 +201,7 @@ def ocr__invoice_parser(
210201
}
211202
file_ = open(file, "rb")
212203
files = {"document": file_}
213-
params = {"locale": {"language": language}}
214-
response = requests.post(self.url, headers=headers, files=files, params=params)
204+
response = requests.post(self.url, headers=headers, files=files)
215205
original_response = response.json()
216206

217207
file_.close()
@@ -546,9 +536,8 @@ def ocr__financial_parser(
546536
}
547537
file_ = open(file, "rb")
548538
files = {"document": file_}
549-
params = {"locale": {"language": language}}
550539
response = requests.post(
551-
self.url_financial, headers=headers, files=files, params=params
540+
self.url_financial, headers=headers, files=files
552541
)
553542
original_response = response.json()
554543

0 commit comments

Comments
 (0)