evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Enumerate openai.Models to validate config (#99438)

Browse source
This commit is contained in:
Joseph 2023-11-24 04:14:44 -08:00 • committed by GitHub
parent 62473936e2
commit df025b5993
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
homeassistant/components/openai_conversation/__init__.py
View file

@ -89,7 +89,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
try:
await hass.async_add_executor_job(
partial(
openai.Engine.list,
openai.Model.list,
api_key=entry.data[CONF_API_KEY],
request_timeout=10,
)

2
homeassistant/components/openai_conversation/config_flow.py
View file

@ -60,7 +60,7 @@ async def validate_input(hass: HomeAssistant, data: dict[str, Any]) -> None:
Data has the keys from STEP_USER_DATA_SCHEMA with values provided by the user.
"""
openai.api_key = data[CONF_API_KEY]
await hass.async_add_executor_job(partial(openai.Engine.list, request_timeout=10))
await hass.async_add_executor_job(partial(openai.Model.list, request_timeout=10))
class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):

2
tests/components/openai_conversation/conftest.py
View file

@ -25,7 +25,7 @@ def mock_config_entry(hass):
async def mock_init_component(hass, mock_config_entry):
"""Initialize integration."""
with patch(
"openai.Engine.list",
"openai.Model.list",
):
assert await async_setup_component(hass, "openai_conversation", {})
await hass.async_block_till_done()

4
tests/components/openai_conversation/test_config_flow.py
View file

@ -32,7 +32,7 @@ async def test_form(hass: HomeAssistant) -> None:
assert result["errors"] is None
with patch(
"homeassistant.components.openai_conversation.config_flow.openai.Engine.list",
"homeassistant.components.openai_conversation.config_flow.openai.Model.list",
), patch(
"homeassistant.components.openai_conversation.async_setup_entry",
return_value=True,
@ -88,7 +88,7 @@ async def test_form_invalid_auth(hass: HomeAssistant, side_effect, error) -> Non
)
with patch(
"homeassistant.components.openai_conversation.config_flow.openai.Engine.list",
"homeassistant.components.openai_conversation.config_flow.openai.Model.list",
side_effect=side_effect,
):
result2 = await hass.config_entries.flow.async_configure(

2
tests/components/openai_conversation/test_init.py
View file

@ -140,7 +140,7 @@ async def test_template_error(
},
)
with patch(
"openai.Engine.list",
"openai.Model.list",
), patch("openai.ChatCompletion.acreate"):
await hass.config_entries.async_setup(mock_config_entry.entry_id)
await hass.async_block_till_done()