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