diff --git a/Partner-Core/src/main/java/work/slhaf/partner/module/communication/ReplyDispatcher.kt b/Partner-Core/src/main/java/work/slhaf/partner/module/communication/ReplyDispatcher.kt
index 45b374c8..5e2b1d9d 100644
--- a/Partner-Core/src/main/java/work/slhaf/partner/module/communication/ReplyDispatcher.kt
+++ b/Partner-Core/src/main/java/work/slhaf/partner/module/communication/ReplyDispatcher.kt
@@ -4,7 +4,7 @@ import kotlinx.coroutines.*
 import kotlinx.coroutines.channels.Channel
 import work.slhaf.partner.framework.agent.interaction.AgentRuntime
 import work.slhaf.partner.framework.agent.interaction.data.InteractionEvent.EventStatus
-import work.slhaf.partner.framework.agent.interaction.data.Reply
+import work.slhaf.partner.framework.agent.interaction.data.ReplyEvent
 import work.slhaf.partner.framework.agent.model.StreamChatMessageConsumer
 import kotlin.time.Duration.Companion.milliseconds
 
@@ -55,11 +55,11 @@ object ReplyDispatcher {
         if (content.isEmpty()) {
             return
         }
-        val event = Reply(
+        val event = ReplyEvent(
             status = EventStatus.RUNNING,
             target = target,
             content = content,
-            mode = Reply.ContentMode.APPEND,
+            mode = ReplyEvent.ContentMode.APPEND,
             done = false
         )
         AgentRuntime.response(event)
diff --git a/Partner-Core/src/main/java/work/slhaf/partner/runtime/gateway/WebSocketGateway.java b/Partner-Core/src/main/java/work/slhaf/partner/runtime/gateway/WebSocketGateway.java
index 457a3edd..84df7c6f 100644
--- a/Partner-Core/src/main/java/work/slhaf/partner/runtime/gateway/WebSocketGateway.java
+++ b/Partner-Core/src/main/java/work/slhaf/partner/runtime/gateway/WebSocketGateway.java
@@ -1,5 +1,6 @@
 package work.slhaf.partner.runtime.gateway;
 
+import com.alibaba.fastjson2.JSONException;
 import com.alibaba.fastjson2.JSONObject;
 import lombok.ToString;
 import lombok.extern.slf4j.Slf4j;
@@ -121,9 +122,13 @@ public class WebSocketGateway extends WebSocketServer implements AgentGateway<In
 
     @Override
     public void onMessage(WebSocket webSocket, String s) {
-        InputData inputData = JSONObject.parseObject(s, InputData.class);
-        userSessions.put(inputData.getSource(), webSocket); // 注册连接
-        receive(inputData);
+        try {
+            InputData inputData = JSONObject.parseObject(s, InputData.class);
+            userSessions.put(inputData.getSource(), webSocket); // 注册连接
+            receive(inputData);
+        } catch (JSONException ignored) {
+            log.warn("Invalid input format: {}", s);
+        }
     }
 
     @Override
diff --git a/Partner-Framework/src/main/java/work/slhaf/partner/framework/agent/interaction/data/InteractionEvent.kt b/Partner-Framework/src/main/java/work/slhaf/partner/framework/agent/interaction/data/InteractionEvent.kt
index 43059f2e..f98436d6 100644
--- a/Partner-Framework/src/main/java/work/slhaf/partner/framework/agent/interaction/data/InteractionEvent.kt
+++ b/Partner-Framework/src/main/java/work/slhaf/partner/framework/agent/interaction/data/InteractionEvent.kt
@@ -44,7 +44,7 @@ sealed class InteractionEvent {
 
 }
 
-data class Reply @JvmOverloads constructor(
+data class ReplyEvent @JvmOverloads constructor(
     override val status: EventStatus,
     override val target: String,
     val content: String,
@@ -60,7 +60,7 @@ data class Reply @JvmOverloads constructor(
     }
 }
 
-data class Module(
+data class ModuleEvent(
     override val status: EventStatus,
     override val target: String,
     val data: Data
@@ -73,7 +73,7 @@ data class Module(
     )
 }
 
-data class System @JvmOverloads constructor(
+data class SystemEvent @JvmOverloads constructor(
     override val status: EventStatus,
     override val target: String,
     val title: String,