sparrow-parse 0.4.8__py3-none-any.whl → 0.4.9__py3-none-any.whl

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.
sparrow_parse/__init__.py CHANGED
@@ -1 +1 @@
1
- __version__ = '0.4.8'
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,4 +1,4 @@
1
- sparrow_parse/__init__.py,sha256=peVX1WWE_zKFqMg4P-uTzqbB0sWuEULK5ffxVASwHhs,21
1
+ sparrow_parse/__init__.py,sha256=HVLYuTxwxepBhsoeROAeognEnSW-LkbZ_mSNgbAcTqA,21
2
2
  sparrow_parse/__main__.py,sha256=Xs1bpJV0n08KWOoQE34FBYn6EBXZA9HIYJKrE4ZdG78,153
3
3
  sparrow_parse/extractors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
4
4
  sparrow_parse/extractors/vllm_extractor.py,sha256=PDLgLlKiq3Bv-UOQTzX3AgxNOLcEU2EniGAXLjMC30U,7820
@@ -11,9 +11,9 @@ sparrow_parse/vllm/huggingface_inference.py,sha256=EJnG6PesGKMc_0qGPN8ufE6pSnhAg
11
11
  sparrow_parse/vllm/inference_base.py,sha256=4mwGoAY63MB4cHZpV0czTkJWEzimmiTzqqzKmLNzgjw,820
12
12
  sparrow_parse/vllm/inference_factory.py,sha256=FTM65O-dW2WZchHOrNN7_Q3-FlVoAc65iSptuuUuClM,1166
13
13
  sparrow_parse/vllm/local_gpu_inference.py,sha256=aHoJTejb5xrXjWDIGu5RBQWEyRCOBCB04sMvO2Wyvg8,628
14
- sparrow_parse/vllm/mlx_inference.py,sha256=eFyi42-ju28Eb6bzFj0xa5UUaOV68SR12vL4O3vaD9s,4899
15
- sparrow_parse-0.4.8.dist-info/METADATA,sha256=mV7l0xksEDcbwuWHEnczmSwjVm_sdDox0RHyScXAfP8,6432
16
- sparrow_parse-0.4.8.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
17
- sparrow_parse-0.4.8.dist-info/entry_points.txt,sha256=8CrvTVTTcz1YuZ8aRCYNOH15ZOAaYLlcbYX3t28HwJY,54
18
- sparrow_parse-0.4.8.dist-info/top_level.txt,sha256=n6b-WtT91zKLyCPZTP7wvne8v_yvIahcsz-4sX8I0rY,14
19
- sparrow_parse-0.4.8.dist-info/RECORD,,
14
+ sparrow_parse/vllm/mlx_inference.py,sha256=jlycb0WwpQKGkfCrYYgirOdXrqEV6RROB86dmBvAlrU,4961
15
+ sparrow_parse-0.4.9.dist-info/METADATA,sha256=vunHfk8AaOMZGvmR-5oeph45PBfJMiiYKRbSBGZ-AGA,6432
16
+ sparrow_parse-0.4.9.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
17
+ sparrow_parse-0.4.9.dist-info/entry_points.txt,sha256=8CrvTVTTcz1YuZ8aRCYNOH15ZOAaYLlcbYX3t28HwJY,54
18
+ sparrow_parse-0.4.9.dist-info/top_level.txt,sha256=n6b-WtT91zKLyCPZTP7wvne8v_yvIahcsz-4sX8I0rY,14
19
+ sparrow_parse-0.4.9.dist-info/RECORD,,