inferencesh 0.1.9__py3-none-any.whl → 0.1.11__py3-none-any.whl

inferencesh/sdk.py

@@ -39,7 +39,6 @@ class File(BaseModel):
     def __init__(self, **data):
         super().__init__(**data)
         if self._is_url(self.path):
-            print(f"Downloading URL: {self.path}")
             self._download_url()
         elif not os.path.isabs(self.path):
             self.path = os.path.abspath(self.path)
@@ -49,26 +48,44 @@ class File(BaseModel):
         """Check if the path is a URL."""
         parsed = urllib.parse.urlparse(path)
         return parsed.scheme in ('http', 'https')
-    
     def _download_url(self) -> None:
         """Download the URL to a temporary file and update the path."""
         original_url = self.path
-        # Create a temporary file with a suffix based on the URL path
-        suffix = os.path.splitext(urllib.parse.urlparse(original_url).path)[1]
-        tmp_file = tempfile.NamedTemporaryFile(delete=False, suffix=suffix)
-        self._tmp_path = tmp_file.name
-        
-        # Set up request with user agent
-        headers = {
-            'User-Agent': 'inference.sh'
-        }
-        req = urllib.request.Request(original_url, headers=headers)
-        
-        # Download the file
-        print(f"Downloading URL: {original_url} to {self._tmp_path}")
-        with urllib.request.urlopen(req) as response, open(self._tmp_path, 'wb') as out_file:
-            out_file.write(response.read())
-        self.path = self._tmp_path
+        tmp_file = None
+        try:
+            # Create a temporary file with a suffix based on the URL path
+            suffix = os.path.splitext(urllib.parse.urlparse(original_url).path)[1]
+            tmp_file = tempfile.NamedTemporaryFile(delete=False, suffix=suffix)
+            self._tmp_path = tmp_file.name
+            
+            # Set up request with user agent
+            headers = {
+                'User-Agent': (
+                    'Mozilla/5.0 (Windows NT 10.0; Win64; x64) '
+                    'AppleWebKit/537.36 (KHTML, like Gecko) '
+                    'Chrome/91.0.4472.124 Safari/537.36'
+                )
+            }
+            req = urllib.request.Request(original_url, headers=headers)
+            
+            # Download the file
+            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())
+                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)}")
+            except IOError as e:
+                raise RuntimeError(f"Failed to write downloaded file to {self._tmp_path}: {str(e)}")
+        except Exception as e:
+            # Clean up temp file if something went wrong
+            if tmp_file is not None and hasattr(self, '_tmp_path'):
+                try:
+                    os.unlink(self._tmp_path)
+                except:
+                    pass
+            raise RuntimeError(f"Error downloading URL {original_url}: {str(e)}")
 
     def __del__(self):
         """Cleanup temporary file if it exists."""

{inferencesh-0.1.9.dist-info → inferencesh-0.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: inferencesh
-Version: 0.1.9
+Version: 0.1.11
 Summary: inference.sh Python SDK
 Author: Inference Shell Inc.
 Author-email: "Inference Shell Inc." <hello@inference.sh>

inferencesh-0.1.11.dist-info/RECORD

@@ -0,0 +1,8 @@
+inferencesh/__init__.py,sha256=pR0MXSJe41LgJkjGK-jhZR7LjqCFdRZtNTV6qcjYSTI,123
+inferencesh/sdk.py,sha256=Czs31xNdlm9LB_PdCo5Vnknn46c98votVsRbv9gHezU,4870
+inferencesh-0.1.11.dist-info/LICENSE,sha256=OsgqEWIh2el_QMj0y8O1A5Q5Dl-dxqqYbFE6fszuR4s,1086
+inferencesh-0.1.11.dist-info/METADATA,sha256=RbW4TasZukB7XGe7wZsk6rR8pwuTs_fCiaMLvecmL-k,2584
+inferencesh-0.1.11.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+inferencesh-0.1.11.dist-info/entry_points.txt,sha256=6IC-fyozAqW3ljsMLGCXxJ0_ui2Jb-2fLHtoH1RTnEE,45
+inferencesh-0.1.11.dist-info/top_level.txt,sha256=TSMHg3T1ThMl1HGAWmzBClwOYH1ump5neof9BfHIwaA,12
+inferencesh-0.1.11.dist-info/RECORD,,

inferencesh-0.1.9.dist-info/RECORD

@@ -1,8 +0,0 @@
-inferencesh/__init__.py,sha256=pR0MXSJe41LgJkjGK-jhZR7LjqCFdRZtNTV6qcjYSTI,123
-inferencesh/sdk.py,sha256=QOHT8As_SCGI93s6hd8i8tizOzSZ2bMneQCIc45IrXw,3959
-inferencesh-0.1.9.dist-info/LICENSE,sha256=OsgqEWIh2el_QMj0y8O1A5Q5Dl-dxqqYbFE6fszuR4s,1086
-inferencesh-0.1.9.dist-info/METADATA,sha256=n0_HTF7QFx9jhssc9EOJ5ByP74rWBLa7suS7Cbpn6Bc,2583
-inferencesh-0.1.9.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-inferencesh-0.1.9.dist-info/entry_points.txt,sha256=6IC-fyozAqW3ljsMLGCXxJ0_ui2Jb-2fLHtoH1RTnEE,45
-inferencesh-0.1.9.dist-info/top_level.txt,sha256=TSMHg3T1ThMl1HGAWmzBClwOYH1ump5neof9BfHIwaA,12
-inferencesh-0.1.9.dist-info/RECORD,,