File size: 1,831 Bytes
246d201 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
import { delay, WebSocketHandler, ws } from "msw";
import { toSocketIo } from "@mswjs/socket.io-binding";
import { AgentState } from "#/types/agent-state";
import { InitConfig } from "#/types/core/variances";
import { SESSION_HISTORY } from "./session-history.mock";
import {
generateAgentStateChangeObservation,
emitMessages,
emitAssistantMessage,
} from "./mock-ws-helpers";
const isInitConfig = (data: unknown): data is InitConfig =>
typeof data === "object" &&
data !== null &&
"action" in data &&
data.action === "initialize";
const chat = ws.link(`ws://${window?.location.host}/socket.io`);
export const handlers: WebSocketHandler[] = [
chat.addEventListener("connection", (connection) => {
const io = toSocketIo(connection);
// @ts-expect-error - accessing private property for testing purposes
const { url }: { url: URL } = io.client.connection;
const conversationId = url.searchParams.get("conversation_id");
io.client.emit("connect");
if (conversationId) {
emitMessages(io, SESSION_HISTORY["1"]);
io.client.emit(
"oh_event",
generateAgentStateChangeObservation(AgentState.AWAITING_USER_INPUT),
);
}
io.client.on("oh_action", async (_, data) => {
if (isInitConfig(data)) {
io.client.emit(
"oh_event",
generateAgentStateChangeObservation(AgentState.INIT),
);
} else {
io.client.emit(
"oh_event",
generateAgentStateChangeObservation(AgentState.RUNNING),
);
await delay(2500);
emitAssistantMessage(io, "Hello!");
io.client.emit(
"oh_event",
generateAgentStateChangeObservation(AgentState.AWAITING_USER_INPUT),
);
}
});
}),
];
|