inferencesh 0.4.10__tar.gz → 0.4.12__tar.gz

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

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

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

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

@@ -1,5 +1,6 @@
-from typing import Optional, Union, Any
-from pydantic import BaseModel, Field, PrivateAttr, model_validator
+from typing import Optional, Union, Any, Type
+from pydantic import BaseModel, Field, PrivateAttr, model_validator, GetCoreSchemaHandler, GetJsonSchemaHandler
+from pydantic_core import CoreSchema, core_schema
 import mimetypes
 import os
 import urllib.request
@@ -8,9 +9,6 @@ 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."""
     
@@ -238,27 +236,43 @@ class File(BaseModel):
             self.size = self._get_file_size()  # Always update size
             self.filename = self._get_filename()
 
-    @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
-        return {
-            "oneOf": [
-                {"type": "string"},  # Accept string input
-                schema  # Accept full object input
-            ]
-        } 
+    # @classmethod
+    # def __get_pydantic_core_schema__(
+    #     cls, source: Type[Any], handler: GetCoreSchemaHandler
+    # ) -> CoreSchema:
+    #     """Generates a Pydantic Core schema for validation of this File class"""
+    #     # Get the default schema for our class
+    #     schema = handler(source)
         
+    #     # Create a proper serialization schema that includes the type
+    #     serialization = core_schema.plain_serializer_function_ser_schema(
+    #         lambda x: x.uri if x.uri else x.path,
+    #         return_schema=core_schema.str_schema(),
+    #         when_used="json",
+    #     )
+        
+    #     return core_schema.json_or_python_schema(
+    #         json_schema=core_schema.union_schema([
+    #             core_schema.str_schema(),  # Accept string input
+    #             schema,  # Accept full object input
+    #         ]),
+    #         python_schema=schema,
+    #         serialization=serialization,
+    #     )
+
     @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 {
+        cls, schema: CoreSchema, handler: GetJsonSchemaHandler
+    ) -> dict[str, Any]:
+        """Customizes the JSON schema generation for this File class"""
+        json_schema = handler(schema)
+        json_schema = handler.resolve_ref_schema(json_schema)
+        json_schema.update({
+            "$id": "/schemas/File",
             "oneOf": [
                 {"type": "string"},
-                schema
+                json_schema
             ]
-        }
+        })
+        return json_schema
     

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

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