inferencesh 0.1.23__py3-none-any.whl → 0.1.24__py3-none-any.whl

inferencesh/sdk.py

@@ -1,4 +1,4 @@
-from typing import Optional, Union, ClassVar
+from typing import Optional, Union
 from pydantic import BaseModel, ConfigDict, PrivateAttr, model_validator
 import mimetypes
 import os
@@ -114,6 +114,25 @@ class File(BaseModel):
         populate_by_name=True
     )
 
+    @classmethod
+    def __get_validators__(cls):
+        # First yield the default validators
+        yield cls.validate
+
+    @classmethod
+    def validate(cls, value):
+        """Convert string values to File objects."""
+        if isinstance(value, str):
+            # If it's a string, treat it as a uri
+            return cls(uri=value)
+        elif isinstance(value, cls):
+            # If it's already a File instance, return it as is
+            return value
+        elif isinstance(value, dict):
+            # If it's a dict, use normal validation
+            return cls(**value)
+        raise ValueError(f'Invalid input for File: {value}')
+
     @model_validator(mode='after')
     def check_uri_or_path(self) -> 'File':
         """Validate that either uri or path is provided."""
@@ -170,7 +189,7 @@ class File(BaseModel):
             if tmp_file is not None and hasattr(self, '_tmp_path'):
                 try:
                     os.unlink(self._tmp_path)
-                except:
+                except (OSError, IOError):
                     pass
             raise RuntimeError(f"Error downloading URL {original_url}: {str(e)}")
 
@@ -179,7 +198,7 @@ class File(BaseModel):
         if hasattr(self, '_tmp_path') and self._tmp_path:
             try:
                 os.unlink(self._tmp_path)
-            except:
+            except (OSError, IOError):
                 pass
 
     def _populate_metadata(self) -> None:

{inferencesh-0.1.23.dist-info → inferencesh-0.1.24.dist-info}/METADATA

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

inferencesh-0.1.24.dist-info/RECORD

@@ -0,0 +1,8 @@
+inferencesh/__init__.py,sha256=hbKkgHCh0lCdhWyHs3FHHRd8JfLeHkTd1bT4v79Fi8M,192
+inferencesh/sdk.py,sha256=u8GbIAn5clg_9zHUy99zu2AidgBION4f1KcMJnmjChg,12417
+inferencesh-0.1.24.dist-info/licenses/LICENSE,sha256=OsgqEWIh2el_QMj0y8O1A5Q5Dl-dxqqYbFE6fszuR4s,1086
+inferencesh-0.1.24.dist-info/METADATA,sha256=FM54ThOJtpP14Z1-DN2ZV0Mv0g73md2u8kW15mJfSE4,2612
+inferencesh-0.1.24.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
+inferencesh-0.1.24.dist-info/entry_points.txt,sha256=6IC-fyozAqW3ljsMLGCXxJ0_ui2Jb-2fLHtoH1RTnEE,45
+inferencesh-0.1.24.dist-info/top_level.txt,sha256=TSMHg3T1ThMl1HGAWmzBClwOYH1ump5neof9BfHIwaA,12
+inferencesh-0.1.24.dist-info/RECORD,,

{inferencesh-0.1.23.dist-info → inferencesh-0.1.24.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.0.0)
+Generator: setuptools (80.3.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

inferencesh-0.1.23.dist-info/RECORD

@@ -1,8 +0,0 @@
-inferencesh/__init__.py,sha256=hbKkgHCh0lCdhWyHs3FHHRd8JfLeHkTd1bT4v79Fi8M,192
-inferencesh/sdk.py,sha256=MdGlTaZITc9nuE94jLU01L1LWsC5RKj-OpFre-x6CeI,11738
-inferencesh-0.1.23.dist-info/licenses/LICENSE,sha256=OsgqEWIh2el_QMj0y8O1A5Q5Dl-dxqqYbFE6fszuR4s,1086
-inferencesh-0.1.23.dist-info/METADATA,sha256=_S9CelUY7dd_6ECy-n3Dj-K562rN3cvQ5f1WLvXP5HQ,2612
-inferencesh-0.1.23.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
-inferencesh-0.1.23.dist-info/entry_points.txt,sha256=6IC-fyozAqW3ljsMLGCXxJ0_ui2Jb-2fLHtoH1RTnEE,45
-inferencesh-0.1.23.dist-info/top_level.txt,sha256=TSMHg3T1ThMl1HGAWmzBClwOYH1ump5neof9BfHIwaA,12
-inferencesh-0.1.23.dist-info/RECORD,,