Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove deprecated DALL-E image formats (#122388)

pull/122296/head
Denis Shulyaka 2024-07-22 22:57:48 +03:00 committed by GitHub
parent 489457c47b
commit ee30510b23
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 57 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
homeassistant/components/openai_conversation/__init__.py
View File

@ -2,8 +2,6 @@
from __future__ import annotations from __future__ import annotations
from typing import Literal, cast
import openai import openai
import voluptuous as vol import voluptuous as vol
@ -20,11 +18,7 @@ from homeassistant.exceptions import (
HomeAssistantError, HomeAssistantError,
ServiceValidationError, ServiceValidationError,
) )
from homeassistant.helpers import ( from homeassistant.helpers import config_validation as cv, selector
config_validation as cv,
issue_registry as ir,
selector,
)
from homeassistant.helpers.typing import ConfigType from homeassistant.helpers.typing import ConfigType
from .const import DOMAIN, LOGGER from .const import DOMAIN, LOGGER
@ -53,32 +47,11 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
client: openai.AsyncClient = entry.runtime_data client: openai.AsyncClient = entry.runtime_data
if call.data["size"] in ("256", "512", "1024"):
ir.async_create_issue(
hass,
DOMAIN,
"image_size_deprecated_format",
breaks_in_ha_version="2024.7.0",
is_fixable=False,
is_persistent=True,
learn_more_url="https://www.home-assistant.io/integrations/openai_conversation/",
severity=ir.IssueSeverity.WARNING,
translation_key="image_size_deprecated_format",
)
size = "1024x1024"
else:
size = call.data["size"]
size = cast(
Literal["256x256", "512x512", "1024x1024", "1792x1024", "1024x1792"],
size,
) # size is selector, so no need to check further
try: try:
response = await client.images.generate( response = await client.images.generate(
model="dall-e-3", model="dall-e-3",
prompt=call.data["prompt"], prompt=call.data["prompt"],
size=size, size=call.data["size"],
quality=call.data["quality"], quality=call.data["quality"],
style=call.data["style"], style=call.data["style"],
response_format="url", response_format="url",
@ -102,7 +75,7 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
), ),
vol.Required("prompt"): cv.string, vol.Required("prompt"): cv.string,
vol.Optional("size", default="1024x1024"): vol.In( vol.Optional("size", default="1024x1024"): vol.In(
("1024x1024", "1024x1792", "1792x1024", "256", "512", "1024") ("1024x1024", "1024x1792", "1792x1024")
), ),
vol.Optional("quality", default="standard"): vol.In(("standard", "hd")), vol.Optional("quality", default="standard"): vol.In(("standard", "hd")),
vol.Optional("style", default="vivid"): vol.In(("vivid", "natural")), vol.Optional("style", default="vivid"): vol.In(("vivid", "natural")),

27
tests/components/openai_conversation/test_init.py
View File

@ -60,33 +60,6 @@ from tests.common import MockConfigEntry
"style": "natural", "style": "natural",
}, },
), ),
(
{"prompt": "Picture of a dog", "size": "256"},
{
"prompt": "Picture of a dog",
"size": "1024x1024",
"quality": "standard",
"style": "vivid",
},
),
(
{"prompt": "Picture of a dog", "size": "512"},
{
"prompt": "Picture of a dog",
"size": "1024x1024",
"quality": "standard",
"style": "vivid",
},
),
(
{"prompt": "Picture of a dog", "size": "1024"},
{
"prompt": "Picture of a dog",
"size": "1024x1024",
"quality": "standard",
"style": "vivid",
},
),
], ],
) )
async def test_generate_image_service( async def test_generate_image_service(