Electric-Special/ha-core
1
0
Fork 0
mirror of https://github.com/Electric-Special/ha-core.git synced 2026-03-21 03:03:17 +01:00
Code Issues Packages Projects Releases Wiki Activity

OpenAI: Increase max iterations for AI Task (#162599)

Browse Source
This commit is contained in:
Paulus Schoutsen
2026-02-13 13:16:26 -05:00
committed by GitHub
parent 490514c274
commit f9f2f39a3c
2 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
homeassistant/components/openai_conversation/ai_task.py
View File

@@ -71,7 +71,9 @@ class OpenAITaskEntity(
chat_log: conversation.ChatLog,
) -> ai_task.GenDataTaskResult:
"""Handle a generate data task."""
await self._async_handle_chat_log(chat_log, task.name, task.structure)
await self._async_handle_chat_log(
chat_log, task.name, task.structure, max_iterations=1000
)
if not isinstance(chat_log.content[-1], conversation.AssistantContent):
raise HomeAssistantError(

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

@@ -481,6 +481,7 @@ class OpenAIBaseLLMEntity(Entity):
structure_name: str | None = None,
structure: vol.Schema | None = None,
force_image: bool = False,
max_iterations: int = MAX_TOOL_ITERATIONS,
) -> None:
"""Generate an answer for the chat log."""
options = self.subentry.data
@@ -632,7 +633,7 @@ class OpenAIBaseLLMEntity(Entity):
client = self.entry.runtime_data
# To prevent infinite loops, we limit the number of iterations
for _iteration in range(MAX_TOOL_ITERATIONS):
for _iteration in range(max_iterations):
try:
stream = await client.responses.create(**model_args)