0.1.38 bugfix

Author: sorryhyun

backend/providers/claude/__init__.py

@@ -5,6 +5,25 @@
 for use with the ChitChats multi-provider abstraction layer.
 """
 
+# Monkey-patch the SDK's message parser to handle rate_limit_event gracefully.
+# The SDK raises MessageParseError for unknown message types like rate_limit_event,
+# which kills the entire response stream. This patch converts them to SystemMessage instead.
+import claude_agent_sdk._internal.client as _cl  # noqa: E402
+import claude_agent_sdk._internal.message_parser as _mp  # noqa: E402
+from claude_agent_sdk.types import SystemMessage as _SystemMessage  # noqa: E402
+
+_original_parse_message = _mp.parse_message
+
+
+def _patched_parse_message(data):
+    if isinstance(data, dict) and data.get("type") == "rate_limit_event":
+        return _SystemMessage(subtype="rate_limit", data=data)
+    return _original_parse_message(data)
+
+
+_mp.parse_message = _patched_parse_message
+_cl.parse_message = _patched_parse_message  # type: ignore[attr-defined]  # Patch the already-imported reference
+
 from .client import ClaudeClient
 from .parser import ClaudeStreamParser
 from .provider import ClaudeClientPool, ClaudeProvider

backend/providers/claude/parser.py

@@ -83,7 +83,9 @@ def parse_message(
 
         # Extract session_id from SystemMessage
         if isinstance(message, SystemMessage):
-            if isinstance(message.data, dict) and "session_id" in message.data:
+            if message.subtype == "rate_limit":
+                logger.warning(f"Rate limited by API (concurrent usage?): {message.data}")
+            elif isinstance(message.data, dict) and "session_id" in message.data:
                 new_session_id = message.data["session_id"]
                 logger.debug(f"Extracted session_id: {new_session_id}")