Update app.py

app.py

@@ -86,13 +86,31 @@ async def proxy(request: Request):
             try:
                 response_str = bytes(response).decode("utf-8")
                 logger.debug("Decoded response: %s", response_str)
-                # If we can decode and it looks like a complete JSON-RPC response, break
-                if response_str.strip().endswith("}") and "\"jsonrpc\"" in response_str and "\"result\"" in response_str:
-                    logger.info("Complete JSON-RPC response detected")
-                    break
+                # More robust JSON-RPC detection
+                # Check if we have a complete JSON object
+                if len(response_str.strip()) > 0:
+                    # Try to parse as JSON
+                    import json
+                    try:
+                        json_data = json.loads(response_str)
+                        # Check if it's a valid JSON-RPC response
+                        if isinstance(json_data, dict) and "jsonrpc" in json_data:
+                            # Check if it's a response (has id or error)
+                            if "id" in json_data or "error" in json_data:
+                                logger.info("Complete JSON-RPC response detected")
+                                break
+                    except json.JSONDecodeError as e:
+                        # If JSON parsing fails, continue reading
+                        logger.debug("JSON parsing failed: %s", str(e))
+                        # Continue reading
+                        continue
+                else:
+                    # Empty response, continue reading
+                    continue
             except Exception as e:
                 logger.warning("Decoding failed: %s", str(e))
                 # Continue reading
+                continue
         except Exception as e:
             logger.error("Error reading response: %s", str(e))
             break