feat(workbench-shell): ✨ Add dynamic chart artifact rendering to thread messages

package.json

@@ -30,6 +30,7 @@
     "class-variance-authority": "^0.7.1",
     "clsx": "^2.1.1",
     "cmdk": "^1.1.1",
+    "dompurify": "^3.4.2",
     "lucide-react": "^0.542.0",
     "motion": "^12.36.0",
     "nanoid": "^5.1.6",
@@ -42,11 +43,15 @@
     "tailwind-merge": "^3.3.1",
     "tw-animate-css": "^1.3.7",
     "use-stick-to-bottom": "^1.1.3",
+    "vega": "^6.2.0",
+    "vega-embed": "^7.1.0",
+    "vega-lite": "^6.4.3",
     "xterm": "^5.3.0"
   },
   "devDependencies": {
     "@tailwindcss/vite": "^4.1.12",
     "@tauri-apps/cli": "^2",
+    "@types/dompurify": "^3.0.5",
     "@types/react": "^19.1.8",
     "@types/react-dom": "^19.1.6",
     "@vitejs/plugin-react": "^4.6.0",

src-tauri/migrations/20260506000000_message_parts_json.sql

@@ -0,0 +1 @@
+ALTER TABLE messages ADD COLUMN parts_json TEXT;

src-tauri/src/commands/thread.rs

@@ -417,6 +417,7 @@ mod tests {
             run_id: None,
             role: role.into(),
             content_markdown: content.into(),
+            parts_json: None,
             message_type: "plain_message".into(),
             status: "completed".into(),
             metadata_json: None,
@@ -436,6 +437,7 @@ mod tests {
                 run_id: None,
                 role: "system".into(),
                 content_markdown: "summary".into(),
+                parts_json: None,
                 message_type: "summary_marker".into(),
                 status: "completed".into(),
                 metadata_json: Some(serde_json::json!({ "kind": "context_summary" }).to_string()),
@@ -465,6 +467,7 @@ mod tests {
                 run_id: None,
                 role: "tool".into(),
                 content_markdown: "tool output".into(),
+                parts_json: None,
                 message_type: "plain_message".into(),
                 status: "completed".into(),
                 metadata_json: None,

src-tauri/src/core/agent_run_compaction.rs

@@ -52,6 +52,7 @@ pub(crate) async fn persist_clear_context_reset_to_pool(
             run_id: None,
             role: "user".to_string(),
             content_markdown: "/clear".to_string(),
+            parts_json: None,
             message_type: "plain_message".to_string(),
             status: "completed".to_string(),
             metadata_json: Some(command_metadata.to_string()),
@@ -64,6 +65,7 @@ pub(crate) async fn persist_clear_context_reset_to_pool(
             run_id: None,
             role: "system".to_string(),
             content_markdown: "Context is now reset".to_string(),
+            parts_json: None,
             message_type: "summary_marker".to_string(),
             status: "completed".to_string(),
             metadata_json: Some(reset_metadata.to_string()),
@@ -228,6 +230,7 @@ impl AgentRunManager {
             run_id: None,
             role: "user".to_string(),
             content_markdown: command_display_text.clone(),
+            parts_json: None,
             message_type: "plain_message".to_string(),
             status: "completed".to_string(),
             metadata_json: Some(command_metadata.to_string()),
@@ -245,6 +248,7 @@ impl AgentRunManager {
             run_id: None,
             role: "system".to_string(),
             content_markdown: "Context is now reset".to_string(),
+            parts_json: None,
             message_type: "summary_marker".to_string(),
             status: "completed".to_string(),
             metadata_json: Some(reset_metadata.to_string()),
@@ -356,6 +360,7 @@ impl AgentRunManager {
                     run_id: None,
                     role: "system".to_string(),
                     content_markdown: summary,
+                    parts_json: None,
                     message_type: "summary_marker".to_string(),
                     status: "completed".to_string(),
                     metadata_json: Some(summary_metadata.to_string()),
@@ -418,6 +423,7 @@ impl AgentRunManager {
                         run_id: None,
                         role: "system".to_string(),
                         content_markdown: heuristic_summary,
+                        parts_json: None,
                         message_type: "summary_marker".to_string(),
                         status: "completed".to_string(),
                         metadata_json: Some(summary_metadata.to_string()),

src-tauri/src/core/agent_run_event_handler.rs

@@ -360,6 +360,7 @@ impl AgentRunManager {
                 run_id: Some(run_id.to_string()),
                 role: "assistant".to_string(),
                 content_markdown: String::new(),
+                parts_json: None,
                 message_type: "plain_message".to_string(),
                 status: "streaming".to_string(),
                 metadata_json: None,
@@ -414,6 +415,7 @@ impl AgentRunManager {
                 run_id: Some(run_id.to_string()),
                 role: "assistant".to_string(),
                 content_markdown: String::new(),
+                parts_json: None,
                 message_type: "reasoning".to_string(),
                 status: "streaming".to_string(),
                 metadata_json: None,

src-tauri/src/core/agent_run_manager.rs

@@ -200,6 +200,7 @@ impl AgentRunManager {
                     run_id: None,
                     role: "user".to_string(),
                     content_markdown: display_prompt.unwrap_or_else(|| prompt.to_string()),
+                    parts_json: None,
                     message_type: "plain_message".to_string(),
                     status: "completed".to_string(),
                     metadata_json: prompt_metadata.map(|value| value.to_string()),
@@ -257,7 +258,10 @@ impl AgentRunManager {
             }
 
             let (runtime_tx, runtime_rx) = mpsc::unbounded_channel::<ThreadStreamEvent>();
-            let runtime_finish_rx = self.runtime.start_session(spec, runtime_tx).await?;
+            let runtime_finish_rx = self
+                .runtime
+                .start_session(spec, runtime_tx, Arc::clone(&self.active_runs))
+                .await?;
             self.spawn_runtime_event_loop(run_id.clone(), runtime_rx);
             self.spawn_runtime_finish_watchdog(run_id.clone(), runtime_finish_rx);
 
@@ -515,6 +519,7 @@ impl AgentRunManager {
             run_id: None,
             role: "system".to_string(),
             content_markdown: "Context is now reset".to_string(),
+            parts_json: None,
             message_type: "summary_marker".to_string(),
             status: "completed".to_string(),
             metadata_json: Some(
@@ -534,6 +539,7 @@ impl AgentRunManager {
             run_id: None,
             role: "assistant".to_string(),
             content_markdown: crate::core::plan_checkpoint::plan_markdown(plan_metadata),
+            parts_json: None,
             message_type: "plan".to_string(),
             status: "completed".to_string(),
             metadata_json: serde_json::to_string(plan_metadata).ok(),

src-tauri/src/core/agent_run_manager_tests.rs

@@ -607,6 +607,7 @@ pub(super) mod tests {
                 run_id: None,
                 role: "user".into(),
                 content_markdown: "oldest user message".into(),
+                parts_json: None,
                 message_type: "plain_message".into(),
                 status: "completed".into(),
                 metadata_json: None,
@@ -619,6 +620,7 @@ pub(super) mod tests {
                 run_id: None,
                 role: "assistant".into(),
                 content_markdown: "newer assistant reply".into(),
+                parts_json: None,
                 message_type: "plain_message".into(),
                 status: "completed".into(),
                 metadata_json: None,
@@ -631,6 +633,7 @@ pub(super) mod tests {
                 run_id: None,
                 role: "user".into(),
                 content_markdown: "newest user follow-up".into(),
+                parts_json: None,
                 message_type: "plain_message".into(),
                 status: "completed".into(),
                 metadata_json: None,
@@ -1410,6 +1413,7 @@ pub(super) mod tests {
             run_id: None,
             role: "user".into(),
             content_markdown: "请帮我分析标题生成策略".into(),
+            parts_json: None,
             message_type: "plain_message".into(),
             status: "completed".into(),
             metadata_json: None,
@@ -1432,6 +1436,7 @@ pub(super) mod tests {
             run_id: None,
             role: "user".into(),
             content_markdown: "Need a short title".into(),
+            parts_json: None,
             message_type: "plain_message".into(),
             status: "completed".into(),
             metadata_json: None,
@@ -1457,6 +1462,7 @@ pub(super) mod tests {
             run_id: None,
             role: "assistant".into(),
             content_markdown: "Let's decide whether to keep fallback behavior.".into(),
+            parts_json: None,
             message_type: "plain_message".into(),
             status: "completed".into(),
             metadata_json: None,

src-tauri/src/core/agent_session.rs

@@ -123,16 +123,28 @@ pub struct AgentSession {
     pub(crate) checkpoint_requested: AtomicBool,
     pub(crate) abort_signal: tiycore::agent::AbortSignal,
     context_compression_state: Arc<StdMutex<ContextCompressionRuntimeState>>,
+    /// Shared reference to the active-runs map so we can read the in-memory
+    /// `streaming_message_id` without querying the database (avoids race).
+    pub(crate) active_runs: Arc<
+        tokio::sync::Mutex<
+            std::collections::HashMap<String, crate::core::agent_run_manager::ActiveRun>,
+        >,
+    >,
 }
 
 impl AgentSession {
-    pub fn new(
+    pub(crate) fn new(
         pool: SqlitePool,
         tool_gateway: Arc<ToolGateway>,
         helper_orchestrator: Arc<HelperAgentOrchestrator>,
         event_tx: mpsc::UnboundedSender<ThreadStreamEvent>,
         spec: AgentSessionSpec,
         max_turns: usize,
+        active_runs: Arc<
+            tokio::sync::Mutex<
+                std::collections::HashMap<String, crate::core::agent_run_manager::ActiveRun>,
+            >,
+        >,
     ) -> Arc<Self> {
         Arc::new_cyclic(|weak_self| {
             let agent = Arc::new(Agent::with_model(spec.model_plan.primary.model.clone()));
@@ -158,6 +170,7 @@ impl AgentSession {
                 checkpoint_requested: AtomicBool::new(false),
                 abort_signal: tiycore::agent::AbortSignal::new(),
                 context_compression_state,
+                active_runs,
             }
         })
     }