inferencesh 0.4.8__tar.gz → 0.4.10__tar.gz

{inferencesh-0.4.8/src/inferencesh.egg-info → inferencesh-0.4.10}/PKG-INFO

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

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

{inferencesh-0.4.8 → inferencesh-0.4.10}/src/inferencesh/models/file.py

@@ -8,6 +8,8 @@ import hashlib
 from pathlib import Path
 from tqdm import tqdm
 
+from pydantic import GetCoreSchemaHandler, GetJsonSchemaHandler, JsonSchemaValue
+from pydantic_core import CoreSchema, core_schema
 
 class File(BaseModel):
     """A class representing a file in the inference.sh ecosystem."""
@@ -239,21 +241,24 @@ class File(BaseModel):
     @classmethod
     def model_json_schema(cls, **kwargs):
         schema = super().model_json_schema(**kwargs)
+        schema["$id"] = "/schemas/File"
         # Create a schema that accepts either a string or the full object
-        base_schema = {
-            "anyOf": [
+        return {
+            "oneOf": [
                 {"type": "string"},  # Accept string input
-                {
-                    "type": "object",
-                    "properties": schema["properties"],
-                    "title": schema.get("title", "File"),
-                    "description": "A class representing a file in the inference.sh ecosystem."
-                }
+                schema  # Accept full object input
+            ]
+        } 
+        
+    @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
             ]
         }
-        # If this is the top-level schema, return as is
-        if not kwargs.get("ref_template"):
-            return base_schema
-        # If this is being used in $defs, include it in the schema
-        schema.update(base_schema)
-        return schema
+    

{inferencesh-0.4.8 → inferencesh-0.4.10/src/inferencesh.egg-info}/PKG-INFO

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