inferencesh 0.2.0__tar.gz → 0.2.1__tar.gz

{inferencesh-0.2.0/src/inferencesh.egg-info → inferencesh-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: inferencesh
-Version: 0.2.0
+Version: 0.2.1
 Summary: inference.sh Python SDK
 Author: Inference Shell Inc.
 Author-email: "Inference Shell Inc." <hello@inference.sh>
@@ -13,6 +13,7 @@ Requires-Python: >=3.7
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: pydantic>=2.0.0
+Requires-Dist: tqdm>=4.67.0
 Provides-Extra: test
 Requires-Dist: pytest>=7.0.0; extra == "test"
 Requires-Dist: pytest-cov>=4.0.0; extra == "test"

{inferencesh-0.2.0 → inferencesh-0.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "inferencesh"
-version = "0.2.0"
+version = "0.2.1"
 description = "inference.sh Python SDK"
 authors = [
     {name = "Inference Shell Inc.", email = "hello@inference.sh"},
@@ -18,6 +18,7 @@ classifiers = [
 ]
 dependencies = [
     "pydantic>=2.0.0",
+    "tqdm>=4.67.0",
 ]
 
 [project.urls]

{inferencesh-0.2.0 → inferencesh-0.2.1}/src/inferencesh/sdk.py

@@ -15,6 +15,7 @@ from enum import Enum
 import shutil
 from pathlib import Path
 import hashlib
+from tqdm import tqdm
 
 
 # inspired by https://github.com/pydantic/pydantic/issues/7580
@@ -125,7 +126,6 @@ class File(BaseModel):
     @model_validator(mode='before')
     @classmethod
     def convert_str_to_file(cls, values):
-        print(f"check_uri_or_path input: {values}")
         if isinstance(values, str):  # Only accept strings
             return {"uri": values}
         elif isinstance(values, dict):
@@ -176,11 +176,22 @@ class File(BaseModel):
             }
             req = urllib.request.Request(original_url, headers=headers)
             
-            # Download the file
+            # Download the file with progress bar
             print(f"Downloading URL: {original_url} to {self._tmp_path}")
             try:
-                with urllib.request.urlopen(req) as response, open(self._tmp_path, 'wb') as out_file:
-                    out_file.write(response.read())
+                with urllib.request.urlopen(req) as response:
+                    total_size = int(response.headers.get('content-length', 0))
+                    block_size = 1024  # 1 Kibibyte
+                    
+                    with tqdm(total=total_size, unit='iB', unit_scale=True) as pbar:
+                        with open(self._tmp_path, 'wb') as out_file:
+                            while True:
+                                buffer = response.read(block_size)
+                                if not buffer:
+                                    break
+                                out_file.write(buffer)
+                                pbar.update(len(buffer))
+                            
                 self.path = self._tmp_path
             except (urllib.error.URLError, urllib.error.HTTPError) as e:
                 raise RuntimeError(f"Failed to download URL {original_url}: {str(e)}")

{inferencesh-0.2.0 → inferencesh-0.2.1/src/inferencesh.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: inferencesh
-Version: 0.2.0
+Version: 0.2.1
 Summary: inference.sh Python SDK
 Author: Inference Shell Inc.
 Author-email: "Inference Shell Inc." <hello@inference.sh>
@@ -13,6 +13,7 @@ Requires-Python: >=3.7
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: pydantic>=2.0.0
+Requires-Dist: tqdm>=4.67.0
 Provides-Extra: test
 Requires-Dist: pytest>=7.0.0; extra == "test"
 Requires-Dist: pytest-cov>=4.0.0; extra == "test"

{inferencesh-0.2.0 → inferencesh-0.2.1}/src/inferencesh.egg-info/requires.txt

@@ -1,4 +1,5 @@
 pydantic>=2.0.0
+tqdm>=4.67.0
 
 [test]
 pytest>=7.0.0