inferencesh 0.4.15__py3-none-any.whl → 0.4.16__py3-none-any.whl

inferencesh/client.py

@@ -366,7 +366,8 @@ class Inference:
             task_id = task["id"]
             
             # Stream updates
-            for update in client.run(params, stream=True):
+            stream = client.run(params, stream=True)
+            for update in stream:
                 print(f"Status: {update.get('status')}")
                 if update.get('status') == TaskStatus.COMPLETED:
                     print(f"Result: {update.get('output')}")
@@ -794,7 +795,8 @@ class AsyncInference:
             task_id = task["id"]
             
             # Stream updates
-            async for update in await client.run(params, stream=True):
+            stream = await client.run(params, stream=True)
+            for update in stream:
                 print(f"Status: {update.get('status')}")
                 if update.get('status') == TaskStatus.COMPLETED:
                     print(f"Result: {update.get('output')}")

inferencesh/models/file.py

@@ -266,11 +266,15 @@ class File(BaseModel):
     ) -> dict[str, Any]:
         """Generate a simple JSON schema that accepts either a string or an object"""
         json_schema = handler(schema)
-        # Don't resolve refs here - that's what was causing the recursion
+        if "$ref" in json_schema:
+            # If we got a ref, resolve it to the actual schema
+            json_schema = handler.resolve_ref_schema(json_schema)
+        
+        # Add string as an alternative without recursion
         return {
             "$id": "/schemas/File",
             "oneOf": [
-                json_schema,
+                {k: v for k, v in json_schema.items() if k != "$ref"},  # Remove any $ref to prevent recursion
                 {"type": "string"}
             ]
         }

{inferencesh-0.4.15.dist-info → inferencesh-0.4.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: inferencesh
-Version: 0.4.15
+Version: 0.4.16
 Summary: inference.sh Python SDK
 Author-email: "Inference Shell Inc." <hello@inference.sh>
 Project-URL: Homepage, https://github.com/inference-sh/sdk

{inferencesh-0.4.15.dist-info → inferencesh-0.4.16.dist-info}/RECORD

@@ -1,15 +1,15 @@
 inferencesh/__init__.py,sha256=dY3l3yCkWoMtGX0gNXgxFnrprFRl6PPWjH8V7Qedx5g,772
-inferencesh/client.py,sha256=6wTCLqLq-QapvjCjMg8ZE3BQyg8iTL8hv8UU7t-oxmE,39360
+inferencesh/client.py,sha256=sMgr6vVPD84P3LpQfjmuKqlRpbgOUmWLFQ9IJM5kWzc,39410
 inferencesh/models/__init__.py,sha256=FDwcdtT6c4hbRitymjmN-hZMlQa8RbKSftkZZyjtUXA,536
 inferencesh/models/base.py,sha256=eTwRvXAjMGh6b8AUXWKSGpRyeScAkPs6bNYY8AXKSz8,5505
-inferencesh/models/file.py,sha256=35bnHcBB4Whi7Pg7HcmKUSm2FhRO6RPbmgXfhT0Op6Y,11506
+inferencesh/models/file.py,sha256=H4s-A2RWfNTMCcLqSLpqNHvGfSvYEnbZaJj-rWftrL0,11743
 inferencesh/models/llm.py,sha256=N5eP549tWYh5qlbAwMbMTcHwx-bET7OCvxJ2SlSLfVk,27948
 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.15.dist-info/licenses/LICENSE,sha256=OsgqEWIh2el_QMj0y8O1A5Q5Dl-dxqqYbFE6fszuR4s,1086
-inferencesh-0.4.15.dist-info/METADATA,sha256=DRHSLO6_-kedTgY0KBsZlPts1AfWL0bwie36dJkPKS0,5406
-inferencesh-0.4.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-inferencesh-0.4.15.dist-info/entry_points.txt,sha256=6IC-fyozAqW3ljsMLGCXxJ0_ui2Jb-2fLHtoH1RTnEE,45
-inferencesh-0.4.15.dist-info/top_level.txt,sha256=TSMHg3T1ThMl1HGAWmzBClwOYH1ump5neof9BfHIwaA,12
-inferencesh-0.4.15.dist-info/RECORD,,
+inferencesh-0.4.16.dist-info/licenses/LICENSE,sha256=OsgqEWIh2el_QMj0y8O1A5Q5Dl-dxqqYbFE6fszuR4s,1086
+inferencesh-0.4.16.dist-info/METADATA,sha256=uGxSDfljhrZy_IQG83Y9E09gbuu-cbNSikBP7TjI9rI,5406
+inferencesh-0.4.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+inferencesh-0.4.16.dist-info/entry_points.txt,sha256=6IC-fyozAqW3ljsMLGCXxJ0_ui2Jb-2fLHtoH1RTnEE,45
+inferencesh-0.4.16.dist-info/top_level.txt,sha256=TSMHg3T1ThMl1HGAWmzBClwOYH1ump5neof9BfHIwaA,12
+inferencesh-0.4.16.dist-info/RECORD,,