File tree Expand file tree Collapse file tree
crates/zenii-core/src/tools Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -150,11 +150,19 @@ impl Tool for ChannelSendTool {
150150 match self . registry . send ( channel, msg. clone ( ) ) . await {
151151 Ok ( ( ) ) => {
152152 // Persist outgoing message in channel session + notify frontend
153- let recipient_id = msg. metadata . get ( recipient_metadata_key ( channel) ) . cloned ( ) ;
153+ let recipient_id =
154+ msg. metadata . get ( recipient_metadata_key ( channel) ) . cloned ( ) ;
154155 if let Some ( ref rid) = recipient_id {
155156 let channel_key = format ! ( "{channel}:{rid}" ) ;
156- if let Ok ( sid) = self . session_map . resolve_session ( & channel_key, channel) . await {
157- let _ = self . session_manager . append_message ( & sid, "assistant" , message) . await ;
157+ if let Ok ( sid) = self
158+ . session_map
159+ . resolve_session ( & channel_key, channel)
160+ . await
161+ {
162+ let _ = self
163+ . session_manager
164+ . append_message ( & sid, "assistant" , message)
165+ . await ;
158166 let _ = self . event_bus . publish ( AppEvent :: ChannelMessageReceived {
159167 channel : channel. to_string ( ) ,
160168 sender : "agent" . to_string ( ) ,
Original file line number Diff line number Diff line change @@ -105,7 +105,7 @@ describe("WebSocket chat streaming", () => {
105105 const ws = MockWebSocket . instances [ 0 ] ;
106106 ws . onmessage ! ( { data : JSON . stringify ( { type : "error" , error : "fail" } ) } ) ;
107107
108- expect ( callbacks . onError ) . toHaveBeenCalledWith ( "fail" ) ;
108+ expect ( callbacks . onError ) . toHaveBeenCalledWith ( "fail" , undefined ) ;
109109 expect ( ws . closed ) . toBe ( true ) ;
110110 } ) ;
111111
@@ -117,6 +117,7 @@ describe("WebSocket chat streaming", () => {
117117
118118 expect ( callbacks . onError ) . toHaveBeenCalledWith (
119119 "Failed to parse WebSocket message" ,
120+ undefined ,
120121 ) ;
121122 } ) ;
122123
You can’t perform that action at this time.
0 commit comments