inferencesh 0.2.12__tar.gz → 0.2.14__tar.gz

{inferencesh-0.2.12/src/inferencesh.egg-info → inferencesh-0.2.14}/PKG-INFO

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

{inferencesh-0.2.12 → inferencesh-0.2.14}/pyproject.toml

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

{inferencesh-0.2.12 → inferencesh-0.2.14}/src/inferencesh/__init__.py

@@ -10,9 +10,11 @@ from .models import (
     ContextMessageRole,
     Message,
     ContextMessage,
-    ContextMessageWithImage,
     LLMInput,
-    LLMInputWithImage,
+    LLMOutput,
+    build_messages,
+    stream_generate,
+    timing_context,
 )
 from .utils import StorageDir, download
 
@@ -24,9 +26,11 @@ __all__ = [
     "ContextMessageRole",
     "Message",
     "ContextMessage",
-    "ContextMessageWithImage",
     "LLMInput",
-    "LLMInputWithImage",
+    "LLMOutput",
+    "build_messages",
+    "stream_generate",
+    "timing_context",
     "StorageDir",
     "download",
 ]

{inferencesh-0.2.12 → inferencesh-0.2.14}/src/inferencesh/models/llm.py

@@ -5,6 +5,7 @@ from queue import Queue
 from threading import Thread
 import time
 from contextlib import contextmanager
+import base64
 
 from .base import BaseAppInput, BaseAppOutput
 from .file import File
@@ -136,6 +137,10 @@ def timing_context():
     finally:
         pass
 
+def image_to_base64_data_uri(file_path):
+    with open(file_path, "rb") as img_file:
+        base64_data = base64.b64encode(img_file.read()).decode('utf-8')
+        return f"data:image/png;base64,{base64_data}"
 
 def build_messages(
     input_data: LLMInput,
@@ -164,7 +169,8 @@ def build_messages(
             message_content.append({"type": "text", "text": text})
         if hasattr(msg, 'image') and msg.image:
             if msg.image.path:
-                message_content.append({"type": "image_url", "image_url": {"url": msg.image.path}})
+                image_data_uri = image_to_base64_data_uri(msg.image.path)
+                message_content.append({"type": "image_url", "image_url": {"url": image_data_uri}})
             elif msg.image.uri:
                 message_content.append({"type": "image_url", "image_url": {"url": msg.image.uri}})
         messages.append({
@@ -181,7 +187,8 @@ def build_messages(
         user_content.append({"type": "text", "text": text})
     if hasattr(input_data, 'image') and input_data.image:
         if input_data.image.path:
-            user_content.append({"type": "image_url", "image_url": {"url": input_data.image.path}})
+            image_data_uri = image_to_base64_data_uri(input_data.image.path)
+            user_content.append({"type": "image_url", "image_url": {"url": image_data_uri}})
         elif input_data.image.uri:
             user_content.append({"type": "image_url", "image_url": {"url": input_data.image.uri}})
     messages.append({"role": "user", "content": user_content})

{inferencesh-0.2.12 → inferencesh-0.2.14/src/inferencesh.egg-info}/PKG-INFO

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