handle a bit more errors

DuckyBlender · Oct 5, 2024 · 6d62a45 · 6d62a45
1 parent c25c49b
commit 6d62a45
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/src/apis/together.rs b/src/apis/together.rs
@@ -1,3 +1,4 @@
+use anyhow::Result;
 use reqwest::Client;
 use serde::{Deserialize, Serialize};
 use std::env;
@@ -44,12 +45,7 @@ impl TogetherClient {
         }
     }
 
-    pub async fn submit_request(
-        &self,
-        prompt: ImageRequest,
-    ) -> Result<ImageResponse, reqwest::Error> {
-        // base64 image
-
+    pub async fn submit_request(&self, prompt: ImageRequest) -> Result<ImageResponse> {
         let response = self
             .client
             .post("https://api.together.xyz/v1/images/generations")
@@ -62,6 +58,10 @@ impl TogetherClient {
             .send()
             .await?;
 
+        if !response.status().is_success() {
+            return Err(anyhow::anyhow!("Status code: {}", response.status()));
+        }
+
         let response = response.json::<ImageResponse>().await?;
 
         Ok(response)

diff --git a/src/main.rs b/src/main.rs
@@ -181,6 +181,10 @@ async fn handle_command(
         let res = client.submit_request(request).await;
         if let Err(e) = res {
             error!("Failed to submit request: {:?}", e);
+            bot.send_message(message.chat.id, format!("error: {e:?}"))
+                .reply_parameters(ReplyParameters::new(message.id))
+                .await
+                .unwrap();
             return Ok(lambda_http::Response::builder()
                 .status(200)
                 .body(String::new())