Update 2026-05-13 16:43:53

2026-05-13 16:43:53 +08:00
parent 6af5c584f4
commit afd7c5fe85
490 changed files with 850 additions and 922 deletions
@@ -0,0 +1,23 @@
+from contextvars import ContextVar
+from typing import Any, Callable, Awaitable, Dict, Optional
+
+# The current session ID processing the request
+current_session_id: ContextVar[str] = ContextVar("current_session_id", default="")
+
+# A callback to send progress updates to the frontend during tool execution
+current_progress_callback: ContextVar[Optional[Callable[[str], Awaitable[None]]]] = ContextVar("current_progress_callback", default=None)
+
+# A payload dictionary to store visualization results generated by tools
+# This will be picked up by the stream handler and sent to the frontend
+current_viz_data: ContextVar[Optional[Dict[str, Any]]] = ContextVar("current_viz_data", default=None)
+
+# Store the last queried data so the Visualization Tool can access it
+current_data: ContextVar[Optional[list]] = ContextVar("current_data", default=None)
+
+# The data source requested by the user or bound to the session
+current_data_source: ContextVar[str] = ContextVar("current_data_source", default="postgres")
+
+# Any file URL attached to the request
+current_file_url: ContextVar[Optional[str]] = ContextVar("current_file_url", default=None)
+
+current_knowledge_base_id: ContextVar[Optional[str]] = ContextVar("current_knowledge_base_id", default=None)