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

OpenAI to report when running out of funds (#139088)

Browse Source
This commit is contained in:
Paulus Schoutsen 2025-02-23 04:51:25 -05:00 committed by GitHub
parent 0797c3228b
commit 91668e99e3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 28 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
homeassistant/components/openai_conversation/conversation.py
View File

@ -287,6 +287,9 @@ class OpenAIConversationEntity(
try:
result = await client.chat.completions.create(**model_args)
except openai.RateLimitError as err:
LOGGER.error("Rate limited by OpenAI: %s", err)
raise HomeAssistantError("Rate limited or insufficient funds") from err
except openai.OpenAIError as err:
LOGGER.error("Error talking to OpenAI: %s", err)
raise HomeAssistantError("Error talking to OpenAI") from err

31
tests/components/openai_conversation/test_conversation.py
View File

@ -4,7 +4,7 @@ from collections.abc import Generator
from unittest.mock import AsyncMock, patch
from httpx import Response
from openai import RateLimitError
from openai import AuthenticationError, RateLimitError
from openai.types.chat.chat_completion_chunk import (
ChatCompletionChunk,
Choice,
@ -94,23 +94,42 @@ async def test_entity(
)
@pytest.mark.parametrize(
("exception", "message"),
[
(
RateLimitError(
response=Response(status_code=429, request=""), body=None, message=None
),
"Rate limited or insufficient funds",
),
(
AuthenticationError(
response=Response(status_code=401, request=""), body=None, message=None
),
"Error talking to OpenAI",
),
],
)
async def test_error_handling(
hass: HomeAssistant, mock_config_entry: MockConfigEntry, mock_init_component
hass: HomeAssistant,
mock_config_entry: MockConfigEntry,
mock_init_component,
exception,
message,
) -> None:
"""Test that we handle errors when calling completion API."""
with patch(
"openai.resources.chat.completions.AsyncCompletions.create",
new_callable=AsyncMock,
side_effect=RateLimitError(
response=Response(status_code=None, request=""), body=None, message=None
),
side_effect=exception,
):
result = await conversation.async_converse(
hass, "hello", None, Context(), agent_id=mock_config_entry.entry_id
)
assert result.response.response_type == intent.IntentResponseType.ERROR, result
assert result.response.error_code == "unknown", result
assert result.response.speech["plain"]["speech"] == message, result.response.speech
async def test_conversation_agent(