client

Author: okaris

examples/run.py

@@ -0,0 +1,44 @@
+import asyncio
+import os
+from datetime import datetime
+from typing import List
+
+from inferencesh import Inference, TaskStatus
+
+def main() -> None:
+    api_key = "YOUR_INFERENCESH_API_KEY"
+    client = Inference(api_key=api_key)
+    
+    app = "infsh/text-templating"
+
+    try:
+        result = client.run_sync(
+            {
+                "app": app,
+                "input": {
+                    "template": "{1} / {2}",
+                    "strings": [
+                        "god",
+                        "particle",
+                    ]
+                },
+                "worker_selection_mode": "private",
+            },
+        )
+        
+
+        # Print final result
+        if result.get("status") == TaskStatus.COMPLETED:
+            print(f"\n✓ task completed successfully!")
+            print(f"result: {result.get('output', {}).get('result')}")
+        else:
+            status = result.get("status")
+            status_name = TaskStatus(status).name if status is not None else "UNKNOWN"
+            print(f"\n✗ task did not complete. final status: {status_name}")
+
+    except Exception as exc:  # noqa: BLE001
+        print(f"\nerror during run_sync: {type(exc).__name__}: {exc}")
+        raise  # Re-raise to see full traceback
+
+if __name__ == "__main__":
+    main()

pyproject.toml

@@ -19,6 +19,8 @@ classifiers = [
 dependencies = [
     "pydantic>=2.0.0",
     "tqdm>=4.67.0",
+    # Required for the synchronous client and examples
+    "requests>=2.31.0",
 ]
 
 [project.urls]
@@ -42,3 +44,7 @@ test = [
     "pytest>=7.0.0",
     "pytest-cov>=4.0.0",
 ]
+async = [
+    "aiohttp>=3.9.0; python_version >= '3.8'",
+    "aiofiles>=23.2.1; python_version >= '3.8'",
+]

setup.py

@@ -10,6 +10,8 @@
     package_dir={"": "src"},
     install_requires=[
         "pydantic>=2.0.0",
+        "tqdm>=4.67.0",
+        "requests>=2.31.0",
     ],
     python_requires=">=3.7",
     classifiers=[

src/inferencesh/__init__.py

@@ -17,6 +17,7 @@
     timing_context,
 )
 from .utils import StorageDir, download
+from .client import Inference, AsyncInference, UploadFileOptions, TaskStatus
 
 __all__ = [
     "BaseApp",
@@ -33,4 +34,8 @@
     "timing_context",
     "StorageDir",
     "download",
+    "Inference",
+    "AsyncInference",
+    "UploadFileOptions",
+    "TaskStatus",
 ]