my-aws-helpers 3.1.0.dev3__tar.gz → 3.1.0.dev4__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (27) hide show
  1. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/PKG-INFO +1 -1
  2. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers/bedrock.py +4 -0
  3. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers.egg-info/PKG-INFO +1 -1
  4. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/setup.py +1 -1
  5. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/MANIFEST.in +0 -0
  6. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/README.md +0 -0
  7. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers/api.py +0 -0
  8. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers/auth.py +0 -0
  9. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers/cognito.py +0 -0
  10. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers/dynamo.py +0 -0
  11. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers/errors.py +0 -0
  12. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers/event.py +0 -0
  13. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers/logging.py +0 -0
  14. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers/prompts/__init__.py +0 -0
  15. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers/prompts/markdown_system_prompt.txt +0 -0
  16. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers/prompts/transactions_headers_prompt.txt +0 -0
  17. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers/prompts/transactions_headers_prompt_v2.txt +0 -0
  18. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers/prompts/transactions_prompt.txt +0 -0
  19. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers/s3.py +0 -0
  20. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers/sfn.py +0 -0
  21. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers.egg-info/SOURCES.txt +0 -0
  22. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers.egg-info/dependency_links.txt +0 -0
  23. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers.egg-info/requires.txt +0 -0
  24. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers.egg-info/top_level.txt +0 -0
  25. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers.egg-info/zip-safe +0 -0
  26. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/setup.cfg +0 -0
  27. {my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/tests/test_event.py +0 -0
{my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: my_aws_helpers
3
- Version: 3.1.0.dev3
3
+ Version: 3.1.0.dev4
4
4
  Summary: AWS Helpers
5
5
  Home-page: https://github.com/JarrodMccarthy/aws_helpers.git
6
6
  Author: Jarrod McCarthy
{my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers/bedrock.py RENAMED
@@ -180,6 +180,7 @@ class Bedrock:
180
180
  except Exception as e:
181
181
  self.logger.error(f"Could not get pix map for page {i}")
182
182
  continue
183
+ skip_page_zero = False
183
184
  header_ocr_result = None
184
185
  if len(image_bytes_list) > 1:
185
186
  headers_prompt = self._get_prompt(prompt_type=PromptType.transaction_headers.value)
@@ -190,6 +191,7 @@ class Bedrock:
190
191
  self.logger.info(f"No ocr result returned when getting headers {PromptType.transaction_headers.value}")
191
192
  headers = header_ocr_result.content.get("headers")
192
193
  if (len(headers) < 1) or (headers is None):
194
+ skip_page_zero = True
193
195
  continue
194
196
  else:
195
197
  break
@@ -201,6 +203,8 @@ class Bedrock:
201
203
  self.logger.info("Got Prompt")
202
204
  results = list()
203
205
 
206
+ if skip_page_zero:
207
+ image_bytes_list = image_bytes_list[1:] #page zero often has account summary info
204
208
  results = self._parallel_ocr(image_bytes_list=image_bytes_list, prompt=transactions_prompt)
205
209
 
206
210
  # for i, image_bytes in enumerate(image_bytes_list):
{my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/my_aws_helpers.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: my-aws-helpers
3
- Version: 3.1.0.dev3
3
+ Version: 3.1.0.dev4
4
4
  Summary: AWS Helpers
5
5
  Home-page: https://github.com/JarrodMccarthy/aws_helpers.git
6
6
  Author: Jarrod McCarthy
{my_aws_helpers-3.1.0.dev3 → my_aws_helpers-3.1.0.dev4}/setup.py RENAMED
@@ -3,7 +3,7 @@ from setuptools import find_namespace_packages, setup
3
3
 
4
4
  base_path = os.path.abspath(os.path.dirname(__file__))
5
5
 
6
- version = "3.1.0.dev3"
6
+ version = "3.1.0.dev4"
7
7
 
8
8
  setup(
9
9
  name="my_aws_helpers",