Propagate session ID in LLM and MCP requests (#5165)

2025-10-31 18:18:20 +08:00
parent bcbcd33cfb
commit 0515c76e84
10 changed files with 449 additions and 19 deletions
@@ -164,10 +164,15 @@ fn get_agent_messages(
        } else {
            None
        };
-        let mut stream = agent
-            .reply(conversation.clone(), session_config, None)
-            .await
-            .map_err(|e| anyhow!("Failed to get reply from agent: {}", e))?;
+
+        let session_id = session_config.as_ref().map(|s| s.id.clone());
+        let mut stream = crate::session_context::with_session_id(session_id, async {
+            agent
+                .reply(conversation.clone(), session_config, None)
+                .await
+        })
+        .await
+        .map_err(|e| anyhow!("Failed to get reply from agent: {}", e))?;
        while let Some(message_result) = stream.next().await {
            match message_result {
                Ok(AgentEvent::Message(msg)) => conversation.push(msg),