slhaf/Partner
1
0
Fork 0
mirror of https://github.com/slhaf/Partner.git synced 2026-05-12 16:53:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(gateway): print warn log while receiving invalid input

Browse Source
This commit is contained in:
slhafzjw
2026-04-20 15:36:23 +08:00
parent e1a5a07e2a
commit ac715602a6
3 changed files with 14 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
Partner-Core/src/main/java/work/slhaf/partner/module/communication/ReplyDispatcher.kt
View File

@@ -4,7 +4,7 @@ import kotlinx.coroutines.*
import kotlinx.coroutines.channels.Channel import kotlinx.coroutines.channels.Channel
import work.slhaf.partner.framework.agent.interaction.AgentRuntime 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.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 work.slhaf.partner.framework.agent.model.StreamChatMessageConsumer
import kotlin.time.Duration.Companion.milliseconds import kotlin.time.Duration.Companion.milliseconds
@@ -55,11 +55,11 @@ object ReplyDispatcher {
if (content.isEmpty()) { if (content.isEmpty()) {
return return
} }
val event = Reply( val event = ReplyEvent(
status = EventStatus.RUNNING, status = EventStatus.RUNNING,
target = target, target = target,
content = content, content = content,
mode = Reply.ContentMode.APPEND, mode = ReplyEvent.ContentMode.APPEND,
done = false done = false
) )
AgentRuntime.response(event) AgentRuntime.response(event)

5
Partner-Core/src/main/java/work/slhaf/partner/runtime/gateway/WebSocketGateway.java
View File

@@ -1,5 +1,6 @@
package work.slhaf.partner.runtime.gateway; package work.slhaf.partner.runtime.gateway;
import com.alibaba.fastjson2.JSONException;
import com.alibaba.fastjson2.JSONObject; import com.alibaba.fastjson2.JSONObject;
import lombok.ToString; import lombok.ToString;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
@@ -121,9 +122,13 @@ public class WebSocketGateway extends WebSocketServer implements AgentGateway<In
@Override @Override
public void onMessage(WebSocket webSocket, String s) { public void onMessage(WebSocket webSocket, String s) {
try {
InputData inputData = JSONObject.parseObject(s, InputData.class); InputData inputData = JSONObject.parseObject(s, InputData.class);
userSessions.put(inputData.getSource(), webSocket); // 注册连接 userSessions.put(inputData.getSource(), webSocket); // 注册连接
receive(inputData); receive(inputData);
} catch (JSONException ignored) {
log.warn("Invalid input format: {}", s);
}
} }
@Override @Override

6
Partner-Framework/src/main/java/work/slhaf/partner/framework/agent/interaction/data/InteractionEvent.kt
View File

@@ -44,7 +44,7 @@ sealed class InteractionEvent {
} }
data class Reply @JvmOverloads constructor( data class ReplyEvent @JvmOverloads constructor(
override val status: EventStatus, override val status: EventStatus,
override val target: String, override val target: String,
val content: 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 status: EventStatus,
override val target: String, override val target: String,
val data: Data 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 status: EventStatus,
override val target: String, override val target: String,
val title: String, val title: String,