fix: output err if step failed

release-notes-05
Meng Zhang 2023-11-02 16:15:06 -07:00
parent 4c7eae584e
commit eb34850a5e
1 changed files with 3 additions and 2 deletions

View File

@ -74,8 +74,9 @@ impl AsyncTextInferenceEngine {
let mut engine = self.engine.lock().await;
let Ok(result) = engine.as_mut().unwrap().step() else {
panic!("Failed to evaluation");
let result = match engine.as_mut().unwrap().step() {
Ok(result) => result,
Err(err) => panic!("Failed to step: {}", err),
};
for ffi::StepOutput { request_id, text } in result {