Fixup last import bits to get the example working

zephraph · Feb 17, 2025 · 2fe281c · 2fe281c
1 parent 79f22a2
commit 2fe281c
Showing 1 changed file with 17 additions and 5 deletions.
diff --git a/src/clients/python/src/webview_python/__init__.py b/src/clients/python/src/webview_python/__init__.py
@@ -10,6 +10,17 @@
 import msgspec
 import sys
 
+__all__ = [
+    "WebViewMessage",
+    "WebViewRequest",
+    "WebViewOptions",
+    "WebViewContentHtml",
+    "WebViewContentUrl",
+    "WebViewNotification",
+    "WebViewResponse",
+    "IpcNotification",
+]
+
 # Import schemas
 from .schemas.WebViewMessage import (
     NotificationMessage,
@@ -25,10 +36,11 @@
     ErrResponse,
     Response as WebViewResponse,
     Notification as WebViewNotification,
-    WebViewMessage,
+    Message as WebViewMessage,
+    IpcNotification,
 )
 from .schemas.WebViewRequest import (
-    WebViewRequest,
+    Request as WebViewRequest,
     GetVersionRequest,
     EvalRequest,
     SetTitleRequest,
@@ -46,9 +58,9 @@
     Size,
 )
 from .schemas.WebViewOptions import (
-    WebViewOptions,
-    WebViewContentHtml,
-    WebViewContentUrl,
+    Options as WebViewOptions,
+    ContentHtml as WebViewContentHtml,
+    ContentUrl as WebViewContentUrl,
 )
 
 # Constants