From 6061a7d0601c3d4bc3654de41037e127e92d0136 Mon Sep 17 00:00:00 2001
From: leavesster <11785335+leavesster@users.noreply.github.com>
Date: Sat, 31 Jan 2026 14:29:34 +0800
Subject: [PATCH] feat: export TypedDict types for type checking

Add missing TypedDict types to __all__ in context.py:
- ToNode, ToFlow: output targeting types
- QueryBlockResponse: query block response type
- FlowDownstream, NodeDownstream, Downstream: downstream query types
- OOMOL_LLM_ENV, HostInfo: environment info types

This allows users to import these types for proper type annotations.
---
 oocana/oocana/context.py | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

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: