fix release

okaris · okaris · commit fcbbc39465bc · 2025-06-23T15:46:31.000Z
diff --git a/src/inferencesh/__init__.py b/src/inferencesh/__init__.py
@@ -10,9 +10,11 @@
     ContextMessageRole,
     Message,
     ContextMessage,
-    ContextMessageWithImage,
     LLMInput,
-    LLMInputWithImage,
+    LLMOutput,
+    build_messages,
+    stream_generate,
+    timing_context,
 )
 from .utils import StorageDir, download
 
@@ -24,9 +26,11 @@
     "ContextMessageRole",
     "Message",
     "ContextMessage",
-    "ContextMessageWithImage",
     "LLMInput",
-    "LLMInputWithImage",
+    "LLMOutput",
+    "build_messages",
+    "stream_generate",
+    "timing_context",
     "StorageDir",
     "download",
 ]
