inferencesh 0.4.7__tar.gz → 0.4.9__tar.gz

{inferencesh-0.4.7/src/inferencesh.egg-info → inferencesh-0.4.9}/PKG-INFO

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

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

{inferencesh-0.4.7 → inferencesh-0.4.9}/src/inferencesh/models/file.py

@@ -7,6 +7,8 @@ import urllib.parse
 import hashlib
 from pathlib import Path
 from tqdm import tqdm
+from pydantic_core import core_schema
+from pydantic_core.core_schema import GetJsonSchemaHandler, JsonSchemaValue
 
 
 class File(BaseModel):
@@ -242,10 +244,21 @@ class File(BaseModel):
         schema["$id"] = "/schemas/File"
         # Create a schema that accepts either a string or the full object
         return {
-            "anyOf": [
+            "oneOf": [
                 {"type": "string"},  # Accept string input
                 schema  # Accept full object input
-            ],
-            "title": "File",
-            "description": "A class representing a file in the inference.sh ecosystem."
-        } 
+            ]
+        } 
+        
+    @classmethod
+    def __get_pydantic_json_schema__(
+        cls, core_schema: core_schema.CoreSchema, handler: GetJsonSchemaHandler
+    ) -> JsonSchemaValue:
+        schema = handler(core_schema)  # this is the "normal" schema
+        return {
+            "oneOf": [
+                {"type": "string"},
+                schema
+            ]
+        }
+    

{inferencesh-0.4.7 → inferencesh-0.4.9/src/inferencesh.egg-info}/PKG-INFO

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