inferencesh 0.1.4__py3-none-any.whl

inferencesh/__init__.py

@@ -0,0 +1,5 @@
+"""inference.sh Python SDK package."""
+
+__version__ = "0.1.2"
+
+from .sdk import BaseApp, BaseAppInput, BaseAppOutput, File 

inferencesh/sdk.py

@@ -0,0 +1,75 @@
+from typing import Optional, Union
+from pydantic import BaseModel, ConfigDict
+import mimetypes
+import os
+
+class BaseAppInput(BaseModel):
+    pass
+
+class BaseAppOutput(BaseModel):
+    pass
+
+class BaseApp(BaseModel):
+    model_config = ConfigDict(arbitrary_types_allowed=True)
+    extra = "allow"
+    async def setup(self):
+        pass
+
+    async def run(self, app_input: BaseAppInput) -> BaseAppOutput:
+        raise NotImplementedError("run method must be implemented")
+
+    async def unload(self):
+        pass
+
+
+class File(BaseModel):
+    """A class representing a file in the inference.sh ecosystem."""
+    path: str  # Absolute path to the file
+    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
+    
+    def __init__(self, **data):
+        super().__init__(**data)
+        if not os.path.isabs(self.path):
+            self.path = os.path.abspath(self.path)
+        self._populate_metadata()
+    
+    def _populate_metadata(self) -> None:
+        """Populate file metadata from the path if it exists."""
+        if os.path.exists(self.path):
+            if not self.mime_type:
+                self.mime_type = self._guess_mime_type()
+            if not self.size:
+                self.size = self._get_file_size()
+            if not self.filename:
+                self.filename = self._get_filename()
+    
+    @classmethod
+    def from_path(cls, path: Union[str, os.PathLike]) -> 'File':
+        """Create a File instance from a file path."""
+        return cls(path=str(path))
+    
+    def _guess_mime_type(self) -> Optional[str]:
+        """Guess the MIME type of the file."""
+        return mimetypes.guess_type(self.path)[0]
+    
+    def _get_file_size(self) -> int:
+        """Get the size of the file in bytes."""
+        return os.path.getsize(self.path)
+    
+    def _get_filename(self) -> str:
+        """Get the base filename from the path."""
+        return os.path.basename(self.path)
+    
+    def exists(self) -> bool:
+        """Check if the file exists."""
+        return os.path.exists(self.path)
+    
+    def refresh_metadata(self) -> None:
+        """Refresh all metadata from the file."""
+        self._populate_metadata()
+    
+    class Config:
+        """Pydantic config"""
+        arbitrary_types_allowed = True 

inferencesh-0.1.4.dist-info/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Inference Shell Inc.
+
+This software is licensed for use only in conjunction with inference.sh services and platforms.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software to use it with inference.sh services, subject to the following conditions:
+
+1. The software may only be used in conjunction with inference.sh services and platforms.
+2. Redistribution and modification are permitted only for use with inference.sh services.
+3. This copyright notice and permission notice shall be included in all copies or
+   substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

inferencesh-0.1.4.dist-info/METADATA

@@ -0,0 +1,100 @@
+Metadata-Version: 2.2
+Name: inferencesh
+Version: 0.1.4
+Summary: inference.sh Python SDK
+Author: Inference Shell Inc.
+Author-email: "Inference Shell Inc." <hello@inference.sh>
+Project-URL: Homepage, https://github.com/inference-sh/sdk
+Project-URL: Bug Tracker, https://github.com/inference-sh/sdk/issues
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: pydantic>=2.0.0
+Dynamic: author
+Dynamic: requires-python
+
+# inference.sh CLI
+
+Helper package for inference.sh Python applications.
+
+## Installation
+
+```bash
+pip install infsh
+```
+
+## File Handling
+
+The `File` class provides a standardized way to handle files in the inference.sh ecosystem:
+
+```python
+from infsh import File
+
+# Basic file creation
+file = File(path="/path/to/file.png")
+
+# File with explicit metadata
+file = File(
+    path="/path/to/file.png",
+    mime_type="image/png",
+    filename="custom_name.png",
+    size=1024  # in bytes
+)
+
+# Create from path (automatically populates metadata)
+file = File.from_path("/path/to/file.png")
+
+# Check if file exists
+exists = file.exists()
+
+# Access file metadata
+print(file.mime_type)  # automatically detected if not specified
+print(file.size)       # file size in bytes
+print(file.filename)   # basename of the file
+
+# Refresh metadata (useful if file has changed)
+file.refresh_metadata()
+```
+
+The `File` class automatically handles:
+- MIME type detection
+- File size calculation
+- Filename extraction from path
+- File existence checking
+
+## Creating an App
+
+To create an inference app, inherit from `BaseApp` and define your input/output types:
+
+```python
+from infsh import BaseApp, BaseAppInput, BaseAppOutput, File
+
+class AppInput(BaseAppInput):
+    image: str  # URL or file path to image
+    mask: str   # URL or file path to mask
+
+class AppOutput(BaseAppOutput):
+    image: File
+
+class MyApp(BaseApp):
+    async def setup(self):
+        # Initialize your model here
+        pass
+
+    async def run(self, app_input: AppInput) -> AppOutput:
+        # Process input and return output
+        result_path = "/tmp/result.png"
+        return AppOutput(image=File(path=result_path))
+
+    async def unload(self):
+        # Clean up resources
+        pass
+```
+
+The app lifecycle has three main methods:
+- `setup()`: Called when the app starts, use it to initialize models
+- `run()`: Called for each inference request
+- `unload()`: Called when shutting down, use it to free resources

inferencesh-0.1.4.dist-info/RECORD

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

inferencesh-0.1.4.dist-info/WHEEL

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

inferencesh-0.1.4.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+infsh = infsh.__main__:cli

inferencesh-0.1.4.dist-info/top_level.txt

@@ -0,0 +1 @@
+inferencesh