jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-23 16:57:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve type hints in openai_conversation tests (#123811)

Browse Source
This commit is contained in:
epenet 2024-08-13 16:01:24 +02:00 committed by GitHub
parent 04b1d2414d
commit 135f15fdc3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/components/openai_conversation/conftest.py
View File

@ -13,7 +13,7 @@ from tests.common import MockConfigEntry
@pytest.fixture
def mock_config_entry(hass):
def mock_config_entry(hass: HomeAssistant) -> MockConfigEntry:
"""Mock a config entry."""
entry = MockConfigEntry(
title="OpenAI",
@ -27,7 +27,9 @@ def mock_config_entry(hass):
@pytest.fixture
def mock_config_entry_with_assist(hass, mock_config_entry):
def mock_config_entry_with_assist(
hass: HomeAssistant, mock_config_entry: MockConfigEntry
) -> MockConfigEntry:
"""Mock a config entry with assist."""
hass.config_entries.async_update_entry(
mock_config_entry, options={CONF_LLM_HASS_API: llm.LLM_API_ASSIST}
@ -36,7 +38,9 @@ def mock_config_entry_with_assist(hass, mock_config_entry):
@pytest.fixture
async def mock_init_component(hass, mock_config_entry):
async def mock_init_component(
hass: HomeAssistant, mock_config_entry: MockConfigEntry
) -> None:
"""Initialize integration."""
with patch(
"openai.resources.models.AsyncModels.list",