From 6b3a8b69bc5a6883ff837c8f1e536f968db0991b Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 20 Jun 2024 18:18:09 -0300 Subject: [PATCH] refactor: Update AnthropicModelComponent to handle ImportError gracefully --- src/backend/base/langflow/components/models/AnthropicModel.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/components/models/AnthropicModel.py b/src/backend/base/langflow/components/models/AnthropicModel.py index 9b4b4dca3..dd7ccb4f5 100644 --- a/src/backend/base/langflow/components/models/AnthropicModel.py +++ b/src/backend/base/langflow/components/models/AnthropicModel.py @@ -98,9 +98,9 @@ class AnthropicModelComponent(LCModelComponent): try: from anthropic import BadRequestError except ImportError: - return + return None if isinstance(exception, BadRequestError): message = exception.body.get("error", {}).get("message") # type: ignore if message: return message - return + return None