Mirrored_Repos/AppFlowy
1
0
Fork 0
mirror of https://github.com/AppFlowy-IO/AppFlowy.git synced 2024-08-30 18:12:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: show ai service error (#5675)

Browse Source
This commit is contained in:
Nathan.fooo
2024-07-02 13:26:53 +08:00
committed by GitHub
parent 8c1520b273
commit 7a3f013a87
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
frontend/rust-lib/flowy-chat/src/chat.rs
View File

@ -140,12 +140,14 @@ impl Chat {
send_notification(&chat_id, ChatNotification::StreamChatMessageError) send_notification(&chat_id, ChatNotification::StreamChatMessageError)
.payload(pb) .payload(pb)
.send(); .send();
break; return Err(err);
}, },
} }
} }
}, },
Err(err) => { Err(err) => {
error!("[Chat] failed to stream answer: {}", err);
let _ = text_sink.send(format!("error:{}", err)).await;
let pb = ChatMessageErrorPB { let pb = ChatMessageErrorPB {
chat_id: chat_id.clone(), chat_id: chat_id.clone(),
error_message: err.to_string(), error_message: err.to_string(),
@ -153,10 +155,15 @@ impl Chat {
send_notification(&chat_id, ChatNotification::StreamChatMessageError) send_notification(&chat_id, ChatNotification::StreamChatMessageError)
.payload(pb) .payload(pb)
.send(); .send();
return Err(err);
}, },
} }
send_notification(&chat_id, ChatNotification::FinishStreaming).send(); send_notification(&chat_id, ChatNotification::FinishStreaming).send();
if stream_buffer.lock().await.is_empty() {
return Ok(());
}
let answer = cloud_service let answer = cloud_service
.save_answer( .save_answer(
&workspace_id, &workspace_id,
@ -166,7 +173,6 @@ impl Chat {
) )
.await?; .await?;
Self::save_answer(uid, &chat_id, &user_service, answer)?; Self::save_answer(uid, &chat_id, &user_service, answer)?;
Ok::<(), FlowyError>(()) Ok::<(), FlowyError>(())
}); });