Merge pull request #14 from 0xPlaygrounds/misc/add-http-error-logging

misc: Add logging of http errors
0xPlaygrounds · Sep 5, 2024 · 7af3603 · 7af3603
2 parents e26a2e0 + 8076a3a
commit 7af3603
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?;