fix init

okaris · okaris · commit 49d9eb6e4e8a · 2025-06-23T14:29:15.000Z
diff --git a/src/inferencesh/models/__init__.py b/src/inferencesh/models/__init__.py
@@ -6,9 +6,11 @@
     ContextMessageRole,
     Message,
     ContextMessage,
-    ContextMessageWithImage,
     LLMInput,
-    LLMInputWithImage,
+    LLMOutput,
+    build_messages,
+    stream_generate,
+    timing_context,
 )
 
 __all__ = [
@@ -19,7 +21,9 @@
     "ContextMessageRole",
     "Message",
     "ContextMessage",
-    "ContextMessageWithImage",
     "LLMInput",
-    "LLMInputWithImage",
+    "LLMOutput",
+    "build_messages",
+    "stream_generate",
+    "timing_context",
 ] 
