From 12376a2338692bf0579de2db647c42049ac992b8 Mon Sep 17 00:00:00 2001
From: Paulus Schoutsen <balloob@gmail.com>
Date: Wed, 21 May 2025 18:54:29 -0400
Subject: [PATCH] Mark LLMs that support streaming as such (#145405)

---
 homeassistant/components/anthropic/conversation.py       | 1 +
 homeassistant/components/ollama/conversation.py          | 9 ++++++---
 .../components/openai_conversation/conversation.py       | 1 +
 3 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/homeassistant/components/anthropic/conversation.py b/homeassistant/components/anthropic/conversation.py
index 7e1fda467a8..bfdd4bfd361 100644
--- a/homeassistant/components/anthropic/conversation.py
+++ b/homeassistant/components/anthropic/conversation.py
@@ -326,6 +326,7 @@ class AnthropicConversationEntity(
 
     _attr_has_entity_name = True
     _attr_name = None
+    _attr_supports_streaming = True
 
     def __init__(self, entry: AnthropicConfigEntry) -> None:
         """Initialize the agent."""
diff --git a/homeassistant/components/ollama/conversation.py b/homeassistant/components/ollama/conversation.py
index ab9e05b5fbe..6c507030ad3 100644
--- a/homeassistant/components/ollama/conversation.py
+++ b/homeassistant/components/ollama/conversation.py
@@ -89,9 +89,11 @@ def _parse_tool_args(arguments: dict[str, Any]) -> dict[str, Any]:
 
 
 def _convert_content(
-    chat_content: conversation.Content
-    | conversation.ToolResultContent
-    | conversation.AssistantContent,
+    chat_content: (
+        conversation.Content
+        | conversation.ToolResultContent
+        | conversation.AssistantContent
+    ),
 ) -> ollama.Message:
     """Create tool response content."""
     if isinstance(chat_content, conversation.ToolResultContent):
@@ -172,6 +174,7 @@ class OllamaConversationEntity(
     """Ollama conversation agent."""
 
     _attr_has_entity_name = True
+    _attr_supports_streaming = True
 
     def __init__(self, entry: ConfigEntry) -> None:
         """Initialize the agent."""
diff --git a/homeassistant/components/openai_conversation/conversation.py b/homeassistant/components/openai_conversation/conversation.py
index d55ffc2df0c..a129400194b 100644
--- a/homeassistant/components/openai_conversation/conversation.py
+++ b/homeassistant/components/openai_conversation/conversation.py
@@ -231,6 +231,7 @@ class OpenAIConversationEntity(
 
     _attr_has_entity_name = True
     _attr_name = None
+    _attr_supports_streaming = True
 
     def __init__(self, entry: OpenAIConfigEntry) -> None:
         """Initialize the agent."""