fix proxy decode bug

Shirley125 · Shirley125 · commit 8a725eb58461 · 2025-10-25T12:55:05.000+08:00
Signed-off-by: CHEN &lt;116010019@link.cuhk.edu.cn&gt;
diff --git a/examples/disaggregated_prefill_v1/load_balance_proxy_layerwise_server_example.py b/examples/disaggregated_prefill_v1/load_balance_proxy_layerwise_server_example.py
@@ -558,7 +558,7 @@ async def generate_stream():
                             released_kv = True
                         try:
                             chunk_str = chunk.decode("utf-8").strip()
-                        except Exception:
+                        except UnicodeDecodeError:
                             logger.debug(
                                 f"Skipping chunk: {chunk}")
                             yield chunk
diff --git a/examples/disaggregated_prefill_v1/load_balance_proxy_server_example.py b/examples/disaggregated_prefill_v1/load_balance_proxy_server_example.py
@@ -541,7 +541,7 @@ async def generate_stream():
                             released_kv = True
                         try:
                             chunk_str = chunk.decode("utf-8").strip()
-                        except Exception:
+                        except UnicodeDecodeError:
                             logger.debug(
                                 f"Skipping chunk: {chunk}")
                             yield chunk
