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

Use template selector for prompt template in OpenAI (#87205)

Browse source 
* Use template selector for prompt template in OpenAI

* Fix tests

* Do not parse template result
This commit is contained in:
Franck Nijhof 2023-02-02 21:20:10 +01:00 • committed by GitHub
parent 22698b1cc5
commit f9d7d65f3a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 18 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -18,8 +18,7 @@ from homeassistant.data_entry_flow import FlowResult
from homeassistant.helpers.selector import (
NumberSelector,
NumberSelectorConfig,
TextSelector,
TextSelectorConfig,
TemplateSelector,
)
from .const import (
@ -132,9 +131,7 @@ def openai_config_option_schema(options: MappingProxyType[str, Any]) -> dict:
if not options:
options = DEFAULT_OPTIONS
return {
vol.Required(CONF_PROMPT, default=options.get(CONF_PROMPT)): TextSelector(
TextSelectorConfig(multiline=True)
),
vol.Required(CONF_PROMPT, default=options.get(CONF_PROMPT)): TemplateSelector(),
vol.Required(CONF_MODEL, default=options.get(CONF_MODEL)): str,
vol.Required(CONF_MAX_TOKENS, default=options.get(CONF_MAX_TOKENS)): int,
vol.Required(CONF_TOP_P, default=options.get(CONF_TOP_P)): NumberSelector(