jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-28 15:47:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

OpenAI Conversation split out chat log processing (#145129)

Browse Source
This commit is contained in:
Paulus Schoutsen 2025-05-18 01:00:24 -04:00 committed by GitHub
parent f07265ece4
commit 6fd9857666
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 20 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -274,7 +274,7 @@ class OpenAIConversationEntity(
user_input: conversation.ConversationInput, user_input: conversation.ConversationInput,
chat_log: conversation.ChatLog, chat_log: conversation.ChatLog,
) -> conversation.ConversationResult: ) -> conversation.ConversationResult:
"""Call the API.""" """Process the user input and call the API."""
options = self.entry.options options = self.entry.options
try: try:
@ -287,6 +287,24 @@ class OpenAIConversationEntity(
except conversation.ConverseError as err: except conversation.ConverseError as err:
return err.as_conversation_result() return err.as_conversation_result()
await self._async_handle_chat_log(chat_log)
intent_response = intent.IntentResponse(language=user_input.language)
assert type(chat_log.content[-1]) is conversation.AssistantContent
intent_response.async_set_speech(chat_log.content[-1].content or "")
return conversation.ConversationResult(
response=intent_response,
conversation_id=chat_log.conversation_id,
continue_conversation=chat_log.continue_conversation,
)
async def _async_handle_chat_log(
self,
chat_log: conversation.ChatLog,
) -> None:
"""Generate an answer for the chat log."""
options = self.entry.options
tools: list[ToolParam] | None = None tools: list[ToolParam] | None = None
if chat_log.llm_api: if chat_log.llm_api:
tools = [ tools = [
@ -357,7 +375,7 @@ class OpenAIConversationEntity(
raise HomeAssistantError("Error talking to OpenAI") from err raise HomeAssistantError("Error talking to OpenAI") from err
async for content in chat_log.async_add_delta_content_stream( async for content in chat_log.async_add_delta_content_stream(
user_input.agent_id, _transform_stream(chat_log, result, messages) self.entity_id, _transform_stream(chat_log, result, messages)
): ):
if not isinstance(content, conversation.AssistantContent): if not isinstance(content, conversation.AssistantContent):
messages.extend(_convert_content_to_param(content)) messages.extend(_convert_content_to_param(content))
@ -365,15 +383,6 @@ class OpenAIConversationEntity(
if not chat_log.unresponded_tool_results: if not chat_log.unresponded_tool_results:
break break
intent_response = intent.IntentResponse(language=user_input.language)
assert type(chat_log.content[-1]) is conversation.AssistantContent
intent_response.async_set_speech(chat_log.content[-1].content or "")
return conversation.ConversationResult(
response=intent_response,
conversation_id=chat_log.conversation_id,
continue_conversation=chat_log.continue_conversation,
)
async def _async_entry_update_listener( async def _async_entry_update_listener(
self, hass: HomeAssistant, entry: ConfigEntry self, hass: HomeAssistant, entry: ConfigEntry
) -> None: ) -> None: