Adjust typing of AbstractConversationAgent.supported_languages (#91648)
* Adjust typing of AbstractConversationAgent.supported_languages * Update test
This commit is contained in:
parent
5e9bbeb4ad
commit
eabbe8969d
3 changed files with 6 additions and 5 deletions
|
@ -3,7 +3,7 @@ from __future__ import annotations
|
|||
|
||||
from abc import ABC, abstractmethod
|
||||
from dataclasses import dataclass
|
||||
from typing import Any, TypedDict
|
||||
from typing import Any, Literal, TypedDict
|
||||
|
||||
from homeassistant.core import Context
|
||||
from homeassistant.helpers import intent
|
||||
|
@ -51,7 +51,7 @@ class AbstractConversationAgent(ABC):
|
|||
|
||||
@property
|
||||
@abstractmethod
|
||||
def supported_languages(self) -> list[str]:
|
||||
def supported_languages(self) -> list[str] | Literal["*"]:
|
||||
"""Return a list of supported languages."""
|
||||
|
||||
@abstractmethod
|
||||
|
|
|
@ -3,6 +3,7 @@ from __future__ import annotations
|
|||
|
||||
from functools import partial
|
||||
import logging
|
||||
from typing import Literal
|
||||
|
||||
import openai
|
||||
from openai import error
|
||||
|
@ -71,9 +72,9 @@ class OpenAIAgent(conversation.AbstractConversationAgent):
|
|||
return {"name": "Powered by OpenAI", "url": "https://www.openai.com"}
|
||||
|
||||
@property
|
||||
def supported_languages(self) -> list[str]:
|
||||
def supported_languages(self) -> list[str] | Literal["*"]:
|
||||
"""Return a list of supported languages."""
|
||||
return [MATCH_ALL]
|
||||
return MATCH_ALL
|
||||
|
||||
async def async_process(
|
||||
self, user_input: conversation.ConversationInput
|
||||
|
|
|
@ -148,4 +148,4 @@ async def test_conversation_agent(
|
|||
agent = await conversation._get_agent_manager(hass).async_get_agent(
|
||||
mock_config_entry.entry_id
|
||||
)
|
||||
assert agent.supported_languages == ["*"]
|
||||
assert agent.supported_languages == "*"
|
||||
|
|
Loading…
Add table
Reference in a new issue