diff --git a/any_parser/base.py b/any_parser/base.py index ec1eec2..7b0cd83 100644 --- a/any_parser/base.py +++ b/any_parser/base.py @@ -11,7 +11,7 @@ CAMBIO_PARSE_URL = ( "https://qreije6m7l.execute-api.us-west-2.amazonaws.com/v1/cambio_api/parse" ) -COMBIO_INSTRUCT_URL = ( +CAMBIO_INSTRUCT_URL = ( "https://qreije6m7l.execute-api.us-west-2.amazonaws.com/v1/cambio_api/instruction" ) @@ -21,7 +21,7 @@ def __init__(self, apiKey) -> None: self._uploadurl = CAMBIO_UPLOAD_URL self._extracturl = CAMBIO_EXTRACT_URL self._parseurl = CAMBIO_PARSE_URL - self._instructurl = COMBIO_INSTRUCT_URL + self._instructurl = CAMBIO_INSTRUCT_URL self._request_header = {"x-api-key": apiKey} def setAPIKey(self, apiKey): @@ -95,8 +95,7 @@ def _request_info_extraction(self, user_id, job_id, s3_key, mode, prompt=""): "userId": user_id, "jobId": job_id, "fileKey": s3_key, - "user_prompt": prompt, - "use_textract": "True" if mode == "advanced" else "False", + "userPrompt": prompt, } response = requests.post( self._parseurl, headers=self._request_header, json=payload @@ -115,8 +114,7 @@ def _request_instruction_extraction(self, user_id, job_id, s3_key, mode, prompt= "userId": user_id, "jobId": job_id, "fileKey": s3_key, - "user_prompt": prompt, - "use_textract": "True" if mode == "advanced" else "False", + "userPrompt": prompt, } response = requests.post( self._instructurl, headers=self._request_header, json=payload