Compare commits

...

1 Commits

Author SHA1 Message Date
Paulus Schoutsen
cbd50c4177 OpenAI: Increase max iterations for AI Task 2026-02-08 20:56:34 -05:00
2 changed files with 5 additions and 2 deletions

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(

View File

@@ -479,6 +479,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
@@ -628,7 +629,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)