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

Make openai conversation prompt template more readable + test case (#86676)

Browse source
This commit is contained in:
Paulus Schoutsen 2023-01-26 05:04:15 -05:00 • committed by GitHub
parent 021ac84405
commit ff91fb7d74
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 19 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -50,6 +50,17 @@ async def test_default_prompt(hass, mock_init_component):
model="Test Model 3A", model="Test Model 3A",
suggested_area="Test Area 2", suggested_area="Test Area 2",
) )
device = device_reg.async_get_or_create(
config_entry_id="1234",
connections={("test", "9876-disabled")},
name="Test Device 3",
manufacturer="Test Manufacturer 3",
model="Test Model 3A",
suggested_area="Test Area 2",
)
device_reg.async_update_device(
device.id, disabled_by=device_registry.DeviceEntryDisabler.USER
)
with patch("openai.Completion.create") as mock_create: with patch("openai.Completion.create") as mock_create:
await conversation.async_converse(hass, "hello", None, Context()) await conversation.async_converse(hass, "hello", None, Context())