inferencesh 0.1.5__py3-none-any.whl → 0.1.7__py3-none-any.whl

inferencesh/sdk.py

@@ -1,7 +1,10 @@
-from typing import Optional, Union
+from typing import Optional, Union, ClassVar
 from pydantic import BaseModel, ConfigDict
 import mimetypes
 import os
+import urllib.request
+import urllib.parse
+import tempfile
 
 class BaseAppInput(BaseModel):
     pass
@@ -10,7 +13,11 @@ class BaseAppOutput(BaseModel):
     pass
 
 class BaseApp(BaseModel):
-    model_config = ConfigDict(arbitrary_types_allowed=True, extra='allow')
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra='allow'
+    )
+
     async def setup(self):
         pass
 
@@ -23,17 +30,45 @@ class BaseApp(BaseModel):
 
 class File(BaseModel):
     """A class representing a file in the inference.sh ecosystem."""
-    path: str  # Absolute path to the file
+    path: str  # Absolute path to the file or URL
     mime_type: Optional[str] = None  # MIME type of the file
     size: Optional[int] = None  # File size in bytes
     filename: Optional[str] = None  # Original filename if available
+    _tmp_path: Optional[str] = None  # Internal storage for temporary file path
     
     def __init__(self, **data):
         super().__init__(**data)
-        if not os.path.isabs(self.path):
+        if self._is_url(self.path):
+            self._download_url()
+        elif not os.path.isabs(self.path):
             self.path = os.path.abspath(self.path)
         self._populate_metadata()
     
+    def _is_url(self, path: str) -> bool:
+        """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
+        
+        # Download the file
+        urllib.request.urlretrieve(original_url, self._tmp_path)
+        self.path = self._tmp_path
+
+    def __del__(self):
+        """Cleanup temporary file if it exists."""
+        if hasattr(self, '_tmp_path') and self._tmp_path:
+            try:
+                os.unlink(self._tmp_path)
+            except:
+                pass
+
     def _populate_metadata(self) -> None:
         """Populate file metadata from the path if it exists."""
         if os.path.exists(self.path):

{inferencesh-0.1.5.dist-info → inferencesh-0.1.7.dist-info}/METADATA

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

inferencesh-0.1.7.dist-info/RECORD

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

inferencesh-0.1.5.dist-info/RECORD

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