misc: Add logging of http errors

0xPlaygrounds · Sep 5, 2024 · 8076a3a · 8076a3a
1 parent e26a2e0
commit 8076a3a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/rig-core/src/providers/cohere.rs b/rig-core/src/providers/cohere.rs
@@ -208,6 +208,7 @@ impl embeddings::EmbeddingModel for EmbeddingModel {
             }))
             .send()
             .await?
+            .error_for_status()?
             .json::<ApiResponse<EmbeddingResponse>>()
             .await?;
 
@@ -503,6 +504,7 @@ impl completion::CompletionModel for CompletionModel {
             )
             .send()
             .await?
+            .error_for_status()?
             .json::<ApiResponse<CompletionResponse>>()
             .await?;
 

diff --git a/rig-core/src/providers/openai.rs b/rig-core/src/providers/openai.rs
@@ -266,6 +266,7 @@ impl embeddings::EmbeddingModel for EmbeddingModel {
             }))
             .send()
             .await?
+            .error_for_status()?
             .json::<ApiResponse<EmbeddingResponse>>()
             .await?;
 
@@ -525,6 +526,7 @@ impl completion::CompletionModel for CompletionModel {
             )
             .send()
             .await?
+            .error_for_status()?
             .json::<ApiResponse<CompletionResponse>>()
             .await?;