Merge pull request #29 from Mng-dev-ai/fix/streaming-done-exit

Fix streaming completion exit

Author: Mng-dev-ai

anthropic_bridge/providers/openai/client.py

@@ -327,6 +327,7 @@ async def _stream_response(
                                 "output_tokens": resp_usage.get("output_tokens", 0),
                             }
                             break
+                            break
 
         except Exception as e:
             yield self._sse(

anthropic_bridge/providers/openrouter/client.py

@@ -182,13 +182,15 @@ async def _stream_openrouter(
                 lines = buffer.split("\n")
                 buffer = lines.pop()
 
+                done = False
                 for line in lines:
                     line = line.strip()
                     if not line or not line.startswith("data: "):
                         continue
 
                     data_str = line[6:]
                     if data_str == "[DONE]":
+                        done = True
                         break
 
                     try:
@@ -397,6 +399,8 @@ async def _stream_openrouter(
                                     await get_reasoning_cache().set(
                                         t["id"], current_reasoning_details.copy()
                                     )
+                if done:
+                    break
 
         if thinking_started:
             yield self._sse(

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "anthropic-bridge"
-version = "0.1.25"
+version = "0.1.26"
 description = "Bridge Anthropic API to OpenRouter and OpenAI models."
 license = "MIT"
 requires-python = ">=3.11"