jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

refactor: Update get_message_from_openai_exception to handle None return value

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-19 20:01:14 -03:00
commit b8b14bb36a
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/base/langflow/base/models/exceptions.py
View file

@ -1,4 +1,4 @@
def get_message_from_openai_exception(exception: Exception) -> str:
def get_message_from_openai_exception(exception: Exception) -> str | None:
"""
Get a message from an OpenAI exception.
@ -13,7 +13,7 @@ def get_message_from_openai_exception(exception: Exception) -> str:
except ImportError:
return
if isinstance(exception, BadRequestError):
message = exception.body.get("message")
message = exception.body.get("message") # type: ignore
if message:
return message
return