GeneralBots/botserver
2
1
Fork 0
Code Activity Actions 4

fix: ensure websocket_session_id and channel context are set before tool execution so TALK can route messages to frontend

Browse source 
Co-authored-by: Qwen-Coder <qwen-coder@alibabacloud.com>
This commit is contained in:
Rodrigo Rodriguez (Pragmatismo) 2026-04-13 18:04:02 -03:00
parent 5a24137a5b
commit 99572f0dc5
1 changed files with 37 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
src/core/bot/tool_executor.rs
View file

@ -339,12 +339,48 @@ impl ToolExecutor {
user_id: &Uuid,
) -> ToolExecutionResult {
let tool_call_id = format!("direct_{}", Uuid::new_v4());
info!(
"[TOOL_EXEC] Direct tool invocation: '{}' for bot '{}', session '{}'",
tool_name, bot_name, session_id
);
// Ensure websocket_session_id and channel are set on the session so TALK can route correctly
{
let mut sm = state.session_manager.lock().await;
if let Ok(Some(sess)) = sm.get_session_by_id(*session_id) {
let needs_update = if let serde_json::Value::Object(ref map) = sess.context_data {
!map.contains_key("websocket_session_id") || !map.contains_key("channel")
} else {
true
};
if needs_update {
let mut updated = sess.clone();
if let serde_json::Value::Object(ref mut map) = updated.context_data {
if !map.contains_key("websocket_session_id") {
map.insert(
"websocket_session_id".to_string(),
serde_json::Value::String(session_id.to_string()),
);
}
if !map.contains_key("channel") {
map.insert(
"channel".to_string(),
serde_json::Value::String("web".to_string()),
);
}
} else {
let mut map = serde_json::Map::new();
map.insert("websocket_session_id".to_string(), serde_json::Value::String(session_id.to_string()));
map.insert("channel".to_string(), serde_json::Value::String("web".to_string()));
updated.context_data = serde_json::Value::Object(map);
}
let context_json = serde_json::to_string(&updated.context_data).unwrap_or_default();
let _ = sm.update_session_context(session_id, user_id, context_json);
}
}
}
let tool_call = ParsedToolCall {
id: tool_call_id.clone(),
tool_name: tool_name.to_string(),