inferencesh 0.4.15__tar.gz → 0.4.17__tar.gz

{inferencesh-0.4.15/src/inferencesh.egg-info → inferencesh-0.4.17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: inferencesh
-Version: 0.4.15
+Version: 0.4.17
 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 → inferencesh-0.4.17}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "inferencesh"
-version = "0.4.15"
+version = "0.4.17"
 description = "inference.sh Python SDK"
 authors = [
     {name = "Inference Shell Inc.", email = "hello@inference.sh"},

{inferencesh-0.4.15 → inferencesh-0.4.17}/src/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-0.4.15 → inferencesh-0.4.17}/src/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 → inferencesh-0.4.17}/src/inferencesh/models/llm.py

@@ -14,6 +14,7 @@ class ContextMessageRole(str, Enum):
     USER = "user"
     ASSISTANT = "assistant" 
     SYSTEM = "system"
+    TOOL = "tool"
 
 
 class Message(BaseAppInput):

{inferencesh-0.4.15 → inferencesh-0.4.17/src/inferencesh.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: inferencesh
-Version: 0.4.15
+Version: 0.4.17
 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 → inferencesh-0.4.17}/tests/test_sdk.py

@@ -159,4 +159,19 @@ def test_file_cleanup(monkeypatch):
         tmp_path = file._tmp_path
         assert os.path.exists(tmp_path)
         del file
-        assert not os.path.exists(tmp_path) 
+        assert not os.path.exists(tmp_path) 
+        
+def test_file_schema():
+    file = File(uri="https://example.com/test.txt")
+    print(file.model_json_schema())
+    assert file.model_json_schema() is not None
+    assert file.model_json_schema()["$id"] == "/schemas/File"
+    assert file.model_json_schema()["oneOf"] is not None
+    assert file.model_json_schema()["oneOf"][0] is not None
+    assert file.model_json_schema()["oneOf"][0]["type"] == "object"
+    assert file.model_json_schema()["oneOf"][0]["properties"] is not None
+    assert file.model_json_schema()["oneOf"][0]["properties"]["uri"] is not None
+    assert file.model_json_schema()["oneOf"][0]["properties"]["path"] is not None
+    assert file.model_json_schema()["oneOf"][0]["properties"]["content_type"] is not None
+    assert file.model_json_schema()["oneOf"][0]["properties"]["size"] is not None
+    assert file.model_json_schema()["oneOf"][0]["properties"]["filename"] is not None