more specific try except

bespokelabsai · Dec 4, 2024 · 232dab5 · 232dab5
1 parent 2ba4bdf
commit 232dab5
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/src/bespokelabs/curator/request_processor/litellm_online_request_processor.py b/src/bespokelabs/curator/request_processor/litellm_online_request_processor.py
@@ -91,11 +91,18 @@ class User(BaseModel):
                 f"Model {self.model} supports structured output via instructor, response: {response}"
             )
             return True
-        except Exception as e:
-            logger.warning(
-                f"Model {self.model} does not support structured output via instructor: {e}"
-            )
-            return False
+        except instructor.exceptions.InstructorRetryException as e:
+            if "litellm.AuthenticationError" in str(e):
+                logger.warning(
+                    f"Please provide a valid API key for model {self.model}."
+                )
+                raise e
+            else:
+                logger.warning(
+                    f"Model {self.model} does not support structured output via instructor: {e} {type(e)} {e.__cause__}"
+                )
+                return False
+
 
     def estimate_output_tokens(self) -> int:
         """Estimate the number of tokens in the model's response.