diff --git a/oocana/oocana/context.py b/oocana/oocana/context.py
index 40f4cac..48d7c62 100644
--- a/oocana/oocana/context.py
+++ b/oocana/oocana/context.py
@@ -16,7 +16,21 @@
 import logging
 import hashlib
 
-__all__ = ["Context", "HandleDefDict", "BlockJob", "BlockExecuteException"]
+__all__ = [
+    "Context",
+    "HandleDefDict",
+    "BlockJob",
+    "BlockExecuteException",
+    # TypedDict exports for type checking
+    "ToNode",
+    "ToFlow",
+    "QueryBlockResponse",
+    "FlowDownstream",
+    "NodeDownstream",
+    "Downstream",
+    "OOMOL_LLM_ENV",
+    "HostInfo",
+]
 
 
 def string_hash(text: str) -> str: