inferencesh 0.2.4__tar.gz → 0.2.6__tar.gz

{inferencesh-0.2.4/src/inferencesh.egg-info → inferencesh-0.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: inferencesh
-Version: 0.2.4
+Version: 0.2.6
 Summary: inference.sh Python SDK
 Author: Inference Shell Inc.
 Author-email: "Inference Shell Inc." <hello@inference.sh>

{inferencesh-0.2.4 → inferencesh-0.2.6}/pyproject.toml

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

{inferencesh-0.2.4 → inferencesh-0.2.6}/src/inferencesh/sdk.py

@@ -264,6 +264,18 @@ 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)
+        
+        # Create a schema that accepts either a string or the full object
+        return {
+            "oneOf": [
+                {"type": "string"},  # Accept string input
+                schema  # Accept full object input
+            ]
+        }
+
 
 class ContextMessageRole(str, Enum):
     USER = "user"
@@ -333,6 +345,16 @@ class LLMInput(BaseAppInput):
     )
 
 class LLMInputWithImage(LLMInput):
+    context: list[ContextMessageWithImage] = Field(
+        description="The context to use for the model",
+        examples=[
+            [
+                {"role": "user", "content": [{"type": "text", "text": "What is the capital of France?"}, {"type": "image", "url": "https://example.com/image.jpg"}]}, 
+                {"role": "assistant", "content": [{"type": "text", "text": "The capital of France is Paris."}]}
+            ],
+        ],
+        default=[]
+    )
     image: Optional[File] = Field(
         description="The image to use for the model",
         default=None

{inferencesh-0.2.4 → inferencesh-0.2.6/src/inferencesh.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: inferencesh
-Version: 0.2.4
+Version: 0.2.6
 Summary: inference.sh Python SDK
 Author: Inference Shell Inc.
 Author-email: "Inference Shell Inc." <hello@inference.sh>