inferencesh 0.4.1__py3-none-any.whl → 0.4.2__py3-none-any.whl

inferencesh/models/file.py

@@ -4,7 +4,6 @@ import mimetypes
 import os
 import urllib.request
 import urllib.parse
-import tempfile
 import hashlib
 from pathlib import Path
 from tqdm import tqdm
@@ -119,12 +118,10 @@ class File(BaseModel):
             return
             
         print(f"Downloading URL: {original_url} to {cache_path}")
-        tmp_file = None
         try:
-            # Download to temporary file first to avoid partial downloads in cache
-            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 to a temporary filename in the final directory
+            tmp_path = str(cache_path) + '.tmp'
+            self._tmp_path = tmp_path
             
             # Set up request with user agent
             headers = {
@@ -176,8 +173,8 @@ class File(BaseModel):
                                     # If we read the whole body at once, exit loop
                                     break
                             
-                # Move the temporary file to the cache location
-                os.replace(self._tmp_path, cache_path)
+                # Rename the temporary file to the final name
+                os.rename(self._tmp_path, cache_path)
                 self._tmp_path = None  # Prevent deletion in __del__
                 self.path = str(cache_path)
             except (urllib.error.URLError, urllib.error.HTTPError) as e:
@@ -186,7 +183,7 @@ class File(BaseModel):
                 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'):
+            if hasattr(self, '_tmp_path') and self._tmp_path:
                 try:
                     os.unlink(self._tmp_path)
                 except (OSError, IOError):

{inferencesh-0.4.1.dist-info → inferencesh-0.4.2.dist-info}/METADATA

@@ -1,8 +1,7 @@
 Metadata-Version: 2.4
 Name: inferencesh
-Version: 0.4.1
+Version: 0.4.2
 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
@@ -21,9 +20,7 @@ Requires-Dist: pytest-cov>=4.0.0; extra == "test"
 Provides-Extra: async
 Requires-Dist: aiohttp>=3.9.0; python_version >= "3.8" and extra == "async"
 Requires-Dist: aiofiles>=23.2.1; python_version >= "3.8" and extra == "async"
-Dynamic: author
 Dynamic: license-file
-Dynamic: requires-python
 
 # inference.sh sdk
 
@@ -38,47 +35,91 @@ pip install infsh
 ## client usage
 
 ```python
-from infsh import Inference, TaskStatus
+from inferencesh import Inference, TaskStatus
 
-# create client
+# Create client
 client = Inference(api_key="your-api-key")
 
-# simple usage - wait for result
-result = client.run({
-    "app": "your-app",
-    "input": {"key": "value"},
-    "variant": "default"
-})
-print(f"output: {result['output']}")
-
-# get task info without waiting
-task = client.run(params, wait=False)
-print(f"task id: {task['id']}")
-
-# stream updates (recommended)
-for update in client.run(params, stream=True):
-    status = update.get("status")
-    print(f"status: {TaskStatus(status).name}")
+# Simple synchronous usage
+try:
+    task = client.run({
+        "app": "your-app",
+        "input": {"key": "value"},
+        "infra": "cloud",
+        "variant": "default"
+    })
     
-    if status == TaskStatus.COMPLETED:
-        print(f"output: {update.get('output')}")
-        break
-    elif status == TaskStatus.FAILED:
-        print(f"error: {update.get('error')}")
-        break
-
-# async support
+    print(f"Task ID: {task.get('id')}")
+
+    if task.get("status") == TaskStatus.COMPLETED:
+        print("✓ Task completed successfully!")
+        print(f"Output: {task.get('output')}")
+    else:
+        status = task.get("status")
+        status_name = TaskStatus(status).name if status is not None else "UNKNOWN"
+        print(f"✗ Task did not complete. Final status: {status_name}")
+
+except Exception as exc:
+    print(f"Error: {type(exc).__name__}: {exc}")
+    raise  # Re-raise to see full traceback
+
+# Streaming updates (recommended)
+try:
+    for update in client.run(
+        {
+            "app": "your-app",
+            "input": {"key": "value"},
+            "infra": "cloud",
+            "variant": "default"
+        },
+        stream=True  # Enable streaming updates
+    ):
+        status = update.get("status")
+        status_name = TaskStatus(status).name if status is not None else "UNKNOWN"
+        print(f"Status: {status_name}")
+        
+        if status == TaskStatus.COMPLETED:
+            print("✓ Task completed!")
+            print(f"Output: {update.get('output')}")
+            break
+        elif status == TaskStatus.FAILED:
+            print(f"✗ Task failed: {update.get('error')}")
+            break
+        elif status == TaskStatus.CANCELLED:
+            print("✗ Task was cancelled")
+            break
+
+except Exception as exc:
+    print(f"Error: {type(exc).__name__}: {exc}")
+    raise  # Re-raise to see full traceback
+
+# Async support
 async def run_async():
-    from infsh import AsyncInference
+    from inferencesh import AsyncInference
     
     client = AsyncInference(api_key="your-api-key")
     
-    # simple usage
-    result = await client.run(params)
+    # Simple usage
+    result = await client.run({
+        "app": "your-app",
+        "input": {"key": "value"},
+        "infra": "cloud",
+        "variant": "default"
+    })
     
-    # stream updates
-    async for update in await client.run(params, stream=True):
-        print(f"status: {TaskStatus(update['status']).name}")
+    # Stream updates
+    async for update in await client.run(
+        {
+            "app": "your-app",
+            "input": {"key": "value"},
+            "infra": "cloud",
+            "variant": "default"
+        },
+        stream=True
+    ):
+        status = update.get("status")
+        status_name = TaskStatus(status).name if status is not None else "UNKNOWN"
+        print(f"Status: {status_name}")
 ```
 
 ## file handling

{inferencesh-0.4.1.dist-info → inferencesh-0.4.2.dist-info}/RECORD

@@ -2,14 +2,14 @@ inferencesh/__init__.py,sha256=dY3l3yCkWoMtGX0gNXgxFnrprFRl6PPWjH8V7Qedx5g,772
 inferencesh/client.py,sha256=6wTCLqLq-QapvjCjMg8ZE3BQyg8iTL8hv8UU7t-oxmE,39360
 inferencesh/models/__init__.py,sha256=FDwcdtT6c4hbRitymjmN-hZMlQa8RbKSftkZZyjtUXA,536
 inferencesh/models/base.py,sha256=4gZQRi8J7y9U6PrGD9pRIehd1MJVJAqGakPQDs2AKFM,3251
-inferencesh/models/file.py,sha256=uh1czgk0KFl_9RHTODX0PkdnI42MSU8QMJR_I4lVKI4,10556
+inferencesh/models/file.py,sha256=V8p5JwCzrXQMSHPsXOf8eeGxTHLFQpQqpG7AL3v0wKo,10374
 inferencesh/models/llm.py,sha256=GLcEkDizBbgcfc-zC719wDe44th3EGf3FpKERjIAPE8,27755
 inferencesh/utils/__init__.py,sha256=-xiD6uo2XzcrPAWFb_fUbaimmnW4KFKc-8IvBzaxNd4,148
 inferencesh/utils/download.py,sha256=DRGBudiPVa5bDS35KfR-DYeGRk7gO03WOelnisecwMo,1815
 inferencesh/utils/storage.py,sha256=E4J8emd4eFKdmdDgAqzz3TpaaDd3n0l8gYlMHuY8yIU,519
-inferencesh-0.4.1.dist-info/licenses/LICENSE,sha256=OsgqEWIh2el_QMj0y8O1A5Q5Dl-dxqqYbFE6fszuR4s,1086
-inferencesh-0.4.1.dist-info/METADATA,sha256=n6TOxwlvMj3fgl4Bsuq5rSyllKhI6dJT5reXXtCQQRk,4084
-inferencesh-0.4.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-inferencesh-0.4.1.dist-info/entry_points.txt,sha256=6IC-fyozAqW3ljsMLGCXxJ0_ui2Jb-2fLHtoH1RTnEE,45
-inferencesh-0.4.1.dist-info/top_level.txt,sha256=TSMHg3T1ThMl1HGAWmzBClwOYH1ump5neof9BfHIwaA,12
-inferencesh-0.4.1.dist-info/RECORD,,
+inferencesh-0.4.2.dist-info/licenses/LICENSE,sha256=OsgqEWIh2el_QMj0y8O1A5Q5Dl-dxqqYbFE6fszuR4s,1086
+inferencesh-0.4.2.dist-info/METADATA,sha256=7pTRbdMbhqbSc9xGqrwasgYkuHuS1NlcbrN0izdRNvk,5405
+inferencesh-0.4.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+inferencesh-0.4.2.dist-info/entry_points.txt,sha256=6IC-fyozAqW3ljsMLGCXxJ0_ui2Jb-2fLHtoH1RTnEE,45
+inferencesh-0.4.2.dist-info/top_level.txt,sha256=TSMHg3T1ThMl1HGAWmzBClwOYH1ump5neof9BfHIwaA,12
+inferencesh-0.4.2.dist-info/RECORD,,