sparrow-parse 0.4.8__tar.gz → 0.4.9__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 (25) hide show
  1. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/PKG-INFO +1 -1
  2. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/setup.py +1 -1
  3. sparrow-parse-0.4.9/sparrow_parse/__init__.py +1 -0
  4. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/sparrow_parse/vllm/mlx_inference.py +6 -5
  5. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/sparrow_parse.egg-info/PKG-INFO +1 -1
  6. sparrow-parse-0.4.8/sparrow_parse/__init__.py +0 -1
  7. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/README.md +0 -0
  8. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/setup.cfg +0 -0
  9. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/sparrow_parse/__main__.py +0 -0
  10. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/sparrow_parse/extractors/__init__.py +0 -0
  11. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/sparrow_parse/extractors/vllm_extractor.py +0 -0
  12. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/sparrow_parse/helpers/__init__.py +0 -0
  13. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/sparrow_parse/helpers/pdf_optimizer.py +0 -0
  14. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/sparrow_parse/processors/__init__.py +0 -0
  15. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/sparrow_parse/processors/table_structure_processor.py +0 -0
  16. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/sparrow_parse/vllm/__init__.py +0 -0
  17. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/sparrow_parse/vllm/huggingface_inference.py +0 -0
  18. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/sparrow_parse/vllm/inference_base.py +0 -0
  19. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/sparrow_parse/vllm/inference_factory.py +0 -0
  20. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/sparrow_parse/vllm/local_gpu_inference.py +0 -0
  21. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/sparrow_parse.egg-info/SOURCES.txt +0 -0
  22. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/sparrow_parse.egg-info/dependency_links.txt +0 -0
  23. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/sparrow_parse.egg-info/entry_points.txt +0 -0
  24. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/sparrow_parse.egg-info/requires.txt +0 -0
  25. {sparrow-parse-0.4.8 → sparrow-parse-0.4.9}/sparrow_parse.egg-info/top_level.txt +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: sparrow-parse
3
- Version: 0.4.8
3
+ Version: 0.4.9
4
4
  Summary: Sparrow Parse is a Python package (part of Sparrow) for parsing and extracting information from documents.
5
5
  Home-page: https://github.com/katanaml/sparrow/tree/main/sparrow-data/parse
6
6
  Author: Andrej Baranovskij
@@ -8,7 +8,7 @@ with open("requirements.txt", "r", encoding="utf-8") as fh:
8
8
 
9
9
  setup(
10
10
  name="sparrow-parse",
11
- version="0.4.8",
11
+ version="0.4.9",
12
12
  author="Andrej Baranovskij",
13
13
  author_email="andrejus.baranovskis@gmail.com",
14
14
  description="Sparrow Parse is a Python package (part of Sparrow) for parsing and extracting information from documents.",
@@ -0,0 +1 @@
1
+ __version__ = '0.4.9'
@@ -4,8 +4,9 @@ from mlx_vlm.utils import load_image
4
4
  from sparrow_parse.vllm.inference_base import ModelInference
5
5
  import os
6
6
  import json
7
- from rich import print
7
+ from rich.console import Console
8
8
 
9
+ console = Console()
9
10
 
10
11
  class MLXInference(ModelInference):
11
12
  """
@@ -20,7 +21,7 @@ class MLXInference(ModelInference):
20
21
  :param model_name: Name of the model to load.
21
22
  """
22
23
  self.model_name = model_name
23
- print(f"MLXInference initialized for model: {model_name}")
24
+ console.print(f"MLXInference initialized for model: {model_name}")
24
25
 
25
26
 
26
27
  @staticmethod
@@ -32,7 +33,7 @@ class MLXInference(ModelInference):
32
33
  :return: Tuple containing the loaded model and processor.
33
34
  """
34
35
  model, processor = load(model_name)
35
- print(f"Loaded model: {model_name}")
36
+ console.print(f"Loaded model: {model_name}")
36
37
  return model, processor
37
38
 
38
39
 
@@ -53,7 +54,7 @@ class MLXInference(ModelInference):
53
54
  formatted_json = json.loads(cleaned_text)
54
55
  return json.dumps(formatted_json, indent=2)
55
56
  except json.JSONDecodeError as e:
56
- print(f"Failed to parse JSON in MLX inference backend: {e}")
57
+ console.print(f"Failed to parse JSON in MLX inference backend: {e}")
57
58
  return output_text
58
59
 
59
60
 
@@ -121,7 +122,7 @@ class MLXInference(ModelInference):
121
122
  )
122
123
  results.append(self.process_response(response))
123
124
 
124
- print("Inference completed successfully for: ", file_path)
125
+ console.print("Inference completed successfully for: ", file_path)
125
126
 
126
127
  return results
127
128
 
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: sparrow-parse
3
- Version: 0.4.8
3
+ Version: 0.4.9
4
4
  Summary: Sparrow Parse is a Python package (part of Sparrow) for parsing and extracting information from documents.
5
5
  Home-page: https://github.com/katanaml/sparrow/tree/main/sparrow-data/parse
6
6
  Author: Andrej Baranovskij
@@ -1 +0,0 @@
1
- __version__ = '0.4.8'
File without changes
File without changes