chore: log stream error

This commit is contained in:
Nathan 2025-03-22 17:35:10 +08:00
parent 210055de62
commit 7fef4fc7af
3 changed files with 8 additions and 2 deletions

View file

@ -32,7 +32,12 @@ where
return Err(AppResponseError::new(ErrorCode::Internal, body));
}
let stream = resp.bytes_stream().map_err(AppResponseError::from);
let stream = resp.bytes_stream().map_err(|err| {
AppResponseError::new(
ErrorCode::SerdeError,
format!("Error reading response stream: {}", err),
)
});
let stream = check_first_item_response_error(stream).await?;
Ok(JsonStream::<T, _, AppResponseError>::new(stream))
}

View file

@ -22,7 +22,7 @@ use tracing::{error, instrument, trace};
pub fn ai_completion_scope() -> Scope {
web::scope("/api/ai/{workspace_id}")
.service(web::resource("/complete/stream").route(web::post().to(stream_complete_text_handler)))
.service(web::resource("v2/complete/stream").route(web::post().to(stream_complete_v2_handler)))
.service(web::resource("/v2/complete/stream").route(web::post().to(stream_complete_v2_handler)))
.service(web::resource("/summarize_row").route(web::post().to(summarize_row_handler)))
.service(web::resource("/translate_row").route(web::post().to(translate_row_handler)))
.service(web::resource("/local/config").route(web::get().to(local_ai_config_handler)))

View file

@ -347,6 +347,7 @@ async fn answer_stream_v2_handler(
)
},
Err(err) => {
trace!("[Chat] stream answer failed: {}", err);
state.metrics.ai_metrics.record_failed_stream_count(1);
Ok(
HttpResponse::ServiceUnavailable()