Fix the bug of non-standard SSE format in Claude API response data pa…

…rsing and improve robustness.
yym68686 · Jul 23, 2024 · 5200484 · 5200484
1 parent 47f4668
commit 5200484
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/response.py b/response.py
@@ -104,7 +104,9 @@ async def fetch_claude_response_stream(client, url, headers, payload, model):
                 # print(line)
 
                 if line.startswith("data:"):
-                    line = line[6:]
+                    line = line[5:]
+                    if line.startswith(" "):
+                        line = line[1:]
                     resp: dict = json.loads(line)
                     message = resp.get("message")
                     if message: