refactor error handling

neowu · Apr 22, 2024 · cf18902 · cf18902
1 parent 9afcf22
commit cf18902
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/gcloud/vertex.rs b/src/gcloud/vertex.rs
@@ -107,7 +107,7 @@ impl Vertex {
         let (tx, rx) = channel(64);
         let handle = tokio::spawn(read_response_stream(response, tx));
         let function_call = self.process_response(rx, handler).await;
-        let _ = tokio::try_join!(handle)?;
+        handle.await??;
 
         Ok(function_call)
     }

diff --git a/src/openai/chatgpt.rs b/src/openai/chatgpt.rs
@@ -96,7 +96,7 @@ impl ChatGPT {
         let handle = tokio::spawn(read_event_source(source, tx));
 
         let function_call = self.process_event(rx, handler).await;
-        let _ = tokio::try_join!(handle)?;
+        handle.await??;
 
         Ok(function_call)
     }