Skip to content

Commit 65874ba

Browse files
committed
fix: add missing ignored_exception_types to OpenAiImageGenerationDriver
1 parent f4b0730 commit 65874ba

File tree

1 file changed

+14
-1
lines changed

1 file changed

+14
-1
lines changed

griptape/drivers/image_generation/openai_image_generation_driver.py

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from typing import TYPE_CHECKING, Literal, Optional
55

66
import openai
7-
from attrs import define, field, fields_dict
7+
from attrs import Factory, define, field, fields_dict
88

99
from griptape.drivers.image_generation import BaseImageGenerationDriver
1010
from griptape.utils.decorators import lazy_property
@@ -86,6 +86,19 @@ class OpenAiImageGenerationDriver(BaseImageGenerationDriver):
8686
_client: Optional[openai.OpenAI] = field(
8787
default=None, kw_only=True, alias="client", metadata={"serializable": False}
8888
)
89+
ignored_exception_types: tuple[type[Exception], ...] = field(
90+
default=Factory(
91+
lambda: (
92+
openai.BadRequestError,
93+
openai.AuthenticationError,
94+
openai.PermissionDeniedError,
95+
openai.NotFoundError,
96+
openai.ConflictError,
97+
openai.UnprocessableEntityError,
98+
),
99+
),
100+
kw_only=True,
101+
)
89102

90103
@image_size.validator # pyright: ignore[reportAttributeAccessIssue, reportOptionalMemberAccess]
91104
def validate_image_size(self, attribute: str, value: str | None) -> None:

0 commit comments

Comments
 (0)