From 2f3781bfc22a89b8a967a3c1a31b41c634f4ace2 Mon Sep 17 00:00:00 2001
From: chihlasm <michael@chihlas.com>
Date: Sat, 28 Mar 2026 23:02:35 +0000
Subject: [PATCH] feat: add generate_text_stream to AnthropicProvider for SSE
 support

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
---
 backend/app/core/ai_provider.py | 38 +++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)

diff --git a/backend/app/core/ai_provider.py b/backend/app/core/ai_provider.py
index 7453ec9c..84d4c33b 100644
--- a/backend/app/core/ai_provider.py
+++ b/backend/app/core/ai_provider.py
@@ -7,6 +7,7 @@ backends for JSON generation used by the AI Flow Builder.
 
 import logging
 from abc import ABC, abstractmethod
+from collections.abc import AsyncIterator
 
 from app.core.config import settings
 
@@ -54,6 +55,26 @@ class AIProvider(ABC):
         """
         ...
 
+    async def generate_text_stream(
+        self,
+        system_prompt: str,
+        messages: list[dict[str, str]],
+        max_tokens: int = 4096,
+    ) -> "AsyncIterator[str]":
+        """Stream a text response token by token.
+
+        Args:
+            system_prompt: System-level instruction for the model.
+            messages: List of message dicts with "role" and "content" keys.
+            max_tokens: Maximum output tokens.
+
+        Yields:
+            Text chunks as they are generated.
+        """
+        raise NotImplementedError("Streaming not supported for this provider")
+        # Make this an async generator to satisfy type checker
+        yield ""  # pragma: no cover
+
 
 class GeminiProvider(AIProvider):
     """Google Gemini provider using the google-genai SDK."""
@@ -221,6 +242,23 @@ class AnthropicProvider(AIProvider):
         # Anthropic doesn't differentiate between JSON and text mode
         return await self.generate_json(system_prompt, messages, max_tokens)
 
+    async def generate_text_stream(
+        self,
+        system_prompt: str,
+        messages: list[dict[str, str]],
+        max_tokens: int = 4096,
+    ) -> AsyncIterator[str]:
+        client = _get_anthropic_client(self._api_key, self._timeout)
+
+        async with client.messages.stream(
+            model=self._model,
+            max_tokens=max_tokens,
+            system=system_prompt,
+            messages=messages,
+        ) as stream:
+            async for text in stream.text_stream:
+                yield text
+
 
 def get_ai_provider(model: str | None = None) -> AIProvider:
     """Factory that returns the configured AI provider.